U f{jpY@sddlZddlmZddlmZddlmZmZddlm Z ddl m Z m Z m Z ddlmZd d lmZd d lmZd d lmZd d lmZmZmZmZmZmZmZd dlmZddZ ddZ!d&ddZ"ed'ddZ#ej$dej%dej&diZ'ddZ(ed(d d!Z)d"d#Z*ed)d$d%Z+dS)*N)FIFF)_check_ch_keys)_has_eeg_average_ref_projmake_eeg_average_ref_proj) setup_proj) pick_types pick_channelspick_channels_forward)BaseRaw)Evoked) BaseEpochs)pinv)loggerwarnverbose_validate_type_check_preload _check_optionfill_docDEFAULTScCs2||g}|||i|j|gdd|S)aAdd a copy of a channel specified by ch_name. Input data can be in the form of Raw, Epochs or Evoked. The instance object is modified inplace. Parameters ---------- inst : instance of Raw | Epochs | Evoked Data containing the EEG channels ch_name : str Name of the channel to copy. new_ch_name : str Name given to the copy of the channel. Returns ------- inst : instance of Raw | Epochs | Evoked The data with a copy of a given channel. TZforce_update_info)copyr Zrename_channels add_channels)instch_nameZ new_ch_nameZnew_instrO/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/reference.py _copy_channelsr c sjtdt|}dd|Dddd}tjf|}|dkrZfdd|D}d }nd }t|d krxtd |fg}tjd D]f\}ddksdtjkrt d| |qdstfdd||Dd krt dq|D]}jd |=qt jddd\_ } tj|dd} tt| |d krfjtjjd<W5QRX|S)z!Prepare instance for referencing.zApplying a referencecSsi|] }|dqSTr.0type_rrr 8sz+_check_before_reference..FmegZref_megNcsg|]}j|qSrch_namesr#irrr =sz+_check_before_reference..zEEG channels foundzchannels suppliedrzNo %s to apply the reference toprojsdesczAverage EEG referencekindz3Removing existing average EEG reference projection.Zactivecs g|]}|ddkr|qS)dataZ col_namesr)r#ch)projrrr-RszInactive signal space projection (SSP) operators are present that operate on sensors involved in the desired referencing scheme. These projectors need to be applied using the apply_proj() method function before the desired reference can be set.)Z add_eeg_refactivateTorderedcustom_ref_applied)r _get_ch_typerinfolen ValueError enumeraterZFIFFV_PROJ_ITEM_EEG_AVREFrappend RuntimeErrorr _projectorr r)npZ intersect1d_unlockFIFFV_MNE_CUSTOM_REF_ON) rref_fromref_toch_typech_dictZeeg_idxextraZprojs_to_remover+_Zref_to_channelsr)rr3r_check_before_reference2sJ         rIautoc CsJt||||}t|dkr>t|dks,t|}t|j|dd}t|j|dd}|j}|d|ddfjddd}|d|ddf|8<|ddddf}|dk rBt||dd}|dd }|jdt|kst|t j|ddd } t | d d } || } t j| |d|ddfddd } |d|ddf| 7<nd}||fS) z&Apply a custom EEG referencing scheme.rTr5.N)keepdimsZsolr1)ZaxisrLgư>)Zrtol) rIr:AssertionErrorr r)_dataZmeanr shaper@r) rrCrDforwardrEZ ref_namesr1ref_dataGZGaZGa_invZRaZVparrr_apply_referencens*   "rSTc Cs6t|dt|tttfdt|tr,|g}|D]}||jdkr0td|q0t|jdddrjt d|rv| }t|t t fr|j }tt||jd f}t||f}||_ nt|tr.|j }|j\}}}t|t||f}t|j|||fd d |f}|j||t||d d }||_ ntd t|t|jd} |jd ddk rdd|jd D} t|d kst| t|krtdtj} tdnbt| dd| ddtdf} t|jddgdD](} | dd|jd| ddd<qntdtj} td|D]L}|tjtj | d | d d ddtj!tj"| d } |jd#| |j$q,t%d t|d }t|t rt&|j'd gt|f|_'t(|j)D](\}}t|t*||g|j)|<qn,t|tr|j+}t|t*||g|_+|j,t-||ddd|S) aAdd reference channels to data that consists of all zeros. Adds reference channels to data that were not included during recording. This is useful when you need to re-reference your data to different channels. These added channels will consist of all zeros. Parameters ---------- inst : instance of Raw | Epochs | Evoked Instance of Raw or Epochs with EEG channels and reference channel(s). %(ref_channels)s copy : bool Specifies whether the data will be copied (True) or modified in-place (False). Defaults to True. Returns ------- inst : instance of Raw | Epochs | Evoked Data with added EEG reference channels. add_reference_channels ref_channelsr)z%Channel %s already specified in inst.r.T)Z check_activez*Average reference already applied to data.rF)orderz4inst should be Raw, Epochs, or Evoked instead of %s.digNcSs*g|]"}|dtjkr|ddkr|qS)r0identr)rZFIFFV_POINT_EEG)r#dlrrrr-s z*add_reference_channels.. z9The locations of multiple reference channels are ignored.rrF)r'eegexcludechsloczULocation for this channel is unknown; consider calling set_montage() again if needed.g?g) r coil_typer0ZlognoZscannocalrangeZunit_mulunitZ coord_framera)rUrr).rrlisttuplestr isinstancer9r;rr>rr r rNr@zerosr:rOZvstackrZreshape TypeErrortypegetfullnanrZ concatenaterrrZFIFFV_COIL_EEGZ FIFFV_EEG_CHZ FIFF_UNIT_VZFIFFV_COORD_HEADr=Z_update_redundantZarangeZhstackZ_calsr<Z _read_picksmaxpicksZ_check_consistencyset_eeg_reference)rrUrr2r1refsxyznchanZ ref_dig_locZ ref_dig_arrayidxZ chan_infoZrange_pirrrrrrTs        (     rTonoffZCSDcCsBt|tttfd|jd}|tjtjfkr>tdt |fdS)NInstancer7z>Cannot set new reference on data with custom reference type %r) rr rr r9rrBFIFFV_MNE_CUSTOM_REF_OFFr> _ref_dict)rZcurrent_customrrr_check_can_reref s raverageFc sddlm}t|r|dkr.td|ftjdrFtdnjd}z:jtj jd<W5QRX t jdd Wn6t k rj|jd<W5QRXYnXj rtd d fS~|rnt|}d d |Dddd} fddtjf| D} |dkrBt||dnd }|dkrftd|d| }|gkr|tdn tdtdd|Ddt|| ||dS)a]Specify which reference to use for EEG data. Use this function to explicitly specify the desired reference for EEG. This can be either an existing electrode or a new virtual channel. This function will re-reference the data according to the desired reference. Note that it is also possible to re-reference the signal using a Laplacian (LAP) "reference-free" transformation using the :func:`.compute_current_source_density` function. Parameters ---------- inst : instance of Raw | Epochs | Evoked Instance of Raw or Epochs with EEG channels and reference channel(s). %(ref_channels_set_eeg_reference)s copy : bool Specifies whether the data will be copied (True) or modified in-place (False). Defaults to True. %(projection_set_eeg_reference)s %(ch_type_set_eeg_reference)s %(forward_set_eeg_reference)s %(verbose)s Returns ------- inst : instance of Raw | Epochs | Evoked Data with EEG channels re-referenced. If ``ref_channels='average'`` and ``projection=True`` a projection will be added instead of directly re-referencing the data. ref_data : array Array of reference data subtracted from EEG channels. This will be ``None`` if ``projection=True`` or ``ref_channels='REST'``. %(set_eeg_reference_see_also_notes)s r )ForwardrzMSetting projection=True is only supported for ref_channels="average", got %r.r.zTAn average reference projection was already added. The data has been left untouched.r7F)r4zlAverage reference projection was added, but has not been applied yet. Use the apply_proj method to apply it.NcSsi|] }|dqSr!rr"rrrr%_sz%set_eeg_reference..r&csg|]}j|qSrr(r*r,rrr-asz%set_eeg_reference..RESTz forward when ref_channels="REST")rrz Applying z reference.z8EEG data marked as already having the desired reference.zApplying a custom css|]}td|VqdS)titlesNrr"rrr psz$set_eeg_reference..)rE)rPrrr;rr9rrArr~Zadd_projr ExceptionZpreloadrrr8rrrhrS) rrUrZ projectionrErPrrr7rFZch_selrr,rrss\'               rscCst|tttfdd}t|tr.zJNumber of channel names must equal the number of anodes/cathodes (got %d).z{There is already a channel named "%s", please specify a different name for the bipolar channel using the ch_name parameter.cSsg|]}iqSrr)r#rHrrrr-szMNumber of channel info dictionaries must equal the number of anodes/cathodes.rJ)rCrDrEsfreq)rr)r)rZch_typesch_infoF)nameZ check_mincSsi|]\}}|dkr||qS)rrr#kvrrrr%sz)set_bipolar_reference..r`rccSsi|]\}}|dkr||qS))r`r)Zbadsrxrrrrrrr%s ) first_sampr)eventstminevent_idmetadatar)rcommentnaver0Trz, cSsg|]}|qSrr)r#rrrrr-0sz&Added the following bipolar channels: )rrr?)- meas_inforiorepochsrevokedrrrjrgr:r;zipr)rrIr@rkr<indexr9Zget_channel_typesritemsupdaterZFIFFV_COIL_EEG_BIPOLARrArNr rrrrrrrrrjoinrsetattrset drop_channels)rZanodeZcathoderrZ drop_refsrrrrrrr2rrZ multiplierryZref_infoZch_idxZanr9Zan_idxZan_chsZ pick_inforQZref_instZadded_channels attr_namerrrrset_bipolar_references=                    r)NNrJ)T)rTFrJNN)NNTTN),numpyr@ constantsrrrr3rrrpickrr r baser rr rrfixesrutilsrrrrrrrdefaultsrr rIrSrTrBr~ZFIFFV_MNE_CUSTOM_REF_CSDrrrsr8rrrrrsP       $ < * l  c