Module:Piechart/sandbox

style="]]..style..[["

]]..aria..[[

>]]

return html

end

-- Render pie-item

-- (previous is a sum of previous values)

function priv.renderItem(previous, entry, options)

local value = entry.value

local label = entry.label

local bcolor = entry.bcolor

-- value too small to see

if (value < 0.03) then

mw.log('value too small', value, label)

mw.addWarning("pie chart: too small ↆ "..value.."% ("..label..")")

return ""

end

-- minimize transformation defects

if value < 10 then

if previous > 1 then

previous = previous - 0.01

end

value = value + 0.02

else

if previous > 1 then

previous = previous - 0.1

end

value = value + 0.2

end

-- force sum to be below 100% (needed due to value errors)

if previous + value > 100 then

if previous >= 100 then

mw.log('previous is already at 100', value, label)

return ""

end

value = 100 - previous

end

local html = ""

local size = ''

-- mw.logObject({'v,p,l', value, previous, label})

if (value >= 50) then

html = priv.sliceWithClass('pie50', 50, value, previous, bcolor, label)

elseif (value >= 25) then

html = priv.sliceWithClass('pie25', 25, value, previous, bcolor, label)

elseif (value >= 12.5) then

html = priv.sliceWithClass('pie12-5', 12.5, value, previous, bcolor, label)

elseif (value >= 7) then

html = priv.sliceWithClass('pie7', 7, value, previous, bcolor, label)

elseif (value >= 5) then

html = priv.sliceWithClass('pie5', 5, value, previous, bcolor, label)

else

-- 0-5%

local cutIndex = priv.round(value*10)

if cutIndex < 1 then

cutIndex = 1

end

local cut = p.cuts[cutIndex]

local transform = priv.rotation(previous)

html = priv.sliceX(cut, transform, bcolor, label)

end

-- mw.log(html)

return html

end

-- round to int

function priv.round(number)

return math.floor(number + 0.5)

end

-- render full slice with specific class

function priv.sliceWithClass(sizeClass, sizeStep, value, previous, bcolor, label)

local transform = priv.rotation(previous)

local html = ""

html = html .. priv.sliceBase(sizeClass, transform, bcolor, label)

-- mw.logObject({'sliceWithClass:', sizeClass, sizeStep, value, previous, bcolor, label})

if (value > sizeStep) then

local extra = value - sizeStep

transform = priv.rotation(previous + extra)

-- mw.logObject({'sliceWithClass; extra, transform', extra, transform})

html = html .. priv.sliceBase(sizeClass, transform, bcolor, label)

end

return html

end

-- render single slice

function priv.sliceBase(sizeClass, transform, bcolor, label)

local style = bcolor

if transform ~= "" then

style = style .. '; ' .. transform

end

return '\n\t

'

end

-- small slice cut to fluid size.

-- range in theory: 0 to 24.(9)% reaching 24.(9)% for cut = +inf

-- range in practice: 0 to 5%

function priv.sliceX(cut, transform, bcolor, label)

local path = 'clip-path: polygon(0% 0%, '..cut..'% 0%, 0 100%)'

return '\n\t

'

end

-- translate value to turn rotation (v=100 => 1.0turn)

function priv.rotation(value)

if (value > 0.001) then

local f = string.format("%.7f", value / 100)

f = f:gsub("(%d)0+$", "%1") -- remove trailing zeros

return "transform: rotate("..f.."turn)"

end

return ''

end

-- Language sensitive float, small numbers.

function priv.formatNum(value)

local lang = mw.language.getContentLanguage()

-- doesn't do precision :(

-- local v = lang:formatNum(value)

local v = ""

if (value < 10) then

v = string.format("%.2f", value)

else

v = string.format("%.1f", value)

end

if (lang:getCode() == 'pl') then

v = v:gsub("%.", ",")

end

return v

end

-- Format large values.

function priv.formatLargeNum(value)

local lang = mw.language.getContentLanguage()

-- add thusands separators

local v = lang:formatNum(value)

return v

end

-- Testing formatLargeNum.

-- p.__priv.test_formatLargeNum()

function priv.test_formatLargeNum()

mw.log("must not add fractional part")

mw.log( p.__priv.formatLargeNum(12) )

mw.log( p.__priv.formatLargeNum(123) )

mw.log("should preserve fractional part for small numbers")

mw.log( p.__priv.formatLargeNum(1.1) )

mw.log( p.__priv.formatLargeNum(1.12) )

mw.log( p.__priv.formatLargeNum(12.1) )

mw.log("can preserve long fractional part")

mw.log( p.__priv.formatLargeNum(1.1234) )

mw.log( p.__priv.formatLargeNum(1.12345) )

mw.log("should add separators above 1k")

mw.log( p.__priv.formatLargeNum(999) )

mw.log( p.__priv.formatLargeNum(1234) )

mw.log( p.__priv.formatLargeNum(12345) )

mw.log( p.__priv.formatLargeNum(123456) )

mw.log( p.__priv.formatLargeNum(1234567) )

mw.log("must handle large float, but might round values")

mw.log( p.__priv.formatLargeNum(1234.123) )

mw.log( p.__priv.formatLargeNum(12345.123) )

mw.log( p.__priv.formatLargeNum(123456.123) )

mw.log( p.__priv.formatLargeNum(1234567.123) )

end

--[[

Prepare final label.

Typical tpl:

"Abc: $v"

will result in:

"Abc: 23%" -- when values are percentages

"Abc: 1234 (23%)" -- when values are autoscaled

Advanced tpl:

"Abc: $d ($p)" -- only works with autoscale

]]

function priv.prepareLabel(tpl, entry)

-- static tpl

if tpl and not string.find(tpl, '$') then

return tpl

end

-- format % value without %

local p = priv.formatNum(entry.value)

-- default template

if not tpl then

tpl = "$v"

end

local label = ""

if entry.raw then

local d = priv.formatLargeNum(entry.raw)

label = tpl:gsub("%$p", p .. "%%"):gsub("%$d", d):gsub("%$v", d .. " (" .. p .. "%%)")

else

label = tpl:gsub("%$v", p .. "%%")

end

return label

end

-- default colors

-- source: https://colorbrewer2.org/#type=diverging&scheme=PRGn&n=6

local colorGroupSize = 3 -- must be at least 3

local colorGroups = 4

local colorPalette = {

-- green (from dark)

'#1b7837',

'#7fbf7b',

'#d9f0d3',

-- violet

'#762a83',

'#af8dc3',

'#e7d4e8',

-- red

'#d73027',

'#fc8d59',

'#fee090',

-- blue

'#4575b4',

'#91bfdb',

'#e0f3f8',

}

local lastColor = '#fff'

-- background color from entry or the default colors

function priv.backColor(entry, no, total)

if (type(entry.color) == "string") then

-- Remove unsafe characters from entry.color

local sanitizedColor = entry.color

:gsub('#', '#') -- workaround Module:Political_party issue reported on talk

:gsub("[^a-zA-Z0-9#%-]", "")

return 'background:' .. sanitizedColor

else

local color = priv.defaultColor(no, total)

return 'background:' .. color

end

end

-- color from the default colors

function priv.defaultColor(no, total)

local color = lastColor

if no <= 0 then

return color

end

local size = #colorPalette

if not total or total == 0 then

total = size + 1

end

local colorNo = priv.defaultColorNo(no, total, size)

if colorNo > 0 then

color = colorPalette[colorNo]

end

return color

end

-- gets color number from default colors

-- trys to return a light color as the last one

-- 0 means white-ish color should be used

function priv.defaultColorNo(no, total, size)

local color = 0 -- special, lastColor

if total == 1 then

color = 1

elseif total <= colorGroupSize * (colorGroups - 1) then

if no < total then

color = no

else

local groupIndex = ((no - 1) % colorGroupSize)

if groupIndex == 0 then -- dark

color = no+1

elseif groupIndex == 1 then -- med

color = no+1

else

color = no

end

end

elseif no < total then

color = ((no - 1) % size) + 1

end

return color

end

--[[

Testing defaultColorNo:

p.__priv.test_defaultColorNo(1, 12)

p.__priv.test_defaultColorNo(2, 12)

p.__priv.test_defaultColorNo(3, 12)

p.__priv.test_defaultColorNo(4, 12)

p.__priv.test_defaultColorNo(5, 12)

p.__priv.test_defaultColorNo(6, 12)

]]

function priv.test_defaultColorNo(total, size)

for no=1,total do

local color = priv.defaultColorNo(no, total, size)

mw.logObject({no=no, color=color})

end

end

--[[

trim string

note:

`(s:gsub(...))` returns only a string

`s:gsub(...)` returns a string and a number

]]

function priv.trim(s)

return (s:gsub("^%s+", ""):gsub("%s+$", ""))

end

--[[

Extract text from simple wikitext.

For now only works with links.

]]

-- Tests:

-- mw.log(p.extract_text("sweets: $v"))

-- mw.log(p.extract_text("sandwiches: $v"))

-- mw.log(p.extract_text("sandwich]]es: $v"))

-- mw.log(p.extract_text("sandwiches: $v"))

function p.extract_text(label)

label = label

-- replace links with pipe (e.g., sweets)

:gsub("%[%%+|(.-)%]%]", "%1")

-- replace simple links without pipe (e.g., sandwich)

:gsub("%[%[(.-)%]%]", "%1")

-- remove templates?

-- :gsub("{.-}", "")

-- remove tags

:gsub("<[^>]+>", "")

-- escape special chars just in case

:gsub("<", "<"):gsub(">", ">")

:gsub("'", "'"):gsub("\"", """)

return label

end

--[[

Parse classic template params into JSON.

From:

|label1=cookies: $v |value1=11 |color1=goldenrod

|label2=sweets: $v |value2=20 |color2=darkred

To:

{"value":11,"color":"goldenrod","label":"cookies: $v"},

{"value":20,"color":"darkred","label":"sweets: $v"},

]]

function p.parseEnumParams(frame)

local args = frame:getParent().args

local result = {}

local i = 1

local sum = 0.0

local hasCustomColor = false -- has last custom color

while args["value" .. i] do

-- value is required in this mode; it's also assumed to be 0..100

local entry = { value = tonumber(args["value" .. i]) or 0 }

-- label and color is optional

local label = args["label" .. i]

if label and label ~= "" then

entry.label = label

end

hasCustomColor = false

local color = args["color" .. i]

if color and color ~= "" then

entry.color = color

hasCustomColor = true

end

table.insert(result, entry)

sum = sum + entry.value

i = i + 1

end

-- re-loop to set values in labels

local willAutoscale = priv.willAutoscale(sum)

for _, entry in ipairs(result) do

local label = entry.label

if label and not label:find("%$v") then

-- autoscale will be forced, so use $v in labels

if willAutoscale then

entry.label = label .. " $v"

else

entry.label = label .. " (" .. entry.value .. "%)"

end

end

end

-- tracking data errors

priv.sumErrorTracking(sum, result)

-- support other value mapping

local lang = mw.language.getContentLanguage()

local langOther = "Other"

if (lang:getCode() == 'pl') then

langOther = "Inne"

end

local colorOther = "#FEFDFD" -- white-ish for custom colors for best chance and contrast

local otherValue = 100 - sum

if args["other"] and args["other"] ~= "" then

if otherValue < 0.001 then

otherValue = 0

end

local otherEntry = { label = (args["other-label"] or langOther) .. " ("..priv.formatNum(otherValue).."%)" }

if args["other-color"] and args["other-color"] ~= "" then

otherEntry.color = args["other-color"]

else

otherEntry.color = colorOther

end

table.insert(result, otherEntry)

elseif otherValue > 0.01 then

if hasCustomColor then

table.insert(result, {visible = false, label = langOther .. " ($v)", color = colorOther})

else

table.insert(result, {visible = false, label = langOther .. " ($v)"})

end

end

local jsonString = mw.text.jsonEncode(result)

return jsonString

end

-- Function to check if a value is true-ish

local trueValues = { ["true"] = true, ["1"] = true, ["on"] = true, ["yes"] = true }

function priv.isTrueishValue(value)

-- should return nil for empty args (i.e. undefined i.e. default)

if not value or value == "" then return nil end

value = priv.trim(value)

if value == "" then return nil end

-- other non-empty are false

return trueValues[value:lower()] or false

end

--[[

Parse classic template params into JSON with chart meta data.

]]

function p.parseMetaParams(frame)

local args = frame:getParent().args

local meta = {}

-- default meta for value1..n parameters

-- ...and for thumb right/left

local thumb = args["thumb"]

if args["value1"] or (thumb and (thumb == "right" or thumb == "left")) then

meta.size = 200

meta.legend = true

end

-- explicit meta param

if args["meta"] then

local decodeSuccess, tempMeta = pcall(function()

return mw.text.jsonDecode(args["meta"], mw.text.JSON_TRY_FIXING)

end)

if not decodeSuccess then

mw.log('invalid meta parameter')

else

meta = tempMeta

end

end

if args["size"] then meta.size = tonumber(args["size"]) end

if args["radius"] and tonumber(args["radius"]) then

meta.size = 2 * tonumber(args["radius"])

end

if args["autoscale"] then meta.autoscale = priv.isTrueishValue(args["autoscale"]) end

if args["legend"] then meta.legend = priv.isTrueishValue(args["legend"]) end

if args["ariahidechart"] then meta.ariahidechart = priv.isTrueishValue(args["ariahidechart"]) end

if args["direction"] and args["direction"] ~= "" then

meta.direction = args["direction"]:gsub("[^a-z0-9%-]", "")

end

if args["width"] and args["width"] ~= "" then

meta.width = args["width"]:gsub("[^a-z0-9%-]", "")

end

if args["caption"] and args["caption"] ~= "" then

meta.caption = args["caption"]

end

if args["footer"] and args["footer"] ~= "" then

meta.footer = args["footer"]

end

return mw.text.jsonEncode(meta)

end

return p