gen.py

import os

N = 1

def decls():
    global N
    s = ""
    for i in range(1, N + 1):
        s = s + f"\tlocal a{i}: integer = {i}" + '\n'
    return s

def capture():
    global N
    s = ""
    for i in range(1, N + 1):
        s = s + f"\t\ta{i} = a{i} * 2" + '\n'
    return s


def genfile():
    f = open("temp/out.pln", "w")

    template = f"""
local m: module = {{}}

function m.bench(N: integer): boolean
{decls()}

    local f: () -> () = function ()
    {capture()}
    end

    for i = 1, N do
        f()
    end
    return true
end

return m"""
    f.write(template)
    f.close()
    os.system("./pallenec --emit-c temp/out.pln")
    os.system("./pallenec temp/out.pln -o temp/out_no_merge.so")
    os.system(f"python3 edit.py {N}")
    os.system("./pallenec --compile-c temp/out.c")

N = int(input("#upvalues (1 - 200): "))
genfile()