U f{jS@sbddlmZddlmZmZddlmZmZmZm Z m Z ddl m Z ddZ dd d Zd d ZdS))FIFF)find_taghas_tag) write_int start_block end_blockwrite_float_matrixwrite_name_list)loggercCsF|d|d|d<|d<|d|d|d<|d<|dj|d<dS)z Transpose mat inplace (no copy).ncolnrow col_names row_namesdataN)T)matrL/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/matrix.py_transpose_named_matrix sr Fc Cs|dtjkrpt|dD]<}|d|dtjkrt|d||r|d|}qqt|d|dSn t||st|d|dSt|||}|dkrtdn|j}|j \}} t||tj }|dk r|j|krtdt||tj }|dk r|j| krtdt||tj }|dk r6|j d ng} t||tj}|dk r^|j d ng} t|| | | |d } |rt| | S) avRead named matrix from the given node. Parameters ---------- fid : file The opened file descriptor. node : dict The node in the tree. matkind : int The type of matrix. transpose : bool If True, transpose the matrix. Default is False. %(verbose)s Returns ------- mat: dict The matrix data blockZnchildchildrenz.Desired named matrix (kind = %d) not availableNzMatrix data missingz@Number of rows in matrix data and FIFF_MNE_NROW tag do not matchzCNumber of columns in matrix data and FIFF_MNE_NCOL tag do not match:)r r rrr)rFIFFB_MNE_NAMED_MATRIXrangerr infor ValueErrorrshape FIFF_MNE_NROW FIFF_MNE_NCOLFIFF_MNE_ROW_NAMESsplitFIFF_MNE_COL_NAMESdictr) fidnodeZmatkindindentZ transposektagrr r rrrrrr_read_named_matrixsH      r*cCsB|d|d}|dj|krr|t|dj}||djkrN|dkrNd|}td|d|d||dj|ft|tjt|tj|dt|tj|dt |ddkrt |d|dkrtdt |tj |dt |d dkr"t |d |dkrtd t |tj |d t |||dt|tjd S) zWrite named matrix from the given node. Parameters ---------- fid : file The opened file descriptor. kind : int The kind of the matrix. matkind : int The type of matrix. r r rrzCannot write matrix: row (%i) and column (%i) total element (%i) mismatch with data size (%i), appears to be off by a factor of %gxrz$len(mat["row_names"]) != mat["nrow"]rz$len(mat["col_names"]) != mat["ncol"]N)sizefloatrrrrrrr lenr r!r#rr)r%kindrZn_totratiorrrwrite_named_matrixXs0  r1N)rF) constantsrr)rrwriterrrrr utilsr rr*r1rrrrs   D