Модул:Демография

Документация за този модул може да бъде създадена на Модул:Демография/doc

local p = {}

local function toBgNum(n)
	n = tostring(n)
	n = n:gsub('^%d+', function(cap)
		if #cap > 4 then
			cap = string.reverse(string.gsub(string.reverse(cap), '(%d%d%d)', '%1 '))
		end
		return cap
	end):gsub('%.', ',')
	return mw.text.trim(n)
end

local function renderTimeline(t)
	local height = 25 * (#t + 2)
	local mval = 1
	local bardata = ''
	local plotdata1 = ''
	local plotdata2 = ''

	local y, v
	for i = 1, #t do
		y = t[i].year
		v = t[i].val
		mval = math.max(mval, v)
		bardata = bardata .. '  bar:' .. y .. ' text:' .. y .. '\n'
		plotdata1 = plotdata1 .. '  bar:' .. y .. ' from:0 till:' .. v .. '\n'
		plotdata2 = plotdata2 .. '  bar:' .. y .. ' at:' .. v .. ' fontsize:7 text:"' .. toBgNum(v) .. '" shift:(-12.5,5)\n'
	end

	-- some ugly code ahead :)
	local result = '\n'
			-- timeline color options
			.. 'Colors=' .. '\n'
  			.. ' id:a value:gray(0.9)' .. '\n'
			.. ' id:b value:gray(0.7)' .. '\n'
			.. ' id:c value:rgb(1,1,1)' .. '\n'
			.. ' id:d value:rgb(0.6,0.7,1)' .. '\n\n'
			-- variuos other timeline options
			.. 'ImageSize = width:auto barincrement:40 height:' .. height .. '\n'
			.. 'PlotArea = left:50 bottom:30 top:30 right:30' .. '\n'
			.. 'DateFormat = x.y' .. '\n'
			.. 'Period = from:0 till:' .. mval .. '\n'
			.. 'TimeAxis = orientation:vertical' .. '\n'
			.. 'AlignBars = justify' .. '\n\n'
			-- bar data
			.. 'BarData=' .. '\n'
			.. bardata .. '\n'
			-- first plot data
			.. 'PlotData=' .. '\n'
			.. ' color:d width:20 align:left' .. '\n'
			.. plotdata1 .. '\n'
			-- second plot data
			.. 'PlotData=' .. '\n'
			.. plotdata2
	-- end of result

	return result
end

local function renderTable(t, order, link, notimeline, align, timeline)
	if notimeline == '1' or notimeline == 'да' or notimeline == 'y' or notimeline == 'yes' or notimeline == 'true' then
		notimeline = true
	else
		notimeline = false
	end

	if order ~= '' then order = '<br><small>(в ' .. order .. ')</small>' end

	local position = {
		['left'] = 'floatleft',
		['вляво'] = 'floatleft',
		['л'] = 'floatleft',
		['ляво'] = 'floatleft',
		['right'] = 'floatright',
		['вдясно'] = 'floatright',
		['д'] = 'floatright',
		['дясно'] = 'floatright',
		['center'] = 'centered',
		['ц'] = 'centered',
		['център'] = 'centered',
	}

	local class
	if align ~= '' and position[align] then
		class = 'wikitable ' .. position[align]
	else
		class = 'wikitable'
	end

	local tablebegin, tableend
	local tablecontent = ''
	local rowstyle = ''
	local style = ''

	-- some ugly code ahead :)
	if notimeline then -- notimeline option is true, so plain sortable table
		tablebegin = '<table class="' .. class .. ' sortable" style="text-align:center; white-space:nowrap">\n'
			.. '<tr><th>Година на<br>преброяване</th><th>Численост' .. order.. '</th></tr>\n'
		-- end of tablebegin
		tableend = '</table>'
	else -- here a timeline have to be added, so there has to be 2 tables - the sortable one and the outter table which holds both the inner table and the timeline
		-- explicitly add a style because the inner table has no border
		rowstyle = ' style="border-top:1px #aaa solid"'
		style = ' style="border-right:1px #aaa solid"'

		tablebegin = '<table class="' .. class .. '">\n' -- outer table
			.. '<tr><td style="padding:0">\n' -- cell that will hold the inner table; no padding for the inner table
			.. '<table class="wikitable sortable" style="margin:0; border:0; text-align:center; white-space:nowrap">\n' -- sortable inner table with removed margins and border
			.. '<tr><th' .. style .. '>Година на<br>преброяване</th><th>Численост' .. order.. '</th></tr>\n'
		-- end of tablebegin
		tableend = '</table>\n' -- closing of the inner table
			.. '</td><td style="text-align:center">' .. timeline .. '</td></tr>\n' -- timeline cell
			.. '</table>' -- closing of the outer table
		-- end of tableend
	end

	local y, v, attryear, attrnum
	for i = 1, #t do
		y = t[i].year
		v = t[i].val
		attryear = ' data-sort-value="' .. y .. '"' .. style
		attrnum = ' data-sort-value="' .. v .. '" style="text-align:right"'
		if link ~= '' then
			if link ~= '0' and link ~= 'не' and link ~= 'без' and link ~= 'no' then
				y = '[[' .. mw.ustring.gsub(link, '%%година%%', tostring(y)) .. '|' .. y .. ']]'
			end
		else
			y = '[[' .. y .. ']]'
		end
		tablecontent = tablecontent .. '<tr' .. rowstyle .. '><td' .. attryear .. '>' .. y .. '</td><td' .. attrnum .. '>' .. toBgNum(v) .. '</td></tr>\n'
	end

	return tablebegin .. tablecontent .. tableend
end

function p.main(frame)
	local args = frame.args
	local pargs = frame:getParent().args

	local order = args['численост-порядък'] or pargs['численост-порядък'] or ''
	local link = args['препратка'] or pargs['препратка'] or ''
	local notimeline = args['без-графика'] or pargs['без-графика'] or ''
	local align = args['подравняване'] or pargs['подравняване'] or ''
	local stats = {}
	local ref
	local result = ''

	for k, v in pairs(pargs) do
		k = tonumber(k)
		if mw.ustring.match(tostring(k), '^%d%d%d%d$') and (k >= 1650 and k <= os.date('*t').year) then
			v = mw.text.killMarkers(v)
			v = mw.ustring.gsub(v, '&.-;', '') -- remove any HTML entities
			v = mw.ustring.gsub(v, '%s', '') -- remove any whitespace characters
			v = mw.ustring.gsub(v, ',', '.') -- replace commas with dots
			v = mw.ustring.gsub(v, '[^.%d]', '') -- remove any character that isn't a dot or a digit
			local _, count = mw.ustring.gsub(v, '%.', '')
			if count > 1 then v = mw.ustring.gsub(v, '%.', '', count - 1) end
			v = tonumber(v) -- if unsuccessful then it will be nil; the next "if" will be skipped
			if v then
				table.insert(stats, {year = k, val = v})
			end
		end
	end

	if #stats > 0 then
		table.sort(stats, function(a, b) return a.year < b.year end) -- sort in ascending order (years)
		local timeline
		if mw.isSubsting() then
			timeline = '<timeline>' .. renderTimeline(stats) .. '</timeline>'
		else
			timeline = frame:extensionTag('timeline', renderTimeline(stats))
		end
		result = renderTable(stats, order, link, notimeline, align, timeline)
	end

	return result
end

return p