\ K]c@`s>dZddlmZmZmZddlZddlZddlmZm Z ddl mZm Z ddl m Z mZejd.kZd efd YZd Zdd Zd ZdZdefdYZdefdYZdefdYZdefdYZdefdYZdZdefdYZdZdefdYZ de fd YZ!d!e!fd"YZ"e!d#Z#d$Z$d%Z%d&efd'YZ&d(efd)YZ'd*efd+YZ(d,e(fd-YZ)e'Z*e*j+Z,e*j-Z.e*j/Z0dS(/s Define typing templates i(tprint_functiontdivisiontabsolute_importN(t MethodTypet FunctionTypei(ttypestutils(t TypingErrort InternalErrorit SignaturecB`seZdZdZddZdZdZdZd Z d Z d Z d Z e d ZdZdZRS(se The signature of a function call or operation, i.e. its argument types and return type. t return_typetargstrecvrtpysigcC`sFt|trt|}n||_||_||_||_dS(N(t isinstancetlistttupleR R R R (tselfR R R R ((s5lib/python2.7/site-packages/numba/typing/templates.pyt__init__s    c K`sDtd|jd|jd|jd|j}|j|t|S(snCopy and replace the given attributes provided as keyword arguments. Returns an updated copy. R R R R (tdictR R R R tupdateR (Rtkwargstcurstate((s5lib/python2.7/site-packages/numba/typing/templates.pytreplace"s     cC`s|j|j|j|jfS(s. Needed because of __slots__. (R R R R (R((s5lib/python2.7/site-packages/numba/typing/templates.pyt __getstate__-scC`s"|\|_|_|_|_dS(s. Needed because of __slots__. N(R R R R (Rtstate((s5lib/python2.7/site-packages/numba/typing/templates.pyt __setstate__3scC`st|j|jfS(N(thashR R (R((s5lib/python2.7/site-packages/numba/typing/templates.pyt__hash__9scC`sYt|trU|j|jkoT|j|jkoT|j|jkoT|j|jkSdS(N(RR R R R R (Rtother((s5lib/python2.7/site-packages/numba/typing/templates.pyt__eq__<s cC`s ||k S(N((RR((s5lib/python2.7/site-packages/numba/typing/templates.pyt__ne__CscC`sd|j|jfS(Ns%s -> %s(R R (R((s5lib/python2.7/site-packages/numba/typing/templates.pyt__repr__FscC`s |jdk S(sa Whether this signature represents a bound method or a regular function. N(R tNone(R((s5lib/python2.7/site-packages/numba/typing/templates.pyt is_methodIscC`sz|jdk r|St|jd|jd|jd}t|jjjd}t j d|d|jj |_|S(sE Convert this signature to a bound method signature. R iit parameterstreturn_annotationN( R R!t signatureR R RR R#tvaluesRt pySignatureR$(Rtsigtparams((s5lib/python2.7/site-packages/numba/typing/templates.pyt as_methodQs  cC`s6|jdkr|St|j|jf|j}|S(sI Convert this signature to a regular function signature. N(R R!R%R R (RR(((s5lib/python2.7/site-packages/numba/typing/templates.pyt as_functionbs(R R R R N(t__name__t __module__t__doc__t __slots__R!RRRRRRRR tpropertyR"R*R+(((s5lib/python2.7/site-packages/numba/typing/templates.pyR s        cC`s4tf}td|dt|}t|||S(Ntkeytcases(tConcreteTemplateRRttype(tnameR1t signaturest baseclassestgvars((s5lib/python2.7/site-packages/numba/typing/templates.pytmake_concrete_templatels c`sPfd}d|f}tf}td|d|d|}t|||S(sK Create a callable template with the given key and typer function. c`sS(N((R(ttyper(s5lib/python2.7/site-packages/numba/typing/templates.pytgenericvss%s_CallableTemplateR1R;R (tCallableTemplateRR4(R1R:R R;R5tbasest class_dict((R:s5lib/python2.7/site-packages/numba/typing/templates.pytmake_callable_templaters   cO`s2|jdd}| stt||d|S(NR (tpopR!tAssertionErrorR (R R tkwsR ((s5lib/python2.7/site-packages/numba/typing/templates.pyR%s c`s|j}g}x<|jD].\}} | j| jkr|j|qqW|rh|t| } n|} |j} |rx5t|D]$\} } |t|| | | s((R#titemstkindt KEYWORD_ONLYtappendtlentcopyt enumeratetbindR&R5tdefaulttVAR_POSITIONALRCtgettemptyRAR(R R RBtnormal_handlertdefault_handlertstararg_handlerR)tkwonlyR5tpt bind_argstbind_kwstidxtntiRERO((RFs5lib/python2.7/site-packages/numba/typing/templates.pytfold_argumentss4  "   "#tFunctionTemplatecB`s/eZeZeZdZdZdZRS(cC`s ||_dS(N(tcontext(RR_((s5lib/python2.7/site-packages/numba/typing/templates.pyRscC`s?i|jd6|jd6}|jj|j||||}|S(Ntunsafe_castingtexact_match_required(R`RaR_tresolve_overloadR1(RR2R RBtoptionstselected((s5lib/python2.7/site-packages/numba/typing/templates.pyt_selects    cC`sCt|j}t|tr?|jdks3t|j}n|S(sy Return the key for looking up the implementation for the given signature on the target context. N(R4R1RRtim_selfR!RAtim_func(RR(R1((s5lib/python2.7/site-packages/numba/typing/templates.pyt get_impl_keys  ( R,R-tTrueR`tFalseRaRReRh(((s5lib/python2.7/site-packages/numba/typing/templates.pyR^s   tAbstractTemplatecB`seZdZdZRS(s Defines method ``generic(self, args, kws)`` which compute a possible signature base on input types. The signature does not have to match the input types. It is compared against the input types afterwards. cC`st|d}|||}|dk r]t|ts]tdj|t|q]n| rtd|Drd}tt ||}| st|||}n|S(NR;s9generic() must return a Signature or None. {} returned {}cs`s!|]}t|tjVqdS(N(RRtOptional(RDtx((s5lib/python2.7/site-packages/numba/typing/templates.pys scS`s!t|tjr|jS|SdS(N(RRRlR4(Rm((s5lib/python2.7/site-packages/numba/typing/templates.pyt unpack_opts( tgetattrR!RR RAtformatR4tanyRtmap(RR RBR;R(Rn((s5lib/python2.7/site-packages/numba/typing/templates.pytapplys   (R,R-R.Rs(((s5lib/python2.7/site-packages/numba/typing/templates.pyRksR<cB`seZdZdZdZRS(sy Base class for a template defining a ``generic(self)`` method returning a callable to be called with the actual ``*args`` and ``**kwargs`` representing the call signature. The callable has to return a return type, a full signature, or None. The signature does not have to match the input types. It is compared against the input types afterwards. c C`st|d}|}|||}|dkrtd|Dryd}tt||}|||}n|dkrdSny |j}Wn tk rtj|}nX|j ||}|j rt dnt |t s2t |tjstd|fnt||j}n|jdk rP|j|_nt|jt|jkrt|jjt|j } |jd| }n||_|g} |j| |j|j S(NR;cs`s!|]}t|tjVqdS(N(RRRl(RDRm((s5lib/python2.7/site-packages/numba/typing/templates.pys scS`s!t|tjr|jS|SdS(N(RRRlR4(Rm((s5lib/python2.7/site-packages/numba/typing/templates.pyRn ssunsupported call signatures1invalid return type for callable template: got %rR#(RoR!RqRRrR tAttributeErrorRt pysignatureRNRRRR RtTypet TypeErrorR%R R RKR#R&RRe( RR RBR;R:R(RnR tboundR#R2((s5lib/python2.7/site-packages/numba/typing/templates.pyRss<       "  N(R,R-R.R!R Rs(((s5lib/python2.7/site-packages/numba/typing/templates.pyR<sR3cB`seZdZdZRS(sg Defines attributes "cases" as a list of signature to match against the given input types. cC`s"t|d}|j|||S(NR2(RoRe(RR RBR2((s5lib/python2.7/site-packages/numba/typing/templates.pyRs:s(R,R-R.Rs(((s5lib/python2.7/site-packages/numba/typing/templates.pyR34st_OverloadFunctionTemplatecB`s;eZdZdZdZdZdZdZRS(s; A base class of templates for overload functions. cC`s]tj|}tj|}d}||\}}}||\} } } d} | ||f} d}|}| }|r| s d|| f}t|q ny| r | | j|  }y||j|dd }Wq tk rd}||| |df}t|q Xntr2d}n d }||krsd | |||f}t||ng|D]}|j^qz}g| D]}|j^q}||krd }||| ||| f}t|ng|D]}|j^qg| D]}|j^q k}|sYd }||| ||| f}t|ndS( NcS`sg}g}d}x|jjD]}|jtjjkr|j||jtjj krh|}q|jtjj krd}t ||qq"|j|q"W|||fS(NsXThe use of VAR_KEYWORD (e.g. **kwargs) is unsupported. (offending argument name is '%s')( R!R#R&RORt pyParameterRRRJRHRPt VAR_KEYWORDR(R(RBR tpos_argRmtmsg((s5lib/python2.7/site-packages/numba/typing/templates.pytget_args_kwargsVs  s9Typing signature: %s Implementation signature: %ss.Typing and implementation arguments differ in sVAR_POSITIONAL (e.g. *args) argument kind (offending argument name is '%s') found in the typing function signature, but is not in the implementing function signature. %siis)argument names. %s First difference: '%s'cS`st|t|A}d|S(NsDifference: %s(tset(ttypingt implementingtdiff((s5lib/python2.7/site-packages/numba/typing/templates.pytgen_diffscS`sdS(N((RR((s5lib/python2.7/site-packages/numba/typing/templates.pyRssargument names. %s %sskeyword argument names. %s %ss&keyword argument default values. %s %s(RRuRtindext ValueErrort_IS_PY3R5RO(Rt typing_funct impl_funct typing_sigtimpl_sigR~tty_argstty_kwstty_postim_argstim_kwstim_postsig_fmttsig_strt err_prefixtatbR}t specializedRRmttytimtsame((s5lib/python2.7/site-packages/numba/typing/templates.pyt_validate_sigsDsP       8cC`sl|j||\}}|dkr(dStj|}|j|j||}|j||j|j<|S(sv Type the overloaded function by compiling the appropriate implementation for the given args. N( t _get_implR!Rt Dispatchert get_call_typeR_t get_overloadt_compiled_overloadsR (RR RBtdispt disp_typeR(((s5lib/python2.7/site-packages/numba/typing/templates.pyR;s cC`st|jt|t|jf}y|j|\}}Wn,tk ri|j|||\}}nX||fS(sGet implementation given the argument types. Returning a Dispatcher object. The Dispatcher object is cached internally in `self._impl_cache`. (R_RRGt _impl_cachetKeyErrort _build_impl(RR RBt cache_keytimpl((s5lib/python2.7/site-packages/numba/typing/templates.pyRs $ c C`s ddlm}|j||}|dkr?d|j|rs(Rtregistry_itemst_registrations(RR ((R s5lib/python2.7/site-packages/numba/typing/templates.pyRps cc`s'x t|j|D] }|VqWdS(N(tnextR (RR5R((s5lib/python2.7/site-packages/numba/typing/templates.pytnew_registrationsus(R,R-R.RR(((s5lib/python2.7/site-packages/numba/typing/templates.pyRas  tRegistryLoadercB`seZdZdZRS(s6 An incremental loader for a typing registry. RRR(RRR(R,R-R.R (((s5lib/python2.7/site-packages/numba/typing/templates.pyRzs(i(1R.t __future__RRRRRRRRRRterrorsRRt version_infoRtobjectR R9R!R?R%R]R^RkR<R3RyRRRRRRRRRRRRRtbuiltin_registryRtinferRt infer_getattrRt infer_global(((s5lib/python2.7/site-packages/numba/typing/templates.pytsD  \   3 !;  % (L,  "A