U f{jO? @sFddlZddlmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZmZddlmZmZmZmZmZmZdd lmZdd lmZdd lmZdd lmZdd l m!Z!m"Z"m#Z#ddl$m%Z%ed%ddZ&edddddZ'ed&ddZ(ed'ddZ)ddZ*ed(ddZ+ed)dd Z,edd!d"gdfd#d$Z-dS)*N)Epochs) check_fnameloggerverbose _check_option _check_fname)FIFF) fiff_open) pick_typespick_types_forward) Projection_has_eeg_average_ref_proj _read_projmake_projectormake_eeg_average_ref_proj _write_proj)start_and_end_file)make_fixed_length_events) parallel_func)_check_n_samples)is_fixed_orient_subject_from_forwardconvert_forward_solution) _make_stcc Cs8t|ddt|\}}}|}t||}W5QRX|S)aTRead projections from a FIF file. Parameters ---------- fname : str The name of file containing the projections vectors. It should end with -proj.fif or -proj.fif.gz. %(verbose)s Returns ------- projs : list The list of projection vectors. See Also -------- write_proj projectionz -proj.fifz -proj.fif.gzz _proj.fifz _proj.fif.gz)rr r)fnamerfftree_fidprojsr#G/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/proj.py read_projs  r%F) overwriterc Cs:t||d}t|ddt|}t||W5QRXdS)aWrite projections to a FIF file. Parameters ---------- fname : str The name of file containing the projections vectors. It should end with -proj.fif or -proj.fif.gz. projs : list The list of projection vectors. %(overwrite)s .. versionadded:: 1.0 %(verbose)s .. versionadded:: 1.0 See Also -------- read_proj )r&rrN)rrrr)rr"r&rr!r#r#r$ write_proj4s   r'separatec s\ddlm}t|dddd} t|dddd} t|ddddd } td |d d g|d kr||krftd d ddg} d}t|dddd} |dkrt| dkrtdd}n dddg} |dkrt| dkrtdd}|dkrt| dkrtdd}|dkrt| dkrtdd}|dfdd| | | fD\} }}g}t|||g| | | g| ||g| D]\}}}}|dkrqf||dd|f}|j |dd\}}}|ddd|f}|| }|d|}t t|j |D]n\}\}}t |d|tjddfd|jd}d|||df}td|td||tj|d}||qqf|S)NrlinalggradFbads)megref_megexcludemagT)r-eegr.r/r-r(combinedzAn_grad (%d) must be equal to n_mag (%d) when using meg="combined"r1zBNo MEG channels found for joint estimation. Forcing n_grad=n_mag=0ZplanarZaxialz*No gradiometers found. Forcing n_grad to 0z*No magnetometers found. Forcing n_mag to 0z)No EEG channels found. Forcing n_eeg to 0ch_namesc3s |]}fdd|DVqdS)csg|] }|qSr#r#).0kr4r#r$ tsz+_compute_proj...Nr#)r5indr7r#r$ tsz _compute_proj..) full_matricesr)Z col_names row_namesdataZnrowZncolz%s-%s-PCA-%02dzAdding projection: %s)Zactiver=desckindZ explained_var)scipyr*r r ValueErrorlenrinfozipsvdsum enumerateTdictnpnewaxissizer r ZFIFFV_PROJ_ITEM_FIELDappend) r=rCn_gradn_magn_eeg desc_prefixr-rr*Zgrad_indZmag_indZeeg_indkindsZ grad_namesZ mag_namesZ eeg_namesr"nr9namesr>Zdata_indUZSexp2r Zexp_varr6uvarZ proj_dataZ this_descprojr#r7r$ _compute_projQs|              rYc Cst||}|j} | dks,tt| dkr2d} n*t| dkrXtt| d} nd} |dkrvd| |j|jf}t ||j |||||dS)a'Compute SSP (signal-space projection) vectors on epoched data. %(compute_ssp)s Parameters ---------- epochs : instance of Epochs The epochs containing the artifact. n_grad : int Number of vectors for gradiometers. n_mag : int Number of vectors for magnetometers. n_eeg : int Number of vectors for EEG channels. %(n_jobs)s Number of jobs to use to compute covariance. desc_prefix : str | None The description prefix to use. If None, one will be created based on the event_id, tmin, and tmax. 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 ------- projs: list List of projection vectors. See Also -------- compute_proj_raw, compute_proj_evoked Nr0rzMultiple-eventsz%s-%-.3f-%-.3fr-) _compute_cov_epochsevent_idrBlistkeysstrvaluestmintmaxrYrC) epochsrNrOrPn_jobsrQr-rr=r^r#r#r$compute_proj_epochss) rgcspttj|\}}|fdd|D}t|}|dkr@td|jdt|j}}t|||t|}|S)zCompute epochs covariance.c3s|]}||jVqdS)N)rH)r5eZp_funr#r$r:sz&_compute_cov_epochs..rzNo good epochs foundZnchan) rrJdotrB RuntimeErrorrCtimesrrF)rerfparallelr=Zn_epochsZn_chanZ n_samplesr#rir$r]sr]c CsJt|j|jj}|dkr2d|jd|jdf}t||j|||||dS)aCompute SSP (signal-space projection) vectors on evoked data. %(compute_ssp)s Parameters ---------- evoked : instance of Evoked The Evoked obtained by averaging the artifact. n_grad : int Number of vectors for gradiometers. n_mag : int Number of vectors for magnetometers. n_eeg : int Number of vectors for EEG channels. desc_prefix : str | None The description prefix to use. If None, one will be created based on tmin and tmax. .. versionadded:: 0.17 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 ------- projs : list List of projection vectors. See Also -------- compute_proj_raw, compute_proj_epochs Nz %-.3f-%-.3frr\)rJrjr=rHrlrYrC)ZevokedrNrOrPrQr-rr=r#r#r$compute_proj_evokeds (roc  CsX|dk rt||jd|jd}t|d|||} t|jddddddd} t|| dd|d|jd| ||dd d }t|| }|j}|s|j|jd}nt| |d d }|r| |d n|j}t ||j}|dd||f\}}t |||j d t ||j}|j}||jd}||jd}d ||f}t||||||| d }|S)aUCompute SSP (signal-space projection) vectors on continuous data. %(compute_ssp)s Parameters ---------- raw : instance of Raw A raw object to use the data from. start : float Time (in sec) to start computing SSP. stop : float Time (in sec) to stop computing SSP. None will go to the end of the file. duration : float Duration (in sec) to chunk data into for SSP If duration is None, data will not be chunked. n_grad : int Number of vectors for gradiometers. n_mag : int Number of vectors for magnetometers. n_eeg : int Number of vectors for EEG channels. reject : dict | None Epoch rejection configuration (see Epochs). flat : dict | None Epoch flat configuration (see Epochs). %(n_jobs)s Number of jobs to use to compute covariance. 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 ------- projs: list List of projection vectors. See Also -------- compute_proj_epochs, compute_proj_evoked NZsfreqiTr,)r-r1ZeogZecgZemgr/?F)rcrdpicksrejectflatZbaselinerXrzRaw-%-.3f-%-.3fr\)rJroundrCrr rr]Zn_timesmaxZ time_as_indexminrshaperjrHrY)rawstartstopdurationrNrOrPrsrtrfr-reventsrrrer=rCrlrQr"r#r#r$compute_proj_rawsB2    r~r+fixedc Cs:ddlm}td|dddgtd|dd d d d d dgt|ddrLtd|dkrft|dd|d}nt||d|d}t|ddddd|drt|rtd|dd}|dkr|dkst|st |dg}ng}|dkr|n||}d d dg} |dk rPt ||dddd\} } } || kr4t | |}n| dkrftd||fn|| krftd||j \} }|d}t |}t|D]R}|ddd|d|d f}|d kr|j|ddd!}|dkr|d||<n||ddd"f}|d kr |||<n|d kr(||d||<n|d krHd#||d||<n|d kr~|t |ddd"f| }||||<n^|t | |ddd"f}|d kr||||<n(|dkrd#||||<n td$|q|d%kr|t |}t|}d&d'|d(D}t|ddt jf||d(jd)d#|d*S)+aCompute sensitivity map. Such maps are used to know how much sources are visible by a type of sensor, and how much projections shadow some sources. Parameters ---------- fwd : Forward The forward operator. projs : list List of projection vectors. ch_type : 'grad' | 'mag' | 'eeg' The type of sensors to use. mode : str The type of sensitivity map computed. See manual. Should be 'free', 'fixed', 'ratio', 'radiality', 'angle', 'remaining', or 'dampening' corresponding to the argument --map 1, 2, 3, 4, 5, 6 and 7 of the command mne_sensitivity_map. exclude : list of str | str List of channels to exclude. If empty do not exclude any (default). If 'bads', exclude channels in fwd['info']['bads']. %(verbose)s Returns ------- stc : SourceEstimate | VolSourceEstimate The sensitivity map as a SourceEstimate or VolSourceEstimate instance for visualization. rr)ch_typer1r+r0modefreerratioZ radialityZangle remainingZ dampeningT)origz1fwd should must be computed with free orientationF)r-r1r/)surf_oriZ force_fixedcopyrrz>Error converting solution, please notify mne-python developersZsolr=NrCr<)Zinclude_activez@No valid projectors found for channel type %s, cannot compute %sz%No projectors used, cannot compute %sr)r;Z compute_uvrZrqzUnknown mode type (got %s))rrcSsg|] }|dqS)Zvertnor#)r5sr#r#r$r8sz#sensitivity_map..srcrp)rcZtstepsubject)r@r*rrrAr rrkrrrrJrjrxemptyrangerEZnormrvrrrKr?)Zfwdr"rrr/rr*ZgainZeeg_aveZresidual_typesrXZncomprUZ n_sensorsZ n_dipolesZ n_locationssensitivity_mapr6ZggrgzcoprZverticesr#r#r$rXs                       r)N)r(N)rZrZrZNNr(N)rZrZrZNr(N) rNrrZrZrNNNr(N).numpyrJrerutilsrrrrrZ io.constantsr Zio.openr Zio.pickr r Zio.projr rrrrrZio.writereventrrmrZcovrforwardrrrZsource_estimaterr%r'rYrgr]ror~rr#r#r#r$sZ           @ 6 . Q