require('strict')local fn = require('Module:Formatnum')local mm = require('Module:Math')local p =local pargs =local args=local data=
p.main = function(frame) -- called from template pargs = frame:getParent.args local output
if output then return frame:preprocess(output) end return p.errormsg("No valid options")endfunction p.getPattern(section) local pattern = '(.-)' return patternendfunction p.getPopulationData(frame)
local page = "List of countries by population (United Nations)" data['dates'] = data['total'] = data.total['latest'] = 0 data.total['previous'] = 0 data.total['projected'] = 0 --local total = 0 --local totalProjected = 0 local count = 0 local title = mw.title.new(page) --, ns) -- creates object if page doesn't exist (and valid page name) --TODO: could use mw.title.makeTitle, but that needs ns local content = title:getContent -- get dates for value in string.gmatch(content, p.getPattern("date_1")) do data.dates['latest']=value -- date of latest data end for value in string.gmatch(content, p.getPattern("date")) do data.dates['previous']=mw.getContentLanguage:formatDate('j F Y', value) end data.dates['today'] = mw.getContentLanguage:formatDate('j F Y') -- today's date (for formatting see https://www.mediawiki.org/wiki/Help:Extension:ParserFunctions#.23time) -- get population data for each country passes as parameter for k,v in pairs(args) do local country = mw.text.trim(v) -- get population data from section local section = country .. "_1" for value in string.gmatch(content, p.getPattern(section)) do
count=count+1 data[count] = data[count]['country'] = country data[count]['populationString'] = frame:preprocess(value) local raw = string.gsub(data[count]['populationString'], ",", "") -- strip formatting from number string data[count]['populationNumber'] = tonumber(raw) data.total['latest'] = data.total['latest'] + data[count]['populationNumber'] local section = country .. "_0" for value2 in string.gmatch(content, p.getPattern(section)) do data[count]['populationString2'] = frame:preprocess(value2) local raw = string.gsub(data[count]['populationString2'], ",", "") -- strip formatting from number string data[count]['populationNumber2'] = tonumber(raw) data.total['previous'] = data.total['previous'] + data[count]['populationNumber2'] data[count]['populationIncrement']=data[count]['populationNumber'] - data[count]['populationNumber2'] data[count]['populationGrowth'] =data[count]['populationIncrement']/data[count]['populationNumber2'] data[count]['populationDouble'] = p.getPopulationDoubleTime(data[count]['populationNumber'],data[count]['populationNumber2'])
data[count]['populationProjected'] = p.getPopulationProjection(data[count]['populationNumber'],data[count]['populationNumber2']) data.total['projected'] = data.total['projected'] + data[count]['populationProjected'] end
end end return trueend
-- estimate time to double population based on latest growth ratefunction p.getPopulationDoubleTime(latest,previous) local growth = (latest - previous) / previous local doubleTime = math.log(2) / math.log(1 + growth) return doubleTimeend
-- estimate today's population based on latest growth ratefunction p.getPopulationProjection(latest,previous) local ay = (mw.getCurrentFrame:callParserFunction - mw.getCurrentFrame:callParserFunction) /60/60/24/365.2425 -- number of years since first date until today local projected = math.pow(previous, 1 - ay) * math.pow(latest, ay) return projectedend
--sort rows by population (defaults to latest population) TODO add options for sorting columns function p.sortPopulationData(ByColumn) local sort_function = function(a,b) if (tonumber(a.populationNumber) > tonumber(b.populationNumber)) then -- primary sort on 'population' -> a before b return true end end table.sort(data, sort_function)end--Function to get flag icon and handle special cases 1. There is an issue of non-standard sizes when used with static rank column The three countries with extra height (and the required size parameter) are Nepal/NPL (size=12px), Switzerland/CHE (size=15px), Vatican/VAT (size=15px) a few have lower default heights so it doesn't matter (Poland, New Caledonia) 2. Alias, e.g. NEP->NPL, TRI->TTO function p.getFlagLabel(countryCode) local output local templateArgs = local size if countryCode
"VAT" then size="15px" elseif countryCode
-- simple version --output = mw.getCurrentFrame:expandTemplate -- method with fixed-height div and overflow output = '
' .. mw.getCurrentFrame:expandTemplate .. '
' return output end
--output table of data as Wikitext table function p.tabulateDataWikitext(frame)
local output local i = 1 -- output table output = '
| -class=wrap' output = output --headers (top row) .. '\n!rowspan=2 |
.. '\n!rowspan=2 | Country' .. '\n!rowspan=2 | Projected population (' .. data['dates']['today'] .. ')' .. '\n!rowspan=2 | Pct of total' .. '\n!colspan=2 | UN Population estimates' .. '\n!colspan=2 | Annual growth' .. '\n!rowspan=2 | Doubling time (years)' .. '\n | -' -- headers (second row) .. '\n!' .. data.dates['latest'] .. '\n | ' .. data.dates['previous'] .. '\n!Increment' .. '\n | Rate' while (data[i]) do -- add rows output = output .. '\n | -\n | ' .. i output = output .. '\n | ' .. frame:expandTemplate output = output .. '\n | ' .. mm._precision_format(data[i]['populationProjected'],0) output = output .. '\n | ' .. mm._precision_format(data[i]['populationProjected']/data.total['projected']*100,2) .. "%" -- projected output = output .. '\n | ' .. data[i]['populationString'] output = output .. '\n | ' .. data[i]['populationString2'] output = output .. '\n | ' .. mm._precision_format(data[i]['populationIncrement'],0) output = output .. '\n | ' .. mm._round(data[i]['populationGrowth']*100,2) .. "%" output = output .. '\n | ' .. mm._round(data[i]['populationDouble'],0) i=i+1 end local newcell = '\n! style="text-align:right;" | ' output = output .. '\n | -' -- totals row .. '\n! | Total' .. newcell .. fn.formatNum(mm._round(data.total['projected'],0),"en",0) .. newcell .. '100%' .. newcell .. fn.formatNum(data.total['latest'],"en",0) .. newcell .. fn.formatNum(data.total['previous'],"en",0) .. newcell .. fn.formatNum(data.total['latest']-data.total['previous'],"en",0) .. newcell .. fn.formatNum((data.total['latest']-data.total['previous'])/data.total['previous']*100,"en",2).."%" .. newcell .. mm._precision_format(p.getPopulationDoubleTime(data.total['latest'],data.total['previous']),0) output = output .. '\n |
|---|
--output table of data as use Lua HTML Libraryfunction p.tabulateData(frame) local hideYearsCols = frame.args['hide_years'] or false local doublingFootnote = frame.args['doubling_note'] or "" local growthFootnote = frame.args['growth_note'] or "" local i = 1 local static = mw.html.create('table'):addClass('wikitable') static:tag('tr'):tag('th'):attr('rowspan', 1):wikitext('
'):cssText('border-bottom-color:#eaecf0;') static:tag('tr'):tag('th'):wikitext('
'):cssText('border-top-color:#eaecf0;') while (data[i]) do -- add rows static:tag('tr'):tag('td'):wikitext(i) i=i+1 end static:tag('tr'):tag('th'):wikitext('
') local numRows=i-1
local tbl = mw.html.create('table'):addClass('wikitable') -- start table :addClass('sortable') :addClass('mw-datatable') :addClass('static-row-numbers') -- prefix with row numbers using css and templatestyles :addClass('nowrap') :css('text-align','right') local row = tbl:tag('tr') -- header row :addClass('static-row-header') -- for templatestyles in some skins --:tag('th'):attr('rowspan', 2):wikitext('#') row :tag('th'):attr('rowspan', 2):wikitext('Country') :tag('th'):attr('rowspan', 2):wikitext('Projected population
(' .. data['dates']['today'] .. ')') :tag('th'):attr('rowspan', 2):wikitext('Pct of
total') if not hideYearsCols then row :tag('th'):attr('colspan', 2):wikitext('UN Population estimates') end row :tag('th'):attr('colspan', 2):wikitext('Annual growth'..growthFootnote) :tag('th'):attr('rowspan', 2):wikitext('Doubling time
(years)'..doublingFootnote) row = tbl:tag('tr') -- headers (second row) :addClass('static-row-header') -- for templatestyles in some skins if not hideYearsCols then row :tag('th'):wikitext(data.dates['latest']) :tag('th'):wikitext(data.dates['previous']) end row :tag('th'):wikitext('Increment') :tag('th'):wikitext('Rate') i = 1 while (data[i]) do -- add country rows local row=tbl:tag('tr') --row :tag('td'):wikitext(i) row :tag('td'):cssText("text-align:left;") :wikitext(p.getFlagLabel(data[i]['country'])) :tag('td'):wikitext(mm._precision_format(data[i]['populationProjected'],0)) :tag('td'):wikitext(mm._precision_format(data[i]['populationProjected']/data.total['projected']*100,2) .. "%") -- % of projected if not hideYearsCols then row :tag('td'):wikitext(data[i]['populationString']) :tag('td'):wikitext(data[i]['populationString2']) end row :tag('td'):wikitext(mm._precision_format(data[i]['populationIncrement'],0)) :tag('td'):wikitext(mm._precision_format(data[i]['populationGrowth']*100,2) .. "%") :tag('td'):wikitext(mm._precision_format(data[i]['populationDouble'],0)) i=i+1 end local style = row = tbl:tag('tr') -- totals row :addClass('static-row-header') -- for templatestyles in some skins --:tag('th') :wikitext row :tag('th') :wikitext('Total') :tag('th'):css(style):wikitext(fn.formatNum(mm._round(data.total['projected'],0),"en",0)) :tag('th'):css(style):wikitext('100%') if not hideYearsCols then row :tag('th'):css(style):wikitext(fn.formatNum(data.total['latest'], "en",0)) :tag('th'):css(style):wikitext(fn.formatNum(data.total['previous'],"en",0)) end row :tag('th'):css(style):wikitext(fn.formatNum(data.total['latest'] - data.total['previous'],"en",0)) :tag('th'):css(style):wikitext(fn.formatNum((data.total['latest'] - data.total['previous']) / data.total['previous'] * 100,"en",2).."%") :tag('th'):css(style):wikitext(mm._precision_format(p.getPopulationDoubleTime(data.total['latest'],data.total['previous']),0)) -- return tostring(tbl) -- return table without row numbers -- use separate column for static row numbers -- return '
| ' .. tostring(static) .. '\n | ' .. tostring(tbl) .. '\n |
--currently the main entry function takes list of country codes gets population data from "List of countries by population (United Nations)" outputs sorted table function p.populations(frame) args = frame.args --TODO handle parent args for template local page = "List of countries by population (United Nations)"
local title = mw.title.new(page) --, ns) -- creates object if page doesn't exist (and valid page name) --TODO: could use mw.title.makeTitle, but that needs ns local output = "" if title and title.exists then local content = title:getContent
if not p.getPopulationData(frame) then return p.errormsg("Error retrieving data.") end p.sortPopulationData("latest") --output = p.tabulateDataWikitext(frame) -- version building table with Wikitext output = p.tabulateData(frame) -- version building table with mw.html library else return '
No page title found' endlocal test = "test: " local number=5435.12345 test= fn.formatNum(5435.12345,"en",0) --test= frame:expandTemplate --test=frame:callParserFunction return output --.. testend-- function for pie chartfunction p.piechart(frame) args = frame.args --TODO handle parent args for template local page = "List of countries by population (United Nations)"
local title = mw.title.new(page) --, ns) -- creates object if page doesn't exist (and valid page name) --TODO: could use mw.title.makeTitle, but that needs ns local output = "" if title and title.exists then local content = title:getContent
if not p.getPopulationData(frame) then return p.errormsg("Error retrieving data.") end p.sortPopulationData("latest") --output = p.tabulateDataWikitext(frame) -- version building table with Wikitext output = p.makePieChart(frame) -- version building table with mw.html library else return '
No page title found' endreturn output --.. testend
function p.makePieChart(frame) --local args=frame.args local templateArgs = templateArgs['caption'] = args['caption'] or "" --'South American population by country' --.. ' (top 8)' templateArgs['thumb'] = args['thumb'] or "right" templateArgs['other'] = args['other'] or nil local maxSlices = tonumber(args['slices']) -- nil if not a number if type(maxSlices) ~= "number" or maxSlices > 30 or maxSlices < 1 then maxSlices = 30 -- limit of template -- get number from data end
local i=1 while data[i] and i <= maxSlices do --templateArgs['label'..i] = data[i]['country'] templateArgs['label'..i] = mw.getCurrentFrame:expandTemplate templateArgs['value'..i] = mm._round(data[i]['populationNumber']/data.total['latest']*100,1) templateArgs['color'..i] = args['color'..i] or nil i=i+1 end
--caption= South American population by country (top 8)|other = yes|label1 = |value1 = |label2 = |value2 = |label3 = |value3 = |label4 = |value4 = |label5 = |value5 = |label6 = |value6 = |label7 = |value7 = |label8 = |value8 = }} local chart = mw.getCurrentFrame:expandTemplate return chart
end
function p.firstToUpper(str) return (str:gsub("^%l", string.upper))endp.errormsg = function (message) return '
' .. message .. '' end-- Test why was the sort being applied to the wrong level? Fixed function p.test(frame) -- meant test
local tbl = mw.html.create('table'):addClass('wikitable'):addClass('sortable'):addClass('mw-datatable') :css('text-align','right') tbl:tag('tr') -- header row :tag('th'):attr('rowspan', 2):wikitext('#') :tag('th'):attr('rowspan', 2):wikitext('A') :tag('th'):attr('rowspan', 2):wikitext('B') :tag('th'):attr('colspan', 2):wikitext('C+D'):addClass('unsortable') :tag('th'):attr('colspan', 2):wikitext('E+F'):addClass('unsortable') :tag('th'):attr('rowspan', 2):wikitext('G') tbl:tag('tr') -- headers (second row) :tag('th'):wikitext('C'):addClass('sortable') :tag('th'):wikitext('D'):addClass('sortable') :tag('th'):wikitext('E'):addClass('sortable') :tag('th'):wikitext('F') :addClass('sortable') local i = 1 while (i<5) do -- add rows tbl:tag('tr') :tag('td'):wikitext(i) :tag('td'):wikitext("A"..i) :tag('td'):wikitext("B"..i) :tag('td'):wikitext(tostring(math.fmod(5-i,2)) .. 'C' .. i) :tag('td'):wikitext("D"..i) :tag('td'):wikitext("E"..i) :tag('td'):wikitext(tostring(math.fmod(5-i,2)) .. 'F' .. i) :tag('td'):wikitext("G"..i) i=i+1 end local output = '
.. '\n | A' .. '\n | B' .. '\n | C+D' .. '\n | E+F' .. '\n | G' .. '\n | -' -- headers (second row) .. '\n!C' .. '\n | D' .. '\n!E' .. '\n | F' i=1 while (i<5) do -- add rows output = output .. '\n | -\n | ' .. i output = output .. '\n | A' .. i output = output .. '\n | B' .. i output = output .. '\n | ' .. tostring(math.fmod(5-i,2)) .. 'C' .. i output = output .. '\n | D' .. i output = output .. '\n | E' .. i output = output .. '\n | ' .. tostring(math.fmod(5-i,2)) .. 'F' .. i output = output .. '\n | G' .. i i=i+1 end output = output .. '\n | |||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
return output .. tostring(tbl)end
-- function for static rank column
function p.rank(frame)
--args = frame.args -- for module TODO allow invoke to work args = frame:getParent.args -- parent arguments for template args = frame.args -- invoke arguments for template local caption = args['caption'] local valign = args['valign'] or "top" local rowHeader = args['row-header'] local headerPadding = args['header-padding'] or "0px" local textAlign = args['text-align'] or "right" local style = args['style'] or "" local headerHeight = args['header-height'] or "" local headerLines = args['header-lines'] or 1 local headerText = args['header-text'] or "" local rows = tonumber(args['rows']) or 0 local rowHeader = args['row-header'] local rowHeight = args['row-height'] local marginRight = "0px" if rowHeader then marginRight = "-8px" end local headerValign = "bottom" if rowHeader then headerValign = "center" end -- copied from template; should be middle? local linebreaks = "" if headerLines then local i=0 while i
--|+| valign= |
| style=height: valign= | ]] local heightClass = "static-rank-col" if rowHeight and rowHeight "large" then heightClass = "static-rank-col-large" end local output = '\n
return p.templateStyle(frame, "Static column begin/styles.css") .. output endfunction p.templateStyle(frame, src) return frame:extensionTag('templatestyles', ,);endreturn p |
|---|