optimise Visitor.ScopeTrackingTransform

Cython/Compiler/Visitor.pxd

@@ -21,6 +21,11 @@ cdef class CythonTransform(VisitorTransform):
     cdef public context
     cdef public current_directives
 
+cdef class ScopeTrackingTransform(CythonTransform):
+    cdef public scope_type
+    cdef public scope_node
+    cdef visit_scope(self, node, scope_type)
+
 cdef class EnvTransform(CythonTransform):
     cdef public list env_stack
 

Cython/Compiler/Visitor.py

@@ -291,8 +291,8 @@ class CythonTransform(VisitorTransform):
 
 class ScopeTrackingTransform(CythonTransform):
     # Keeps track of type of scopes
-    scope_type = None # can be either of 'module', 'function', 'cclass', 'pyclass'
-    scope_node = None
+    #scope_type: can be either of 'module', 'function', 'cclass', 'pyclass', 'struct'
+    #scope_node: the node that owns the current scope
     
     def visit_ModuleNode(self, node):
         self.scope_type = 'module'