Module:Harmonics in ED

From XenReference

Documentation for this module may be created at Module:Harmonics in ED/doc

local p = {}
local primes = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89}
local p_cols = {"prime2", "prime3", "prime5", "prime7", "prime11", "prime13", "prime17", "prime19", "prime23", "prime29", "prime31", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime", "higherPrime"}

local function steps(et, harm) -- steps of harmonic
    return math.floor((math.log(harm) / math.log(2) * et) + 0.5)
end

local function steps_re(et, harm) -- steps of reduced harmonic
    return steps(et, harm) - (math.floor(math.log(harm) / math.log(2)) * et)
end

local function rel_err(et, harm) -- relative error of harmonic
    st = math.log(harm) / math.log(2) * et
    return math.floor(st + 0.5) - st
end

local function abs_err(et, harm) -- absolute error of harmonic
    return 1200 / et * rel_err(et, harm)
end

local function rel_col(err) -- color used for relative error
    abs_err = math.abs(err)
    if(abs_err == 0) then
        return "acc0"
    end
    if(abs_err < 0.03125) then
        return "acc1"
    end
    return "acc" .. math.floor(abs_err * 12) + 2
end

function p.table(frame) -- making the table itself
    local p_len = #primes
    local et = math.floor(frame.args["et"])
    local nowiki = frame.args["nowiki"]
    local p_lim = math.floor(frame.args["p_lim"])

    local tab = "{| class=\"wikitable\"\n"
    if (tonumber(nowiki) == 1) then
    tab = "<nowiki>" .. tab end
    tab = tab .. "|+Approximation of prime harmonics in "
    tab = tab .. et .. "edo\n"
    tab = tab .. "! colspan=\"2\" |Harmonic\n"
    for i = 1,p_len do
        tab = tab .. "! class=\""
        tab = tab .. p_cols[i] .. "\" |" .. primes[i] .. "\n"
        if primes[i] >= p_lim then break end
    end
    tab = tab .. "|-\n! rowspan=\"2\" |Error\n!Absolute (¢)\n"
    for i = 1,p_len do
        tab = tab .. "| "
        if rel_err(et, primes[i]) > 0 then tab = tab .. "+" end
        tab = tab .. string.format("%.1f", abs_err(et, primes[i])) .. "\n"
        if primes[i] >= p_lim then break end
    end
    tab = tab .. "|-\n!Relative (%)\n"
    for i = 1,p_len do
        local er = rel_err(et, primes[i])
        tab = tab .. "| class=\"" .. rel_col(er) .. "\" | "
        if er > 0 then tab = tab .. "+" end
        tab = tab .. string.format("%.1f", er * 100) .. "\n"
        if primes[i] >= p_lim then break end
    end
    tab = tab .. "|-\n! colspan=\"2\" |Steps\n(reduced)\n"
    for i = 1,p_len do
        tab = tab .. "|" .. steps(et, primes[i]) .. "\n(" .. steps_re(et, primes[i]) .. ")\n"
        if primes[i] >= p_lim then break end
    end
    tab = tab .. "|}"
    if (tonumber(nowiki) == 1) then
    tab = tab .. "</nowiki>" end
    return tab
end

return p