U f{j8@sddlZddlmZddlZddlZddlmZmZddlm Z m Z m Z ddl m Z ddlmZdd lmZedd d ZddZGddde ZdS)N)path)loggerfill_doc)_read_segments_file_find_channels _create_chs)BaseRaw) _empty_info)FIFFFc Cst||||||||dS)aRead Nicolet data as raw object. ..note:: This reader takes data files with the extension ``.data`` as an input. The header file with the same file name stem and an extension ``.head`` is expected to be found in the same directory. Parameters ---------- input_fname : str Path to the data file (ending with ``.data`` not ``.head``). ch_type : str Channel type to designate to the data channels. Supported data types include 'eeg', 'dbs'. eog : list | tuple | 'auto' Names of channels or list of indices that should be designated EOG channels. If 'auto', the channel names beginning with ``EOG`` are used. Defaults to empty tuple. ecg : list or tuple | 'auto' Names of channels or list of indices that should be designated ECG channels. If 'auto', the channel names beginning with ``ECG`` are used. Defaults to empty tuple. emg : list or tuple | 'auto' Names of channels or list of indices that should be designated EMG channels. If 'auto', the channel names beginning with ``EMG`` are used. Defaults to empty tuple. misc : list or tuple Names of channels or list of indices that should be designated MISC channels. Defaults to empty tuple. %(preload)s %(verbose)s Returns ------- raw : instance of Raw A Raw object containing the data. See Also -------- mne.io.Raw : Documentation of attribute and methods. )eogecgemgmiscpreloadverbose) RawNicolet) input_fnamech_typerrrrrrr r U/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/nicolet/nicolet.pyread_raw_nicolets , rc Cs:t|\}}|dkr&td|d|d}tdt}t|dv} | D]j} | d\} } | dkr|| d d d } n4| d krt| } n"| d krt | } n| dkrt| } | || <qNW5QRX|d} |dkrt | d}|dkrt | d}|dkrt | d}|d\}}|d}|d}|dd\}}t t |dt |d t |dt |dt |d t |t |}t |d}t |df|d<|dkrtj}tj}n |dkrtj}tj}ntdt|d dt| }t| ||||||||d<d|d <|d!d"|d#<d$|_|||fS)%z'Extract info from Nicolet header files.z.dataz%File name should end with .data not "z".z.headzReading header...r=Z elec_names,Zconversion_factor)Z num_channelsZrec_idZadm_idZpat_id num_samplesZstart_tsautoZEOGZECGZEMG-:r.rZ sample_freqZ meas_dateZeegZseegzBChannel type not recognized. Available types are 'eeg' and 'seeg'.gư>ZchsgZhighpassZsfreqg@ZlowpassF)rsplitext ValueErrorrinfodictopensplitfloatintrdatetimer calendartimegm utctimetupler ZFIFFV_COIL_EEGZ FIFFV_EEG_CHZ FIFFV_SEEG_CH TypeErrornprepeatlenr Z _unlockedZ_update_redundant)fnamerrrrr extensionheader header_infoZfidlinevarvalueZch_namesdatetimesecZmsecr%Zch_coilZch_kindcalsr r r_get_nicolet_infoAsn           "    r>cs*eZdZdZd fdd ZddZZS) raRaw object from Nicolet file. Parameters ---------- input_fname : str Path to the Nicolet file. ch_type : str Channel type to designate to the data channels. Supported data types include 'eeg', 'seeg'. eog : list | tuple | 'auto' Names of channels or list of indices that should be designated EOG channels. If 'auto', the channel names beginning with ``EOG`` are used. Defaults to empty tuple. ecg : list or tuple | 'auto' Names of channels or list of indices that should be designated ECG channels. If 'auto', the channel names beginning with ``ECG`` are used. Defaults to empty tuple. emg : list or tuple | 'auto' Names of channels or list of indices that should be designated EMG channels. If 'auto', the channel names beginning with ``EMG`` are used. Defaults to empty tuple. misc : list or tuple Names of channels or list of indices that should be designated MISC channels. Defaults to empty tuple. %(preload)s %(verbose)s See Also -------- mne.io.Raw : Documentation of attribute and methods. r FNc sTt|}t||||||\} } | ddg} tt|j| ||g| g| d|ddS)Nrrr*) filenamesZ raw_extras last_sampsZ orig_formatr)rabspathr>superr__init__) selfrrrrrrrrr%r6r@ __class__r rrCs   zRawNicolet.__init__c Cst||||||||dd dS)zRead a chunk of raw data.zrr r r rs$     /?