\ K]c @sdZddlmZddlZddlmZddlmZmZddl m Z m Z edd'Z ej dZeZdZedZedZedZedZdZeeZdefdYZdefdYZdefdYZeje egZejeZdefdYZdefdYZ d e fd!YZ!d"e fd#YZ"ie!d$6e"d%6Z#d&Z$dS((s3 Calling conventions for Numba-compiled functions. i(t namedtupleN(tir(tcgutilsttypesi(tPYOBJECTtGENERIC_POINTERtStatustcodetis_oktis_nonetis_errortis_stop_iterationt is_python_exct is_user_exct excinfoptri cCstjt|S(N(RtConstantt errcode_t(R((s5lib/python2.7/site-packages/numba/targets/callconv.pyt _const_int#siiit BaseCallConvcBskeZdZdZdZdZdZdZdZdZ dZ d Z d Z RS( cCs ||_dS(N(tcontext(tselfR((s5lib/python2.7/site-packages/numba/targets/callconv.pyt__init__5scCsA|tjkr|j|n||kr|jj||d|}tj||j}|j|3|jj ||j |j }|j ||WdQX|j|nt |tjs%||j kr|jj||d|d|j }n|jj ||j |}|j ||ntdj||dS(Ntvaluetfromtyttotysreturning {0} for {1}(Rtnonetreturn_native_noneRt make_helperRt as_bool_bittvalidtif_thentget_return_valuettypetdatat return_valuet isinstancetOptionaltcasttNotImplementedErrortformat(RtbuildertrettytvaltyRtoptvaltvalidbittretval((s5lib/python2.7/site-packages/numba/targets/callconv.pytreturn_optional_value8s$   cCs|j|tdS(N(t_return_errcode_rawt RETCODE_NONE(RR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyRUscCs|j|tdS(N(R/t RETCODE_EXC(RR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyt return_excXscCs|j|tdS(N(R/tRETCODE_STOPIT(RR(((s5lib/python2.7/site-packages/numba/targets/callconv.pytreturn_stop_iteration[scCs |jj|j}|jS(sQ Get the actual type of the return argument for Numba type *ty*. (Rtdata_model_managertget_return_typet as_pointer(Rttytrestype((s5lib/python2.7/site-packages/numba/targets/callconv.pyR6^scCs|j|}||_|S(sS Initialize and return a call helper object for the given builder. (t_make_call_helpert_BaseCallConv__call_helper(RR(tch((s5lib/python2.7/site-packages/numba/targets/callconv.pytinit_call_helperes cCs|jS(N(R;(RR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyt_get_call_helpermsc Cs |jj}|j|j`|j|j|j}tj|tj |||j |WdQX|j |WdQX|j|j |j d|j |WdQX|j|j|j |WdQX|jdd|j ||j|dS(sT Given a non-ok *status*, raise the corresponding Python exception. NtPyExc_StopIterationtPyExc_SystemErrors*unknown error when calling native function(tfunctiontappend_basic_blockRR t err_cleart unserializeRRt if_likelyt is_not_nullt raise_objecttbranchR t err_set_noneR terr_set_stringtposition_at_end(RR(tapitstatustbbendtexc((s5lib/python2.7/site-packages/numba/targets/callconv.pyt raise_errorps"     cCs.|j|}|j|}|j||S(s Get the decoded (unpacked) Python arguments with *argtypes* from LLVM function *func*. A tuple of LLVM values is returned. (t get_argumentst_get_arg_packertfrom_arguments(RR(targtypestfunctraw_argstarginfo((s5lib/python2.7/site-packages/numba/targets/callconv.pytdecode_argumentsscCs|jj|S(sF Get an argument packer for the given argument types. (Rtget_arg_packer(RRT((s5lib/python2.7/site-packages/numba/targets/callconv.pyRRs( t__name__t __module__RR.RR2R4R6R=R>RPRXRR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyR3s          tMinimalCallConvcBsteZdZdZdZd d d dZdZdZdZ dZ e dZ d Z d ZRS( s A minimal calling convention, suitable for e.g. GPU targets. The implemented function signature is: retcode_t (*, ... ) The return code will be one of the RETCODE_* constants or a function-specific user exception id (>= RETCODE_USEREXC). Caller is responsible for allocating a slot for the return value (passed as a pointer in the first argument). cCstS(N(t_MinimalCallHelper(RR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyR:scCsp|jjd}|j|jjksLtt|jt|jjf|j|||j|tdS(Ni( RAtargsR tpointeetAssertionErrortstrtstoreR/t RETCODE_OK(RR(R-tretptr((s5lib/python2.7/site-packages/numba/targets/callconv.pyR"s $c Cs|dk r2t|t r2td|fn|dk rdt|t rdtd|fn|dk r|j}|dkr|}n||j|jf}d|krd}qnd}|j |}|j |||} |j |t | dS(Ns-exc should be None or exception class, got %rs(exc_args should be None or tuple, got %r( tNonet issubclasst BaseExceptiont TypeErrorR#ttuplet_raw_function_nametfilenametlineR>t_add_exceptionR/R( RR(ROtexc_argstloct func_nametfnametlocinfot call_helpertexc_id((s5lib/python2.7/site-packages/numba/targets/callconv.pytreturn_user_excs"      cCs|j||jdS(N(R/R(RR(RM((s5lib/python2.7/site-packages/numba/targets/callconv.pytreturn_status_propagatescCs/t|trt|}n|j|dS(N(R#tintRtret(RR(R((s5lib/python2.7/site-packages/numba/targets/callconv.pyR/sc Cs|jd|t}|jd|t}|j||}|j|}|jd|t}|jd|t}|jd|t} td|d|d|d|d|d| d |d d } | S( s? Given a return *code*, get a Status instance. s==s>=RRR R R R R RN( t icmp_signedRcR0tor_tnot_R1R3tRETCODE_USEREXCRRe( RR(RtnormRtokterrROR R RM((s5lib/python2.7/site-packages/numba/targets/callconv.pyt_get_return_statuss   cCsJ|j|}t|j}|j|}tjt|g|}|S(sQ Get the implemented Function type for *restype* and *argtypes*. (RRtlisttargument_typesR6Rt FunctionTypeR(RR9RTRWtresptrtfnty((s5lib/python2.7/site-packages/numba/targets/callconv.pytget_function_types cCs`| s t|j|}|j|j|g|D]}d|^q5d|jd_|S(sA Set names and attributes of function arguments. sarg.s.reti(R`RRt assign_namesRQR^tname(RtfnR^t fe_argtypestnoaliasRWta((s5lib/python2.7/site-packages/numba/targets/callconv.pytdecorate_functions  cCs |jdS(s@ Get the Python-level arguments of LLVM *func*. i(R^(RRU((s5lib/python2.7/site-packages/numba/targets/callconv.pyRQscCs|jdjj}tj||}|jtj|||j|}|j||}|gt |} |j || } |j || } |j |} |j j||| } | | fS(s3 Call the Numba-compiled *callee*. i(R^R R_Rt alloca_onceRbtget_null_valueRRt as_argumentsRtcallRtloadRtget_returned_value(RR(tcalleetrestytargtysR^R)t retvaltmpRWtrealargsRRMR-tout((s5lib/python2.7/site-packages/numba/targets/callconv.pyt call_function sN(RZR[t__doc__R:R"ReRuRvR/RRtFalseRRQR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyR\s        R]cBs)eZdZdZdZdZRS(s A call helper object for the "minimal" calling convention. User exceptions are represented as integer codes and stored in a mapping for retrieval from the caller. cCs i|_dS(N(t exceptions(R((s5lib/python2.7/site-packages/numba/targets/callconv.pyR#scCs-t|jt}|||f|j|<|S(s Parameters ---------- exc : exception type exc_args : None or tuple exception args locinfo : tuple location information (tlenRt FIRST_USEREXC(RRORnRrRt((s5lib/python2.7/site-packages/numba/targets/callconv.pyRm&s cCs;y|j|SWn%tk r6d|}t|ffSXdS(Ns#unknown error %d in native function(RtKeyErrort SystemError(RRttmsg((s5lib/python2.7/site-packages/numba/targets/callconv.pyt get_exception5s   (RZR[RRRmR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyR]s  t CPUCallConvcBseZdZejdZdZdZddddZ dZ dZ dZ dZ ed Zd Zd Zd Zd ZRS(s The calling convention for CPU targets. The implemented function signature is: retcode_t (*, excinfo **, ... ) The return code will be one of the RETCODE_* constants. If RETCODE_USEREXC, the exception info pointer will be filled with a pointer to a constant struct describing the raised exception. Caller is responsible for allocating slots for the return value and the exception info pointer (passed as first and second arguments, respectively). icCsdS(N(Re(RR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyR:RscCsr|j|j}|j|jjksNtt|jt|jjf|j|||j|tdS(N( t_get_return_argumentRAR R_R`RaRbR/Rc(RR(R-Rd((s5lib/python2.7/site-packages/numba/targets/callconv.pyR"Us $c CsB|dk r2t|t r2td|fn|dk rdt|t rdtd|fn|dkr|t}n|jj|}|dk r|j}|dkr|}n||j |j f}d|krd}qnd}|||f}|j |} |j |j } |j| | |j|tdS(Ns-exc should be None or exception class, got %rs(exc_args should be None or tuple, got %r(ReRfRgRhR#RiRtget_python_apiRjRkRltserialize_objectt_get_excinfo_argumentRARbR/R|( RR(RORnRoRptpyapiRqRrt struct_gvtexcptr((s5lib/python2.7/site-packages/numba/targets/callconv.pyRu\s,        cCs<|j|j}|j|j||j||jdS(N(RRARbRR/R(RR(RMR((s5lib/python2.7/site-packages/numba/targets/callconv.pyRv}scCs|j|dS(N(Rx(RR(R((s5lib/python2.7/site-packages/numba/targets/callconv.pyR/sc Cs|jd|t}|jd|t}|jd|t}|jd|t}|j||}|j|} |jd|t} |j| |t j t t j }t d|d|d| d|d|d| d |d |} | S( sP Given a return *code* and *excinfoptr*, get a Status instance. s==s>=RRR R R R R R(RyRcR0R1R3RzR{R|tselectRRt excinfo_ptr_tt UndefinedR( RR(RRR}RROR R~RR RM((s5lib/python2.7/site-packages/numba/targets/callconv.pyRs$   cCsV|j|}t|j}|j|}tjt|tjtg|}|S(sQ Get the implemented Function type for *restype* and *argtypes*. ( RRRRR6RRRt PointerTypeR(RR9RTRWRR((s5lib/python2.7/site-packages/numba/targets/callconv.pyRs  c Cs|j|}|j|j|g|D]}d|^q(|j|}d|_|jd|jd|j|}d|_|jd|jd|r|j|}xC|D]8}t|jt j r|jd|jdqqWn|S(sU Set names of function arguments, and add useful attributes to them. sarg.Rdt nocaptureRtexcinfo( RRRRQRRt add_attributeRR#R RR( RRR^RRRWRtretargtexcarg((s5lib/python2.7/site-packages/numba/targets/callconv.pyRs$        cCs |jdS(s@ Get the Python-level arguments of LLVM *func*. i(R^(RRU((s5lib/python2.7/site-packages/numba/targets/callconv.pyRQscCs |jdS(Ni(R^(RRU((s5lib/python2.7/site-packages/numba/targets/callconv.pyRscCs |jdS(Ni(R^(RRU((s5lib/python2.7/site-packages/numba/targets/callconv.pyRscCs|j|jj}tj||}|jtj||tj|tjt dd}|j |} t | j ||}||g|} |j || } |j|| |j|} |j|} |jj||| }| |fS(s3 Call the Numba-compiled *callee*. RR(Rt function_typeR_RRRbRRRt excinfo_tRRRRRRRRR(RR(RRRR^R)RRRWRRRMR-R((s5lib/python2.7/site-packages/numba/targets/callconv.pyRs  N(RZR[Rt itertoolstcountt _status_idsR:R"ReRuRvR/RRRRRQRRR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyRAs         t ErrorModelcBs eZdZdddZRS(cCs ||_dS(N(t call_conv(RR((s5lib/python2.7/site-packages/numba/targets/callconv.pyRscCs.|jr&|jj|t||tStSdS(N(traise_on_fp_zero_divisionRRutZeroDivisionErrortTrueR(RR(RnRo((s5lib/python2.7/site-packages/numba/targets/callconv.pytfp_zero_divisions  N(RZR[RReR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyRs tPythonErrorModelcBseZdZeZRS(sL The Python error model. Any invalid FP input raises an exception. (RZR[RRR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyRstNumpyErrorModelcBseZdZeZRS(s6 In the Numpy error model, floating-point errors don't raise an exception. The FPU exception state is inspected by Numpy at the end of a ufunc's execution and a warning is raised if appropriate. Note there's no easy way to set the FPU exception state from LLVM. Instructions known to set an FP exception can be optimized away: https://llvm.org/bugs/show_bug.cgi?id=6050 http://lists.llvm.org/pipermail/llvm-dev/2014-September/076918.html http://lists.llvm.org/pipermail/llvm-commits/Week-of-Mon-20140929/237997.html (RZR[RRR(((s5lib/python2.7/site-packages/numba/targets/callconv.pyRs tpythontnumpycCst||jS(sF Create an error model instance for the given target context. (t error_modelsR(t model_nameR((s5lib/python2.7/site-packages/numba/targets/callconv.pytcreate_error_models(RRR R R R R R(%Rt collectionsRRtllvmliteRtnumbaRRtbaseRRRtIntTypetint32_tRRRcR1R0R3RR|tobjectRR\R]tLiteralStructTypeRRRRRRRRR(((s5lib/python2.7/site-packages/numba/targets/callconv.pytsH        k~!