[pypy-svn] rev 1653 - pypy/trunk/src/pypy/translator

[arigo at codespeak.net]

Author: arigo
Date: Thu Oct  9 19:31:04 2003
New Revision: 1653

Added:
   pypy/trunk/src/pypy/translator/gencl.py   (contents, props changed)
Log:
A Common Lisp generator! From sanxiyn. Still preliminary but already nice to 
have :-)


Added: pypy/trunk/src/pypy/translator/gencl.py
==============================================================================
--- (empty file)
+++ pypy/trunk/src/pypy/translator/gencl.py	Thu Oct  9 19:31:04 2003
@@ -0,0 +1,111 @@
+import autopath
+from pypy.translator.flowmodel import *
+
+class Op:
+    def __init__(self, gen, op):
+        self.str = gen.str
+        self.opname = op.opname
+        self.args = op.args
+        self.result = op.result
+    def __call__(self):
+        meth_name = "op_" + self.opname
+        meth_default = self.op_default
+        meth = getattr(self, meth_name, meth_default)
+        meth()
+    def op_default(self):
+        print self.opname, "is missing"
+    def op_mod(self):
+        result, arg1, arg2 = map(self.str, (self.result,) + self.args)
+        print "(setq", result, "(mod", arg1, arg2, "))"
+
+class GenCL:
+    def __init__(self, fun):
+        self.fun = fun
+        self.blockref = {}
+    def str(self, obj):
+        if isinstance(obj, Variable):
+            return obj.pseudoname
+    def emit(self):
+        self.emit_defun(self.fun)
+    def emit_defun(self, fun):
+        print "(defun", fun.functionname, "(",
+        for arg in fun.get_args():
+            print self.str(arg),
+        print ")"
+        print "(block nil"
+        self.emit_block(fun.startblock)
+        print ")"
+        print ")"
+    def emit_block(self, block):
+        print "(tagbody"
+        nb = len(self.blockref)
+        tag = self.blockref.setdefault(block, nb)
+        if tag != nb:
+            print "(go", "tag" + str(tag), ")"
+            print ")" # close tagbody
+            return
+        print "tag" + str(tag)
+        for op in block.operations:
+            emit_op = Op(self, op)
+            emit_op()
+        self.dispatch_branch(block.branch)
+        print ")"
+    def dispatch_branch(self, branch):
+        if isinstance(branch, Branch):
+            self.emit_branch(branch)
+        elif isinstance(branch, ConditionalBranch):
+            self.emit_conditional_branch(branch)
+        elif isinstance(branch, EndBranch):
+            self.emit_end_branch(branch)
+        else:
+            print branch.__class__, "is missing"
+    def emit_branch(self, branch):
+        source = branch.args
+        target = branch.target.input_args
+        for item in zip(source, target):
+            init, var = map(self.str, item)
+            print "(setq", var, init, ")"
+        self.emit_block(branch.target)
+    def emit_conditional_branch(self, branch):
+        # XXX: Fix this
+        print "(if (not (zerop", self.str(branch.condition), "))"
+        self.emit_branch(branch.ifbranch)
+        self.emit_branch(branch.elsebranch)
+        print ")"
+    def emit_end_branch(self, branch):
+        retval = self.str(branch.returnvalue)
+        print "(return", retval, ")"
+
+def my_gcd(a, b):
+    r = a % b
+    while r:
+        a = b
+        b = r
+        r = a % b
+    return b
+
+import sys
+from pypy.objspace.flow import Space
+from vpath.adapter.process import exec_cmd
+from cStringIO import StringIO
+
+def test(func, *args):
+    fun = Space().build_flow(func)
+    gen = GenCL(fun)
+    out = StringIO()
+    sys.stdout = out
+    gen.emit()
+    sys.stdout = sys.__stdout__
+    fp = file("test.lisp", "w")
+    fp.write(out.getvalue())
+    print >>fp, "(write (", fun.functionname,
+    for arg in args:
+        print >>fp, str(arg),
+    print >>fp, "))"
+    fp.close()
+    output = exec_cmd("clisp test.lisp")
+    print "Python:", func(*args)
+    print "Lisp:", output
+
+if __name__ == "__main__":
+    test(my_gcd, 96, 64)