From: Robert Bradshaw <robertwb@math.washington.edu>
Date: Sat, 17 Jan 2009 08:39:38 +0000 (-0800)
Subject: Verbose code writing metaclass
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=ff15efef78975abfd96b60165e5c3f6d367e2d12;p=cython.git

Verbose code writing metaclass
---

diff --git a/Cython/Compiler/Nodes.py b/Cython/Compiler/Nodes.py
index b2e3019c..957843bc 100644
--- a/Cython/Compiler/Nodes.py
+++ b/Cython/Compiler/Nodes.py
@@ -69,11 +69,35 @@ def embed_position(pos, docstring):
     doc.encoding = encoding
     return doc
 
+
+from Code import CCodeWriter
+from types import FunctionType
+
+def write_func_call(func):
+    def f(*args, **kwds):
+        if len(args) > 1 and isinstance(args[1], CCodeWriter):
+            node, code = args[:2]
+            code.putln('/* %s.%s %s */' % (node.__class__.__name__, func.__name__, node.pos[1:]))
+        return func(*args, **kwds)
+    return f
+
+class VerboseCodeWriter(type):
+    # Set this as a metaclass to trace function calls in code.
+    def __new__(cls, name, bases, attrs):
+        attrs = dict(attrs)
+        for mname, m in attrs.items():
+            if isinstance(m, FunctionType):
+                attrs[mname] = write_func_call(m)
+        return super(VerboseCodeWriter, cls).__new__(cls, name, bases, attrs)
+
+
 class Node(object):
     #  pos         (string, int, int)   Source file position
     #  is_name     boolean              Is a NameNode
     #  is_literal  boolean              Is a ConstNode
     
+    __metaclass__ = VerboseCodeWriter
+    
     is_name = 0
     is_literal = 0
     temps = None