[pypy-svn] r7637 - pypy/trunk/src/goal

[mgedmin at codespeak.net]

Author: mgedmin
Date: Tue Nov 23 18:31:15 2004
New Revision: 7637

Modified:
   pypy/trunk/src/goal/translate_pypy.py
Log:
Mark functions that get or return SomeObject even when annotation fails.



Modified: pypy/trunk/src/goal/translate_pypy.py
==============================================================================
--- pypy/trunk/src/goal/translate_pypy.py	(original)
+++ pypy/trunk/src/goal/translate_pypy.py	Tue Nov 23 18:31:15 2004
@@ -50,13 +50,13 @@
         a.simplify()
         t.frozen = True   # cannot freeze if we don't have annotations
 
-        if not options['-no-mark-some-objects']:
-            find_someobjects(a)
-
 
-def find_someobjects(annotator):
+def find_someobjects(translator):
     """Find all functions in that have SomeObject in their signature."""
-    translator = annotator.translator
+    annotator = translator.annotator
+    if not annotator:
+        return # no annotations available
+
     translator.highlight_functions = {}
 
     def is_someobject(var):
@@ -178,6 +178,10 @@
         from pypy.translator.tool.pygame.flowviewer import TranslatorLayout
         from pypy.translator.tool.pygame.graphdisplay import GraphDisplay
         import pygame
+
+        if not options['-no-mark-some-objects']:
+            find_someobjects(t)
+
         display = GraphDisplay(TranslatorLayout(t))
         async_quit = display.async_quit
         return display.run, async_quit, pygame.quit