U f{j:@sddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZdidd Zejd ejd ejd ejd ejdejdejdejdejdejdejdejdejdejdejdejdej dej!dej"dej#dej$dej%dej&dej'dej(dej)dej*dej+dej,diZ-dej.dej/d ifd!ej0d"ej1d#ej2d$ej3d%ej4d&ej5d'ifd!ej6d ej7d ej8d ej9d(ifd)Z:e d*d+Z;gdfd,d-Zd2d3Z?djd4d5Z@d6d7ZAe dkd;d<ZBe dld>d?ZCd@dAZDgd:fdBdCZEe ggdd=dfdDdEZFddd=dddggfdFdGZGe dmdHdIZHgd:dd=fdJdKZIdLdMZJe dNdOZKe dndPdQZLdodRdSZMeNd=d=d=ddddddddddd=dddd=d=d=dTZOePeQeOd gZRdUZSdVZTdWeTdXZUdYeTdZZVd[eTZWdpd\d]ZXdqd^d_ZYdrdadbZZdcddZ[dsdedfZ\dtdgdhZ]dS)u)deepcopyN)FIFF)loggerverbose_validate_typefill_doc _ensure_int _check_optionwarnFc CsFtttjtjdttjtjdttjdttjtjtjdttj tjtjdttj tjtjdttj tjtjdttj tjdttj tjdttjtjdttjtjdttjtjdttjtjdttjdttjdttjdttjdttjdttjdttjtjtjtjtjtjtjtjtj tj!g dttj"tjtj#dttj"tjtj$dttj"tj%tj&dttj"tj'dttj"tj(tj)dttj"tj(tj*dttjtj+tj,dd}|rBttj-d}ttj.d}tttj/dttj0dttj0tjdtf||tfd tji||||||d }|1D]\}}||2|q(|S) aReturn all known channel types, and associated FIFF constants. Parameters ---------- include_defaults : bool Whether to include default values for "unit" and "coil_type" for all entries (see Notes). Defaults are generally based on values normally present for a VectorView MEG system. Defaults to ``False``. Returns ------- channel_types : dict The keys are channel type strings, and the values are dictionaries of FIFF constants for "kind", and possibly "unit" and "coil_type". Notes ----- Values which might vary within a channel type across real data recordings are excluded unless ``include_defaults=True``. For example, "ref_meg" channels may have coil type ``FIFFV_COIL_MAGNES_OFFDIAG_REF_GRAD``, ``FIFFV_COIL_VV_MAG_T3``, etc (depending on the recording system), so no "coil_type" entry is given for "ref_meg" unless ``include_defaults`` is requested. )kindunit)r )r r coil_type)r r)gradmagref_megeegseegdbsecogeogemgecgrespbiomiscstimexcisystiasgofdipolechpifnirs_cw_amplitudefnirs_fd_ac_amplitudefnirs_fd_phasefnirs_odhbohbrcsd)r)r)rrr) rrrrrrrrrr')3dictr FIFFV_MEG_CH FIFF_UNIT_T_M FIFF_UNIT_TFIFFV_REF_MEG_CH FIFFV_EEG_CHZ FIFF_UNIT_VFIFFV_COIL_EEG FIFFV_SEEG_CH FIFFV_DBS_CH FIFFV_ECOG_CH FIFFV_EOG_CH FIFFV_EMG_CH FIFFV_ECG_CH FIFFV_RESP_CH FIFFV_BIO_CH FIFFV_MISC_CH FIFFV_STIM_CH FIFFV_EXCI_CH FIFFV_SYST_CH FIFFV_IAS_CHFIFFV_GOODNESS_FITFIFFV_DIPOLE_WAVE FIFFV_QUAT_0 FIFFV_QUAT_1 FIFFV_QUAT_2 FIFFV_QUAT_3 FIFFV_QUAT_4 FIFFV_QUAT_5 FIFFV_QUAT_6 FIFFV_HPI_G FIFFV_HPI_ERR FIFFV_HPI_MOVFIFFV_FNIRS_CHFIFFV_COIL_FNIRS_CW_AMPLITUDE FIFFV_COIL_FNIRS_FD_AC_AMPLITUDEZ FIFF_UNIT_RADFIFFV_COIL_FNIRS_FD_PHASEFIFFV_COIL_FNIRS_ODZ FIFF_UNIT_MOLFIFFV_COIL_FNIRS_HBOFIFFV_COIL_FNIRS_HBRZFIFF_UNIT_V_M2FIFFV_COIL_EEG_CSDFIFFV_COIL_NONEZFIFF_UNIT_NONEZFIFFV_COIL_VV_PLANAR_T1ZFIFFV_COIL_VV_MAG_T3itemsupdate)Zinclude_defaultsbaseZ coil_noneZ unit_nonedefaultskeyvaluerZJ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/pick.pyget_channel_type_constantss        7      r\megrrrrrrrrrr rrrrr#r"r!rfnirsrrrrr(r)r$r%r&r'r*)r]r^rcCsn|d|}zt|d}Wn,tk rHtd|d|dfYnX|tkrjt|\}}|||}|S)aGet channel type. Parameters ---------- %(info_not_none)s idx : int Index of channel. Returns ------- type : str Type of channel. Will be one of:: {'grad', 'mag', 'eeg', 'csd', 'stim', 'eog', 'emg', 'ecg', 'ref_meg', 'resp', 'exci', 'ias', 'syst', 'misc', 'seeg', 'dbs', 'bio', 'chpi', 'dipole', 'gof', 'ecog', 'hbo', 'hbr'} chsr z*Unknown channel type (%s) for channel "%s"ch_name) _first_ruleKeyError ValueError _second_rules)infoidxchZ first_kindrXZ second_rulerZrZr[ channel_types    rhcCs8tt|t|krtdt|t||st|tsDt|}t|tsVt|}g}t|D].\}}t|dks~||krb||krb||qbnt|t st |}t|dkrt |}t|t st |}t t }}|D]4}||kr||kr|| |q||qt|r,t d|ft |t S)aPick channels by names. Returns the indices of ``ch_names`` in ``include`` but not in ``exclude``. Parameters ---------- ch_names : list of str List of channels. include : list of str List of channels to include (if empty include all available). .. note:: This is to be treated as a set. The order of this list is not used or maintained in ``sel``. exclude : list of str List of channels to exclude (if empty do not exclude any channel). Defaults to []. ordered : bool If true (default False), treat ``include`` as an ordered list rather than a set, and any channels from ``include`` are missing in ``ch_names`` an error will be raised. .. versionadded:: 0.18 Returns ------- sel : array of int Indices of good channels. See Also -------- pick_channels_regexp, pick_types z0ch_names is not a unique list, picking is unsaferz6Missing channels from ch_names required by include: %s)lennpunique RuntimeError_check_excludes_includes isinstanceset enumerateappendlistindexrcarrayint)ch_namesincludeexcludeorderedselknamemissingrZrZr[ pick_channelss<"         r~cs t|fddt|DS)atPick channels using regular expression. Returns the indices of the good channels in ch_names. Parameters ---------- ch_names : list of str List of channels. regexp : str The regular expression. See python standard module for regular expressions. Returns ------- sel : array of int Indices of good channels. See Also -------- pick_channels Examples -------- >>> pick_channels_regexp(['MEG 2331', 'MEG 2332', 'MEG 2333'], 'MEG ...1') [0] >>> pick_channels_regexp(['MEG 2331', 'MEG 2332', 'MEG 2333'], 'MEG *') [0, 1, 2] csg|]\}}|r|qSrZ)match).0r{r|rrZr[ 0s z(pick_channels_regexp..)recompilerp)rvregexprZrr[pick_channels_regexps rcCsz|dkr dS|dtjkr\|dkr&dS|dkr@|ddr@dS|dkrv|ddrvdSn|d krv|dtjkrvdSd S) zTriage an MEG pick type.Trrplanar1r`2planar23rF)rr-endswithr.)rgr]rZrZr[_triage_meg_pick3srcCs|dkr dS|dtjkr&d|kr&dS|dtjkr@d|kr@dS|dtjkrZd|krZdS|dtjkrtd|krtdS|dtjkrd|krdS|dtjkrd|krdSd S) zTriage an fNIRS pick type.Trr(r)r$r%r&r'F)rrPrQrLrMrNrO)rgr^warnedrZrZr[_triage_fnirs_pickCs*rcCsDt|tr@ddddg}||r"dgng7}||kr@td||fdS)zEnsure a valid meg type.rrrrautoz+meg value must be one of %s or bool, not %sN)rnstrrc)r] allow_autoZ allowed_typesrZrZr[_check_meg_typeYs  rcCsTt|d||dkr$tdn,|dkr:|dg}nt|ttfsPtd|S)Nrez+exclude must be a list of strings or "bads"badszexclude must either be "bads" or a list of strings. If only one channel is to be excluded, use [ch_name] instead of passing ch_name.)r_check_consistencyrcgetrnrrtuple)rerxrZrZr[_check_info_excludecs  rrrZrc$st|ttfdt|}d}tj|tdt|ddt|t|tr|dkrdkoddk otdd ko|d k }||||||| | | | | |||||||fD]$}t|tsd }t | t |qt ||||||| | | | | |||||||d }t|trd D]}|||<q t|tr.)rboolrrrjZzerosrrnrircformattyper+_FNIRS_CH_TYPES_SPLITrangerhrbAssertionErrorrrrr,r/r0wherertrur~)$rer]rrrrrrrrr#rr rrr"r!rrr^r*rrwrxZ selectionrparamwZ param_dictrXrr{ch_typeZsel_kindZ myincluderzrZrr[ pick_typesqsL             "  rTc sddlm}|r n|dkr0St|dkrDtdttttd|}|t|krtd|t|ft dgdkrfd d |D}||\}}t|dkrt d tdf gd<W5QRX fd d |Dd <W5QRX fdd dDd<dkrtd} | D]~fdd tddD} fdd | D} d| } | d<t| dd<| dd<dd| dd<q^ | d<W5QRXS)aRestrict an info structure to a selection of channels. Parameters ---------- %(info_not_none)s sel : list of int | None Indices of channels to include. If None, all channels are included. copy : bool If copy is False, info is modified inplace. %(verbose)s Returns ------- res : dict Info structure restricted to a selection of channels. r)_bad_chans_compNrz No channels match the selection.rvz-Found %d / %d unique names, sel is not uniquercsg|]}d|qSrrZ)rrfrerZr[r"szpick_info..zURemoving %d compensators from info because not all compensation channels were picked.csg|]}d|qS)r_rZrrrZr[r+sr_csg|]}|dkr|qSrrZrrgrrZr[r-s rcs g|]\}}|dkr|qSrrZ)rr{nrrZr[r1s data row_namescsg|]}dd|qS)rrrZ)ri)crZr[r3srowcalsnrow) meas_inforrcopyrircrjrkarangerrre_unlock_update_redundantrrp) rerzrrrZn_uniquerv_Z comps_missingrZrow_idxrrrZ)rrer[ pick_infosN               rcCs*|D] }|d|dtjkrdSqdS)zDetermine if KIT ref channels are chosen. This is currently only used by make_forward_solution, which cannot run when KIT reference channels are included. r_rTF)rZFIFFV_COIL_KIT_REF_MAG)repicksprZrZr[ _has_kit_refs@srcCst|dkrt|dkr|St||jdd}t|jd||d}t|dkrTtdt|}t|j||_|j|ddf|_|S)aQPick channels from evoked data. Parameters ---------- orig : Evoked object One evoked dataset. include : list of str, (optional) List of channels to include (if empty, include all available). exclude : list of str | str List of channels to exclude. If empty do not exclude any (default). If 'bads', exclude channels in orig.info['bads']. Defaults to 'bads'. Returns ------- res : instance of Evoked Evoked data restricted to selected channels. If include and exclude are empty it returns orig without copy. rTre allow_badsrv)rwrxz*Warning : No channels match the selection.N)rirmrer~rcrrr)origrwrxrzresrZrZr[pick_channels_evokedLs  rc s|dt|dkr4t|dkr4|r0|S|St||ddd}t|dd|||d}t|dd|||d}|rt|n|t|t|krtd t|}|dkrtd td |d fdd |ddfdd <d|ddfd<|dd<fdd|D|d <dd<d fdd|Ddd<W5QRXd fddddDdd<ddk rdd |ddfdd <d|ddfd<|dd<fdd|Ddd<S)a7Pick channels from forward operator. Parameters ---------- orig : dict A forward solution. include : list of str List of channels to include (if empty, include all available). Defaults to []. exclude : list of str | 'bads' Channels to exclude (if empty, do not exclude any). Defaults to []. If 'bads', then exclude bad channels in orig. ordered : bool If true (default False), treat ``include`` as an ordered list rather than a set. .. versionadded:: 0.18 copy : bool If True (default), make a copy. .. versionadded:: 0.19 %(verbose)s Returns ------- res : dict Forward solution restricted to selected channels. If include and exclude are empty it returns orig without copy. rerTrsolrrwrxryrvzZForward solution and functional data appear to have different channel names, please check.zNothing remains after pickingz. %d out of %d channels remain after pickingrrNZ _orig_solrcsg|]}dd|qS)rrrZrfwdrZr[rsz)pick_channels_forward..csg|]}dd|qS)rer_rZrrrZr[rsr_csg|]}|kr|qSrZrZ)rbrrZr[rsrsol_gradZ_orig_sol_gradcsg|]}dd|qS)rrrZrrrZr[rs) rrirrmr~rrcrrerr) rrwrxryrrZsel_solZsel_infoZnuserZ)rvrr[pick_channels_forwardwsT     $ "   rc sR|dt||||||||d } t| dkr6tdfdd| D} t|| S)a%Pick by channel type and names from a forward operator. Parameters ---------- orig : dict A forward solution. meg : bool | str If True include MEG channels. If string it can be 'mag', 'grad', 'planar1' or 'planar2' to select only magnetometers, all gradiometers, or a specific type of gradiometer. eeg : bool If True include EEG channels. ref_meg : bool If True include CTF / 4D reference channels. seeg : bool If True include stereotactic EEG channels. ecog : bool If True include electrocorticography channels. dbs : bool If True include deep brain stimulation channels. include : list of str List of additional channels to include. If empty do not include any. exclude : list of str | str List of channels to exclude. If empty do not exclude any (default). If 'bads', exclude channels in orig['info']['bads']. Returns ------- res : dict Forward solution restricted to selected channel types. re)rrrrrwrxrzNo valid channels foundcsg|]}d|qSrrZrrrZr[rsz&pick_types_forward..)rrircr) rr]rrrrrrwrxrzZinclude_ch_namesrZrr[pick_types_forwards!  rc CsddtD}|jttttttttdt||dddd}|D]2}t||}|D]}||krd|||qdqN|S)aGet indices of channels by type. Parameters ---------- %(info_not_none)s %(picks_all)s Returns ------- idx_by_type : dict A dictionary that maps each channel type to a (possibly empty) list of channel indices. cSsi|]}|dkr|tqS))r]r^rrrrXrZrZr[ sz+channel_indices_by_type..)rrr(r)r$r%r&r'allrZT)nonerx allow_empty)_PICK_TYPES_KEYSrUrr _picks_to_idxrhkeysrq)rerZ idx_by_typer{rrXrZrZr[channel_indices_by_types$  rc s|r@dkr&ddkr&d=dkr@ddkr@d=|dkrPdn|}td|||d}dsd|dd|fn d|}fd d |D}fd d dD}|d<|d<|d<t|d <S) aKPick channels from covariance matrix. Parameters ---------- orig : Covariance A covariance. include : list of str, (optional) List of channels to include (if empty, include all available). exclude : list of str, (optional) | 'bads' Channels to exclude (if empty, do not exclude any). Defaults to 'bads'. ordered : bool If True (default False), ensure that the order of the channels in the modified instance matches the order of ``include``. .. versionadded:: 0.20.0 copy : bool If True (the default), return a copy of the covariance matrix with the modified channels. If False, channels are modified in-place. .. versionadded:: 0.20.0 Returns ------- res : dict Covariance solution restricted to selected channels. methodNZloglikrnamesrZdiagrcsg|]}d|qSrrZrrrZr[r>sz%pick_channels_cov..csg|]}|dkr|qSrrZ)rr|rrZr[r?s dim)rr~ri) rrwrxryrrzrrrrZrr[pick_channels_covs$ , rcCstdd|dgDS)z7Determine of mag and grad should be dealt with jointly.css*|]"}|dididdVqdS)Zmax_infoZsss_infoZin_orderrN)r)rphrZrZr[ Msz&_mag_grad_dependent..Z proc_history)anyrrrZrZr[_mag_grad_dependentIs rcsxtddtt}tt}tddtD||}td|dkrXtdftfddt d DS) aCheck whether a certain channel type is in an info object. Parameters ---------- %(info_not_none)s ch_type : str the channel type to be checked for Returns ------- has_ch_type : bool Whether the channel type is present or not. rrcSsg|]}|dkr|qSr]rZrrZrZr[rdsz%_contains_ch_type..Nz;Cannot check for channels of type "%s" because info is Nonec3s|]}t|kVqdS)Nrh)riirrerZr[rjsz$_contains_ch_type..r) rrr_MEG_CH_TYPES_SPLITrsortedrr rcrr)rerZ meg_extrasZ fnirs_extrasZvalid_channel_typesrZrr[_contains_ch_typeQs    rc st|tdt||}|dkr&t|}gddtDt|dD]}|d|d|krDt||}z||WqDtk r|dkr|d|}t ||r|dt j krĈd |q|dt j krd |nYqDXqDfd d tDtd ddkst |rptddrptddrpddttddddgfdd DS)aJGet data channel indices as separate list of tuples. Parameters ---------- %(info_not_none)s meg_combined : bool | 'auto' Whether to return combined picks for grad and mag. Can be 'auto' to choose based on Maxwell filtering status. ref_meg : bool If True include CTF / 4D reference channels exclude : list of str | str List of channels to exclude. If 'bads' (default), exclude channels in info['bads']. Returns ------- picks_list : list of tuples The list of tuples of picks and the type string. rrcSsi|] }|tqSrZrrrrZrZr[rsz"_picks_by_type..rr_r`rrrcs g|]}|t|tfqSrZ)rjrtrurZ picks_listrZr[rsz"_picks_by_type..Nr)rrrrr]cSsg|]}t|dr|qS)rrirrrZrZr[rs )rrrr_DATA_CH_TYPES_SPLITrrhrqrbrrr.r-rriinsertrjrkZ concatenatepop)reZ meg_combinedrrxr{Z this_typergrZrr[_picks_by_typensB      * rcCsrddlm}t|ttttjfsn|dkrXt||s.rvT)rxrrxr)rrr)rerxrrZrZr[_pick_data_or_icasrrc Csddlm}d}t||r$|d} nt|dd}|} | dks@t|} d} |d kr~t|trpt| }d | f} n|}d |f} t|trt| |}t |}t |dkrtj gtd n|}|j dkrt d |j f|jjdkrt|||||| || }|r|d}|d}|jjdkr0td|jf~ |t}t |dkr^|s^t d| f|| krt d| | f|| krt d| | f|| ;}|r||fS|S)z Convert and check pick validity.rrFrrezan int or InforNz, treated as range(%d)zNone, treated as "%s"rzpicks must be 1D, got %sD)SU)ruzApicks must be a list of int or list of str, got a data type of %sz6No appropriate channels found for the given picks (%r)zAll picks must be >= %d, got %rz+All picks must be < n_channels (%d), got %r)rrrnr rrurjrsliceZ atleast_1drirtndimrcrchar_picks_str_to_idxr TypeErrorZastyper) rerrrxrr return_kindrpicked_ch_type_or_genericZn_chan orig_picks extra_reprrZrZr[rsj              rc Cs t|trtdt}t|dkr|ddkr|ddkrh|dkrL|dn|} t|d|d| d}n4|dd krt|||d }n|dd krt||d}t|dkr|d kr|std t||fg} t} |D]>} z| |d | Wqtk r| | YqXqd } t}t dd}t t }}|D]P} | t krVd|| <n6| tkrl|| hO}n | tkr|| hO}n | } qTq>t }t|dkr|dds|D]}|t t||d|dO}qt|dkr|ddst|dkrt|d|d<n t||d<t|fd|i|}t|dkrTtt |t |B}|| |f}dd|D}t|dkr|stdt||t| | ftgt}n4t|dkrtdnt|t|dd}t|  }t| dkr|std| d|r||fS|S)z'Turn a list of str into ndarray of int.z@picks as str can only be used when measurement info is availablerr)rr data_or_icarrrvrxrrrNzApicks (%s) yielded no channels, consider passing picks explicitlyFrTr])r]rrxr^rxcSsg|]}t|dkqS)rrrrZrZr[rsz%_picks_str_to_idx..zpicks (%s) could not be interpreted as channel names (no channel "%s"), channel types (no type "%s"), or a generic type (just "all" or "data")zgSome channel names are ambiguously equivalent to channel types, cannot use string-based picks for thesez Channel(s) zH could not be picked, because they are not present in the info instance.)rnrurcrrrir~rrreprrqrsr+rorrrrrrsumrrjrtrlrr )rerrxrrrrrZ picks_genericZ use_excludeZ bad_namesZ picks_namerZbad_typeZ picks_typerr]r^Z extra_picksZuse_megZ all_picksZ any_foundrrZrZr[r5s                    rcsr|dkrtj|gd}fdd|D}||dkr\fddjdD}|dk rn|S)z5Return an instance with picked and excluded channels.Trcsg|]}jd|qSrrrrinstrZr[rsz_pick_inst..rcsg|]}|jdkr|qSrrrr rZr[rsN)rrrer~Z drop_channels)r rrxrZ pick_namesrZr r[ _pick_insts  r cs`|rdnd}t||ddd}fdd|D}|rBdd|D}|r\td d |D}|S) z/Get the data channel types in an info instance.rrrZF)rcsg|]}t|qSrZrr rrZr[rsz&_get_channel_types..cSsg|]}|tkr|qSrZ)rrrZrZr[rscSsi|] }||qSrZrZrrZrZr[rsz&_get_channel_types..)rrrr)rerrkZ only_data_chsrZch_typesrZrr[_get_channel_typess r)F)F)FFFFFFrFFFFFFFFFFFFFFrZrN)rZTN)N)FFr)NF)rTF)rZ)rrFTF)T)NFF)^rrrnumpyrj constantsrutilsrrrr r r r r\r,r/r0r;r5r6r7r8r:r<r>r=r2r3r9rArBrCrDrErFrGrHrIrJr@r?r4rKrar-r.rPrQrLrMrNrOr1ZFIFFV_COIL_EEG_BIPOLARrSrRrdrhr~rrrrrrrrrrrrrrrrrmr+rrrrrrrZ_DATA_CH_TYPES_ORDER_DEFAULTZ_VALID_CHANNEL_TYPESrrrrrr rrZrZrZr[s  $ f!  !C"   @ + T +  4  7   Gj