a iDf$u@sUdZddlmZddlZddlZddlZddlmZddlZ ddl Z ddl m Z ddlmZmZmZddlmZddlmZdd lmZz ddlZWneydZYn0d Zd Zd Zd ed<ze jje fZ!Wne"ye fZ!Yn0ddZ#ddZ$ddZ%dGddZ&e&dddZ'e&Z(e&ddZ)e'e(e)gZ*ddZ+d d!Z,d"d#Z-d$d%Z.dHd'd(Z/d)d*Z0d+d,Z1dId/d0Z2dJd1d2Z3dKd3d4Z4Gd5d6d6e j5Z6d7d8Z7d9d:Z8d;d<Z9d=d>Z:d?d@Z;dAdBZdS)Lz0DatetimeIndex 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_ERRORScCsd|d|dS)Nz(?P<>))namepatternrrf/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/xarray/coding/cftimeindex.pynamedQsrcCs d|dS)Nz(?:z)?r)xrrroptionalUsrcCs$|sdS|dtt|ddS)Nr)rtrailing_optional)xsrrrrYsr\-T\:cCsnd|ddf|ddf|ddf|ddf|ddfg}g}|D]&\}}}||rL|ndt||q6d t|d S) N)Nyearz\d{4}monthz\d{2}dayhourminutesecondr^$)appendrr)date_sep datetime_septime_seppiecesZ pattern_listseprZ sub_patternrrr build_pattern_sr.r)r)r+ )r*cCs8tD] }t||}|r|Sqtd|dS)Nz4no ISO-8601 or cftime-string-like match for string: ) _PATTERNSrematch groupdict ValueError)Zdatetime_stringrr2rrrparse_iso8601_likeus r5cCsjtdurtd|ddd}t|}i}dD](}||d}|dur,t|||<|}q,|jfi||fS)NNo module named 'cftime'r)r r!r"r#r$r%)cftimeModuleNotFoundErrorr5getintreplace) date_typeZtimestrdefaultresultr;attrvalue resolutionrrr_parse_iso8601_with_resos   rBcCsp|dkr2||jdd||jdddtddfS|dkr|jdkrb||jdddtdd}n||j|jddtdd}||j|jd|fS|dkr||j|j|j}||tdddfS|d kr||j|j|j|j}||tddd fS|d kr,||j|j|j|j|j}||tddd fS|d krh||j|j|j|j|j|j}||tdddfStdS)zGeneralization of pandas.tseries.index.DatetimeIndex._parsed_string_to_bounds for use with non-standard calendars and cftime.datetime objects. r r microsecondsr! r")daysrDr#)hoursrDr$)minutesrDr%)secondsrDN)r rr!r"r#r$r%KeyError)r<rAparsedendstartrrr_parsed_string_to_boundssB    rOcstjfdd|DtjdS)z(Adapted from pandas.tslib.get_date_fieldcsg|]}t|qSr)getattr.0datefieldrr z"get_date_field..dtype)nparrayint64)Z datetimesrUrrTrget_date_fieldsr]0.0cs$|ffdd }|_||_t|S)z1Adapted from pandas.tseries.index._field_accessorcsRtdurtdttjt|kr.t|jStddd|dtjddS)Nr6zThe z!rz2 accessor requires a minimum version of cftime of z . Found an installed version of .)r7r8r __version__r]_data ImportError)selfmin_cftime_versionrrrfs  z_field_accessor..f)__name____doc__property)r docstringrdrfrrer_field_accessors rkcCs|jjrt|jdSdSdS)Nr)rasizetypercrrr get_date_typesrocsrtdurtdt|dkrn|d}t|t|tjsHtddtfdd|Dsntd|ddS)Nr6rzrWz-assert_all_valid_date_type..zGCFTimeIndex requires using datetime objects of all the same type. Got )r7r8lenrmrqdatetime TypeErrorall)datasamplerrsrassert_all_valid_date_types"  r{, , cCs|d|tt||S)z&Format a single row from format_times.r/)joinmapstr)timesindent separatorrow_endrrr format_rowsrcCst|tt|d}tt||}d} t|D]T} | dkrD|n|} | |dkrX|n|} || || d|} | t| | || d7} q4| S)z)Format values of cftimeindex as pd.Index.rrr)rrr)maxCFTIME_REPR_LENGTHrumathceilranger)index max_widthoffsetrfirst_row_offsetZintermediate_row_end last_row_endZ n_per_rowZn_rowsZrepresentationrowrrZ times_for_rowrrr format_timess   rcCsHd|jdt||j|jd}dd|D}||}|S)z.Format attributes of CFTimeIndex for __repr__.')rYlengthcalendarfreqcSsg|]\}}|d|qS)=r)rRkvrrrrVrWz format_attrs..)rYrurritemsr~)rrattrs attrs_strrrr format_attrss  rcsjeZdZdZeddZeddZeddZedd Zed d Z ed d Z eddZ edddZ edddZ edddZeeZdOddZddZddZdd Zd!d"Zd#d$Zfd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0d1d2d3Zd4d5Zd6d7Zd8d9Z d:d;Z!dPd=d>Z"d?d@Z#edAdBZ$edCdDZ%edEdFZ&dGdHZ'dIdJZ(dKdLZ)dMdNZ*Z+S)Q 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 datetimeZdayofyrz'The ordinal day of year of the datetimez1.0.2.1ZdayofwkzThe day of week of the datetimeZ daysinmonthz/The number of days in the month of the datetimez1.1.0.0NcKsJt||dur t|dr |j}t|}tj|dd|_||_i|_|S)NrOrX) r{hasattrrobject__new__rZr[ra_cache)clsryrkwargsr>rrrr>s zCFTimeIndex.__new__c Cs t|j}td}t|d}t|tkr>t|j||dd}nRt|jdt||ddd}t|jt d|||d}d|d |d |g}t |}|d |d |d }t||krt|||kr| ddd |d}|d |dd |d|d }|S)zA Return a string representation for this object. display_widthr)rrN,)rrr r/z...z([z], rr}z], r) rmrgr ruITEMS_IN_REPR_MAX_ELSE_ELLIPSISrvalues"REPR_ELLIPSIS_SHOW_ITEMS_FRONT_ENDr~rr;) rc klass_namerrZdatastrZ front_strZend_strrZ full_repr_strrrr__repr__Is8     "zCFTimeIndex.__repr__c Cst|j||\}}|j}|jrt|rZ||dkr>||dksV||dkrZ||dkrZt|j|dd}|j|dd}t||S||k}||k} t || @S)aYAdapted 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") Size: 8B array([1]) Coordinates: * time (time) object 8B 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") Size: 8B array(1) Coordinates: time datetime64[ns] 8B 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") Size: 8B array([1]) Coordinates: * time (time) datetime64[ns] 8B 2001-01-01T01:00:00 rrFleft)sideright) rOr<rais_monotonic_increasingrurKZ searchsortedslicerZZ flatnonzero) rcrArLrNrMrrrZlhs_maskZrhs_maskrrr_partial_date_sliceps&+     zCFTimeIndex._partial_date_slicecCs@t|j|\}}z|||}Wnty:t|Yn0|S)zAAdapted from pandas.tseries.index.DatetimeIndex._get_string_slice)rBr<rrK)rckeyrLrAlocrrr_get_string_slices  zCFTimeIndex._get_string_slicec Cs|j|d|d}|j|d|d}t|j||j}t|j||j}|jr`||k|dkB}n||k|dkB}t|||} |dur||| |} | S)z.Adapted from pandas.Index._get_nearest_indexerpad)limitZbackfillrFN)Z get_indexerabsrrrZwhere_filter_indexer_tolerance) rctargetr toleranceZ left_indexerZ right_indexerZleft_distancesZright_distances conditionindexerrrr_get_nearest_indexersz CFTimeIndex._get_nearest_indexercCsJt|tjr"t|j||j}nt|j||}t||k|d}|S)z3Adapted from pandas.Index._filter_indexer_tolerancerF)rqpdIndexrrrZr)rcrrrZdistancerrrrs  z%CFTimeIndex._filter_indexer_tolerancecs$t|tr||St|SdS)z7Adapted from pandas.tseries.index.DatetimeIndex.get_locN)rqrrsuperget_locrcr __class__rrrs  zCFTimeIndex.get_loccCsbt|ts|St|j|\}}t|j||\}}|jrRt|dkrR|dkrN|S|S|dkr^|S|S)zXAdapted from pandas.tseries.index.DatetimeIndex._maybe_cast_slice_bound rr)rqrrBr<rOZis_monotonic_decreasingru)rclabelrrLrArNrMrrr_maybe_cast_slice_bounds z#CFTimeIndex._maybe_cast_slice_boundcCsXt|jttkr |j|St|trD|j||j|j |j S|j| |SdS)z9Adapted from pandas.tseries.index.DatetimeIndex.get_valueN) rZZasarrayrYboolZilocrqrZ slice_indexerrNstopstepr)rcZseriesrrrr get_values   zCFTimeIndex.get_valuec CsRz2||}t|p0t|tkp0t|tjo0|jWStt t fyLYdS0dS)zKAdapted from pandas.tseries.base.DatetimeIndexOpsMixin.__contains__FN) rr rmrrqrZZndarrayrlrKrwr4)rcrr>rrr __contains__s  zCFTimeIndex.__contains__cCs ||S)z-Needed for .loc based partial-string indexing)rrrrrcontainsszCFTimeIndex.containsz int | floatzstr | timedelta)nrcCsPt|tr|||St|tr>> index = xr.cftime_range("2000", periods=1, freq="ME") >>> index CFTimeIndex([2000-01-31 00:00:00], dtype='object', length=1, calendar='standard', freq=None) >>> index.shift(1, "ME") 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_offsetz6'freq' must be of type str or datetime.timedelta, got r_N)rqrrxarray.coding.cftime_offsetsrrw)rcrrrrrrshifts"     zCFTimeIndex.shiftcCs&t|tjr|}tt||SrprqrTimedeltaIndexto_pytimedeltarrZr[rcotherrrr__add__0s zCFTimeIndex.__add__cCs&t|tjr|}t|t|Srprrrrr__radd__5s zCFTimeIndex.__radd__cCst|rtt||St|tjr>> 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) zBConverting a CFTimeIndex with dates from a non-standard calendar, z, 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) rarlrZ DatetimeIndexrrrwarningswarnRuntimeWarning)rcunsafeZnptimesrrrrto_datetimeindexSs(   zCFTimeIndex.to_datetimeindexcstfdd|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') csg|]}|qSr)strftimerQ date_formatrrrVrWz(CFTimeIndex.strftime..)rrra)rcrrrrrszCFTimeIndex.strftimecsTddlm|jjs$tjgtjdS|dddtjfdd|jDtjdS)z@Convert to integers with units of microseconds since 1970-01-01.r) exact_cftime_datetime_differencerXrcsg|]}t|qSr)_total_microsecondsrQepochrrrrVsz$CFTimeIndex.asi8..) Zxarray.core.resample_cftimerrarlrZr[r\r<rrnrrrasi8s  zCFTimeIndex.asi8cCs ddlm}|jjsdS||S)z0The calendar used by the datetimes in the index.r)rN)xarray.coding.timesrrarl)rcrrrrrs zCFTimeIndex.calendarcCs$ddlm}|jjdkrdS||S)z-The frequency used by the dates in the index.r) infer_freqN)Zxarray.coding.frequenciesrrarl)rcrrrrrs  zCFTimeIndex.freqc Csrddlm}m}|jjs&tt|S||}t||sFt |dt | }|j }|||}t ||j|jS)z%Round dates using a specified method.r) CFTIME_TICKSrz is a non-fixed frequency)rrrrarlrrZr[rqr4rZ as_timedeltar_cftimeindex_from_i8r<r) rcrmethodrrrunitrZroundedrrr_round_via_methods   zCFTimeIndex._round_via_methodcCs ||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 )r _floor_intrcrrrrfloorszCFTimeIndex.floorcCs ||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 )r _ceil_intrrrrrszCFTimeIndex.ceilcCs ||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 )r_round_to_nearest_half_evenrrrrroundszCFTimeIndex.round)N)F),rg __module__ __qualname__rhrkr r!r"r#r$r%rZ dayofyearZ dayofweekZ days_in_monthriror<rrrrrrrrrrrrrrrrrrrrrrrrr __classcell__rrrrrsX         '@    - 9   rcCst||\}}|Srp)rB)r<Z datetime_strrS_rrr_parse_iso8601_without_resosrcs$tfdd|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 csg|]}t|qSr)r)rRsrsrrrV+rWz2_parse_array_of_cftime_strings..)rZr[ZravelZreshapeshape)stringsr<rrsr_parse_array_of_cftime_stringss rcCst|}t|dtS)z.re)rZr[r)rr<rdatesrrrr5s rcCs|tddS)zCompute the total number of microseconds of a datetime.timedelta. Parameters ---------- delta : datetime.timedelta Input timedelta. Returns ------- int rrCr)deltarrrrJs rcCs|t||SCopied from pandas.rZ remainderrrrrrrYsrcCs|t| |Sr r rrrrr^srcCsj|drt||d|St||\}}t||dkt||dk|d}||d7<||S)r rr)rrZdivmod logical_or logical_and)rrZquotientr maskrrrrcs r)rrr)Nr^)rr|r})r|rr}r)r|)?rh __future__rrr1rrvrnumpyrZZpandasrZpackaging.versionrrrrrZxarray.core.commonr Zxarray.core.optionsr Zxarray.core.utilsr r7rbrrr__annotations__errorsZOutOfBoundsTimedelta OverflowErrorrAttributeErrorrrrr.Z_BASIC_PATTERNZ_EXTENDED_PATTERNZ_CFTIME_PATTERNr0r5rBrOr]rkror{rrrrrrrrrrrrrrrrrst)                ,    z