Module:SB tree: Difference between revisions
No edit summary |
ArrowHead294 (talk | contribs) mNo edit summary |
||
| (24 intermediate revisions by 2 users not shown) | |||
| Line 1: | Line 1: | ||
local p = {} | local p = {} | ||
local yesno = require("Module:Yesno") | |||
-- Function that constructs a sequence of ratios according to the Stern-Brocot tree | -- Function that constructs a sequence of ratios according to the Stern-Brocot tree | ||
| Line 6: | Line 8: | ||
-- Start and stop ratio may be any two ratios, but the default values are 1/1 and 1/0 | -- Start and stop ratio may be any two ratios, but the default values are 1/1 and 1/0 | ||
-- Depth may be specified to be any value, but the default is 4; this is how many generations | -- Depth may be specified to be any value, but the default is 4; this is how many generations | ||
-- after the initial ratios to produce successive layers of the tree | -- after the initial ratios to produce successive layers of the tree, so depth 4 is 5 layers | ||
-- Edge extend iteratively finds the mediants of the first two and last two ratios; default is 0 | -- Edge extend iteratively finds the mediants of the first two and last two ratios; default is 0 | ||
-- Transpiled from python code to lua with aid of ChatGPT | -- Transpiled from python code to lua with aid of ChatGPT | ||
| Line 64: | Line 66: | ||
-- This is nearly identical to the sb_tree_ratios function, except only the depth | -- This is nearly identical to the sb_tree_ratios function, except only the depth | ||
-- and edge extend are needed | -- and edge extend are needed | ||
-- | -- Transpiled from python code to lua with aid of ChatGPT | ||
function p.sb_tree_depths(depth, edge_extend) | function p.sb_tree_depths(depth, edge_extend) | ||
-- Default parameter values | -- Default parameter values | ||
| Line 70: | Line 72: | ||
edge_extend = edge_extend or 0 | edge_extend = edge_extend or 0 | ||
-- Initial depths of the | -- Initial depths of the initial ratios are 1 for lua programming purposes | ||
local tree = {1, 1} | local tree = {1, 1} | ||
| Line 77: | Line 79: | ||
local new_tree = {} | local new_tree = {} | ||
-- Depths of the first and mediant of first and second ratio are added in pairs | |||
-- The last ratio is added after the for loop itself | |||
for j = 1, #tree - 1 do | for j = 1, #tree - 1 do | ||
local depth_1 = tree[j] | local depth_1 = tree[j] | ||
| Line 90: | Line 94: | ||
-- For extending the edges of the tree | -- For extending the edges of the tree | ||
-- When edge-extending, the first and last depths must always be 1 | |||
for i = 1, edge_extend do | for i = 1, edge_extend do | ||
-- Extend at end | -- Extend at end | ||
tree[#tree] = tree[#tree - 1] + 1 | tree[#tree] = tree[#tree - 1] + 1 | ||
table.insert(tree, | table.insert(tree, 1) | ||
-- Extend at front | -- Extend at front | ||
tree[1] = tree[2] + 1 | tree[1] = tree[2] + 1 | ||
table.insert(tree, 1, | table.insert(tree, 1, 1) | ||
end | end | ||
| Line 103: | Line 108: | ||
end | end | ||
-- | -- Helper function that parses ratios entered as "p/q", returning it as an array { p, q } | ||
-- Test function that produces the ratios of the SB tree as a | function p.parse_ratio(ratio_unparsed) | ||
local ratio = {} | |||
for number in string.gmatch(ratio_unparsed, '([^/]+)') do | |||
table.insert(ratio, number) | |||
end | |||
return ratio | |||
end | |||
-- Test function that produces the ratios of the SB tree as a multi-column table, using different arguments | |||
-- To try this out, add the following text (not as a lua comment): | -- To try this out, add the following text (not as a lua comment): | ||
-- {{#invoke:SB_tree|sb_table}} | -- {{#invoke: SB_tree|sb_table}} | ||
function p.sb_table(frame) | function p.sb_table(frame) | ||
-- Call the sb function | -- Call the sb function | ||
-- Start/stop ratios are the same, depth is deeper, and edge-extend is allowed | -- Start/stop ratios are the same, depth is deeper, and edge-extend is allowed | ||
local depth = 4 | local depth = frame.args["Depth"] or 4 | ||
local edge_extend = 0 | local edge_extend = frame.args["Edge Extend"] or 2 | ||
local start_ratio_unparsed = frame.args["Start Ratio"] or "1/1" | |||
local stop_ratio_unparsed = frame.args["Stop Ratio"] or "1/0" | |||
local stagger_ratios = (frame.args["Stagger Ratios"] and string.lower(frame.args["Stagger Ratios"])) == "true" and true or false | |||
-- Parse ratios | |||
local start_ratio = p.parse_ratio(start_ratio_unparsed) | |||
local stop_ratio = p.parse_ratio(stop_ratio_unparsed) | |||
-- Get the ratios and depths | |||
local sb_tree_depths = p.sb_tree_depths(depth, edge_extend) | local sb_tree_depths = p.sb_tree_depths(depth, edge_extend) | ||
local sb_tree_ratios = p.sb_tree_ratios(depth, | local sb_tree_ratios = p.sb_tree_ratios(depth, start_ratio, stop_ratio, edge_extend) | ||
-- Create the table | -- Create the table | ||
result = | result = "{| class=\"wikitable\"\n|-\n" | ||
.. "! " | |||
-- Create the header cell | |||
if stagger_ratios then | |||
-- Create the | result = result .. "colspan=\"" .. depth + edge_extend + 1 .. "\" | " | ||
result = result .. | end | ||
result = result .. "Ratios\n" | |||
-- Create the individual rows | -- Create the individual rows | ||
| Line 128: | Line 151: | ||
result = result .. "|-\n" | result = result .. "|-\n" | ||
-- Create the cells for each row | -- Create the cells for each row | ||
-- | if stagger_ratios then | ||
-- Stagger ratios | |||
for j = 1, depth + edge_extend + 1 do | |||
result = result .. "| " | |||
if j == sb_tree_depths[i] then | |||
result = result .. " | result = result .. sb_tree_ratios[i][1] .. "/" .. sb_tree_ratios[i][2] | ||
end | |||
result = result .. "\n" | |||
end | end | ||
else | |||
-- Don't stagger ratios | |||
result = result .. "| " .. sb_tree_ratios[i][1] .. "/" .. sb_tree_ratios[i][2] .. "\n" | |||
end | end | ||
end | end | ||
| Line 141: | Line 169: | ||
result = result .. "|}" | result = result .. "|}" | ||
return result | -- Debugger option | ||
local debugg = yesno(frame.args["debug"]) | |||
if debugg == true then | |||
result = "<syntaxhighlight lang=\"wikitext\">" .. result .. "</syntaxhighlight>" | |||
end | |||
return frame:preprocess(result) | |||
end | end | ||
return p | return p | ||