Difference between revisions of "Parameters"
From Unofficial Stationeers Wiki
(Fix disparate parameters logic) |
(Doc some of the module's inner workings; add sorting to strMap and sort compare's output) |
||
Line 20: | Line 20: | ||
end | end | ||
− | local function strMap(iter, transform) | + | local function strMap(iter, transform, sort) |
local returnTable = {} | local returnTable = {} | ||
for _, v in pairs(iter) do | for _, v in pairs(iter) do | ||
table.insert(returnTable, transform(v)) | table.insert(returnTable, transform(v)) | ||
+ | end | ||
+ | if sort then | ||
+ | table.sort(returnTable) | ||
end | end | ||
return table.concat(returnTable) | return table.concat(returnTable) | ||
Line 40: | Line 43: | ||
local function normaliseParams(parameters) | local function normaliseParams(parameters) | ||
local paramsNorm = {} | local paramsNorm = {} | ||
+ | -- Prepare a key lookup metatable, which will hold the original | ||
+ | -- parameter names for each normalised parameter, e.g. | ||
+ | -- [test] = {TEST, Test}. paramIndex functions like a Python | ||
+ | -- defaultdict, where the default is a table. | ||
local paramIndex = setmetatable({}, {__index = function(t, k) | local paramIndex = setmetatable({}, {__index = function(t, k) | ||
if not rawget(t, k) then | if not rawget(t, k) then | ||
Line 54: | Line 61: | ||
paramsNorm = removeDuplicates(paramsNorm) | paramsNorm = removeDuplicates(paramsNorm) | ||
+ | -- Overload key lookup in paramsNorm. While [[Module:Set]] will | ||
+ | -- operate on the table (which is to say, the normalised parameters | ||
+ | -- array), key access will be by way of the paramIndex metatable. | ||
setmetatable(paramsNorm, {__index = paramIndex}) | setmetatable(paramsNorm, {__index = paramIndex}) | ||
return paramsNorm | return paramsNorm | ||
Line 71: | Line 81: | ||
%s]], | %s]], | ||
strMap(Set.valueIntersection(baseParams, otherParams), | strMap(Set.valueIntersection(baseParams, otherParams), | ||
− | function(v) return string.format('* %s\n', v) end), | + | function(v) return string.format('* %s\n', v) end, true), |
strMap(Set.valueIntersection(baseNormParams, otherNormParams), | strMap(Set.valueIntersection(baseNormParams, otherNormParams), | ||
function(v) return string.format('* %s < %s [%s]\n', | function(v) return string.format('* %s < %s [%s]\n', | ||
Line 77: | Line 87: | ||
table.concat(otherNormParams[v], '; '), | table.concat(otherNormParams[v], '; '), | ||
v) | v) | ||
− | end), | + | end, true), |
strMap(Set.valueComplement(otherNormParams, baseNormParams), | strMap(Set.valueComplement(otherNormParams, baseNormParams), | ||
function(v) return strMap(baseNormParams[v], | function(v) return strMap(baseNormParams[v], | ||
− | function(s) return string.format('* %s\n', s) end) | + | function(s) return string.format('* %s\n', s) end, true) |
− | end)) | + | end, true)) |
end | end | ||
Revision as of 06:33, 8 June 2015
-- This module implements [[Template:Parameters]]. -- [SublimeLinter luacheck-globals:mw] local p = {} local removeDuplicates = require('Module:TableTools').removeDuplicates local function makeInvokeFunction(funcName) return function(frame) local getArgs = require('Module:Arguments').getArgs return p[funcName](getArgs(frame)) end end local function extractParams(source) local parameters = {} for parameter in string.gmatch(source, '{{{(.-)%f[}|<>]') do table.insert(parameters, parameter) end return removeDuplicates(parameters) end local function strMap(iter, transform, sort) local returnTable = {} for _, v in pairs(iter) do table.insert(returnTable, transform(v)) end if sort then table.sort(returnTable) end return table.concat(returnTable) end function p._code(args) local title = args._base or mw.title.getCurrentTitle().baseText return string.format([[{{%s %s}}]], title, strMap(extractParams(args.base), function(s) return string.format('| %s = \n', s) end)) end function p._compare(args) local Set = require('Module:Set') local function normaliseParams(parameters) local paramsNorm = {} -- Prepare a key lookup metatable, which will hold the original -- parameter names for each normalised parameter, e.g. -- [test] = {TEST, Test}. paramIndex functions like a Python -- defaultdict, where the default is a table. local paramIndex = setmetatable({}, {__index = function(t, k) if not rawget(t, k) then rawset(t, k, {}) end return rawget(t, k) end}) for _, parameter in pairs(parameters) do table.insert(paramsNorm, string.lower(string.gsub(parameter, '%A', ''))) table.insert(paramIndex[ string.lower(string.gsub(parameter, '%A', ''))], parameter) end paramsNorm = removeDuplicates(paramsNorm) -- Overload key lookup in paramsNorm. While [[Module:Set]] will -- operate on the table (which is to say, the normalised parameters -- array), key access will be by way of the paramIndex metatable. setmetatable(paramsNorm, {__index = paramIndex}) return paramsNorm end local baseParams = extractParams(args.base) local otherParams = extractParams(args.other) local baseNormParams = normaliseParams(Set.valueComplement( otherParams, baseParams)) local otherNormParams = normaliseParams(otherParams) return string.format([[Identical: %s Similar: %s Disparate: %s]], strMap(Set.valueIntersection(baseParams, otherParams), function(v) return string.format('* %s\n', v) end, true), strMap(Set.valueIntersection(baseNormParams, otherNormParams), function(v) return string.format('* %s < %s [%s]\n', table.concat(baseNormParams[v], '; '), table.concat(otherNormParams[v], '; '), v) end, true), strMap(Set.valueComplement(otherNormParams, baseNormParams), function(v) return strMap(baseNormParams[v], function(s) return string.format('* %s\n', s) end, true) end, true)) end function p._demo(args) local title = args._base and ('|_template=' .. args._base) or '' return string.format('{{Parameter names example%s|%s}}', title, table.concat(extractParams(args.base), '|')) end function p._list(args) return strMap(extractParams(args.base), function(s) return string.format('* %s\n', s) end) end p.code = makeInvokeFunction('_code') p.compare = makeInvokeFunction('_compare') p.demo = makeInvokeFunction('_demo') p.list = makeInvokeFunction('_list') return p