o 5c@srdZddlmZddlZddlmZddlmZm Z m Z m Z m Z ddl mZdddZ ddddZdddZdS)zj Helper functions to generate range-like data for DatetimeArray (and possibly TimedeltaArray/PeriodArray) ) annotationsN)i8max) BaseOffsetOutOfBoundsDatetime Timedelta TimestampiNaT)nptstartTimestamp | Timedelta | Noneendperiods int | Nonefreqrreturnnpt.NDArray[np.intp]c Csv|dur|jnd}|dur|jnd}|j}|dur4|dur4|dur4|}||||||dd}n,|durG|durG|}t|||dd}n|dur\|dur\||}t|||dd}ntdtjdd Kz tj|||tjd } Wn-ty|g} | d |kr| | d || d |kstj | dd tjd } Yn wWd| SWd| S1swY| 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. Returns ------- ndarray[np.int64] Representing nanoseconds. Nr )sider zEat least 'start' or 'end' should be specified if a 'period' is given.raiseover)dtype) valuenanos_generate_range_overflow_safe ValueErrornperrstatearangeint64FloatingPointErrorappendarray) r r r ristartiendstridebevaluesxdrr,P/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/core/arrays/_ranges.pygenerate_regular_rangesB"        r.endpointintr'rstrc Csr|dvsJtt}d|d|d|}tjdd)zt|tt|}Wnty=}zt||d}~wwWdn1sHwYt||kr[t||||S|dkrg|d krg|dkss|dkrw|d krw|dkrwt||d kr||kr|||krt|||d ||S|d }||} d| kr|ksnJ| |||ft||||} 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 r Cannot generate range with = and periods=rrNrr r rr) ruint64rrabsr"r$_generate_range_overflow_safe_signedr) r/r r'ri64maxmsgaddenderr mid_periods remainingmidpointr,r,r-rNs>   &rc Cs6|dvsJ|dkr|d9}tjddrt|t|}zt||}|tkr-t|WWdSttfyAYnw|dkrJ|dksT|dkrR|dksTJ|dkrt|t|}tt}||ksmJ||t|kr|WdSWdn1swYtd|d |d |) z A special case for _generate_range_overflow_safe where `periods * stride` can be calculated without overflowing int64 bounds. r2r rrrNrr3r4r5) rrr!r OverflowErrorr"r6rr)r/r r'rr;resultr9r,r,r-r8s6  $  $r8) r r r r r rrrrr)r ) r/r0r r0r'r0rr1rr0)__doc__ __future__rnumpyrpandas._libs.librpandas._libs.tslibsrrrrrpandas._typingr r.rr8r,r,r,r-s    ; C