U f{j6@s(ddlZddlZddlmZddlmZmZddl m Z m Z m Z ddl mZddlmZdd lmZdd lmZd d Zd dZe ddddddddddddeddddddgdddddddd dddd!dfd"d#Ze ddd$ddddddddeddd%ejddgddd&dd'dd ddddd!dfd(d)ZdS)*N)Epochs)compute_proj_evokedcompute_proj_epochs)loggerverbosewarn) pick_types)make_eeg_average_ref_proj)find_ecg_events)find_eog_eventscCs||kr||=dS)ztAux function. Use this function when preparing rejection parameters instead of directly deleting keys. N)Zdict_keyrrT/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/ssp.py _safe_del_keysrc* Cs|r |n|}~||r$g}n"t|jd}tdt||r^t|j}|||dkrj|}|dksvt td||dkrt || |||||| d\} }!}!nt ||||| | |d} | j d d krt d |d g| f|rgfnd Stdt|j}"|"d|7<|dk rtt|"ddddddd krNt|dtt|"ddddddd krvt|dtt|"ddddddd krt|dtt|"ddddddd krt|d|dk rptt|"ddddddd krt|dtt|"ddddddd kr t|dtt|"ddddddd krHt|dtt|"ddddddd krpt|dt|"dd k}#t|"dddd|#dd}$|j|| |$| | ||ddddd t|| d||dd|$||dd }%|%j}&|%jj d d kr t dg| f|r|&fnd S| r.|%}'t|'||||d}(nt|%|||| |d }(|(D]})|d!|)d"|)d"<qF||(td#|| f|r|&fnd S)$z5Compute SSP/PCA projections for ECG or EOG artifacts.projsz)Including %d SSP projectors from raw fileN)ECGEOGzRunning %s SSP computationr)ch_nameevent_idl_freqh_freqtstart qrs_threshold filter_length)rrrrrrrr zNo z events foundrzComputing projectorbadsgradF)megeegeogref_megexcludemagTrr comps)rrr ecgr!r"g?z zero-doubleZfirwin2) picksrn_jobsmethod iir_paramsZl_trans_bandwidthZh_trans_bandwidthZphaseZ fir_design)ZbaselineZpreloadr&rejectflatprojzNo good epochs found)n_gradn_magn_eegr)r-r.r/r'r-desczDone.)copyZ load_datacpdeepcopyinforlenr appendAssertionErrorr r shaperr rfilterrdrop_logeventsaveragerrextend)*moderaw raw_eventtmintmaxr-r.r/rrr=rr'rr*r+ravg_refno_projrZ exg_l_freqZ exg_h_freqrr filter_methodr)return_drop_logr2rrrZeeg_projr<_Zmy_infor!r&epochsr;ZevokedZev_projsprrr_compute_exg_projs                           rKgɿg?g?gA@TZ10sg|=gAfc=g-C6 ?gMb0?)rr#rr Fi#gautoZfirZseparatecCsBtd|||||||||| | | | | |||||||||||||||S)a Compute SSP (signal-space projection) vectors for ECG artifacts. %(compute_proj_ecg)s .. note:: Raw data will be loaded if it hasn't been preloaded already. Parameters ---------- raw : mne.io.Raw Raw input file. raw_event : mne.io.Raw or None Raw file to use for event detection (if None, raw is used). tmin : float Time before event in seconds. tmax : float Time after event in seconds. n_grad : int Number of SSP vectors for gradiometers. n_mag : int Number of SSP vectors for magnetometers. n_eeg : int Number of SSP vectors for EEG. l_freq : float | None Filter low cut-off frequency for the data channels in Hz. h_freq : float | None Filter high cut-off frequency for the data channels in Hz. average : bool Compute SSP after averaging. Default is True. filter_length : str | int | None Number of taps to use for filtering. %(n_jobs)s ch_name : str | None Channel to use for ECG detection (Required if no ECG found). reject : dict | None Epoch rejection configuration (see Epochs). flat : dict | None Epoch flat configuration (see Epochs). bads : list List with (additional) bad channels. avg_ref : bool Add EEG average reference proj. no_proj : bool Exclude the SSP projectors currently in the fiff file. event_id : int ID to use for events. ecg_l_freq : float Low pass frequency applied to the ECG channel for event detection. ecg_h_freq : float High pass frequency applied to the ECG channel for event detection. tstart : float Start artifact detection after tstart seconds. qrs_threshold : float | str Between 0 and 1. qrs detection threshold. Can also be "auto" to automatically choose the threshold that generates a reasonable number of heartbeats (40-160 beats / min). filter_method : str Method for filtering ('iir' or 'fir'). iir_params : dict | None Dictionary of parameters to use for IIR filtering. See mne.filter.construct_iir_filter for details. If iir_params is None and method="iir", 4th order Butterworth will be used. copy : bool If False, filtering raw data is done in place. Defaults to True. return_drop_log : bool If True, return the drop log. .. versionadded:: 0.15 meg : str Can be 'separate' (default) or 'combined' to compute projectors for magnetometers and gradiometers separately or jointly. If 'combined', ``n_mag == n_grad`` is required and the number of projectors computed for MEG will be ``n_mag``. .. versionadded:: 0.18 %(verbose)s Returns ------- proj : list Computed SSP projectors. ecg_events : ndarray Detected ECG events. drop_log : list The drop log, if requested. See Also -------- find_ecg_events create_ecg_epochs Notes ----- Filtering is applied to the ECG channel while finding events using ``ecg_l_freq`` and ``ecg_h_freq``, and then to the ``raw`` instance using ``l_freq`` and ``h_freq`` before creation of the epochs used to create the projectors. rrK)r@rArBrCr-r.r/rrr=rr'rr*r+rrDrErZ ecg_l_freqZ ecg_h_freqrrrFr)r2rGrrrrrcompute_proj_ecgs@lrPg?gMb@?i cCsBtd|||||||||| | | || | |||||||d||||||S)a Compute SSP (signal-space projection) vectors for EOG artifacts. %(compute_proj_eog)s .. note:: Raw data must be preloaded. Parameters ---------- raw : mne.io.Raw Raw input file. raw_event : mne.io.Raw or None Raw file to use for event detection (if None, raw is used). tmin : float Time before event in seconds. tmax : float Time after event in seconds. n_grad : int Number of SSP vectors for gradiometers. n_mag : int Number of SSP vectors for magnetometers. n_eeg : int Number of SSP vectors for EEG. l_freq : float | None Filter low cut-off frequency for the data channels in Hz. h_freq : float | None Filter high cut-off frequency for the data channels in Hz. average : bool Compute SSP after averaging. Default is True. filter_length : str | int | None Number of taps to use for filtering. %(n_jobs)s reject : dict | None Epoch rejection configuration (see Epochs). flat : dict | None Epoch flat configuration (see Epochs). bads : list List with (additional) bad channels. avg_ref : bool Add EEG average reference proj. no_proj : bool Exclude the SSP projectors currently in the fiff file. event_id : int ID to use for events. eog_l_freq : float Low pass frequency applied to the E0G channel for event detection. eog_h_freq : float High pass frequency applied to the EOG channel for event detection. tstart : float Start artifact detection after tstart seconds. filter_method : str Method for filtering ('iir' or 'fir'). iir_params : dict | None Dictionary of parameters to use for IIR filtering. See mne.filter.construct_iir_filter for details. If iir_params is None and method="iir", 4th order Butterworth will be used. ch_name : str | None If not None, specify EOG channel name. copy : bool If False, filtering raw data is done in place. Defaults to True. return_drop_log : bool If True, return the drop log. .. versionadded:: 0.15 meg : str Can be 'separate' (default) or 'combined' to compute projectors for magnetometers and gradiometers separately or jointly. If 'combined', ``n_mag == n_grad`` is required and the number of projectors computed for MEG will be ``n_mag``. .. versionadded:: 0.18 %(verbose)s Returns ------- proj: list Computed SSP projectors. eog_events: ndarray Detected EOG events. drop_log : list The drop log, if requested. See Also -------- find_eog_events create_eog_epochs Notes ----- Filtering is applied to the EOG channel while finding events using ``eog_l_freq`` and ``eog_h_freq``, and then to the ``raw`` instance using ``l_freq`` and ``h_freq`` before creation of the epochs used to create the projectors. rrNrO)r@rArBrCr-r.r/rrr=rr'r*r+rrDrErZ eog_l_freqZ eog_h_freqrrFr)rr2rGrrrrrcompute_proj_eogs@grR)r2r3numpynprIrr,rrutilsrrrZio.pickr ior r%r r r rrKdictrPinfrRrrrrs      p s