U f{j %@sddlZddlmZddlmZmZddlmZm Z m Z m Z ddl m Z ddlmZe dd dZe dddZddZe dddZdS)N) peak_finder) pick_types pick_channels)loggerverbose_pl_validate_type filter_data)Epochs 10sFc  Cst||} td| |r dnd}|j| |dd\} } | |jd|j} t| ||||jd|j|||| d } t| | dddf|jt | dddf<| S) aLocate EOG artifacts. .. note:: To control true-positive and true-negative detection rates, you may adjust the ``thresh`` parameter. Parameters ---------- raw : instance of Raw The raw data. event_id : int The index to assign to found events. l_freq : float Low cut-off frequency to apply to the EOG channel in Hz. h_freq : float High cut-off frequency to apply to the EOG channel in Hz. filter_length : str | int | None Number of taps to use for filtering. %(ch_name_eog)s tstart : float Start detection after tstart seconds. reject_by_annotation : bool Whether to omit data that is annotated as bad. thresh : float | None Threshold to trigger the detection of an EOG event. This controls the thresholding of the underlying peak-finding algorithm. Larger values mean that fewer peaks (i.e., fewer EOG events) will be detected. If ``None``, use the default of ``(max(eog) - min(eog)) / 4``, with ``eog`` being the filtered EOG signal. %(verbose)s Returns ------- eog_events : array Events. See Also -------- create_eog_epochs compute_proj_eog z)EOG channel index for this subject is: %sZomitNT)picksreject_by_annotationZ return_timesZsfreq) event_idl_freqh_freq sampling_rate first_samp filter_lengthtstartthreshrr) _get_eog_channel_indexrinfoget_datar_find_eog_eventsnproundZastypeint)rawrrrrch_namerrrreog_indseogtimes eog_eventsr(T/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/eog.pyfind_eog_eventss.-   r*c  sFtdtdddtfdd|D} ttj| ddd } t| } t|| ||d d d d d dd } td| t | } t |} t t | t t | krt| | d |dd\}}nt| | d |dd\}}|| 7}t|}td|t||t|t |t|t gj}|S)zFind EOG events.zOFiltering the data to remove DC offset to help distinguish blinks from saccades-g@g?cs,g|]$}t|ddddddddd qS)rN? zero-doublehannfirwin2F)phase fir_window fir_designrr ).0xrZfmaxrr(r) `sz$_find_eog_events..rr)ZaxisNr-r.r/r0)r1r2r3z8Now detecting blinks and generating corresponding events)ZextremazNumber of EOG events detected: )rrrminimumarraysqrtsumZargmaxr Zmeanr!absmaxminrlenZzerosZonesT)r%rrrrrrrrrZfilteogtempZindexmaxZn_samples_startr'_Zn_eventsr(r6r)rTsP           rc st|dttfdd|dkr^tjddddddddd }|jdkrJtd fd d |D}nt|trp|g}n|}|dk r‡fd d |D}|rt d t |dd |t j |d}tdt |dd ||S)zGet EOG channel indices.Nr#)typesZ item_nameFTZbads)ZmegZeegZstimr%ZecgZemgZref_megexcluderzNo EOG channel(s) foundcsg|]}j|qSr(ch_names)r4iinstr(r)r7sz*_get_eog_channel_index..csg|]}|jkr|qSr(rF)r4r#rIr(r)r7s zThe specified EOG channelz cannot be found: z, )includezUsing EOG channelz: )r strlistrrsize RuntimeError isinstancecopy ValueErrorr joinrrGr)r#rJr$rG not_foundr(rIr)rs,    rr-TcCs<t|||||| | d}t|||||d|| || | | |d }|S)az Conveniently generate epochs around EOG artifact events. %(create_eog_epochs)s Parameters ---------- raw : instance of Raw The raw data. %(ch_name_eog)s event_id : int The index to assign to found events. %(picks_all)s tmin : float Start time before event. tmax : float End time after event. l_freq : float Low pass frequency to apply to the EOG channel while finding events. h_freq : float High pass frequency to apply to the EOG channel while finding events. reject : dict | None Rejection parameters based on peak-to-peak amplitude. Valid keys are 'grad' | 'mag' | 'eeg' | 'eog' | 'ecg'. If reject is None then no rejection is done. Example:: reject = dict(grad=4000e-13, # T / m (gradiometers) mag=4e-12, # T (magnetometers) eeg=40e-6, # V (EEG channels) eog=250e-6 # V (EOG channels) ) flat : dict | None Rejection parameters based on flatness of signal. Valid keys are 'grad' | 'mag' | 'eeg' | 'eog' | 'ecg', and values are floats that set the minimum acceptable peak-to-peak amplitude. If flat is None then no rejection is done. baseline : tuple or list of length 2, or None The time interval to apply rescaling / baseline correction. If None do not apply it. If baseline is (a, b) the interval is between "a (s)" and "b (s)". If a is None the beginning of the data is used and if b is None then b is set to the end of the interval. If baseline is equal to (None, None) all the time interval is used. If None, no correction is applied. preload : bool Preload epochs or not. %(reject_by_annotation_epochs)s .. versionadded:: 0.14.0 thresh : float Threshold to trigger EOG event. %(decim)s .. versionadded:: 0.21.0 %(verbose)s Returns ------- eog_epochs : instance of Epochs Data epoched around EOG events. See Also -------- find_eog_events compute_proj_eog Notes ----- Filtering is only applied to the EOG channel while finding events. The resulting ``eog_epochs`` will have no filtering applied (i.e., have the same filter properties as the input ``raw`` instance). )r#rrrrrF) eventsrtmintmaxZprojrejectflatrbaselinepreloadrdecim)r*r )r"r#rrrWrXrrrYrZr[r\rrr]rrVZ eog_epochsr(r(r)create_eog_epochss$M r^) rrrrNrFNN)rr+NN)NrNrUr-rrNNNTTNrN)numpyrZ _peak_finderrrrutilsrrr r filterr Zepochsr r*rrr^r(r(r(r)sF    C 0