Aluode/PerceptionLabPortable
0
1o
2 '�Yi�! � @ s� d Z ddlmZ ddlZddlmZ ddlmZ ddl m3Z4m Z ddl m
Z
mZ dd lmZmZ dd5lmZmZ dd lmZmZ ddd�Zddd�Zdd� ZdS )z Tools for data interpolation.� )�chainN)�connected_components� )� create_info)�6BaseEpochs� EpochsArray)�Evoked� EvokedArray)�BaseRaw�RawArray)� _cart_to_sph� _sph_to_cart)� _ensure_int�_validate_type� �?Tc s d� kr7dksn t d�� ���ttt�dd� | D ����}t| d t�r/dd� | D ��ndd� | D ��g �|D ]� ��t� �fdd �t | �D ��t8��� � q:��fd9d�t |�D �}|rhd d� | D �} | D ]}t t|j d �t|� �r|j
|d
� ||j d < qj| S )a] Interpolate or mark bads consistently for a list of instances.10 11 Once called on a list of instances, the instances can be concatenated12 as they will have the same list of bad channels.13 14 Parameters15 ----------16 insts : list17 The list of instances (Evoked, Epochs or Raw) to consider18 for interpolation. Each instance should have marked channels.19 interp_thresh : float20 A float between 0 and 1 (default) that specifies the fraction of time21 a channel should be good to be eventually interpolated for certain22 instances. For example if 0.5, a channel which is good at least half23 of the time will be interpolated in the instances where it is marked24 as bad. If 1 then channels will never be interpolated and if 0 all bad25 channels will be systematically interpolated.26 copy : bool27 If True then the returned instances will be copies.28 29 Returns30 -------31 insts_bads : list32 The list of instances, with the same channel(s) marked as bad in all of33 them, possibly with some formerly bad channels interpolated.34 r � z+interp_thresh must be between 0 and 1, got c S s g | ]}|j d �qS )�bads��info��.0�inst� r ��E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\mne/preprocessing/interpolate.py�35<listcomp>2 � z!equalize_bads.<locals>.<listcomp>c S s g | ] }t |�t |j� �qS r ��len�timesr r r r r 4 s c S s g | ]}t |j��qS r r r r r r r 6 r c 3 s* � | ]\}}� |j d vr�| V qdS )r Nr )r �kr )�ch_name� durationsr r � <genexpr>; s � ��z equalize_bads.<locals>.<genexpr>c s g | ] \}}� | �k r|�qS r r )r r �ch)�36good_times�
interp_threshr r r C s c S s g | ]}|� � �qS r )�copyr r r r r E � r )�exclude)�37ValueError�list�setr �
from_iterable�38isinstancer �append�sum� enumerate�npr r �interpolate_bads)Zinstsr% r& Zall_badsZ bads_keepr r )r r! r$ r% r �
equalize_bads s. ��� r3 � c s� t � tttf� t|d�}|dkrtd|� d���� �� }|du r%td��|�� }|d dkr8td |d � ���� j d39 }t40� � j d41< t t t
|� ���t�t��t��f�}|D ] }t��|d �}t��|d �} d ||| f< d || |f< qXt|d d
�\}42���fdd�t ��D �} � fdd�| D �} | D ]}
t|
�|kr�tdd�|
�� d���q�t� }t � }t| �D ]�\}}� fdd�|D �}
|�|
�}t|d |
�}td|d � �g� j d d�}||d d d dd�< � j|
d �}t� t��rtj|dd �� d d �}t ||� j d �}n=t� t��r%tj|d d �� t|�d d �}t ||� j d �}n tj|dd �� d d �}t!tj|dd �� d d �|� j � j"� j#d!�}||d|d � �< q�� j$t43|�%� �d"d#� t44|�� j d45< � �&� � �'t46|�(� �� |� j d47< � S )$a� Interpolate bridged electrode pairs.48 49 Because bridged electrodes contain brain signal, it's just that the50 signal is spatially smeared between the two electrodes, we can51 make a virtual channel midway between the bridged pairs and use52 that to aid in interpolation rather than completely discarding the53 data from the two channels.54 55 Parameters56 ----------57 inst : instance of Epochs, Evoked, or Raw58 The data object with channels that are to be interpolated.59 bridged_idx : list of tuple60 The indices of channels marked as bridged with each bridged61 pair stored as a tuple.62 bad_limit : int63 The maximum number of electrodes that can be bridged together64 (included) and interpolated. Above this number, an error will be65 raised.66 67 .. versionadded:: 1.268 69 Returns70 -------71 inst : instance of Epochs, Evoked, or Raw72 The modified data object.73 74 See Also75 --------76 mne.preprocessing.compute_bridged_electrodes77 � bad_limitr zEArgument 'bad_limit' should be a strictly positive integer. Provided z is invalid.Nz&No channel positions found in ``inst``Z coord_frame�headz2Montage channel positions must be in ``head`` got r r F)Zdirectedc s, g | ]}�fd d�t �� |k�d D ��qS )c � g | ]}� | �qS r r )r �j)�nodesr r r � r' �=interpolate_bridged_electrodes.<locals>.<listcomp>.<listcomp>r )r1 �where�r r )�labelsr9 r r r � s , z2interpolate_bridged_electrodes.<locals>.<listcomp>c s g | ] }� fd d�|D ��qS )c � g | ]}� j j| �qS r �r Zch_namesr< �r r r r � � r: r )r � group_idxr@ r r r � s �z
The channels z, ze are bridged together and form a large area of bridged electrodes. Interpolation might be inaccurate.c r> r r? r< r@ r r r � rA �ch_poszvirtual ZsfreqZeegZchs�loc� )Zpicks�Zaxis�����)�78first_samp)�tmin)rI �nave�kindT)Zforce_update_info))r r79 r r r r) Z get_montage� RuntimeErrorZ
get_positionsr r* �sortedr+ r r1 Zzerosr Z searchsortedr �join�dictr0 �union�_find_centroid_spherer �get_datar- �averageZreshaper rH r rI r rJ rK Z add_channels�valuesr2 Z
drop_channels�keys)r Z bridged_idxr5 Zmontage�posZ bads_origZG_denseZbridgeZidx0Zidx1�_Z80groups_idxZ groups_names� group_namesZ virtual_chsr r rB Z pos_virtualZ virtual_info�dataZ81virtual_chr )r r= r9 r � interpolate_bridged_electrodesO s� 82�� �83 84� ��85 �86rZ c sh t �� fdd�|D ��}t|�}t j|dd�}t|�}t j|dd�d }||d< t|�ddd�f }|S )a Compute the centroid position between N electrodes.87 88 The centroid should be determined in spherical "head" coordinates which is89 more accurante than cutting through the scalp by averaging in cartesian90 coordinates.91 92 A simple way is to average the location in cartesian coordinate, convert93 to spehrical coordinate and replace the radius with the average radius of94 the N points in spherical coordinates.95 96 Parameters97 ----------98 ch_pos : OrderedDict99 The position of all channels in cartesian coordinates.100 group_names : list | tuple101 The name of the N electrodes used to determine the centroid.102 103 Returns104 -------105 pos_centroid : array of shape (3,)106 The position of the centroid in cartesian coordinates.107 c r7 r r )r r �rC r r r � r' z)_find_centroid_sphere.<locals>.<listcomp>r rF )r r N)r1 �arrayr rS r
)rC rX Zcartesian_positionsZsphere_positionsZcartesian_pos_centroidZsphere_pos_centroidZ108avg_radiusZ pos_centroidr r[ r rQ � s rQ )r T)r4 )�__doc__� itertoolsr �numpyr1 Zscipy.sparse.csgraphr Z_fiff.meas_infor Zepochsr r Zevokedr r �ior109 r Z110transformsr r
�utilsr r r3 rZ rQ r r r r �<module> s 111112; x