Module:Weather box/row

local math_mod = require('Module:Math')

local traceText local Value Value = { lang = mw.getContentLanguage, getDisplay = function (self, second) if not self:isValid then return nil end local display = self.string if display == 'trace' then if second then -- If a cell displays "cm (inch)", show "trace" not "trace (trace)". return nil end return traceText or 'trace' end if math.abs(self.number) >= 1000 then display = self.lang:formatNum(math.abs(self.number)) if self.number < 0 then display = '−' .. display end elseif self.number < 0 then display = '−' .. display:sub(2) end return display end, getPrecision = function (self) local result = rawget(self, 'precision') if not result then if self:isValid then result = math.max(0, math_mod._precision(self.string)) else result = 0 end rawset(self, 'precision', result) end return result end, isValid = function (self) return self.number ~= nil and self.number ~= -9999 end, new = function (v) local val, str, precision if type(v) == 'string' then if v == 'trace' then val, str, precision = 0, 'trace', 0 else val, str = math_mod._cleanNumber(v) end elseif type(v) == 'number' then val, str = v, tostring(v) end if not val then val, str = -9999, '' end return setmetatable({			number = val,			string = str,			precision = precision,		}, Value) end, converts = { in2cm = { factor = 2.54             }, in2mm = { factor = 25.4             }, cm2in = { factor = 1/2.54, p2max = 1 }, mm2in = { factor = 1/25.4, p2max = 0 }, },	setConvert = function (self, invalue, units) -- Use method modified from Module:Convert to determine precision. if invalue.string == 'trace' then self.number, self.string, self.precision = 0, 'trace', 0 return end local convert = self.converts[units] or error('Unknown units') local outnum = invalue.number * convert.factor local precision = invalue:getPrecision if outnum > 0 then local adjust = math.log10(1/convert.factor) + math.log10(2) local p1 = math.floor(precision + adjust) local p2 = 1 - math.floor(math.log10(outnum)) if convert.p2max then p2 = math.min(p2, convert.p2max) end precision = math.max(p1, p2) end self:setNumberRounded(outnum, precision) end, setNumberRounded = function (self, number, precision) if precision > 2 then precision = 2 end self.number = math_mod._round(number, precision) if precision < 0 then self.string = tostring(self.number) else local fmt = '%.' .. string.format('%d', precision) .. 'f'			self.string = string.format(fmt, self.number) end end, } Value.__index = Value

local function checkFlag(flag, default) if flag == nil then return default elseif type(flag) == 'boolean' then return flag elseif type(flag) == 'string' then flag = flag:lower if flag == '0' or flag == 'false' or				flag == '' or flag == 'no' or				flag == 'n' then return false else return true end else return error('Flag type not valid') end end

local function makeLine(label, first_values, second_values, color_values) local result = {'|- style="text-align: center;"\n! scope="row" style="height: 16px;" | ', label, "\n"} for i = 1, 13 do		local color_str = color_values[i] if i == 13 then table.insert(result, table.concat( {'|style="', color_str, ' border-left-width:medium" | '} )) else table.insert(result, table.concat( {'|style="', color_str, '" | '} )) end local display = first_values[i]:getDisplay if display then table.insert(result, display) if second_values ~= nil then display = second_values[i]:getDisplay(true) if display then table.insert(result, " (" .. display .. ")")				end end else table.insert(result, '—') end table.insert(result, "\n") end return table.concat(result) end

local function getInputs(args, group_name, suffix, include_space) local month_names = { 'Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec', 'year' } local str local values = {} if suffix == nil then for i, mon in ipairs(month_names) do			if include_space then str = args[ mon .. ' ' .. group_name ] or '' else str = args[ mon .. group_name ] or '' end values[i] = Value.new(str) end else for i, mon in ipairs(month_names) do			local value, updated for var, suf in ipairs(suffix) do				if include_space then str = args[ mon .. ' ' .. group_name .. ' ' .. suf ] else str = args[ mon .. group_name .. ' ' .. suf ] end if str ~= nil and str ~= '' then value = Value.new(str) value.variant = var updated = true break end end if not updated then value = Value.new value.variant = 0 end values[i] = value end end return values end

local function getAnnualValue(values, mode) if mode == 'avg' or mode == 'sum' then local total = 0 local p1, p2, variant p1 = 0 for i = 1, 12 do			if not values[i]:isValid then return Value.new end if not variant then local var = values[i].variant if var and var ~= 0 then variant = var end end p2 = values[i]:getPrecision if p2 > p1 then p1 = p2			end total = total + values[i].number end local value = Value.new(total) if mode == 'avg' then value:setNumberRounded(total / 12, p1) end value.variant = variant return value elseif mode == 'min' then local target for i = 1, 12 do			if values[i]:isValid then if target == nil or values[i].number < target.number then target = values[i] end end end return target or Value.new elseif mode == 'max' then local target for i = 1, 12 do			if values[i]:isValid then if target == nil or values[i].number > target.number then target = values[i] end end end return target or Value.new else error('Unrecognized Annual Mode') end end

local function reconcileTemperature(C_values, F_values) for i = 1, 13 do		local p		if C_values[i].string == '' then if F_values[i]:isValid then p = F_values[i]:getPrecision C_values[i]:setNumberRounded((F_values[i].number - 32)*5/9, p)			end elseif F_values[i].string == '' then if C_values[i]:isValid then p = C_values[i]:getPrecision F_values[i]:setNumberRounded(C_values[i].number*9/5 + 32, p)			end end end end

local function reconcilePrecipitation(M_values, I_values, prefer_cm) local v_class = 0 for i = 1, 13 do		if M_values[i].variant == 1 then v_class = 1 elseif M_values[i].variant == 2 then v_class = 2 end end if v_class == 0 then if prefer_cm then v_class = 1 else v_class = 2 end end for i = 1, 13 do		local units if M_values[i].string == '' then if I_values[i]:isValid then if v_class == 1 then units = 'in2cm' else units = 'in2mm' end M_values[i]:setConvert(I_values[i], units) M_values[i].variant = v_class end elseif I_values[i].string == '' then if M_values[i]:isValid then if M_values[i].variant == 1 then units = 'cm2in' else units = 'mm2in' end I_values[i]:setConvert(M_values[i], units) end end end end

local function _buildRow(definition, args, options) options = options or {} local wbc = require('Module:Weather box/colors' .. (options.sandbox or '')) local mode = (definition.mode or 'basic'):lower local group_name = definition.group_name local first_values, second_values local color_values local color_scheme = definition.color_scheme or 't'	local scale_factor = math_mod._cleanNumber(definition.scale_factor) or 1 local date_mode = checkFlag(definition.date_mode, false) local label = definition.label or '' local annual_mode = (definition.annual_mode or 'avg'):lower local include_space = checkFlag(definition.include_space, true) local second_line = checkFlag(definition.second_line, false) local prefer_cm = checkFlag(definition.prefer_cm, false) local imperial_first = checkFlag(args['imperial first']) local metric_first = checkFlag(args['metric first']) local wantSingleLine = options.wantSingleLine or checkFlag(args['single line']) local trace = args.trace if trace and trace ~= '' then traceText = trace end if imperial_first == nil then imperial_first = metric_first == nil and true or not metric_first end if mode == 'basic' then first_values = getInputs(args, group_name, nil, include_space) second_values = nil elseif mode == 'temperature' then first_values = getInputs(args, group_name, {'C'}, include_space) second_values = getInputs(args, group_name, {'F'}, include_space) reconcileTemperature(first_values, second_values) elseif mode == "precipitation" then first_values = getInputs(args, group_name, {'cm', 'mm'}, include_space) second_values = getInputs(args, group_name, {'inch'}, include_space) reconcilePrecipitation(first_values, second_values, prefer_cm) else error('Requested mode not recognized') end local good = false for i = 1, 13 do		if first_values[i].string ~= '' then good = true break end end if not good then return '' end if first_values[13].string == '' then first_values[13] = getAnnualValue(first_values, annual_mode) end if second_values ~= nil then if second_values[13].string == '' then second_values[13] = getAnnualValue(second_values, annual_mode) end end color_scheme = wbc.interpret_color_code(color_scheme) color_values = {} local month_adj = { 31/30, 28.25/30, 31/30, 1, 31/30, 1, 31/30, 31/30, 1, 31/30, 1, 31/30, 365.25/30 }	for i = 1, 13 do		if first_values[i]:isValid then local adj = scale_factor if date_mode then adj = adj / month_adj[i] end if mode == "precipitation" then if first_values[i].variant == 1 then adj = adj * 10 end end table.insert(color_values, color_scheme(first_values[i].number * adj)) else table.insert(color_values, color_scheme(nil)) end end if imperial_first and second_values ~= nil then first_values, second_values = second_values, first_values end if not wantSingleLine then if second_line and second_values ~= nil then first_values = second_values end second_values = nil end return makeLine(label, first_values, second_values, color_values) end

local function buildRow(frame) return _buildRow(frame.args, frame:getParent.args) -- row definition, template parameters end

return { buildRow = buildRow, _buildRow = _buildRow, }