Line 1: |
Line 1: |
| + | -- [[Category:Todo]]: have Module:Logger to expose syntax warnings, similarly to Module:DropList (use it there as well). |
| + | |
| local getArgs = require("Module:GetArgs") | | local getArgs = require("Module:GetArgs") |
− | local format = require("Module:StringInterpolation").format | + | local Utils = require("Module:Core") |
− | local find = require("Module:Utils").find | + | local format = Utils.format |
| + | local BaseData = require("Module:BaseData") |
| + | |
| + | local MapBranchingTable = BaseData{ |
| + | |
| + | _grammar = { |
| + | split = "^%s*(.+)%s*->%s*(.+)%s*$", |
| + | comma_list = "[^,]+", |
| + | node_and_colors = "^%s*([%a%d]+)%s*/%s*(%S+)%s*$", |
| + | node = "^%s*([%a%d]+)%s*$", |
| + | digit_node = "^%s*%d%s*$", |
| + | }, |
| | | |
− | local templates = {
| + | _id = "mapbranchingtable", -- not supporting unique ids for now |
| + | _title = "Branching Rules", |
| + | _width = "auto", |
| + | _start = "'''Start'''", |
| | | |
− | table_header = [[{| class="wikitable mw-collapsible mw-collapsed typography-xl-optout branching-table" style="width:100%;" | + | -- .branching-table is defined in MediaWiki:Common.css |
− | |- | + | _template = [[{| class="wikitable typography-xl-optout branching-table" style="width:${width};" |
− | !colspan="3"|<span style="float:left;padding-left:10px;">Branching Rules</span> | + | |- class="mw-customtoggle-${id}" style="cursor:pointer;" |
− | |-
| + | !colspan="3"|${title} |
| + | |- class="mw-collapsible ${collapsed}" id="mw-customcollapsible-${id}" |
| !colspan="2"|Nodes||Rules | | !colspan="2"|Nodes||Rules |
| + | ${rows} |
| + | |}]], |
| + | |
| + | _row_start_template = [[|- class="mw-collapsible ${collapsed}" id="mw-customcollapsible-${id}" |
| + | |rowspan="${rowspan}" style="text-align:center;vertical-align:middle;width:50px;"|${from}]], |
| + | |
| + | _row_separator_template = [[|- class="mw-collapsible ${collapsed}" id="mw-customcollapsible-${id}" |
| ]], | | ]], |
| | | |
− | table_footer = "\n|}", | + | _row_template = [[${separator}|class="mw-collapsible ${collapsed}" id="mw-customcollapsible-${id}" style="text-align:center;width:50px;"|${to}${rules}]], |
| + | |
| + | _rules = "\n|rowspan=\"${span}\" style=\"padding:10px;\"|\n${rules}", |
| + | |
| + | -- .kcRoute is defined in MediaWiki:Common.css |
| + | _node_template = [[<div class="maproute" style="vertical-align:middle"><div class="${class}" style="${style}">${label}</div></div>]], |
| + | |
| + | _node_color1 = [[background: ${color};]], |
| | | |
− | table_row_start = [[|- | + | _node_color2 = [[background: linear-gradient(90deg, ${color1} 50%, ${color2} 50%)]], |
− | |rowspan="${rowspan}" style="text-align:center;vertical-align:middle;width:10%"|${from}]],
| |
| | | |
− | table_row = [[${separator}|style="text-align:center;width:10%;"|${to} | + | _node_color3 = [[background: linear-gradient(-30deg, transparent 60%, ${color1} 60%) 0% 100%, linear-gradient(30deg, transparent 60%, ${color2} 60%) 100% 100%, linear-gradient(${color3}, ${color3}) 0% 100%, linear-gradient(${color3}, ${color3}) 100% 100%, linear-gradient(${color1}, ${color1}) 0% 0%, linear-gradient(${color2}, ${color2}) 100% 0%; background-repeat: no-repeat; background-size: 50% 50%;]], |
− | |
| |
− | ${rules}]], | |
| | | |
− | battle_node_label = [[<div class="kcRoute" style="vertical-align:middle"><div class="kcRouteNode" style="background:${color};">${label}</div></div>]], | + | _node_color4 = [[background: linear-gradient(${color1}, ${color1}) 0% 0%, linear-gradient(${color2}, ${color2}) 100% 0%, linear-gradient(${color3}, ${color3}) 100% 100%, linear-gradient(${color4}, ${color4}) 0% 100%; background-size: 50% 50%; background-repeat: no-repeat;]], |
| + | |
| + | _node_colors = { |
| + | grey = "grey", |
| + | battle = "#FF1744", -- Red A400 |
| + | resource = "#64DD17", -- Light Green A700 |
| + | storm = "#EA80FC", -- Purple A100 |
| + | empty = "#40C4FF", -- Light Blue A200 |
| + | night = "#7E57C2", -- Deep Purple 400 |
| + | nightday_night = { "battle", "night", "night", "night" }, |
| + | night_battle_empty = { "night", "battle", "empty" }, |
| + | ambush = { "battle", "orange", "battle", "orange" }, |
| + | }, |
| | | |
| } | | } |
| | | |
− | local node_colors = {
| + | function make_id_from_title(title) |
− | grey = "grey", | + | return title:gsub("%s", ""):lower() |
− | battle = "#FF1744", -- Red A400
| + | end |
− | resource = "#64DD17", -- Light Green A700
| |
− | storm = "#EA80FC", -- Purple A100
| |
− | empty = "#40C4FF", -- Light Blue A200
| |
− | }
| |
| | | |
− | function formatNodeLabel(label, color) | + | function MapBranchingTable:parse_node(node_string) |
− | if label == "0" then | + | local node, colors = node_string:match(self._grammar.node_and_colors) |
− | return "'''Start'''" | + | if not node then |
− | elseif label:match("%d") then
| + | node = node_string:match(self._grammar.node) |
− | return format{templates.battle_node_label, label = label, color = node_colors.grey}
| |
| else | | else |
− | return format{templates.battle_node_label, label = label, color = node_colors[color] or color or node_colors.battle} | + | colors = mw.text.split(colors, "%s*/%s*") |
| end | | end |
| + | return node, colors |
| end | | end |
| | | |
− | function formatTable(args) | + | function MapBranchingTable:parse(args) |
− | local branching = { index = {}, } | + | self._vars = { |
| + | id = args.id or args.title and make_id_from_title(args.title) or self._id, |
| + | title = args.title or self._title, |
| + | width = args.width or self._width, |
| + | branching = { index = {} }, |
| + | collapsed = args.expand and "" or "mw-collapsed", |
| + | } |
| + | local branching = self._vars.branching |
| for route, rules in pairs(args) do | | for route, rules in pairs(args) do |
− | local from, to = route:match("(%S+)%s*->%s*(%S+)") | + | if type(route) ~= "number" then |
− | local from_color = from:match("%S+/(%S+)")
| + | local from_node, to_nodes = route:match(self._grammar.split) |
− | local to_color = to:match("%S+/(%S+)")
| + | if from_node and to_nodes then |
− | from = from_color and from:match("(%S+)/") or from
| + | local from_node_colors |
− | to = to_color and to:match("(%S+)/") or to
| + | from_node, from_node_colors = self:parse_node(from_node) |
− | if not find(branching.index, from) then
| + | local first_node = nil |
− | table.insert(branching.index, from)
| + | for to_node in string.gmatch(to_nodes, self._grammar.comma_list) do |
| + | local to_node_colors |
| + | to_node, to_node_colors = self:parse_node(to_node) |
| + | if from_node and to_node then |
| + | Utils.insertNew(branching.index, from_node) |
| + | if not branching[from_node] then |
| + | branching[from_node] = { |
| + | colors = from_node_colors, |
| + | index = {}, |
| + | indexes = { index = {}, groups = {} } |
| + | } |
| + | end |
| + | if not first_node then |
| + | first_node = to_node |
| + | branching[from_node][to_node] = { |
| + | colors = to_node_colors, |
| + | rules = rules, |
| + | span = 1 |
| + | } |
| + | if not branching[from_node].indexes.groups[first_node] then |
| + | table.insert(branching[from_node].indexes.index, first_node) |
| + | branching[from_node].indexes.groups[first_node] = { first_node } |
| + | end |
| + | else |
| + | branching[from_node][first_node].span = branching[from_node][first_node].span + 1 |
| + | branching[from_node][to_node] = { colors = to_node_colors } |
| + | if branching[from_node].indexes.groups[first_node] then |
| + | Utils.insertNew(branching[from_node].indexes.groups[first_node], to_node) |
| + | end |
| + | end |
| + | end |
| + | end |
| + | end |
| end | | end |
− | if not branching[from] then
| + | end |
− | branching[from] = { color = from_color, index = {} } | + | local sorting |
| + | if args.order then |
| + | local order_nodes = mw.text.split(args.order, "%s*,%s*") |
| + | sorting = function(a, b) |
| + | local ai = Utils.ifind(order_nodes, a) |
| + | local bi = Utils.ifind(order_nodes, b) |
| + | if ai and bi then |
| + | return ai < bi |
| + | else |
| + | return a < b |
| + | end |
| end | | end |
− | branching[from][to] = { color = to_color, rules = rules } | + | else |
− | if not find(branching[from].index, to) then
| + | sorting = function(a, b) |
− | table.insert(branching[from].index, to) | + | return a < b |
| end | | end |
| end | | end |
− | table.sort(branching.index) | + | table.sort(branching.index, sorting) |
| for _, from in ipairs(branching.index) do | | for _, from in ipairs(branching.index) do |
− | table.sort(branching[from].index) | + | table.sort(branching[from].indexes.index, sorting) |
| + | for _, first_node in ipairs(branching[from].indexes.index) do |
| + | for _, node in ipairs(branching[from].indexes.groups[first_node]) do |
| + | table.insert(branching[from].index, node) |
| + | end |
| + | end |
| end | | end |
| + | end |
| + | |
| + | function MapBranchingTable:format_color(color) |
| + | return self._node_colors[color] or color or self._node_colors.battle |
| + | end |
| + | |
| + | function MapBranchingTable:format_node(label, colors) |
| + | local style |
| + | if label:match(self._grammar.digit_node) then |
| + | style = format{self._node_color1, color = self._node_colors.grey} |
| + | elseif not colors or #colors == 0 then |
| + | style = format{self._node_color1, color = self._node_colors.battle} |
| + | elseif #colors == 1 then |
| + | if type(self._node_colors[colors[1]]) == "table" then |
| + | return self:format_node(label, self._node_colors[colors[1]]) |
| + | end |
| + | style = format{self._node_color1, color = self:format_color(colors[1])} |
| + | elseif #colors == 2 then |
| + | style = format{ |
| + | self._node_color2, |
| + | color1 = self:format_color(colors[1]), |
| + | color2 = self:format_color(colors[2]), |
| + | } |
| + | elseif #colors == 3 then |
| + | style = format{ |
| + | self._node_color3, |
| + | color1 = self:format_color(colors[1]), |
| + | color2 = self:format_color(colors[2]), |
| + | color3 = self:format_color(colors[3]), |
| + | } |
| + | elseif #colors >= 4 then |
| + | style = format{ |
| + | self._node_color4, |
| + | color1 = self:format_color(colors[1]), |
| + | color2 = self:format_color(colors[2]), |
| + | color3 = self:format_color(colors[3]), |
| + | color4 = self:format_color(colors[4]), |
| + | } |
| + | end |
| + | return label == "0" and self._start or format{ |
| + | self._node_template, |
| + | label = label, |
| + | class = #label > 1 and "mapnodewide" or "mapnode", |
| + | style = style |
| + | } |
| + | end |
| + | |
| + | function MapBranchingTable:format_rows() |
| + | local branching = self._vars.branching |
| local rows = {} | | local rows = {} |
− | for _, from in ipairs(branching.index) do | + | for _, from_node in ipairs(branching.index) do |
| table.insert(rows, format{ | | table.insert(rows, format{ |
− | templates.table_row_start, | + | self._row_start_template, |
− | rowspan = #branching[from].index, | + | rowspan = #branching[from_node].index, |
− | from = formatNodeLabel(from, branching[from].color), | + | from = self:format_node(from_node, branching[from_node].colors), |
| + | id = self._vars.id, |
| + | collapsed = self._vars.collapsed, |
| }) | | }) |
| local first = true | | local first = true |
− | for _, to in ipairs(branching[from].index) do | + | for _, to_node in ipairs(branching[from_node].index) do |
| table.insert(rows, format{ | | table.insert(rows, format{ |
− | templates.table_row, | + | self._row_template, |
− | separator = first and "" or "|-\n", | + | separator = first and "" or format{ |
− | to = formatNodeLabel(to, branching[from][to].color), | + | self._row_separator_template, |
− | rules = branching[from][to].rules, | + | id = self._vars.id, |
| + | collapsed = self._vars.collapsed, |
| + | }, |
| + | to = self:format_node(to_node, branching[from_node][to_node].colors), |
| + | rules = branching[from_node][to_node].rules and format{ |
| + | self._rules, |
| + | rules = branching[from_node][to_node].rules, |
| + | span = branching[from_node][to_node].span, |
| + | } or "", |
| + | id = self._vars.id, |
| + | collapsed = self._vars.collapsed, |
| }) | | }) |
| first = false | | first = false |
| end | | end |
| end | | end |
− | return templates.table_header .. table.concat(rows, "\n") .. templates.table_footer | + | self._vars.rows = table.concat(rows, "\n") |
| end | | end |
| | | |
− | local MapBranchingTable = {}
| + | function MapBranchingTable:format(args) |
| + | self:parse(args) |
| + | self:format_rows() |
| + | return format(self._template, self._vars) |
| + | end |
| | | |
− | function MapBranchingTable.format(frame, args_) | + | function MapBranchingTable.Table(frame, args) |
− | local args = args_ or getArgs{frame = frame:getParent()} | + | return MapBranchingTable:format(args or getArgs{frame = frame:getParent()}) |
− | return formatTable(args)
| |
| end | | end |
| | | |
− | MapBranchingTable.t = MapBranchingTable.format(nil, { ["0 -> 1"] = "Fixed route", ["1 -> B/green"] = "Random", ["1 -> C"] = "Random" })
| + | Utils.registerTableTests(MapBranchingTable, { |
| + | { |
| + | ["0 -> 1"] = "Fixed route", |
| + | ["1 -> C/empty, D"] = "Random", |
| + | ["1 -> A, B/battle, X/empty"] = "Random", |
| + | ["C -> F, G/battle/empty"] = "...", |
| + | ["B -> D/red/green/blue, E/red/green/blue/yellow"] = "...", |
| + | ["X -> Y/nightday_night, Z/night_battle_empty"] = "...", |
| + | ["Z1 -> Z2, Z3"] = "...", |
| + | ["ZZ1 -> ZZ2, ZZ3"] = "...", |
| + | ["title"] = "A Custom Title", |
| + | ["order"] = "C, A, 1, B", |
| + | "?", |
| + | } |
| + | }) |
| + | -- p.run_table_tests() |
| | | |
| return MapBranchingTable | | return MapBranchingTable |