Benutzer:Lövberg/Spielwiese/Modul:DiagrammEinwohnerentwicklung

p = {}

function p._bargraph(data1, data2)

   data = {data1, data2}
   
   -- Daten überprüfen und zusammenstellen
   bardata = {}
   
   is_stacked = false
   is_first_series = true

   years = {}

   for _, series in ipairs(data) do
       entries = mw.text.split(series, '/')
       for _, entry in ipairs(entries) do
           entrydata =  mw.text.split(entry , '=')
           if #entrydata == 2 then
               year = entrydata[1]
               value = tonumber(entrydata[2])
               if #entrydata == 2 and tonumber(year) ~= nil and value ~= nil then
                   is_new_year = true
                   if is_first_series then
                       bardata[year] = {value, -1}
                   else
                       is_stacked = true
                       barentry = bardata[year]
                       if barentry then
                           bardata[year][2] = value
                           is_new_year = false
                       else
                           bardata[year] = {-1, value}
                       end
                   end
                   if is_new_year then
                        years[#years + 1] = year
                   end
               else
                   -- TODO: error
               end
           else
               -- TODO: error
           end
       end
       is_first_series = false
   end

   table.sort(years)
   
   graphdef = {'Colors=',
               '  id:minor value:gray(0.9)',
               '  id:major value:gray(0.7)',
               '  id:bar value:rgb(0.6,0.7,0.8)',
               '%s',
               'ImageSize  = width:%d height:250',
               'PlotArea = left:50 bottom:30 top:30 right:30',
               'TimeAxis = orientation:vertical',
               'AlignBars = justify',
               'Period = from:0 till:%d',
               'ScaleMajor = gridcolor:major start:0 increment:%d',
               'ScaleMinor = gridcolor:minor start:0 increment:%d'}

   max_value = 0
   barcount = 0

   for _, year in ipairs(years) do
       values = bardata[year]

       prev_value = 0
       sum = 0
      
       is_first_value = true
       textPos = values[2] > 0 and -12 or 3
       
       for _, value in ipairs(values) do
           if value > 0 then
               if is_first_value then
                    barcount = barcount + 1

                   graphdef[#graphdef + 1] = 'BarData='
                   graphdef[#graphdef + 1] = mw.ustring.format('  bar:%d text:%d', year, year)
               end

               sum = sum + value
               
               graphdef[#graphdef + 1] = 'PlotData='
               graphdef[#graphdef + 1] = mw.ustring.format('  color:%s width:30 align:center', is_first_value and 'bar' or 'bar2')
               graphdef[#graphdef + 1] = mw.ustring.format('  bar:%d from:%d till:%d', year, prev_value, sum)
               graphdef[#graphdef + 1] = mw.ustring.format('  bar:%d at:%d text:%d shift:(1,%d)', year, sum, value, textPos)
               if not is_first_value then
                   graphdef[#graphdef + 1] = mw.ustring.format('  bar:%d at:%d text:%d shift:(1,3)', year, sum, sum)
               end
               
               prev_value = value
           end
           is_first_value = false
       end
       max_value = math.max(max_value, sum)
   end

   -- Keine (gültigen) Daten
   if barcount == 0 then
       return nil
   end

   scale = {200000, 50000}
   if max_value <= 100 then
      scale = {10, 5}
   elseif max_value <= 200 then
      scale = {20, 5}
   elseif max_value <= 500 then
      scale = {50, 10}
   elseif max_value <= 1000 then
      scale = {100, 25}
   elseif max_value <= 2000 then
      scale = {200, 50}
   elseif max_value <= 5000 then
      scale = {500, 100}
   elseif max_value <= 10000 then
      scale = {1000, 250}
   elseif max_value <= 20000 then
      scale = {2000, 500}
   elseif max_value <= 50000 then
      scale = {5000, 1000}
   elseif max_value <= 100000 then
      scale = {10000, 2500}
   elseif max_value <= 200000 then
      scale = {20000, 5000}
   elseif max_value <= 1000000 then
      scale = {100000, 25000}
   end

   max_value = math.ceil(max_value * 1.1 / scale[1]) * scale[1]

   return mw.ustring.format(table.concat(graphdef, '\n'), is_stacked and '  id:bar2 value:rgb(0.7,0.8,0.9)' or , barcount * 40 + 70, max_value, scale[1], scale[2])

end

function p.bargraph(frame)

   result = p._bargraph(frame.args[1], frame.args[2])
   if result ~= nil then
       return frame:extensionTag('timeline', result)
   else
       return ''
   end

end

return p