U f{jH @sTddlmZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZdd lmZmZdd lmZmZdd lmZmZmZmZd d lmZmZmZd dlmZm Z m!Z!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,ddl-m.Z.m/Z/ddZ0ddZ1ddZ2ddZ3d*ddZ4d+ddZ5e%d,d"d#Z6e%d-d&dd'd(d)Z7dS).)deepcopyN) _check_origin)make_ad_hoc_cov)FIFF) pick_types pick_info)_simplify_info)_has_eeg_average_ref_projmake_projector) get_head_surfget_meg_helmet_surf)transform_surface_to read_trans _find_trans _ensure_trans)_create_meg_coils_create_eeg_els_read_coil_defs) _do_self_dots_do_surface_dots_get_legen_table_do_cross_dots)loggerverbose _check_option _reg_pinv_pl path_like) EpochsArray BaseEpochs)Evoked EvokedArrayc Csxddlm}d}t|tdddd}|dkr0d nd \}}t|d |d d \} } |td d| jd| |dd} ||| | fS)zSet up dot products.r)interp1dgQ?g+=g-a=gư>)maggradeegfast)2Znearest)dZlinearFrrZaxis)Zscipy.interpolater$rdictrnpZlinspaceshape) modeinfocoilsch_typer$int_radnoiseZn_coeffinterpZlutn_factlut_funr:_/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/forward/_field_interpolation.py _setup_dotss  r<c CsZtd|d}|dt}t||d}t|jt|d|}|d}|ds\tt dt |d  }t|jt||}|d d d krt ||d \} |d <n.|d dkst |d t||d \} |d <t|jt| |} |j| } t|d| } |ddkrVt|rVtd| jdd| tj| dd8} | S)zDo the hairy computations.z# Preparing the mapping matrix...ch_namesprojsr self_dotsr6diag?data pinv_methodtsvdmissnesttikhonov surface_dotskindr'z0 The map has an average electrode reference (z channels)r-)rr2getlistr r/dotTNotImplementedErrorr@sqrtZravel _pinv_truncAssertionError_pinv_tikhonovr r0Zmean) fmdr2r=r>Zproj_opZ proj_dotsZ noise_covZwhitenerZ whitened_dotsinvZ inv_whitenedZinv_whitened_projZ mapping_matr:r:r;_compute_mapping_matrix)s.  rUc Csddlm}|j|dd\}}}t|}||d}t|d|kddd}td|t||d||dfd|d |}|d d d |f||d |j }||fS) zDCompute pseudoinverse, truncating at most "miss" fraction of varexp.r)linalgF)Z full_matricesr,rArz? Truncating at %d/%d components to omit less than %g (%0.2g)N) ZscipyrVZsvdr/Zcumsumwhererr2lenrM) xrErVusvZvarexpnrTr:r:r;rPUs   &rPc Cs>t||dd\}}}td|dt|d|d||fS)N)regZrankz Truncating at /u% components and regularizing with α=z0.1e)rrr2rX)rYr^rT_r]r:r:r;rRes"rRc Cstdd|dD}|tdd|dDO}tj|krZ|ttjgO}|ttjg8}t|}t|dkr|dtjtjfkst|dtjkrdnd}|dkrtd d }t |dd |d |}t |dd |d |} d } d}nDt |d}t |d} d} d}t |dr(t |ds(t dt ||}t||||\} } } }tdt|d|dt|dt| d |||| |dd}tdt|dt| d|dt| d t| d || ||| |j}dd|dD}t|||| ||| || d }t||}|S)aFind mapping from one set of channels to another. Parameters ---------- info_from : instance of Info The measurement data to interpolate from. info_to : instance of Info The measurement info to interpolate to. mode : str Either `'accurate'` or `'fast'`, determines the quality of the Legendre polynomial expansion used. `'fast'` should be sufficient for most applications. origin : array-like, shape (3,) | str Origin of the sphere in the head coordinate frame and in meters. Can be ``'auto'``, which means a head-digitization-based origin fit. Default is ``(0., 0., 0.04)``. Returns ------- mapping : array, shape (n_to, n_from) A mapping matrix. css|]}|dVqdSrINr:.0chr:r:r; sz+_map_meg_or_eeg_channels..chscss|]}|dVqdSrar:rbr:r:r;resrrr'megFr+normal dev_head_trD-C6?rGg?r>zIinfo_to must have an average EEG reference projector if info_from has onez Computing dot products for  z channelz...N)n_jobsz! Computing cross products for u → cSsg|] }|dqS)ch_namer:)rccr:r:r; sz,_map_meg_or_eeg_channels..) rIr=originr6r?rHr5rErC)setrZFIFFV_REF_MEG_CHZ FIFFV_MEG_CHsortedrXZ FIFFV_EEG_CHrQrrrr RuntimeErrorrr<rr2upperrrrrMr.rU) info_frominfo_tor1rprEZ info_kindsrIZ templatesZ coils_fromZcoils_torCr5r6r9r8r?Z cross_dotsr=rSmappingr:r:r;_map_meg_or_eeg_channelsms          * 4  rxr&r(cCstd|ddgt|jdddd}t|j|ddd}t|dkrHtdt|j|}t|j|}t||d |d }t|d r|}n|j }|j } | d kr|t j d d d d f}t j |jdt||jd f|jd} t|| D] \} } t || || d d <qt|j|dd} | dD]}|dd7<q| | t|tr| d ks\t| d} t| | |jd|j|jd}n,t|tstt| | |j|j|j|j d}|S)alCompute virtual evoked using interpolated fields in mag/grad channels. Parameters ---------- inst : instance of mne.Evoked or mne.Epochs The evoked or epochs object. ch_type : str The destination channel type. It can be 'mag' or 'grad'. mode : str Either `'accurate'` or `'fast'`, determines the quality of the Legendre polynomial expansion used. `'fast'` should be sufficient for most applications. Returns ------- inst : instance of mne.EvokedArray or mne.EpochsArray The transformed evoked object containing only virtual channels. r4r%r&TFrgr'Zref_megrzNo channels matching the destination channel type found in info. Please pass an evoked containingboth the original and destination channels. Only the locations of the destination channels will be used for interpolation.r{g{Gz?)rpr1get_datarN)dtype)selcopyrfrm_v)tmincommentnave)reventsevent_idmetadata)!rrr2rX ValueErrorrrxhasattrr|rBndimr/Znewaxisemptyr0r}ziprLZ_update_redundantZ_check_consistency isinstancer"rQr#timesrrr!r rrrr)instr4r1Z pick_fromZpick_torurvrwrBrZdata_dZd_r2rdZinst_r:r:r;_as_meg_type_insts^          rrgrzc stfdddDstddkr.tdtd|dd g}ttd |t||}td |d d g|d krt|dddd} tdnt|dddd} tdt | dkrt d|d} t t || }| |d<|d krt |dd|d} d} d} nt|d} d} d} t||| |\}}}}tdt | | ft|d| |||||}tt d}tdt |t|d| |||||| }t||d| |||||| d }td!t|||d"<||d#<|d$=|d%=|d&=|d'=|S)(aRe-map M/EEG data to a surface. Parameters ---------- %(info_not_none)s surf : dict The surface to map the data to. The required fields are `'rr'`, `'nn'`, and `'coord_frame'`. Must be in head coordinates. ch_type : str Must be either `'meg'` or `'eeg'`, determines the type of field. trans : None | dict If None, no transformation applied. Should be a Head<->MRI transformation. mode : str Either `'accurate'` or `'fast'`, determines the quality of the Legendre polynomial expansion used. `'fast'` should be sufficient for most applications. %(n_jobs)s origin : array-like, shape (3,) | str Origin of the sphere in the head coordinate frame and in meters. The default is ``'auto'``, which means a head-digitization-based origin fit. %(verbose)s Returns ------- mapping : array A n_vertices x n_sensors array that remaps the MEG or EEG data, as `new_data = np.dot(mapping, data)`. c3s|]}|kVqdS)Nr:)rckeysurfr:r;re2sz(_make_surface_mapping..)rrnnz!surf must have both "rr" and "nn"Z coord_framezEThe surface coordinate frame must be specified in surf["coord_frame"]r1Zaccurater(headr4rgr'TFryzPrepare MEG mapping...zPrepare EEG mapping...rzcannot map, no channels foundrirfrhr3rjZ electrodesgMbP?z#Computing dot products for %i %s...rz2Computing dot products for %i surface locations...r=) rIrr=r3rpr6r?rHr5rEzField mapping data readyrBrr?rHr5rE)allKeyErrorrrrrrrr2rXrsrr rrr<rr/Zarangerr.rU)r2rr4transr1rlrprZ orig_surfZpicksrir3Ztype_strrEr5r6r9r8r?r~rHrSr:rr;_make_surface_mappingsx!         rautohelmet)bemr) head_sourcerc  s<j} |dkr"fdddD} ntd|ddg|g} |dkrJt||}d| krn|dkrntd | dt| d krtd |dk rt|trt |}t |d d }td|dd gg} | D]8}|dkr|dkrt | |}nt || |d}| |qt}t| | D],\}}tj||||||d}| |q |S)a7Compute surface maps used for field display in 3D. Parameters ---------- evoked : Evoked | Epochs | Raw The measurement file. Need to have info attribute. trans : str | 'auto' | None The full path to the ``*-trans.fif`` file produced during coregistration. If present or found using 'auto' the maps will be in MRI coordinates. If None, map for EEG data will not be available. subject : str | None The subject name corresponding to FreeSurfer environment variable SUBJECT. If None, map for EEG data will not be available. subjects_dir : str The path to the freesurfer subjects reconstructions. It corresponds to Freesurfer environment variable SUBJECTS_DIR. ch_type : None | 'eeg' | 'meg' If None, a map for each available channel type will be returned. Else only the specified type will be used. mode : 'accurate' | 'fast' Either ``'accurate'`` or ``'fast'``, determines the quality of the Legendre polynomial expansion used. ``'fast'`` should be sufficient for most applications. meg_surf : 'helmet' | 'head' Should be ``'helmet'`` or ``'head'`` to specify in which surface to compute the MEG field map. The default value is ``'helmet'``. origin : array-like, shape (3,) | 'auto' Origin of the sphere in the head coordinate frame and in meters. Can be ``'auto'``, which means a head-digitization-based origin fit. Default is ``(0., 0., 0.04)``. .. versionadded:: 0.11 %(n_jobs)s %(head_source)s .. versionadded:: 1.1 %(verbose)s Returns ------- surf_maps : list The surface maps to be used for field plots. The list contains separate ones for MEG and EEG (if both MEG and EEG are present). Ncsg|]}|kr|qSr:r:)rctevokedr:r;rosz"make_field_map..)r'rgr4r'rgrz*No trans file available. EEG data ignored.rzNo data available for mapping.rZmrimeg_surfr)source subjects_dir)rlrpr1)r2rrrremoverXrsrrrrr r appendrKrr)rrsubjectrr4r1rrprlrrr2typesZsurfsZ this_typerZ surf_mapsZ this_surfZthis_mapr:rr;make_field_mapsJ2         r)N)r&r()rgNr(NrzN)rNNNr(rrzN)8rrnumpyr/rrZcovrZ io.constantsrZio.pickrrZ io.meas_infor Zio.projr r Zsurfacer r Z transformsrrrrZ _make_forwardrrrZ _lead_dotsrrrrutilsrrrrrrZepochsr r!rr"r#r<rUrPrRrxrrrr:r:r:r; sJ       , U O m