jTdZddlmZddlmZddlZddlmZddl m Z m Z m Z m Z mZerddlmZ dddZ dddZddZdS)zj Helper functions to generate range-like data for DatetimeArray (and possibly TimedeltaArray/PeriodArray) ) annotations) TYPE_CHECKINGN)i8max) BaseOffsetOutOfBoundsDatetime Timedelta TimestampiNaT)nptnsstartTimestamp | Timedelta | Noneendperiods int | Nonefreqrunitstrreturnnpt.NDArray[np.intp]cZ||jnd}||jnd}|jt|} ||d}n)#t$r}t d|d|d|d}~wwxYwt |j} ||||} | || z | z| zz| dzzdz} nF|||} t | || d } n,|||| z} t | || d } nt d tjd 5 tj | | | tj } nq#t$rd| g} | d| kr*| | d| z| d| k*tj | ddtj } YnwxYwdddn #1swxYwY| S)a Generate a range of dates or timestamps with the spans between dates described by the given `freq` DateOffset. Parameters ---------- start : Timedelta, Timestamp or None First point of produced date range. end : Timedelta, Timestamp or None Last point of produced date range. periods : int or None Number of periods in produced date range. freq : Tick Describes space between dates in produced date range. unit : str, default "ns" The resolution the output is meant to represent. Returns ------- ndarray[np.int64] Representing the given resolution. NF)round_okzfreq=z is incompatible with unit=z,. Use a lower freq or a higher unit instead.r )siderzEat least 'start' or 'end' should be specified if a 'period' is given.raiseover)dtype)_valuenanosras_unit ValueErrorint_generate_range_overflow_safenperrstatearangeint64FloatingPointErrorappendarray)r rrrristartiendtderrstridebevaluesxdrs nC:\Users\Terasoftware\OneDrive\Desktop\faahhh\fyndo\fyndo\venv\Lib\site-packages\pandas/core/arrays/_ranges.pygenerate_regular_ranger8s:# 8U\\DF 23::dDJJ 4B ZZuZ - -  9D 9 9T 9 9 9    ^^F 6 $  f$v- -! ;a ?      )!Wf7 K K K  g 6M )!Wf5 I I I S    ' " " " 8 8 8Yq!V28<<F "D#"F #A+FF FF  F$'F$endpointr%r2rcD|dvsJtjt}d|d|d|}tjd5 tj|tjtj|z}n"#t $r}t ||d}~wwxYw dddn #1swxYwYtj||krt||||S|dkr |d kr|dks|dcxkr|krnn|d krt ||d kr+||z |cxkr|krnnt||z |d z ||S|d z}||z } d| cxkr|ksnJ| |||ftt||||} t| | ||S) a[ Calculate the second endpoint for passing to np.arange, checking to avoid an integer overflow. Catch OverflowError and re-raise as OutOfBoundsDatetime. Parameters ---------- endpoint : int nanosecond timestamp of the known endpoint of the desired range periods : int number of periods in the desired range stride : int nanoseconds between periods in the desired range side : {'start', 'end'} which end of the range `endpoint` refers to Returns ------- other_end : int Raises ------ OutOfBoundsDatetime r rCannot generate range with = and periods=rrNrr rrr) r'uint64rr(absr+r$_generate_range_overflow_safe_signedr&r%) r9rr2ri64maxmsgaddendr1 mid_periods remainingmidpoints r7r&r&as8 # #### Yu  F O O Ox O Og O OC ' " " "44 4Yw''")BF6NN*C*CCFF! 4 4 4%c** 3 4  444444444444444 vf~~ 3HgvtTTT Q,  47?  vz  1           "&%-  "#&&&  8f,         - v w{FD   Q,K+%I yJJJJ7JJJJJY6$JJJJ0;PTUUVVH (9fd K KKs6B-;A>=B-> BBBB--B14B1c|dvsJ|dkr|dz}tjd5tj|tj|z} tj||z}|tkrtt |cdddS#t tf$rYnwxYw|dkr|dks|dkr|dksJ|dkrtj|tj|z}tjt}||ksJ||tj|zkrt |cdddSdddn #1swxYwYtd|d |d |) z A special case for _generate_range_overflow_safe where `periods * stride` can be calculated without overflowing int64 bounds. r;rr rrNrr<r=r>) r'r(r*r OverflowErrorr%r+r?rr)r9rr2rrDresulturesultrBs r7rArAs- # #### u}"  ' " " "$$'""RXf%5%55 Xh''&0F~ $$#v;;$$$$$$$$#M2    D  Ox1}O&1*OQOOO A: $ i))BIf,=,==GYu%%FV# # # #&29V#4#444 $7||9$$$$$$$$$$$$$$$$$$$$$$$< MdMMXMMGMM  s0*E7BB*'E)B**BEE E)r ) r rrrrrrrrrrr)r ) r9r%rr%r2r%rrrr%)__doc__ __future__rtypingrnumpyr'pandas._libs.librpandas._libs.tslibsrrrr r pandas._typingr r8r&rAr7rUs!#""""" """"""#"""""" FFFFFT;B@L@L@L@L@LF++++++rT