\ K]c@@sddlmZmZddlmZddlZddlZddlZddl m Z ddl mZddl mZddl mZddlmZdd lmZmZdd lmZdd l mZmZmZdd lmZmZd dlmZdejfdYZ dejfdYZ!ej"ej#e ej"ej$e!ej"ej%ej&dZ'dZ(dZ)de*fdYZ+dZ,dZ-dZ.dZ/dej0fdYZ1dZ2de3fd YZ4ej5d!ej6fd"YZ7e4j8j9ej#d#Z:e4j8j;ej#d$Z<d%Z=e4j8j>ej%ej?ej@d&ZAdS('i(tabsolute_importtprint_function(t OrderedDictN(tir(ttypes(t cpu_target(tnjit(t templates(tdefault_managertmodels(timputils(tcgutilstutilsterrors(texec_tSequencei(t_boxt InstanceModelcB@seZdZRS(cC@srtj|}tjdt|}dtj|fdtj|fg}tt|j|||dS(NsOpaque.tmeminfotdata( Rt ClassDataTypetOpaquetstrtMemInfoPointertCPointertsuperRt__init__(tselftdmmtfe_typt cls_data_tytdtypetmembers((s2lib/python2.7/site-packages/numba/jitclass/base.pyRs (t__name__t __module__R(((s2lib/python2.7/site-packages/numba/jitclass/base.pyRstInstanceDataModelcB@seZdZRS(cC@s]|j}g|jjD]\}}t||f^q}tt|j|||dS(N(t class_typetstructtitemst _mangle_attrRR#R(RRRtclstytktvR ((s2lib/python2.7/site-packages/numba/jitclass/base.pyR)s 4(R!R"R(((s2lib/python2.7/site-packages/numba/jitclass/base.pyR#(scC@sd|S(sZ Mangle attributes. The resulting name does not startswith an underscore '_'. tm_((tname((s2lib/python2.7/site-packages/numba/jitclass/base.pyR'4ss3 def ctor({args}): return __numba_cls_({args}) cC@su|j}g}x_|jD]Q\}}|j|j@|jkrQ|j|qd|j}tj|qW|S(sI Returns list of positional and keyword argument names in order. s(%s argument type unsupported in jitclass(t parametersR&tkindtPOSITIONAL_OR_KEYWORDtappendR tUnsupportedError(tfn_sigtparamstargsR)R*tmsg((s2lib/python2.7/site-packages/numba/jitclass/base.pyt_getargsEs  t JitClassTypecB@s2eZdZdZdZdZdZRS(s# The type of any jitclass. cC@st|dkr!tdn|\}t|trHtdnd|ks`tdtj||||}|j|S(Nis must have exactly one base classscannot subclass from a jitclassR$smissing "class_type" attr(tlent TypeErrort isinstanceR7tAssertionErrorttypet__new__t _set_init(tclsR,tbasestdcttbasetoutcls((s2lib/python2.7/site-packages/numba/jitclass/base.pyR=Xs  cC@s|jjjd}tj|}t|d}||_tjddj |}i|d6}t |||d}t ||_ dS(s Generate a wrapper for calling the constructor from pure Python. Note the wrapper will only accept positional arguments. RiR4s, t __numba_cls_tctorN( R$t instance_typetmethodsR t pysignatureR6t _ctor_sigt_ctor_templatetformattjoinRRt_ctor(R?tinittinit_sigR4t ctor_sourcetglblsRE((s2lib/python2.7/site-packages/numba/jitclass/base.pyR>cs    cC@s)t|tjr%|jj|jkStS(N(R:RtBoxt _numba_type_R$tFalse(R?tinstance((s2lib/python2.7/site-packages/numba/jitclass/base.pyt__instancecheck__tscO@s<|jjd||}|j|j|jd|jS(Ni(RItbindtNonetapply_defaultsRMR4tkwargs(R?R4RZRW((s2lib/python2.7/site-packages/numba/jitclass/base.pyt__call__ys (R!R"t__doc__R=R>RVR[(((s2lib/python2.7/site-packages/numba/jitclass/base.pyR7Ts   cC@snxg|jD]Y\}}t|ts>td|fnt|tjs td|fq q WdS(Ns#spec keys should be strings, got %rs2spec values should be Numba type instances, got %r(R&R:RR9RtType(tspecR)R*((s2lib/python2.7/site-packages/numba/jitclass/base.pyt_validate_specs cC@sdt}xT|jD]F\}}|jdrR|jd rRd||}n|||s cs@s0|]&\}}t|tr||fVqdS(N(R:tproperty(RoR)R*((s2lib/python2.7/site-packages/numba/jitclass/base.pys s c3@s9|]/\}}|kr|kr||fVqdS(N((RoR)R*(RGtprops(s2lib/python2.7/site-packages/numba/jitclass/base.pys s sname shadowing: {0}s, R\ts(class members are not yet supported: {0}sdeleter is not supported: {0}tgettsetR$N(!R:RRR_RfR!treversedtinspecttgetmrotupdatet__dict__tdictR&Rtt NameErrorRKRLtpopt_drop_ignored_attrstkeysR9tfdelRXRtfgettfsettConstructorTemplateR7Rttyping_contextt insert_globalttarget_contexttregister(R?R^t class_ctortbuildertclsdcttbaseclstotherstshadowedt docstringR5R R)R*t jitmethodstjitpropsRAR$t typingctxt targetctx((RGRqs2lib/python2.7/site-packages/numba/jitclass/base.pytregister_class_typesT  % !       RcB@seZdZdZRS(s8 Base class for jitclass constructor templates. c C@sp|jj}|jd}|jf|}tj|}|j|j||}tj ||j d}|S(NRi( tkeyRFRtget_reference_typeRt Dispatchert get_call_typeRjRt signatureR4( RR4tkwsRFREt boundargst disp_typetsigRe((s2lib/python2.7/site-packages/numba/jitclass/base.pytgenerics  (R!R"R\R(((s2lib/python2.7/site-packages/numba/jitclass/base.pyRscC@stdddg}xm|jD]_\}}t|tjtjfrY|j|q"t|ddt kr"|j|q"q"Wx|D] }||=qWdS(Nt __weakref__R"Ryt __objclass__( RtR&R:RmtBuiltinFunctionTypetBuiltinMethodTypetaddRhRXtobject(RAtdropR)R*((s2lib/python2.7/site-packages/numba/jitclass/base.pyR}s   t ClassBuildercB@sGeZdZejZeZdZdZ dZ dZ RS(s A jitclass builder for a mutable jitclass. This will register typing and implementation hooks to the given typing and target contexts. cC@s(||_||_||_||_dS(N(R$RGRR(RR$RGRR((s2lib/python2.7/site-packages/numba/jitclass/base.pyRs   cC@s0|j|j|jj|jj|jdS(s7 Register to the frontend and backend. N(t_register_methodstclass_impl_registryR$RFRtinstall_registry(R((s2lib/python2.7/site-packages/numba/jitclass/base.pyR s  cC@sJxC|jD]8}||jkr |j|||jj|q q WdS(sN Register method implementations for the given instance type. N(Rtimplemented_methodst_implement_methodR(RtregistryRFtmeth((s2lib/python2.7/site-packages/numba/jitclass/base.pyR+sc@s~fdfd}jd}|dkrF||n4|jtjftjtjtjdS(Nc@sfd}|S(Nc @sp|jd}|j}tj|}|j||}|||}t||tj|||j|S(Ni( R4RRRt get_functionRlR timpl_ret_new_reft return_type( RjRRR4RFtmethodRRkRe(tattr(s2lib/python2.7/site-packages/numba/jitclass/base.pytimp;s    ((R(R(s2lib/python2.7/site-packages/numba/jitclass/base.pytget_imp:s c@sd|tt|}tj|dtjffdY}tjtjftjtj tj tj|tjtj tj dS(Ns__%s__t GetSetItemc@seZfdZRS(c@sg|d}t|tjrc|jkrc|j}tj|}|j|j||}|SdS(Ni(R:RtClassInstanceTypeRRRRj(RR4RRURRR(t _dunder_meth(s2lib/python2.7/site-packages/numba/jitclass/base.pyRLs  (R!R"R((R(s2lib/python2.7/site-packages/numba/jitclass/base.pyRJs( RhtoperatorRt infer_globaltAbstractTemplateR t lower_builtinRRtVarArgtAny(tgetsettopR(R(Rs2lib/python2.7/site-packages/numba/jitclass/base.pyt_getsetitem_genFs   Ratgetitemtsetitem(RR(tstriptlowerRRRR(RRRRtdunder_stripped((RRs2lib/python2.7/site-packages/numba/jitclass/base.pyR8s   ( R!R"R\R tRegistryRRtRRRRR(((s2lib/python2.7/site-packages/numba/jitclass/base.pyRs    tClassAttributecB@seZejZdZRS(c@sjkrjSjkr}j}tj|dtjffdY}tj|Sjkrj}|d}tj|jj fi}|j SdS(NtMethodTemplatec@s)eZjfZfdZRS(c@s5ft|}j|j||}|jS(N(ttupleRRjt as_method(RR4RR(RRU(s2lib/python2.7/site-packages/numba/jitclass/base.pyRxs(R!R"RR((RRRUR(s2lib/python2.7/site-packages/numba/jitclass/base.pyRusRs( R%RRRRRt BoundFunctionRRRjR(RRURRRtimpdcttgetterR((RRRURs2lib/python2.7/site-packages/numba/jitclass/base.pytgeneric_resolveks  (  (R!R"RRRR(((s2lib/python2.7/site-packages/numba/jitclass/base.pyRgs c C@s9||jkrz|j||d|}|j}|j||jd|}tj|||j|t|t|S||j kr |j |d}t j d|} t j|} | j|j|gi} |j| | } | ||g} t|| tj||| j| Stdj|dS(s4 Generic getattr() for @jitclass instances. tvaluetrefRssattribute {0!r} not implementedN(R%t make_helperRtmake_data_helpert get_data_typeR timpl_ret_borrowedRhR'RRRRXRRRRRRlRRtNotImplementedErrorRK( RjRttypRRtinstt data_pointerRRRt dispatcherRkRe((s2lib/python2.7/site-packages/numba/jitclass/base.pyt get_attr_impls$     cC@sk|j\}}|\}}||jkr|j||d|} | j} |j||jd| } |j|} t| t|} t| t|||j j || ||j j || | n||j krR|j |d}t j|}|j|j||fi}|j||}||||ft||ntdj|dS(s4 Generic setattr() for @jitclass instances. RRRtsattribute {0!r} not implementedN(R4R%RRRRRhR'tsetattrtnrttincreftdecrefRRRRRRRlRRK(RjRRR4RRtvaltyttargettvalRtdata_ptrRt attr_typetoldvaluetsetterRRk((s2lib/python2.7/site-packages/numba/jitclass/base.pyt set_attr_impls,     c C@s |jtj}|jtj}tjtj|||g}dj|j}|j |d|}|j rtj |j }|j } |j| } |j|jd| j} |j|| d| } |jj|| | j|jn|S(Ns _Dtor.{0}R,iR(tget_value_typeRtvoidptrtuintptllvmirRntVoidTypeRKR,tget_or_insert_functiontis_declarationt IRBuildertappend_basic_blockRtbitcastR4t as_pointerRRRt _getvaluetret_void( RjtmoduleRFt llvoidptrtllsizet dtor_ftypetfnametdtor_fnRt alloc_fe_typet alloc_typetptrR((s2lib/python2.7/site-packages/numba/jitclass/base.pytimp_dtors      cC@sl|j}|j|j}|j|}|jj||jtj|t||j |}|jj ||}|j ||j }|j tj|||j||} || _|| _|jf|j} |jd} tj| } |j| tj| } t|| | jgt|}| ||| j}tj||||S(s7 Generic constructor (__new__) for jitclasses. R(RRtget_abi_sizeofRtmeminfo_alloc_dtort get_constantRRRRt meminfo_dataRRtstoreR tget_null_valueRRRR4RRRtvoidRlRtlistR R(RjRRR4tinst_typRt alloc_sizeRRt inst_structRORNRRktrealargstret((s2lib/python2.7/site-packages/numba/jitclass/base.pyt ctor_impls0         (Bt __future__RRt collectionsRRRmRvRtllvmliteRRtnumbatnumba.targets.registryRRt numba.typingRtnumba.datamodelRR t numba.targetsR R R R t numba.sixRRRrRt StructModelRR#RRRt ClassTypet OpaqueModelR'RJR6R<R7R_RfRlRRRR}RRt infer_getattrtAttributeTemplateRRtlower_getattr_genericRtlower_setattr_genericRRRRRR (((s2lib/python2.7/site-packages/numba/jitclass/base.pytsJ     0 O U  (