Aluode/PerceptionLabPortable
0
1o
2 '�Yi@
� @ sd d dl ZddlmZmZ ddlmZ ddlmZ ddl m3Z4m Z e dd5dd �d d
��Z dd� Z
dS )� N� )�
_picks_to_idx� pick_info)�6Projection)�verbose� )�_prep_mf_coils�7_sss_basis�meg�badsZaccurate)�accuracyr c sD t � |d|dd�}t� |�� ~tdd|d�}t� d|d�}t|d �}|� d kr6td8|� d � d � d ���t||�} ~� fd
d�t�t� | � j9dd��d D �}10|11r\td|12� d���| tj j | dd� } t
|�} t| �| jd ksuJ �g } t| | j�D ]"\}
}t� d d|tjdd�f � d dd�}| �td||
d�� q}| S )a� Generate projectors to perform homogeneous/harmonic correction to data.13 14 Remove environmental fields from magnetometer data by assuming it is15 explained as a homogeneous :footcite:`TierneyEtAl2021` or harmonic field16 :footcite:`TierneyEtAl2022`. Useful for arrays of OPMs.17 18 Parameters19 ----------20 %(info)s21 order : int22 The order of the spherical harmonic basis set to use. Set to 1 to use23 only the homogeneous field component (default), 2 to add gradients, 324 to add quadrature terms, etc.25 picks : str | array_like | slice | None26 Channels to include. Default of ``'meg'`` (same as None) will select27 all non-reference MEG channels. Use ``('meg', 'ref_meg')`` to include28 reference sensors as well.29 exclude : list | 'bads'30 List of channels to exclude from HFC, only used when picking31 based on types (e.g., exclude="bads" when picks="meg").32 Specify ``'bads'`` (the default) to exclude all channels marked as bad.33 accuracy : str34 Can be ``"point"``, ``"normal"`` or ``"accurate"`` (default), defines35 which level of coil definition accuracy is used to generate model.36 %(verbose)s37 38 Returns39 -------40 %(projs)s41 42 See Also43 --------44 mne.io.Raw.add_proj45 mne.io.Raw.apply_proj46 47 Notes48 -----49 To apply the projectors to a dataset, use50 ``inst.add_proj(projs).apply_proj()``.51 52 .. versionadded:: 1.453 54 References55 ----------56 .. footbibliography::57 r58 F)�none�excludeZ with_ref_meg)� r r r )�originZ int_orderZ ext_order)Z59ignore_refr � ZnchanzOnly �/z, picks could be interpreted as MEG channels.c s g | ]}� d | �qS )�ch_names� )�.0�pick��infor �E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\mne/preprocessing/hfc.py�60<listcomp>K s �z$compute_proj_hfc.<locals>.<listcomp>r )Zaxisz=The following channel(s) generate non-finite projectors:61 z62Please exclude from picks!r N)Z col_namesZ row_names�dataZncolZnrow)Zactiver �desc)r r �dictr �len�63ValueErrorr �np�where�isfinite�anyZlinalgZnorm� _label_basis�shape�zip�TZnewaxis�appendr )r �orderZpicksr r r �expZcoilsZn_chs�SZ bad_chans�labelsZprojs�labelZvecZ proj_datar r r �compute_proj_hfc
sF 264 �6566 ����r. c C s dd� t �d| d �D �S )z0Give basis vectors names for Projection() class.c S s6 g | ]}t �d | |d �D ]67}d|� d|� ��qqS )�����r zHFC: l=z m=�r Zarange)r �L�mr r r r e s ���z _label_basis.<locals>.<listcomp>r r0 )r) r r r r$ c s �r$ )r r68 r )�numpyr Z69_fiff.pickr r Z70_fiff.projr �utilsr Zmaxwellr r r. r$ r r r r �<module> s �� U