degrowth/c
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

codegen: factor out head generation

Browse Source
This commit is contained in:
Veit Heller
2018-05-06 15:27:45 +02:00
parent a5ce351f91
commit 063ecdd3d8
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
c.py
View File

@@ -213,19 +213,21 @@ class CodeGen:
res = "{}\n{}".format(res, self.generate(node)) res = "{}\n{}".format(res, self.generate(node))
return res return res
def generate_head(self, node):
args = ", ".join("int {}".format(n) for n in node.args)
return "int {}({})".format(node.name, args)
def generate_proto(self, node): def generate_proto(self, node):
case = node.__class__ case = node.__class__
if case == DefNode: if case == DefNode:
args = ", ".join("int {}".format(n) for n in node.args) return "{};".format(self.generate_head(node))
return "int {}({});".format(node.name, args)
return "" return ""
def generate(self, node): def generate(self, node):
case = node.__class__ case = node.__class__
if case == DefNode: if case == DefNode:
args = ", ".join("int {}".format(n) for n in node.args)
body = self.generate(node.body) body = self.generate(node.body)
return "int {}({}) {{ return {}; }}".format(node.name, args, body) return "{} {{ return {}; }}".format(self.generate_head(node), body)
if case == IntegerNode: if case == IntegerNode:
return str(node.value) return str(node.value)
if case == CallNode: if case == CallNode: