File size: 2,230 Bytes
16a8ff1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | ##################### UFuncDefinition ######################
cdef extern from *:
ctypedef int npy_intp
struct PyObject
PyObject* __Pyx_NewRef(object)
{{inline_func_declaration}}
# variable names have to come from tempita to avoid duplication
@cname("{{func_cname}}")
cdef void {{func_cname}}(char **args, const npy_intp *dimensions, const npy_intp* steps, void* data) except * {{"nogil" if will_be_called_without_gil else ""}}:
cdef npy_intp i
cdef npy_intp n = dimensions[0]
{{for idx, tp in enumerate(in_types)}}
cdef char* in_{{idx}} = args[{{idx}}]
cdef {{tp.empty_declaration_code(pyrex=True)}} cast_in_{{idx}}
{{endfor}}
{{for idx, tp in enumerate(out_types)}}
cdef char* out_{{idx}} = args[{{idx+len(in_types)}}]
cdef {{tp.empty_declaration_code(pyrex=True)}} cast_out_{{idx}}
{{endfor}}
{{for idx in range(len(out_types)+len(in_types))}}
cdef npy_intp step_{{idx}} = steps[{{idx}}]
{{endfor}}
{{"with gil" if (not nogil and will_be_called_without_gil) else "if True"}}:
for i in range(n):
{{for idx, tp in enumerate(in_types)}}
{{if tp.is_pyobject}}
cast_in_{{idx}} = (<{{tp.empty_declaration_code(pyrex=True)}}>(<void**>in_{{idx}})[0])
{{else}}
cast_in_{{idx}} = (<{{tp.empty_declaration_code(pyrex=True)}}*>in_{{idx}})[0]
{{endif}}
{{endfor}}
{{", ".join("cast_out_{}".format(idx) for idx in range(len(out_types)))}} = \
{{inline_func_call}}({{", ".join("cast_in_{}".format(idx) for idx in range(len(in_types)))}})
{{for idx, tp in enumerate(out_types)}}
{{if tp.is_pyobject}}
(<void**>out_{{idx}})[0] = <void*>__Pyx_NewRef(cast_out_{{idx}})
{{else}}
(<{{tp.empty_declaration_code(pyrex=True)}}*>out_{{idx}})[0] = cast_out_{{idx}}
{{endif}}
{{endfor}}
{{for idx in range(len(in_types))}}
in_{{idx}} += step_{{idx}}
{{endfor}}
{{for idx in range(len(out_types))}}
out_{{idx}} += step_{{idx+len(in_types)}}
{{endfor}}
|