U f{j#@sdZddlZddlZddlmZddlZddlmZmZddl m Z ddl Z ddl mZmZmZmZmZddlmZdd lmZdd lmZdd l mZmZdd lmZmZdd lmZddl m!Z!eGdddeZ"ddZ#ddZ$dFddZ%ddZ&e'dddZ(e'dddd Z)e'dd!d!d"Z*e'd#d$d%d"Z+d&d'd'd(d)d*d*d*d'd'd*d*d*d+ Z,d,d-Z-d.d/Z.ed0d1Z/edGd4d5Z0ddd6d7d8Z1d9d:d;Z2dd?d?eZ5d@dAZ6dHdBdCZ7dDdEZ8dS)Iz,Conversion tool from BrainVision EEG to FIF.N)datetimetimezone)StringIO)verboseloggerwarnfill_doc_DefaultEventParser)FIFF) _empty_info)BaseRaw)_read_segments_file _mult_cal_one) Annotationsread_annotations)make_dig_montage)HEAD_SIZE_DEFAULTcs.eZdZdZed fdd Zd d ZZS) RawBrainVisionaRaw object from Brain Vision EEG file. Parameters ---------- vhdr_fname : str Path to the EEG header file. eog : list or tuple Names of channels or list of indices that should be designated EOG channels. Values should correspond to the header file. Default is ``('HEOGL', 'HEOGR', 'VEOGb')``. misc : list or tuple of str | 'auto' Names of channels or list of indices that should be designated MISC channels. Values should correspond to the electrodes in the header file. If 'auto', units in header file are used for inferring misc channels. Default is ``'auto'``. scale : float The scaling factor for EEG data. Unless specified otherwise by header file, units are in microvolts. Default scale factor is 1. %(preload)s %(verbose)s Attributes ---------- impedances : dict A dictionary of all electrodes and their impedances. See Also -------- mne.io.Raw : Documentation of attribute and methods. ZHEOGLZHEOGRZVEOGbauto?FNc std|t|}t|d}|dkr2dnd} t||||\} } } } }}}}t| d}t| tr| dkrd}t t | d D]}|t | 7}qt |gd d |D}t |d }n| d krtn:t | d}|dtj|}t| }d}|||}W5QRXt| tr(dn| }t|| | |d}tt|j| |d g| g||||g|d||t|\}}}}|}t||jd|_t|| d}||| r|t dS)Nz Extracting parameters from %s....ahdrTFrbFrZ skiplinescSsg|] }t|qS)len).0linerr]/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/brainvision/brainvision.py Ssz+RawBrainVision.__init__..Cnchansingle)offsetsfmtorder n_samples)Z last_samps filenames orig_formatpreloadr raw_extras orig_units meas_datesfreq)!rinfoopabspathsplitext _get_hdr_infoopen isinstancedictrangeintrreadlinenpZcumsumNotImplementedErrorseekosSEEK_ENDtell_fmt_byte_dictsuperr__init__Z set_montage _aux_hdr_info splitlines_parse_impedance impedancesrZset_annotationsZ drop_channels_AHDR_CHANNEL_NAME) self vhdr_fnameeogmiscscaler-r hdr_fnameext ahdr_formatr2 data_fnamer(r)r* mrk_fnamemontager/fZn_skipiir' n_data_chZ dtype_bytesr,r.settingscfgZcinfo_Zsplit_settingsZannots __class__rr!rEAsd        zRawBrainVision.__init__c Csz|j|d}|j|d} |j|ddkrHt||||||||n.t| trxt| } t||||||||| |d n|j|d} t|j|d} | | |t |||f} t ||D]}| d}|}t| trd | kr| d d kr|d d }d |kr"|}nd |kr8|d }ntd dd|D| d||f<qW5QRXt|| |||dS)zRead a chunk of raw data. orig_nchanr(r)r$)dtype n_channelsr'rASCIIZ decimalsymbol., zVUnknown BrainVision data format encountered. Please contact the MNE-Python developers.cSsg|] }t|qSrfloat)rpartrrr!r"sz5RawBrainVision._read_segment_file..N) _raw_extras_read_segments_cr8str_fmt_dtype_dictrr7 _filenamesr?r=emptyr:r<decodestripr9replacesplit RuntimeErrorr)rKdataidxfistartstopcalsmultrXr(r_r'fidblockrWr Z line_datarrr!_read_segment_filevs@        &z!RawBrainVision._read_segment_file)rrrFN)__name__ __module__ __qualname____doc__rrEr| __classcell__rrr\r!r s4rc Cs|j|d}|j|d} t| } t| } |j|d} t| ||f} t|j|ddd^}t|tr|t |j |j n|}|D]4}| || || |t || ||| |<qW5QRXt|| |||dS)z"Read chunk of vectorized raw data.r*r(r^rr) bufferingN)rhrkrCr=Zzerosr7rlr8sliceZarangervrwr?fromfiler)rawrsrtrurvrwrxryr*r(r_n_bytesr`r{rzZidsZch_idrrr!ris"ric Cst|d}|}W5QRX|dddd}t|dzNtd|ddtjtj @}d}|rx| d }|d krd }||}Wnt k r|d }YnXtd |tj}|st tt tt tdfS||d}td|}|r|d|}td|tj }ttt} } } d} |D]} | d}|dd\}}}}|dd}|dd}| dkrt|dkr|dkr|d} |rt|nd}| || t|d | |d|qHt | t | t | | fS)aRead annotations from a vmrk/amrk file. Parameters ---------- fname : str vmrk/amrk file to be read. Returns ------- onset : array, shape (n_annots,) The onsets in seconds. duration : array, shape (n_annots,) The onsets in seconds. description : array, shape (n_annots,) The description of each annotation. date_str : str The recording time as a string. Defaults to empty string if no recording time is found. rasciiignore rmarker Codepage=(.+)utf-8r#ANSIcp1252latin-1z\[Marker Infos\]Nz^\[.*\]$z ^Mk\d+=(.*)rc\1z New Segmentr/)r7readrnrqro_check_bv_versionresearch IGNORECASE MULTILINEgroupUnicodeDecodeErrorr=arraylistendrvfindallrprisdigitr;append)fnamerztxtheader cp_settingcodepagemZmk_txtitemsonsetduration descriptiondate_strr2Z info_datamtypeZmdescZ this_onsetZ this_durationrrr! _read_mrksT    $    "  rrc Cst|\}}}}t|}|dkrtt|dd}t|sNt|dd}td|t|\}}}} | d}tj |t d|}tj |t d|}t ||||d} | S) aCreate Annotations from BrainVision vmrk/amrk. This function reads a .vmrk or .amrk file and makes an :class:`mne.Annotations` object. Parameters ---------- fname : str | object The path to the .vmrk/.amrk file. sfreq : float | 'auto' The sampling frequency in the file. It's necessary as Annotations are expressed in seconds and vmrk/amrk files are in samples. If set to 'auto' then the sfreq is taken from the .vhdr/.ahdr file that has the same name (without file extension). So data.vmrk/amrk looks for sfreq in data.vhdr or, if it does not exist, in data.ahdr. Returns ------- annotations : instance of Annotations The annotations present in the file. rr.vhdrrz$Finding 'sfreq' from header file: %sr1r_)rrr orig_time) r_str_to_meas_dater3r5existsrr2rFr=rrfr) rr1rrrrrrPr[r2 annotationsrrr!_read_annotations_brainvision s  rcCsxd}d}|dkst|dkr,td|dtddD],}|||f}t||d k r6|Sq6t|||fd S) zCheck the header version.z MNE-Python currently only supports %s versions 1.0 and 2.0, got unparsable %r. Contact MNE-Python developers for support.zEBrain ?Vision( Core| V-Amp)? Data( Exchange)? %s File,? Version %s\.0)rrrzMissing header in z file.r#rN)AssertionErrorrr: capitalizerr)rkindZ _data_errZ_data_reversionZthis_rerrr!r9s  rrr$)Z MULTIPLEXEDZ VECTORIZEDshortr;r&)ZINT_16ZINT_32Z IEEE_FLOAT_32r)rr;r&zgMbP?g& .>r#) VµVuVmVnVr$u°Czn/auµSZuSZARUSNc Csr|}|dkrdSzt|d}Wn:tk r^}zdt|krLWY dSW5d}~XYnX|jtjd}|S)N)r0Z00000000000000000000z%Y%m%d%H%M%S%fzdoes not match format)tzinfo)rorstrptime ValueErrorrjrprutc)rr0errr!rbs  rc Cs:t|d}|}d}|dd}t|d|}zJtd|ddtjtj @}|rl| d}|dkrxd }||}Wnt k r|d }YnXW5QRX| d d kr| d \}}n |d }}t}t|}||W5QRXd} || sd} d|| d} t| } d| _||| | fS)zAux function for _get_hdr_info.rrrrrrr#rrrz [Comment]rrz Common Infosz Common infosg.AZSamplingIntervalF)r7r<rnrorrrrrrrrfindrq configparser ConfigParserr read_file has_sectiongetfloatr _unlocked) rPrVrrrYrparamsrZrzcinfostrr1r2rrr!rFvs>       rFcIs t|}t|d}|dk}|dkr2td|t|\}}}d|_|d} | tkrhtd| t| } |d} | d krd d } | t krtd | t | } n(| d krtdfdd dD} t |} t | |d} t | |d}d}t |d}|}W5QRX|D]6}t||}|r.|d}t||d<qnq.d|d<|d}|r|d7}d}| d krz|d}WnZtjk rtdt | d*}|dd|}|t| |}W5QRXYnXdg|}t|}t|}|tjt}t}t}d D]\}} t td!|dd}!| !d"} t"| d#kr| d$7} n| d%dkrd&| d%<| dd#\}"}#}$}%|"#d'd"}"|"||<|"||!<|$dkr|%sd(}$nd)}$|%#d*d}%|%||"<t|$||!<t$|%d|||!<|%d+krH|%d kr8t%j&nt%j'||"<qH|rtt(|t(<t(|d<d,|t(<d)|d<d|d<|d-krt)|*n|}d}&+d.rd%d/l,m-}'t)}(t)})t)}*d.D]}+||+d},|).|,d0d1|+d!d"D\}-}.}/t/|.}0t/|/}1|-dkr(|-t0d29}-|'t1|-|1|0ggd}2|2dk2rl|,t)||krl|*.|,|(.|2qt1|(d3}(t3tt4|)|(d4d5}&t"|*dkr||*7}td65|*t6|7rt8d7|9}d}3d8|kr@|:d8}3||3dd}d9\}4}5t;|D]$\}3}6t}8td?qn|7}3qv|3 rg}9g}:d@};dA||3krd}dC|=|5k}?t;|dD]\}@}+|r8|@t"|kr8|+t(kr8q|3|7krP||3|@}An ||7|@}A|A=|+dkspt>|8rt!|;|+}B|r&d)dtj@|dG|dG<nd}D|>rdHd1|:D}:tAtj1|:tjBdI|dG<d)dtj@|dG|dG<|dGdFkrt"t?|:dkrd>}DnLdJd1|:D}:tCtj1|:tjBdI|dG<|dGdFkrt"t?|:dkrd>}D|Dr&tdK|dGt"|9dkr8nt"t?|9dkr|9ddLkrZn0t|9d|dM<|? rd)dtj@|dM|dM<n*d}E|? r dNd1|9D}9tCtj1|9tjBdI|dM<zd)dtj@|dM|dM<Wn>tDk rt"t?|9dk r|dOdP|dM<d>}EnYnXn^dQd1|9D}9tAtj1|9tjBdI|dM<tE|dM r~|dOdP|dM<t"t?|9dk r~d>}E|E r|dM|dOdPk rdR}Fnd}FtdS|dM|FftFGdTg|dU<t;|D]\}3},|,|k s|3|k s|3||k rt%jH}Gt%jI}Ht%jJ}%nz|,|k s4|3|k s4|3||k r\t%jK}Gt%jI}H|,|k rT||,}%nt%j'}%n0|,dVk rzt%jL}Gt%jI}Ht%j'}%nt%jM}Gt%jN}Ht%jJ}%|dU.t|,|H|G|3d|3d||3||3tOdWtj|%t%jPt%jQdX  qd>|_|R|| | | |||&|fS)Ya4Extract all the information from the header file. Parameters ---------- hdr_fname : str Raw EEG header to be read. eog : list of str Names of channels that should be designated EOG channels. Names should correspond to the header file. misc : list or tuple of str | 'auto' Names of channels or list of indices that should be designated MISC channels. Values should correspond to the electrodes in the header file. If 'auto', units in header file are used for inferring misc channels. Default is ``'auto'``. scale : float The scaling factor for EEG data. Unless specified otherwise by header file, units are in microvolts. Default scale factor is 1. Returns ------- %(info_not_none)s data_fname : str Path to the binary data file. fmt : str The format of the binary data file. order : str Orientation of the binary data. n_samples : int Number of data points in the binary data file. mrk_fname : str Path to the marker file. montage : DigMontage Coordinates of the channels, if present in the header file. orig_units : dict Dictionary mapping channel names to their units as specified in the header file. Example: {'FC1': 'nV'} rr)rrzOThe header file must be given to read the data, not a file with extension '%s'.TZDataOrientationz$Data Orientation %s is not supportedZ DataFormatBINARYz Binary InfosZ BinaryFormatzDatatype %s is not supportedr$zdBrainVision files with ASCII data in vectorized order (i.e. channels in rows) are not supported yet.csi|]}|d|qS) ASCII Infos)get)rkeyrZrr! sz!_get_hdr_info..rZDataFileZ MarkerFilez-^Mk\d+=New Segment,.*,\d+,\d+,-?\d+,(\d{20})$rrr0NZNumberOfChannelsr#Z DataPointszQNo info on DataPoints found. Inferring number of samples from the data file size.rr rz Channel Infoszch(\d+)rcr)rrrrrrÂ)rrrrrrrZ Coordinates) _sph_to_cartcSsg|] }t|qSrre)rcrrr!r"Xsz!_get_hdr_info..g@@head)Zch_pos coord_framezNo coordinate information found for channels {}. Setting channel types to misc. To avoid this warning, set channel types explicitly.zMissing channel unitsZChannels)rrz#\s+NamezS o f t w a r e F i l t e r sz#\s+Low Cutoff)r#r Fz\Online software filter detected. Using software filter settings and ignoring hardware valuesz\s+zResolution / Unitz\s\s+z[s])NaNOffDChighpasscSs"g|]}|dkrt|ntjqS)rrrrfr=ZInfrZfiltrrr!r"srcSs g|]}|dkrt|ndqS)rrrerrrr!r"szgChannels contain different highpass filters. Lowest (weakest) filter setting (%0.2f Hz) will be stored.rrrlowpasscSs g|]}|dkrt|ndqS)rrrerrrr!r"sr1g@cSs"g|]}|dkrt|ntjqS)rrrrrr!r"sz, Nyquist limitziChannels contain different lowpass filters. Highest (weakest) filter setting (%0.2f Hz%s) will be stored.z!Setting channel info structure...ZchszSTI 014 ) ch_name coil_typerZlognoZscannocalr:locunitZunit_mulr)Srfr3r5IOErrorrFrr_orientation_dictr> _fmt_dictoptionsdirnamejoinr7 readlinesrrrorgetintr NoOptionErrorrr?rBrCr=rmfillnanr9rr;rqrrp _unit_dictr Z FIFF_UNIT_CELZFIFF_UNIT_NONErJrkeysrZ transformsrrZdeg2radrrallrzipformatisnananyrrrGindex enumeratematchrrsetpimaxZfloat64minZeroDivisionErrorisinfrr2Z FIFFV_EOG_CHZFIFFV_COIL_NONEZ FIFF_UNIT_VZ FIFFV_MISC_CHZ FIFFV_STIM_CHZ FIFFV_EEG_CHZFIFFV_COIL_EEGfullZFIFF_UNITM_NONEZFIFFV_COORD_HEADZ_update_redundant)IrPrMrNrOrQrRrYrr2r)Z data_formatr(pathrSrTregexpZ tmp_mrk_flinesr rrr%r*rzrZch_namesrxrangesZch_dictZmisc_chsr/Zchanpropsnnamer[ resolutionrrUrZ montage_posZ montage_namesZto_miscchrZradthetaphiZpolazposrtZhp_colZlp_colsettingZidx_ampZfilter_list_has_ch_namerrdividershiftrZhp_sZlp_siZline_ampZ ch_name_partsZ real_shiftZheterogeneous_hp_filterZheterogeneous_lp_filterZnyquistrrrrr!r6s'                                          "           ""      r6rFcCst||||||dS)aReader for Brain Vision EEG file. Parameters ---------- vhdr_fname : str Path to the EEG header file. eog : list or tuple of str Names of channels or list of indices that should be designated EOG channels. Values should correspond to the header file Default is ``('HEOGL', 'HEOGR', 'VEOGb')``. misc : list or tuple of str | 'auto' Names of channels or list of indices that should be designated MISC channels. Values should correspond to the electrodes in the header file. If 'auto', units in header file are used for inferring misc channels. Default is ``'auto'``. scale : float The scaling factor for EEG data. Unless specified otherwise by header file, units are in microvolts. Default scale factor is 1. %(preload)s %(verbose)s Returns ------- raw : instance of RawBrainVision A Raw object containing BrainVision data. See Also -------- mne.io.Raw : Documentation of attribute and methods. rLrMrNrOr-r)rrrrr!read_raw_brainvisionVs "rri)zEvent/z Stimulus/Sz Response/RzOptic/Oii)z New Segment/zSyncStatus/Sync Oni'Z AHDR_CHANNELcs eZdZdZfddZZS)_BVEventParserzDParse standard brainvision events, accounting for non-standard ones.cslt}|dd}|dd}|rB||krBt|||}n&|tkrTt|}ntt|j|td}|S)z=Parse BrainVision event codes (like `Stimulus/S 11`) to ints.N)offset) _BV_EVENT_IO_OFFSETSrorr;_OTHER_ACCEPTED_MARKERSrDr__call__ _OTHER_OFFSET)rKrr'Z maybe_digitrcoder\rr!r$s   z_BVEventParser.__call__)r}r~rrr$rrrr\r!rsrcCs<tdd|D}tttt}t||dkS)NcSsg|]}|dqS)z 0123456789 )rstrip)rddrrr!r"sz#_check_bv_annot..r)rr"runionr#r)Z descriptionsZmarkers_basenameZ bv_markersrrr!_check_bv_annots   r*cCst|}dd|D}t}t|dkr||d}|d}|ddd}d}|dk rdd|d d D} |j| d| d| d dd }||ddD]} t d | } | r| d } | } t d| d}t|rt |dnt j|d}|dk r|d|i| dkr@d|kr@||dn<| dkrdd|krd||dnd|kr|||d||| <qqq|S)a{Parse impedances from the header file. Parameters ---------- settings : list The header settings lines from the VHDR/AHDR file. recording_date : datetime.datetime | None The date of the recording as extracted from the VMRK/AMRK file. Returns ------- impedances : dict A dictionary of all electrodes and their impedances. cSs$g|]}|dr|dr|qS)z Impedance [z :) startswithendswithrrrrr!r"s  z$_parse_impedance..rr#[]NcSsg|] }t|qSr)r;r-rrr!r"sr:r )hourminutesecond microsecondz[ a-zA-Z0-9_+-]+:z[-+]?\d*\.\d+|\d+r)impZimp_unitZ imp_meas_timeZRef ReferenceZGndGroundData)_parse_impedance_rangesr9rrrqlstripr'rprrrrrfr=rupdate)rYZrecording_daterZimpedance_setting_linesrIrtZimpedance_settingZimpedance_unitZimpedance_timeZ meas_timerrZ channel_nameZchannel_imp_lineZ imp_as_numberZchannel_impedancerrr!rHsL      rHcCsdd|D}t}|rt|dkr`|d}dD](}t|dt|d|dd ||<q4n:|D]4}|}t|d t|d |d d ||d<qd|S) a/Parse the selected electrode impedance ranges from the header. Parameters ---------- settings : list The header settings lines from the VHDR/AHDR file. Returns ------- electrode_imp_ranges : dict A dictionary of impedance ranges for each type of electrode. cSsg|]}d|kr|qS)z$Selected Impedance Measurement Ranger)ritemrrr!r"sz+_parse_impedance_ranges..r#r)r8r6r7r)Zimp_lower_boundZimp_upper_boundZimp_range_unit )r9rrqrf)rYZimpedance_rangesZelectrode_imp_rangesZ img_rangeZelectrode_typeZelectrode_rangerrr!r9s"       r9)r)rrrFN)N)9rrr@os.pathr r3rrriornumpyr=utilsrrrr r constantsr Z meas_infor baserrrrrrZchannelsrdefaultsrrrirrrr9rrrCrkrrrFr6rr"r#r%rJrr*rHr9rrrr!s        \ , 7 + & >