def make_wrappers():
from pypytools.codegen import Code
code = Code()
names = [
'all', 'any', 'argmax', 'argmin', 'argsort', 'choose', 'clip', 'copy',
'cumprod', 'cumsum', 'diagonal', 'max', 'min', 'nonzero', 'prod',
'ptp', 'put', 'ravel', 'repeat', 'reshape', 'round', 'squeeze', 'sum',
'swapaxes', 'transpose'
]
#
for name in names:
ns = code.new_scope(name=name)
with code.def_(name, ['myarray'], '*args', '**kwargs'):
ns.w('return myarray.{name}(*args, **kwargs)')
#
code.compile()
gbl = globals()
for name in names:
gbl[name] = code[name]
class ModuleGenerator(object):
def __init__(self, request, convert_case, pyx, version_check, standalone):
self.code = Code(pyx=pyx)
self.request = request
self.convert_case = convert_case
self.pyx = pyx
self.version_check = version_check
self.standalone = standalone
self.allnodes = {} # id -> node
self.children = defaultdict(list) # nodeId -> nested nodes
self.importnames = {} # filename -> import name
self.extra_annotations = defaultdict(list) # obj -> [ann]
self.field_override = {} # obj -> obj
def register_extra_annotation(self, obj, ann):
self.extra_annotations[obj].append(ann.annotation)
def register_field_override(self, origin, target):
self.field_override[origin] = target
def has_annotation(self, obj, anncls):
annotations = self.extra_annotations.get(obj, [])
if obj.annotations is not None:
annotations += obj.annotations
for ann in annotations:
if ann.id == anncls.__id__:
# XXX: probably "annotation" should be taken by the
# constructor
res = anncls()
res.annotation = ann
res.target = obj
return res
return None
def w(self, *args, **kwargs):
self.code.w(*args, **kwargs)
def block(self, *args, **kwargs):
return self.code.block(*args, **kwargs)
def register_import(self, fname):
name = py.path.local(fname).purebasename
name = name.replace('+', 'PLUS')
name = '_%s_capnp' % name
if name in self.importnames.values():
# avoid name clashes
name = '%s_%s' % (name, len(self.filenames))
self.importnames[fname] = name
return name
def generate(self):
self.request.emit(self)
return self.code.build()
def _dump_node(self, node):
def visit(node, deep=0):
print('%s%s: %s' % (' ' * deep, node.which(), node.displayName))
for child in self.children[node.id]:
visit(child, deep + 2)
visit(node)
def _convert_name(self, name):
name = ensure_unicode(name) if PY3 else name
if self.convert_case:
return from_camel_case(name)
else:
return name
def _field_name(self, field):
name = self._convert_name(field.name)
name = self._mangle_name(name)
return name
def _mangle_name(self, name):
if name in keyword.kwlist:
return name + '_'
return name
def declare_enum(self, compile_name, name, items):
# this method cannot go on Node__Enum because it's also called by
# Node__Struct (for __tag__)
items = list(map(repr, items))
ns = self.code.new_scope()
ns.name = compile_name
ns.members = "(%s,)" % (', '.join(items))
ns.prebuilt = [
ns.format('{name}({i})', i=i) for i in range(len(items))
]
ns.prebuilt = ', '.join(ns.prebuilt)
ns.prebuilt = ns.format('({prebuilt},)')
with ns.block("{cdef class} {name}(_BaseEnum):"):
ns.w("__members__ = {members}")
#
# define the _new staticmethod, to create new instances.
if self.pyx:
# on CPython, make a tuple of prebuilt instances for the
# statically known values
ns.w("@staticmethod")
with ns.block(
'cdef _new(long x, __prebuilt={prebuilt}):') as ns:
ns.ww("""
try:
return __prebuilt[x]
except IndexError:
return {name}(x)
""")
else:
# on PyPy, always create a new object, so that the JIT will be
# able to make it virtual
ns.w("@staticmethod")
with ns.block('def _new(x):') as ns:
ns.w('return {name}(x)')
ns.w("_fill_enum({name})")
def def_property(self, ns, name, src):
if self.pyx:
with ns.block('property {name}:', name=name):
with ns.block('def __get__(self):'):
ns.ww(src)
else:
ns.w('@property')
with ns.block('def {name}(self):', name=name):
ns.ww(src)
ns.w()
def test_scope_formatting(self):
code = Code()
ns = code.new_scope(y=52)
ns.x = 42
ns.w('x == {x}, y == {y}')
assert code.build() == 'x == 42, y == 52'
def test_new_scope(self):
code = Code()
code.global_scope.x = 42
code.global_scope.y = 52
ns = code.new_scope(y=53, z=63)
assert ns.__dict__ == dict(x=42, y=53, z=63)