U f{j/L@s`ddlZddlmZmZmZGdddeZddZGdd d Z dd d Z Gd ddeZ dS)N) _ensure_intverboseloggerc@s*eZdZdZd ddZddZddZd S) _Interp2aInterpolate between two points. Parameters ---------- control_points : array, shape (n_changes,) The control points (indices) to use. values : callable | array, shape (n_changes, ...) Callable that takes the control point and returns a list of arrays that must be interpolated. interp : str Can be 'zero', 'linear', 'hann', or 'cos2' (same as hann). Notes ----- This will process data using overlapping windows of potentially different sizes to achieve a constant output value using different 2-point interpolation schemes. For example, for linear interpolation, and window sizes of 6 and 17, this would look like:: 1 _ _ |\ / '-. .-' | \ / '-. .-' | x |-.-| | / \ .-' '-. |/ \_.-' '-. 0 +----|----|----|----|--- 0 5 10 15 20 25 hanncs^tt|_tt|j|js0tdt|jdkrFtd|jdk sltd|jddft |tj r~|g}t |t t fr |D]b}|dkst |tj stdt|f|dk r|jdt|jkrtdt|j|jdfq|fdd }|}||_d|_d|_d|_d|_|_|_d }||krTtd ||f||_dS) Nz(Control points must be sorted and uniquerz"Must be at least one control pointz,All control points must be positive (got %s)z7All entries in "values" must be ndarray or None, got %szYValues, if provided, must be the same length as the number of control points (%s), got %scs(t|kddfddDS)Nrcs g|]}|dk r|ndqSN.0vidxr G/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/_ola.py Hsz2_Interp2.__init__..val..)npwhere)ptcontrol_pointsZ use_valuesrrvalFsz_Interp2.__init__..val)Zcos2linearzerorz"interp must be one of %s, got "%s")rarrayintZravelrZ array_equalunique ValueErrorlenall isinstancendarraylisttuple TypeErrortypeshapevaluesn_last _position _left_idx_left_right _use_interp_interp)selfrr'interpr rZ known_typesr rr__init__-sJ  z_Interp2.__init__c csd|_t|d}|j}|j|}td||j|ft|t}|jdkrtdd|j df| |j d|_t |j dkr|j|_ d}|j|j |j kr2t|j |j |j|}td|td|}||j|kst||rtd||<||jddgV|j|7_||7}|jd7_t|j |kd}t |dkrdt |j dg}|d}t|j |} |jtt | dd7_t| D]\} } | |j ks|j dkr0|j dk r| |j dkst|j |_|j d7_ d|_|j |j d} td|j d| f| | |_ |j dk s@t|j |j } |j |j d}|jdkr|| }|jd krd|_nR|jd krtjd d |d d|_n0ttjdtjd|d d|_|j|j9_t|||j}|dkrtd|j|| |f|j| }|dks t|jdkr2d}n"|j|||}|j|ksTtt|||}||j|ksvt||rtd||<||j|j |gV|j|7_||7}q|j |j |jkr^||j}|dkr^td|t||d}||rtd||<|j dk s*t||j ddgV|j|7_||7}|jd7_|j|ksnt||ks|t|st|j||kstdS)z/Feed data and get interpolators as a generator.rn_ptszFeed %s (%s-%s)Nz Eval @ %s (%s)rz Left ZOH %sTrr?F)Zendpointg@z Interp %s %s (%s-%s)z Right ZOH %s)r(rr)rdebugrzerosboolr+rr'rr,r*minslicesizeAssertionErroranyrarangemax enumerater-r.Zlinspacecospir)r/r2Zoriginal_positionstopusedZn_usedn_useZthis_slZstop_right_idxZ left_idxsZbiZleft_idxZeval_ptZ left_pointZ right_pointZ interp_spanZ interp_startZ this_interpr r rfeed_generatorUs                     z_Interp2.feed_generatorcsd}|D]}|dkr0fdd|dD}t|D]\}}|dk r8|ddkrx|d|dtjf|d|df<q8|d|dtjf|d|d|dtjfd |d|d|df<q8q|dk st|S) z&Feed data and get interpolated values.Ncs*g|]"}|dk r"t|jfndqSr )remptyr&r r2r rrsz!_Interp2.feed..rr.rr3)rEr?rZnewaxisr;)r/r2Z out_arraysoZaiZarrr rGrfeeds   $  z _Interp2.feedN)r)__name__ __module__ __qualname____doc__r1rErJr r r rrs (drcCsXt|tjr|g}t|ttfr:tdd|Dr:t|}t|sTtdt |f|S)Ncss|]}t|tjVqdSr )r rr!)r sr r r sz_check_store..z#store must be callable, got type %s) r rr!r"r#r_Storercallabler$r%)storer r r _check_stores rTc@sBeZdZdZed ddddZedd Zeddd d ZdS) _COLAadConstant overlap-add processing helper. Parameters ---------- process : callable A function that takes a chunk of input data with shape ``(n_channels, n_samples)`` and processes it. store : callable | ndarray A function that takes a completed chunk of output data. Can also be an ``ndarray``, in which case it is treated as the output data in which to store the results. n_total : int The total number of samples. n_samples : int The number of samples per window. n_overlap : int The overlap between windows. window : str The window to use. Default is "hann". tol : float The tolerance for COLA checking. Notes ----- This will process data using overlapping windows to achieve a constant output value. For example, for ``n_total=27``, ``n_samples=10``, ``n_overlap=5`` and ``window='triang'``:: 1 _____ _______ | \ /\ /\ / | \ / \ / \ / | x x x | / \ / \ / \ | / \/ \/ \ 0 +----|----|----|----|----|- 0 5 10 15 20 25 This produces four windows: the first three are the requested length (10 samples) and the last one is longer (12 samples). The first and last window are asymmetric. r绽|=N)rc  Csddlm} t|d}t|d}t|d}|dkr@td|f|dkrVtd|f|dkrltd|ft||_t||_~~||jkrtd |j|ft|std t |f||_ |j|j|_ t ||_ d|_d|_|_t|tr|nd } | ||j|jd d d|_|jt|j|j|j | |d_td||jd |j |_|j|j|_||jd} ||jd<t|}t|jd krdnd} tdt|j| |j||j|| f~~ | dkrtd| |fdS)Nr) get_window n_samples n_overlapn_totalzn_samples must be > 0, got %szn_overlap must be >= 0, got %szn_total must be >= 0, got %szRNumber of samples per window (%d) must be at most the total number of samples (%s)z%process must be callable, got type %sZcustomrrH)Zfftbins)tolrOz_ Processing %4d data chunk%s of (at least) %0.1f sec with %0.1f sec overlap and %s windowingz< The final %0.3f sec will be lumped into the final window)Z scipy.signalrWrrr _n_samplesZ _n_overlaprRr$r%_process_steprT_store_idx _in_buffers _out_buffersr str_window _check_colarr=startsstopsfloatrrinfo)r/processrSrZrXrYZsfreqZwindowr[rrW window_namedeltaplr r rr1sf             z_COLA.__init__cCs|j|j|jdjdS)z%dz=data (shape %s) exceeded expected total buffer size (%s > %s)rZconstantz * Processing %d->%dcs g|]}|ddfqS.N)copy)r Zin_)this_lenr rrtsz_COLA.feed..c3s.|]&}|jdko jknVqdS)r\N)r&r:)r proc)rs this_windowr rrPvsz_COLA.feed..zinternal indexing errorcs*g|]"}t|jddf|jqS)Nr\)rr6r&dtyper rI)max_lenr rr|s.z1 - Shifting input/output buffers by %d samplescsg|]}|ddfqSrqr rw)rnr rrsr4) rcrrr?r rr!ndimr$r%rFr&rvrr5rpZ concatenaterirbrhrfrrpadranger`r^r RuntimeErrorr_rdr>ra)r/rZdataskwargsZdidatastartrBoffsetrDZ this_procoutsoioutZ next_startobr )rnrxrsrurrJ?s   "      " $     &   z _COLA.feed)rrV) rKrLrMrNrr1propertyrprJr r r rrUs*1 rUrVcstjfddt|Ddd}|dkrX|d|| d7<t|}tt||}||krtd||d||f|S)z@Check whether the Constant OverLap Add (COLA) constraint is met.cs$g|]}||dqS)rr )r iistepwinr rrsz_check_cola..r)ZaxisNzesegment length %d with step %d for %s window type does not provide a constant output (%g%% deviation)d)rsumr{Zmedianr>absr)rZnpersegrrmr[ZbinsumsconstZ deviationr rrrgs   &  rgc@s&eZdZdZddddZddZdS)rQzStore data in chunks.N)pickscGsLt|D],\}}t|tjr&|jdkrtd|fq||_d|_||_dS)Nrz&outs[oi] must be >= 1D ndarray, got %sr) r?r rr!ryr$rrr)r/rrrrr r rr1sz_Storer.__init__cstt|jks(tfddDs0tdtf}|jdk rL||jf7}|jdjd}|t|j|f7}t |jD]\}}|||<q~||_dS)Nc3s&|]}|jddjdkVqdS)r\rN)r&)r rrr rrPsz#_Storer.__call__..zBad outsrr\) rrrrEllipsisrrr&r9zip)r/rrrBZo1o2r rr__call__s   z_Storer.__call__)rKrLrMrNr1rr r r rrQs rQ)rV) numpyrutilsrrrobjectrrTrUrgrQr r r rsD 5