okameron/okameron.lua at main · TalonFloof/okameron · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
function getdirectory(p)
	for i = #p, 1, -1 do
		if p:sub(i,i) == "/" then
			return p:sub(1,i)
		end
	end

	return "./"
end
local sd = getdirectory(arg[0])

local args = {...}

arch = "okami1041"

local scan = dofile(sd.."scanner.lua")
local parse = dofile(sd.."parser.lua")
local irgen = dofile(sd.."irgen.lua")

local startup = ""

useXrSDK = false

while #args > 0 and string.sub(args[1],1,1) == "-" do
    if string.sub(args[1],2,6) == "arch=" then
        arch = string.sub(args[1],7)
    elseif string.sub(args[1],2,9) == "startup=" then
        local file = io.open(string.sub(args[1],10),"rb")
        startup = file:read("*a")
        file:close()
    elseif string.sub(args[1],2,#args[1]) == "xrsdk" then
        useXrSDK = true
    else
        print("Unknown option \""..args[1].."\"")
        return
    end
    table.remove(args,1)
end

local codegen = dofile(sd.."codegen-"..arch..".lua")

local tokens = {}

function serialize_list (list,newlines,subNewline)
    local str = "("
    for _,i in ipairs(list) do
        if type(i) == "table" then
            str = str .. serialize_list(i,subNewline,subNewline)
        else
            str = str .. tostring(i)
        end
        str = str .. (newlines and "\n" or " ")
    end
    if #list > 0 then
        str = string.sub(str,1,#str-1)
    end
    str = str .. ")"
    return str
end

local nl = string.char(10)
function serialize_table (tabl, indent)
    indent = indent and (indent.."  ") or ""
    local str = ''
    str = str .. indent.."{"..nl
    for key, value in pairs (tabl) do
        local pr = (type(key)=="string") and ('["'..key..'"]=') or ""
        if type (value) == "table" then
            str = str..pr..serialize_table (value, indent)
        elseif type (value) == "string" then
            str = str..indent..pr..'"'..tostring(value)..'",'..nl
        else
            str = str..indent..pr..tostring(value)..','..nl
        end
    end
    str = str .. indent.."},"..nl
    return str
end

local function addToTokens(t)
    while #t > 0 do
        table.insert(tokens,table.remove(t,1))
    end
end

for _,i in ipairs(args) do
    local file = io.open(i,"rb")
    addToTokens(scan(i,file:read("*a")))
    file:close()
end
local tree, asm = parse(tokens,systemWordSize)
asm = startup .. "\n" .. asm
--io.stderr:write(serialize_list(tree,true,false).."\n")
local ircode = irgen(tree,systemWordSize)
--io.stderr:write(serialize_list(ircode[1],true,false).."\n\n\n")
--io.stderr:write(serialize_list(ircode[2],true,false).."\n")
codegen(ircode,asm)