\ K]c@`s5ddlmZmZmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZmZddlmZmZddlmZmZmZmZddlmZmZmZmZejZd e fd YZ!e!d ddZ"d e fd YZ#de fdYZ$de$fdYZ%de%fdYZ&de&fdYZ'de%fdYZ(de&fdYZ)de&fdYZ*de&fdYZ+de&fdYZ,d e&fd!YZ-d"e&fd#YZ.d$e&fd%YZ/d&e'fd'YZ0d(e'fd)YZ1d*e'fd+YZ2d,e'fd-YZ3d.e'fd/YZ4d0e&fd1YZ5d2e&fd3YZ6d4e%fd5YZ7d6e&fd7YZ8d8e$fd9YZ9d:e$fd;YZ:d<e$fd=YZ;d>e$fd?YZ<d@e$fdAYZ=dBe fdCYZ>dDe fdEYZ?dFe fdGYZ@dHe fdIYZAdJe fdKYZBdLe fdMYZCdNe fdOYZDeDZEdS(Pi(tprint_functiontdivisiontabsolute_import(t defaultdictN(t FunctionTypetBuiltinFunctionType(tconfigterrorsi(tBINOPS_TO_OPERATORStINPLACE_BINOPS_TO_OPERATORStUNARY_BUITINS_TO_OPERATORStOPERATORS_TO_BUILTINS(tNotDefinedErrortRedefinedErrortVerificationErrortConstantInferenceErrortLoccB`seZdZejdZd edZe dZ dZ dZ dZ dZdZd Zd d Zd d ZRS(sSource location sdef\s+(\w+)\(.*cC`s1||_||_||_d|_||_dS(s Arguments: filename - name of the file line - line in file col - column maybe_decorator - Set to True if location is likely a jit decorator N(tfilenametlinetcoltNonetlinestmaybe_decorator(tselfRRRR((s'lib/python2.7/site-packages/numba/ir.pyt__init__s     cC`s||j|jdtS(NR(Rt firstlinenotTrue(tclstfunc_id((s'lib/python2.7/site-packages/numba/ir.pytfrom_function_id+scC`sd|j|j|jfS(Ns!Loc(filename=%s, line=%s, col=%s)(RRR(R((s'lib/python2.7/site-packages/numba/ir.pyt__repr__/s cC`sA|jdk r)d|j|j|jfSd|j|jfSdS(Ns %s (%s:%s)s%s (%s)(RRRR(R((s'lib/python2.7/site-packages/numba/ir.pyt__str__3scC`sWd}|j}x>t||jd D]%}|jjdr*|}Pq*q*W|S(Nisdef (Rt get_linestreversedRtstript startswith(Rtfn_nameRtx((s'lib/python2.7/site-packages/numba/ir.pyt_find_definition9s cC`s:|j}|r2|jj|jjdSdSdS(Ni(R&t _defmatchertmatchR"tgroupsR(Rtdefn((s'lib/python2.7/site-packages/numba/ir.pyt_raw_function_nameFs  cC`s1|jdkr*tj|j|_n|jS(N(RRt linecachetgetlinest _get_path(R((s'lib/python2.7/site-packages/numba/ir.pyR NscC`sLd}ytjj|j}Wn&tk rGtjj|j}nX|S(N(RtostpathtrelpathRt ValueErrortabspath(RR0((s'lib/python2.7/site-packages/numba/ir.pyR.Us  icC`sL|j}|j}|jrdg|}|r|rd||krtd|}|d}|||!}d}x0t|D]"\} } d| kr| }PqqW||}qng} |r|rd} |td|||!}t} x#|D]} d| krt} qqW| sd}x2t||d D]} d| kr8| }Pq8q8W|r| j || | }| j dd|d qn|r| j |d | j t j |d | |d }| j d|t j d qn| s d } nt jd d}||j|t jdj| f}|S(Ntsdef ii cS`s:d}x-tjtjt|D]}|d7}q"W|S(Nii(t itertoolst takewhiletstrtisspace(tstringtspacesR%((s'lib/python2.7/site-packages/numba/ir.pyt count_spacess"it is it^s#s File "%s", line %d:s %s(R RRtmaxt enumeratetFalseRRR!tappendtextendt _termcolort highlighttindicateRR.tcodetjoin(Rt nlines_upRtuse_linettmplinestmin_linetmax_linetselectedtindextidxR%tretR;t def_foundR$R:terrttmp((s'lib/python2.7/site-packages/numba/ir.pyt strformatcsT               $ +cC`st||j||S(s9 Return a new Loc with this line number. (ttypeR(RRR((s'lib/python2.7/site-packages/numba/ir.pyt with_linenosN(t__name__t __module__t__doc__tretcompileR'RR@Rt classmethodRRRR&R+R R.RTRV(((s'lib/python2.7/site-packages/numba/ir.pyRs       Osunknown locationtVarMapcB`sPeZdZdZdZdZdZdZdZdZ RS(cC`s i|_dS(N(t_con(R((s'lib/python2.7/site-packages/numba/ir.pyRscC`s/||jkrt|n ||j|sics`s%|]\}}d||fVqdS(s%s=%sN((RtkRz((s'lib/python2.7/site-packages/numba/ir.pys ss*%sR4s call %s(%s)Rs%s %s %scs`s%|]\}}d||fVqdS(s%s=%sN((RRRz((s'lib/python2.7/site-packages/numba/ir.pys ss%s(%s)(RRGRRtDIFF_IRRRtsortedRRtfilterRRRRtoperatortcontainsR Rc( RRt pres_orderRRtarglistRRR((s'lib/python2.7/site-packages/numba/ir.pyRs 3%!3cC`s|j|jS(N(RuR(R((s'lib/python2.7/site-packages/numba/ir.pyRpscC`std|d|jdS(Ns%sR(RR(R((s'lib/python2.7/site-packages/numba/ir.pytinfer_constantsN(RWRXRYRRRR\RRRRRRRRRRRRRRRRRRRRRpR(((s'lib/python2.7/site-packages/numba/ir.pyR#s4         tSetItemcB`s eZdZdZdZRS(s target[index] = value cC`s|t|tstt|ts*tt|ts?tt|tsTt||_||_||_||_dS(N(RqRrRRttargetRNRR(RRRNRR((s'lib/python2.7/site-packages/numba/ir.pyRs   cC`sd|j|j|jfS(Ns %s[%s] = %s(RRNR(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRR(((s'lib/python2.7/site-packages/numba/ir.pyRs t StaticSetItemcB`s eZdZdZdZRS(s( target[constant index] = value cC`st|tstt|t s+tt|ts@tt|tsUtt|tsjt||_||_||_||_||_dS(N( RqRrRRRRNRRR(RRRNRRR((s'lib/python2.7/site-packages/numba/ir.pyRs    cC`sd|j|j|jfS(Ns %s[%r] = %s(RRNR(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRR(((s'lib/python2.7/site-packages/numba/ir.pyRs tDelItemcB`s eZdZdZdZRS(s del target[index] cC`s^t|tstt|ts*tt|ts?t||_||_||_dS(N(RqRrRRRRNR(RRRNR((s'lib/python2.7/site-packages/numba/ir.pyRs   cC`sd|j|jfS(Ns del %s[%s](RRN(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRR(((s'lib/python2.7/site-packages/numba/ir.pyR s tSetAttrcB`seZdZdZRS(cC`s|t|tstt|ts*tt|ts?tt|tsTt||_||_||_||_dS(N( RqRrRR7RRRRR(RRRRR((s'lib/python2.7/site-packages/numba/ir.pyRs   cC`sd|j|j|jfS(Ns (%s).%s = %s(RRR(R((s'lib/python2.7/site-packages/numba/ir.pyR&s(RWRXRR(((s'lib/python2.7/site-packages/numba/ir.pyRs tDelAttrcB`seZdZdZRS(cC`s^t|tstt|ts*tt|ts?t||_||_||_dS(N(RqRrRR7RRRR(RRRR((s'lib/python2.7/site-packages/numba/ir.pyR+s   cC`sd|j|jfS(Ns del (%s).%s(RR(R((s'lib/python2.7/site-packages/numba/ir.pyR3s(RWRXRR(((s'lib/python2.7/site-packages/numba/ir.pyR*s tStoreMapcB`seZdZdZRS(cC`s|t|tstt|ts*tt|ts?tt|tsTt||_||_||_||_dS(N(RqRrRRtdcttkeyRR(RRRRR((s'lib/python2.7/site-packages/numba/ir.pyR8s   cC`sd|j|j|jfS(Ns %s[%s] = %s(RRR(R((s'lib/python2.7/site-packages/numba/ir.pyRBs(RWRXRR(((s'lib/python2.7/site-packages/numba/ir.pyR7s tDelcB`seZdZdZRS(cC`s@t|tstt|ts*t||_||_dS(N(RqR7RRRR(RRR((s'lib/python2.7/site-packages/numba/ir.pyRGs cC`s d|jS(Nsdel %s(R(R((s'lib/python2.7/site-packages/numba/ir.pyRMs(RWRXRR(((s'lib/python2.7/site-packages/numba/ir.pyRFs tRaisecB`s)eZeZdZdZdZRS(cC`sL|dks!t|ts!tt|ts6t||_||_dS(N(RRqRrRRt exceptionR(RRR((s'lib/python2.7/site-packages/numba/ir.pyRTs! cC`s d|jS(Nsraise %s(R(R((s'lib/python2.7/site-packages/numba/ir.pyRZscC`sgS(N((R((s'lib/python2.7/site-packages/numba/ir.pyR]s(RWRXRR~RRR(((s'lib/python2.7/site-packages/numba/ir.pyRQs  t StaticRaisecB`s/eZdZeZdZdZdZRS(s Raise an exception class and arguments known at compile-time. Note that if *exc_class* is None, a bare "raise" statement is implied (i.e. re-raise the current exception). cC`sv|dks!t|ts!tt|ts6t|dksWt|tsWt||_||_||_dS(N( RRqRURRRtt exc_classtexc_argsR(RRRR((s'lib/python2.7/site-packages/numba/ir.pyRis !!  cC`sZ|jdkrdS|jdkr0d|jfSd|jdjtt|jfSdS(Ntraisesraise %ss raise %s(%s)s, (RRRRGtmaptrepr(R((s'lib/python2.7/site-packages/numba/ir.pyRqs  cC`sgS(N((R((s'lib/python2.7/site-packages/numba/ir.pyRzs(RWRXRYRR~RRR(((s'lib/python2.7/site-packages/numba/ir.pyRas   tReturncB`s/eZdZeZdZdZdZRS(s Return to caller. cC`sLt|ts!tt|t|ts6t||_||_dS(N(RqRrRRURRR(RRR((s'lib/python2.7/site-packages/numba/ir.pyRs! cC`s d|jS(Ns return %s(R(R((s'lib/python2.7/site-packages/numba/ir.pyRscC`sgS(N((R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRR~RRR(((s'lib/python2.7/site-packages/numba/ir.pyR~s   tJumpcB`s)eZdZdZdZdZRS(s Unconditional branch. cC`s+t|tst||_||_dS(N(RqRRRR(RRR((s'lib/python2.7/site-packages/numba/ir.pyRs cC`s d|jS(Nsjump %s(R(R((s'lib/python2.7/site-packages/numba/ir.pyRscC`s |jgS(N(R(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRRR(((s'lib/python2.7/site-packages/numba/ir.pyRs  tBranchcB`s)eZdZdZdZdZRS(s Conditional branch. cC`sRt|tstt|ts*t||_||_||_||_dS(N(RqRrRRtcondttruebrtfalsebrR(RRRRR((s'lib/python2.7/site-packages/numba/ir.pyRs    cC`sd|j|j|jfS(Nsbranch %s, %s, %s(RRR(R((s'lib/python2.7/site-packages/numba/ir.pyRscC`s|j|jgS(N(RR(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRRR(((s'lib/python2.7/site-packages/numba/ir.pyRs  tAssigncB`s eZdZdZdZRS(s Assign to a variable. cC`s^t|tstt|ts*tt|ts?t||_||_||_dS(N(RqRmRRrRRRR(RRRR((s'lib/python2.7/site-packages/numba/ir.pyRs   cC`sd|j|jfS(Ns%s = %s(RR(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRR(((s'lib/python2.7/site-packages/numba/ir.pyRs tPrintcB`s eZdZdZdZRS(s Print some values. cC`std|Dst|dks=t|ts=tt|tsRtt||_||_i|_ ||_ dS(Ncs`s|]}t|tVqdS(N(RqRr(RR%((s'lib/python2.7/site-packages/numba/ir.pys s( tallRRRqRrRRtRRtconstsR(RRRR((s'lib/python2.7/site-packages/numba/ir.pyRs!  cC`sddjd|jDS(Ns print(%s)s, cs`s|]}t|VqdS(N(R7(RRz((s'lib/python2.7/site-packages/numba/ir.pys s(RGR(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRR(((s'lib/python2.7/site-packages/numba/ir.pyRs tYieldcB`s#eZdZdZdZRS(cC`sIt|tstt|ts*t||_||_||_dS(N(RqRrRRRRRN(RRRRN((s'lib/python2.7/site-packages/numba/ir.pyRs   cC`sd|jfS(Nsyield %s(R(R((s'lib/python2.7/site-packages/numba/ir.pyRscC`s |jgS(N(R(R((s'lib/python2.7/site-packages/numba/ir.pyRps(RWRXRRRp(((s'lib/python2.7/site-packages/numba/ir.pyRs  t EnterWithcB`s)eZdZdZdZdZRS(sEnter a "with" context cC`sRt|tstt|ts*t||_||_||_||_dS(s Parameters ---------- contextmanager : IR value begin, end : int The beginning and the ending offset of the with-body. loc : int Source location N(RqRrRRtcontextmanagertbegintendR(RRRRR((s'lib/python2.7/site-packages/numba/ir.pyRs    cC`sdj|jS(Ns enter_with {}(tformatR(R((s'lib/python2.7/site-packages/numba/ir.pyRscC`s |jgS(N(R(R((s'lib/python2.7/site-packages/numba/ir.pyRps(RWRXRYRRRp(((s'lib/python2.7/site-packages/numba/ir.pyRs  tArgcB`s#eZdZdZdZRS(cC`s^t|tstt|ts*tt|ts?t||_||_||_dS(N(RqR7RRRR_RNR(RR_RNR((s'lib/python2.7/site-packages/numba/ir.pyRs   cC`sd|j|jfS(Nsarg(%d, name=%s)(RNR_(R((s'lib/python2.7/site-packages/numba/ir.pyR scC`std|d|jdS(Ns%sR(RR(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRRR(((s'lib/python2.7/site-packages/numba/ir.pyRs  tConstcB`s&eZedZdZdZRS(cC`s4t|tst||_||_||_dS(N(RqRRRRtuse_literal_type(RRRR((s'lib/python2.7/site-packages/numba/ir.pyRs  cC`sdt|jj|jfS(Ns const(%s, %s)(RURRW(R((s'lib/python2.7/site-packages/numba/ir.pyRscC`s|jS(N(R(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRRRR(((s'lib/python2.7/site-packages/numba/ir.pyRs  tGlobalcB`s,eZdZdZdZdZRS(cC`s4t|tst||_||_||_dS(N(RqRRR_RR(RR_RR((s'lib/python2.7/site-packages/numba/ir.pyR!s  cC`sd|j|jfS(Nsglobal(%s: %s)(R_R(R((s'lib/python2.7/site-packages/numba/ir.pyR'scC`s|jS(N(R(R((s'lib/python2.7/site-packages/numba/ir.pyR*scC`s"t|j|jtj|jS(N(RR_RtcopytdeepcopyR(Rtmemo((s'lib/python2.7/site-packages/numba/ir.pyt __deepcopy__-s(RWRXRRRR(((s'lib/python2.7/site-packages/numba/ir.pyR s   tFreeVarcB`s)eZdZdZdZdZRS(sm A freevar, as loaded by LOAD_DECREF. (i.e. a variable defined in an enclosing non-global scope) cC`sgt|tstt|ts*tt|ts?t||_||_||_||_dS(N( RqRRR7RRNR_RR(RRNR_RR((s'lib/python2.7/site-packages/numba/ir.pyR9s   cC`sd|j|jfS(Nsfreevar(%s: %s)(R_R(R((s'lib/python2.7/site-packages/numba/ir.pyREscC`s|jS(N(R(R((s'lib/python2.7/site-packages/numba/ir.pyRHs(RWRXRYRRR(((s'lib/python2.7/site-packages/numba/ir.pyR3s RrcB`s8eZdZdZdZdZedZRS(st Attributes ----------- - scope: Scope - name: str - loc: Loc Definition location cC`sj|dks!t|ts!tt|ts6tt|tsKt||_||_||_dS(N( RRqtScopeRR7RtscopeR_R(RRR_R((s'lib/python2.7/site-packages/numba/ir.pyRXs !  cC`sd|j|jfS(Ns Var(%s, %s)(R_R(R((s'lib/python2.7/site-packages/numba/ir.pyRascC`s|jS(N(R_(R((s'lib/python2.7/site-packages/numba/ir.pyRdscC`s|jjdS(Nt$(R_R#(R((s'lib/python2.7/site-packages/numba/ir.pytis_tempgs(RWRXRYRRRtpropertyR(((s'lib/python2.7/site-packages/numba/ir.pyRrLs    RcB`s)eZdZdZdZdZRS(s A low-level "intrinsic" function. Suitable as the callable of a "call" expression. The given *name* is backend-defined and will be inserted as-is in the generated low-level IR. The *type* is the equivalent Numba signature of calling the intrinsic. cC`s(||_||_d|_||_dS(N(R_RURRR(RR_RUR((s'lib/python2.7/site-packages/numba/ir.pyRvs   cC`sd|j|j|jfS(NsIntrinsic(%s, %s, %s)(R_RUR(R((s'lib/python2.7/site-packages/numba/ir.pyR|scC`s|jS(N(R_(R((s'lib/python2.7/site-packages/numba/ir.pyRs(RWRXRYRRR(((s'lib/python2.7/site-packages/numba/ir.pyRls  RcB`sheZdZdZdZdZdZdZedZ dZ e dZ d Z RS( s Attributes ----------- - parent: Scope Parent scope - localvars: VarMap Scope-local variable map - loc: Loc Start of scope location cC`sg|dks!t|ts!tt|ts6t||_t|_||_t t |_ dS(N( RRqRRRtparentR]t localvarsRRRt redefined(RRR((s'lib/python2.7/site-packages/numba/ir.pyRs !   cC`s5td|d|d|}|jj|j||S(s# Define a variable RR_R(RrRRaR_(RR_RRz((s'lib/python2.7/site-packages/numba/ir.pyRascC`s6||jkr)d||j|f}n|j|S(sC Refer to a variable. Returns the latest version. s%s.%d(Rt get_exact(RR_((s'lib/python2.7/site-packages/numba/ir.pyRcscC`sHy|jj|SWn-tk rC|jr=|jj|SnXdS(sr Refer to a variable. The returned variable has the exact name (exact variable version). N(RRcR t has_parentR(RR_((s'lib/python2.7/site-packages/numba/ir.pyRs   cC`sw||jkr)d||j|f}ntd|d|d|}||jkrc|j||S|jj|SdS(Ns%s.%dRR_R(RRrRRaRc(RR_RRz((s'lib/python2.7/site-packages/numba/ir.pyt get_or_defines cC`s{||jkr|j||S|s5|jj|S|j|}|d|j||j|qtd|jD}td||fd|qWdS(Ntdumpcs`s|]}t|VqdS(N(R7(RRz((s'lib/python2.7/site-packages/numba/ir.pys %ss %-40s %stfile(tsyststdoutRthasattrRRRptprint(RRRt inst_vars((s'lib/python2.7/site-packages/numba/ir.pyRs cC`s |jdS(Ni(R(R((s'lib/python2.7/site-packages/numba/ir.pyt terminator(scC`s|jo|jdjS(Ni(RR}(R((s'lib/python2.7/site-packages/numba/ir.pyt is_terminated,scC`sL|jstdnx-|jd D]}|jr&tdq&q&WdS(NsMissing block terminatoris&Terminator before the last instruction(R RRR}(RR((s'lib/python2.7/site-packages/numba/ir.pytverify0s   cC`s-|jj|}|jj|d|dS(s. Insert *stmt* after *other*. iN(RRNR(RtstmttotherRN((s'lib/python2.7/site-packages/numba/ir.pyt insert_after9scC`s;t|tst|js$t|jjd|dS(Ni(RqR{RR RR(RR ((s'lib/python2.7/site-packages/numba/ir.pytinsert_before_terminator@scC`sd|jfS(Ns(R(R((s'lib/python2.7/site-packages/numba/ir.pyREsN(RWRXRYRRRRRRRRARRRRR R R RRR(((s'lib/python2.7/site-packages/numba/ir.pyRs"          tLoopcB`s&eZdZdZdZdZRS(sDescribes a loop-block tentrytexitcC`s||_||_dS(N(RR(RRR((s'lib/python2.7/site-packages/numba/ir.pyRNs cC`s|j|jf}d|S(NsLoop(entry=%s, exit=%s)(RR(RR((s'lib/python2.7/site-packages/numba/ir.pyRRs(RR(RWRXRYt __slots__RR(((s'lib/python2.7/site-packages/numba/ir.pyRIs tWithcB`s&eZdZdZdZdZRS(sDescribes a with-block RRcC`s||_||_dS(N(RR(RRR((s'lib/python2.7/site-packages/numba/ir.pyR\s cC`s|j|jf}d|S(NsWith(entry=%s, exit=%s)(RR(RR((s'lib/python2.7/site-packages/numba/ir.pyR`s(RR(RWRXRYRRR(((s'lib/python2.7/site-packages/numba/ir.pyRWs t FunctionIRcB`skeZdZdZd d edZdZdZdZ edZ d dZ d dZ RS( cC`sM||_||_||_||_||_||_||_|jdS(N(tblockst is_generatorRRt arg_countt arg_namest _definitionst_reset_analysis_variables(RRRRRt definitionsRR((s'lib/python2.7/site-packages/numba/ir.pyRgs       cC`sAddlm}|j||_d|_d|_i|_dS(Ni(R(R4RtConstantInferencet_constsRtgenerator_infotvariable_lifetimetblock_entry_vars(RR((s'lib/python2.7/site-packages/numba/ir.pyRts   cC`s|t|}tj|}||_|j|_|rFt|_n|dk r^||_n|dk rv||_n|j |j j |_ |S(s Derive a new function IR from this one, using the given blocks, and possibly modifying the argument count and generator flag. Post-processing will have to be run again on the new IR. N( tminRRRR@RRRRRRtderive(RRRRtforce_non_generatort firstblocktnew_ir((s'lib/python2.7/site-packages/numba/ir.pyR$s        cC`stj|}i}i}xV|jjD]E\}}|j}|||<||jkr+|j|||yield point #%d: live variables = %s, weak live variables = %s( RRR RRt state_varst yield_pointsRt live_varstweak_live_vars(RRtgiRNtyp((s'lib/python2.7/site-packages/numba/ir.pytdump_generator_infos ""N( RWRXRRRR@R$RR*RR-RR5(((s'lib/python2.7/site-packages/numba/ir.pyRes      t UndefinedTypecB`s eZdZdZdZRS(cC`s5|j}|dk r|Stj|}||_|S(N(t _singletonRtobjectt__new__(Rtobj((s'lib/python2.7/site-packages/numba/ir.pyR9s    cC`sdS(Nt Undefined((R((s'lib/python2.7/site-packages/numba/ir.pyRsN(RWRXRR7R9R(((s'lib/python2.7/site-packages/numba/ir.pyR6s (Ft __future__RRRt collectionsRRR5R/R,RgRZRtwarningsRttypesRRtnumbaRRtutilsRR R R R R RRt termcolorRCR8Rt unknown_locR]RmRnR{RRRRRRRRRRRRRRRRRRRRRRRrRRRRRRR6R(((s'lib/python2.7/site-packages/numba/ir.pytsh         ""  !   ^hy