@@ -1,7 +1,10 @@ +SYMBOLS = ['do', 'defun', 'puts', 'printf', 'strlen'] def is_list(l): return type(l) == type([]) def is_int(n): return type(n) == type(1) +def is_atom(a): + return a in SYMBOLS class Compiler: @@ -14,6 +17,8 @@ class Compiler: self.PTR_SIZE = 4 def get_arg(self, a): + if is_atom(a): + return ['atom', a] if is_int(a): return ['int', a] if is_list(a): @@ -28,6 +33,20 @@ class Compiler: def defun(self, name, args, body): self.global_functions[name] = [args, body] + + def ifelse(self, cond, if_branch, else_branch): + self.compile_exp(cond) + print("\ttestl\t%eax, %eax") + self.seq += 2 + else_branch_seq = self.seq - 1 + end_if_branch_seq = self.seq + print("\tje\t.L" + str(else_branch_seq)) + self.compile_exp(if_branch) + print("\tjmp\t.L" + str(end_if_branch_seq)) + print(".L" + str(else_branch_seq) + ":") + self.compile_exp(else_branch) + print(".L" + str(end_if_branch_seq) + ":") + def output_functions(self): for name, data in self.global_functions.items(): @@ -47,39 +66,61 @@ class Compiler: print(".LC" + str(seq) + ":") print("\t.string \"" + str(c) + "\"") - def compile_exp(self, exp): - if not exp or len(exp) == 0: - return False - if exp[0] == 'defun': - return self.defun(*exp[1:]) - if exp[0] == 'do': - exp.pop(0) - map(self.compile_exp, exp) - return True - - call = str(exp[0]) - stack_adjustment = self.PTR_SIZE + int(round((len(exp) - 1 + 0.5) * self.PTR_SIZE / (4.0 * self.PTR_SIZE))) * (4 * self.PTR_SIZE) - if exp[0] != 'do': - print("\tsubl\t$" + str(stack_adjustment) + ", %esp") + def compile_lambda(self, args, body): + name = "lambda__" + str(self.seq) + self.seq += 1 + self.defun(name, args, body) + print("\tmovl\t$" + str(name) + ",%eax") + return ['subexpr'] + + def compile_eval_arg(self, arg): + atype, aparam = self.get_arg(arg) + if atype == 'strconst': + return "$.LC" + str(aparam) + if atype == 'int': + return "$" + str(aparam) + if atype == 'atom': + return str(aparam) + return "%eax" + + def compile_call(self, func, args): + stack_adjustment = self.PTR_SIZE + int(round(((len(args) + 0.5) * self.PTR_SIZE / (4.0 * self.PTR_SIZE)))) * (4 * self.PTR_SIZE) + print("\tsubl\t$" + str(stack_adjustment) + ", %esp") count = 0 - for a in exp[1:]: - atype, aparam = self.get_arg(a) - if exp[0] != 'do': - if atype == 'strconst': - param = "$.LC" + str(aparam) - elif atype == 'int': - param = "$" + str(aparam) - else: - param = "%eax" - + for a in args: + param = self.compile_eval_arg(a) if count > 0: - i = count * self.PTR_SIZE + i = count * 4 else: i = "" - print("\tmovl\t" + str(param) + ", " + str(i) + "(%esp)") + print("\tmovl\t" + str(param) + "," + str(i) + "(%esp)") count += 1 - print("\tcall\t" + str(call)) + res = self.compile_eval_arg(func) + if res == "%eax": + res = "*%eax" + print("\tcall\t" + str(res)) print("\taddl\t$" + str(stack_adjustment) + ", %esp") + return ['subexpr'] + + def compile_do(self, exp): + map(self.compile_exp, exp) + return ['subexpr'] + + def compile_exp(self, exp): + if not exp or len(exp) == 0: + return False + if exp[0] == 'do': + self.compile_do(exp[1:]) + return True + if exp[0] == 'defun': + return self.defun(*exp[1:]) + if exp[0] == 'if': + return self.ifelse(*exp[1:]) + if exp[0] == 'lambda': + return self.compile_lambda(*exp[1:]) + if exp[0] == 'call': + return self.compile_call(exp[1], exp[2]) + return self.compile_call(exp[0], exp[1:]) def compile_main(self, exp): print(""" @@ -108,14 +149,23 @@ main: def compile(self, exp): self.compile_main(['do', self.DO_BEFORE, exp, self.DO_AFTER]) -DO_BEFORE = ['do', - ['defun', 'hello_world', [], ['puts', "Hello World"]] -] +DO_BEFORE = [] DO_AFTER = [] +""" +prog = ['do', + ['if', ['strlen',""], + ['puts', "IF: The string was not empty"], + ['puts', "ELSE: The string was empty"] + ], + ['if', ['strlen',"Test"], + ['puts', "Second IF: The string was not empty"], + ['puts', "Second IF: The string was empty"] + ] +] +""" prog = ['do', - ['printf', "'hello world' takes %ld bytes\\n", ['strlen', "hello world"]], - ['printf', "The above should show _%ld_ bytes\\n",11] + ['call', ['lambda', [], ["puts", "Test"]], [] ] ] compiler = Compiler(DO_BEFORE, DO_AFTER) pycompiler.py 3800f0d0363311610a91a68b3dcb217d421e7f2e Michael Matuzak mmatuzak@gmail.com http://github.com/emkay/pycompile/commit/8fac24d017e7605c81c7d6759547d0c1725e7881 8fac24d017e7605c81c7d6759547d0c1725e7881 2009-05-02T13:52:02-07:00 2009-05-02T13:52:02-07:00 adding lambdas. adding if branches. adding a symbol table. 5fe8a295877401b9b68ce6fee286f844f282cb5b Michael Matuzak mmatuzak@gmail.com