J/PhkddlZddlZddlZddlZddlmZmZmZddl m Z ddl m Z ddl mZddlmZeeeejdddZejr?ejd ejejd ejejd ejejd ejejd ejejdejejdejejdej ejdej!ejdej"ejdej#ejdej$ejdej%ejdej&eje'ej(iZ)n=ejd ej*ejd ej+ejd ej,ejd ej-ejd ej.ejdej/ejdej0ejdej1ejdej2ejdej3ejdej4ejdej5ejdej6ejdej7eje'ej(iZ)ej8dej9Z:ej8dej9Z;ejdj<Z=dZ>dZ?dZ@ejAdejBdejCdejDd iZEd!ZFd"ZGd#ZHd$ZId%ZJd&ZKd'ZLd(ZMd)ZNGd*d+ejOd,d-ZPd.ZQd?d0ZRd1ZSd2ZTd3ZUd4ZVd5ZWd@d6ZXd@d7ZYd8ZZd9Z[d:Z\d;Z]d<Z^d=Z_d>Z`dS)AN)errorstypesconfig) signature)npdatetime_helpers) TypingError) is_nonelike.boolint8int16int32int64uint8uint16uint32uint64float32float64float16 complex64 complex128z[<>=\|]([a-z])(\d+)?$z[<>=\|]([mM])8?(\[([a-z]+)\])?$U1c&t|j}|stj||}|d}|dkrc|jdvrtjd|jtz}|t|dks Jdtj |S|dkr>|j}|t|dks Jdtj |Stj|) NrUz=|z%Does not support non-native byteorderzUnicode char size mismatchSzChar size mismatch) re_typestrmatchstrrNumbaNotImplementedErrorgroups byteorderitemsizesizeof_unicode_charintrUnicodeCharSeqCharSeq)dtypemr#typecodecounts V/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/numba/np/numpy_support.py_from_str_dtyper/Es##A 5-e444 XXZZFayH3 ?$ & &13>?? ?"55F1I&&&(D&&&#E*** SF1I&&&(<&&&}U###-e444cHt|j}|stj||}|d}|dpd}|dkrt j|S|dkrt j|Stj|)Nrr r+M) re_datetimestrr r!rr"r#r NPTimedelta NPDatetime)r*r+r#r,units r._from_datetime_dtyper8^sUY''A 5-e444 XXZZFayH !9?D3 &&& S%%%-e444r0c,t|tur/t|tjrtj|}n t |ddt |S t|S#t$rYnwxYw |j }|dvrt|S|dvrt|S|dvr;|j 4t|j d}tj||jSn#t"$rYnwxYwt%j|)z Return a Numba Type instance corresponding to the given Numpy *dtype*. NumbaNotImplementedError is raised on unsupported Numpy dtypes. fieldsNSUmMVr)type issubclassnpgenericr*getattrfrom_struct_dtype FROM_DTYPEKeyErrorcharr/r8subdtype from_dtyper NestedArrayshapeAttributeErrorrr")r*rFsubtypes r.rHrHms0  E{{dz%<< $ ' ' 3 ''' %        ;z 4<<"5)) ) 4<<'.. . 3;;5>5 !233G$Wek:: :        )% 0 00s$' A44 BBC22 C?>C?M8m8rrctj|}t|tjtjtjfr!t jt|St|tj rt jdSt|tj tj frTtt|}|jrt j|d|jdSt j|St|tjtjfr8tt|}t jd||jfzSt|tjrt'|St|tjrt+|jSt|tjjrt+|jSt|tjrt+|jSt|tjrr7r)r(r-Recordas_struct_dtype EnumMemberas_dtypenpytypesDType NumberClassrItuplerJPyObjectobjectrr")nbtypeletterspecmsgs r.r]r]s/ _V $ $F&5=%-EFF%xF $$$&5=**x}}&5+U->?@@$"4<<0 ; $8 <== =8F## #&5=%*>?@@9"4<<0x&&,!77888&%,'''v&&&&%*++& %%%&%..//& %%%&%+,,& %%%&%+,,&&fl(;(;<x~~&%.)) x  < < TypeErrorrr"rH)valr*s r.map_arrayscalar_typers#rz"" N  NHT#YY''EE N N N13ACG993MNN N N e  s !A/A5c6t|tjSN)rTr@ndarray)rs r.is_arrayrs c2: & &&r0cJ|jdrd}n|jdrd}nd}|S)N C_CONTIGUOUSC F_CONTIGUOUSFA)flags)rlayouts r. map_layoutrs8 y  > " Mr0ctd}d}t|D]3\}}t|tjr|j|kr |}|j}4|J|S)z Given the array-compatible input types to an operation (e.g. ufunc), select the appropriate input for wrapping the operation output, according to each input's __array_priority__. An index into *inputs* is returned. z-infN)float enumeraterTrArrayCompatiblearray_priority)inputsmax_prioselected_indexindextys r.select_array_wrapperrsrV}}HNv&&)) r r50 1 1 )!H,,"N(H  % % % r0c|t|}||f}|d|i}|3|jtjjkr|St jd||jS)a. Given the array-compatible input types to an operation (e.g. ufunc), and the operation's formal output type (a types.Array instance), resolve the actual output type using the typing *context*. This uses a mechanism compatible with Numpy's __array_priority__ / __array_wrap__. __array_wrap__Nz__array_wrap__ failed for )rresolve_function_typerrArrayrr return_type)contextr formal_outputselected_inputargssigs r.resolve_output_typersz0889N = (D  ' '(8$ C CC {  (EK,F F F !   44!IJJJ ?r0cddlm}|j} |||v}nF#t$r9d|j|jzD}dtfd|D}YnwxYw|S)aRReturn whether the *loop* for the *ufunc* is supported -in nopython-. *loop* should be a UFuncLoopSpec instance, and *ufunc* a numpy ufunc. For ufuncs implemented using the ufunc_db, it is supported if the ufunc_db contains a lowering definition for 'loop' in the 'ufunc' entry. For other ufuncs, it is type based. The loop will be considered valid if it only contains the following letter types: '?bBhHiIlLqQfd'. Note this is legacy and when implementing new ufuncs the ufunc_db should be preferred, as it allows for a more fine-grained incremental support. r)ufunc_dbcg|] }|j S)rF.0xs r. z(supported_ufunc_loop..EsMMMafMMMr0z ?bBhHiIlLqQfdc3 K|]}|vV dSrr)rrxsupported_typess r. z'supported_ufunc_loop..Is(FFaQ/1FFFFFFr0)numba.npr ufunc_sigget_ufunc_inforE numpy_inputs numpy_outputsall)ufunclooprloop_sigsupported_loop loop_typesrs @r.supported_ufunc_loopr+s"!!!!!~HG "X%<%i") rc$d|jDS)Nc,g|]}t|Srr]rs r.rz.UFuncLoopSpec.numpy_inputs..\s111 111r0)rselfs r.rzUFuncLoopSpec.numpy_inputsZs11T[1111r0c$d|jDS)Nc,g|]}t|Srrrs r.rz/UFuncLoopSpec.numpy_outputs..`s222 222r0)outputsrs r.rzUFuncLoopSpec.numpy_outputs^s22T\2222r0N)__name__ __module__ __qualname____doc__ __slots__propertyrrrr0r.rrNsYI 22X233X333r0r_UFuncLoopSpec)rrrct|dkrt|dg|RSttj|g|RS)Nrr)lenrrTuple)out_tysin_tyss r._ufunc_loop_sigrcsM 7||q-f----W--77777r0safectj|}tj|}|r|jdvr |jdvrdStj|||S)aW A variant of np.can_cast() that can allow casting any integer to any real or complex type, in case the operation has mixed-kind inputs. For example we want `np.power(float32, int32)` to be computed using SP arithmetic and return `float32`. However, `np.sqrt(int32)` should use DP arithmetic and return `float64`. iucfT)r@r*kindcan_cast)from_tohas_mixed_inputscastings r.ufunc_can_castrjsU HUOOE "BEJ$..27d??t ;ub' * **r0c|d|j}||jd}t||jksJ d|D}n#tj$rYdSwxYw d|D}n#tj$rYdSwxYwt d|Dot d|D}d}d}|jD]*} | d|j} |jr| |j dng} d| vr0d | vr5d } t|| D]=\} }| jd vs|d vr| j|krd } n"t| j||d sd } n>| r:t|| D])\} }| jd vrt|| j|dsd } n*| rg ||| }||| }| ddks| dkr|||| |j }t||| cS#tj$rY&wxYw,dS)ajFind the appropriate loop to be used for a ufunc based on the types of the operands ufunc - The ufunc we want to check arg_types - The tuple of arguments to the ufunc, including any explicit output(s). return value - A UFuncLoopSpec identifying the loop, or None if no matching loop is found. Nc,g|]}t|Srrrs r.rz,ufunc_find_matching_loop..s;;;!(1++;;;r0c,g|]}t|Srrrs r.rz,ufunc_find_matching_loop..s===18A;;===r0c3(K|] }|jdvVdS)rNrrrs r.rz+ufunc_find_matching_loop..( 5 5BGtO 5 5 5 5 5 5r0c3(K|] }|jdvVdS)rNrrs r.rz+ufunc_find_matching_loop..rr0ct|t|ksJdt||D}|d|t|dDz }|S)z Return a list of Numba types representing *ufunc_letters*, except when the letter designates a datetime64 or timedelta64, in which case the type is taken from *numba_types*. cbg|],\}}|dvr|n ttj|-S)r<rHr@r*)rtpres r.rzBufunc_find_matching_loop..choose_types..sPDDDR~~:bhv6F6F+G+GDDDr0cPg|]#}ttj|$Srr)rres r.rzBufunc_find_matching_loop..choose_types..s@BBBRXf--..BBBr0N)rrz) numba_types ufunc_lettersrs r. choose_typesz.ufunc_find_matching_loop..choose_typess =!!S%5%55555DD#&{M#B#BDDD BB -c+.>.>.?.? @BBB B r0cd}fd}|dkr9tfdDrdj}|||}n|S|S|dkr'dj}dj} ||| |S|dkr'dj} dj}||| |S|dd krdj}|||}|Sd S) a Sets the output unit of a datetime type based on the input units Timedelta is a special dtype that requires the time unit to be specified (day, month, etc). Not every operation with timedelta inputs leads to an output of timedelta output. However, for those that do, the unit of output must be inferred based on the units of the inputs. At the moment this function takes care of two cases: a) where all inputs are timedelta with the same unit (mm), and therefore the output has the same unit. This case is used for arr.sum, and for arr1+arr2 where all arrays are timedeltas. If in the future this needs to be extended to a case with mixed units, the rules should be implemented in `npdatetime_helpers` and called from this function to set the correct output unit. b) where left operand is a timedelta, i.e. the "m?" case. This case is used for division, eg timedelta / int. At the time of writing, Numba does not support addition of timedelta and other types, so this function does not consider the case "?m", i.e. where timedelta is the right operand to a non-timedelta left operand. To extend it in the future, just add another elif clause. cg}|D]d}t|tjr3|jdkr(|tj|O||e|S)Nr2)rTrr5r7rp)rr7 new_outputsouts r. make_specificzLufunc_find_matching_loop..set_output_dt_units..make_specificstK , ,c5#455,#(b..&&u'8'>'>????&&s++++ r0cPg}|D]}t|tjrn|jdkrct j||}|$t dddzd|dzd|dz|tj||||S) Nr2zufunc 'z ' is not zsupported between z datetime64[z] zand timedelta64[rR)rTrr6r7r combine_datetime_timedelta_unitsrrp)rdt_unittd_unitrrr7 ufunc_names r.make_datetime_specificzUufunc_find_matching_loop..set_output_dt_units..make_datetime_specificsK , ,c5#344 ,R-N**D|)*IJ*I*I*I*>+?*C*C*C*C+D+HW*G*G*G+H,,,  &&u'7'='=>>>>&&s++++ r0mmc3DK|]}|jdjkVdS)rN)r7)rinprs r.rzHufunc_find_matching_loop..set_output_dt_units..s0@@#38vay~-@@@@@@r0rr<rMmr+N)rr7) rr ufunc_inputsrrrr7rrrs ` ` r.set_output_dt_unitsz5ufunc_find_matching_loop..set_output_dt_unitss22        " 4  @@@@@@@@@ ay~+mGT::   T ! !QinGQinG))'7GDD D T ! !QinGQinG))'7GDD D !_ # #!9>D'-66K  $ #r0eOTr<Fr same_kindrr+r) ninrrr"anyrnoutrzrFrrr)r arg_types input_types output_typesnp_input_typesnp_output_typesrrr candidater ufunc_outputsfoundouterinnerrrs r.ufunc_find_matching_loopr|sJUYJ'KUYZZ(L    ) ) ) );;{;;;  *tt== ===  *tt 5 5n 5 5 555 6 5 5n 5 5 555   OOOn[3A3A  %), 38:E 5:+,,//2 ,    ,    ==  LE5zT!!Ud]]:&&!EE'$EJ$4f>>     #O] C C   uJd**"5%*#3[BB+"EE  A A%k<@@&,|]CC ?c))\T-A-A11&'2>PPG%VWi@@@@@ 2     A( 4s5 AAA A++A>=A>=GG'&G'c|jSr)isalignedstruct)structs r._is_aligned_structr7s  !!r0c|jrd}tj|g}|jD][\}}|dd\}}t |dkr|dnd}t |}|||d} ||| f\|j} t|} tj || | S)z:Convert a NumPy structured dtype to Numba Record type z-dtypes that contain object are not supported.Nr )r>rqrr) hasobjectrr"r:itemsrrHrpr%rrrZ) r*rgr:nameinfo elemdtyperqrrrinfosrsrus r.rCrC;s 3=-c222 Fl((** % % d"2A2hFt99>>Qt  " "    tUm$$$$ >D ''G <g . ..r0ct|tjr|j}tj|z}||S)z; Get a ctypes array of *nbytes* starting at *ptr*. )rTctypesc_void_pvaluec_byte from_address)ptrnbytesarrtys r._get_bytes_bufferr)Ys?#v''i MF "E   c " ""r0cHtjt|||Sr)r@ frombufferr))r&r'r*s r._get_array_from_ptrr,cs =*377 ? ??r0cddlm} |j}n#t$rYnwxYw|t j|}t |tjr|td|}nt |tj r||j }t |tj sJt|j}|||krtd|d||}tj|tj}ntd||jt j|tjz}t'||||S) z Return a Numpy array view over the data pointed to by *ptr* with the given *shape*, in C order. If *dtype* is given, it is used as the array's dtype, otherwise the array's dtype is inferred from *ptr*'s type. r) from_ctypesNz*explicit dtype required for void* argumentzmismatching dtype 'z' for pointer zexpected a ctypes pointer, got )r*)numba.core.typing.ctypes_utilsr._as_parameter_rKr@r*rTr!r"r_Pointer __class__rCPointerr]castr%prodintpr,reshape)r&rJr*r.pptrty ptr_dtyper's r.carrayr;gsw ;:::::         #v''F =HII I  C ) ) F CM**%00000U[))  )!3!3)$uucc+,, , KV_ - -issDEEE ^bge27;;; ;F q&% 0 0 8 8 ? ??s  cnt|ts |ddd}t|||jS)z Return a Numpy array view over the data pointed to by *ptr* with the given *shape*, in Fortran order. If *dtype* is given, it is used as the array's dtype, otherwise the array's dtype is inferred from *ptr*'s type. N)rTr'r;T)r&rJr*s r.farrayr?s: eS ! !ddd  #ue $ $ &&r0cht|}|dz }|dkr#||dkr|dz}|dkr ||dk|dkrdS|||krdSd}||kr#||dkr|dz }||kr ||dk|}||kr+||||z||dz krdS|dz}||k+dS)zvIs the given shape, strides, and itemsize of C layout? Note: The code is usable as a numba-compiled function rr=rTFr)dimsstridesr%ndinneraxouteraxaxs r. is_contiguousrHs  TB1fG B,,4=A--1  B,,4=A--{{t77###uG G  W  2 21  G  W  2 2 B w,, 2;b !WR!V_ 4 45 a w,, 4r0cht|}d}||kr#||dkr|dz }||kr ||dk||krdS|||krdS|dz }||kr#||dkr|dz}||kr ||dk|}||kr+||||z||dzkrdS|dz }||k+dS)zvIs the given shape, strides, and itemsize of F layout? Note: The code is usable as a numba-compiled function rrTFrA)rBrCr%rDfirstaxlastaxrGs r. is_fortranrLs  TBG B,,4=A--1  B,,4=A--"}}t77###u!VF 7  tF|q00!  7  tF|q00 B v++ 2;b !WR!V_ 4 45 a v++ 4r0ctjtjtjtjtjtjtjjf}t||S)zs Returns True if the type of 'arr' is supported by the Numba `np.asarray` implementation, False otherwise. ) rrSequencer StringLiteralNumberrX containersListTyperT)arroks r.type_can_asarrayrUsA +u~u{E4G , u'7'@ BB c2  r0ctjtjtjtjtjtjf}t||S)z Returns True if the type of 'typ' is a scalar type, according to NumPy rules. False otherwise. https://numpy.org/doc/stable/reference/arrays.scalars.html#built-in-scalar-types )rrXrP UnicodeTyperOr5r6rT)typrTs r.type_is_scalarrYs9 -u'8%:M  U- /B c2  r0ct|ttjfs"t d|dS)z2Raises TypingError if the value is not an integer.z{} must be an integerN)rTr'rrVrr~)vrs r.check_is_integerr\sE a#u}- . .@188>>???@@r0c`||kp(tj|otj| Sr)r@isnanabs r. lt_floatsrbs* q5 5RXa[[4!_5r0cRtj|jratj|jrFtj|jrdStj|jrdS|j|jkSdStj|jrdStj|jr+tj|jr|j|jkSdStj|jrdS|j|jkrdS|j|jkr|j|jkSdS)NFT)r@r^realimagr_s r. lt_complexrfs x! 8AF   x +u8AF##+46AF?*5 8AF   !4x !8AF##!6AF?* 58AF##!4v#t16)) v. 5r0)rr)a collectionsr!renumpyr@ numba.corerrrnumba.core.typing.templatesrrrnumba.core.errorsrnumba.core.cgutilsr ramapr' __version__split numpy_versionUSE_LEGACY_TYPE_SYSTEMr*booleanr rrrrrrrrrrrrrcpyobjectrDnp_bool_np_int8np_int16np_int32np_int64np_uint8 np_uint16 np_uint32 np_uint64 np_float32 np_float64 np_float16 np_complex64 np_complex128compileIrr4r%r&r/r8rHr6r5r)r(rYr]r[rtrrrrrr namedtuplerrrrrrCr)r,r;r?rHrLrUrYr\rbrfrr0r.rs ,,,,,,,,,,111111''''''))))))+*****cc#r~33C88!<==>>   )%-%*5;5;5;5;ELELELU]U]U]u 0%.%JJ, %.%-5>5>5>5>EOEOEOU-U-U-u1 3%.%J,RZ0"$ 7 7 >EEbhtnn-5552 5 5 5111@ d t M3 # "/"/"/J///6 H H H   '''*.   F33333*K*+;+MOO333*888++++$xxxv"""///<###@@@#@#@#@#@L''''BB@@@ 666!!!!!r0