dtUdZddlmZddlZddlZddlZddlmZddlZ ddl Z ddl m Z ddlmZmZmZddlmZddlmZdd lmZ ddlZn #e$rdZYnwxYwd Zd Zd Zd ed< e jje fZ!n#e"$re fZ!YnwxYwdZ#dZ$dZ%d1dZ&e&ddZ'e&Z(e&dZ)e'e(e)gZ*dZ+dZ,dZ-dZ.d2dZ/d Z0d!Z1d3d$Z2 d4d%Z3d5d&Z4Gd'd(e j5Z6d)Z7d*Z8d+Z9d,Z:d-Z;d.ZdS)6z0DatetimeIndex analog for cftime.datetime objects) annotationsN timedelta)Version)_STANDARD_CALENDARScftime_to_nptimeinfer_calendar_name)_contains_cftime_datetimes)OPTIONS) is_scalard ztuple[type[Exception], ...]OUT_OF_BOUNDS_TIMEDELTA_ERRORScd|zdz|zdzS)Nz(?P<>))namepatterns 9lib/python3.11/site-packages/xarray/coding/cftimeindex.pynamedrPs D=3  (3 ..cd|zdzS)Nz(?:z)?r)xs roptionalrTs 19t rcd|sdS|dtt|ddzS)Nr)rtrailing_optional)xss rr r Xs7 r a58-bf5566 66r\-T\:cd|ddf|ddf|ddf|ddf|ddfg}g}|D]0\}}}||r|ndt||z1d t|zd zS) N)Nyearz\d{4}monthz\d{2}dayhourminutesecondr^$)appendrr )date_sep datetime_septime_seppieces pattern_listsepr sub_patterns r build_patternr6^s 7H% 5(# vx( 8X& 8X& FL"(MMT;C/SSR5{3K3KKLLLL "<00 03 66rr)r/r1 )r0ctD]/}tj||}|r|cS0t d|)Nz4no ISO-8601 or cftime-string-like match for string: ) _PATTERNSrematch groupdict ValueError)datetime_stringrr;s rparse_iso8601_liker?tsb%%/22  %??$$ $ $ $ % PPP  rcttd|ddd}t|}i}dD].}||d}|t |||<|}/|jdi||fS)NNo module named 'cftime'r)r&r'r(r)r*r+r)cftimeModuleNotFoundErrorr?getintreplace) date_typetimestrdefaultresultrFattrvalue resolutions r_parse_iso8601_with_resorN~s ~!"<===i1a  G  ( (FGD 4&&   JJGDMJ 7? % %W % %z 11rc|dkr8||jdd||jdzddtdz fS|dkrv|jdkr'||jdzddtdz }n+||j|jdzdtdz }||j|jd|fS|dkr2||j|j|j}||tddzfS|d kr8||j|j|j|j}||tdd zfS|d kr>||j|j|j|j|j}||tdd zfS|d krD||j|j|j|j|j|j}||tddzfSt)zGeneralization of pandas.tseries.index.DatetimeIndex._parsed_string_to_bounds for use with non-standard calendars and cftime.datetime objects. r&r microsecondsr' r()daysrQr))hoursrQr*)minutesrQr+)secondsrQ)r&rr'r(r)r*r+KeyError)rGrMparsedendstarts r_parsed_string_to_boundsr\s# V Ifk1a ( ( IfkAoq! , ,ya/H/H/H H   w   <2  )FK!OQ22YA5N5N5NNCC)FK)91== AAACyflA66;; u   &+v|VZ@@eiQR@@@@@@ v   &+v|VZMMeiabAAAAAA x   Kvz6;   eiCCCCCC x   K L J K M M   eiCCCCCCrcDtjfd|DS)z(Adapted from pandas.tslib.get_date_fieldc0g|]}t|Sr)getattr).0datefields r z"get_date_field..s#@@@dWT5))@@@r)nparray) datetimesrbs `rget_date_fieldrgs( 8@@@@i@@@ A AAr0.0cL|ffd }|_||_t|S)z1Adapted from pandas.tseries.index._field_accessorc ttdttjt|krt |jSt ddd|dtjd)NrAzThe z!rz2 accessor requires a minimum version of cftime of z . Found an installed version of .)rBrCr __version__rg_data ImportError)selfmin_cftime_versionrs rfz_field_accessor..fs >%&@AA A 6% & &'2D*E*E E E!$*d33 3>t=>>(:>>(.(:>>> r)__name____doc__property)r docstringrprqs` r_field_accessorrvsB$6      AJAI A;;rcR|jjrt|jdSdS)Nr)rmsizetype)ros r get_date_typerzs' zDJqM"""trc~ttdt|dkr|d}t|t |tjs"t dtfd|Ds$t d|dSdS)NrArz?CFTimeIndex requires cftime.datetime objects. Got object of {}.c38K|]}t|VdSN) isinstance)r`rLrGs r z-assert_all_valid_date_type..s-BBE:eY//BBBBBBrzJCFTimeIndex requires using datetime objects of all the same type. Got {}.) rBrClenryr~datetime TypeErrorformatall)datasamplerGs @rassert_all_valid_date_typers ~!"<=== 4yy1}}aLL &&/22 --3VI->-> BBBBTBBBBB ::@&,, }  r, , cd|dz|tt|z|zS)z&Format a single row from format_times.r7)joinmapstr)timesindent separatorrow_ends r format_rowrs* C<)..S%99 9G CCrcFt|tt|zzd}tjt||z }d} t |D]B} | dkr|n|} | |dz kr|n|} || |z| dz|z} | t | | || z } C| S)z)Format values of cftimeindex as pd.Index.rrr)rrr)maxCFTIME_REPR_LENGTHrmathceilranger)index max_widthoffsetrfirst_row_offsetintermediate_row_end last_row_end n_per_rown_rowsrepresentationrowrr times_for_rows r format_timesrsI"4s9~~"EFJJI Ys5zzI- . .FNV}}  %(AXX!!6"%!"3"3,,9McIoqI0EEF * &Iw     rcd|jdt|d|jdt|dkr d|jdndd}d|D}||}|S)z.Format attributes of CFTimeIndex for __repr__.'N)dtypelengthcalendarfreqc"g|] \}}|d| S)=r)r`kvs rrcz format_attrs..s&6661A666r)rrrritemsr)rrattrs attrs_strs r format_attrsrs$U[###ZZ/))))%(ZZ1__!EJ!!!!$   E76 666I##I..I rceZdZdZeddZeddZeddZedd Zed d Z ed d Z eddZ edddZ edddZ edddZeeZd7dZdZdZdZdZdZfdZd7d Zd!Zd"Zd#Zd8d(Zd)Zd*Zd+Z d,Z!d9d.Z"d/Z#ed0Z$ed1Z%ed2Z&d3Z'd4Z(d5Z)d6Z*xZ+S): CFTimeIndexafCustom Index for working with CF calendars and dates All elements of a CFTimeIndex must be cftime.datetime objects. Parameters ---------- data : array or CFTimeIndex Sequence of cftime.datetime objects to use in index name : str, default: None Name of the resulting index See Also -------- cftime_range r&zThe year of the datetimer'zThe month of the datetimer(zThe days of the datetimer)zThe hours of the datetimer*zThe minutes of the datetimer+zThe seconds of the datetime microsecondz The microseconds of the datetimedayofyrz'The ordinal day of year of the datetimez1.0.2.1dayofwkzThe day of week of the datetime daysinmonthz/The number of days in the month of the datetimez1.1.0.0Nc t||t|dr|j}t|}t j|d|_||_i|_|S)NrOr) rhasattrrobject__new__rdrerm_cache)clsrrkwargsrJs rrzCFTimeIndex.__new__=sd"4((( ,>,> HIIG && %AAAAYAAA }   - -9~~!777%--c3IfQh3I3IJJ )VVWVV#vax.V)VVVMrct|j||\}}|j}|jrt |r7||dkr ||dks||dkr||dkrt ||d}||d}t||S||k}||k} tj || zS)a5Adapted from pandas.tseries.index.DatetimeIndex._partial_date_slice Note that when using a CFTimeIndex, if a partial-date selection returns a single element, it will never be converted to a scalar coordinate; this is in slight contrast to the behavior when using a DatetimeIndex, which sometimes will return a DataArray with a scalar coordinate depending on the resolution of the datetimes used in defining the index. For example: >>> from cftime import DatetimeNoLeap >>> da = xr.DataArray( ... [1, 2], ... coords=[[DatetimeNoLeap(2001, 1, 1), DatetimeNoLeap(2001, 2, 1)]], ... dims=["time"], ... ) >>> da.sel(time="2001-01-01") array([1]) Coordinates: * time (time) object 2001-01-01 00:00:00 >>> da = xr.DataArray( ... [1, 2], ... coords=[[pd.Timestamp(2001, 1, 1), pd.Timestamp(2001, 2, 1)]], ... dims=["time"], ... ) >>> da.sel(time="2001-01-01") array(1) Coordinates: time datetime64[ns] 2001-01-01 >>> da = xr.DataArray( ... [1, 2], ... coords=[[pd.Timestamp(2001, 1, 1, 1), pd.Timestamp(2001, 2, 1)]], ... dims=["time"], ... ) >>> da.sel(time="2001-01-01") array([1]) Coordinates: * time (time) datetime64[ns] 2001-01-01T01:00:00 rrSleft)sideright) r\rGrmis_monotonic_increasingrrX searchsortedslicerd flatnonzero) rorMrYr[rZrrrlhs_maskrhs_masks r_partial_date_slicezCFTimeIndex._partial_date_sliceosV.dnj&QQ s   ' &5zz q!!cE!HnnE"I%%#b //%%e&%99D&&s&99Eu%% %E>C<~h1222rct|j|\}} |||}n#t$rt|wxYw|S)zAAdapted from pandas.tseries.index.DatetimeIndex._get_string_slice)rNrGrrX)rokeyrYrMlocs r_get_string_slicezCFTimeIndex._get_string_slices`5dncJJ  **:v>>CC   3--   s 1A c||d|}||d|}t|j||jz }t|j||jz }|jr||k|dkz}n ||k|dkz}t j|||} |||| |} | S)z.Adapted from pandas.Index._get_nearest_indexerpad)limitbackfillrS) get_indexerabsrrrdwhere_filter_indexer_tolerance) rotargetr tolerance left_indexer right_indexerleft_distancesright_distances conditionindexers r_get_nearest_indexerz CFTimeIndex._get_nearest_indexers''U'CC ((5(II T[6FGGdk-86=HII  ' T'/9mr>QRII'?:}PR?RSI(9lMBB  44VWiPPGrct|tjr#t|j||jz }nt|j||z }t j||k|d}|S)z3Adapted from pandas.Index._filter_indexer_tolerancerS)r~pdIndexrrrdr)rorrrdistances rrz%CFTimeIndex._filter_indexer_tolerancesi fbh ' ' :4;w/&-?@@HH4;w/&899H(8y0'2>>rct|tr||St|S)z7Adapted from pandas.tseries.index.DatetimeIndex.get_loc)r~rrsuperget_loc)ror __class__s rrzCFTimeIndex.get_locs? c3   ())#.. .77??3'' 'rct|ts|St|j|\}}t |j||\}}|jrt |dkr |dkr|n|S|dkr|n|S)aJAdapted from pandas.tseries.index.DatetimeIndex._maybe_cast_slice_bound Note that we have never used the kind argument in CFTimeIndex and it is deprecated as of pandas version 1.3.0. It exists only for compatibility reasons. We can remove it when our minimum version of pandas is 1.3.0. rr)r~rrNrGr\is_monotonic_decreasingr)rolabelrkindrYrMr[rZs r_maybe_cast_slice_boundz#CFTimeIndex._maybe_cast_slice_bounds%%% L5dneLL -dnj&QQ s  ' 4CIIMM&..33e 3uuC/rcPtj|jtjtkr |j|St |t r1|j||j|j |j S|j| |S)z9Adapted from pandas.tseries.index.DatetimeIndex.get_value) rdasarrayrboolilocr~r slice_indexerr[stopstepr)roseriesrs r get_valuezCFTimeIndex.get_values} :c?? BHTNN 2 2;s# # U # # 2;t11#)SXsxPPQ Q;t||C001 1rc ||}t|p8t|tkp t |t jo|jS#tttf$rYdSwxYw)zKAdapted from pandas.tseries.base.DatetimeIndexOpsMixin.__contains__F) rr ryrr~rdndarrayrxrXrr=)rorrJs r __contains__zCFTimeIndex.__contains__s \\#&&F&!!D<<5(Dvrz22Bv{  )Z0   55 sAAA:9A:c,||S)z-Needed for .loc based partial-string indexing)r )rors rcontainszCFTimeIndex.containss  %%%rn int | floatrstr | timedeltact|tr|||zzSt|trddlm}||||zzSt d|)aEShift the CFTimeIndex a multiple of the given frequency. See the documentation for :py:func:`~xarray.cftime_range` for a complete listing of valid frequency strings. Parameters ---------- n : int, float if freq of days or below Periods to shift by freq : str or datetime.timedelta A frequency string or datetime.timedelta object to shift by Returns ------- CFTimeIndex See Also -------- pandas.DatetimeIndex.shift Examples -------- >>> index = xr.cftime_range("2000", periods=1, freq="M") >>> index CFTimeIndex([2000-01-31 00:00:00], dtype='object', length=1, calendar='standard', freq=None) >>> index.shift(1, "M") CFTimeIndex([2000-02-29 00:00:00], dtype='object', length=1, calendar='standard', freq=None) >>> index.shift(1.5, "D") CFTimeIndex([2000-02-01 12:00:00], dtype='object', length=1, calendar='standard', freq=None) r) to_offsetz9'freq' must be of type str or datetime.timedelta, got {}.)r~rrxarray.coding.cftime_offsetsrrr)ror rrs rshiftzCFTimeIndex.shiftsD dI & & !d(? " c " "  > > > > > >!iioo-- -55;VD\\ rct|tjr|}t t j||zSr}r~rTimedeltaIndexto_pytimedeltarrdreroothers r__add__zCFTimeIndex.__add__4sC eR. / / +((**E28D>>E1222rct|tjr|}t |t j|zSr}rrs r__radd__zCFTimeIndex.__radd__9sC eR. / / +((**E528D>>1222rc t|r$ttj||z St |t jr6ttj||z Sttj|rY t jtj|tj|z S#t$rtdwxYwtSNzcThe time difference exceeds the range of values that can be expressed at the nanosecond resolution.) _contains_datetime_timedeltasrrdrer~rrrr rr=NotImplementedrs r__sub__zCFTimeIndex.__sub__>s ( / / "rx~~566 6 r0 1 1 "rx~~0D0D0F0FFGG G ' 8 8 " ($"(5//)IJJJ1    J  " !s &:C!!C;c tj|tj|z S#t$rt dwxYwr)rrrdrerr=rs r__rsub__zCFTimeIndex.__rsub__NsX $URXd^^%;<< <-   F  s (+AFct|}t|}|tvr1|s/tjd|t dtj|S)aIf possible, convert this index to a pandas.DatetimeIndex. Parameters ---------- unsafe : bool Flag to turn off warning when converting from a CFTimeIndex with a non-standard calendar to a DatetimeIndex (default ``False``). Returns ------- pandas.DatetimeIndex Raises ------ ValueError If the CFTimeIndex contains dates that are not possible in the standard calendar or outside the nanosecond-precision range. Warns ----- RuntimeWarning If converting from a non-standard calendar to a DatetimeIndex. Warnings -------- Note that for non-standard calendars, this will change the calendar type of the index. In that case the result of this method should be used with caution. Examples -------- >>> times = xr.cftime_range("2000", periods=2, calendar="gregorian") >>> times CFTimeIndex([2000-01-01 00:00:00, 2000-01-02 00:00:00], dtype='object', length=2, calendar='standard', freq=None) >>> times.to_datetimeindex() DatetimeIndex(['2000-01-01', '2000-01-02'], dtype='datetime64[ns]', freq=None) zConverting a CFTimeIndex with dates from a non-standard calendar, {!r}, to a pandas.DatetimeIndex, which uses dates from the standard calendar. This may lead to subtle errors in operations that depend on the length of time between dates.r) stacklevel) rr rwarningswarnrRuntimeWarningr DatetimeIndex)rounsafenptimesrs rto_datetimeindexzCFTimeIndex.to_datetimeindexWsvN#4((&t,, . . .v . M ))    (((rcNtjfd|jDS)a Return an Index of formatted strings specified by date_format, which supports the same string format as the python standard library. Details of the string format can be found in `python string format doc `__ Parameters ---------- date_format : str Date format string (e.g. "%Y-%m-%d") Returns ------- pandas.Index Index of formatted strings Examples -------- >>> rng = xr.cftime_range( ... start="2000", periods=5, freq="2MS", calendar="noleap" ... ) >>> rng.strftime("%B %d, %Y, %r") Index(['January 01, 2000, 12:00:00 AM', 'March 01, 2000, 12:00:00 AM', 'May 01, 2000, 12:00:00 AM', 'July 01, 2000, 12:00:00 AM', 'September 01, 2000, 12:00:00 AM'], dtype='object') c:g|]}|Sr)strftime)r`ra date_formats rrcz(CFTimeIndex.strftime..s%KKK{33KKKr)rrrm)ror/s `rr.zCFTimeIndex.strftimes,8xKKKK KKKLLLrcddlm|dddtjfd|jDtjS)z@Convert to integers with units of microseconds since 1970-01-01.r) exact_cftime_datetime_differencercBg|]}t|Sr)_total_microseconds)r`raepochr1s rrcz$CFTimeIndex.asi8..s?   $$D$DUD$Q$QRR   rr)xarray.core.resample_cftimer1rGrdrerint64)ror5r1s @@rasi8zCFTimeIndex.asi8s{ QPPPPPtQ**x      K   (     rc$ddlm}||S)z0The calendar used by the datetimes in the index.r)r )xarray.coding.timesr )ror s rrzCFTimeIndex.calendars' <;;;;;""4(((rc$ddlm}||S)z-The frequency used by the dates in the index.r) infer_freq)xarray.coding.frequenciesr<)ror<s rrzCFTimeIndex.freqs& 988888z$rc ddlm}m}||}t||st |dt |}|j}|||}t||j |j S)z%Round dates using a specified method.r) CFTIME_TICKSrz is a non-fixed frequency) rr?rr~r=r4 as_timedeltar8_cftimeindex_from_i8rGr) rormethodr?rrunitrroundeds r_round_via_methodzCFTimeIndex._round_via_methodsHHHHHHHH4&,// CAAABB B"6#6#6#8#899&&&#GT^TYGGGrc8||tS)aRound dates down to fixed frequency. Parameters ---------- freq : str The frequency level to round the index to. Must be a fixed frequency like 'S' (second) not 'ME' (month end). See `frequency aliases `_ for a list of possible values. Returns ------- CFTimeIndex )rE _floor_intrors rfloorzCFTimeIndex.floors%%dJ777rc8||tS)aRound dates up to fixed frequency. Parameters ---------- freq : str The frequency level to round the index to. Must be a fixed frequency like 'S' (second) not 'ME' (month end). See `frequency aliases `_ for a list of possible values. Returns ------- CFTimeIndex )rE _ceil_intrHs rrzCFTimeIndex.ceils%%dI666rc8||tS)aRound dates to a fixed frequency. Parameters ---------- freq : str The frequency level to round the index to. Must be a fixed frequency like 'S' (second) not 'ME' (month end). See `frequency aliases `_ for a list of possible values. Returns ------- CFTimeIndex )rE_round_to_nearest_half_evenrHs rroundzCFTimeIndex.rounds%%d,GHHHrr})r r rr)F),rr __module__ __qualname__rsrvr&r'r(r)r*r+r dayofyear dayofweek days_in_monthrtrzrGrrrrrrrrrr r rrrr r"r+r.r8rrrErIrrN __classcell__)rs@rrrs ?6#= > >D OG%@ A AE /%!; < ? ?D _X'D E EF _X'D E EF!/-1STTK3>3>3@"(((((0000&222   &&&,,,,\333 333 """ 3)3)3)3)jMMM<   X  ))X)   X H H H888"777"IIIIIIIrrc,t||\}}|Sr})rN)rG datetime_strra_s r_parse_iso8601_without_resorXs&y,??GD! Krctjfd|D|jS)aCreate a numpy array from an array of strings. For use in generating dates from strings for use with interp. Assumes the array is either 0-dimensional or 1-dimensional. Parameters ---------- strings : array of strings Strings to convert to dates date_type : cftime.datetime type Calendar type to use for dates Returns ------- np.array c0g|]}t|Sr)rX)r`srGs rrcz2_parse_array_of_cftime_strings..s$LLLq $Y 2 2LLLr)rdreravelreshapeshape)stringsrGs `r_parse_array_of_cftime_stringsr` sG" 8LLLLGMMOOLLL   ggmrc`tj|}t|dtS)z.9s/UUUUeiSZZ@@@@UUUr)r)rdrer)rrGrdatesr5s @rrArA(sO IdAq ! !E HUUUUfUUU V VE u4 ( ( ((rc(|tdz S)zCompute the total number of microseconds of a datetime.timedelta. Parameters ---------- delta : datetime.timedelta Input timedelta. Returns ------- int rrPr)deltas rr4r4=s 9!,,, ,,rc2|tj||z SCopied from pandas.rd remainderrrCs rrGrGLs BL.. ..rc4|tj| |zSrirkrms rrKrKQs BL&$// //rc|dzrt||dzz |Stj||\}}tj||dzktj||dzk|dz}||xxdz cc<||zS)rjrr)rKrddivmod logical_or logical_and)rrCquotientrlmasks rrMrMVs ax3$!)+T222)FD11Hi =TQY dai0H(UV,!W!W  D TNNNaNNN d?r)r"r#r$)Nrh)rrr)rrrr)r)?rs __future__rrr:r%rrnumpyrdpandasrpackaging.versionrr:rrr xarray.core.commonr xarray.core.optionsr xarray.core.utilsr rBrnrrr__annotations__errorsOutOfBoundsTimedelta OverflowErrorrAttributeErrorrrr r6_BASIC_PATTERN_EXTENDED_PATTERN_CFTIME_PATTERNr9r?rNr\rgrvrzrrrrrrrXr`rrAr4rGrKrMrrrrs}666P#""""" %%%%%% :99999''''''''''''MMMM FFF "%%'"<;;;6&(i&Dm%T""666&3%5"""6///777 7 7 7 7 R888!MOO-S111 . @ 222$)))XBBB *&DDDD2    hIhIhIhIhI"(hIhIhIV ,+++ )))* - - -/// 000     s$A AA%A44A?>A?