o 8ήcU'@spdZddlmZddlmZmZddlmZmZmZddlm Z m Z ddl m Z ddl mZddlmZdd l mZdd lmZmZmZmZdd lmZeeee d d ZeeGdddejZeeeeGdddejZe ejddZe eddZ ddZ! d$ddZ"eeddZ#eeddZ$e eedd Z%e ejed!d"Z&d#S)%zeProvides Numba type, FunctionType, that makes functions as instances of a first-class function type. ) typeof_impl)modelsregister_model)unbox NativeValuebox)lower_constant lower_cast)CFunc)cgutils)ir)types) FunctionTypeUndefinedFunctionTypeFunctionPrototypeWrapperAddressProtocol) DispatchercCsFt|tr |j}t|St|tr|}t|Stdt|j)Nzfunction type from ) isinstancer _sigr signatureNotImplementedErrortype__name__r)valcsigrG/tmp/pip-target-vg8gfxp4/lib/python/numba/experimental/function_type.pytypeof_function_types  rc eZdZdZfddZZS)FunctionProtoModelzIFunctionProtoModel describes the signatures of first-class functions cs|t|tr |j}nt|tr|}ntt||j}fdd|j D}t t ||}t t |||dS)Ncsg|] }|qSr)lookupget_value_type).0tdmmrr +sz/FunctionProtoModel.__init__..)rrftyperrrr!rtyper"atypesr PointerTypesuperr __init__)selfr&fe_typer(rettyargsbe_type __class__r%rr-#s   zFunctionProtoModel.__init__r __module__ __qualname____doc__r- __classcell__rrr3rr sr cr) FunctionModelz>FunctionModel holds addresses of function implementations cs,dtjfdtjfg}tt||||dS)Naddrpyaddr)r voidptrr,r:r-)r.r&r/membersr3rrr-5szFunctionModel.__init__r5rrr3rr:0sr:cCs|j|t|t|jdS)Ninfo)add_dynamic_addridrr)contextbuildertyppyvalrrrlower_constant_dispatcher@s rGcCs"|}t|tr1|j}t|||}|j||t|d|_|j|t |t |j d|_ | St|trPt|||}|j|t |t |j d|_ | St|tr|}||sbJt|||}|j||t|d|_|j|t |t |j d|_ | Std||||)Nr?z3lower_constant_struct_function_type({}, {}, {}, {})) get_preciserr _wrapper_addressr create_struct_proxyrAstrr;rBrrr< _getvaluerr__wrapper_address__check_signaturerrformat)rCrDrErFr;sfuncrrrlower_constant_function_typeFs@       rQcCs|sd}n;t|dr|}n1t|tr!||jksJ|j}n!t|tr6||}|j j }|j |}n t dt|d|t|tsQtdt|d|dkri|dkritdt|d |d |d |S) aReturn the address of a compiled cfunc wrapper function of `func`. Warning: The compiled function must be compatible with the given signature `sig`. If it is not, then result of calling the compiled function is undefined. The compatibility is ensured when passing in a first-class function to a Numba njit compiled function either as an argument or via namespace scoping. Parameters ---------- func : object A Numba cfunc or jit decoreated function or an object that implements the wrapper address protocol (see note below). sig : Signature The expected function signature. Returns ------- addr : int An address in memory (pointer value) of the compiled function corresponding to the specified signature. Note: wrapper address protocol ------------------------------ An object implements the wrapper address protocol iff the object provides a callable attribute named __wrapper_address__ that takes a Signature instance as the argument, and returns an integer representing the address or pointer value of a compiled function for the given signature. rMzget wrapper address of z instance with z%wrapper address must be integer, got z instancerzwrapper address of z- instance must be a positive integer but got z [sig=]) is_precisehasattrrMrr raddressrget_compile_resultfndescllvm_cfunc_wrapper_namelibraryget_pointer_to_functionrrint TypeError ValueError)funcrr;cres wrapper_namerrr_get_wrapper_addressis2!       rb return_excc Cs||}||jt}||}||d}|||||} | ||| f} |dkro|j t || dd,|dkrG|j |n|dkrS||n t|Wd| SWd| S1sjwY| S)znLow-level call to _get_wrapper_address(func, sig). When calling this function, GIL must be acquired. rbignoreFlikelyrc return_nullN)get_python_apiinsert_const_stringmodulerimport_module_noblockobject_getattr_stringdecref unserializeserialize_objectcall_function_objargsif_thenr is_null call_convrcretget_null_objectr) rCrDr_r failure_modepyapimodname numba_mod numba_funcsig_objr;rrrlower_get_wrapper_addresss0        r|cCsx|}t||j|j}t|j|j||jdd}|j||_ |j ||j t j }|j|||_t|S)Nrgrv)rHr rJrCrDr|rrwlong_as_voidptrr;rmr"r r=ptrtointr<rrL)rEobjrrPr;lltyrrrunbox_function_types  rcCs|}t||j|j|d}t|j|jj}|j|j |jj}|jj t |j|ddd|d}|j d||j |jWdn1sQwY|j|||j|}|j||S)N)valueFrezfirst-class function z parent object not setPyExc_MemoryError)rHr rJrCrD alloca_oncerwpyobjinttoptrr<rqrrerr_set_stringrtrustoreloadincref)rErrrP pyaddr_ptrraw_ptrcstrcfuncrrrbox_function_types   rcCs|S)Nr)rCrDfromtytotyrrrr)lower_cast_function_type_to_function_typesrc Cs|}||}t|||}|}t||||jdd}|||_| || || t j } ||| |_|S)Nrcr})rHrhr rJ gil_ensurer|rr~r;rm gil_releaser"r r=rr<rL) rCrDrrrrwrP gil_stater;rrrr&lower_cast_dispatcher_to_function_types      rN)rc)'r8numba.extendingrrrrrrnumba.core.imputilsrr numba.core.ccallbackr numba.corer llvmliter r numba.core.typesrrrrnumba.core.dispatcherrregisterrPrimitiveModelr StructModelr:rGrQrbr|rrrrrrrrsD          "A &