Aluode/PerceptionLabPortable
0
1o
2 '�Yi�, � @ s� d dl Z d dlZddlmZmZ ddlmZmZm Z e �3d�Z e �4d�Z e d!dd��Z
e d"d5d ��Zd d
� Zdd� Zddd�dd�Zdd� Zdddddd�dd�Zdd� Zdd � Zd d � Zd d � ZdS )#� N� )�
_picks_to_idx�6pick_types)�
_check_option�_validate_type�fill_doczS(\d+)_D(\d+) (\d+\.?\d*)zS(\d+)_D(\d+) (\w+)c s$ t �� fdd�t� |g d�D �t�S )a9 Determine the distance between NIRS source and detectors.7 8 Parameters9 ----------10 %(info_not_none)s11 %(picks_all_data)s12 13 Returns14 -------15 dists : array of float16 Array containing distances in meters.17 Of shape equal to number of channels, or shape of picks if supplied.18 c sB g | ] }t j�t j� d | d dd� �dd�dd�d ��qS )�chs�locr � � r )Zaxis)�npZlinalgZnorm�diffZreshape��.0�pick��info� ��E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\mne/preprocessing/nirs/nirs.py�19<listcomp>! s �,��z-source_detector_distances.<locals>.<listcomp>)�exclude)r �arrayr �float)r �picksr r r �source_detector_distances s 20 ��r �{�G�z�?c C s t | �|k S )a� Determine which NIRS channels are short.21 22 Channels with a source to detector distance of less than23 ``threshold`` are reported as short. The default threshold is 0.01 m.24 25 Parameters26 ----------27 %(info_not_none)s28 threshold : float29 The threshold distance for what is considered short in meters.30 31 Returns32 -------33 short : array of bool34 Array indicating which channels are short.35 Of shape equal to number of channels.36 )r )r � thresholdr r r �short_channels+ s r c C sX t | ddgg dd�}t� }|D ]}|�ttt�| d | ��� d ��� qt� |t37�S )z,Return the light frequency for each channel.�fnirs_cw_amplitude�fnirs_odT�r � allow_empty�ch_namesr ) r �list�append�roundr � _S_D_F_RE�match�groupsr r �int)r r �freqsr r r r �_channel_frequenciesA s �* r+ c C sB t | ddgg dd�}g }|D ]}|�| d | �d�d � q|S )z'Return the chromophore of each channel.�hbo�hbrTr r" � � )r r$ �split)r r Zchroma�iir r r �_channel_chromophoreM s38 r2 T�� throw_errors�39check_badsc s� t � ddgg dd�}t � ddgg dd�}t|�dkt|�dk@ r%td|�}t|�r0d }t}|}nd40}t}|}t�|�}|jd krLtd |� d
t |�� ���|j41j dk} t |�}| rj|ddgkritd|� d|� d���nt� t�
|�|�s~td|� d|� d���t|�d dkr�tdt� j�� d�|�}� fdd�|D �}42t�t�|43��r�td|44� d�|�}|D ]A} |�� d | d �} t| �s�td t� j| �� �|�} n"| �� d }
t|�r�|
}
q�|
d vr�td � d | d � �|�} nq�|t�� fd d�|D �� }t|d d d� |d!d d� �D ]\}}� d | d }� d | d }|�|�}|�|�}|�� d d"� \}}}|�� d d"� \}}}t|��rOt|�t|�}}||k�sg||k�sg||d k�sg||d! k�r�td#|� d$|d � d%|d! � d&|� d'|� d(t|�� d)t|�� �|�} n�q
|�r�t|d d d� |d!d d� �D ]+\}}� j| � j| g}t t� d* ��|��}t|�d!k�r�td+|� d,|� ����q�|S )-ai Check channels follow expected fNIRS format.45 46 If the channels are correctly ordered then an array of valid picks47 will be returned.48 49 If throw_errors is True then any errors in fNIRS formatting will be50 thrown to inform the user. If throw_errors is False then an empty array51 will be returned if the channels are not sufficiently formatted.52 r r Tr r, r- r zqMNE does not support a combination of amplitude, optical density, and haemoglobin data in the same raw structure.Z frequencies� chromophore)r z Exactly two z must exist in info, got �UzThe z) in info must be ["hbo", "hbr"], but got z insteadz0 in info must be unique and sorted, but got got r zRNIRS channels not ordered correctly. An even number of NIRS channels is required. z channels wereprovidedc s g | ] }� d | d d �qS )r r r53 r )r r1 r r r r � � z+_check_channels_ordered.<locals>.<listcomp>zjNIRS channels is missing wavelength information in the info["chs"] structure. The encoded wavelengths are �.r Zch_namez^NIRS channels have specified naming conventions. The provided channel name can not be parsed: )r, r- z{NIRS channels have specified naming conventions.Chromophore data must be labeled either hbo or hbr. The failing channel is c � g | ]}� d | �qS �r" r r r r r r � � Nr/ r zhNIRS channels not ordered correctly. Channels must be ordered as source detector pairs with alternating r. z & z , but got SZ_Dz pair z and �badsz,NIRS bad labelling is not consistent, found z but needed )r �len�_throw_or_return_emptyr& � _S_D_H_REr r �shape�54ValueErrorr# Zdtype�kindZ array_equal�uniquer" �any�isnanr' �bool�reprr( Zargsort�zipr �set� intersection� RuntimeError) r � pair_valsr4 r5 �55picks_wave� picks_chromar Z56error_wordZuse_REZis_strZ all_freqsr Z ch_name_info�valuer1 �jj�ch1_name�ch2_nameZch1_reZch2_reZch1_SZch1_DZ ch1_valueZch2_SZch2_DZ ch2_valueZwant�gotr r r �_check_channels_orderedW s� ��5758� ��������� �� ��� 59&6061626364��������� &���rU c C s |rt | ��g S �N)rB )�msgr4 r r r r? � s r? )r4 �fnirs�whichr5 r! c C s� t |dtfd� tdddd�}td|dt|� � |dur@|| }|dkr)d d65gnd |� �}tt| |d ��s@t|� d
|� d���t� t66| ��}|j dkrO|} nt� t | ��} t
| | ||d�}67|68S )zCApply all checks to fNIRS info. Works on all continuous wave types.NrX zoptical densityzcontinuous waver6 )ZodZ cw_amplitude�hbrV rZ r, r- Zfnirs_)rX z must operate on z data, but none was found.r r3 )r �str�dictr �tupler> r rL r rD r+ �sizer2 rU ) r r4 rX rY r5 r! �kindsrC r* rM �outr r r �_validate_nirs_info� s, 69��70�ra c C s� t | dd�}t| d �}t|ddd� |ddd� �D ]"\}}| j| | j| }}||v r4|�|� q||v r=|�|� qt|�| d< | S )z*Spread bad labeling across fnirs channels.F)r5 r= Nr r/ )ra rJ rI r" �add�sorted)r r Znew_badsr1 rQ rR rS r r r �_fnirs_spread_bads s & 71� rd c s� t | ddgg dd�}t | ddgg dd�}t|�dkr t� n
t|�dkr&t� ng g fS t�� fdd �| jD ��}t�� fd72d �| jD ��}d d � |D �}d d � |D �}||fS )
z#Return list of unique optode names.r r Tr r, r- r c �" g | ]
}t � �|��� d ��qS )r �r) r' r( �r �ch��regexr r r 0 � " z'_fnirs_optode_names.<locals>.<listcomp>c re )r/ rf rg ri r r r 1 rk c S � g | ]}d |� ��qS )�Sr )r �sr r r r 3 � c S rl )�Dr )r �dr r r r 4 ro )r r> r& r@ r rD r" )r rN rO �sourcesZ detectorsZ src_namesZ det_namesr ri r �_fnirs_optode_names" s � rs c sT � fdd�| j D ��d�}d� v rtdd�}n d� v r tdd�}| d | d73 | S ) z Find the position of an optode.c s g | ]}� |v �qS r r )r �a��optoder r r ; s z$_optode_position.<locals>.<listcomp>Trm r � rp r74 r r )r" �index�range)r rv �idxZloc_idxr ru r �_optode_position9 s 75r{ c sZ | j � t� dg d�}� fdd�|D ��� fdd�|D �}t|�fdd�d�}| �|� d S ) NT)rX r c s g | ] }� d | � � d �qS )r" r )r0 r r r r r J r8 z!_reorder_nirx.<locals>.<listcomp>c r: r; r r r r r r K r< c s" � � | �� d �| jdd�d fS )Nr r/ )�maxsplit)rx r0 )�name)�prefixesr r �<lambda>N rk z _reorder_nirx.<locals>.<lambda>)�key)r r rc Zreorder_channels)�rawr Z76nirs_namesZ nirs_sortedr )r r~ r �
_reorder_nirxE s 77�r� rV )r )�re�numpyr Z78_fiff.pickr r �utilsr r r �compiler& r@ r r r+ r2 rU r? ra rd rs r{ r� r r r r �<module> s2 7980 81 82 �#