U d{j2+@sddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZdgZdd Zeeejd d d Zdddddd ddZdS)N)units)Time TimeDelta)BinnedTimeSeries) TimeSeries)AstropyUserWarningaggregate_downsamplec Cst|dkrtgSt|dr2t|||Sg}tt|dD]V}||d||dkrz|||||qF|||||||dqF||||ddt|SdS)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)rindicesfunctionresultir\/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/timeseries/downsample.pyr s   $r )timerel_basereturncCs||jdddS)Nseclong)formatZsubfmt)to_value)rrrrr_to_relative_longdouble$s r) time_bin_sizetime_bin_start time_bin_endn_binsaggregate_funcc Cs t|tstd|dk r2t|tjtfs2td|dk rPt|ttfsPt|}|dk rnt|ttfsnt|}|jdd}|dkr|jd}|j r|jd|j }|dkr|dkr|j r|dkrtdq||tj }nt |jd|d}|j rt|dk r6|j rt|dkrt|tj }tt ||}n>|dk rt|j st|} |jdd}| |d<|dd|d d<|dk r|j s|j st |d d|ddkrtd tt||||d } |dkrt j}| j} | j} |dks|j st| }|jd} t| | }t| | }t|j| }||dk||dk@}t |d d|ddkd}|D]4}t t |||k|||d k}d ||<qh||}t ||}t|rt |d d|ddkrt ||||dk}t t|||<t|r4t !dt "t #|dd g}n t $g}t %|}||}|j&D]}|d krhqV||}t|t j'tjfstdtqVt|tjrtjt (t j)||j*d}tjt+|j,|||j*d d||<n2t j-j.||j/d}d |_0t+|||||<d|j0|<|| |<qV| 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 TypeErroruZQuantityrrZilocrZisscalarrsr maximumrintceilZ replicateanywarningswarnrrZnanmeanr!r"r rwhere logical_andZ searchsortedallZarangeZhstackZnonzerodiffruniqueZcolnamesZndarrayrepeatnanr&r valuemaZzerosr'mask)Z time_seriesr r!r"r#r$Z ts_sortedZ time_durationZ bin_size_secZscalar_start_timeZbinnedZ bin_startZbin_endrZ rel_bin_startZ rel_bin_endZrel_ts_sorted_timeZkeepZnoncontiguous_bins_indicesindZdelete_indicesZrel_subset_timerZ indices_startgroupsZunique_indicesZsubsetZcolnamevaluesdatarrrr0s8                 "   * $        )r0numpyr Zastropyrr*Z astropy.timerrZastropy.timeseries.binnedrZastropy.timeseries.sampledrZastropy.utils.exceptionsr__all__r Z longdoublerrrrrrs