def allocate_object(caller, builder, type, env):
"""
Allocate object of type `type`.
"""
if stack_allocate(type):
obj = builder.alloca(ptypes.Pointer(ptypes.Opaque))
return [obj], obj
else:
if env['flypy.target'] != 'cpu':
raise errors.CompileError(
"Cannot heap allocate object of type %s with target %r" % (
type, env['flypy.target']))
return heap_allocate(caller, builder, type, env)
def allocate_object(caller, builder, type, env):
"""
Allocate object of type `type`.
"""
if stack_allocate(type):
obj = builder.alloca(ptypes.Pointer(ptypes.Opaque))
return [obj], obj
else:
if env['flypy.target'] != 'cpu':
raise errors.CompileError(
"Cannot heap allocate object of type %s with target %r" %
(type, env['flypy.target']))
return heap_allocate(caller, builder, type, env)
def __call__(self, *args, **kwargs):
from flypy.representation import byref, stack_allocate
from flypy.conversion import (toctypes, fromctypes, toobject,
fromobject, ctype)
#from flypy.support.ctypes_support import CTypesStruct
#from flypy.types import Function
# Keep this alive for the duration of the call
keepalive = list(args) + list(kwargs.values())
# Order arguments
args = flatargs(self.py_func, args, kwargs)
# Translate
cfunc, restype = self.translate([typeof(x) for x in args.flat])
# Construct flypy values
argtypes = [typeof(x) for x in args]
arg_objs = list(starmap(fromobject, zip(args, argtypes)))
# Map flypy values to a ctypes representation
args = []
for arg, argtype in zip(arg_objs, argtypes):
c_arg = toctypes(arg, argtype, keepalive)
if byref(argtype) and stack_allocate(argtype):
c_arg = ctypes.pointer(c_arg)
args.append(c_arg)
# We need this cast since the ctypes function constructed from LLVM
# IR has different structs (which are structurally equivalent)
c_restype = ctype(restype)
if byref(restype):
c_result = c_restype() # dummy result value
args.append(ctypes.pointer(c_result))
c_restype = None # void
c_signature = ctypes.PYFUNCTYPE(c_restype,
*[type(arg) for arg in args])
cfunc = ctypes.cast(cfunc, c_signature)
# Handle calling convention
if byref(restype):
cfunc(*args)
else:
c_result = cfunc(*args)
# Map ctypes result back to a python value
result = fromctypes(c_result, restype)
result_obj = toobject(result, restype)
return result_obj
def __call__(self, *args, **kwargs):
from flypy.representation import byref, stack_allocate
from flypy.conversion import (
toctypes, fromctypes, toobject, fromobject, ctype)
#from flypy.support.ctypes_support import CTypesStruct
#from flypy.types import Function
# Keep this alive for the duration of the call
keepalive = list(args) + list(kwargs.values())
# Order arguments
args = flatargs(self.py_func, args, kwargs)
# Translate
cfunc, restype = self.translate([typeof(x) for x in args.flat])
# Construct flypy values
argtypes = [typeof(x) for x in args]
arg_objs = list(starmap(fromobject, zip(args, argtypes)))
# Map flypy values to a ctypes representation
args = []
for arg, argtype in zip(arg_objs, argtypes):
c_arg = toctypes(arg, argtype, keepalive)
if byref(argtype) and stack_allocate(argtype):
c_arg = ctypes.pointer(c_arg)
args.append(c_arg)
# We need this cast since the ctypes function constructed from LLVM
# IR has different structs (which are structurally equivalent)
c_restype = ctype(restype)
if byref(restype):
c_result = c_restype() # dummy result value
args.append(ctypes.pointer(c_result))
c_restype = None # void
c_signature = ctypes.PYFUNCTYPE(c_restype, *[type(arg) for arg in args])
cfunc = ctypes.cast(cfunc, c_signature)
# Handle calling convention
if byref(restype):
cfunc(*args)
else:
c_result = cfunc(*args)
# Map ctypes result back to a python value
result = fromctypes(c_result, restype)
result_obj = toobject(result, restype)
return result_obj