Module:ChordVoicings: Difference between revisions

Eufalesio (talk | contribs)
No edit summary
Eufalesio (talk | contribs)
No edit summary
 
Line 1: Line 1:
-- Module:ChordVoicings (tight caps: n rotations per column, ≤ 2^(n-1) voicings)
-- Module:ChordVoicings (ultra-lean, hard-capped)
-- Usage: {{ChordVoicingTable|rchord=4:5:6:7}}
-- Works fast on small chords (n 5). Rotations per column ≤ n. Columns ≤ 2^(n-1).


local p = {}
local p = {}


-- ---------- small utils ----------
-- ---------- tiny utils ----------
local function split_enum(s)
local function split_enum(s) local t={} for num in string.gmatch(s or "", "%d+") do t[#t+1]=tonumber(num) end; return t end
    local t = {}
local function gcd(a,b) while b~=0 do a,b=a%b,b end; return math.abs(a) end
    for num in string.gmatch(s or "", "%d+") do t[#t+1] = tonumber(num) end
local function reduce(t) local g=t[1] for i=2,#t do g=gcd(g,t[i]) end; if g>1 then local u={} for i=1,#t do u[i]=t[i]/g end; return u end; local u={} for i=1,#t do u[i]=t[i] end; return u end
    return t
local function odd(n) while n%2==0 do n=math.floor(n/2) end; return n end
end
local function odds_of(t) local o={} for i=1,#t do o[i]=odd(t[i]) end; return o end
local function gcd(a,b) while b ~= 0 do a, b = a % b, b end; return math.abs(a) end
local function unique_in_order(t) local seen,out={},{} for _,x in ipairs(t) do if not seen[x] then seen[x]=true; out[#out+1]=x end end; return out end
local function gcd_list(t) local g=t[1]; for i=2,#t do g=gcd(g,t[i]) end; return g end
local function to_str(t) local p={} for i=1,#t do p[i]=tostring(t[i]) end; return table.concat(p,":") end
local function odd_part(n) while n % 2 == 0 do n = math.floor(n/2) end; return n end
local function sort_key(t) local p={} for i=1,#t do p[i]=string.format("%03d", t[i]) end; return table.concat(p,"-") end
local function toodds(enum) local o={}; for i=1,#enum do o[i]=odd_part(enum[i]) end; return o end
local function span_ge_4(t) local mn, mx=t[1], t[1] for i=2,#t do if t[i]<mn then mn=t[i] end; if t[i]>mx then mx=t[i] end end; return (mx/mn) >= 4.0 end
local function to_str(t) local p={}; for i=1,#t do p[i]=tostring(t[i]) end; return table.concat(p, ":") end
local function italic(s) return "''"..s.."''" end
local function sort_key(t) local p={}; for i=1,#t do p[i]=string.format("%03d", t[i]) end; return table.concat(p,"-") end
local function boldital(s) return "'''''"..s.."'''''" end
local function span_ratio(t) local mn, mx=t[1], t[1]; for i=2,#t do if t[i]<mn then mn=t[i] end; if t[i]>mx then mx=t[i] end end; return mx/mn end
local function rot2_once(v) local n=#v; local u={} for i=2,n do u[#u+1]=v[i] end; u[#u+1]=v[1]*4; return reduce(u) end
local function clone(t) local u={}; for i=1,#t do u[i]=t[i] end; return u end
local function unique_in_order(t) local seen,out={},{}; for _,x in ipairs(t) do if not seen[x] then seen[x]=true; out[#out+1]=x end end; return out end
 
local function reduce_irreducible(t)
    local g = gcd_list(t)
    if g>1 then
        local u = {}
        for i=1,#t do u[i]=math.floor(t[i]/g) end
        return u
    end
    return clone(t)
end


-- one two-octave rotation
-- apply octave raises for chosen odds (set is {odd -> true})
local function rot2_once(t)
local function apply_octaves(enum, odds, chosen)
    local n = #t
  local out={} for i=1,#enum do local x=enum[i]; if chosen[odds[i]] then x=x*2 end; out[i]=x end; return out
    local u = {}
    for i=2,n do u[#u+1] = t[i] end
    u[#u+1] = t[1]*4
    return reduce_irreducible(u)
end
end


-- generate non-empty subsets (2^(k)-1) and cap explicitly
-- Build rotation map with n steps, covering only wanted row odds
local function all_nonempty_subsets_capped(arr, maxcount)
local function rotation_map_capped(start_vec, want_set, n)
    local res = {}
  local map = {}
    local function rec(i, cur)
  local v = reduce(start_vec)
        if #res >= maxcount then return end
  for step=1,n do
        if i > #arr then
    local f = odd(v[1])
            if #cur > 0 then
    if want_set[f] and not map[f] then
                local add = {}
      map[f] = v
                for k=1,#cur do add[k]=cur[k] end
      -- early exit: have we covered all wanted odds?
                res[#res+1] = add
      local ok=true
            end
      for w,_ in pairs(want_set) do if not map[w] then ok=false; break end end
            return
      if ok then break end
        end
        if #res < maxcount then rec(i+1, cur) end
        if #res < maxcount then
            cur[#cur+1] = arr[i]
            rec(i+1, cur)
            cur[#cur] = nil
        end
     end
     end
     rec(1, {})
     if step==n then break end
    return res
    v = rot2_once(v)
  end
  return map
end
end


local function subset_labels(nonroot_odds, n_total)
-- Deterministic, non-recursive subset listing for k ≤ 4
    -- There are at most 2^(n-1) columns including Root. Non-root subset count ≤ 2^(n-1)-1.
local function subset_columns(nonroot_sorted, n_total)
    local k = #nonroot_odds
  -- total columns ≤ 2^(n-1). We’ll emit in size order: 0 (Root), 1-sets, 2-sets, 3-sets, 4-sets…
    local max_subsets = math.max(0, (2^(n_total-1)) - 1)
  local cols = { {text="Root", set={}} }
  local cap = math.floor(2^(n_total-1)) - 1 -- non-root subsets cap
  local k = #nonroot_sorted
  if cap <= 0 or k == 0 then return cols end


    local nums = {}
  -- size-1
    for i,x in ipairs(nonroot_odds) do nums[i]=x end
  for i=1,math.min(k,cap>0 and k or 0) do
    table.sort(nums)
    cols[#cols+1] = { text="'"..nonroot_sorted[i], set={nonroot_sorted[i]} }
  end
  local used = math.min(k, cap)
  if used >= cap then return cols end


     local labels = {}
  -- size-2 (i<j)
    labels[#labels+1] = { text="Root", set={} }
  for i=1,k-1 do
 
     for j=i+1,k do
    local subsets = all_nonempty_subsets_capped(nums, max_subsets)
      cols[#cols+1] = { text="'"..nonroot_sorted[i].."'"..nonroot_sorted[j], set={nonroot_sorted[i], nonroot_sorted[j]} }
    table.sort(subsets, function(a,b)
      used = used + 1
        if #a ~= #b then return #a < #b end
      if used >= cap then return cols end
        for i=1,math.min(#a,#b) do
            if a[i] ~= b[i] then return a[i] < b[i] end
        end
        return #a < #b
    end)
    for _,set in ipairs(subsets) do
        local parts={}
        for i,od in ipairs(set) do parts[i] = "'" .. tostring(od) end
        labels[#labels+1] = { text=table.concat(parts, ""), set=set }
     end
     end
    return labels
  end
end


local function apply_octaves(enum, odds, chosen_set)
  -- size-3 (i<j<l)
    local out = {}
  for i=1,k-2 do
     for i=1,#enum do
     for j=i+1,k-1 do
         local v = enum[i]
      for l=j+1,k do
        if chosen_set[odds[i]] then v = v * 2 end
         cols[#cols+1] = { text="'"..nonroot_sorted[i].."'"..nonroot_sorted[j].."'"..nonroot_sorted[l], set={nonroot_sorted[i], nonroot_sorted[j], nonroot_sorted[l]} }
         out[i] = v
        used = used + 1
         if used >= cap then return cols end
      end
     end
     end
    return out
  end
end
 
local function italicize(s) return "''" .. s .. "''" end
local function bold_italic(s) return "'''''" .. s .. "'''''" end


-- ---------- rotation map with hard cap: ≤ n rotations ----------
  -- size-4 (i<j<l<m) — practically never hit for n≤5, but included
-- Build map: firstOdd -> vector, performing at most n steps (including initial state).
  for i=1,k-3 do
-- Stop early if we have covered all requested row-odds.
     for j=i+1,k-2 do
local function build_rotation_map_capped(start_vec, want_odds_set, n)
      for l=j+1,k-1 do
    local map = {}
         for m=l+1,k do
     local v = reduce_irreducible(start_vec)
          cols[#cols+1] = { text="'"..nonroot_sorted[i].."'"..nonroot_sorted[j].."'"..nonroot_sorted[l].."'"..nonroot_sorted[m], set={nonroot_sorted[i], nonroot_sorted[j], nonroot_sorted[l], nonroot_sorted[m]} }
    for step = 1, n do
          used = used + 1
         local fodd = odd_part(v[1])
          if used >= cap then return cols end
        if not map[fodd] then
            map[fodd] = v
            -- early exit if all wanted odds are covered
            local all_ok = true
            for od,_ in pairs(want_odds_set) do
                if not map[od] then all_ok=false; break end
            end
            if all_ok then break end
         end
         end
        -- last iteration shouldn't rotate again
      end
        if step == n then break end
        v = rot2_once(v)
     end
     end
    return map
  end
 
  return cols
end
end


-- ---------- main ----------
function p.render(frame)
function p.render(frame)
    local args = frame.args or frame:getParent().args
  local args = frame.args or frame:getParent().args
    local rchord_str = (args.rchord or args[1] or ""):gsub("%s+", "")
  local rchord_str = (args.rchord or args[1] or ""):gsub("%s+","")
    if rchord_str == "" then
  if rchord_str == "" then return '<strong class="error">Missing parameter: rchord</strong>' end
        return "<strong class=\"error\">Missing parameter: rchord</strong>"
    end


    local width = args.width or "120px"
  local enum = split_enum(rchord_str)
    local title = args.title or "Voicings and rotations around two octaves"
  local n = #enum
    local cls   = args.class or "wikitable sortable"
   if n < 3 then return '<strong class="error">rchord must have at least 3 integers</strong>' end


    local enum = split_enum(rchord_str)
  local width = args.width or "120px"
    local n = #enum
  local title = args.title or "Voicings and rotations around two octaves"
    if n < 3 then
  local cls  = args.class or "wikitable sortable"
        return "<strong class=\"error\">rchord must have at least 3 integers</strong>"
    end


    -- odds & rows
  local odds = odds_of(enum)
    local odds = toodds(enum)
  local root_odd = odd(enum[1])
    local root_odd = odd_part(enum[1])
  local odd_rows = unique_in_order(odds)
    local odd_rows = unique_in_order(odds)


    -- which row-odds do we actually need to cover?
  -- rows we actually need
    local want_odds_set = {}
  local want = {}; for _,o in ipairs(odd_rows) do want[o]=true end
    for _,o in ipairs(odd_rows) do want_odds_set[o] = true end


    -- columns: Root + subsets of non-root odds, capped at 2^(n-1)
  -- non-root odds, sorted
    local nonroot_odds_unique = {}
  local nonroot = {}
    for _,o in ipairs(odd_rows) do if o ~= root_odd then nonroot_odds_unique[#nonroot_odds_unique+1] = o end end
  for _,o in ipairs(odd_rows) do if o ~= root_odd then nonroot[#nonroot+1] = o end end
    local col_specs = subset_labels(nonroot_odds_unique, n) -- {text,set}
  table.sort(nonroot)


    -- header
  -- columns (Root + capped subsets)
    local out = {}
  local cols = subset_columns(nonroot, n)
    out[#out+1] = "== " .. title .. " =="
    out[#out+1] = string.format('{| class="%s" style="text-align:center;"', cls)
    out[#out+1] = "|+"
    out[#out+1] = string.format('! style="width:%s;" {{diagonal split header|Rotation|Voicing}}', width)
    for _,spec in ipairs(col_specs) do
        out[#out+1] = string.format('! style="width:%s;" | %s', width, spec.text)
    end


    -- precompute rotation maps with at most n rotations each
  -- header (no external template calls in header to keep it cheap)
    local rotation_maps = {}
  local out = {}
    for c, spec in ipairs(col_specs) do
  out[#out+1] = "== " .. title .. " =="
        local chosen = {}
  out[#out+1] = string.format('{| class="%s" style="text-align:center;"', cls)
        for _,od in ipairs(spec.set) do chosen[od] = true end
  out[#out+1] = "|+"
        local base_voicing = apply_octaves(enum, odds, chosen)
  out[#out+1] = string.format('! style="width:%s;" | Rotation/Voicing', width)
        rotation_maps[c] = build_rotation_map_capped(base_voicing, want_odds_set, n)
  for _,c in ipairs(cols) do
    end
    out[#out+1] = string.format('! style="width:%s;" | %s', width, c.text)
  end


    -- rows
  -- precompute rotation maps for each column, with ≤ n rotations
     for _,rowodd in ipairs(odd_rows) do
  local rotation_maps = {}
        out[#out+1] = "|-"
  for i,c in ipairs(cols) do
        out[#out+1] = string.format('! style="width:%s;" | On %s', width, tostring(rowodd))
    local chosen = {}
     for _,odv in ipairs(c.set) do chosen[odv]=true end
    local base = apply_octaves(enum, odds, chosen)
    rotation_maps[i] = rotation_map_capped(base, want, n)
  end


        for c, spec in ipairs(col_specs) do
  -- rows
            local rotated = rotation_maps[c][rowodd]
  for _,rowodd in ipairs(odd_rows) do
            if not rotated then
    out[#out+1] = "|-"
                -- If the capped cycle didn't hit this odd, fall back to reduced base voicing.
    out[#out+1] = string.format('! style="width:%s;" | On %s', width, tostring(rowodd))
                local chosen = {}
    for i,c in ipairs(cols) do
                for _,od in ipairs(spec.set) do chosen[od] = true end
      local r = rotation_maps[i][rowodd]
                rotated = reduce_irreducible(apply_octaves(enum, odds, chosen))
      if not r then -- capped cycle didn't hit this odd; show reduced base
            end
        local chosen = {}
 
        for _,odv in ipairs(c.set) do chosen[odv]=true end
            local disp = to_str(rotated)
        r = reduce(apply_octaves(enum, odds, chosen))
            local shown
      end
            local is_two_octaves = (span_ratio(rotated) >= 4.0)
      local s = to_str(r)
 
      local shown
            if rowodd == root_odd and #spec.set == 0 then
      if rowodd == root_odd and #c.set == 0 then
                shown = bold_italic(disp)
        shown = boldital(s)
            elseif not is_two_octaves then
      elseif not span_ge_4(r) then
                shown = italicize(disp)
        shown = italic(s)
            else
      else
                shown = disp
        shown = s
            end
      end
 
      out[#out+1] = string.format('| data-sort-value="%s" | %s', sort_key(r), shown)
            out[#out+1] = string.format('| data-sort-value="%s" | %s', sort_key(rotated), shown)
        end
     end
     end
  end


    out[#out+1] = "|}"
  out[#out+1] = "|}"
    out[#out+1] = "Enumerations in italics map to an existing octave-reduced rotation."
  out[#out+1] = "Enumerations in italics map to an existing octave-reduced rotation."
    return table.concat(out, "\n")
  return table.concat(out, "\n")
end
end


return p
return p