Spaces:
Build error
Build error
# | |
# Symbol Table | |
# | |
from __future__ import absolute_import | |
import re | |
import copy | |
import operator | |
try: | |
import __builtin__ as builtins | |
except ImportError: # Py3 | |
import builtins | |
from .Errors import warning, error, InternalError | |
from .StringEncoding import EncodedString | |
from . import Options, Naming | |
from . import PyrexTypes | |
from .PyrexTypes import py_object_type, unspecified_type | |
from .TypeSlots import ( | |
pyfunction_signature, pymethod_signature, richcmp_special_methods, | |
get_special_method_signature, get_property_accessor_signature) | |
from . import Future | |
from . import Code | |
iso_c99_keywords = set( | |
['auto', 'break', 'case', 'char', 'const', 'continue', 'default', 'do', | |
'double', 'else', 'enum', 'extern', 'float', 'for', 'goto', 'if', | |
'int', 'long', 'register', 'return', 'short', 'signed', 'sizeof', | |
'static', 'struct', 'switch', 'typedef', 'union', 'unsigned', 'void', | |
'volatile', 'while', | |
'_Bool', '_Complex'', _Imaginary', 'inline', 'restrict']) | |
def c_safe_identifier(cname): | |
# There are some C limitations on struct entry names. | |
if ((cname[:2] == '__' and not (cname.startswith(Naming.pyrex_prefix) | |
or cname in ('__weakref__', '__dict__'))) | |
or cname in iso_c99_keywords): | |
cname = Naming.pyrex_prefix + cname | |
return cname | |
class BufferAux(object): | |
writable_needed = False | |
def __init__(self, buflocal_nd_var, rcbuf_var): | |
self.buflocal_nd_var = buflocal_nd_var | |
self.rcbuf_var = rcbuf_var | |
def __repr__(self): | |
return "<BufferAux %r>" % self.__dict__ | |
class Entry(object): | |
# A symbol table entry in a Scope or ModuleNamespace. | |
# | |
# name string Python name of entity | |
# cname string C name of entity | |
# type PyrexType Type of entity | |
# doc string Doc string | |
# annotation ExprNode PEP 484/526 annotation | |
# init string Initial value | |
# visibility 'private' or 'public' or 'extern' | |
# is_builtin boolean Is an entry in the Python builtins dict | |
# is_cglobal boolean Is a C global variable | |
# is_pyglobal boolean Is a Python module-level variable | |
# or class attribute during | |
# class construction | |
# is_member boolean Is an assigned class member | |
# is_pyclass_attr boolean Is a name in a Python class namespace | |
# is_variable boolean Is a variable | |
# is_cfunction boolean Is a C function | |
# is_cmethod boolean Is a C method of an extension type | |
# is_builtin_cmethod boolean Is a C method of a builtin type (implies is_cmethod) | |
# is_unbound_cmethod boolean Is an unbound C method of an extension type | |
# is_final_cmethod boolean Is non-overridable C method | |
# is_inline_cmethod boolean Is inlined C method | |
# is_anonymous boolean Is a anonymous pyfunction entry | |
# is_type boolean Is a type definition | |
# is_cclass boolean Is an extension class | |
# is_cpp_class boolean Is a C++ class | |
# is_const boolean Is a constant | |
# is_property boolean Is a property of an extension type: | |
# doc_cname string or None C const holding the docstring | |
# getter_cname string C func for getting property | |
# setter_cname string C func for setting or deleting property | |
# is_self_arg boolean Is the "self" arg of an exttype method | |
# is_arg boolean Is the arg of a method | |
# is_local boolean Is a local variable | |
# in_closure boolean Is referenced in an inner scope | |
# in_subscope boolean Belongs to a generator expression scope | |
# is_readonly boolean Can't be assigned to | |
# func_cname string C func implementing Python func | |
# func_modifiers [string] C function modifiers ('inline') | |
# pos position Source position where declared | |
# namespace_cname string If is_pyglobal, the C variable | |
# holding its home namespace | |
# pymethdef_cname string PyMethodDef structure | |
# signature Signature Arg & return types for Python func | |
# as_variable Entry Alternative interpretation of extension | |
# type name or builtin C function as a variable | |
# xdecref_cleanup boolean Use Py_XDECREF for error cleanup | |
# in_cinclude boolean Suppress C declaration code | |
# enum_values [Entry] For enum types, list of values | |
# qualified_name string "modname.funcname" or "modname.classname" | |
# or "modname.classname.funcname" | |
# is_declared_generic boolean Is declared as PyObject * even though its | |
# type is an extension type | |
# as_module None Module scope, if a cimported module | |
# is_inherited boolean Is an inherited attribute of an extension type | |
# pystring_cname string C name of Python version of string literal | |
# is_interned boolean For string const entries, value is interned | |
# is_identifier boolean For string const entries, value is an identifier | |
# used boolean | |
# is_special boolean Is a special method or property accessor | |
# of an extension type | |
# defined_in_pxd boolean Is defined in a .pxd file (not just declared) | |
# api boolean Generate C API for C class or function | |
# utility_code string Utility code needed when this entry is used | |
# | |
# buffer_aux BufferAux or None Extra information needed for buffer variables | |
# inline_func_in_pxd boolean Hacky special case for inline function in pxd file. | |
# Ideally this should not be necessary. | |
# might_overflow boolean In an arithmetic expression that could cause | |
# overflow (used for type inference). | |
# utility_code_definition For some Cython builtins, the utility code | |
# which contains the definition of the entry. | |
# Currently only supported for CythonScope entries. | |
# error_on_uninitialized Have Control Flow issue an error when this entry is | |
# used uninitialized | |
# cf_used boolean Entry is used | |
# is_fused_specialized boolean Whether this entry of a cdef or def function | |
# is a specialization | |
# TODO: utility_code and utility_code_definition serves the same purpose... | |
inline_func_in_pxd = False | |
borrowed = 0 | |
init = "" | |
annotation = None | |
visibility = 'private' | |
is_builtin = 0 | |
is_cglobal = 0 | |
is_pyglobal = 0 | |
is_member = 0 | |
is_pyclass_attr = 0 | |
is_variable = 0 | |
is_cfunction = 0 | |
is_cmethod = 0 | |
is_builtin_cmethod = False | |
is_unbound_cmethod = 0 | |
is_final_cmethod = 0 | |
is_inline_cmethod = 0 | |
is_anonymous = 0 | |
is_type = 0 | |
is_cclass = 0 | |
is_cpp_class = 0 | |
is_const = 0 | |
is_property = 0 | |
doc_cname = None | |
getter_cname = None | |
setter_cname = None | |
is_self_arg = 0 | |
is_arg = 0 | |
is_local = 0 | |
in_closure = 0 | |
from_closure = 0 | |
in_subscope = 0 | |
is_declared_generic = 0 | |
is_readonly = 0 | |
pyfunc_cname = None | |
func_cname = None | |
func_modifiers = [] | |
final_func_cname = None | |
doc = None | |
as_variable = None | |
xdecref_cleanup = 0 | |
in_cinclude = 0 | |
as_module = None | |
is_inherited = 0 | |
pystring_cname = None | |
is_identifier = 0 | |
is_interned = 0 | |
used = 0 | |
is_special = 0 | |
defined_in_pxd = 0 | |
is_implemented = 0 | |
api = 0 | |
utility_code = None | |
is_overridable = 0 | |
buffer_aux = None | |
prev_entry = None | |
might_overflow = 0 | |
fused_cfunction = None | |
is_fused_specialized = False | |
utility_code_definition = None | |
needs_property = False | |
in_with_gil_block = 0 | |
from_cython_utility_code = None | |
error_on_uninitialized = False | |
cf_used = True | |
outer_entry = None | |
def __init__(self, name, cname, type, pos = None, init = None): | |
self.name = name | |
self.cname = cname | |
self.type = type | |
self.pos = pos | |
self.init = init | |
self.overloaded_alternatives = [] | |
self.cf_assignments = [] | |
self.cf_references = [] | |
self.inner_entries = [] | |
self.defining_entry = self | |
def __repr__(self): | |
return "%s(<%x>, name=%s, type=%s)" % (type(self).__name__, id(self), self.name, self.type) | |
def already_declared_here(self): | |
error(self.pos, "Previous declaration is here") | |
def redeclared(self, pos): | |
error(pos, "'%s' does not match previous declaration" % self.name) | |
self.already_declared_here() | |
def all_alternatives(self): | |
return [self] + self.overloaded_alternatives | |
def all_entries(self): | |
return [self] + self.inner_entries | |
def __lt__(left, right): | |
if isinstance(left, Entry) and isinstance(right, Entry): | |
return (left.name, left.cname) < (right.name, right.cname) | |
else: | |
return NotImplemented | |
class InnerEntry(Entry): | |
""" | |
An entry in a closure scope that represents the real outer Entry. | |
""" | |
from_closure = True | |
def __init__(self, outer_entry, scope): | |
Entry.__init__(self, outer_entry.name, | |
outer_entry.cname, | |
outer_entry.type, | |
outer_entry.pos) | |
self.outer_entry = outer_entry | |
self.scope = scope | |
# share state with (outermost) defining entry | |
outermost_entry = outer_entry | |
while outermost_entry.outer_entry: | |
outermost_entry = outermost_entry.outer_entry | |
self.defining_entry = outermost_entry | |
self.inner_entries = outermost_entry.inner_entries | |
self.cf_assignments = outermost_entry.cf_assignments | |
self.cf_references = outermost_entry.cf_references | |
self.overloaded_alternatives = outermost_entry.overloaded_alternatives | |
self.inner_entries.append(self) | |
def __getattr__(self, name): | |
if name.startswith('__'): | |
# we wouldn't have been called if it was there | |
raise AttributeError(name) | |
return getattr(self.defining_entry, name) | |
def all_entries(self): | |
return self.defining_entry.all_entries() | |
class Scope(object): | |
# name string Unqualified name | |
# outer_scope Scope or None Enclosing scope | |
# entries {string : Entry} Python name to entry, non-types | |
# const_entries [Entry] Constant entries | |
# type_entries [Entry] Struct/union/enum/typedef/exttype entries | |
# sue_entries [Entry] Struct/union/enum entries | |
# arg_entries [Entry] Function argument entries | |
# var_entries [Entry] User-defined variable entries | |
# pyfunc_entries [Entry] Python function entries | |
# cfunc_entries [Entry] C function entries | |
# c_class_entries [Entry] All extension type entries | |
# cname_to_entry {string : Entry} Temp cname to entry mapping | |
# return_type PyrexType or None Return type of function owning scope | |
# is_builtin_scope boolean Is the builtin scope of Python/Cython | |
# is_py_class_scope boolean Is a Python class scope | |
# is_c_class_scope boolean Is an extension type scope | |
# is_closure_scope boolean Is a closure scope | |
# is_passthrough boolean Outer scope is passed directly | |
# is_cpp_class_scope boolean Is a C++ class scope | |
# is_property_scope boolean Is a extension type property scope | |
# scope_prefix string Disambiguator for C names | |
# in_cinclude boolean Suppress C declaration code | |
# qualified_name string "modname" or "modname.classname" | |
# Python strings in this scope | |
# nogil boolean In a nogil section | |
# directives dict Helper variable for the recursive | |
# analysis, contains directive values. | |
# is_internal boolean Is only used internally (simpler setup) | |
is_builtin_scope = 0 | |
is_py_class_scope = 0 | |
is_c_class_scope = 0 | |
is_closure_scope = 0 | |
is_genexpr_scope = 0 | |
is_passthrough = 0 | |
is_cpp_class_scope = 0 | |
is_property_scope = 0 | |
is_module_scope = 0 | |
is_internal = 0 | |
scope_prefix = "" | |
in_cinclude = 0 | |
nogil = 0 | |
fused_to_specific = None | |
return_type = None | |
def __init__(self, name, outer_scope, parent_scope): | |
# The outer_scope is the next scope in the lookup chain. | |
# The parent_scope is used to derive the qualified name of this scope. | |
self.name = name | |
self.outer_scope = outer_scope | |
self.parent_scope = parent_scope | |
mangled_name = "%d%s_" % (len(name), name.replace('.', '_dot_')) | |
qual_scope = self.qualifying_scope() | |
if qual_scope: | |
self.qualified_name = qual_scope.qualify_name(name) | |
self.scope_prefix = qual_scope.scope_prefix + mangled_name | |
else: | |
self.qualified_name = EncodedString(name) | |
self.scope_prefix = mangled_name | |
self.entries = {} | |
self.subscopes = set() | |
self.const_entries = [] | |
self.type_entries = [] | |
self.sue_entries = [] | |
self.arg_entries = [] | |
self.var_entries = [] | |
self.pyfunc_entries = [] | |
self.cfunc_entries = [] | |
self.c_class_entries = [] | |
self.defined_c_classes = [] | |
self.imported_c_classes = {} | |
self.cname_to_entry = {} | |
self.string_to_entry = {} | |
self.identifier_to_entry = {} | |
self.num_to_entry = {} | |
self.obj_to_entry = {} | |
self.buffer_entries = [] | |
self.lambda_defs = [] | |
self.id_counters = {} | |
def __deepcopy__(self, memo): | |
return self | |
def merge_in(self, other, merge_unused=True, whitelist=None): | |
# Use with care... | |
entries = [] | |
for name, entry in other.entries.items(): | |
if not whitelist or name in whitelist: | |
if entry.used or merge_unused: | |
entries.append((name, entry)) | |
self.entries.update(entries) | |
for attr in ('const_entries', | |
'type_entries', | |
'sue_entries', | |
'arg_entries', | |
'var_entries', | |
'pyfunc_entries', | |
'cfunc_entries', | |
'c_class_entries'): | |
self_entries = getattr(self, attr) | |
names = set(e.name for e in self_entries) | |
for entry in getattr(other, attr): | |
if (entry.used or merge_unused) and entry.name not in names: | |
self_entries.append(entry) | |
def __str__(self): | |
return "<%s %s>" % (self.__class__.__name__, self.qualified_name) | |
def qualifying_scope(self): | |
return self.parent_scope | |
def mangle(self, prefix, name = None): | |
if name: | |
return "%s%s%s" % (prefix, self.scope_prefix, name) | |
else: | |
return self.parent_scope.mangle(prefix, self.name) | |
def mangle_internal(self, name): | |
# Mangle an internal name so as not to clash with any | |
# user-defined name in this scope. | |
prefix = "%s%s_" % (Naming.pyrex_prefix, name) | |
return self.mangle(prefix) | |
#return self.parent_scope.mangle(prefix, self.name) | |
def mangle_class_private_name(self, name): | |
if self.parent_scope: | |
return self.parent_scope.mangle_class_private_name(name) | |
return name | |
def next_id(self, name=None): | |
# Return a cname fragment that is unique for this module | |
counters = self.global_scope().id_counters | |
try: | |
count = counters[name] + 1 | |
except KeyError: | |
count = 0 | |
counters[name] = count | |
if name: | |
if not count: | |
# unique names don't need a suffix, reoccurrences will get one | |
return name | |
return '%s%d' % (name, count) | |
else: | |
return '%d' % count | |
def global_scope(self): | |
""" Return the module-level scope containing this scope. """ | |
return self.outer_scope.global_scope() | |
def builtin_scope(self): | |
""" Return the module-level scope containing this scope. """ | |
return self.outer_scope.builtin_scope() | |
def iter_local_scopes(self): | |
yield self | |
if self.subscopes: | |
for scope in sorted(self.subscopes, key=operator.attrgetter('scope_prefix')): | |
yield scope | |
def declare(self, name, cname, type, pos, visibility, shadow = 0, is_type = 0, create_wrapper = 0): | |
# Create new entry, and add to dictionary if | |
# name is not None. Reports a warning if already | |
# declared. | |
if type.is_buffer and not isinstance(self, LocalScope): # and not is_type: | |
error(pos, 'Buffer types only allowed as function local variables') | |
if not self.in_cinclude and cname and re.match("^_[_A-Z]+$", cname): | |
# See http://www.gnu.org/software/libc/manual/html_node/Reserved-Names.html#Reserved-Names | |
warning(pos, "'%s' is a reserved name in C." % cname, -1) | |
entries = self.entries | |
if name and name in entries and not shadow: | |
old_entry = entries[name] | |
# Reject redeclared C++ functions only if they have the same type signature. | |
cpp_override_allowed = False | |
if type.is_cfunction and old_entry.type.is_cfunction and self.is_cpp(): | |
for alt_entry in old_entry.all_alternatives(): | |
if type == alt_entry.type: | |
if name == '<init>' and not type.args: | |
# Cython pre-declares the no-args constructor - allow later user definitions. | |
cpp_override_allowed = True | |
break | |
else: | |
cpp_override_allowed = True | |
if cpp_override_allowed: | |
# C++ function/method overrides with different signatures are ok. | |
pass | |
elif self.is_cpp_class_scope and entries[name].is_inherited: | |
# Likewise ignore inherited classes. | |
pass | |
elif visibility == 'extern': | |
# Silenced outside of "cdef extern" blocks, until we have a safe way to | |
# prevent pxd-defined cpdef functions from ending up here. | |
warning(pos, "'%s' redeclared " % name, 1 if self.in_cinclude else 0) | |
elif visibility != 'ignore': | |
error(pos, "'%s' redeclared " % name) | |
entries[name].already_declared_here() | |
entry = Entry(name, cname, type, pos = pos) | |
entry.in_cinclude = self.in_cinclude | |
entry.create_wrapper = create_wrapper | |
if name: | |
entry.qualified_name = self.qualify_name(name) | |
# if name in entries and self.is_cpp(): | |
# entries[name].overloaded_alternatives.append(entry) | |
# else: | |
# entries[name] = entry | |
if not shadow: | |
entries[name] = entry | |
if type.is_memoryviewslice: | |
from . import MemoryView | |
entry.init = MemoryView.memslice_entry_init | |
entry.scope = self | |
entry.visibility = visibility | |
return entry | |
def qualify_name(self, name): | |
return EncodedString("%s.%s" % (self.qualified_name, name)) | |
def declare_const(self, name, type, value, pos, cname = None, visibility = 'private', api = 0, create_wrapper = 0): | |
# Add an entry for a named constant. | |
if not cname: | |
if self.in_cinclude or (visibility == 'public' or api): | |
cname = name | |
else: | |
cname = self.mangle(Naming.enum_prefix, name) | |
entry = self.declare(name, cname, type, pos, visibility, create_wrapper = create_wrapper) | |
entry.is_const = 1 | |
entry.value_node = value | |
return entry | |
def declare_type(self, name, type, pos, | |
cname = None, visibility = 'private', api = 0, defining = 1, | |
shadow = 0, template = 0): | |
# Add an entry for a type definition. | |
if not cname: | |
cname = name | |
entry = self.declare(name, cname, type, pos, visibility, shadow, | |
is_type=True) | |
entry.is_type = 1 | |
entry.api = api | |
if defining: | |
self.type_entries.append(entry) | |
if not template: | |
type.entry = entry | |
# here we would set as_variable to an object representing this type | |
return entry | |
def declare_typedef(self, name, base_type, pos, cname = None, | |
visibility = 'private', api = 0): | |
if not cname: | |
if self.in_cinclude or (visibility != 'private' or api): | |
cname = name | |
else: | |
cname = self.mangle(Naming.type_prefix, name) | |
try: | |
if self.is_cpp_class_scope: | |
namespace = self.outer_scope.lookup(self.name).type | |
else: | |
namespace = None | |
type = PyrexTypes.create_typedef_type(name, base_type, cname, | |
(visibility == 'extern'), | |
namespace) | |
except ValueError as e: | |
error(pos, e.args[0]) | |
type = PyrexTypes.error_type | |
entry = self.declare_type(name, type, pos, cname, | |
visibility = visibility, api = api) | |
type.qualified_name = entry.qualified_name | |
return entry | |
def declare_struct_or_union(self, name, kind, scope, | |
typedef_flag, pos, cname = None, | |
visibility = 'private', api = 0, | |
packed = False): | |
# Add an entry for a struct or union definition. | |
if not cname: | |
if self.in_cinclude or (visibility == 'public' or api): | |
cname = name | |
else: | |
cname = self.mangle(Naming.type_prefix, name) | |
entry = self.lookup_here(name) | |
if not entry: | |
type = PyrexTypes.CStructOrUnionType( | |
name, kind, scope, typedef_flag, cname, packed) | |
entry = self.declare_type(name, type, pos, cname, | |
visibility = visibility, api = api, | |
defining = scope is not None) | |
self.sue_entries.append(entry) | |
type.entry = entry | |
else: | |
if not (entry.is_type and entry.type.is_struct_or_union | |
and entry.type.kind == kind): | |
warning(pos, "'%s' redeclared " % name, 0) | |
elif scope and entry.type.scope: | |
warning(pos, "'%s' already defined (ignoring second definition)" % name, 0) | |
else: | |
self.check_previous_typedef_flag(entry, typedef_flag, pos) | |
self.check_previous_visibility(entry, visibility, pos) | |
if scope: | |
entry.type.scope = scope | |
self.type_entries.append(entry) | |
if self.is_cpp_class_scope: | |
entry.type.namespace = self.outer_scope.lookup(self.name).type | |
return entry | |
def declare_cpp_class(self, name, scope, | |
pos, cname = None, base_classes = (), | |
visibility = 'extern', templates = None): | |
if cname is None: | |
if self.in_cinclude or (visibility != 'private'): | |
cname = name | |
else: | |
cname = self.mangle(Naming.type_prefix, name) | |
base_classes = list(base_classes) | |
entry = self.lookup_here(name) | |
if not entry: | |
type = PyrexTypes.CppClassType( | |
name, scope, cname, base_classes, templates = templates) | |
entry = self.declare_type(name, type, pos, cname, | |
visibility = visibility, defining = scope is not None) | |
self.sue_entries.append(entry) | |
else: | |
if not (entry.is_type and entry.type.is_cpp_class): | |
error(pos, "'%s' redeclared " % name) | |
entry.already_declared_here() | |
return None | |
elif scope and entry.type.scope: | |
warning(pos, "'%s' already defined (ignoring second definition)" % name, 0) | |
else: | |
if scope: | |
entry.type.scope = scope | |
self.type_entries.append(entry) | |
if base_classes: | |
if entry.type.base_classes and entry.type.base_classes != base_classes: | |
error(pos, "Base type does not match previous declaration") | |
entry.already_declared_here() | |
else: | |
entry.type.base_classes = base_classes | |
if templates or entry.type.templates: | |
if templates != entry.type.templates: | |
error(pos, "Template parameters do not match previous declaration") | |
entry.already_declared_here() | |
def declare_inherited_attributes(entry, base_classes): | |
for base_class in base_classes: | |
if base_class is PyrexTypes.error_type: | |
continue | |
if base_class.scope is None: | |
error(pos, "Cannot inherit from incomplete type") | |
else: | |
declare_inherited_attributes(entry, base_class.base_classes) | |
entry.type.scope.declare_inherited_cpp_attributes(base_class) | |
if scope: | |
declare_inherited_attributes(entry, base_classes) | |
scope.declare_var(name="this", cname="this", type=PyrexTypes.CPtrType(entry.type), pos=entry.pos) | |
if self.is_cpp_class_scope: | |
entry.type.namespace = self.outer_scope.lookup(self.name).type | |
return entry | |
def check_previous_typedef_flag(self, entry, typedef_flag, pos): | |
if typedef_flag != entry.type.typedef_flag: | |
error(pos, "'%s' previously declared using '%s'" % ( | |
entry.name, ("cdef", "ctypedef")[entry.type.typedef_flag])) | |
def check_previous_visibility(self, entry, visibility, pos): | |
if entry.visibility != visibility: | |
error(pos, "'%s' previously declared as '%s'" % ( | |
entry.name, entry.visibility)) | |
def declare_enum(self, name, pos, cname, typedef_flag, | |
visibility = 'private', api = 0, create_wrapper = 0): | |
if name: | |
if not cname: | |
if (self.in_cinclude or visibility == 'public' | |
or visibility == 'extern' or api): | |
cname = name | |
else: | |
cname = self.mangle(Naming.type_prefix, name) | |
if self.is_cpp_class_scope: | |
namespace = self.outer_scope.lookup(self.name).type | |
else: | |
namespace = None | |
type = PyrexTypes.CEnumType(name, cname, typedef_flag, namespace) | |
else: | |
type = PyrexTypes.c_anon_enum_type | |
entry = self.declare_type(name, type, pos, cname = cname, | |
visibility = visibility, api = api) | |
entry.create_wrapper = create_wrapper | |
entry.enum_values = [] | |
self.sue_entries.append(entry) | |
return entry | |
def declare_tuple_type(self, pos, components): | |
return self.outer_scope.declare_tuple_type(pos, components) | |
def declare_var(self, name, type, pos, | |
cname = None, visibility = 'private', | |
api = 0, in_pxd = 0, is_cdef = 0): | |
# Add an entry for a variable. | |
if not cname: | |
if visibility != 'private' or api: | |
cname = name | |
else: | |
cname = self.mangle(Naming.var_prefix, name) | |
if type.is_cpp_class and visibility != 'extern': | |
type.check_nullary_constructor(pos) | |
entry = self.declare(name, cname, type, pos, visibility) | |
entry.is_variable = 1 | |
if in_pxd and visibility != 'extern': | |
entry.defined_in_pxd = 1 | |
entry.used = 1 | |
if api: | |
entry.api = 1 | |
entry.used = 1 | |
return entry | |
def declare_builtin(self, name, pos): | |
return self.outer_scope.declare_builtin(name, pos) | |
def _declare_pyfunction(self, name, pos, visibility='extern', entry=None): | |
if entry and not entry.type.is_cfunction: | |
error(pos, "'%s' already declared" % name) | |
error(entry.pos, "Previous declaration is here") | |
entry = self.declare_var(name, py_object_type, pos, visibility=visibility) | |
entry.signature = pyfunction_signature | |
self.pyfunc_entries.append(entry) | |
return entry | |
def declare_pyfunction(self, name, pos, allow_redefine=False, visibility='extern'): | |
# Add an entry for a Python function. | |
entry = self.lookup_here(name) | |
if not allow_redefine: | |
return self._declare_pyfunction(name, pos, visibility=visibility, entry=entry) | |
if entry: | |
if entry.type.is_unspecified: | |
entry.type = py_object_type | |
elif entry.type is not py_object_type: | |
return self._declare_pyfunction(name, pos, visibility=visibility, entry=entry) | |
else: # declare entry stub | |
self.declare_var(name, py_object_type, pos, visibility=visibility) | |
entry = self.declare_var(None, py_object_type, pos, | |
cname=name, visibility='private') | |
entry.name = EncodedString(name) | |
entry.qualified_name = self.qualify_name(name) | |
entry.signature = pyfunction_signature | |
entry.is_anonymous = True | |
return entry | |
def declare_lambda_function(self, lambda_name, pos): | |
# Add an entry for an anonymous Python function. | |
func_cname = self.mangle(Naming.lambda_func_prefix + u'funcdef_', lambda_name) | |
pymethdef_cname = self.mangle(Naming.lambda_func_prefix + u'methdef_', lambda_name) | |
qualified_name = self.qualify_name(lambda_name) | |
entry = self.declare(None, func_cname, py_object_type, pos, 'private') | |
entry.name = lambda_name | |
entry.qualified_name = qualified_name | |
entry.pymethdef_cname = pymethdef_cname | |
entry.func_cname = func_cname | |
entry.signature = pyfunction_signature | |
entry.is_anonymous = True | |
return entry | |
def add_lambda_def(self, def_node): | |
self.lambda_defs.append(def_node) | |
def register_pyfunction(self, entry): | |
self.pyfunc_entries.append(entry) | |
def declare_cfunction(self, name, type, pos, | |
cname=None, visibility='private', api=0, in_pxd=0, | |
defining=0, modifiers=(), utility_code=None, overridable=False): | |
# Add an entry for a C function. | |
if not cname: | |
if visibility != 'private' or api: | |
cname = name | |
else: | |
cname = self.mangle(Naming.func_prefix, name) | |
entry = self.lookup_here(name) | |
if entry: | |
if not in_pxd and visibility != entry.visibility and visibility == 'extern': | |
# Previously declared, but now extern => treat this | |
# as implementing the function, using the new cname | |
defining = True | |
visibility = entry.visibility | |
entry.cname = cname | |
entry.func_cname = cname | |
if visibility != 'private' and visibility != entry.visibility: | |
warning(pos, "Function '%s' previously declared as '%s', now as '%s'" % (name, entry.visibility, visibility), 1) | |
if overridable != entry.is_overridable: | |
warning(pos, "Function '%s' previously declared as '%s'" % ( | |
name, 'cpdef' if overridable else 'cdef'), 1) | |
if entry.type.same_as(type): | |
# Fix with_gil vs nogil. | |
entry.type = entry.type.with_with_gil(type.with_gil) | |
else: | |
if visibility == 'extern' and entry.visibility == 'extern': | |
can_override = False | |
if self.is_cpp(): | |
can_override = True | |
elif cname: | |
# if all alternatives have different cnames, | |
# it's safe to allow signature overrides | |
for alt_entry in entry.all_alternatives(): | |
if not alt_entry.cname or cname == alt_entry.cname: | |
break # cname not unique! | |
else: | |
can_override = True | |
if can_override: | |
temp = self.add_cfunction(name, type, pos, cname, visibility, modifiers) | |
temp.overloaded_alternatives = entry.all_alternatives() | |
entry = temp | |
else: | |
warning(pos, "Function signature does not match previous declaration", 1) | |
entry.type = type | |
elif not in_pxd and entry.defined_in_pxd and type.compatible_signature_with(entry.type): | |
# TODO: check that this was done by a signature optimisation and not a user error. | |
#warning(pos, "Function signature does not match previous declaration", 1) | |
entry.type = type | |
else: | |
error(pos, "Function signature does not match previous declaration") | |
else: | |
entry = self.add_cfunction(name, type, pos, cname, visibility, modifiers) | |
entry.func_cname = cname | |
entry.is_overridable = overridable | |
if in_pxd and visibility != 'extern': | |
entry.defined_in_pxd = 1 | |
if api: | |
entry.api = 1 | |
if not defining and not in_pxd and visibility != 'extern': | |
error(pos, "Non-extern C function '%s' declared but not defined" % name) | |
if defining: | |
entry.is_implemented = True | |
if modifiers: | |
entry.func_modifiers = modifiers | |
if utility_code: | |
assert not entry.utility_code, "duplicate utility code definition in entry %s (%s)" % (name, cname) | |
entry.utility_code = utility_code | |
if overridable: | |
# names of cpdef functions can be used as variables and can be assigned to | |
var_entry = Entry(name, cname, py_object_type) # FIXME: cname? | |
var_entry.qualified_name = self.qualify_name(name) | |
var_entry.is_variable = 1 | |
var_entry.is_pyglobal = 1 | |
var_entry.scope = entry.scope | |
entry.as_variable = var_entry | |
type.entry = entry | |
return entry | |
def add_cfunction(self, name, type, pos, cname, visibility, modifiers, inherited=False): | |
# Add a C function entry without giving it a func_cname. | |
entry = self.declare(name, cname, type, pos, visibility) | |
entry.is_cfunction = 1 | |
if modifiers: | |
entry.func_modifiers = modifiers | |
if inherited or type.is_fused: | |
self.cfunc_entries.append(entry) | |
else: | |
# For backwards compatibility reasons, we must keep all non-fused methods | |
# before all fused methods, but separately for each type. | |
i = len(self.cfunc_entries) | |
for cfunc_entry in reversed(self.cfunc_entries): | |
if cfunc_entry.is_inherited or not cfunc_entry.type.is_fused: | |
break | |
i -= 1 | |
self.cfunc_entries.insert(i, entry) | |
return entry | |
def find(self, name, pos): | |
# Look up name, report error if not found. | |
entry = self.lookup(name) | |
if entry: | |
return entry | |
else: | |
error(pos, "'%s' is not declared" % name) | |
def find_imported_module(self, path, pos): | |
# Look up qualified name, must be a module, report error if not found. | |
# Path is a list of names. | |
scope = self | |
for name in path: | |
entry = scope.find(name, pos) | |
if not entry: | |
return None | |
if entry.as_module: | |
scope = entry.as_module | |
else: | |
error(pos, "'%s' is not a cimported module" % '.'.join(path)) | |
return None | |
return scope | |
def lookup(self, name): | |
# Look up name in this scope or an enclosing one. | |
# Return None if not found. | |
return (self.lookup_here(name) | |
or (self.outer_scope and self.outer_scope.lookup(name)) | |
or None) | |
def lookup_here(self, name): | |
# Look up in this scope only, return None if not found. | |
return self.entries.get(name, None) | |
def lookup_target(self, name): | |
# Look up name in this scope only. Declare as Python | |
# variable if not found. | |
entry = self.lookup_here(name) | |
if not entry: | |
entry = self.declare_var(name, py_object_type, None) | |
return entry | |
def lookup_type(self, name): | |
entry = self.lookup(name) | |
if entry and entry.is_type: | |
if entry.type.is_fused and self.fused_to_specific: | |
return entry.type.specialize(self.fused_to_specific) | |
return entry.type | |
def lookup_operator(self, operator, operands): | |
if operands[0].type.is_cpp_class: | |
obj_type = operands[0].type | |
method = obj_type.scope.lookup("operator%s" % operator) | |
if method is not None: | |
arg_types = [arg.type for arg in operands[1:]] | |
res = PyrexTypes.best_match([arg.type for arg in operands[1:]], | |
method.all_alternatives()) | |
if res is not None: | |
return res | |
function = self.lookup("operator%s" % operator) | |
function_alternatives = [] | |
if function is not None: | |
function_alternatives = function.all_alternatives() | |
# look-up nonmember methods listed within a class | |
method_alternatives = [] | |
if len(operands)==2: # binary operators only | |
for n in range(2): | |
if operands[n].type.is_cpp_class: | |
obj_type = operands[n].type | |
method = obj_type.scope.lookup("operator%s" % operator) | |
if method is not None: | |
method_alternatives += method.all_alternatives() | |
if (not method_alternatives) and (not function_alternatives): | |
return None | |
# select the unique alternatives | |
all_alternatives = list(set(method_alternatives + function_alternatives)) | |
return PyrexTypes.best_match([arg.type for arg in operands], | |
all_alternatives) | |
def lookup_operator_for_types(self, pos, operator, types): | |
from .Nodes import Node | |
class FakeOperand(Node): | |
pass | |
operands = [FakeOperand(pos, type=type) for type in types] | |
return self.lookup_operator(operator, operands) | |
def use_utility_code(self, new_code): | |
self.global_scope().use_utility_code(new_code) | |
def use_entry_utility_code(self, entry): | |
self.global_scope().use_entry_utility_code(entry) | |
def defines_any(self, names): | |
# Test whether any of the given names are defined in this scope. | |
for name in names: | |
if name in self.entries: | |
return 1 | |
return 0 | |
def defines_any_special(self, names): | |
# Test whether any of the given names are defined as special methods in this scope. | |
for name in names: | |
if name in self.entries and self.entries[name].is_special: | |
return 1 | |
return 0 | |
def infer_types(self): | |
from .TypeInference import get_type_inferer | |
get_type_inferer().infer_types(self) | |
def is_cpp(self): | |
outer = self.outer_scope | |
if outer is None: | |
return False | |
else: | |
return outer.is_cpp() | |
def add_include_file(self, filename, verbatim_include=None, late=False): | |
self.outer_scope.add_include_file(filename, verbatim_include, late) | |
class PreImportScope(Scope): | |
namespace_cname = Naming.preimport_cname | |
def __init__(self): | |
Scope.__init__(self, Options.pre_import, None, None) | |
def declare_builtin(self, name, pos): | |
entry = self.declare(name, name, py_object_type, pos, 'private') | |
entry.is_variable = True | |
entry.is_pyglobal = True | |
return entry | |
class BuiltinScope(Scope): | |
# The builtin namespace. | |
is_builtin_scope = True | |
def __init__(self): | |
if Options.pre_import is None: | |
Scope.__init__(self, "__builtin__", None, None) | |
else: | |
Scope.__init__(self, "__builtin__", PreImportScope(), None) | |
self.type_names = {} | |
for name, definition in sorted(self.builtin_entries.items()): | |
cname, type = definition | |
self.declare_var(name, type, None, cname) | |
def lookup(self, name, language_level=None, str_is_str=None): | |
# 'language_level' and 'str_is_str' are passed by ModuleScope | |
if name == 'str': | |
if str_is_str is None: | |
str_is_str = language_level in (None, 2) | |
if not str_is_str: | |
name = 'unicode' | |
return Scope.lookup(self, name) | |
def declare_builtin(self, name, pos): | |
if not hasattr(builtins, name): | |
if self.outer_scope is not None: | |
return self.outer_scope.declare_builtin(name, pos) | |
else: | |
if Options.error_on_unknown_names: | |
error(pos, "undeclared name not builtin: %s" % name) | |
else: | |
warning(pos, "undeclared name not builtin: %s" % name, 2) | |
def declare_builtin_cfunction(self, name, type, cname, python_equiv=None, utility_code=None): | |
# If python_equiv == "*", the Python equivalent has the same name | |
# as the entry, otherwise it has the name specified by python_equiv. | |
name = EncodedString(name) | |
entry = self.declare_cfunction(name, type, None, cname, visibility='extern', | |
utility_code=utility_code) | |
if python_equiv: | |
if python_equiv == "*": | |
python_equiv = name | |
else: | |
python_equiv = EncodedString(python_equiv) | |
var_entry = Entry(python_equiv, python_equiv, py_object_type) | |
var_entry.qualified_name = self.qualify_name(name) | |
var_entry.is_variable = 1 | |
var_entry.is_builtin = 1 | |
var_entry.utility_code = utility_code | |
var_entry.scope = entry.scope | |
entry.as_variable = var_entry | |
return entry | |
def declare_builtin_type(self, name, cname, utility_code = None, objstruct_cname = None): | |
name = EncodedString(name) | |
type = PyrexTypes.BuiltinObjectType(name, cname, objstruct_cname) | |
scope = CClassScope(name, outer_scope=None, visibility='extern') | |
scope.directives = {} | |
if name == 'bool': | |
type.is_final_type = True | |
type.set_scope(scope) | |
self.type_names[name] = 1 | |
entry = self.declare_type(name, type, None, visibility='extern') | |
entry.utility_code = utility_code | |
var_entry = Entry(name = entry.name, | |
type = self.lookup('type').type, # make sure "type" is the first type declared... | |
pos = entry.pos, | |
cname = entry.type.typeptr_cname) | |
var_entry.qualified_name = self.qualify_name(name) | |
var_entry.is_variable = 1 | |
var_entry.is_cglobal = 1 | |
var_entry.is_readonly = 1 | |
var_entry.is_builtin = 1 | |
var_entry.utility_code = utility_code | |
var_entry.scope = self | |
if Options.cache_builtins: | |
var_entry.is_const = True | |
entry.as_variable = var_entry | |
return type | |
def builtin_scope(self): | |
return self | |
builtin_entries = { | |
"type": ["((PyObject*)&PyType_Type)", py_object_type], | |
"bool": ["((PyObject*)&PyBool_Type)", py_object_type], | |
"int": ["((PyObject*)&PyInt_Type)", py_object_type], | |
"long": ["((PyObject*)&PyLong_Type)", py_object_type], | |
"float": ["((PyObject*)&PyFloat_Type)", py_object_type], | |
"complex":["((PyObject*)&PyComplex_Type)", py_object_type], | |
"bytes": ["((PyObject*)&PyBytes_Type)", py_object_type], | |
"bytearray": ["((PyObject*)&PyByteArray_Type)", py_object_type], | |
"str": ["((PyObject*)&PyString_Type)", py_object_type], | |
"unicode":["((PyObject*)&PyUnicode_Type)", py_object_type], | |
"tuple": ["((PyObject*)&PyTuple_Type)", py_object_type], | |
"list": ["((PyObject*)&PyList_Type)", py_object_type], | |
"dict": ["((PyObject*)&PyDict_Type)", py_object_type], | |
"set": ["((PyObject*)&PySet_Type)", py_object_type], | |
"frozenset": ["((PyObject*)&PyFrozenSet_Type)", py_object_type], | |
"slice": ["((PyObject*)&PySlice_Type)", py_object_type], | |
# "file": ["((PyObject*)&PyFile_Type)", py_object_type], # not in Py3 | |
"None": ["Py_None", py_object_type], | |
"False": ["Py_False", py_object_type], | |
"True": ["Py_True", py_object_type], | |
} | |
const_counter = 1 # As a temporary solution for compiling code in pxds | |
class ModuleScope(Scope): | |
# module_name string Python name of the module | |
# module_cname string C name of Python module object | |
# #module_dict_cname string C name of module dict object | |
# method_table_cname string C name of method table | |
# doc string Module doc string | |
# doc_cname string C name of module doc string | |
# utility_code_list [UtilityCode] Queuing utility codes for forwarding to Code.py | |
# c_includes {key: IncludeCode} C headers or verbatim code to be generated | |
# See process_include() for more documentation | |
# string_to_entry {string : Entry} Map string const to entry | |
# identifier_to_entry {string : Entry} Map identifier string const to entry | |
# context Context | |
# parent_module Scope Parent in the import namespace | |
# module_entries {string : Entry} For cimport statements | |
# type_names {string : 1} Set of type names (used during parsing) | |
# included_files [string] Cython sources included with 'include' | |
# pxd_file_loaded boolean Corresponding .pxd file has been processed | |
# cimported_modules [ModuleScope] Modules imported with cimport | |
# types_imported {PyrexType} Set of types for which import code generated | |
# has_import_star boolean Module contains import * | |
# cpp boolean Compiling a C++ file | |
# is_cython_builtin boolean Is this the Cython builtin scope (or a child scope) | |
# is_package boolean Is this a package module? (__init__) | |
is_module_scope = 1 | |
has_import_star = 0 | |
is_cython_builtin = 0 | |
old_style_globals = 0 | |
def __init__(self, name, parent_module, context): | |
from . import Builtin | |
self.parent_module = parent_module | |
outer_scope = Builtin.builtin_scope | |
Scope.__init__(self, name, outer_scope, parent_module) | |
if name == "__init__": | |
# Treat Spam/__init__.pyx specially, so that when Python loads | |
# Spam/__init__.so, initSpam() is defined. | |
self.module_name = parent_module.module_name | |
self.is_package = True | |
else: | |
self.module_name = name | |
self.is_package = False | |
self.module_name = EncodedString(self.module_name) | |
self.context = context | |
self.module_cname = Naming.module_cname | |
self.module_dict_cname = Naming.moddict_cname | |
self.method_table_cname = Naming.methtable_cname | |
self.doc = "" | |
self.doc_cname = Naming.moddoc_cname | |
self.utility_code_list = [] | |
self.module_entries = {} | |
self.c_includes = {} | |
self.type_names = dict(outer_scope.type_names) | |
self.pxd_file_loaded = 0 | |
self.cimported_modules = [] | |
self.types_imported = set() | |
self.included_files = [] | |
self.has_extern_class = 0 | |
self.cached_builtins = [] | |
self.undeclared_cached_builtins = [] | |
self.namespace_cname = self.module_cname | |
self._cached_tuple_types = {} | |
for var_name in ['__builtins__', '__name__', '__file__', '__doc__', '__path__', | |
'__spec__', '__loader__', '__package__', '__cached__']: | |
self.declare_var(EncodedString(var_name), py_object_type, None) | |
self.process_include(Code.IncludeCode("Python.h", initial=True)) | |
def qualifying_scope(self): | |
return self.parent_module | |
def global_scope(self): | |
return self | |
def lookup(self, name, language_level=None, str_is_str=None): | |
entry = self.lookup_here(name) | |
if entry is not None: | |
return entry | |
if language_level is None: | |
language_level = self.context.language_level if self.context is not None else 3 | |
if str_is_str is None: | |
str_is_str = language_level == 2 or ( | |
self.context is not None and Future.unicode_literals not in self.context.future_directives) | |
return self.outer_scope.lookup(name, language_level=language_level, str_is_str=str_is_str) | |
def declare_tuple_type(self, pos, components): | |
components = tuple(components) | |
try: | |
ttype = self._cached_tuple_types[components] | |
except KeyError: | |
ttype = self._cached_tuple_types[components] = PyrexTypes.c_tuple_type(components) | |
cname = ttype.cname | |
entry = self.lookup_here(cname) | |
if not entry: | |
scope = StructOrUnionScope(cname) | |
for ix, component in enumerate(components): | |
scope.declare_var(name="f%s" % ix, type=component, pos=pos) | |
struct_entry = self.declare_struct_or_union( | |
cname + '_struct', 'struct', scope, typedef_flag=True, pos=pos, cname=cname) | |
self.type_entries.remove(struct_entry) | |
ttype.struct_entry = struct_entry | |
entry = self.declare_type(cname, ttype, pos, cname) | |
ttype.entry = entry | |
return entry | |
def declare_builtin(self, name, pos): | |
if not hasattr(builtins, name) \ | |
and name not in Code.non_portable_builtins_map \ | |
and name not in Code.uncachable_builtins: | |
if self.has_import_star: | |
entry = self.declare_var(name, py_object_type, pos) | |
return entry | |
else: | |
if Options.error_on_unknown_names: | |
error(pos, "undeclared name not builtin: %s" % name) | |
else: | |
warning(pos, "undeclared name not builtin: %s" % name, 2) | |
# unknown - assume it's builtin and look it up at runtime | |
entry = self.declare(name, None, py_object_type, pos, 'private') | |
entry.is_builtin = 1 | |
return entry | |
if Options.cache_builtins: | |
for entry in self.cached_builtins: | |
if entry.name == name: | |
return entry | |
if name == 'globals' and not self.old_style_globals: | |
return self.outer_scope.lookup('__Pyx_Globals') | |
else: | |
entry = self.declare(None, None, py_object_type, pos, 'private') | |
if Options.cache_builtins and name not in Code.uncachable_builtins: | |
entry.is_builtin = 1 | |
entry.is_const = 1 # cached | |
entry.name = name | |
entry.cname = Naming.builtin_prefix + name | |
self.cached_builtins.append(entry) | |
self.undeclared_cached_builtins.append(entry) | |
else: | |
entry.is_builtin = 1 | |
entry.name = name | |
entry.qualified_name = self.builtin_scope().qualify_name(name) | |
return entry | |
def find_module(self, module_name, pos, relative_level=-1): | |
# Find a module in the import namespace, interpreting | |
# relative imports relative to this module's parent. | |
# Finds and parses the module's .pxd file if the module | |
# has not been referenced before. | |
relative_to = None | |
absolute_fallback = False | |
if relative_level is not None and relative_level > 0: | |
# explicit relative cimport | |
# error of going beyond top-level is handled in cimport node | |
relative_to = self | |
while relative_level > 0 and relative_to: | |
relative_to = relative_to.parent_module | |
relative_level -= 1 | |
elif relative_level != 0: | |
# -1 or None: try relative cimport first, then absolute | |
relative_to = self.parent_module | |
absolute_fallback = True | |
module_scope = self.global_scope() | |
return module_scope.context.find_module( | |
module_name, relative_to=relative_to, pos=pos, absolute_fallback=absolute_fallback) | |
def find_submodule(self, name): | |
# Find and return scope for a submodule of this module, | |
# creating a new empty one if necessary. Doesn't parse .pxd. | |
if '.' in name: | |
name, submodule = name.split('.', 1) | |
else: | |
submodule = None | |
scope = self.lookup_submodule(name) | |
if not scope: | |
scope = ModuleScope(name, parent_module=self, context=self.context) | |
self.module_entries[name] = scope | |
if submodule: | |
scope = scope.find_submodule(submodule) | |
return scope | |
def lookup_submodule(self, name): | |
# Return scope for submodule of this module, or None. | |
if '.' in name: | |
name, submodule = name.split('.', 1) | |
else: | |
submodule = None | |
module = self.module_entries.get(name, None) | |
if submodule and module is not None: | |
module = module.lookup_submodule(submodule) | |
return module | |
def add_include_file(self, filename, verbatim_include=None, late=False): | |
""" | |
Add `filename` as include file. Add `verbatim_include` as | |
verbatim text in the C file. | |
Both `filename` and `verbatim_include` can be `None` or empty. | |
""" | |
inc = Code.IncludeCode(filename, verbatim_include, late=late) | |
self.process_include(inc) | |
def process_include(self, inc): | |
""" | |
Add `inc`, which is an instance of `IncludeCode`, to this | |
`ModuleScope`. This either adds a new element to the | |
`c_includes` dict or it updates an existing entry. | |
In detail: the values of the dict `self.c_includes` are | |
instances of `IncludeCode` containing the code to be put in the | |
generated C file. The keys of the dict are needed to ensure | |
uniqueness in two ways: if an include file is specified in | |
multiple "cdef extern" blocks, only one `#include` statement is | |
generated. Second, the same include might occur multiple times | |
if we find it through multiple "cimport" paths. So we use the | |
generated code (of the form `#include "header.h"`) as dict key. | |
If verbatim code does not belong to any include file (i.e. it | |
was put in a `cdef extern from *` block), then we use a unique | |
dict key: namely, the `sortkey()`. | |
One `IncludeCode` object can contain multiple pieces of C code: | |
one optional "main piece" for the include file and several other | |
pieces for the verbatim code. The `IncludeCode.dict_update` | |
method merges the pieces of two different `IncludeCode` objects | |
if needed. | |
""" | |
key = inc.mainpiece() | |
if key is None: | |
key = inc.sortkey() | |
inc.dict_update(self.c_includes, key) | |
inc = self.c_includes[key] | |
def add_imported_module(self, scope): | |
if scope not in self.cimported_modules: | |
for inc in scope.c_includes.values(): | |
self.process_include(inc) | |
self.cimported_modules.append(scope) | |
for m in scope.cimported_modules: | |
self.add_imported_module(m) | |
def add_imported_entry(self, name, entry, pos): | |
if entry.is_pyglobal: | |
# Allow cimports to follow imports. | |
entry.is_variable = True | |
if entry not in self.entries: | |
self.entries[name] = entry | |
else: | |
warning(pos, "'%s' redeclared " % name, 0) | |
def declare_module(self, name, scope, pos): | |
# Declare a cimported module. This is represented as a | |
# Python module-level variable entry with a module | |
# scope attached to it. Reports an error and returns | |
# None if previously declared as something else. | |
entry = self.lookup_here(name) | |
if entry: | |
if entry.is_pyglobal and entry.as_module is scope: | |
return entry # Already declared as the same module | |
if not (entry.is_pyglobal and not entry.as_module): | |
# SAGE -- I put this here so Pyrex | |
# cimport's work across directories. | |
# Currently it tries to multiply define | |
# every module appearing in an import list. | |
# It shouldn't be an error for a module | |
# name to appear again, and indeed the generated | |
# code compiles fine. | |
return entry | |
else: | |
entry = self.declare_var(name, py_object_type, pos) | |
entry.is_variable = 0 | |
entry.as_module = scope | |
self.add_imported_module(scope) | |
return entry | |
def declare_var(self, name, type, pos, | |
cname = None, visibility = 'private', | |
api = 0, in_pxd = 0, is_cdef = 0): | |
# Add an entry for a global variable. If it is a Python | |
# object type, and not declared with cdef, it will live | |
# in the module dictionary, otherwise it will be a C | |
# global variable. | |
if not visibility in ('private', 'public', 'extern'): | |
error(pos, "Module-level variable cannot be declared %s" % visibility) | |
if not is_cdef: | |
if type is unspecified_type: | |
type = py_object_type | |
if not (type.is_pyobject and not type.is_extension_type): | |
raise InternalError( | |
"Non-cdef global variable is not a generic Python object") | |
if not cname: | |
defining = not in_pxd | |
if visibility == 'extern' or (visibility == 'public' and defining): | |
cname = name | |
else: | |
cname = self.mangle(Naming.var_prefix, name) | |
entry = self.lookup_here(name) | |
if entry and entry.defined_in_pxd: | |
#if visibility != 'private' and visibility != entry.visibility: | |
# warning(pos, "Variable '%s' previously declared as '%s'" % (name, entry.visibility), 1) | |
if not entry.type.same_as(type): | |
if visibility == 'extern' and entry.visibility == 'extern': | |
warning(pos, "Variable '%s' type does not match previous declaration" % name, 1) | |
entry.type = type | |
#else: | |
# error(pos, "Variable '%s' type does not match previous declaration" % name) | |
if entry.visibility != "private": | |
mangled_cname = self.mangle(Naming.var_prefix, name) | |
if entry.cname == mangled_cname: | |
cname = name | |
entry.cname = name | |
if not entry.is_implemented: | |
entry.is_implemented = True | |
return entry | |
entry = Scope.declare_var(self, name, type, pos, | |
cname=cname, visibility=visibility, | |
api=api, in_pxd=in_pxd, is_cdef=is_cdef) | |
if is_cdef: | |
entry.is_cglobal = 1 | |
if entry.type.declaration_value: | |
entry.init = entry.type.declaration_value | |
self.var_entries.append(entry) | |
else: | |
entry.is_pyglobal = 1 | |
if Options.cimport_from_pyx: | |
entry.used = 1 | |
return entry | |
def declare_cfunction(self, name, type, pos, | |
cname=None, visibility='private', api=0, in_pxd=0, | |
defining=0, modifiers=(), utility_code=None, overridable=False): | |
if not defining and 'inline' in modifiers: | |
# TODO(github/1736): Make this an error. | |
warning(pos, "Declarations should not be declared inline.", 1) | |
# Add an entry for a C function. | |
if not cname: | |
if visibility == 'extern' or (visibility == 'public' and defining): | |
cname = name | |
else: | |
cname = self.mangle(Naming.func_prefix, name) | |
if visibility == 'extern' and type.optional_arg_count: | |
error(pos, "Extern functions cannot have default arguments values.") | |
entry = self.lookup_here(name) | |
if entry and entry.defined_in_pxd: | |
if entry.visibility != "private": | |
mangled_cname = self.mangle(Naming.var_prefix, name) | |
if entry.cname == mangled_cname: | |
cname = name | |
entry.cname = cname | |
entry.func_cname = cname | |
entry = Scope.declare_cfunction( | |
self, name, type, pos, | |
cname=cname, visibility=visibility, api=api, in_pxd=in_pxd, | |
defining=defining, modifiers=modifiers, utility_code=utility_code, | |
overridable=overridable) | |
return entry | |
def declare_global(self, name, pos): | |
entry = self.lookup_here(name) | |
if not entry: | |
self.declare_var(name, py_object_type, pos) | |
def use_utility_code(self, new_code): | |
if new_code is not None: | |
self.utility_code_list.append(new_code) | |
def use_entry_utility_code(self, entry): | |
if entry is None: | |
return | |
if entry.utility_code: | |
self.utility_code_list.append(entry.utility_code) | |
if entry.utility_code_definition: | |
self.utility_code_list.append(entry.utility_code_definition) | |
def declare_c_class(self, name, pos, defining=0, implementing=0, | |
module_name=None, base_type=None, objstruct_cname=None, | |
typeobj_cname=None, typeptr_cname=None, visibility='private', | |
typedef_flag=0, api=0, check_size=None, | |
buffer_defaults=None, shadow=0): | |
# If this is a non-extern typedef class, expose the typedef, but use | |
# the non-typedef struct internally to avoid needing forward | |
# declarations for anonymous structs. | |
if typedef_flag and visibility != 'extern': | |
if not (visibility == 'public' or api): | |
warning(pos, "ctypedef only valid for 'extern' , 'public', and 'api'", 2) | |
objtypedef_cname = objstruct_cname | |
typedef_flag = 0 | |
else: | |
objtypedef_cname = None | |
# | |
# Look for previous declaration as a type | |
# | |
entry = self.lookup_here(name) | |
if entry and not shadow: | |
type = entry.type | |
if not (entry.is_type and type.is_extension_type): | |
entry = None # Will cause redeclaration and produce an error | |
else: | |
scope = type.scope | |
if typedef_flag and (not scope or scope.defined): | |
self.check_previous_typedef_flag(entry, typedef_flag, pos) | |
if (scope and scope.defined) or (base_type and type.base_type): | |
if base_type and base_type is not type.base_type: | |
error(pos, "Base type does not match previous declaration") | |
if base_type and not type.base_type: | |
type.base_type = base_type | |
# | |
# Make a new entry if needed | |
# | |
if not entry or shadow: | |
type = PyrexTypes.PyExtensionType( | |
name, typedef_flag, base_type, visibility == 'extern', check_size=check_size) | |
type.pos = pos | |
type.buffer_defaults = buffer_defaults | |
if objtypedef_cname is not None: | |
type.objtypedef_cname = objtypedef_cname | |
if visibility == 'extern': | |
type.module_name = module_name | |
else: | |
type.module_name = self.qualified_name | |
if typeptr_cname: | |
type.typeptr_cname = typeptr_cname | |
else: | |
type.typeptr_cname = self.mangle(Naming.typeptr_prefix, name) | |
entry = self.declare_type(name, type, pos, visibility = visibility, | |
defining = 0, shadow = shadow) | |
entry.is_cclass = True | |
if objstruct_cname: | |
type.objstruct_cname = objstruct_cname | |
elif not entry.in_cinclude: | |
type.objstruct_cname = self.mangle(Naming.objstruct_prefix, name) | |
else: | |
error(entry.pos, | |
"Object name required for 'public' or 'extern' C class") | |
self.attach_var_entry_to_c_class(entry) | |
self.c_class_entries.append(entry) | |
# | |
# Check for re-definition and create scope if needed | |
# | |
if not type.scope: | |
if defining or implementing: | |
scope = CClassScope(name = name, outer_scope = self, | |
visibility = visibility) | |
scope.directives = self.directives.copy() | |
if base_type and base_type.scope: | |
scope.declare_inherited_c_attributes(base_type.scope) | |
type.set_scope(scope) | |
self.type_entries.append(entry) | |
else: | |
if defining and type.scope.defined: | |
error(pos, "C class '%s' already defined" % name) | |
elif implementing and type.scope.implemented: | |
error(pos, "C class '%s' already implemented" % name) | |
# | |
# Fill in options, checking for compatibility with any previous declaration | |
# | |
if defining: | |
entry.defined_in_pxd = 1 | |
if implementing: # So that filenames in runtime exceptions refer to | |
entry.pos = pos # the .pyx file and not the .pxd file | |
if visibility != 'private' and entry.visibility != visibility: | |
error(pos, "Class '%s' previously declared as '%s'" | |
% (name, entry.visibility)) | |
if api: | |
entry.api = 1 | |
if objstruct_cname: | |
if type.objstruct_cname and type.objstruct_cname != objstruct_cname: | |
error(pos, "Object struct name differs from previous declaration") | |
type.objstruct_cname = objstruct_cname | |
if typeobj_cname: | |
if type.typeobj_cname and type.typeobj_cname != typeobj_cname: | |
error(pos, "Type object name differs from previous declaration") | |
type.typeobj_cname = typeobj_cname | |
if self.directives.get('final'): | |
entry.type.is_final_type = True | |
# cdef classes are always exported, but we need to set it to | |
# distinguish between unused Cython utility code extension classes | |
entry.used = True | |
# | |
# Return new or existing entry | |
# | |
return entry | |
def allocate_vtable_names(self, entry): | |
# If extension type has a vtable, allocate vtable struct and | |
# slot names for it. | |
type = entry.type | |
if type.base_type and type.base_type.vtabslot_cname: | |
#print "...allocating vtabslot_cname because base type has one" ### | |
type.vtabslot_cname = "%s.%s" % ( | |
Naming.obj_base_cname, type.base_type.vtabslot_cname) | |
elif type.scope and type.scope.cfunc_entries: | |
# one special case here: when inheriting from builtin | |
# types, the methods may also be built-in, in which | |
# case they won't need a vtable | |
entry_count = len(type.scope.cfunc_entries) | |
base_type = type.base_type | |
while base_type: | |
# FIXME: this will break if we ever get non-inherited C methods | |
if not base_type.scope or entry_count > len(base_type.scope.cfunc_entries): | |
break | |
if base_type.is_builtin_type: | |
# builtin base type defines all methods => no vtable needed | |
return | |
base_type = base_type.base_type | |
#print "...allocating vtabslot_cname because there are C methods" ### | |
type.vtabslot_cname = Naming.vtabslot_cname | |
if type.vtabslot_cname: | |
#print "...allocating other vtable related cnames" ### | |
type.vtabstruct_cname = self.mangle(Naming.vtabstruct_prefix, entry.name) | |
type.vtabptr_cname = self.mangle(Naming.vtabptr_prefix, entry.name) | |
def check_c_classes_pxd(self): | |
# Performs post-analysis checking and finishing up of extension types | |
# being implemented in this module. This is called only for the .pxd. | |
# | |
# Checks all extension types declared in this scope to | |
# make sure that: | |
# | |
# * The extension type is fully declared | |
# | |
# Also allocates a name for the vtable if needed. | |
# | |
for entry in self.c_class_entries: | |
# Check defined | |
if not entry.type.scope: | |
error(entry.pos, "C class '%s' is declared but not defined" % entry.name) | |
def check_c_class(self, entry): | |
type = entry.type | |
name = entry.name | |
visibility = entry.visibility | |
# Check defined | |
if not type.scope: | |
error(entry.pos, "C class '%s' is declared but not defined" % name) | |
# Generate typeobj_cname | |
if visibility != 'extern' and not type.typeobj_cname: | |
type.typeobj_cname = self.mangle(Naming.typeobj_prefix, name) | |
## Generate typeptr_cname | |
#type.typeptr_cname = self.mangle(Naming.typeptr_prefix, name) | |
# Check C methods defined | |
if type.scope: | |
for method_entry in type.scope.cfunc_entries: | |
if not method_entry.is_inherited and not method_entry.func_cname: | |
error(method_entry.pos, "C method '%s' is declared but not defined" % | |
method_entry.name) | |
# Allocate vtable name if necessary | |
if type.vtabslot_cname: | |
#print "ModuleScope.check_c_classes: allocating vtable cname for", self ### | |
type.vtable_cname = self.mangle(Naming.vtable_prefix, entry.name) | |
def check_c_classes(self): | |
# Performs post-analysis checking and finishing up of extension types | |
# being implemented in this module. This is called only for the main | |
# .pyx file scope, not for cimported .pxd scopes. | |
# | |
# Checks all extension types declared in this scope to | |
# make sure that: | |
# | |
# * The extension type is implemented | |
# * All required object and type names have been specified or generated | |
# * All non-inherited C methods are implemented | |
# | |
# Also allocates a name for the vtable if needed. | |
# | |
debug_check_c_classes = 0 | |
if debug_check_c_classes: | |
print("Scope.check_c_classes: checking scope " + self.qualified_name) | |
for entry in self.c_class_entries: | |
if debug_check_c_classes: | |
print("...entry %s %s" % (entry.name, entry)) | |
print("......type = ", entry.type) | |
print("......visibility = ", entry.visibility) | |
self.check_c_class(entry) | |
def check_c_functions(self): | |
# Performs post-analysis checking making sure all | |
# defined c functions are actually implemented. | |
for name, entry in self.entries.items(): | |
if entry.is_cfunction: | |
if (entry.defined_in_pxd | |
and entry.scope is self | |
and entry.visibility != 'extern' | |
and not entry.in_cinclude | |
and not entry.is_implemented): | |
error(entry.pos, "Non-extern C function '%s' declared but not defined" % name) | |
def attach_var_entry_to_c_class(self, entry): | |
# The name of an extension class has to serve as both a type | |
# name and a variable name holding the type object. It is | |
# represented in the symbol table by a type entry with a | |
# variable entry attached to it. For the variable entry, | |
# we use a read-only C global variable whose name is an | |
# expression that refers to the type object. | |
from . import Builtin | |
var_entry = Entry(name = entry.name, | |
type = Builtin.type_type, | |
pos = entry.pos, | |
cname = entry.type.typeptr_cname) | |
var_entry.qualified_name = entry.qualified_name | |
var_entry.is_variable = 1 | |
var_entry.is_cglobal = 1 | |
var_entry.is_readonly = 1 | |
var_entry.scope = entry.scope | |
entry.as_variable = var_entry | |
def is_cpp(self): | |
return self.cpp | |
def infer_types(self): | |
from .TypeInference import PyObjectTypeInferer | |
PyObjectTypeInferer().infer_types(self) | |
class LocalScope(Scope): | |
# Does the function have a 'with gil:' block? | |
has_with_gil_block = False | |
# Transient attribute, used for symbol table variable declarations | |
_in_with_gil_block = False | |
def __init__(self, name, outer_scope, parent_scope = None): | |
if parent_scope is None: | |
parent_scope = outer_scope | |
Scope.__init__(self, name, outer_scope, parent_scope) | |
def mangle(self, prefix, name): | |
return prefix + name | |
def declare_arg(self, name, type, pos): | |
# Add an entry for an argument of a function. | |
cname = self.mangle(Naming.var_prefix, name) | |
entry = self.declare(name, cname, type, pos, 'private') | |
entry.is_variable = 1 | |
if type.is_pyobject: | |
entry.init = "0" | |
entry.is_arg = 1 | |
#entry.borrowed = 1 # Not using borrowed arg refs for now | |
self.arg_entries.append(entry) | |
return entry | |
def declare_var(self, name, type, pos, | |
cname = None, visibility = 'private', | |
api = 0, in_pxd = 0, is_cdef = 0): | |
# Add an entry for a local variable. | |
if visibility in ('public', 'readonly'): | |
error(pos, "Local variable cannot be declared %s" % visibility) | |
entry = Scope.declare_var(self, name, type, pos, | |
cname=cname, visibility=visibility, | |
api=api, in_pxd=in_pxd, is_cdef=is_cdef) | |
if entry.type.declaration_value: | |
entry.init = entry.type.declaration_value | |
entry.is_local = 1 | |
entry.in_with_gil_block = self._in_with_gil_block | |
self.var_entries.append(entry) | |
return entry | |
def declare_global(self, name, pos): | |
# Pull entry from global scope into local scope. | |
if self.lookup_here(name): | |
warning(pos, "'%s' redeclared ", 0) | |
else: | |
entry = self.global_scope().lookup_target(name) | |
self.entries[name] = entry | |
def declare_nonlocal(self, name, pos): | |
# Pull entry from outer scope into local scope | |
orig_entry = self.lookup_here(name) | |
if orig_entry and orig_entry.scope is self and not orig_entry.from_closure: | |
error(pos, "'%s' redeclared as nonlocal" % name) | |
orig_entry.already_declared_here() | |
else: | |
entry = self.lookup(name) | |
if entry is None or not entry.from_closure: | |
error(pos, "no binding for nonlocal '%s' found" % name) | |
def lookup(self, name): | |
# Look up name in this scope or an enclosing one. | |
# Return None if not found. | |
entry = Scope.lookup(self, name) | |
if entry is not None: | |
entry_scope = entry.scope | |
while entry_scope.is_genexpr_scope: | |
entry_scope = entry_scope.outer_scope | |
if entry_scope is not self and entry_scope.is_closure_scope: | |
if hasattr(entry.scope, "scope_class"): | |
raise InternalError("lookup() after scope class created.") | |
# The actual c fragment for the different scopes differs | |
# on the outside and inside, so we make a new entry | |
entry.in_closure = True | |
inner_entry = InnerEntry(entry, self) | |
inner_entry.is_variable = True | |
self.entries[name] = inner_entry | |
return inner_entry | |
return entry | |
def mangle_closure_cnames(self, outer_scope_cname): | |
for scope in self.iter_local_scopes(): | |
for entry in scope.entries.values(): | |
if entry.from_closure: | |
cname = entry.outer_entry.cname | |
if self.is_passthrough: | |
entry.cname = cname | |
else: | |
if cname.startswith(Naming.cur_scope_cname): | |
cname = cname[len(Naming.cur_scope_cname)+2:] | |
entry.cname = "%s->%s" % (outer_scope_cname, cname) | |
elif entry.in_closure: | |
entry.original_cname = entry.cname | |
entry.cname = "%s->%s" % (Naming.cur_scope_cname, entry.cname) | |
class GeneratorExpressionScope(Scope): | |
"""Scope for generator expressions and comprehensions. As opposed | |
to generators, these can be easily inlined in some cases, so all | |
we really need is a scope that holds the loop variable(s). | |
""" | |
is_genexpr_scope = True | |
def __init__(self, outer_scope): | |
parent_scope = outer_scope | |
# TODO: also ignore class scopes? | |
while parent_scope.is_genexpr_scope: | |
parent_scope = parent_scope.parent_scope | |
name = parent_scope.global_scope().next_id(Naming.genexpr_id_ref) | |
Scope.__init__(self, name, outer_scope, parent_scope) | |
self.directives = outer_scope.directives | |
self.genexp_prefix = "%s%d%s" % (Naming.pyrex_prefix, len(name), name) | |
# Class/ExtType scopes are filled at class creation time, i.e. from the | |
# module init function or surrounding function. | |
while outer_scope.is_genexpr_scope or outer_scope.is_c_class_scope or outer_scope.is_py_class_scope: | |
outer_scope = outer_scope.outer_scope | |
self.var_entries = outer_scope.var_entries # keep declarations outside | |
outer_scope.subscopes.add(self) | |
def mangle(self, prefix, name): | |
return '%s%s' % (self.genexp_prefix, self.parent_scope.mangle(prefix, name)) | |
def declare_var(self, name, type, pos, | |
cname = None, visibility = 'private', | |
api = 0, in_pxd = 0, is_cdef = True): | |
if type is unspecified_type: | |
# if the outer scope defines a type for this variable, inherit it | |
outer_entry = self.outer_scope.lookup(name) | |
if outer_entry and outer_entry.is_variable: | |
type = outer_entry.type # may still be 'unspecified_type' ! | |
# the parent scope needs to generate code for the variable, but | |
# this scope must hold its name exclusively | |
cname = '%s%s' % (self.genexp_prefix, self.parent_scope.mangle(Naming.var_prefix, name or self.next_id())) | |
entry = self.declare(name, cname, type, pos, visibility) | |
entry.is_variable = True | |
if self.parent_scope.is_module_scope: | |
entry.is_cglobal = True | |
else: | |
entry.is_local = True | |
entry.in_subscope = True | |
self.var_entries.append(entry) | |
self.entries[name] = entry | |
return entry | |
def declare_pyfunction(self, name, pos, allow_redefine=False): | |
return self.outer_scope.declare_pyfunction( | |
name, pos, allow_redefine) | |
def declare_lambda_function(self, func_cname, pos): | |
return self.outer_scope.declare_lambda_function(func_cname, pos) | |
def add_lambda_def(self, def_node): | |
return self.outer_scope.add_lambda_def(def_node) | |
class ClosureScope(LocalScope): | |
is_closure_scope = True | |
def __init__(self, name, scope_name, outer_scope, parent_scope=None): | |
LocalScope.__init__(self, name, outer_scope, parent_scope) | |
self.closure_cname = "%s%s" % (Naming.closure_scope_prefix, scope_name) | |
# def mangle_closure_cnames(self, scope_var): | |
# for entry in self.entries.values() + self.temp_entries: | |
# entry.in_closure = 1 | |
# LocalScope.mangle_closure_cnames(self, scope_var) | |
# def mangle(self, prefix, name): | |
# return "%s->%s" % (self.cur_scope_cname, name) | |
# return "%s->%s" % (self.closure_cname, name) | |
def declare_pyfunction(self, name, pos, allow_redefine=False): | |
return LocalScope.declare_pyfunction(self, name, pos, allow_redefine, visibility='private') | |
class StructOrUnionScope(Scope): | |
# Namespace of a C struct or union. | |
def __init__(self, name="?"): | |
Scope.__init__(self, name, None, None) | |
def declare_var(self, name, type, pos, | |
cname = None, visibility = 'private', | |
api = 0, in_pxd = 0, is_cdef = 0, | |
allow_pyobject=False, allow_memoryview=False): | |
# Add an entry for an attribute. | |
if not cname: | |
cname = name | |
if visibility == 'private': | |
cname = c_safe_identifier(cname) | |
if type.is_cfunction: | |
type = PyrexTypes.CPtrType(type) | |
entry = self.declare(name, cname, type, pos, visibility) | |
entry.is_variable = 1 | |
self.var_entries.append(entry) | |
if type.is_pyobject and not allow_pyobject: | |
error(pos, "C struct/union member cannot be a Python object") | |
elif type.is_memoryviewslice and not allow_memoryview: | |
# Memory views wrap their buffer owner as a Python object. | |
error(pos, "C struct/union member cannot be a memory view") | |
if visibility != 'private': | |
error(pos, "C struct/union member cannot be declared %s" % visibility) | |
return entry | |
def declare_cfunction(self, name, type, pos, | |
cname=None, visibility='private', api=0, in_pxd=0, | |
defining=0, modifiers=(), overridable=False): # currently no utility code ... | |
if overridable: | |
error(pos, "C struct/union member cannot be declared 'cpdef'") | |
return self.declare_var(name, type, pos, | |
cname=cname, visibility=visibility) | |
class ClassScope(Scope): | |
# Abstract base class for namespace of | |
# Python class or extension type. | |
# | |
# class_name string Python name of the class | |
# scope_prefix string Additional prefix for names | |
# declared in the class | |
# doc string or None Doc string | |
def __init__(self, name, outer_scope): | |
Scope.__init__(self, name, outer_scope, outer_scope) | |
self.class_name = name | |
self.doc = None | |
def lookup(self, name): | |
entry = Scope.lookup(self, name) | |
if entry: | |
return entry | |
if name == "classmethod": | |
# We don't want to use the builtin classmethod here 'cause it won't do the | |
# right thing in this scope (as the class members aren't still functions). | |
# Don't want to add a cfunction to this scope 'cause that would mess with | |
# the type definition, so we just return the right entry. | |
entry = Entry( | |
"classmethod", | |
"__Pyx_Method_ClassMethod", | |
PyrexTypes.CFuncType( | |
py_object_type, | |
[PyrexTypes.CFuncTypeArg("", py_object_type, None)], 0, 0)) | |
entry.utility_code_definition = Code.UtilityCode.load_cached("ClassMethod", "CythonFunction.c") | |
self.use_entry_utility_code(entry) | |
entry.is_cfunction = 1 | |
return entry | |
class PyClassScope(ClassScope): | |
# Namespace of a Python class. | |
# | |
# class_obj_cname string C variable holding class object | |
is_py_class_scope = 1 | |
def mangle_class_private_name(self, name): | |
return self.mangle_special_name(name) | |
def mangle_special_name(self, name): | |
if name and name.startswith('__') and not name.endswith('__'): | |
name = EncodedString('_%s%s' % (self.class_name.lstrip('_'), name)) | |
return name | |
def lookup_here(self, name): | |
name = self.mangle_special_name(name) | |
return ClassScope.lookup_here(self, name) | |
def declare_var(self, name, type, pos, | |
cname = None, visibility = 'private', | |
api = 0, in_pxd = 0, is_cdef = 0): | |
name = self.mangle_special_name(name) | |
if type is unspecified_type: | |
type = py_object_type | |
# Add an entry for a class attribute. | |
entry = Scope.declare_var(self, name, type, pos, | |
cname=cname, visibility=visibility, | |
api=api, in_pxd=in_pxd, is_cdef=is_cdef) | |
entry.is_pyglobal = 1 | |
entry.is_pyclass_attr = 1 | |
return entry | |
def declare_nonlocal(self, name, pos): | |
# Pull entry from outer scope into local scope | |
orig_entry = self.lookup_here(name) | |
if orig_entry and orig_entry.scope is self and not orig_entry.from_closure: | |
error(pos, "'%s' redeclared as nonlocal" % name) | |
orig_entry.already_declared_here() | |
else: | |
entry = self.lookup(name) | |
if entry is None: | |
error(pos, "no binding for nonlocal '%s' found" % name) | |
else: | |
# FIXME: this works, but it's unclear if it's the | |
# right thing to do | |
self.entries[name] = entry | |
def declare_global(self, name, pos): | |
# Pull entry from global scope into local scope. | |
if self.lookup_here(name): | |
warning(pos, "'%s' redeclared ", 0) | |
else: | |
entry = self.global_scope().lookup_target(name) | |
self.entries[name] = entry | |
def add_default_value(self, type): | |
return self.outer_scope.add_default_value(type) | |
class CClassScope(ClassScope): | |
# Namespace of an extension type. | |
# | |
# parent_type CClassType | |
# #typeobj_cname string or None | |
# #objstruct_cname string | |
# method_table_cname string | |
# getset_table_cname string | |
# has_pyobject_attrs boolean Any PyObject attributes? | |
# has_memoryview_attrs boolean Any memory view attributes? | |
# has_cpp_class_attrs boolean Any (non-pointer) C++ attributes? | |
# has_cyclic_pyobject_attrs boolean Any PyObject attributes that may need GC? | |
# property_entries [Entry] | |
# defined boolean Defined in .pxd file | |
# implemented boolean Defined in .pyx file | |
# inherited_var_entries [Entry] Adapted var entries from base class | |
is_c_class_scope = 1 | |
is_closure_class_scope = False | |
has_pyobject_attrs = False | |
has_memoryview_attrs = False | |
has_cpp_class_attrs = False | |
has_cyclic_pyobject_attrs = False | |
defined = False | |
implemented = False | |
def __init__(self, name, outer_scope, visibility): | |
ClassScope.__init__(self, name, outer_scope) | |
if visibility != 'extern': | |
self.method_table_cname = outer_scope.mangle(Naming.methtab_prefix, name) | |
self.getset_table_cname = outer_scope.mangle(Naming.gstab_prefix, name) | |
self.property_entries = [] | |
self.inherited_var_entries = [] | |
def needs_gc(self): | |
# If the type or any of its base types have Python-valued | |
# C attributes, then it needs to participate in GC. | |
if self.has_cyclic_pyobject_attrs and not self.directives.get('no_gc', False): | |
return True | |
base_type = self.parent_type.base_type | |
if base_type and base_type.scope is not None: | |
return base_type.scope.needs_gc() | |
elif self.parent_type.is_builtin_type: | |
return not self.parent_type.is_gc_simple | |
return False | |
def needs_tp_clear(self): | |
""" | |
Do we need to generate an implementation for the tp_clear slot? Can | |
be disabled to keep references for the __dealloc__ cleanup function. | |
""" | |
return self.needs_gc() and not self.directives.get('no_gc_clear', False) | |
def get_refcounted_entries(self, include_weakref=False, | |
include_gc_simple=True): | |
py_attrs = [] | |
py_buffers = [] | |
memoryview_slices = [] | |
for entry in self.var_entries: | |
if entry.type.is_pyobject: | |
if include_weakref or (self.is_closure_class_scope or entry.name != "__weakref__"): | |
if include_gc_simple or not entry.type.is_gc_simple: | |
py_attrs.append(entry) | |
elif entry.type == PyrexTypes.c_py_buffer_type: | |
py_buffers.append(entry) | |
elif entry.type.is_memoryviewslice: | |
memoryview_slices.append(entry) | |
have_entries = py_attrs or py_buffers or memoryview_slices | |
return have_entries, (py_attrs, py_buffers, memoryview_slices) | |
def declare_var(self, name, type, pos, | |
cname = None, visibility = 'private', | |
api = 0, in_pxd = 0, is_cdef = 0): | |
if is_cdef: | |
# Add an entry for an attribute. | |
if self.defined: | |
error(pos, | |
"C attributes cannot be added in implementation part of" | |
" extension type defined in a pxd") | |
if not self.is_closure_class_scope and get_special_method_signature(name): | |
error(pos, | |
"The name '%s' is reserved for a special method." | |
% name) | |
if not cname: | |
cname = name | |
if visibility == 'private': | |
cname = c_safe_identifier(cname) | |
if type.is_cpp_class and visibility != 'extern': | |
type.check_nullary_constructor(pos) | |
self.use_utility_code(Code.UtilityCode("#include <new>")) | |
entry = self.declare(name, cname, type, pos, visibility) | |
entry.is_variable = 1 | |
self.var_entries.append(entry) | |
if type.is_memoryviewslice: | |
self.has_memoryview_attrs = True | |
elif type.is_cpp_class: | |
self.has_cpp_class_attrs = True | |
elif type.is_pyobject and (self.is_closure_class_scope or name != '__weakref__'): | |
self.has_pyobject_attrs = True | |
if (not type.is_builtin_type | |
or not type.scope or type.scope.needs_gc()): | |
self.has_cyclic_pyobject_attrs = True | |
if visibility not in ('private', 'public', 'readonly'): | |
error(pos, | |
"Attribute of extension type cannot be declared %s" % visibility) | |
if visibility in ('public', 'readonly'): | |
# If the field is an external typedef, we cannot be sure about the type, | |
# so do conversion ourself rather than rely on the CPython mechanism (through | |
# a property; made in AnalyseDeclarationsTransform). | |
entry.needs_property = True | |
if not self.is_closure_class_scope and name == "__weakref__": | |
error(pos, "Special attribute __weakref__ cannot be exposed to Python") | |
if not (type.is_pyobject or type.can_coerce_to_pyobject(self)): | |
# we're not testing for coercion *from* Python here - that would fail later | |
error(pos, "C attribute of type '%s' cannot be accessed from Python" % type) | |
else: | |
entry.needs_property = False | |
return entry | |
else: | |
if type is unspecified_type: | |
type = py_object_type | |
# Add an entry for a class attribute. | |
entry = Scope.declare_var(self, name, type, pos, | |
cname=cname, visibility=visibility, | |
api=api, in_pxd=in_pxd, is_cdef=is_cdef) | |
entry.is_member = 1 | |
entry.is_pyglobal = 1 # xxx: is_pyglobal changes behaviour in so many places that | |
# I keep it in for now. is_member should be enough | |
# later on | |
self.namespace_cname = "(PyObject *)%s" % self.parent_type.typeptr_cname | |
return entry | |
def declare_pyfunction(self, name, pos, allow_redefine=False): | |
# Add an entry for a method. | |
if name in richcmp_special_methods: | |
if self.lookup_here('__richcmp__'): | |
error(pos, "Cannot define both % and __richcmp__" % name) | |
elif name == '__richcmp__': | |
for n in richcmp_special_methods: | |
if self.lookup_here(n): | |
error(pos, "Cannot define both % and __richcmp__" % n) | |
if name == "__new__": | |
error(pos, "__new__ method of extension type will change semantics " | |
"in a future version of Pyrex and Cython. Use __cinit__ instead.") | |
entry = self.declare_var(name, py_object_type, pos, | |
visibility='extern') | |
special_sig = get_special_method_signature(name) | |
if special_sig: | |
# Special methods get put in the method table with a particular | |
# signature declared in advance. | |
entry.signature = special_sig | |
entry.is_special = 1 | |
else: | |
entry.signature = pymethod_signature | |
entry.is_special = 0 | |
self.pyfunc_entries.append(entry) | |
return entry | |
def lookup_here(self, name): | |
if not self.is_closure_class_scope and name == "__new__": | |
name = EncodedString("__cinit__") | |
entry = ClassScope.lookup_here(self, name) | |
if entry and entry.is_builtin_cmethod: | |
if not self.parent_type.is_builtin_type: | |
# For subtypes of builtin types, we can only return | |
# optimised C methods if the type if final. | |
# Otherwise, subtypes may choose to override the | |
# method, but the optimisation would prevent the | |
# subtype method from being called. | |
if not self.parent_type.is_final_type: | |
return None | |
return entry | |
def declare_cfunction(self, name, type, pos, | |
cname=None, visibility='private', api=0, in_pxd=0, | |
defining=0, modifiers=(), utility_code=None, overridable=False): | |
if get_special_method_signature(name) and not self.parent_type.is_builtin_type: | |
error(pos, "Special methods must be declared with 'def', not 'cdef'") | |
args = type.args | |
if not type.is_static_method: | |
if not args: | |
error(pos, "C method has no self argument") | |
elif not self.parent_type.assignable_from(args[0].type): | |
error(pos, "Self argument (%s) of C method '%s' does not match parent type (%s)" % | |
(args[0].type, name, self.parent_type)) | |
entry = self.lookup_here(name) | |
if cname is None: | |
cname = c_safe_identifier(name) | |
if entry: | |
if not entry.is_cfunction: | |
warning(pos, "'%s' redeclared " % name, 0) | |
else: | |
if defining and entry.func_cname: | |
error(pos, "'%s' already defined" % name) | |
#print "CClassScope.declare_cfunction: checking signature" ### | |
if entry.is_final_cmethod and entry.is_inherited: | |
error(pos, "Overriding final methods is not allowed") | |
elif type.same_c_signature_as(entry.type, as_cmethod = 1) and type.nogil == entry.type.nogil: | |
# Fix with_gil vs nogil. | |
entry.type = entry.type.with_with_gil(type.with_gil) | |
elif type.compatible_signature_with(entry.type, as_cmethod = 1) and type.nogil == entry.type.nogil: | |
if (self.defined and not in_pxd | |
and not type.same_c_signature_as_resolved_type(entry.type, as_cmethod = 1, as_pxd_definition = 1)): | |
# TODO(robertwb): Make this an error. | |
warning(pos, | |
"Compatible but non-identical C method '%s' not redeclared " | |
"in definition part of extension type '%s'. " | |
"This may cause incorrect vtables to be generated." % ( | |
name, self.class_name), 2) | |
warning(entry.pos, "Previous declaration is here", 2) | |
entry = self.add_cfunction(name, type, pos, cname, visibility='ignore', modifiers=modifiers) | |
else: | |
error(pos, "Signature not compatible with previous declaration") | |
error(entry.pos, "Previous declaration is here") | |
else: | |
if self.defined: | |
error(pos, | |
"C method '%s' not previously declared in definition part of" | |
" extension type '%s'" % (name, self.class_name)) | |
entry = self.add_cfunction(name, type, pos, cname, visibility, modifiers) | |
if defining: | |
entry.func_cname = self.mangle(Naming.func_prefix, name) | |
entry.utility_code = utility_code | |
type.entry = entry | |
if u'inline' in modifiers: | |
entry.is_inline_cmethod = True | |
if (self.parent_type.is_final_type or entry.is_inline_cmethod or | |
self.directives.get('final')): | |
entry.is_final_cmethod = True | |
entry.final_func_cname = entry.func_cname | |
return entry | |
def add_cfunction(self, name, type, pos, cname, visibility, modifiers, inherited=False): | |
# Add a cfunction entry without giving it a func_cname. | |
prev_entry = self.lookup_here(name) | |
entry = ClassScope.add_cfunction(self, name, type, pos, cname, | |
visibility, modifiers, inherited=inherited) | |
entry.is_cmethod = 1 | |
entry.prev_entry = prev_entry | |
return entry | |
def declare_builtin_cfunction(self, name, type, cname, utility_code = None): | |
# overridden methods of builtin types still have their Python | |
# equivalent that must be accessible to support bound methods | |
name = EncodedString(name) | |
entry = self.declare_cfunction(name, type, None, cname, visibility='extern', | |
utility_code=utility_code) | |
var_entry = Entry(name, name, py_object_type) | |
var_entry.qualified_name = name | |
var_entry.is_variable = 1 | |
var_entry.is_builtin = 1 | |
var_entry.utility_code = utility_code | |
var_entry.scope = entry.scope | |
entry.as_variable = var_entry | |
return entry | |
def declare_property(self, name, doc, pos): | |
entry = self.lookup_here(name) | |
if entry is None: | |
entry = self.declare(name, name, py_object_type, pos, 'private') | |
entry.is_property = 1 | |
entry.doc = doc | |
entry.scope = PropertyScope(name, | |
outer_scope = self.global_scope(), parent_scope = self) | |
entry.scope.parent_type = self.parent_type | |
self.property_entries.append(entry) | |
return entry | |
def declare_inherited_c_attributes(self, base_scope): | |
# Declare entries for all the C attributes of an | |
# inherited type, with cnames modified appropriately | |
# to work with this type. | |
def adapt(cname): | |
return "%s.%s" % (Naming.obj_base_cname, base_entry.cname) | |
entries = base_scope.inherited_var_entries + base_scope.var_entries | |
for base_entry in entries: | |
entry = self.declare( | |
base_entry.name, adapt(base_entry.cname), | |
base_entry.type, None, 'private') | |
entry.is_variable = 1 | |
self.inherited_var_entries.append(entry) | |
# If the class defined in a pxd, specific entries have not been added. | |
# Ensure now that the parent (base) scope has specific entries | |
# Iterate over a copy as get_all_specialized_function_types() will mutate | |
for base_entry in base_scope.cfunc_entries[:]: | |
if base_entry.type.is_fused: | |
base_entry.type.get_all_specialized_function_types() | |
for base_entry in base_scope.cfunc_entries: | |
cname = base_entry.cname | |
var_entry = base_entry.as_variable | |
is_builtin = var_entry and var_entry.is_builtin | |
if not is_builtin: | |
cname = adapt(cname) | |
entry = self.add_cfunction(base_entry.name, base_entry.type, | |
base_entry.pos, cname, | |
base_entry.visibility, base_entry.func_modifiers, inherited=True) | |
entry.is_inherited = 1 | |
if base_entry.is_final_cmethod: | |
entry.is_final_cmethod = True | |
entry.is_inline_cmethod = base_entry.is_inline_cmethod | |
if (self.parent_scope == base_scope.parent_scope or | |
entry.is_inline_cmethod): | |
entry.final_func_cname = base_entry.final_func_cname | |
if is_builtin: | |
entry.is_builtin_cmethod = True | |
entry.as_variable = var_entry | |
if base_entry.utility_code: | |
entry.utility_code = base_entry.utility_code | |
class CppClassScope(Scope): | |
# Namespace of a C++ class. | |
is_cpp_class_scope = 1 | |
default_constructor = None | |
type = None | |
def __init__(self, name, outer_scope, templates=None): | |
Scope.__init__(self, name, outer_scope, None) | |
self.directives = outer_scope.directives | |
self.inherited_var_entries = [] | |
if templates is not None: | |
for T in templates: | |
template_entry = self.declare( | |
T, T, PyrexTypes.TemplatePlaceholderType(T), None, 'extern') | |
template_entry.is_type = 1 | |
def declare_var(self, name, type, pos, | |
cname = None, visibility = 'extern', | |
api = 0, in_pxd = 0, is_cdef = 0, defining = 0): | |
# Add an entry for an attribute. | |
if not cname: | |
cname = name | |
entry = self.lookup_here(name) | |
if defining and entry is not None: | |
if entry.type.same_as(type): | |
# Fix with_gil vs nogil. | |
entry.type = entry.type.with_with_gil(type.with_gil) | |
elif type.is_cfunction and type.compatible_signature_with(entry.type): | |
entry.type = type | |
else: | |
error(pos, "Function signature does not match previous declaration") | |
else: | |
entry = self.declare(name, cname, type, pos, visibility) | |
entry.is_variable = 1 | |
if type.is_cfunction and self.type: | |
if not self.type.get_fused_types(): | |
entry.func_cname = "%s::%s" % (self.type.empty_declaration_code(), cname) | |
if name != "this" and (defining or name != "<init>"): | |
self.var_entries.append(entry) | |
return entry | |
def declare_cfunction(self, name, type, pos, | |
cname=None, visibility='extern', api=0, in_pxd=0, | |
defining=0, modifiers=(), utility_code=None, overridable=False): | |
class_name = self.name.split('::')[-1] | |
if name in (class_name, '__init__') and cname is None: | |
cname = "%s__init__%s" % (Naming.func_prefix, class_name) | |
name = '<init>' | |
type.return_type = PyrexTypes.CVoidType() | |
# This is called by the actual constructor, but need to support | |
# arguments that cannot by called by value. | |
type.original_args = type.args | |
def maybe_ref(arg): | |
if arg.type.is_cpp_class and not arg.type.is_reference: | |
return PyrexTypes.CFuncTypeArg( | |
arg.name, PyrexTypes.c_ref_type(arg.type), arg.pos) | |
else: | |
return arg | |
type.args = [maybe_ref(arg) for arg in type.args] | |
elif name == '__dealloc__' and cname is None: | |
cname = "%s__dealloc__%s" % (Naming.func_prefix, class_name) | |
name = '<del>' | |
type.return_type = PyrexTypes.CVoidType() | |
if name in ('<init>', '<del>') and type.nogil: | |
for base in self.type.base_classes: | |
base_entry = base.scope.lookup(name) | |
if base_entry and not base_entry.type.nogil: | |
error(pos, "Constructor cannot be called without GIL unless all base constructors can also be called without GIL") | |
error(base_entry.pos, "Base constructor defined here.") | |
prev_entry = self.lookup_here(name) | |
entry = self.declare_var(name, type, pos, | |
defining=defining, | |
cname=cname, visibility=visibility) | |
if prev_entry and not defining: | |
entry.overloaded_alternatives = prev_entry.all_alternatives() | |
entry.utility_code = utility_code | |
type.entry = entry | |
return entry | |
def declare_inherited_cpp_attributes(self, base_class): | |
base_scope = base_class.scope | |
template_type = base_class | |
while getattr(template_type, 'template_type', None): | |
template_type = template_type.template_type | |
if getattr(template_type, 'templates', None): | |
base_templates = [T.name for T in template_type.templates] | |
else: | |
base_templates = () | |
# Declare entries for all the C++ attributes of an | |
# inherited type, with cnames modified appropriately | |
# to work with this type. | |
for base_entry in \ | |
base_scope.inherited_var_entries + base_scope.var_entries: | |
#constructor/destructor is not inherited | |
if base_entry.name in ("<init>", "<del>"): | |
continue | |
#print base_entry.name, self.entries | |
if base_entry.name in self.entries: | |
base_entry.name # FIXME: is there anything to do in this case? | |
entry = self.declare(base_entry.name, base_entry.cname, | |
base_entry.type, None, 'extern') | |
entry.is_variable = 1 | |
entry.is_inherited = 1 | |
self.inherited_var_entries.append(entry) | |
for base_entry in base_scope.cfunc_entries: | |
entry = self.declare_cfunction(base_entry.name, base_entry.type, | |
base_entry.pos, base_entry.cname, | |
base_entry.visibility, api=0, | |
modifiers=base_entry.func_modifiers, | |
utility_code=base_entry.utility_code) | |
entry.is_inherited = 1 | |
for base_entry in base_scope.type_entries: | |
if base_entry.name not in base_templates: | |
entry = self.declare_type(base_entry.name, base_entry.type, | |
base_entry.pos, base_entry.cname, | |
base_entry.visibility) | |
entry.is_inherited = 1 | |
def specialize(self, values, type_entry): | |
scope = CppClassScope(self.name, self.outer_scope) | |
scope.type = type_entry | |
for entry in self.entries.values(): | |
if entry.is_type: | |
scope.declare_type(entry.name, | |
entry.type.specialize(values), | |
entry.pos, | |
entry.cname, | |
template=1) | |
elif entry.type.is_cfunction: | |
for e in entry.all_alternatives(): | |
scope.declare_cfunction(e.name, | |
e.type.specialize(values), | |
e.pos, | |
e.cname, | |
utility_code=e.utility_code) | |
else: | |
scope.declare_var(entry.name, | |
entry.type.specialize(values), | |
entry.pos, | |
entry.cname, | |
entry.visibility) | |
return scope | |
class PropertyScope(Scope): | |
# Scope holding the __get__, __set__ and __del__ methods for | |
# a property of an extension type. | |
# | |
# parent_type PyExtensionType The type to which the property belongs | |
is_property_scope = 1 | |
def declare_pyfunction(self, name, pos, allow_redefine=False): | |
# Add an entry for a method. | |
signature = get_property_accessor_signature(name) | |
if signature: | |
entry = self.declare(name, name, py_object_type, pos, 'private') | |
entry.is_special = 1 | |
entry.signature = signature | |
return entry | |
else: | |
error(pos, "Only __get__, __set__ and __del__ methods allowed " | |
"in a property declaration") | |
return None | |
class CConstScope(Scope): | |
def __init__(self, const_base_type_scope): | |
Scope.__init__( | |
self, | |
'const_' + const_base_type_scope.name, | |
const_base_type_scope.outer_scope, | |
const_base_type_scope.parent_scope) | |
self.const_base_type_scope = const_base_type_scope | |
def lookup_here(self, name): | |
entry = self.const_base_type_scope.lookup_here(name) | |
if entry is not None: | |
entry = copy.copy(entry) | |
entry.type = PyrexTypes.c_const_type(entry.type) | |
return entry | |
class TemplateScope(Scope): | |
def __init__(self, name, outer_scope): | |
Scope.__init__(self, name, outer_scope, None) | |
self.directives = outer_scope.directives | |