U f{j@s.ddlZddlmZmZmZedddZdS)N)loggerverbose_plcCs6t|}|j}|jdks"|dkr*td|dkrZt|t|d}td|f|dksft |dkrv||}t |}t t j ||dk<t|dd|d ddkdd }t|dd |||ddf}tdg||d gf}~|j}t|} |dkrz| } d } | } tt |dd } | ddkrd}| d| d krt|dd |ddf}t|dd |ddf}|d 8}n6d}| d| d kr|d d}|d d}|d 8}tt|d }tj|tjd }t|}d}||d kr|d 7}| rT|||dksL| |d|krT| } d } ||d krfq||| kr||| |kr|}||} |d 7}| s| |||krd} ||} |||<| ||<|d 7}n||| kr ||} q |d| kr4|d| |kr4|d ||<|d||<|d 7}n(| s\| | kr\|||<| ||<|d 7}||d|}|d|}n2t|}||}|| |kr||}ng}g}|dkr|d9}t|tjst|d}t|tjst|d}t|dkrtdntdt|t|f||fS)aNoise-tolerant fast peak-finding algorithm. Parameters ---------- x0 : 1d array A real vector from the maxima will be found (required). thresh : float | None The amount above surrounding data for a peak to be identified. Larger values mean the algorithm is more selective in finding peaks. If ``None``, use the default of ``(max(x0) - min(x0)) / 4``. extrema : {-1, 1} 1 if maxima are desired, -1 if minima are desired (default = maxima, 1). %(verbose)s Returns ------- peak_loc : array The indices of the identified peaks in x0. peak_mag : array The magnitude of the identified peaks. Notes ----- If repeated values are found the first is identified as the peak. Conversion from initial Matlab code from: Nathanael C. Yoder (ncyoder@purdue.edu) Examples -------- >>> import numpy as np >>> from mne.preprocessing import peak_finder >>> t = np.arange(0, 3, 0.01) >>> x = np.sin(np.pi*t) - np.sin(0.5*np.pi*t) >>> peak_locs, peak_mags = peak_finder(x) # doctest: +SKIP >>> peak_locs # doctest: +SKIP array([36, 260]) # doctest: +SKIP >>> peak_mags # doctest: +SKIP array([0.36900026, 1.76007351]) # doctest: +SKIP rrz,The input data must be a non empty 1D vectorNz&Peak finder automatic threshold: %0.2g)rrrFg@)ZdtypeTgint64Zfloat64zNo significant peaks foundzFound %d significant peak%s)npZ asanyarraysizendim ValueErrormaxminrdebugAssertionErrordiffZfinfofloatepswhereZ concatenatesignintceilZzerosr Zargmax isinstanceZndarrayZ atleast_1dZastypeleninfor)Zx0ZthreshZextremarsZdx0indxlengthmin_magZtemp_magZ found_peakZleft_minZsignDxiiZmaxPeaksZpeak_locZpeak_magZc_indZtemp_locZ peak_indsZ peak_magsZx_indr#]/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/_peak_finder.py peak_finders+   *$                 r%)NrN)numpyr utilsrrrr%r#r#r#r$s