a .;b@sdZddlZdddddddd d d d d dddZedejZddZddZ ddddddddddd d! Z d"d#Z d$d%Z d&d'Z d(d)Zd*d+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 )YMWDhmsZmsusnsZpsZfsasZnatcCst|dkt|dkkS)zB Whether the *src* and *dest* units are of the same kind. rDATETIME_UNITSsrcdestr:lib/python3.9/site-packages/numba/np/npdatetime_helpers.py same_kind(sr!cCsXt|}t|}||krdS|dkr(dS||kr4dS|dkr@dS|dkrT|dkrTdSdS)NTrFrrrrrr can_cast_timedelta_units/sr")rr )rr)r)r<)rr$)r)r r%)r r%)r r%)r r%)r r%) rrrrrrrr r r r cCs`|dkr dS|}d}||krLzt|\}}Wnty@YdS0||9}q||krX|SdSdS)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)Z big_unit_codeZsmall_unit_codecfactorZmultrrr _get_conversion_multiplierUs   r*cCstt|t|S)zl Return an integer multiplier allowing to convert from timedeltas of *src_unit* to *dest_unit*. )r*r)Zsrc_unitZ dest_unitrrr get_timedelta_conversion_factorosr+cCst|}t|}|dks |dkr*|ddfS|dkrJ|dkrJtd||fd\}}|dkr|dkrpd}d }d}q|dkrd}d }d}n2|dkr|dkrd}d }d}n|dkrd}d }d}||krt||}|d usJ||f||||fSt||}|d us J||f||||fSd S)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. rrrz1cannot combine datetime64(%r) and timedelta64(%r))rrrri:ii ii@N)r RuntimeErrorr*) datetime_unittimedelta_unit dt_unit_code td_unit_codeZ dt_factorZ td_factorr)rrr !get_datetime_timedelta_conversionxsD   r1cCsPt|}t|}|dkr|S|dkr(|S|dkr<|dkrsH  4