gdZddlZdddddddd d d d d dddZejdejZdZdZ dddddddddddd Z d Z d!Z d"Z d#Zd$Zd%Zd&ZdS)'z Helper functions for np.timedelta64 and np.datetime64. For now, multiples-of-units (for example timedeltas expressed in tens of seconds) are not supported. N )YMWDhmsmsusnspsfsasnatcJt|dkt|dkkS)zB Whether the *src* and *dest* units are of the same kind. rDATETIME_UNITSsrcdests Q/granary/edev/quantum/lib/python3.11/site-packages/numba/np/npdatetime_helpers.py same_kindr&(s# 3 ! #)=)A BBct|}t|}||krdS|dkrdS||krdS|dkrdS|dkr|dkrdSdS)NTrFrr r"s r%can_cast_timedelta_unitsr)/sk  C $ D d{{t byyt Tzzu rzzu axxD1HHu 4r')rr )rr)r)r<)rr+)r )r r,)r r,)r r,)r r,)rr,) rrrrrrr r r r r c|dkrdS|}d}||kr. t|\}}n#t$rYdSwxYw||z}||k.||kr|SdS)z Return an integer multiplier allowing to convert from *big_unit_code* to *small_unit_code*. None is returned if the conversion is not possible through a simple integer multiplication. rrN)_factorsKeyError) big_unit_codesmall_unit_codecfactormults r%_get_conversion_multiplierr5UsqA F o   qkGAtt   44  $ o   O ts % 33cNtt|t|S)zl Return an integer multiplier allowing to convert from timedeltas of *src_unit* to *dest_unit*. )r5r!)src_unit dest_units r%get_timedelta_conversion_factorr9os( &nX&>&4Y&? A AAr'ct|}t|}|dks|dkr|ddfS|dkr|dkrtd|d|dd\}}|dkr|d krd }d }d }n,|dkrd }d }d}n|dkr|d krd }d }d }n |dkrd }d}d}||kr&t||}| J||f|||z|fSt||}| J||f||||zfS)a Compute a possible conversion for combining *datetime_unit* and *timedelta_unit* (presumably for adding or subtracting). Return (result unit, integer datetime multiplier, integer timedelta multiplier). RuntimeError is raised if the combination is impossible. rrrzcannot combine datetime64(z) and timedelta64())rrrri:ii ii@)r! RuntimeErrorr5) datetime_unittimedelta_unit dt_unit_code td_unit_code dt_factor td_factorr3s r%!get_datetime_timedelta_conversionrCxs"-0L!.1Lr\R//a""aLA--l+mm^^^=>> >Iyq 1  &IILL Q  &IIL    1  &I ILL Q  &I$IL|##+L,GG!!L,#?!!!y619<<+L,GG!!L,#?!!!iV);;;r'ct|}t|}|dkr|S|dkr|S|dkr|dkrdS||kr|S|S)z Return the unit result of combining *datetime_unit* with *timedelta_unit* (e.g. by adding or subtracting). None is returned if combining those units is forbidden. rrNr )r=r>r?r@s r% combine_datetime_timedelta_unitsrEsk "-0L!.1Lr   aLA--tl""r'cjt|}t|}|dkr|S|dkr|S||kr|S|S)z9 Get the best (i.e. finer-grained) of two units. rr )unit_aunit_babs r% get_best_unitrKsG vAvABww Bww 1uu Mr'cdSNrIrJs r%datetime_minimumrPDr'cdSrMrNrOs r%datetime_maximumrSrQr')__doc__numpynpr! timedelta64astypeint64NATr&r)r.r5r9rCrErKrPrSrNr'r%r[s?  $bnU""28,,CCC0   4AAA1<1<1