\ K]c@`s]ddlmZmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZmZmZmZddlmZddl mZmZmZddlmZddlmZmZdd lmZdd l m!Z!m"Z"d d l#m$Z$m%Z%d e&fdYZ'de&fdYZ(de(fdYZ)ej*ddddfZ+de&fdYZ,dej-fdYZ.de.fdYZ-de.fdYZ/de/fd YZ0d!e/fd"YZ1d#e1fd$YZ2ej3e'e4d%ej5DdS(&i(tprint_functiontdivisiontabsolute_importN(tdeepcopy(t _dispatchertcompilertutilsttypestconfigterrors(tdefault_type_manager(tsigutilst serializettyping(tfold_arguments(tPurposettypeof(tget_code_object(tcreate_bound_methodtreraisei(t NullCachet FunctionCachet OmittedArgcB`s/eZdZdZdZedZRS(sC A placeholder for omitted arguments with a default value. cC`s ||_dS(N(tvalue(tselfR((s/lib/python2.7/site-packages/numba/dispatcher.pyt__init__scC`sd|jfS(Nsomitted arg(%r)(R(R((s/lib/python2.7/site-packages/numba/dispatcher.pyt__repr__ scC`stj|jS(N(RtOmittedR(R((s/lib/python2.7/site-packages/numba/dispatcher.pyt _numba_type_#s(t__name__t __module__t__doc__RRtpropertyR(((s/lib/python2.7/site-packages/numba/dispatcher.pyRs  t_FunctionCompilercB`sPeZdZdZdZdZdZdZdZdZ RS(cC`sO||_||_||_||_tj|j|_||_i|_dS(N( tpy_funct targetdescrt targetoptionstlocalsRt pysignaturetpysigtpipeline_classt _failed_cache(RR"R#R$R%R(((s/lib/python2.7/site-packages/numba/dispatcher.pyR)s     cC`sFd}d}d}t|j|||||}|j|fS(s Given positional and named argument types, fold keyword arguments and resolve defaults by inserting types.Omitted() instances. A (pysig, argument types) tuple is returned. cS`s|S(N((tindextparamR((s/lib/python2.7/site-packages/numba/dispatcher.pytnormal_handler=scS`s tj|S(N(RR(R*R+tdefault((s/lib/python2.7/site-packages/numba/dispatcher.pytdefault_handler?scS`s tj|S(N(RtTuple(R*R+tvalues((s/lib/python2.7/site-packages/numba/dispatcher.pytstararg_handlerAs(RR'(RtargstkwsR,R.R1((s/lib/python2.7/site-packages/numba/dispatcher.pytfold_argument_types6s    cC`s,|j||\}}|r"|S|dS(N(t_compile_cached(RR2t return_typetstatustretval((s/lib/python2.7/site-packages/numba/dispatcher.pytcompileKscC`st||f}yt|j|fSWntk r:nXy|j||}Wn*tjk r}}||j|si((R t_tmt collectionst OrderedDictt overloadsR"Rt func_codet__code__R:RWt __defaults__tlistR0t IndexErrorR;RZtVAR_POSITIONALRt DispatcherRt get_pointert _fold_argsRtdocRft_compiling_counterRtfinalizet_make_finalizer( Rt arg_countR"R't can_fallbacktexact_match_requiredtargnamestdefault_valuestdefargstlastargt has_stararg((s/lib/python2.7/site-packages/numba/dispatcher.pyRs,        cC`s|j|jjdS(N(t_clearRttclear(R((s/lib/python2.7/site-packages/numba/dispatcher.pyt_reset_overloadss c`s.|j|jtjfd}|S(sq Return a finalizer function that will release references to related compiled functions. c`sP|r dSx<jD].}yj|jWqtk rGqXqWdS(N(R0tremove_user_functiont entry_pointR<(t shutting_downRO(Rtt targetctx(s/lib/python2.7/site-packages/numba/dispatcher.pyt finalizers  (RtRRR(RR((RtRs/lib/python2.7/site-packages/numba/dispatcher.pyRs   cC`s t|jS(sA Returns a list of compiled function signatures. (RxRt(R((s/lib/python2.7/site-packages/numba/dispatcher.pyt signaturesscC`s7g|jjD]#}|j r|j r|j^qS(N(RtR0t objectmodet interpmodet signature(RRO((s/lib/python2.7/site-packages/numba/dispatcher.pytnopython_signaturesscC`s0| s"t|jdks"t| |_dS(s@Disable the compilation of new signatures at call time. iN(RVRRht _can_compile(RRp((s/lib/python2.7/site-packages/numba/dispatcher.pytdisable_compile s"cC`s^t|jj}g|D]}|j^q}|j||j|j|j||j|scO`sv| s td}g}xO|D]G}t|trT|jtj|jq#|j|j|q#Wy|jt |SWnt j k r}g}xt |D]\}} t| tr| jn| } yt | tj} Wn,tk r} |j|t| fqX| d kr|j|d| ffqqW|rt|jddjd|D} |j| n||dnt jk r}||dnt jt jt jfk r}||dnt jk r}||d natk rq}tjrht|d rht jd }|jd j|j |qhn|nXd S(s For internal use. Compile a specialized version of the function for the given *args* and *kws*, and return the resulting callable. cS`sbtjr6tj|}|jdj|j|ntjrH|ntt ||ddS(s{ Rewrite and raise Exception `e` with help supplied based on the specified issue_type. tN( Rt SHOW_HELPR t error_extrast patch_messagetjoinR2tFULL_TRACEBACKSRttypeRL(RAt issue_typethelp_msg((s/lib/python2.7/site-packages/numba/dispatcher.pyt error_rewriteLs      s'cannot determine Numba type of value %rsC This error may have been caused by the following argument(s): %s s cs`s%|]\}}d||fVqdS(s- argument %d: %sN((Rotiterr((s/lib/python2.7/site-packages/numba/dispatcher.pys tsR tunsupported_errort interpretertconstant_inferenceRt reportableRN(!Rht isinstanceRtappendRRRt typeof_pyvalR9R:R R>t enumerateRRtargumentt ValueErrortstrRLtrstripRRtUnsupportedErrortNotDefinedErrortRedefinedErrortVerificationErrortConstantInferenceErrort ExceptionRRthasattrRR2(RR2R3RtargtypesRRAt failed_argsRtargRpttpt typeof_exctmsgR((s/lib/python2.7/site-packages/numba/dispatcher.pyt_compile_for_argsEsR        #c`sC|dk r&j|j}|jStfdjDS(Nc3`s$|]}|j|fVqdS(N(t inspect_llvm(RoR(R(s/lib/python2.7/site-packages/numba/dispatcher.pys s(RLRttlibraryt get_llvm_strtdictR(RRtlib((Rs/lib/python2.7/site-packages/numba/dispatcher.pyRs  c`sC|dk r&j|j}|jStfdjDS(Nc3`s$|]}|j|fVqdS(N(t inspect_asm(RoR(R(s/lib/python2.7/site-packages/numba/dispatcher.pys s(RLRtRt get_asm_strRR(RRR((Rs/lib/python2.7/site-packages/numba/dispatcher.pyRs  c K`s1|jdt}|jdd}|j}|d k rPi|j||6}n|s|d krntj}nxtj|D]g\}}td|j j |fd|tddd|t|j d|tddd|q~WnA|d k rt d nd d l m} | |d |d|Sd S(s" print or return annotated source with Numba intermediate IR Pass `pretty=True` to attempt color highlighting, and HTML rendering in Jupyter and IPython by returning an Annotate Object. `file` must be None if used in conjunction with `pretty=True`. tprettytstyleR-s%s %stfilet-iPt=s$`file` must be None if `pretty=True`i(tAnnotateRN(tgetR;RtRLtsyststdoutRt iteritemstprintR"Rttype_annotationRtpretty_annotateR( RRRRjRRRttvertresR((s/lib/python2.7/site-packages/numba/dispatcher.pyt inspect_typess"      c`s|dk rqj|}|j}dkr=|jj}n'dkrX|jj}n |jj}|j|Stfdj DS(s For inspecting the CFG of the function. By default the CFG of the user function is showed. The *show_wrapper* option can be set to "python" or "cfunc" to show the python wrapper function or the *cfunc* wrapper function, respectively. tpythontcfuncc3`s*|] }|j|dfVqdS(t show_wrapperN(t inspect_cfg(RoR(RR(s/lib/python2.7/site-packages/numba/dispatcher.pys sN( RLRtRtfndesctllvm_cpython_wrapper_nametllvm_cfunc_wrapper_namet mangled_nametget_function_cfgRR(RRRRORtfname((RRs/lib/python2.7/site-packages/numba/dispatcher.pyRs       cC`s|dkr|jn|g}tj}xe|D]]}|j|}|j}|jjdt|jj d|j f}|j |||sN(RLRtR4RR(RR((Rs/lib/python2.7/site-packages/numba/dispatcher.pyt get_metadatas N(%RRRR?R}R!RRR tweakreftWeakValueDictionaryRRrtdequeRtFUNCTION_CACHE_SIZER)RLR&RRtPipelineRR RRRRt classmethodR%RR tglobal_compiler_lockR9R1R2R9R:(((s/lib/python2.7/site-packages/numba/dispatcher.pyR{s.  (     t LiftedCodecB`sAeZdZeZdZdZdZej dZ RS(s Implementation of the hidden dispatcher objects used for lifted code (a lifted loop is really compiled as a separate function). c C`st||_d|_||_||_||_||_tj||jj |jj j |jj j dt dtdS(NRR(tfunc_irRLt lifted_fromRRRBR%RnRRRtfuncR'R?R;(RRCRRRBR%((s/lib/python2.7/site-packages/numba/dispatcher.pyRs       cC`s |jjjS(s5Return the starting line number of the loop. (RCtloctline(R((s/lib/python2.7/site-packages/numba/dispatcher.pytget_source_locationscC`sdS(sPre-compile actions N((RR2R6RB((s/lib/python2.7/site-packages/numba/dispatcher.pyt _pre_compile scC`s |j|j}tj|\}}|jjt|}|dk rS|jS|j ||||j j }t j d|jd|jd|d|d|d|d|jdd d |jd t }|jdk r|j r|jn|j||jSWdQXdS( NRRRCR2R6RBR%tliftedRDtis_lifted_loop((RRBR RRtRR:RLRRIRCtcopyRt compile_irRRR%RDR?RKRMR(RRRBR2R6R.tcloned_func_irRO((s/lib/python2.7/site-packages/numba/dispatcher.pyR9s(         ( RRRR;R}RRHRIRRAR9(((s/lib/python2.7/site-packages/numba/dispatcher.pyRBs    t LiftedLoopcB`seZdZRS(cC`s|j stddS(NsEnable looplift flags is on(tenable_loopliftRh(RR2R6RB((s/lib/python2.7/site-packages/numba/dispatcher.pyRI4s(RRRI(((s/lib/python2.7/site-packages/numba/dispatcher.pyRO3st LiftedWithcB`s eZedZdZRS(cC`s tj|S(N(RR{(R((s/lib/python2.7/site-packages/numba/dispatcher.pyR9scC`sn|jr|jt|nd}|jj}dj|}tj|d|d|j }||||fS(s Get a typing.ConcreteTemplate for this dispatcher and the given *args* and *kws* types. This enables the resolving of the return type. A (template, pysig, args, kws) tuple is returned. sCallTemplate({0})R@RN( RR9R:RLR"RRR RR(RR2R3R'RRYR((s/lib/python2.7/site-packages/numba/dispatcher.pyR=s  (RRR RR(((s/lib/python2.7/site-packages/numba/dispatcher.pyRQ8stObjModeLiftedWithcB`s2eZdZedZdZdZRS(cO`sk|jdd|_tt|j|||jjsItdn|jdkrgt dndS(Nt output_typess expecting `flags.force_pyobject`s`output_types` must be provided( tpopRLRSRSRQRRBtforce_pyobjectRR\(RR2Rj((s/lib/python2.7/site-packages/numba/dispatcher.pyRTs  cC`s tj|S(N(RtObjModeDispatcher(R((s/lib/python2.7/site-packages/numba/dispatcher.pyR\scC`s| s t|j|tjgt|}|jrO|jt|ntj |j |g}d}|j j }dj|}tj|d|d|}||||fS(s Get a typing.ConcreteTemplate for this dispatcher and the given *args* and *kws* types. This enables the resolving of the return type. A (template, pysig, args, kws) tuple is returned. sCallTemplate({0})R@RN(Rht_legalize_arg_typesRtffi_forced_objectRVRR9R:R RRSRLR"RRR(RR2R3RR'RRYR((s/lib/python2.7/site-packages/numba/dispatcher.pyR`s    cC`sxt|ddD]r\}}t|tjrRd}tj|j|qt|tjrd}tj|j|qqWdS(Ntstartis>Does not support list type inputs into with-context for arg {}sBDoes not support function type inputs into with-context for arg {}(RRRtListR R>RR{(RR2RRR((s/lib/python2.7/site-packages/numba/dispatcher.pyRWxs(RRRR RRRW(((s/lib/python2.7/site-packages/numba/dispatcher.pyRRSs  cc`s$|]}t||jfVqdS(N(RR(Rott((s/lib/python2.7/site-packages/numba/dispatcher.pys s(6t __future__RRRRrRtoststructRR"R;RLRtnumbaRRRRRR tnumba.typeconv.rulesR R R R tnumba.typing.templatesRtnumba.typing.typeofRRtnumba.bytecodeRt numba.sixRRtcachingRRtobjectRR!RRt namedtupleRbRfR{RnRBRORQRRt typeof_initRt number_domain(((s/lib/python2.7/site-packages/numba/dispatcher.pyts@       .T&`C6