IR-e1+ddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZdgZdZd ed ed ejfd Zdddddd dZdS)N)units)Time TimeDelta)BinnedTimeSeries) TimeSeries)AstropyUserWarningaggregate_downsamplec xt|dkrtjgSt|dr(tj|||Sg}t t|dz D]z}||dz||dzkr+|||||E|||||||dz{||||ddtj|S)z Manual reduceat functionality for cases where Numpy functions don't have a reduceat. It will check if the input function has a reduceat and call that if it does. rreduceatN)lennparrayhasattrr rangeappend)rindicesfunctionresultis =lib/python3.11/site-packages/astropy/timeseries/downsample.pyr r s/  7||qx|| : & &  x))%99:::s7||a'(( L LAq1u~a// hhuWQZ'899:::: hhuWQZ'!a%.-H'IJJKKKK hhuWR[]]344555xtimerel_basereturnc6||z ddS)Nseclong)formatsubfmt)to_value)rrs r_to_relative_longdoubler#$s! 8O % %U6 % B BBr) time_bin_sizetime_bin_start time_bin_endn_binsaggregate_funcc  t|tstd|0t|tjt fstd|+t|t t fst |}|+t|t t fst |}|jdd}| |jd}|j r|jd|z j }|S|Q|j r$|td||z tj z}n&tj |jd|d}|j r|M|j rE|C|tj }ttj||z }n5|3|j s,|} |d}| |d<|dd|d d<|P|j sI|j sBtj|d d|ddkrt'jd t*t-|||| } | tj}| j} | j} ||j st5| }|jd} t7| | }t7| | }t7|j| }||dk||dkz}tj|d d|ddkd}|D]E}tjtj|||k|||d zk}d ||<F||}tj||}t5|rqtj|d d|ddkrItj||||dk}tj t5|||<t5|rDtj!dtj"tj#|dd zg}ntj$g}tj%|}||}|j&D])}|d kr ||}t|tj'tjfst'jdt*St|tjrltjtj(tj)||j*}tjtW|j,|||j*d ||<nKtj-.||j/}d |_0tW|||||<d|j0|<|| |<+| S)aZ Downsample a time series by binning values into bins with a fixed size or custom sizes, using a single function to combine the values in the bin. Parameters ---------- time_series : :class:`~astropy.timeseries.TimeSeries` The time series to downsample. time_bin_size : `~astropy.units.Quantity` or `~astropy.time.TimeDelta` ['time'], optional The time interval for the binned time series - this is either a scalar value (in which case all time bins will be assumed to have the same duration) or as an array of values (in which case each time bin can have a different duration). If this argument is provided, ``time_bin_end`` should not be provided. time_bin_start : `~astropy.time.Time` or iterable, optional The start time for the binned time series - this can be either given directly as a `~astropy.time.Time` array or as any iterable that initializes the `~astropy.time.Time` class. This can also be a scalar value if ``time_bin_size`` or ``time_bin_end`` is provided. Defaults to the first time in the sampled time series. time_bin_end : `~astropy.time.Time` or iterable, optional The times of the end of each bin - this can be either given directly as a `~astropy.time.Time` array or as any iterable that initializes the `~astropy.time.Time` class. This can only be given if ``time_bin_start`` is provided or its default is used. If ``time_bin_end`` is scalar and ``time_bin_start`` is an array, time bins are assumed to be contiguous; the end of each bin is the start of the next one, and ``time_bin_end`` gives the end time for the last bin. If ``time_bin_end`` is an array and ``time_bin_start`` is scalar, bins will be contiguous. If both ``time_bin_end`` and ``time_bin_start`` are arrays, bins do not need to be contiguous. If this argument is provided, ``time_bin_size`` should not be provided. n_bins : int, optional The number of bins to use. Defaults to the number needed to fit all the original points. If both ``time_bin_start`` and ``time_bin_size`` are provided and are scalar values, this determines the total bins within that interval. If ``time_bin_start`` is an iterable, this parameter will be ignored. aggregate_func : callable, optional The function to use for combining points in the same bin. Defaults to np.nanmean. Returns ------- binned_time_series : :class:`~astropy.timeseries.BinnedTimeSeries` The downsampled time series. z"time_series should be a TimeSeriesNz3'time_bin_size' should be a Quantity or a TimeDeltarr z_With single 'time_bin_start' either 'n_bins', 'time_bin_size' or time_bin_end' must be providedT)copyr zaOverlapping bins should be avoided since they can lead to double-counting of data during binning.)r$r%r&r'Frz.Skipping column {0} since it has a mix-in type)unit)dtype)1 isinstancer TypeErroruQuantityrrilocrisscalarrsrmaximumr"intceil replicateanywarningswarnrrnanmeanr%r&rr#where logical_and searchsortedallarangehstacknonzerodiffruniquecolnamesndarrayrepeatnanr+r valuemazerosr,mask) time_seriesr$r%r&r'r( ts_sorted time_duration bin_size_secscalar_start_timebinned bin_startbin_endr rel_bin_start rel_bin_endrel_ts_sorted_timekeepnoncontiguous_bins_indicesinddelete_indicesrel_subset_timer indices_startgroupsunique_indicessubsetcolnamevaluesdatas rr r 0sn k: . .><===  I.** MNNN!*^dIEV*W*W!n-- <$ AR(S(SL))  #I"*B"+n<A !5  " N~H !. 6 < :inR&8.:LMML 7  $% H>#0#9#9!##>#>L )E!F!FGGF  %( 7$2!!-!7!7T!7!B!B$5q!%1#2#%6qrr"% "+ ~abb)L",==>>  MF"    #%! F%I!G~^4~Y~a H+Ix@@M)'8<N*N!O!OPQ!R)%% N"[%55"]37%;;     %^(.Ook?;;G 7||?}QRR0K4DDEE? ]=!>  7||Arz"''*:*:;;A>BCDD"Yw''Nt_F? f   &2:qz":;;  M@BT     faj ) ) *:bi77fkJJJD#$:v~>> RW$$$D 5;;vV\;::DDI#+FFN#K#KD ()DIn %w Mr)r9numpyrastropyrr/ astropy.timerrastropy.timeseries.binnedrastropy.timeseries.sampledrastropy.utils.exceptionsr__all__r longdoubler#r rrrms((((((((666666111111777777 ! "   ( C$ C$ C2= C C C C ]]]]]]]r