U f{jL7@sdZddlmZddlZddlmZmZmZddl m Z ddl m Z m Z ddlmZdd lmZdd lmZdd lmZe d(ddZe d)ddZddZddZddZddZe d*ddZe d+ddZd d!Zd"d#Zd,d$d%Z d&d'Z!dS)-z0Compute resolution matrix for linear estimators.)deepcopyN)pick_channels_forward EvokedArraySourceEstimate)FIFF)loggerverbose)convert_forward_solution) apply_inverse)_prepare_label_extraction)LabeldSPMqq?c s||}t||}|ddfdd|ddD}t||dd}|dd }t||||d } | |} td | j| S) aCompute resolution matrix for linear inverse operator. Parameters ---------- forward : instance of Forward Forward Operator. inverse_operator : instance of InverseOperator Inverse operator. method : 'MNE' | 'dSPM' | 'sLORETA' Inverse method to use (MNE, dSPM, sLORETA). lambda2 : float The regularisation parameter. %(verbose)s Returns ------- resmat: array, shape (n_orient_inv * n_dipoles, n_orient_fwd * n_dipoles) Resolution matrix (inverse operator times forward operator). The result of applying the inverse operator to the forward operator. If source orientations are not fixed, all source components will be computed (i.e. for n_orient_inv > 1 or n_orient_fwd > 1). The columns of the resolution matrix are the point-spread functions (PSFs) and the rows are the cross-talk functions (CTFs). infobadscsg|]}|kr|qSr).0cbads_invra/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/minimum_norm/resolution_matrix.py 6sz2make_inverse_resolution_matrix..ch_namesT)orderedZsoldata)methodlambda2z*Dimensions of resolution matrix: %d by %d.)_convert_forward_match_invr!_get_matrix_from_inverse_operatordotrrshape) forwardinverse_operatorrrr invfwdrZ leadfieldinvmatresmatrrrmake_inverse_resolution_matrixs    r(c Cst||||dkrd}t||} |dd} |dd} | | g} |dkrP|j}g} g}| D]h}|dd|f}|dk rt||}d}|dk rt||||\}}t|| dd d }| |||q\t| dkr|} t|dkr|}|dk r| |fS| SdS) aGet point-spread (PSFs) or cross-talk (CTFs) functions. Parameters ---------- resmat : array, shape (n_dipoles, n_dipoles) Forward Operator. src : Source Space Source space used to compute resolution matrix. %(idx_pctf)s func : str ('psf' | 'ctf') Whether to produce PSFs or CTFs. Defaults to psf. %(mode_pctf)s %(n_comp_pctf_n)s %(norm_pctf)s %(return_pca_vars_pctf)s %(verbose)s Returns ------- %(stcs_pctf)s %(pca_vars_pctf)s TmaxrvertnoctfN?)tminZtstep)_check_get_psf_ctf_params_vertices_for_get_psf_ctfT_normalise_psf_ctf_summarise_psf_ctfrappendlen)r'srcidxfuncmoden_compnormreturn_pca_varsr Z verts_allZ vertno_lhZ vertno_rhr*ZstcsZpca_varsvertsfuncsZpca_varstcrrr _get_psf_ctfBs<         rAcCs<|dkr |dkr d|}t||dkr8|r8d}t|dS)z9Check input parameters of _get_psf_ctf() for consistency.)Nsummeanr+zn_comp must be 1 for mode=%s.pcaz/SVD variances can only be returned if mode=pca.N) ValueError)r:r;r=msgrrrr0s  r0cCst|dtkrptd||dddd\}}g}|D]<}t|tkrXt|d|df}n t|}||q0nt|dtkr|}n|g}|S)z=Get vertices in source space for PSFs/CTFs in _get_psf_ctf().rNrCF)r@labelsr7r:Z allow_emptyZ use_sparser+)typer r listnpZ concatenatearrayr5)r8r7Z verts_labs_r>vZ this_vertsrrrr1s&   r1cCsN|dkr&t| |}||}n$|dkrJtjj|dd}||}|S)z&Normalise PSFs/CTFs in _get_psf_ctf().r)r<rZaxis)r)minrJlinalgr<)r?r<maxvalnormsrrrr3s  r3cCsZddlm}d}|dkrvttj|dd tj|dd}|dkr\t|}|| d}n|}|dd|f}n|dkrtjj|dd} |dkrt| }|| d}n| }|dd|f}n|dkrtj |dd}np|d krtj |dd}nX|d krR|j |d d \} } } | ddd|f}|rR| | } d | d|| }||fS)z$Summarise PSFs/CTFs across vertices.r)rPNrQrNr+ZmaxnormrBrCrDF)Z full_matricesd) ZscipyrPrJmaximumrOr)ZargsortZargmaxr<rBrCZsvd)r?r:r;r=rPZs_varZabsvalsZsortidxZmaxidxrRusrLs2rrrr4s6 "   r4r+Fc Cst|||d||||dS)aGet point-spread (PSFs) functions for vertices. Parameters ---------- resmat : array, shape (n_dipoles, n_dipoles) Forward Operator. src : instance of SourceSpaces Source space used to compute resolution matrix. %(idx_pctf)s %(mode_pctf)s %(n_comp_pctf_n)s %(norm_pctf)s %(return_pca_vars_pctf)s %(verbose)s Returns ------- %(stcs_pctf)s %(pca_vars_pctf)s Zpsfr9r:r;r<r=rAr'r7r8r:r;r<r=r rrrget_point_spreadsr[c Cst|||d||||dS)aGet cross-talk (CTFs) function for vertices. Parameters ---------- resmat : array, shape (n_dipoles, n_dipoles) Forward Operator. src : instance of SourceSpaces Source space used to compute resolution matrix. %(idx_pctf)s %(mode_pctf)s %(n_comp_pctf_n)s %(norm_pctf)s %(return_pca_vars_pctf)s %(verbose)s Returns ------- %(stcs_pctf)s %(pca_vars_pctf)s r,rXrYrZrrrget_cross_talk sr\cCs`t|}t|}|s\|s\|dddk }|rH|ds\t|dd}n|dr\t|dd}|S)zEnsure forward and inverse operators match. Inverse operator and forward operator must have same surface orientations, but can have different source orientation constraints. Z orient_priorrr.surf_oriT)r]F)_check_fixed_oriallr )r%r$ is_fixed_invZ is_fixed_fwdZ is_loose_invrrrr%s rc Cs8t|d}|d|d<|d|d<W5QRX|S)zGet a usable dict.rg@@ZsfreqZprojs)rZ_unlock)r#rrrr _prepare_info?s   racs$t||t|}|dt}|dd}fdd|D}t|}t||dd} t|} | rjd} nd } t| |||| d } | j} tj | || j d d } | j d kr | d }| d }| j }| |d|d |d} t|| d stt|| ds ttdt| j | S)aGet inverse matrix from an inverse operator. Currently works only for fixed/loose orientation constraints For loose orientation constraint, the CTFs are computed for the normal component (pick_ori='normal'). Parameters ---------- inverse_operator : instance of InverseOperator The inverse operator. forward : instance of Forward The forward operator. method : 'MNE' | 'dSPM' | 'sLORETA' Inverse methods (for apply_inverse). lambda2 : float The regularization parameter (for apply_inverse). Returns ------- invmat : array, shape (n_dipoles, n_channels) Inverse matrix associated with inverse operator and specified parameters. rrrcsg|]}|qSr)index)rchZ ch_names_invrrrpsz5_get_matrix_from_inverse_operator..r-)rr/NZvector)rrpick_orir+rN)rr+)rfrrr zDimension of Inverse Matrix: %s)rrar6rJeyerr^r rdeletendimcopyr!ZreshapeZ array_equalAssertionErrorrrstr)r#r"rrZinfo_invZ n_chs_invrZ ch_idx_badsZid_matZev_idr`reZ invmat_opr&Zv0o1Zv3o2r!rrdrrKs6      rcCs|dtjk}|S)z@Check if inverse or forward was computed for fixed orientations.Z source_ori)rZFIFFV_MNE_FREE_ORI)instZis_fixedrrrr^sr^)rrN)N)Nr+FFN)Nr+FFN)rr)"__doc__rkrnumpyrJrrrZ io.constantsrutilsrr Zforward.forwardr Z minimum_normr Zsource_estimater labelr r(rAr0r1r3r4r[r\rrarr^rrrrsJ       . K   *    V