U |e@sdZddlZdddddddd d d d d dddZedejZddZddZ ddddddddddd d! Z d"d#Z d$d%Z d&d'Z d(d)Zd*d+Zd,d-Zd.d/ZdS)0z 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 )YMWDhmsmsusnspsfsasZnatcCst|dkt|dkkS)zB Whether the *src* and *dest* units are of the same kind. rDATETIME_UNITSsrcdestr"X/var/www/website-v5/atlas_env/lib/python3.8/site-packages/numba/np/npdatetime_helpers.py same_kind(sr$cCsXt|}t|}||krdS|dkr(dS||kr4dS|dkr@dS|dkrT|dkrTdSdS)NTrFrrrr"r"r#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 cCsb|dkr dS|}d}||krNzt|\}}Wntk rBYdSX||9}q||krZ|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_codecfactormultr"r"r#_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_unitr"r"r#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 k st||f||||fSt||}|d k s t||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.AssertionError) datetime_unittimedelta_unit dt_unit_code td_unit_codeZ dt_factorZ td_factorr,r"r"r#!get_datetime_timedelta_conversionxsD   r6cCsPt|}t|}|dkr|S|dkr(|S|dkr<|dkr)__doc__numpynpr timedelta64astypeint64ZNATr$r%r)r.r/r6r7r:r=r>r"r"r"r#sL  4