o Td@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 )YMWDhmsmsusnspsfsasnatcCst|dkt|dkkS)zB Whether the *src* and *dest* units are of the same kind. rDATETIME_UNITSsrcdestr#n/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/np/npdatetime_helpers.py same_kind(sr%cCsXt|}t|}||krdS|dkrdS||krdS|dkr dS|dkr*|dkr*dSdS)NTrFrrr r#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|dkrdS|}d}||kr)zt|\}}Wn ty YdSw||9}||ks||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_codecfactormultr#r#r$_get_conversion_multiplierUs  r1cCstt|t|S)zl Return an integer multiplier allowing to convert from timedeltas of *src_unit* to *dest_unit*. )r1r)src_unit dest_unitr#r#r$get_timedelta_conversion_factorosr4cCst|}t|}|dks|dkr|ddfS|dkr%|dkr%td||fd\}}|dkrC|dkr8d}d }d}n$|dkrBd}d }d}n|dkr\|dkrRd}d }d}n |dkr\d}d }d}||krvt||}|d usoJ||f||||fSt||}|d usJ||f||||fS)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 RuntimeErrorr1) datetime_unittimedelta_unit dt_unit_code td_unit_code dt_factor td_factorr/r#r#r$!get_datetime_timedelta_conversionxsF   r<cCsLt|}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)r6r7r8r9r#r#r$ combine_datetime_timedelta_unitssr=cCs8t|}t|}|dkr|S|dkr|S||kr|S|S)z9 Get the best (i.e. finer-grained) of two units. rr)unit_aunit_babr#r#r$ get_best_unitsrBcCdSNr#r@rAr#r#r$datetime_minimumrFcCrCrDr#rEr#r#r$datetime_maximumrGrH)__doc__numpynpr timedelta64astypeint64NATr%r&r*r1r4r<r=rBrFrHr#r#r#r$sN  4