| Index: third_party/cython/src/Cython/Compiler/UtilityCode.py
|
| diff --git a/third_party/cython/src/Cython/Compiler/UtilityCode.py b/third_party/cython/src/Cython/Compiler/UtilityCode.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..b7de364f9180616c46bfc252015d6258e1f6a652
|
| --- /dev/null
|
| +++ b/third_party/cython/src/Cython/Compiler/UtilityCode.py
|
| @@ -0,0 +1,177 @@
|
| +from TreeFragment import parse_from_strings, StringParseContext
|
| +import Symtab
|
| +import Naming
|
| +import Code
|
| +
|
| +class NonManglingModuleScope(Symtab.ModuleScope):
|
| +
|
| + def __init__(self, prefix, *args, **kw):
|
| + self.prefix = prefix
|
| + self.cython_scope = None
|
| + Symtab.ModuleScope.__init__(self, *args, **kw)
|
| +
|
| + def add_imported_entry(self, name, entry, pos):
|
| + entry.used = True
|
| + return super(NonManglingModuleScope, self).add_imported_entry(
|
| + name, entry, pos)
|
| +
|
| + def mangle(self, prefix, name=None):
|
| + if name:
|
| + if prefix in (Naming.typeobj_prefix, Naming.func_prefix, Naming.var_prefix, Naming.pyfunc_prefix):
|
| + # Functions, classes etc. gets a manually defined prefix easily
|
| + # manually callable instead (the one passed to CythonUtilityCode)
|
| + prefix = self.prefix
|
| + return "%s%s" % (prefix, name)
|
| + else:
|
| + return Symtab.ModuleScope.mangle(self, prefix)
|
| +
|
| +class CythonUtilityCodeContext(StringParseContext):
|
| + scope = None
|
| +
|
| + def find_module(self, module_name, relative_to = None, pos = None,
|
| + need_pxd = 1):
|
| +
|
| + if module_name != self.module_name:
|
| + if module_name not in self.modules:
|
| + raise AssertionError("Only the cython cimport is supported.")
|
| + else:
|
| + return self.modules[module_name]
|
| +
|
| + if self.scope is None:
|
| + self.scope = NonManglingModuleScope(self.prefix,
|
| + module_name,
|
| + parent_module=None,
|
| + context=self)
|
| +
|
| + return self.scope
|
| +
|
| +
|
| +class CythonUtilityCode(Code.UtilityCodeBase):
|
| + """
|
| + Utility code written in the Cython language itself.
|
| +
|
| + The @cname decorator can set the cname for a function, method of cdef class.
|
| + Functions decorated with @cname('c_func_name') get the given cname.
|
| +
|
| + For cdef classes the rules are as follows:
|
| + obj struct -> <cname>_obj
|
| + obj type ptr -> <cname>_type
|
| + methods -> <class_cname>_<method_cname>
|
| +
|
| + For methods the cname decorator is optional, but without the decorator the
|
| + methods will not be prototyped. See Cython.Compiler.CythonScope and
|
| + tests/run/cythonscope.pyx for examples.
|
| + """
|
| +
|
| + is_cython_utility = True
|
| +
|
| + def __init__(self, impl, name="__pyxutil", prefix="", requires=None,
|
| + file=None, from_scope=None, context=None):
|
| + # 1) We need to delay the parsing/processing, so that all modules can be
|
| + # imported without import loops
|
| + # 2) The same utility code object can be used for multiple source files;
|
| + # while the generated node trees can be altered in the compilation of a
|
| + # single file.
|
| + # Hence, delay any processing until later.
|
| + if context is not None:
|
| + impl = Code.sub_tempita(impl, context, file, name)
|
| + self.impl = impl
|
| + self.name = name
|
| + self.file = file
|
| + self.prefix = prefix
|
| + self.requires = requires or []
|
| + self.from_scope = from_scope
|
| +
|
| + def get_tree(self, entries_only=False, cython_scope=None):
|
| + from AnalysedTreeTransforms import AutoTestDictTransform
|
| + # The AutoTestDictTransform creates the statement "__test__ = {}",
|
| + # which when copied into the main ModuleNode overwrites
|
| + # any __test__ in user code; not desired
|
| + excludes = [AutoTestDictTransform]
|
| +
|
| + import Pipeline, ParseTreeTransforms
|
| + context = CythonUtilityCodeContext(self.name)
|
| + context.prefix = self.prefix
|
| + context.cython_scope = cython_scope
|
| + #context = StringParseContext(self.name)
|
| + tree = parse_from_strings(self.name, self.impl, context=context,
|
| + allow_struct_enum_decorator=True)
|
| + pipeline = Pipeline.create_pipeline(context, 'pyx', exclude_classes=excludes)
|
| +
|
| + if entries_only:
|
| + p = []
|
| + for t in pipeline:
|
| + p.append(t)
|
| + if isinstance(p, ParseTreeTransforms.AnalyseDeclarationsTransform):
|
| + break
|
| +
|
| + pipeline = p
|
| +
|
| + transform = ParseTreeTransforms.CnameDirectivesTransform(context)
|
| + # InterpretCompilerDirectives already does a cdef declarator check
|
| + #before = ParseTreeTransforms.DecoratorTransform
|
| + before = ParseTreeTransforms.InterpretCompilerDirectives
|
| + pipeline = Pipeline.insert_into_pipeline(pipeline, transform,
|
| + before=before)
|
| +
|
| + if self.from_scope:
|
| + def scope_transform(module_node):
|
| + module_node.scope.merge_in(self.from_scope)
|
| + return module_node
|
| +
|
| + transform = ParseTreeTransforms.AnalyseDeclarationsTransform
|
| + pipeline = Pipeline.insert_into_pipeline(pipeline, scope_transform,
|
| + before=transform)
|
| +
|
| + (err, tree) = Pipeline.run_pipeline(pipeline, tree, printtree=False)
|
| + assert not err, err
|
| + return tree
|
| +
|
| + def put_code(self, output):
|
| + pass
|
| +
|
| + @classmethod
|
| + def load_as_string(cls, util_code_name, from_file=None, **kwargs):
|
| + """
|
| + Load a utility code as a string. Returns (proto, implementation)
|
| + """
|
| + util = cls.load(util_code_name, from_file, **kwargs)
|
| + return util.proto, util.impl # keep line numbers => no lstrip()
|
| +
|
| + def declare_in_scope(self, dest_scope, used=False, cython_scope=None,
|
| + whitelist=None):
|
| + """
|
| + Declare all entries from the utility code in dest_scope. Code will only
|
| + be included for used entries. If module_name is given, declare the
|
| + type entries with that name.
|
| + """
|
| + tree = self.get_tree(entries_only=True, cython_scope=cython_scope)
|
| +
|
| + entries = tree.scope.entries
|
| + entries.pop('__name__')
|
| + entries.pop('__file__')
|
| + entries.pop('__builtins__')
|
| + entries.pop('__doc__')
|
| +
|
| + for name, entry in entries.iteritems():
|
| + entry.utility_code_definition = self
|
| + entry.used = used
|
| +
|
| + original_scope = tree.scope
|
| + dest_scope.merge_in(original_scope, merge_unused=True,
|
| + whitelist=whitelist)
|
| + tree.scope = dest_scope
|
| +
|
| + for dep in self.requires:
|
| + if dep.is_cython_utility:
|
| + dep.declare_in_scope(dest_scope)
|
| +
|
| + return original_scope
|
| +
|
| +def declare_declarations_in_scope(declaration_string, env, private_type=True,
|
| + *args, **kwargs):
|
| + """
|
| + Declare some declarations given as Cython code in declaration_string
|
| + in scope env.
|
| + """
|
| + CythonUtilityCode(declaration_string, *args, **kwargs).declare_in_scope(env)
|
|
|