Modulo:Demografia

local mWikidata = require('Modulo:Wikidata')

local p = {}

local function dump(t, ...)
    local args = {...}
    for _, s in ipairs(args) do
        table.insert(t, s)
    end
end

local function max_array(array)
    local max = 0
    for _,p in ipairs(array) do
        if p > max then max = p end
    end
    return max
end

-- Converte una stringa di valori numerici separati da "," in una lista
-- Eventuali valori non numerici sono ignorati
local function read_array(args)
    local years = {}
    local populations =  {}
    if args.anni then
        local list = mw.text.split(string.gsub(args.anni, "%s", ""), ",")
        for _,s in ipairs(list) do
            years[#years+1] = tonumber(s) or 0
        end
        if args.popolazione then
            local list = mw.text.split(string.gsub(args.popolazione, "%s", ""), ",")
            for _,s in ipairs(list) do
                populations[#populations+1] = tonumber(s) or 0
            end
        end
    else -- vecchio metodo di inserimento
        for i=1,20 do
            local index = tonumber(i)
            local year = tonumber(args['a' .. index])
            if year then
                years[#years+1] = year
                populations[#populations+1] = tonumber(args['p' .. index]) or 0
            end
        end
    end
    return years, populations
end

local function get_wikidata(from)
    local populations_raw = mWikidata._getProperty({'P1082', from=from, rank='normal'}, true)
    local populations_reverse = {}
    local years = {}
    for i =1, #populations_raw do -- per il momento inserisco come anni una sequenza numerica
        populations_reverse[i] = populations_raw[#populations_raw-i+1]
        years[i] = i
    end
    return years,populations_reverse
end

local function default_table(popmax)
    local default = {}
    local log_p = math.ceil(math.log10(popmax))
    if log_p == 0 then log_p = 1 end
    local first_digit = math.floor(popmax / 10 ^ (log_p - 1))
    if first_digit == 9 then
        default.popmax = 10 ^ (log_p)
    else
        default.popmax = (first_digit +1 ) * 10 ^ (log_p-1)
    end
    if log_p < 2 then log_p = 2 end
    if first_digit <=5 then
        default.passo1 = (10 ^ (log_p-1)) / 2
        default.passo2 = 10 ^ (log_p-2)
    else
        default.passo1 = (10 ^ (log_p-1))
        default.passo2 = 2 * 10 ^ (log_p-2)     
    end
    return default
end

function p._demografia(args)
    local years,populations = read_array(args)
    local from_wikidata = false
    if #years == 0 then
        years,populations = get_wikidata(args.from)
        from_wikidata = true
    end
    if #years == 0 then return '' end
    local popmax = max_array(populations)
    local is_thousand = false
    if from_wikidata and popmax >= 900000 then
        popmax = popmax / 1000
        for index,population in ipairs(populations) do
            populations[index] = population / 1000
            is_thousand = true
        end
    end
    -- calcolo default per la popolazione
    local default = default_table(popmax)
    -- generazione del grafico
    local graph = {}
    dump(graph, 'Colors=\n id:lightgrey value:gray(0.9)\n id:darkgrey  value:gray(0.7)\n id:sfondo value:rgb(1,1,1)\n id:barra value:rgb(0.6,0.7,0.8)\n\n')
    dump(graph, 'ImageSize = width:', args.dimx or '488', ' height:', args.dmy or '373', '\n')
    dump(graph, 'PlotArea   = left:50 bottom:50 top:30 right:30\nDateFormat = x.y\n')
    dump(graph, 'Period = from:0 till:', args.popmax or tostring(default.popmax), '\n')
    dump(graph, 'TimeAxis   = orientation:vertical\nAlignBars  = justify\n')
    dump(graph, 'ScaleMajor = gridcolor:darkgrey increment:', args.passo1 or tostring(default.passo1), ' start:0\n' )
    dump(graph, 'ScaleMinor = gridcolor:lightgrey increment:', args.passo2 or tostring(default.passo2), ' start:0\n')
    dump(graph, 'BackgroundColors = canvas:sfondo\n')
    dump(graph, 'BarData=\n')
    for _,year in ipairs(years) do
        dump(graph, string.format(' bar: %d text:%d\n', year, year))
    end
    dump(graph, '\n')
    dump(graph, 'PlotData=\n color:barra width:20 align:left\n\n')
    for i =1,#years do
        dump(graph, string.format(' bar:%d from:0 till:%d\n', years[i], populations[i]))
    end
    dump(graph, '\n')
    local lang = mw.language.new('it')
    dump(graph, 'PlotData=\n')
    for i=1,#years do
        local p_formatted = lang:formatNum(math.floor(tonumber(populations[i]) + 0,5) )
        dump(graph, string.format(' bar: %d at: %d fontsize:S text: %s shift:(-10,5)|n.d. shift:(-8,5)\n',
                                    years[i], populations[i], p_formatted))
    end
    dump(graph, '\n')
    if args.fonte then
        dump(graph, 'TextData=\n fontsize:S pos:(20,20)\n text:fonte ', args.fonte, '\n')
    end
    local draw_graph
    if args.raw then
        draw_graph = '<nowiki>'.. table.concat(graph) .. '</nowiki>'
    else
        draw_graph = mw.getCurrentFrame():extensionTag('timeline', table.concat(graph))
    end
    local titolo = "''Abitanti censiti''\n"
    if is_thousand then
        titolo =  "''Abitanti censiti (migliaia)''\n"
    end
    if args.titolo then
        titolo = "''" .. args.titolo .. "''\n"
    end
    return titolo .. draw_graph
end

function p.demografia(frame)
    local args = require('Modulo:Arguments').getArgs(frame)
    return p._demografia(args)
end

return p