U f{j*@sddlmZddlmZmZddlmZddlZ ddl m Z m Z m Z mZddlmZmZddlmZdd lmZmZmZmZGd d d eZd d ZddZedddZddZede jde ddfddZ!dgdgfddZ"ddZ#dS) N)BytesIOSEEK_SET)GzipFile) read_tag_inforead_tagTag_call_dict_names) make_dir_tree dir_tree_find)FIFF)loggerverbose _file_likewarnc@sFeZdZdZddZddZddZdd Zefd d Z dd dZ dS) _NoCloseReadzDCreate a wrapper that will not close when used as a context manager.cCs ||_dSNfid)selfrrJ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/open.py__init__sz_NoCloseRead.__init__cCs|jSrrrrrr __enter__sz_NoCloseRead.__enter__cCsdSrr)rtype_value tracebackrrr__exit__sz_NoCloseRead.__exit__cCsdSrrrrrrclosesz_NoCloseRead.closecCs|j||Sr)rseek)roffsetwhencerrrr!"sz_NoCloseRead.seekcCs |j|Sr)rread)rsizerrrr%%sz_NoCloseRead.readN)r$) __name__ __module__ __qualname____doc__rrrr rr!r%rrrrrs rcCsht|rt|}|dnHt|}t|ddkrPtdt |d}ntdt |d}|S)z+Open a FIF file with no additional parsing.rrz.gzz Using gziprbzUsing normal I/O) rrr!stropsplitextlowerrdebugropen)fnamerrrr _fiff_get_fid)s     r3c Cspt|tj}d}|D]T}d}|dD]2}|jtjkrbt||j}t|j}|tj krbd}q\|jtj krt||j}t t ||j}|jtjkr&|dk rq&t||jj} t |\} } | d} | d} | | d| }|sd} | dkr0| dkr0t | d| d| | | | ddf}q&t | d| d| | | | ddf}q&|dk rqlq|S) z%Get the next filename in split files.N directory.-rr$rz%s-%d.%s)r r Z FIFFB_REFkindZ FIFF_REF_ROLErposintdataZFIFFV_ROLE_NEXT_FILEZFIFF_REF_FILE_NAMEr-joindirnameZFIFF_REF_FILE_NUMsplitfindrfindisdigit)rr2treeZ nodes_listZ next_fnameZnodesenttagZroleZnext_numpathbaseidxZidx2Znum_strrrr_get_next_fname9sL           rGFcCs:t|}zt|||WStk r4|YnXdS)aXOpen a FIF file. Parameters ---------- fname : str | fid Name of the fif file, or an opened file (will seek back to 0). preload : bool If True, all data from the file is read into a memory buffer. This requires more memory, but can be faster for I/O operations that require frequent seeks. %(verbose)s Returns ------- fid : file The file descriptor of the open file. tree : fif tree The tree is a complex structure filled with dictionaries, lists and tags. directory : list A list of tags. N)r3 _fiff_open Exceptionr )r2preloadrrrrr fiff_opends rKc Csz|r |}t|}W5QRXt|}dt|d}|jtjkrRt|d|jtj krlt|d|j dkrt|dt |}|jtj krt|dt d|t|j}d}|dkrt ||}|dkrtd |n |j} d }|rN|ddt} |jdkrN|} t|}|dkr:qNn| |_| |qt|| \} } t d |d|| | fS) Nzfile z does notz start with a file id tagz have a directory pointerz$ Creating tag directory for %s...TrzKFIF tag directory missing at the end of the file, possibly corrupted file: Fz[done])rr%rreprr7r Z FIFF_FILE_ID ValueErrortypeZFIFFT_ID_STRUCTr&rZFIFF_DIR_POINTERrr0r9r:rr!listnexttellr8appendr ) r2rrJZfid_oldrCprefixZdirposZ read_slowZdir_tagr4r8rA_rrrrHsJ           rHz c Csz|ttfkrtdt|tr&t|}t|\}}} tj|d<|} t| ||d|||d} W5QRX|tkrvd | } | S)aShow FIFF information. This function is similar to mne_show_fiff. Parameters ---------- fname : str Filename to evaluate. indent : str How to indent the lines. read_limit : int Max number of bytes of data to read from a tag. Can be np.inf to always read all data (helps test read completion). max_str : int Max number of characters of string representation to print for each tag's data. output : type Either str or list. str is a convenience output for printing. tag : int | None Provide information about this tag. If None (default), all information is shown. %(verbose)s Returns ------- contents : str The contents of the file. zoutput must be list or strblockr)indentlevel read_limitmax_strtag_id ) rPr,rN isinstancer9rKr Z FIFFB_ROOT _show_treer;) r2rXrZr[outputrCrfrAr4routrrr show_fiffs     rcFIFF_Z FIFF_UNITcs8tfddtD}t|dkr4dg}|S)zFind matching values.csHg|]@\}|krtfddDrtfddDsqS)c3s|]}|kVqdSrr).0fmtkrr sz(_find_type...c3s|]}|kVqdSrr)reexcrgrrris)any)revexcludefmtsrrgr s z_find_type..rz???)r9r itemslen)rrornvalsrrmr _find_types  rtc Csddlm}||}||d} |tt|dddt|ddgdg} d } |d kst| dt|rxd } |d d k rFd d|d Ddg} dd|d D} dd|d D}dd|d D}d}d }t| d d| dd ||| D]8\}}}}}| s||krqt ||d|}|d ks@||krjzt ||}Wnt k rhd }YnX||kr~|d7}qt|dgd}|dkrdt|ddnd}d}|j d k rdt|j d |}t |j tjr|j jdkr|dt|j j7}nt |j tr.|dtt|j 7}nt |j trT|dtt|j 7}njt |j ttfr~|dtt|j 7}n@||j r|d|j |j jf7}n|dtt|j 7}|sdn|}t|d|f}| | |t|dd|dt|d ||g7} | dd!d"| d<d}d }q|| krFd } | sVdg} d}|d#D]"}| t||||d|||7} q^| S)$zShow FIFF tree.r)sparserrWz = /ZFIFFB_)roFNTr4cSsg|] }|jqSr)r7rerBrrrrpsz_show_tree..r$cSsg|] }|jqSr)rOrwrrrrp scSsg|] }|jqSr)r&rwrrrrp scSsg|] }|jqSr)r8rwrrrrp srdxz: z ... array size=z ... dict len=z ... str len=z ... list len=z ... sparse (%s) shape=%sz ... type=z>>>>>>>>>>>>>>>>>>>>BADz?%s?z (zb %s)r]¶children)Zscipyrur,r9r;rtstrip startswithziprrrIr:r^npZndarrayr&dictrrrPtupleissparseZ getformatshaperOr getreplacer_)rrArXrYrZr[r\ruZthis_idtZnext_idtrbZ tag_foundkindstypessizesZposscounterZgoodrhknr&r8rrCZ this_typeprependZpostpendbranchrrrr_s     "    r_)FN)$os.pathrDr-iorrgziprnumpyrrCrrrr rAr r constantsr utilsrrrrobjectrr3rGrKrHinfr,rcrtr_rrrrs*   + = .