\ K]c@@sRddlmZmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl mZm Z ddl mZmZddlmZddlmZmZdd l mZdd lmZd efd YZd efdYZdefdYZdefdYZdefdYZdS(i(tprint_functiontabsolute_import(t defaultdictN(ttypesterrors(t Conversiontrulesi(t templates(ttypeoftPurpose(tutils(tSequencetRatingcB@s)eZdZdZdZdZRS(tpromotet safe_converttunsafe_convertcC@sd|_d|_d|_dS(Ni(R RR(tself((s3lib/python2.7/site-packages/numba/typing/context.pyt__init__s  cC@s|j|j|jfS(s]Returns a tuple suitable for comparing with the worse situation start first. (RRR (R((s3lib/python2.7/site-packages/numba/typing/context.pytastuplescC@sbt|t|k rtSt}|j|j|_|j|j|_|j|j|_|S(N(ttypetNotImplementedR R RR(Rtothertrsum((s3lib/python2.7/site-packages/numba/typing/context.pyt__add__"s (R RR(t__name__t __module__t __slots__RRR(((s3lib/python2.7/site-packages/numba/typing/context.pyR s  t CallStackcB@sVeZdZdZdZdZejdZdZ dZ dZ RS(s# A compile-time call stack cC@sg|_tj|_dS(N(t_stackt threadingtRLockt_lock(R((s3lib/python2.7/site-packages/numba/typing/context.pyR1s cC@s|jt||dS(sy Returns item in the stack where index=0 is the top and index=1 is the second item from the top. i(Rtlen(Rtindex((s3lib/python2.7/site-packages/numba/typing/context.pyt __getitem__5scC@s t|jS(N(R R(R((s3lib/python2.7/site-packages/numba/typing/context.pyt__len__<scc@s~|j|j|r*d}t|n|jj|jjt|||z dVWd|jj|jj XdS(Ns2compiler re-entrant to the same function signature( tmatchtfunct RuntimeErrorRtacquireRtappendt CallFrametpoptrelease(Rt typeinfertfunc_idtargstmsg((s3lib/python2.7/site-packages/numba/typing/context.pytregister?s   cc@s/x(|D] }|jj|kr|VqqWdS(sg Yields frame that matches the function object starting from the top of stack. N(R-R%(Rtpy_functframe((s3lib/python2.7/site-packages/numba/typing/context.pytfinditerMs cC@s0yt|j|SWntk r+dSXdS(sZ Returns the first result from `.finditer(py_func)`; or None if no match. N(tnextR3t StopIteration(RR1((s3lib/python2.7/site-packages/numba/typing/context.pyt findfirstVs cC@s1x*|j|D]}|j|kr|SqWdS(s Returns first function that matches *py_func* and the arguments types in *args*; or, None if no match. N(R3R.(RR1R.R2((s3lib/python2.7/site-packages/numba/typing/context.pyR$_s( RRt__doc__RR"R#t contextlibtcontextmanagerR0R3R6R$(((s3lib/python2.7/site-packages/numba/typing/context.pyR,s    R)cB@s eZdZdZdZRS(s# A compile-time call frame cC@s||_||_||_dS(N(R,R-R.(RR,R-R.((s3lib/python2.7/site-packages/numba/typing/context.pyRms  cC@sdj|j|jS(NsCallFrame({}, {})(tformatR-R.(R((s3lib/python2.7/site-packages/numba/typing/context.pyt__repr__rs(RRR7RR;(((s3lib/python2.7/site-packages/numba/typing/context.pyR)is t BaseContextcB@sReZdZdZdZdZdZdZdZd#dZ dZ d Z d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ e!e"dZ#dZ$e!e!e"d Z%d!Z&d"Z'RS($sEA typing context for storing function typing constrain template. cC@s\i|_tt|_tt|_tj|_tj |_ t |_ |j dS(N(t _registriesRtlistt _functionst _attributesR t UniqueDictt_globalsRtdefault_type_managerttmRt callstacktinit(R((s3lib/python2.7/site-packages/numba/typing/context.pyRzs   cC@sdS(sQ Initialize the typing context. Can be overriden by subclasses. N((R((s3lib/python2.7/site-packages/numba/typing/context.pyRFtcC@s|j|jdS(sy Refresh context with new declarations from known registries. Useful for third-party extensions. N(tload_additional_registriest_load_builtins(R((s3lib/python2.7/site-packages/numba/typing/context.pytrefreshs c C@sg}g}t}t|tjrF|j\}}|j|nw||jkrxe|j|D]4}|p{t|d}|jt|dgqcWnd}|j |j d||rdg}x'|D]}|j dj |qWn|r |j dndj |S( sH Returns a string description of the type of a function tgenerictcasess2No type info available for {func!r} as a callable.R%sKnown signatures:s * {0}s * parameterizeds ( tFalset isinstanceRtCallabletget_call_signaturestextendR?thasattrtgetattrR(R:tjoin( RR%tdesctdefnstparamtsigsttplR/tsig((s3lib/python2.7/site-packages/numba/typing/context.pytexplain_function_types&   cC@sy|j|||}Wn"tjk r=}|}d}nXd}|dk rT|S|j|||}|dkr|dk r|n|S(st Resolve function type *func* for argument types *args* and *kws*. A signature is returned. N(t_resolve_user_function_typeRt TypingErrortNonet_resolve_builtin_function_type(RR%R.tkwstrestetlast_exception((s3lib/python2.7/site-packages/numba/typing/context.pytresolve_function_types   c C@s||jkr|j|}x|D]|}xsttgD]e}|rW|j||}n4g|D]}tj|^q^} |j| |}|dk r6|Sq6Wq#WndS(N(R?tTrueRMtapplyRt unliteralR^( RR%R.R`RVtdefntsupport_literalsRatat fixedargs((s3lib/python2.7/site-packages/numba/typing/context.pyR_s  " cC@s|j|}|dk r$|}nt|tjrj|j|d}|dk rj|j|||Snt|tjr|j|||SdS(Nt__call__( t_lookup_globalR^RNRtTypetresolve_getattrRdROt get_call_type(RR%R.R`tliteralstfunctyt func_type((s3lib/python2.7/site-packages/numba/typing/context.pyR\s   cc@s|||jkr/xf|j|D] }|VqWnIxFt|jD]5}||jkr?x|j|D] }|VqbWq?q?WdS(sE Get matching AttributeTemplates for the Numba type. N(R@Rt__mro__(Rttyptattrinfotcls((s3lib/python2.7/site-packages/numba/typing/context.pyt_get_attribute_templatess c@sfd}||}|dk r.|S|tj|}|dk rS|St|tjrj|}|dk r|SndS(s Resolve getting the attribute *attr* (a string) on the Numba type. The attribute's type is returned, or None if resolution failed. c@s@x9j|D](}|j|}|dk r|SqWdS(N(RxtresolveR^(RuRvtret(tattrR(s3lib/python2.7/site-packages/numba/typing/context.pytcores N(R^RRgRNtModuletresolve_module_constants(RRuR{R|touttattrty((R{Rs3lib/python2.7/site-packages/numba/typing/context.pyRos    cC@sRxK|j|D]:}|j||}|dk rtjtj||SqWdS(s Resolve setting the attribute *attr* (a string) on the *target* type to the given *value* type. A function signature is returned, or None if resolution failed. N(RxRyR^Rt signatureRtvoid(RttargetR{tvalueRvt expectedty((s3lib/python2.7/site-packages/numba/typing/context.pytresolve_setattrs cC@sDt|tj st|||f}d}|jd||S(Ntstatic_getitem((RNRRntAssertionErrorRd(RRR!R.R`((s3lib/python2.7/site-packages/numba/typing/context.pytresolve_static_getitem#s cC@sGt|tj st||||f}i}|jd||S(Ntstatic_setitem(RNRRnRRd(RRR!RR.R`((s3lib/python2.7/site-packages/numba/typing/context.pytresolve_static_setitem)scC@sRt|tjst||jtj}|j||||fi}|S(N(RNRRnRtresolve_value_typetoperatortsetitemRp(RRR!RtfntyRZ((s3lib/python2.7/site-packages/numba/typing/context.pytresolve_setitem/scC@s=||f}i}|jtj}|j|||}|S(N(RRtdelitemRp(RRR!R.R`RRZ((s3lib/python2.7/site-packages/numba/typing/context.pytresolve_delitem5s  cC@sSt|tjstt|j|}y|j|SWntk rNnXdS(sb Resolve module-level global constants. Return None or the attribute type N(RNRR}RRStpymodRt ValueError(RRuR{tattrval((s3lib/python2.7/site-packages/numba/typing/context.pyR~<s  cC@s]yt|tjSWnBtk rXtjj|rRttjj|tjSnXdS(s Return the numba type of a Python value that is being used as a function argument. Integer types will all be considered int64, regardless of size. ValueError is raised for unsupported types. N(RR targumentRtnumbatcudat is_cuda_arrayt as_cuda_array(Rtval((s3lib/python2.7/site-packages/numba/typing/context.pytresolve_argument_typeHs  cC@s~yt|tj}Wn"tk r:}tj|}nX|St|tjrU|S|j |}|dk rt|S|dS(s Return the numba type of a Python value that is being used as a runtime constant. ValueError is raised for unsupported types. N( RR tconstantRR terase_tracebackRNRtExternalFunctiont_get_global_typeR^(RRttyRbt typeof_exc((s3lib/python2.7/site-packages/numba/typing/context.pyRXs cC@sB|j|}|dk r|St|tjr>tj|SdS(N(RmR^RNtpytypest ModuleTypeRR}(RtgvR((s3lib/python2.7/site-packages/numba/typing/context.pyRqs  cC@sFddlm}m}m}ddlm}m}|jtjdS(Ni(tbuiltinst arraydeclt npdatetime(t ctypes_utilstbufproto( RGRRRRRtinstall_registryRtbuiltin_registry(RRRRRR((s3lib/python2.7/site-packages/numba/typing/context.pyRIxscC@sdS(sS Load target-specific registries. Can be overriden by subclasses. N((R((s3lib/python2.7/site-packages/numba/typing/context.pyRHRGcC@s>y|j|}Wn-tk r@tj|}||j|ARGiis s#Ambiguous overloading for %s %s: %sN( RR R.RR^R(RtsortRTtmaptstrR(RRRLR.R`tallow_ambiguousRRtoptionst candidatestcasetratingt best_ratetbestttiedRR/((s3lib/python2.7/site-packages/numba/typing/context.pytresolve_overload(s0    # ! cG@s`d}t|d|}|d}x4|dD](}|j||}|dkr0Pq0q0W|S(NcS@st|ddS(sfUses bitwidth to order numeric-types. Fallback to stable, deterministic sort. tbitwidthi(RS(tobj((s3lib/python2.7/site-packages/numba/typing/context.pytkeyfuncZsRii(tsortedt unify_pairsR^(RttypelistRtunifiedttp((s3lib/python2.7/site-packages/numba/typing/context.pyt unify_typesWs   cC@s>||kr|S|tjkr#|S|tjkr6|S|j||}|dk rX|S|j||}|dk rz|S|jd|d|}|dk r|tjkr|S|jd|d|}|dk r|tjkr|St|tjs t|tjr:tj |}tj |}|j ||SdS(sr Try to unify the two given types. A third type is returned, or None in case of failure. RRN( Rt undefinedtunifyR^RRRRNtLiteralRgR(RtfirsttsecondRR((s3lib/python2.7/site-packages/numba/typing/context.pyRgs.   $N((RRR7RRFRJR[RdR_R^R\RxRoRRRRRR~RRRRIRHRRmRRRRRRRReRMRRRRR(((s3lib/python2.7/site-packages/numba/typing/context.pyR<vsJ                         - tContextcB@seZdZRS(c C@sddlm}m}m}m}m}m}m}m}m } |j |j |j |j |j |j |j |j |j |j |j |j |j |j |j |j |j | j dS(Ni( t cffi_utilst cmathdecltenumdecltlistdecltmathdecltnpydeclt randomdecltsetdecltdictdecl( RGRRRRRRRR R RR( RRRRRRRRR R ((s3lib/python2.7/site-packages/numba/typing/context.pyRHs@ (RRRH(((s3lib/python2.7/site-packages/numba/typing/context.pyRs(t __future__RRt collectionsRRRRRR8RRRtnumba.typeconvRRRGRRR R t numba.sixR tobjectR RR)R<R(((s3lib/python2.7/site-packages/numba/typing/context.pyts(      =