U f{jL @sdZddlmZddlZddlmZmZmZddl m Z ddl m Z dd lm Z mZmZmZdd lmZdd lmZdd lmZd dlmZmZmZmZmZdddZedddZddZeGdddeZ dS)z*Conversion tool from Neuroscan CNT to FIF.)pathN)warnfill_doc _check_option)_topo_to_sphere)FIFF) _mult_cal_one_find_channels _create_chsread_str) _empty_info)BaseRaw) Annotations) _read_teeg_get_event_parser_session_date_2_meas_date$_compute_robust_event_table_position CNTEventType3int16c Cshd}d}|fdd}t|dR}||tj|ddd\}||tj|ddd\}t||d }W5QRXt|d}t||d } W5QRXt| jd } t|d }||d || j } W5QRXt | | } | st t t t d S|tj dd| Dt d|t| d|d} tj dd| Dt d}t dd| D}t | |||d dSd S)aUCNT Annotation File Reader. This method opens the .cnt files, searches all the metadata to construct the annotations and parses the event table. Notice that CNT files, can point to a different file containing the events. This case when the event table is separated from the main .cnt is not supported. Parameters ---------- fname: str path to cnt file containing the annotations. data_format : 'int16' | 'int32' Defines the data format the data is read in. Returns ------- annot : instance of Annotations The annotations. rxcSsH|dkr dnd}|tkr$|||9}|dd|}|||}|dS)NrrKr)r)offset n_channels event_type data_formatn_bytesZ event_timer"M/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/cnt/cnt.py_translating_function0s   z4_read_annotations_cnt.._translating_functionrbrQsz)_read_annotations_cnt..r)rrr cSsg|]}t|ddqS)ZLatencyg)getattrr+r"r"r#r.VscSsg|]}t|jqSr")strZStimTyper+r"r"r#r.Ys)onsetduration descriptionZ orig_time)openseeknpZ frombufferreadrrrr total_lengthlistrarrayfloattype)fnamer ZSETUP_NCHANNELS_OFFSETZSETUP_RATE_OFFSETr$r)rsfreqZevent_table_posZteegZ event_parserbufferZ my_eventsr1r2r3r"r"r#_read_annotations_cntsL        r@r"automm/dd/yyFc Cst|||||||||d S)a- Read CNT data as raw object. .. Note:: 2d spatial coordinates (x, y) for EEG channels are read from the file header and fit to a sphere to compute corresponding z-coordinates. If channels assigned as EEG channels have locations far away from the head (i.e. x and y coordinates don't fit to a sphere), all the channel locations will be distorted (all channels that are not assigned with keywords ``eog``, ``ecg``, ``emg`` and ``misc`` are assigned as EEG channels). If you are not sure that the channel locations in the header are correct, it is probably safer to replace them with :meth:`mne.io.Raw.set_montage`. Montages can be created/imported with: - Standard montages with :func:`mne.channels.make_standard_montage` - Montages for `Compumedics systems `_ with :func:`mne.channels.read_dig_dat` - Other reader functions are listed under *See Also* at :class:`mne.channels.DigMontage` Parameters ---------- input_fname : str Path to the data file. eog : list | tuple | 'auto' | 'header' Names of channels or list of indices that should be designated EOG channels. If 'header', VEOG and HEOG channels assigned in the file header are used. If 'auto', channel names containing 'EOG' are used. Defaults to empty tuple. misc : list | tuple Names of channels or list of indices that should be designated MISC channels. Defaults to empty tuple. ecg : list | tuple | 'auto' Names of channels or list of indices that should be designated ECG channels. If 'auto', the channel names containing 'ECG' are used. Defaults to empty tuple. emg : list | tuple Names of channels or list of indices that should be designated EMG channels. If 'auto', the channel names containing 'EMG' are used. Defaults to empty tuple. data_format : 'auto' | 'int16' | 'int32' Defines the data format the data is read in. If 'auto', it is determined from the file header using ``numsamples`` field. Defaults to 'auto'. date_format : 'mm/dd/yy' | 'dd/mm/yy' Format of date in the header. Defaults to 'mm/dd/yy'. %(preload)s %(verbose)s Returns ------- raw : instance of RawCNT. The raw data. See Also -------- mne.io.Raw : Documentation of attribute and methods. Notes ----- .. versionadded:: 0.12 )eogmiscecgemgr date_formatpreloadverbose)RawCNT) input_fnamerCrDrErFr rGrHrIr"r"r# read_raw_cnt`sC rLc/ Csd}t}t|ddd} | dt| d} | r@t| nd} | dt| d} t| dkrp| dnd} t| dkr| d nd} | d d t| d }|d krtj }n|d krtj }ntj }t| d }|dkrtj }n|dkrtj }nd}| dt| d}dt| dt| df}t||}| dtj| dd dd}| dtj| dd dd}|dkr| dddtj| dd dD}| dtj| dd d d}tj| dd d d}| d!tj| d"d dd}| d#tj| d$d dd}| d d tj| d$d dd}t| |d%}| d&tj| d'd dd|d(<||krf||}n||d)|}td*|d+d,d-g|d+kr|dks|||d.krtd/d }|||}n |||}n|d,krd nd0}|||}tj| d"d dd|d1<|d1d kr,|d1|<nd |d1<tttttf\}}}} }!t}"t|D]<}#| |d)|#t| d}$||$| |d)|#d0tj| d2d ddr|"|$| |d)|#d3tj| d$d d}%|%d d 9<|!|%| |d)|#d4|tj| dd dd| |d)|#d5tj| d$d dd}&| |d)|#d6tj| d$d dd}'||'|&d7d8qdW5QRXt|}(|d kr||(d9<|d kr||(d:<|| || | d;})|d+krt|d<}|d+krt|d=}|d+kr&t|d>}t||tjtj||||} d?dt| D}*t|!|*}+tt| dftj},|+|,dddd@f<t | |,D]\}-}.|-j!|.dAq|j!t"|||dBt#|dkrdnt#|}|(j!|||"|)| dCdD|(_$|(%|(|fS)EzRead the cnt header.rr%r bufferingyrrMFRLNz%s %s rr&r'countrheadericSsg|]}|dkr|qS)rr")r,idxr"r"r#r.sz!_get_cnt_info..i2ii1)r\i`gi@ZlowpassZhighpass)handidsex first_name last_nameZEOGZECGZEMGcSs"g|]\}}|dtjkr|qS)Z coil_type)r FIFFV_COIL_EEG)r,r^chr"r"r#r.+sr)loc) baselines n_samplesr!) meas_dater3bads subject_infochsF)&dictr4r5r isdigitintsplitlenr ZFIFFV_SUBJ_SEX_MALEZFIFFV_SUBJ_SEX_FEMALEZFIFFV_SUBJ_SEX_UNKNOWNZFIFFV_SUBJ_HAND_RIGHTZFIFFV_SUBJ_HAND_LEFTrr6fromfilerrrr9rangeappendrr r rnZ FIFFV_EEG_CH enumeraterfullnanzipupdater:r0Z _unlockedZ_update_redundant)/rKrCrErFrDr rGZ data_offsetcnt_infor)Z patient_idZ patient_namermrlrkriZ session_labelZ session_datersrr>Zlowpass_toggleZhighpass_togglerrZ lowcutoffZ highcutoffZ event_offset data_sizer!Zch_namescalsrqrvposrtZch_idxZch_nameZxyZ sensitivitycalinforuZeegsZcoordsZlocsrorpr"r"r# _get_cnt_infos                           $          rcs*eZdZdZd fdd Zd d ZZS) rJa Raw object from Neuroscan CNT file. .. Note:: The channel positions are read from the file header. Channels that are not assigned with keywords ``eog``, ``ecg``, ``emg`` and ``misc`` are assigned as eeg channels. All the eeg channel locations are fit to a sphere when computing the z-coordinates for the channels. If channels assigned as eeg channels have locations far away from the head (i.e. x and y coordinates don't fit to a sphere), all the channel locations will be distorted. If you are not sure that the channel locations in the header are correct, it is probably safer to use a (standard) montage. See :func:`mne.channels.make_standard_montage` Parameters ---------- input_fname : str Path to the CNT file. eog : list | tuple 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. misc : list | tuple Names of channels or list of indices that should be designated MISC channels. Defaults to empty tuple. ecg : list | tuple 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 | tuple 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. data_format : 'auto' | 'int16' | 'int32' Defines the data format the data is read in. If 'auto', it is determined from the file header using ``numsamples`` field. Defaults to 'auto'. date_format : 'mm/dd/yy' | 'dd/mm/yy' Format of date in the header. Defaults to 'mm/dd/yy'. %(preload)s stim_channel : bool | None Add a stim channel from the events. Defaults to None to trigger a future warning. .. warning:: This defaults to True in 0.18 but will change to False in 0.19 (when no stim channel synthesis will be allowed) and be removed in 0.20; migrate code to use :func:`mne.events_from_annotations` instead. .. versionadded:: 0.18 %(verbose)s See Also -------- mne.io.Raw : Documentation of attribute and methods. r"rArBFNc  std|ddg|dkrd} nd} t|}t||||||| \} } | ddg} tt|j| ||g| g| d| d | d d krd nd }|t||ddS)NrGrBzdd/mm/yyz%d/%m/%y %H:%M:%Sz%m/%d/%y %H:%M:%Srrrry) filenamesZ raw_extras last_sampsZ orig_formatrIr!rrbr)r ) rrabspathrsuperrJ__init__Zset_annotationsr@)selfrKrCrDrErFr rGrHrIZ _date_formatrrr __class__r"r#rys.    zRawCNT.__init__c" Cs|j|d}d|j|kr4|d} |j|d} n|} d} |j|d} |j|d} |j|d} | dkrrd nd }| | }||| }td | ||}t||}|| }t|j|d d d}|d| d||| td ||| D]|}|t|| || |f}||}t||f}|d ksH|| d krL|nd }|| dkrf||7}|| ||}||}tj |||d}|j || | fdd}t|| |f}t |D]0\}}t || |d| }||d| |f<qd|j|kr||} ||}!| | |!|d<|||||f||<||| |dddf8<t |dd||f||||qW5QRXdS)z>Take a chunk of raw data, multiply by mult or cals, and store.Z orig_nchanZ stim_channelrNrcrqr!rrazs.       H G