Module:Sandbox/Izno/Pie chart

Module documentation[create] [purge]

You might want to create a documentation page for this Scribunto module.
Editors can experiment in this module's sandbox (create | mirror) and testcases (create) pages.
Add categories to the /doc subpage. Subpages of this module.

local p = {}
local compress_sparse_array = require('Module:TableTools').compressSparseArray

local function arg_or_default(arg, default)
	if arg and mw.text.trim(arg) ~= '' then
		return arg
	else
		return default
	end
end

local incoming_args = {
	['other-color'] = 'other_color_default()',
	style = nil,
	thumb = nil,
	labeln = nil,
	valuen = nil,
	colorn = 'get_graph_color(n)',
	caption = nil,
	other = nil,
	footer = nil
}

local graph_colors = {
	'#1F78B4',
	'#33A02C',
	'#FF7F00',
	'#6A3D9A',
	'#B15928',
	'#A6CEE3',
	'#B2DF8A',
	'#FB9A99',
	'#FDBF6F',
	'#CAB2D6',
	'#FFFF99',
	'#FEEBE2',
	'#A9A9A9'
}
	
local function get_graph_color(nth_item)
	return graph_colors[math.max(math.min(nth_item, #graph_colors), 1)]
end

local function get_css_for_slices(args)
	local slices = {}
	local degree_ratio = 360/100

	for k, v in pairs(args) do
		if k:match('value%d+') then
			local number = k:find('(%d+)')
			slices[number] = { value = tonumber(v) * degree_ratio }
		end
	end
	
	-- second loop because we need to guarantee that we've found all the potential slices
	-- before attempting to add their labels and colors
	for k, v in pairs(args) do
		if k:match('label%d+') then
			local number = k:find('(%d)')
			if slices[number] then -- if we don't have a value, then no label set
				slices[number]['label'] = v
			end
		end
		if k:match('color%d+') then
			local number = k:find('(%d)')
			if slices[number] then -- if we don't have a value, then no color set
				slices[number]['color'] = v
			end
		end
	end
	
	compress_sparse_array(slices)
	local conical_css_for_slices = {}
	local accumulated_value = 0
	
	for i, slice in ipairs(slices) do
		local value = slice.value
		local color = slice.color or get_graph_color(i)
		local next_accumulated_value = accumulated_value + value
		
		if i == 1 then
			-- we only need the end degree for the first
			table.insert(conical_css_for_slices, color .. ' ' .. value .. 'deg,')
		elseif i > 1 and i < #slices then
			-- need both start and end for the first, start is the previous end
			table.insert(conical_css_for_slices, color .. ' ' .. accumulated_value .. 'deg ' .. next_accumulated_value .. 'deg,')
		elseif i == #slices then
			-- we only need the start degree for the last
			table.insert(conical_css_for_slices, color .. ' ' .. accumulated_value .. 'deg')
		end
		
		accumulated_value = next_accumulated_value
	end
	return 'background: conic-gradient(' .. table.concat(conical_css_for_slices) .. ')'
end

function p._main(args)
	
	local root = mw.html.create('div')
	root:addClass('pie-chart thumb')
		:cssText(arg_or_default(args.style, nil))
	
	local thumb = arg_or_default(args.thumb, nil)
	local thumb_class = 'tright'
	if thumb and (thumb == 'center' or thumb == 'none') then
		thumb_class = 'tnone'
	elseif thumb and thumb == 'left' then
		thumb_class = 'tleft'
	end
	root:addClass(thumb_class)
	
	local radius = tonumber(arg_or_default(args.radius, 100))
	local diameter = radius * 2
	root:tag('div')
		:addClass('pie-chart-inner thumbinner')
		:css('width', (diameter + 2) .. 'px')
	
	local chart = mw.html.create('div')
	chart:addClass('pie-chart-chart mw-no-invert')
	chart:css('border-radius', '50%') -- can be in pie chart/styles.css if we have one
	chart:css('width', diameter .. 'px')
	chart:css('height', diameter .. 'px')
	
	local conic_css = get_css_for_slices(args)
	chart:cssText(conic_css)

	if thumb and thumb == 'center' then
		local root_container = mw.html.create('div')
		root_container:addClass('pie-chart-container center')
			:node(root)
		return root_container
	end

	return root
	
end

function p.main(frame)
	return p._main(frame:getParent().args)
end

return p