U f{j*,@sddlZddlmZddlmZddlmZmZm Z ddl m Z m Z m Z ddlmZmZmZmZeddd d d Zd dZddZddZddZeddZddZdS)N)jit)BaseRaw) Annotations_adjust_onset_meas_date_annotations_starts_stops) _picks_to_idx_picks_by_type_get_channel_types)_validate_typeverboselogger_mask_to_onsets_offsets{Gzt?)r csttdtj|dddt|dj}t|dj}|dkrV|dkrVtdt|}t|jj d jd }t t |jd }t }fd d tjddD}td dd\} } t fddt| | D} t tjd t} t tjd t} td|D]\}t jfddt| | Dd d}t t j|d d}|dk r|||k}t||}|jd d}|t |d 7<|jj d}t ||kd}||t d|k||k@d}| t ||ddfd }d| |<|dk r$|||k}t||}|jd d}|t |d 7<|jj d}t ||kd}||t d|k||k@d}| t ||ddfd }d| |<q$t| d}t| d}||}fdd|D}||fS)uAnnotate raw data based on peak-to-peak amplitude. Creates annotations ``BAD_peak`` or ``BAD_flat`` for spans of data where consecutive samples exceed the threshold in ``peak`` or fall below the threshold in ``flat`` for more than ``min_duration``. Channels where more than ``bad_percent`` of the total recording length should be annotated with either ``BAD_peak`` or ``BAD_flat`` are returned in ``bads`` instead. Note that the annotations and the bads are not automatically added to the :class:`~mne.io.Raw` object; use :meth:`~mne.io.Raw.set_annotations` and :class:`info['bads'] ` to do so. Parameters ---------- raw : instance of Raw The raw data. peak : float | dict | None Annotate segments based on **maximum** peak-to-peak signal amplitude (PTP). Valid **keys** can be any channel type present in the object. The **values** are floats that set the maximum acceptable PTP. If the PTP is larger than this threshold, the segment will be annotated. If float, the minimum acceptable PTP is applied to all channels. flat : float | dict | None Annotate segments based on **minimum** peak-to-peak signal amplitude (PTP). Valid **keys** can be any channel type present in the object. The **values** are floats that set the minimum acceptable PTP. If the PTP is smaller than this threshold, the segment will be annotated. If float, the minimum acceptable PTP is applied to all channels. bad_percent : float The percentage of the time a channel can be above or below thresholds. Below this percentage, :class:`~mne.Annotations` are created. Above this percentage, the channel involved is return in ``bads``. Note the returned ``bads`` are not automatically added to :class:`info['bads'] `. Defaults to ``5`` (5%%). min_duration : float The minimum duration (sec) required by consecutives samples to be above ``peak`` or below ``flat`` thresholds to be considered. to consider as above or below threshold. For some systems, adjacent time samples with exactly the same value are not totally uncommon. Defaults to ``0.005`` (5 ms). %(picks_good_data)s %(verbose)s Returns ------- annotations : instance of Annotations The annotated bad segments. bads : list The channels detected as bad. Notes ----- This function does not use a window to detect small peak-to-peak or large peak-to-peak amplitude changes as the ``reject`` and ``flat`` argument from :class:`~mne.Epochs` does. Instead, it looks at the difference between consecutive samples. - When used to detect segments below ``flat``, at least ``min_duration`` seconds of consecutive samples must respect ``abs(a[i+1] - a[i]) ≤ flat``. - When used to detect segments above ``peak``, at least ``min_duration`` seconds of consecutive samples must respect ``abs(a[i+1] - a[i]) ≥ peak``. Thus, this function does not detect every temporal event with large peak-to-peak amplitude, but only the ones where the peak-to-peak amplitude is supra-threshold between consecutive samples. For instance, segments experiencing a DC shift will not be picked up. Only the edges from the DC shift will be annotated (and those only if the edge transitions are longer than ``min_duration``). This function may perform faster if data is loaded in memory, as it loads data one channel type at a time (across all time points), which is typically not an efficient way to read raw data from disk. .. versionadded:: 1.0 rawZ data_or_icabads)excludepeakflatNz@At least one of the arguments 'peak' or 'flat' must not be None.sfreqcs8i|]0\}}tj|ddjdkr|tj|ddqS)T)Z assume_uniquer)npZ intersect1dsize).0ch_typeZ picks_of_type)picks_c/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/annotate_amplitude.py osz&annotate_amplitude..Z bad_acq_skipT)invertcs&g|]\}}tjj||qSr)rZarangetimesrrZonsetendrrr xsz&annotate_amplitude..z.Finding segments below or above PTP threshold.cs$g|]\}}||fdqS)rrr"rrrrr%s)Zaxisdrcs$g|]}|jdkrj|qS)r)infoZch_names)rbadr$rrr%s)r rrr( _check_ptp ValueError_check_bad_percent_check_min_durationr!rintrroundlistr rZ concatenatezipZzeroslenboolr itemsabsdiff_reject_short_segmentssumZnonzerowhereextend_create_annotations)rrr bad_percent min_durationpicksr min_duration_samplesronsetsZendsindexZany_flatZany_peakrdatar6Zflat_Z flat_countZ flat_meanZflat_ch_to_set_badZflat_ch_to_annotateidxZpeak_Z peak_countZ peak_meanZpeak_ch_to_set_badZpeak_ch_to_annotateZannotation_flatZannotation_peak annotationsrr&rannotate_amplitudeszQ                 rEc stdtdfdk rbttsbdkr@td|ddtt||}fdd|Dn@ttrD],\}}|dkrttd|d |d |dqtS) zDCheck the PTP threhsold argument, and converts it to dict if needed.numericNrz Argument 'z1' should define a positive threshold. Provided: 'z'.csi|] }|qSrr)rrptprrrsz_check_ptp..z@' should define positive thresholds. Provided for channel type 'z': ')r dict isinstancer+setr r4)rHnamer(r>Zch_typeskeyvaluerrGrr*s r*cCs>t|ddt|}d|kr(dks:ntd|d|S)zCCheck that bad_percent is a valid percentage and converts to float.rFr<rr'zQArgument 'bad_percent' should define a percentage between 0% and 100%. Provided: z%.r floatr+)r<rrrr,s  r,cCsNt|ddt|}|dkr,td|d||krJtd|d|d|S)zBCheck that min_duration is a valid duration and converts to float.rFr=rzQArgument 'min_duration' should define a positive duration in seconds. Provided: 'z ' seconds.zdArgument 'min_duration' should define a positive duration in seconds shorter than the raw duration (z seconds). Provided: 'rO)r=Z raw_durationrrrr-s  r-cCsL|jtkr|jdkstt|D]&\}}t|\}}t|||||q |S)zDCheck if flat or peak segments are longer than the minimum duration.r)Zdtyper3ndimAssertionError enumerater _mark_inner)Zarrr?kchr@offsetsrrrr7s  r7cCs0t||D] \}}|||kr d|||<q dS)z'Inner loop of _reject_short_segments().FN)r1)Zarr_kr@rWr?startstoprrrrTs rTcCs|dks tt|\}}t|t|}}||jd}|||jd}t||d|gt||jdd}t|||S)z5Create the peak of flat annotations from the any_arr.)rrrZBAD_Z meas_date)Z orig_time)rRrrarrayr(rr2r)Zany_arrkindrZstartsZstopsr@Z durationsZannotrrrr;s   r;)NNrrN)numpyrfixesriorrDrrrZio.pickrr r utilsr r r rrEr*r,r-r7rTr;rrrrs(  %