:jdZddlZdddddddd d d d d dddZejdj ej ZdZdZ dddddddddddd Z d Z d!Z d"Z d#Zd$Zd%Zd&Zy)'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 )YMWDhmsmsusnspsfsasnatc4t|dkt|dkk(S)zB Whether the *src* and *dest* units are of the same kind. rDATETIME_UNITSsrcdests p/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/numba/np/npdatetime_helpers.py same_kindr&(s# 3 ! #)=)A BBcnt|}t|}||k(ry|dk(ry||kDry|dk(ry|dkr|dkDryy)NTrFrr r"s r%can_cast_timedelta_unitsr)/sS  C $ D d{ by Tz rz axD1H r')rr )rr)r)r<)rr+)r )r r,)r r,)r r,)r r,)rr,) rrrrrrr r r r r c~|dk(ry|}d}||kr t|\}}||z}||kr||k(r|Sy#t$rYywxYw)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_multiplierr5UsnA F o  qkGAt $ o  O   s 0 <<c6tt|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&? AAr'c~t|}t|}|dk(s|dk(r|ddfS|dkr|dk\rtd|d|dd\}}|dk(r|d k\rd }d }d }n(|dk(r#d }d }d}n|dk(r|d k\rd }d }d }n |dk(rd }d}d}||k\rt||}| 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_conversionrCxsB"-0L!.1Lr\R/a""aLA-+^=> >Iyq 1 &IIL Q &IIL   1 &I IL Q &I$IL|#+L,G!?L,#??!y619<<+L,G!?L,#??!iV);;;r'cjt|}t|}|dk(r|S|dk(r|S|dkr|dk\ry||kDr|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_unitsrEsZ "-0L!.1Lr  aLA-l"r'cTt|}t|}|dk(r|S|dk(r|S||kDr|S|S)z9 Get the best (i.e. finer-grained) of two units. rr )unit_aunit_babs r% get_best_unitrKsA vAvABw Bw 1u Mr'cyNrIrJs r%datetime_minimumrPr'cyrMrNrOs r%datetime_maximumrSrQr')__doc__numpynpr! timedelta64astypeint64NATr&r)r.r5r9rCrErKrPrSrNr'r%r[s  $bnnU""288,C0  4A1