U f{j%@sddlmZddlZddlmZddlmZddlm Z ddl m Z m Z m Z ddlmZmZd d lmZmZmZd d ZdddZedddZddZddZdS))deepcopyN)FIFF)read_tag) dir_tree_find) start_block end_block write_int)write_named_matrix_read_named_matrix)loggerverbose_plcCsh|dtddkrd|d<nH|dtddkr8d|d<n,|dtddkrTd |d<nt|d|d<d S) zConvert CTF kind to MNE kind.ctfkindZ47314252rkindZ47324252r Z47334252N)int)onerN/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/ctf_comp.py _add_kinds   rrangecalFcCs2dd|D}tt|}tt|}t||f||fdD]\} } } tt| D]} || | } | dkrtd| | | t| f|| | }|||d|||d}| rt d|nt |}|rd|n|}|| | <qTq>||d<||d <|d d d f|d d |d d d f|d d <d S) zGet row and column cals.cSsg|] }|dqS)Zch_namer).0crrr #sz#_calibrate_comp..)FTrzDChannel %s does not appear exactly once in data, found %d instance%sr?rowcalscolcalsNdata) npZzeroslenziprcount RuntimeErrorrindexfloat)compchs row_names col_namesZ mult_keysflipZch_namesZrow_calsZcol_calsnamesZcalsinviipidxvalrrr_calibrate_comp s0   r5cCst|||dS)aRead the CTF software compensation data from the given node. Parameters ---------- fid : file The file descriptor. node : dict The node in the FIF tree. chs : list The list of channels from info['chs'] to match with compensators that are read. %(verbose)s Returns ------- compdata : list The compensation data N)_read_ctf_comp)fidnoder+rrrr read_ctf_comp9sr9cCsddlm}|dkrtn|}g}t|tj}|D]J}t||tj}t|dD]8}|d|j } |d|j } | tj krRt || } qqRt dt| jd} ~ t| t|dD]>}|d|j } |d|j } | tjkrt || } | j} qqd} t| | d <|| d <|| g|| s8t| ||d |d nr rAgetrCrr r?r)r7rHr*r"rrrwrite_ctf_comps&     " rK)rF)N)copyrnumpyr# constantsrrIrtreerwriterrr matrixr r utilsr rrrr5r9r6rKrrrrs      I