Module:SB tree: Difference between revisions

Line 2:

local rat = require('Module:Rational')

-- Function that constructs a Stern-Brocot tree, ~~given~~ any ~~start and end~~ ratios and any ~~depth~~

-- Function that constructs a sequence of ratios according to the Stern-Brocot tree

-- Ratios are entered as arrays, such as {1, 1} for the ratio 1/1 and {1, 0} for the ratio 1/0

-- Mediants are found between each ratio iteratively

-- 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

-- Edge extend recursively finds the mediants of the first two and last two ratios; default is 0

-- Transpiled from python to lua with aid of ChatGPT; may need testing to make sure code works

function p.sb_tree(depth, start_ratio, stop_ratio, edge_extend)

-- Default parameter values

depth = depth or 4

start_ratio = start_ratio or {1, 1}

stop_ratio = stop_ratio or {1, 0}

edge_extend = edge_extend or 0

-- Initial tree are the ratios 1/1 and 1/0

local tree = {start_ratio, stop_ratio}

Line 26:

Line 37:

end

~~local edge_extend_a = edge_extend~~

for i = 1, edge_extend do

for i = 1, ~~edge_extend_a~~ do

local last_ratio_1 = tree[#tree - 1] -- Second last ratio

local last_ratio_2 = tree[#tree] -- Last ratio

Line 44:

Line 54:

end

-- Test function that produces the SB tree as a one-column table

-- Test function that produces the ratios of the SB tree as a one-column table, using default parameters

-- To try this out, add the following text (not as a lua comment):

-- {{#invoke:SB_tree|sb_table}}

function p.sb_table(frame)

result = '{| class="wikitable"\n'

@@ Line 2: / Line 2: @@
 local rat = require('Module:Rational')
--- Function that constructs a Stern-Brocot tree, given any start and end ratios and any depth
+-- Function that constructs a sequence of ratios according to the Stern-Brocot tree
+-- Ratios are entered as arrays, such as {1, 1} for the ratio 1/1 and {1, 0} for the ratio 1/0
+-- Mediants are found between each ratio iteratively
+-- 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
+-- Edge extend recursively finds the mediants of the first two and last two ratios; default is 0
 -- Transpiled from python to lua with aid of ChatGPT; may need testing to make sure code works
 function p.sb_tree(depth, start_ratio, stop_ratio, edge_extend)
+	-- Default parameter values
+	depth = depth or 4
+	start_ratio = start_ratio or {1, 1}
+	stop_ratio = stop_ratio or {1, 0}
+	edge_extend = edge_extend or 0
      -- Initial tree are the ratios 1/1 and 1/0
      local tree = {start_ratio, stop_ratio}
@@ Line 26: / Line 37: @@
      end
-	local edge_extend_a = edge_extend
+     for i = 1, edge_extend do
-     for i = 1, edge_extend_a do
          local last_ratio_1 = tree[#tree - 1]     -- Second last ratio
          local last_ratio_2 = tree[#tree]         -- Last ratio
@@ Line 44: / Line 54: @@
 end
--- Test function that produces the SB tree as a one-column table
+-- Test function that produces the ratios of the SB tree as a one-column table, using default parameters
+-- To try this out, add the following text (not as a lua comment):
+-- {{#invoke:SB_tree|sb_table}}
 function p.sb_table(frame)
 	result = '{| class="wikitable"\n'