[pypy-svn] r72258 - pypy/branch/avm/pypy/translator/avm2

[magcius at codespeak.net]

Author: magcius
Date: Mon Mar 15 20:17:59 2010
New Revision: 72258

Added:
   pypy/branch/avm/pypy/translator/avm2/entrypoint.py   (contents, props changed)
Log:
I forgot to add a file.

Added: pypy/branch/avm/pypy/translator/avm2/entrypoint.py
==============================================================================
--- (empty file)
+++ pypy/branch/avm/pypy/translator/avm2/entrypoint.py	Mon Mar 15 20:17:59 2010
@@ -0,0 +1,64 @@
+
+from pypy.translator.avm2.types_ import Avm2TypeSystem
+from pypy.translator.avm2.database import LowLevelDatabase
+from pypy.translator.cli.node import Node
+from pypy.rpython.ootypesystem import ootype
+
+def get_entrypoint(graph):
+    from pypy.translator.avm2.test.runtest import TestEntryPoint
+    try:
+        ARG0 = graph.getargs()[0].concretetype
+    except IndexError:
+        ARG0 = None
+    if isinstance(ARG0, ootype.List) and ARG0.ITEM is ootype.String:
+        return StandaloneEntryPoint(graph)
+    else:
+        return TestEntryPoint(graph)
+
+class BaseEntryPoint(Node):
+    
+    def set_db(self, db):
+        self.db = db
+        self.cts = Avm2TypeSystem(db)
+
+class StandaloneEntryPoint(BaseEntryPoint):
+    """
+    This class produces a 'main' method that converts the argv in a
+    List of Strings and pass it to the real entry point.
+    """
+    
+    def __init__(self, graph_to_call):
+        self.graph = graph_to_call
+
+    def get_name(self):
+        return 'main'
+
+    def render(self, ilasm):
+        try:
+            ARG0 = self.graph.getargs()[0].concretetype
+        except IndexError:
+            ARG0 = None
+        assert isinstance(ARG0, ootype.List) and ARG0.ITEM is ootype.String,\
+               'Wrong entry point signature: List(String) expected'
+
+        qname = self.cts.graph_to_qname(self.graph)
+        ilasm.emit('findpropstrict', qname)
+        ilasm.emit('callpropvoid', qname, 0)
+        ## ilasm.opcode('pop') # XXX: return this value, if it's an int32
+
+        ## ilasm.call('void [pypylib]pypy.runtime.DebugPrint::close_file()')
+        ## ilasm.opcode('ret')
+        ## ilasm.end_function()
+        self.db.pending_function(self.graph)
+
+class LibraryEntryPoint(BaseEntryPoint):
+    def __init__(self, name, graphs):
+        self.name = name
+        self.graphs = graphs
+
+    def get_name(self):
+        return self.name
+
+    def render(self, ilasm):
+        for graph in self.graphs:
+            self.db.pending_function(graph)