Jump to content

Module:Weather/sandbox

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by Erutuon (talk | contribs) at 04:31, 7 October 2016 (using single quotes in certain cases to avoid having to escape double quotes). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.

export = {}
degree = "°" -- used by add_unitNames()
minus = "−" -- used by makeRow() and makeTable()
thinSpace = mw.ustring.char(0x2009) -- used by makeCell()

-- Error message handling
message = ""

local function addMessage(new_message)
	if show then
		if checkForString(message) then
			message = message .. " " .. new_message
		else
			message = "Notices: " .. new_message
		end
	end
end

-- Input and output parameters
local function getFormat (frame)
	local inputParameter = frame.args.input
	local outputParameter = frame.args.output
	
	if inputParameter == nil then
		error('Please provide the number of values and a unit in the input parameter')
	else
		length = tonumber(string.match(inputParameter, "(%d+)")) -- Find digits in the input parameter.
		inputUnit = string.match(inputParameter, "([CF])") -- C or F
		if string.find(inputParameter, "[^CF%d%s]") then
			addMessage('There are extraneous characters in the <span style="background-color: #EEE; font-family: monospace;">output</span> parameter.')
		end
	end
	
	if inputUnit == "C" then
		outputUnit = "F"
	elseif inputUnit == "F" then
		outputUnit = "C"
	else
		error ("Please provide an input unit in the input parameter: F for Fahrenheit or C for Celsius", 0)
	end
	
	if length == nil then
		error ("getFormat has not found a length value in the input parameter")
	end
	
	if outputParameter == nil then
		addMessage('No output format has been provided in the <span style="background-color: #EEE; font-family: monospace;">output</span> parameter.')
	else
		cellFormat = {}
		local n = 1
		for unit in outputParameter:gmatch("[CF]") do
			cellFormat[n] = unit
			n = n + 1
			if n > 2 then
				break
			end
		end
		local function setFormat(key, formatVariable, formatValue1, formatValue2)
			if string.find(outputParameter, key) then
				cellFormat[formatVariable] = formatValue1
			else
				cellFormat[formatVariable] = formatValue2
			end
		end
		if cellFormat[1] then
			cellFormat.first = cellFormat[1]
		else
			error('C or F not found in output parameter')
		end
		if cellFormat[2] == nil then
			cellFormat["convertUnits"] = "no"
		else
			if cellFormat[2] == cellFormat[1] then
				error('There should not be two of the same unit name in the output parameter.')
			else
				cellFormat["convertUnits"] = "yes"
			end
		end
		setFormat("unit", "unitNames", "yes", "no")
		setFormat("no ?color", "color", "no", "yes")
		setFormat("sort", "sortable", "yes", "no")
		setFormat("full ?size", "smallFont", "no", "yes")
		setFormat("no ?brackets", "brackets", "no", "yes")
		setFormat("round", "decimals", "0", "")
		if string.find(outputParameter, "line break") then
			cellFormat["lineBreak"] = "yes"
		elseif string.find(outputParameter, "one line") then
			cellFormat["lineBreak"] = "no"
		else
			cellFormat["lineBreak"] = "auto"
		end
		if string.find(outputParameter, "one line") and string.find(outputParameter, "line break") then
			error('Place either "one line" or "line break" in the output parameter, not both')
		end
	end
	if frame.args.palette == nil then
		palette = "cool2avg"
	else
		palette = frame.args.palette
	end
	
	if frame.args.messages == "show" then
		show = true
	else
		show = false
	end
	
	return length, inputUnit, outputUnit
end

-- Number and string-handling functions
local function checkForNumber(value)
	return type(tonumber(value)) == "number"
end

function checkForString(string)
	string = tostring(string)
	return string ~= "" and string ~= nil
end

local function round(value, decimals)
	value = tonumber(value)
	if type(value) == "number" then
		local string = string.format("%." .. decimals .. "f", value)
		return string
	elseif value == nil then
		value = "nil"
		addMessage("Format was asked to operate on " .. value .. ", which cannot be converted to a number.", 2)
		return ""
	end
end

local function convert(value, decimals, unit) -- Unit is the unit being converted from. It defaults to inputUnit.
	if not unit then
		unit = inputUnit
	end
	if checkForNumber(value) then
		local value = tonumber(value)
		if unit == "C" then
			addMessage(value .. " " .. degree .. unit .. " was converted.")
			return round(value * 9/5 + 32, decimals)
		elseif unit == "F" then
			addMessage(value .. " " .. degree .. unit .. " was converted.")
			return round((value - 32) * 5/9, decimals)
		else
			error("Input unit not recognized", 2)
		end
	else
		return "" -- Setting result to empty string if value is not a number avoids concatenation errors.
	end
end

-- Input parsing
function makeArray(parameter, array, frame)
	local array = {}
	local n = 1
	for number in parameter:gmatch("%-?%d+%.?%d?") do
		local number = number
		if n == 1 then
			local decimals = number:match("%.(%d+)")
			if decimals == nil then
				precision = "0"
			else
				precision = #decimals
			end
		end
		table.insert(array, n, number)
		n = n + 1
		if n > length then
			break
		end
	end
	if not array[length] then
		addMessage('There are not ' .. length .. ' values in the ' .. parameter .. ' parameter.')
	end
	return array, precision
end

function makeArrays(frame)
	getFormat(frame)
	local parameter_a = frame.args.a
	local parameter_b = frame.args.b
	local parameter_c = frame.args.c
	if parameter_a then
		a = makeArray(parameter_a, a, frame)
	else
		error('Please provide a set of numbers in parameter a')
	end
	if parameter_b then
		b = makeArray(parameter_b, b, frame)
	else
		addMessage('There is no content in parameter <span style="background-color: #EEE; font-family: monospace;">b</span>.')
	end
	if parameter_c then
		c = makeArray(parameter_c, c, frame)
	else
		addMessage('There is no content in parameter <span style="background-color: #EEE; font-family: monospace;">c</span>.')
	end
	return a, b, c
end

-- Color generation

palettes = {
	-- The first three arrays in each palette defines background color using a table of four numbers,
	-- say { 11, 22, 33, 44 } (values in °C).
	-- That means the color is 0 below 11 and above 44, and is 255 from 22 to 33.
	-- The color rises from 0 to 255 between 11 and 22, and falls between 33 and 44.
	cool = {
		{ -42.75,   4.47, 41.5, 60   },
		{ -42.75,   4.47,  4.5, 41.5 },
		{ -90   , -42.78,  4.5, 23   },
		white = { -23.3, 37.8 },
	},
	cool2 = {
		{ -42.75,   4.5 , 41.5, 56   },
		{ -42.75,   4.5 ,  4.5, 41.5 },
		{ -90   , -42.78,  4.5, 23   },
		white = { -23.3, 35 },
	},
	cool2avg = {
		{ -38,   4.5, 25  , 45   },
		{ -38,   4.5,  4.5, 30   },
		{ -70, -38  ,  4.5, 23   },
		white = { -23.3, 25 },
	},
}

local function temperature_color(palette, value, outRGB)
	--[[ Return style for a table cell based on the given value which
		should be a temperature in °C. ]]
	local backgroundColor, textColor
	value = tonumber(value)
	if value == nil then
		backgroundColor, textColor = 'FFF', '000'
		addMessage('Value supplied to <span style="background-color: #EEE; font-family: monospace;">temperature_color</span> is not recognized.')
	else
		local min, max = unpack(palette.white or { -23, 35 })
		if value < min or value >= max then
			textColor = 'FFF'
		else
			textColor = '' -- This assumes that black text color is the default for most readers.
		end

		local backgroundRGB = outRGB or {}
		for i, v in ipairs(palette) do
			local a, b, c, d = unpack(v)
			if value <= a then
				backgroundRGB[i] = 0
			elseif value < b then
				backgroundRGB[i] = (value - a) * 255 / (b - a)
			elseif value <= c then
				backgroundRGB[i] = 255
			elseif value < d then
				backgroundRGB[i] = 255 - ( (value - c) * 255 / (d - c) )
			else
				backgroundRGB[i] = 0
			end
		end
		backgroundColor = string.format('%02X%02X%02X', backgroundRGB[1], backgroundRGB[2], backgroundRGB[3])
	end
	if textColor == "" then
		return backgroundColor
	else
		return backgroundColor, textColor
	end
end

local function colorCSS(backgroundColor, textColor)
	if backgroundColor and textColor then
		return 'background: #' .. backgroundColor .. '; color: #' .. textColor .. ';'
	elseif backgroundColor then
		return 'background: #' .. backgroundColor .. ';'
	else
		return ''
	end
end

local function temperatureColorCSS(palette, value, outRGB)
	return colorCSS(temperature_color(palette, value, outRGB))
end

function temperatureCSS(value, unit, palette)
	local palette = palettes[palette] or palettes.cool
	local value = tonumber(value)
	if value == nil then
		error('The function <span style="background-color: #EEE; font-family: monospace;">temperatureCSS</span> is receiving a nil value')
	else
		if unit == 'C' then
			return colorCSS(temperature_color(palette, value))
		elseif unit == 'F' then
			return colorCSS(temperature_color(palette, convert(value, decimals, 'F')))
		else
			unit_error(unit or "nil")
		end
	end
end

local function styleAttribute(palette, value, outRGB)
	local font_size = "font-size: 85%;"
	local color = temperatureColorCSS(palette, value, outRGB)
	return 'style=\"' .. color .. ' ' .. font_size .. '\"'
end

style_attribute = styleAttribute

function export.temperatureStyle(frame) -- used by Template:Average temperature table/color
	local palette = palettes[frame.args.palette] or palettes.cool
	local unit = frame.args.unit or 'C'
	local value = tonumber(frame.args[1])
	if unit == 'C' then
		return styleAttribute(palette, value)
	elseif unit == 'F' then
		return styleAttribute(palette, convert(value, 1, 'F'))
	else
		unit_error(unit)
	end
end

export.temperature_style = export.temperatureStyle

--[[ ==== Cell, row, table generation ==== ]]
local output_formats = {
	high_low_average_F =
		{ first = "F",
		convertUnits = "yes",
		unitNames = "no",
		color = "yes",
		smallFont = "yes",
		sortable = "yes",
		decimals = "0",
		brackets = "yes",
		lineBreak = "auto", },
	high_low_average_C =
		{ first = "C",
		convertUnits = "yes",
		unitNames = "no",
		color = "yes",
		smallFont = "yes",
		sortable = "yes",
		decimals = "0",
		brackets = "yes",
		lineBreak = "auto", },
	high_low_F =
		{ first = "F",
		convertUnits = "yes",
		unitNames = "no",
		color = "no",
		smallFont = "yes",
		sortable = "no",
		decimals = "",
		brackets = "yes",
		lineBreak = "auto", },
	high_low_C =
		{ first = "C",
		convertUnits = "yes",
		unitNames = "no",
		color = "no",
		smallFont = "yes",
		sortable = "no",
		decimals = "0",
		brackets = "yes",
		lineBreak = "auto", },
	average_F =
		{ first = "F",
		convertUnits = "yes",
		unitNames = "no",
		color = "yes",
		smallFont = "yes",
		sortable = "no",
		decimals = "0",
		brackets = "yes",
		lineBreak = "auto", },
	average_C =
		{ first = "C",
		convertUnits = "yes",
		unitNames = "no",
		color = "yes",
		smallFont = "yes",
		sortable = "no",
		decimals = "0",
		brackets = "yes",
		lineBreak = "auto", },
	}

local function add_unitNames(value, unit)
	if not unit then unit = inputUnit end
	-- Don't add a unit name to an empty string
	value = output_format.unitNames == "yes" and checkForString(value) and value .. "&nbsp;" .. degree .. unit or value
	return value
end

local function ifYes(parameter, realization1, realization2)
	if realization1 then
		if realization2 then
			result = parameter == "yes" and { realization1, realization2 } or { "", "" }
		else
			result = parameter == "yes" and realization1 or ""
		end
	else
		result = ""
		addMessage('<span style="background-color: #EEE; font-family: monospace;">ifYes</span> needs at least one realization')
	end
	return result
end

function makeCell(output_format, a, b, c)
	local cell, cell_content = "", ""
	local colorCSS, other_CSS, title_attribute, sortkey, attribute_separator, converted_units_separator = "", "", "", "", "", "", ""
	local styleAttribute, high_low_separator, brackets, values, converted_units = {"", ""}, {"", ""}, {"", ""}, {"", ""}, {"", ""}
	-- Distinguish styleAttribute variable from styleAttribute function above.
	decimals = ( checkForNumber(output_format.decimals) and output_format.decimals ) or precision
		--[[ Precision is the number of decimals in the first number of the last array.
			This may be a problem for data from Weatherbase,
			which seems to inappropriately remove .0 from numbers that have it. ]]
	
	if checkForNumber(b) and checkForNumber(a) then
		values, high_low_separator = { round(a, decimals), round(b, decimals) }, { thinSpace .. "/" .. thinSpace, ifYes(output_format.convertUnits, thinSpace .. "/" .. thinSpace) }
	elseif checkForNumber(a) then
		values = { round(a, decimals), "" }
	elseif checkForNumber(c) then
		values = { round(c, decimals), "" }
	end
	if output_format.first == inputUnit then
		if output_format.convertUnits == "yes" then
			converted_units = { add_unitNames(convert(values[1], decimals), outputUnit), add_unitNames(convert(values[2], decimals), outputUnit) }
		end
		values = { add_unitNames(values[1]), add_unitNames(values[2]) }
	elseif output_format.first == "C" or output_format.first == "F" then
		if output_format.convertUnits == "yes" then
			converted_units = { add_unitNames(values[1]), add_unitNames(values[2]) }
		end
		values = { add_unitNames(convert(values[1], decimals), outputUnit), add_unitNames(convert(values[2], decimals), outputUnit) }
	else
		if output_format.first == nil then
			output_format.first = "nil"
		end
		addMessage('<span style="background-color: #EEE; font-family: monospace;">' .. output_format.first .. '</span>, the value for <span style="background-color: #EEE; font-family: monospace;">first</span> in <span style="background-color: #EEE; font-family: monospace;">output_format</span> is not recognized.')
	end
	--[[
		Regarding line breaks:
		If there are two values, there will be at least three characters: 9/1.
		If there is one decimal, numbers will be three to five characters long
		and there will be 3 to 10 characters total even without unit conversion:
			1.1, 116.5/88.0.
		If there are units, that adds three characters per number: 25 °C/20 °C.
		In each of these cases, a line break is needed so that table cells are not too wide;
		even more so when more than one of these things are true.
		]]
	if output_format.convertUnits == "yes" then
		brackets = output_format.brackets == "yes" and { "(", ")" } or { "", "" }
		if output_format.lineBreak == "auto" then
			converted_units_separator = ( checkForString(values[2]) or decimals ~= "0" or output_format.show_units == "yes" ) and "<br>" or "&nbsp;"
		else
			converted_units_separator = output_format.lineBreak == "yes" and "<br>" or output_format.lineBreak == "no" and "&nbsp;" or error('Value for lineBreak not recognized')
		end
	end
	
	cell_content = values[1] .. high_low_separator[1] .. values[2] .. converted_units_separator .. brackets[1] .. converted_units[1] .. high_low_separator[2] .. converted_units[2] .. brackets[2]
	
	if checkForNumber(c) then
		colorCSS = output_format.color == "yes" and temperatureCSS(c, inputUnit, palette) or ""
		if checkForNumber(b) and checkForNumber(a) then
			attribute_value = output_format.first == inputUnit and c or convert(c, decimals)
			sortkey = output_format.sortable == "yes" and " data-sort-value=\"" .. attribute_value .. "\"" or ""
			title_attribute = " title=\"Average temperature: " .. attribute_value .. " " .. degree .. output_format.first .. "\""
		end
	elseif checkForNumber(b) then
		colorCSS = ""
	elseif checkForNumber(a) then
		colorCSS = output_format.color == "yes" and temperatureCSS(a, inputUnit, palette) or ""
	else
		addMessage('Neither a nor b nor c are strings.')
	end
	other_CSS = output_format.smallFont == "yes" and "font-size: 85%;" or ""
	if checkForString(colorCSS) or checkForString(other_CSS) then
		styleAttribute = { "style=\"", "\"" }
	end
	
	if checkForString(other_CSS) or checkForString(colorCSS) or checkForString(title_attribute) or checkForString(sortkey) then
		attribute_separator = " | "
	end
	cell = "\n| " .. styleAttribute[1] .. colorCSS .. other_CSS .. styleAttribute[2] .. title_attribute .. sortkey .. attribute_separator .. cell_content
	return cell
end

function export.makeRow(frame)
	makeArrays(frame)
	local output = ""
	if frame.args[1] then
		output = "\n|-"
		output = output .. "\n! " .. frame.args[1]
		if frame.args[2] then
			output = output .. " !! " .. frame.args[2]
		end
	end
	if cellFormat then
		output_format = cellFormat
	end
	if a and b and c then
		for i = 1, length do
			if not output_format then
				output_format = output_formats.high_low_average_F
			end
			output = output .. makeCell(output_format, a[i], b[i], c[i])
		end
	elseif a and b then
		for i = 1, length do
			if not output_format then
				output_format = output_formats.high_low_F
			end
			output = output .. makeCell(output_format, a[i], b[i])
		end
	elseif a then
		for i = 1, length do
			if not output_format then
				output_format = output_formats.average_F
			end
			output = output .. makeCell(output_format, a[i])
		end
	end
	output = mw.ustring.gsub(output, "([%p%s])-(%d)", "%1" .. minus .. "%2")
	return output
end

function export.makeTable(frame)
	makeArrays(frame)
	local output = "{| class=\"wikitable center nowrap\""
	if cellFormat then
		output_format = cellFormat
	end
	if a and b and c then
		for i = 1, length do
			if not output_format then
				output_format = output_formats.high_low_average_F
			end
			output = output .. makeCell(output_format, a[i], b[i], c[i])
		end
	elseif a and b then
		for i = 1, length do
			if not output_format then
				output_format = output_formats.high_low_F
			end
			output = output .. makeCell(output_format, a[i], b[i])
		end
	elseif a then
		for i = 1, length do
			if not output_format then
				output_format = output_formats.average_F
			end
			output = output .. makeCell(output_format, a[i])
		end
	end
	output = mw.ustring.gsub(output, "([%p%s])-(%d)", "%1" .. minus .. "%2")
		--[[  Replaces hyphens that have a punctuation or space character before them and a number after them,
				making sure that hyphens in "data-sort-type" are not replaced with minuses.
				If Lua had (?<=), a capture would not be necessary.  ]]
	output = output .. "\n|}"
	if show then
		output = output .. "\n\n<span style=\"color: red; font-size: 80%; line-height: 100%;\">" .. message .. "</span>"
	end
	return output
end



local chart = [[
{{Graph:Chart
|width=600
|height=180
|xAxisTitle=Celsius
|yAxisTitle=__COLOR
|type=line
|x=__XVALUES
|y=__YVALUES
|colors=__COLOR
}}
]]

function export.show(frame)
	-- For testing, return wikitext to show graphs of how the red/green/blue colors
	-- vary with temperature, and a table of the resulting colors.
	local function collection()
		-- Return a table to hold items.
		return {
			n = 0,
			add = function (self, item)
				self.n = self.n + 1
				self[self.n] = item
			end,
			join = function (self, sep)
				return table.concat(self, sep)
			end,
		}
	end
	local function make_chart(result, color, xvalues, yvalues)
		result:add('\n')
		result:add(frame:preprocess((chart:gsub('__[A-Z]+', {
			__COLOR = color,
			__XVALUES = xvalues:join(','),
			__YVALUES = yvalues:join(','),
		}))))
	end
	local function with_minus(value)
		if value < 0 then
			return minus .. tostring(-value)
		end
		return tostring(value)
	end
	local args = frame.args
	local first = args[1] or -90
	local last = args[2] or 59
	local palette = palettes[args.palette] or palettes.cool
	local xvals, reds, greens, blues = collection(), collection(), collection(), collection()
	local wikitext = collection()
	wikitext:add('{| class="wikitable"\n|-\n')
	local columns = 0
	for celsius = first, last do
		local backgroundRGB = {}
		local style = styleAttribute(palette, celsius, backgroundRGB)
		local R = math.floor(backgroundRGB[1])
		local G = math.floor(backgroundRGB[2])
		local B = math.floor(backgroundRGB[3])
		xvals:add(celsius)
		reds:add(R)
		greens:add(G)
		blues:add(B)
		wikitext:add('| ' .. style .. ' | ' .. with_minus(celsius) .. '\n')
		columns = columns + 1
		if columns >= 10 then
			columns = 0
			wikitext:add('|-\n')
		end
	end
	wikitext:add('|}\n')
	make_chart(wikitext, 'Red', xvals, reds)
	make_chart(wikitext, 'Green', xvals, greens)
	make_chart(wikitext, 'Blue', xvals, blues)
	return wikitext:join()
end

return export