U f»{j†Bã@s–ddlmZddlmZddlmZddlZddlZddl Z ddl Z ddl m Z ddlmZmZddlmZd Zd d „Zd d „ZdZdQdd„ZdZdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Z d#d$„Z!d%d&„Z"d'd(„Z#d)d*„Z$d+d,„Z%d-d.„Z&d/d0„Z'd1d2„Z(dRd3d4„Z)d5d6„Z*d7d8„Z+dSd9d:„Z,edTd;d<„ƒZ-dUd>d?„Z.d@dA„Z/dBdC„Z0dDdE„Z1dVdFdG„Z2dHdI„Z3dJdK„Z4dWdMdN„Z5dOdP„Z6dS)Xé)Úcontextmanager)ÚGzipFileNé)ÚFIFFé)ÚloggerÚ _file_like)Ú_cal_to_julian)réÿÿÿcCs”t|tjƒr||j9}| tj|dd ¡¡| tj|dd ¡¡| tj|dd ¡¡| tjtjdd ¡¡| tj||d ¡¡dS)z Write data.ú>i4©ÚdtypeN) Ú isinstanceÚnpZndarrayÚsizeÚwriteÚarrayÚtobytesrÚFIFFV_NEXT_SEQ)ÚfidÚdataÚkindÚ data_sizeZ FIFFT_TYPEr ©rúK/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/write.pyÚ_writes  rcCsft|tƒrRtddd |dd…d¡}|dkr6tdƒ‚tt|dd…ƒd|ƒ}|dkrbtd ƒ‚|S) z7Convert human-readable bytes to machine-readable bytes.éé)ÚMBÚGBéþÿÿÿNz-split_size has to end with either"MB" or "GB"rìz$split_size cannot be larger than 2GB)rÚstrÚdictÚgetÚ ValueErrorÚintÚfloat)Z split_sizeÚexprrrÚ_get_split_size)s r)iFcCsx| tjtjdd ¡¡| tjtjdd ¡¡| tjddd ¡¡|rVtjntj}| tj|dd ¡¡dS)zWrite a FIFF_NOP.r r rN) rrrrZFIFF_NOPrZ FIFFT_VOIDZFIFFV_NEXT_NONEr)rÚlastZnext_rrrÚ write_nop:s r+r cCs’d}t |¡}|jjdkr6|jdkr6td|j›dƒ‚|jdkrH| ¡nd}|tkrntd|›dt›d|›ƒ‚| d ¡j }t ||||t j d ƒd S) z)Write a 32-bit integer tag to a fif file.éZuibrz$Cannot safely write data with dtype z as intzValue z exceeds maximum allowed (z ) for tag r N) rZasarrayr rrÚ TypeErrorÚmaxÚ INT32_MAXZastypeÚTrrÚ FIFFT_INT)rrrrZmax_valrrrÚ write_intFs  ÿÿ r2cCs,d}tj|ddj}t||||tjdƒdS)z:Write a double-precision floating point tag to a fif file.éú>f8r N)rrr0rrÚ FIFFT_DOUBLE©rrrrrrrÚ write_doubleVsr7cCs,d}tj|ddj}t||||tjdƒdS)z:Write a single-precision floating point tag to a fif file.r,ú>f4r N)rrr0rrÚ FIFFT_FLOATr6rrrÚ write_float]sr:cCs,d}tj|ddj}t||||tjdƒdS)z%Write a dau_pack16 tag to a fif file.rz>i2r N)rrr0rrZFIFFT_DAU_PACK16r6rrrÚwrite_dau_pack16dsr;cCs,d}tj|ddj}t||||tjdƒdS)z8Write a 64 bit complex floating point tag to a fif file.r3ú>c8r N©rrr0rrÚFIFFT_COMPLEX_FLOATr6rrrÚwrite_complex64ksr?cCs,d}tj|ddj}t||||tjdƒdS)z9Write a 128 bit complex floating point tag to a fif file.éú>c16r Nr=r6rrrÚwrite_complex128rsrBcCsHt|ƒdkst‚d}t t|Ž¡}tj|dd}t||||tjdƒdS)z,Write a Julian-formatted date to a FIF file.ér,r r N) ÚlenÚAssertionErrorrÚsumr rrrZ FIFFT_JULIAN)rrrrZjdrrrÚ write_julianys rGcCs6| d¡}t|ƒ}d}|dkr2t||||tj|ƒdS)zWrite a string tag.Úlatin1z>arN)ÚencoderDrrZ FIFFT_STRING)rrrZstr_datarZmy_dtyperrrÚ write_string‚s  rJcCst||d |¡ƒdS)zeWrite a colon-separated list of names. Parameters ---------- data : list of strings ú:N)rJÚjoin)rrrrrrÚwrite_name_list‹srMcCsþd}tj|B}d|jd|jd}| tj|dd ¡¡| tj|dd ¡¡| tj|dd ¡¡| tjtjdd ¡¡| tj|dd ¡¡tj |jdtj d}|j ddd…|d|j…<|j|d<| tj|dd ¡¡t |ƒdS) z3Write a single-precision floating-point matrix tag.é@r,rr r r8Néÿÿÿÿ) rr9rÚndimrrrrrÚemptyÚint32ÚshapeÚcheck_fiff_length)rrÚmatÚ FIFFT_MATRIXZFIFFT_MATRIX_FLOATrÚdimsrrrÚwrite_float_matrix•s  rXcCsþd}tj|B}d|jd|jd}| tj|dd ¡¡| tj|dd ¡¡| tj|dd ¡¡| tjtjdd ¡¡| tj|dd ¡¡tj |jdtj d}|j ddd …|d|j…<|j|d <| tj|dd ¡¡t |ƒdS) z3Write a double-precision floating-point matrix tag.rNr3r,rr r r4NrO) rr5rrPrrrrrrQrRrSrT)rrrUrVZFIFFT_MATRIX_DOUBLErrWrrrÚwrite_double_matrix©s  rYcCsîd}tj|B}d|jd}| tj|dd ¡¡| tj|dd ¡¡| tj|dd ¡¡| tjtjdd ¡¡| tj|dd ¡¡tjdtj d}|j d|d<|j d|d<d |d <| tj|dd ¡¡t |ƒd S) zWrite integer 32 matrix tag.rNr,é r r rCrrrN) rr1rrrrrrrQrRrSrT)rrrUrVZFIFFT_MATRIX_INTrrWrrrÚwrite_int_matrix½s r[cCsþd}tj|B}d|jd|jd}| tj|dd ¡¡| tj|dd ¡¡| tj|dd ¡¡| tjtjdd ¡¡| tj|dd ¡¡tj |jdtj d}|j ddd …|d|j…<|j|d <| tj|dd ¡¡t |ƒdS) zWrite complex 64 matrix tag.rNr3r,rr r r<NrO) rr>rrPrrrrrrQrRrSrT)rrrUrVZFIFFT_MATRIX_COMPLEX_FLOATrrWrrrÚwrite_complex_float_matrixÒs  r\cCsþd}tj|B}d|jd|jd}| tj|dd ¡¡| tj|dd ¡¡| tj|dd ¡¡| tjtjdd ¡¡| tj|dd ¡¡tj |jdtj d}|j ddd …|d|j…<|j|d <| tj|dd ¡¡t |ƒdS) zWrite complex 128 matrix tag.rNr@r,rr r rANrO) rZFIFFT_COMPLEX_DOUBLErrPrrrrrrQrRrSrT)rrrUrVZFIFFT_MATRIX_COMPLEX_DOUBLErrWrrrÚwrite_complex_double_matrixæs  r]csndt ¡}t d|¡}|ddg7}ddlm‰d ‡fdd„|d d d …Dƒ¡}t tj |tj d d ¡}|S) z‹Get (mostly) unique machine ID. Returns ------- ids : array (length 2, int32) The machine identifier used in MNE. s%012xs..s00r©rIócsg|]}ˆ|dƒ‘qS)Ú hex_codecr)Ú.0Úhr^rrÚ szget_machid..NrOr)Úcount) ÚuuidÚgetnodeÚreÚfindallÚcodecsrIrLrZflipudZ frombufferrR)ÚmacZidsrr^rÚ get_machidús    "rkcCs:tt ¡dƒ\}}t|ƒt|dƒ}}tƒtj||dœS)zCreate a new file ID tag.gð?g€„.A)ÚmachidÚversionÚsecsÚusecs)ÚdivmodÚtimer&rkrÚ FIFFC_VERSION)rnrorrrÚget_new_file_id s ÿrscCsÀ|dkrtƒn|}d}| tj|dd ¡¡| tjtjdd ¡¡| tj|dd ¡¡| tjtjdd ¡¡tj|d|dd|dd|d |d gdd}| | ¡¡dS) zWrite fiff id.Nrr r rmrlrrrnro)Ú_generate_meas_idrrrrrZFIFFT_ID_STRUCTr)rrÚid_rZarrrrrÚwrite_ids  þþrvcCst|tj|ƒdS)zWrite a FIFF_BLOCK_START tag.N)r2rZFIFF_BLOCK_START©rrrrrÚ start_block&srxcCst|tj|ƒdS)zWrite a FIFF_BLOCK_END tag.N)r2rZFIFF_BLOCK_ENDrwrrrÚ end_block+srycCs¤t|ƒr*t dt|ƒ¡|}| d¡nLt|ƒ}t |¡d ¡dkrbt d¡t |ddd}nt d ¡t |dƒ}t |t j |ƒt|t jd ƒt|t jd ƒ|S) aROpen a fif file for writing and writes the compulsory header tags. Parameters ---------- fname : string | fid The name of the file to open. It is recommended that the name ends with .fif or .fif.gz. Can also be an already opened file. id_ : dict | None ID to use for the FIFF_FILE_ID. zWriting using %s I/Orrz.gzzWriting using gzipÚwbr)Ú compresslevelzWriting using normal I/OrO)rrÚdebugÚtypeÚseekr"ÚopÚsplitextÚlowerrÚopenrvrZ FIFF_FILE_IDr2ZFIFF_DIR_POINTERZFIFF_FREE_LIST©ÚfnamerurrrrÚ start_file0s     r…c cs*t||d}|Vt|ƒW5QRXdS)z3Start and (if successfully written) close the file.©ruN)r…Úend_filerƒrrrÚstart_and_end_fileQsrˆTcCs$| ¡dkr |r| ¡tdƒ‚dS)z8Ensure our file hasn't grown too large to work properly.r!zoFIFF file exceeded 2GB limit, please split file, reduce split_size (if possible), or save to a different formatN)ÚtellÚcloseÚIOError)rrŠrrrrTYs rTcCs t|ddt|ƒ| ¡dS)z9Write the closing tags to a fif file and closes the file.T)r*N)r+rTrŠ)rrrrr‡cs r‡cCsfd}| tjtjdd ¡¡| tjtjdd ¡¡| tj|dd ¡¡| tjtjdd ¡¡| tj|ddd ¡¡| tj|ddd ¡¡|ddd…dd…f}|ddd…df}| tj|d d ¡¡| tj|d d ¡¡tj  |d¡}|dd…dd…f}|dd…df}| tj|d d ¡¡| tj|d d ¡¡dS) z,Write a coordinate transformation structure.éhr r ÚfromÚtoÚtransNrCr8) rrrrZFIFF_COORD_TRANSrZFIFFT_COORD_TRANS_STRUCTrZlinalgÚinv)rrrZrotÚmoveZ trans_invrrrÚwrite_coord_transjs r’cCs¨d}| tjtjdd ¡¡| tjtjdd ¡¡| tj|dd ¡¡| tjtjdd ¡¡| tj|ddd ¡¡| tj|ddd ¡¡| tj|ddd ¡¡| tj|ddd ¡¡| tj|d dd ¡¡| tj|d dd ¡¡| tj|d dd ¡¡| tj|d dd ¡¡| tj|d dd ¡¡|ddd…}| tj|dd ¡¡| ddt|ƒ¡dS)z1Write a channel information record to a fif file.é`r r ZscannoZlognorÚranger8ÚcalZ coil_typeÚlocÚunitZunit_mulÚch_nameNéz>cór@) rrrrZ FIFF_CH_INFOrZFIFFT_CH_INFO_STRUCTrrD)rÚchrr˜rrrÚ write_ch_info‚s"rœcCs |dk rd}|rt|tjƒ|dk r4t|tj|ƒ|D]º}| t tjd¡  ¡¡| t tj d¡  ¡¡| tj|dd  ¡¡| t tj d¡  ¡¡| t |dd¡  ¡¡| t |dd¡  ¡¡| t |ddd…d ¡  ¡¡q8|rt |tjƒdS) z5Write a set of digitizer data points into a fif file.Nrr r rÚidentÚrrCr8) rxrZ FIFFB_ISOTRAKr2ZFIFF_MNE_COORD_FRAMErrrZFIFF_DIG_POINTrZFIFFT_DIG_POINT_STRUCTrry)rÚdigÚblockZ coord_framerÚdrrrÚwrite_dig_pointsžs   $r¢cCst|||ddS)z:Write a single-precision sparse compressed row matrix tag.Úcsr©Úfmt©Úwrite_float_sparse©rrrUrrrÚwrite_float_sparse_rcs³sr©cCst|||ddS)z=Write a single-precision sparse compressed column matrix tag.Úcscr¤r¦r¨rrrÚwrite_float_sparse_ccs¸sr«ÚautocCs¦ddlm}ddlm}m}|dkr8t||jƒr4dnd}|dkrL|j}|}n |j}|}t||ƒsxtd|  ¡t |ƒfƒ‚|d >} t j | B} |j } |jd} d | d | d | dd } | tj|d d  ¡¡| tj| d d  ¡¡| tj| d d  ¡¡| tjt jd d  ¡¡| tj|jd d  ¡¡| tj|jd d  ¡¡| tj|jd d  ¡¡| |jd|jddg}| tj|d d  ¡¡t|ƒdS)z:Write a single-precision floating-point sparse matrix tag.r)Úsparser)Ú_matrix_coding_CCSÚ_matrix_coding_RCSr¬r£rªzMust write %s, got %sr@r,r r r8rN)Zscipyr­Útagr®r¯rZ csr_matrixZ csc_matrixr-Úupperr}rr9ZnnzrSrrrrrrÚindicesZindptrrT)rrrUr¥r­r®r¯ZneedÚbitsrVZFIFFT_MATRIX_FLOAT_RCSZnnzmZnrowrrWrrrr§½s4     r§cCs.tƒ}tj|d<tƒ|d<t\|d<|d<|S)zGenerate a new meas_id dict.rmrlrnro)r#rrrrkÚ DATE_NONEr†rrrrtàs   rt)F)N)N)N)T)FN)r¬)7Ú contextlibrÚgziprÚos.pathÚpathrrgrqreÚnumpyrÚ constantsrÚutilsrrZutils.numericsr r´rr)Z_NEXT_FILE_BUFFERr+r/r2r7r:r;r?rBrGrJrMrXrYr[r\r]rkrsrvrxryr…rˆrTr‡r’rœr¢r©r«r§rtrrrrÚs\           !    #