2@jdZddlmZddlZddlZddlmZddlmZdZ dZ dZ d d d Z d;d Z d ZdZe edZe edd7d e ej?d8d5d e ej@d9d6d e ejAd:d7d dS)=z_ Overloads for ClassInstanceType for built-in functions that call dunder methods on an object. )wrapsN)overload)ClassInstanceTypec<|dvsJtdd|S)N)xy)listn_argss VC:\Projects\Historical\venv\Lib\site-packages\numba/experimental/jitclass/overloads.py _get_argsr s( V     ::gvg cfd}|S)zr Decorator to add an overload for target that applies when the first argument is a ClassInstanceType. c tfd}turHttj|j}|t t|ksJt|S)NcNt|dtsdS|i|S)Nr) isinstancer)argskwargsfuncs r wrappedz;class_instance_overload..decorator..wrappeds4d1g'899 4((( (r) rcomplexr inspect signature parametersrlenr)rrparamstargets` r decoratorz*class_instance_overload..decorators t ) ) ) )  )  '+G44?@@FYs6{{333333x(((r)rrs` r class_instance_overloadr!s$ ) ) ) ) ) rc6i}t||||S)zC Extract a code-generated function from a string template. )exec)templatename namespaces r extract_templater''s#I9 T?rrr ct}dd|d}tt|dfd}t ||S)zd Register an overload for func that checks for methods __attr__ for each attr in attrs. def func(,z ): pass rc6fdD}t|S)NcFg|]}tdd|dS)r__try_call_method).0attrrr s r zCregister_simple_overload..overload_func..>sA    DG]$]]]F ; ;   r) take_first)rroptionsattrsr s` r overload_funcz/register_simple_overload..overload_func<s>        7##r)rjoinrr'r!)rr r5 arg_namesr$r6s `` r register_simple_overloadr90s &!!I ((9  H  Hf - -..$$$$$/.$ ) "4 ( ( 7 77rc ||jvrbt|}dd|d|dd|dd|ddd }t|d SdS) zo If method is defined for cls_type, return a callable that calls this method. If not, return None. r)r*z): return r.(rNz) r) jit_methodsrr7r')cls_typemethodr r8r$s r r/r/Gs %%%f%%  ((9   aL!$'HHYqrr]$;$; &111 &%rcD||jvrd|d}t|dSdS)z __complex__ needs special treatment as the argument names are kwargs and therefore specific in name and default value. z+ def func(real=0, imag=0): return real.z() rN)r=r')r>r?r$s r try_call_complex_methodrAUsE%%% &111 &%rcbtd|Ds J||D]}||cS dS)z) Take the first non-None option. c3FK|]}|duptj|VdSN)r isfunction)r0os r ztake_first..es6CCaqDy1G.q11CCCCCCrN)all)r4rFs r r3r3as[ CC7CCC C CLLWLL C  =HHH rcht|d}d|jvrd}nd}d}t|||S)N__bool____len__c:tt|SrD)boolrxs r using_len_implz"class_bool..using_len_implpsA<< rcdS)NTr rNs r zclass_bool..usrr/r=r3)rOusing_bool_implrPalways_true_impls r class_boolrVksT%a44OAM!!    %~ o~7G H HHrcBtt|dddS)N __complex__rc:tt|SrD)rfloatrealimags r rRzclass_complex..~swuT{{33rrr)r3rAr[s r class_complexr_zs* m44333  rc$t|ddS)N __contains__rr.rOys r class_containsrds 1na 0 00rctt|dg}d|jvr|dt|S)N __float__ __index__cDt|SrD)rZrgrNs r rRzclass_float..sq{{}}!5!5r)r/r=appendr3rOr4s r class_floatrksDq+../G q}$$55666 w rc|t|dg}|t|dt|S)N__int__rg)r/rir3rjs r class_intrns;q),,-G NN?1k22333 w rc@tt|ddS)N__str__c t|SrD)reprrNs r rRzclass_str..s $q''rr3r/rNs r class_strrts' 9%%  rcBtt|dddS)N__ne__rc||k SrDr rbs r rRzclass_ne..s !q&\rrsrbs r class_nerxs+ 8Q''!!  rcFfd}t||dS)Nctt|ddd}dd|jvrd}nd}t||S)Nr-rc||kSrDr rbs r reflected_implzEregister_reflected_overload..class_lt..reflected_impls 1u rrS)rOrc normal_implr| meth_forwardmeth_reflecteds r class_ltz-register_reflected_overload..class_ltsc%a)>l)>)>)>BB " " " "am 3 3    "N+~666r)r!)rr~rrs `` r register_reflected_overloadrsC 7 7 7 7 7 7"D!!(+++++rabsrhashgelegtlteqaddrfloordivlshiftmulmodnegposinvertpowrshiftsubtruediviadd ifloordivilshiftimulimodipowirshiftisubitruedivandorxoriandiorixor)rr^)B__doc__ functoolsrroperatornumba.core.extendingrnumba.core.typesrrr!r'r9r/rAr3rMrVrr_containsrdrZrkintrnstrrtnerxrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrand_or_rrrrr rr rs))))))...... *3488888. 2 2 2 2 2 2 2 I I I!!"!*++11,+1        %%&% , , ,e$$$e$$$v&&&HKt444HKt444HKt444HKt444HKt444uQ7777*JqAAAA(1====uQ7777uQ7777u---u---(333uQ7777(1====uQ7777)9Q????a@@@@+[*QOOOO)9hqIIIIa@@@@a@@@@a@@@@)9hqIIIIa@@@@*J !LLLLa8888tA6666uQ7777a@@@@ud1====a@@@@@@r