U f{j~@svdZddlmZddlZddlZddlmZddlZddl m Z ddl Z ddl mZmZddlmZmZmZmZmZmZdd lmZdd lmZdd lmZmZmZdd lm Z dd l!m"Z"m#Z#ddl$m%Z%ddl!m&Z&m'Z'm(Z(m)Z)m*Z*ddl+m,Z,dZ-ddZ.Gdddej/Z0ddZ1ddZ2ddZ3ddZ4e&d,d!d"Z5Gd#d$d$eZ6e&d-d%d&Z7d.d'd(Z8d/d*d+Z9dS)0zEGI NetStation Load Function.) OrderedDictN)parse) _read_events_combine_triggers)_get_signalfname _get_ep_info_extract _get_blocks _get_gains_block_r)BaseRaw)FIFF) _empty_info create_info_ensure_meas_date_none_or_dt) setup_proj) _create_chs _mult_cal_one) Annotations)verboseloggerwarn _check_option _check_fname) EvokedArray)ZVREFzVertex Referencec6 Cs$t|}|dd}|dd}t|d}ddg}t||d}d}t|dr\|dd }t||}t|} t|} t||d } | | |dd } t d | } | d krt d | f| d }t|dkrt dd|Dstt|dkrdnd}dD]}t| |D]6\}}||d kr6t d|f||| ||<qt| |tj| |<| d}d}t||}||}||}tjdd| ||9<W5QRX| ||<| |tj| |<q| d}| d| d}||kp>| d| dk p>| ddd | dd dk }|rft dt| dt| df| | t| ddd}d }t| d| dD]2\}}||}t||||||<||7}q|| d<t|d }t|}|d!d jj | d"<|d#} t}!t}"d }#t}$| D]f}%t!|%d$d jj }&|&d%kr$|%d&d jj }'|'"}'|$#|'|!#d'|"#d(|#d}#q$|#| d)krt d*|#| d)fg}(d+|krt||d+d})t|)}*|*d}+| d},t$|+d d|,d do|+d|,dtd,k}-|-s>t d-t|+t|,ft|d.}.t|.}/|/d#} g}0g}1| D]~}%|%d!d jj }2|%d/d j}3d}4|3d k r|3j }4|2d0krd1}5nd2|2krd3}5nd4}5|0#|5|1#|4|(#|2qh| j|0|1|d+d|*d5| j|(||dd |!|"|$d6| S)7zRead mff header.EEGsignalinfozinfo.xmlZ mffVersionZ recordTime)filepathr) eeg_fname info_fnamezD\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}.(\d{6}(?:\d{3})?)[+-]\d{2}:\d{2}NzCould not parse recordTime %r) css|]}|VqdSN) isnumeric).0fr+P/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/egi/egimff.py 9sz#_read_mff_header..r%ii@B) last_samps first_sampszCould not parse epoch time %ssfreqraise)Zover samples_blockr.r/rz5EGI epoch first/last samps could not be parsed: %s %s disk_sampszsensorLayout.xmlnamedevicesensortype)rrnumbereeguV n_channelszHNumber of defined channels (%d) did not match the expected channels (%d)ZPNSr z/PNS and signals samples did not match: %s vs %sz pnsSet.xmlunitZECGecgZEMGemgZbio) pns_types pns_units pns_fnamepns_sample_blocks) pns_namesversiondate chan_type chan_unitnumbers)%ropjoinr lenr rdictupdaterematch RuntimeErrorgroupsallAssertionError enumeratenparrayZuint64mathgcdZerrstateZastypeZint64sumlistfullziparangergetElementsByTagName firstChilddataintencodeappend array_equal)6r! all_filesZeeg_fileZ eeg_info_fileZ info_filepathtagsZversion_and_daterEfnameZ signal_blocksepochsZ summaryinfoZ record_timegfracdivkeyeie numerator denominatorZthis_gcdZ n_samps_blockZn_samps_epochsbadr4offsetfirstlastZn_thisZsensor_layout_fileZsensor_layout_objsensorsrGrHZn_chansrIr7Z sensortypeZsnrDZ pns_fpathZ pns_blocks pns_samplesZsignal_samplesZ same_blocksZpns_fileZpns_objr@rAr5Z unit_elemr=ch_typer+r+r,_read_mff_headers          "    $                        ryc@s0eZdZdZddZddZddZdd Zd S) _FixedOffsetz`Fixed offset in minutes east from UTC. Adapted from the official Python documentation. cCstj|d|_dS)N)minutes)datetime timedelta_offset)selfrsr+r+r,__init__sz_FixedOffset.__init__cCs|jSr')r~rdtr+r+r, utcoffsetsz_FixedOffset.utcoffsetcCsdS)NZMFFr+rr+r+r,tznamesz_FixedOffset.tznamecCs tdS)Nr)r|r}rr+r+r,dstsz_FixedOffset.dstN)__name__ __module__ __qualname____doc__rrrrr+r+r+r,rzs rzc Cst|}t|dd}t|tjdd}W5QRXt|ddkrv|ddd|dd dg\}}|||d<tj|dd }t||| d dddd }|j dddgggdd | ||S)zObtain the headers from the file package mff. Parameters ---------- input_fname : str Path for the file Returns ------- info : dict Main headers set. z /signal1.binrbrrrF Niz%Y-%m-%dT%H:%M:%S.%f%zz%z)rE meas_dt_local utc_offsetZgainbits value_range)Z n_categoriesZ n_segmentsn_events event_codesZcategory_namesZcategory_lengthsZ pre_baseline) ryopenrVfromfileZint32rLr|strptimerMstrftimerN) input_fnameZmff_hdrfidrErtzZtime_nr r+r+r, _read_headers2  $   rcstt||d}|ddkrP|ddkrP|dd|dgt|d}n ddd fd d |d D}d |kr||d 9}|S)z,Calculate calibration info for EEG channels.r$rrrr rGư>r)r;Vcsg|] }|qSr+r+)r)tZ cal_scalesr+r, sz-_get_eeg_calibration_info..rHZgcal)r rJrKrL)r!egi_infoZgainscalsr+rr,_get_eeg_calibration_infos   rcspddlm}t|d}t|sNtdfddt|dD}|dfSd d d d }t |d }t |}| d} t } t } t} t }| D] ddj}  ddj}| dkrЈt|jn| j}|j}fdddD}t |d}||kr|| ||<qt||k}t|dkr@| |q|||| |<q|f| | d| }||fS)zRead channel locations.r)make_dig_montagezcoordinates.xmlz=File coordinates.xml not found, not setting channel locationscsg|]}|dqS)rr+r)i)channel_namingr+r,r sz_read_locs..r<NZlpaZrpaZnasion)zLeft periauricular pointzRight periauricular pointZNasionrIr7r5rr9cs"g|]}t|djjqS)r)floatr_r`ra)r)Zcoord)r7r+r,r sZxyzd)ch_poshsp)Zchannels.montagerrJrKexistsrrrangerVrWrr_rr[rMr`rbrarc flatnonzerorLrd)r!rrrrhch_namesZ dig_ident_maprIZ coordinatesrvrrZnlrZ name_elementZ num_elementr5nrZcoordslocZid_monr+)rr7r, _read_locssR            rc Cst|dD]v\}}||}|d|}tjtjd}||tj}tj} d} |d|dkrjtj} d} ||j | |tj | dq |S) z/Add info for PNS channels to channel info dict.rDr@)r>r?rrAr;g?)calkind coil_typer=) rUindexr FIFFV_ECG_CH FIFFV_EMG_CHget FIFFV_BIO_CHZ FIFF_UNIT_VFIFF_UNIT_NONErNFIFFV_COIL_NONE) chsrrZi_chch_nameidxrxZtype_to_kind_mapch_kindZch_unitZch_calr+r+r,_add_pns_channel_info4s&  rFE%dc Cst||||||||S)a-Read EGI mff binary as raw object. .. note:: This function attempts to create a synthetic trigger channel. See notes below. Parameters ---------- input_fname : str Path to the raw file. eog : list or tuple Names of channels or list of indices that should be designated EOG channels. Default is None. misc : list or tuple Names of channels or list of indices that should be designated MISC channels. Default is None. include : None | list The event channels to be ignored when creating the synthetic trigger. Defaults to None. Note. Overrides `exclude` parameter. exclude : None | list The event channels to be ignored when creating the synthetic trigger. Defaults to None. If None, channels that have more than one event and the ``sync`` and ``TREV`` channels will be ignored. %(preload)s channel_naming : str Channel naming convention for the data channels. Defaults to 'E%%d' (resulting in channel names 'E1', 'E2', 'E3'...). The effective default prior to 0.14.0 was 'EEG %%03d'. %(verbose)s Returns ------- raw : instance of RawMff A Raw object containing EGI mff data. Notes ----- The trigger channel names are based on the arbitrary user dependent event codes used. However this function will attempt to generate a synthetic trigger channel named ``STI 014`` in accordance with the general Neuromag / MNE naming pattern. The event_id assignment equals ``np.arange(n_events) + 1``. The resulting ``event_id`` mapping is stored as attribute to the resulting raw object but will be ignored when saving to a fiff. Note. The trigger channel is artificially constructed based on timestamps received by the Netstation. As a consequence, triggers have only short durations. This step will fail if events are not mutually exclusive. See Also -------- mne.io.Raw : Documentation of attribute and methods. .. versionadded:: 0.15.0 )RawMff)reogmiscincludeexcludepreloadrrr+r+r,_read_raw_egi_mffHs = rcs.eZdZdZed fdd ZddZZS) rz RawMff class.NFrc + st|ddddd}td|t|} |dkr4g}|dkr\tt| ddkd }td t|| \} } t || } td | d d krt | d dkr|dkrddgn|fddt Dg} |dkrt | D]&\} }| dkr| r| | qtt| tkr:tdtn | fddt| d Dfddt Dnfddt Ddfd|ffD]V\}}t|t r|D]"}|krtdt|qn|dk rtd|qtdtddfddt Dttd}t| |d| d <ttfd!dD||_| d dk rt| | d g} | jd| d"d#kstnd|_d| d <g| d$t j!j"}t#| d%}t$||d&<| d'|d'<t| d(d)|d*<t%|| |\}}|t | d | d dk rH| d+t&| t'dtdt|t|g} || d,t&| t'dt| d,g} t(j)}t(j*}t+|| |||d-d-|}fd.dt |D}|D],|,t(j-| t(j.t(j/t(j0d/qt1|| |}||d0<d1|_2|3|dk rF|j4|d2d3t5t6|j7t8}t|r|d0t9|d4dd5}|d0D]<}|d6t(j*k}|d7t8k}|r|r||d4d5d8<q|:t(j;|d9<W5QRXt<|| d:} | | d;<d<}!ttd=d|Dd d<td>d|Dd d?<td@d|Dd dA<t=t&fdBd|!Dtt|stdCd g| dD<|!D]}"| dD t|"qt>| dD| dD<| dDd d kst| dDd#|dEkstd g}#| d"d#dg}$tt t t dF}%tdArt<|| dG| dH<t | dIdJ}&t | dJ}'|&|'dkrtdK|%dL |$d#| d%|%dM d| d%|%dN dOn|&|'krt?dP|&|'f| g|_@| g|_AtBtC|jD||dQ| g|#|$| g|dRt| dSdd| d"dd#D]d\}(})|(|)}*|*d ksNt|*r0|%dL |)dT| d%|%dM |*| d%|%dN dUq0t|%dLr|EtFf|%dVdidS)WzInit the RawMff class.readTr)need_dirz!Reading EGI MFF Header from %s...NrGr:rz Reading events ...z# Assembling measurement info ...rrsyncZTREVcsg|]\}}|kr|qSr+r+r)rk) exclude_listr+r,rs z#RawMff.__init__..rz5Did not find any event code with more than one event.csg|]}|kr|qSr+r+r) exclude_indsr+r,rscsg|]\}}|kr|qSr+r+rinclude_r+r,rscsg|]\}}|kr|qSr+r+r)rr+r,rsrrzCould not find event named z!`%s` must be None or of type listz. Synthesizing trigger channel "STI 014" ...z Excluding events {%s} ...z, csg|]\}}|kr|qSr+r+rrr+r,rs)Z remappingZ new_triggercsg|]}|kr|qSr+r+)r)ro) include_namesr+r,rs r.r3rr0Z meas_daterr6r8 device_infozSTI 014rDr+cs&g|]\}}|ds|kr|qS)ZSTI) startswith)r)rr5)rr+r,rs )Zunit_mulrrrr=rFignoreZ on_missingrrrrr%custom_ref_appliedr# egi_events)r:stipnscSsg|]}|dtjkqSr)r FIFFV_EEG_CHr)chr+r+r,rscSsg|]}|dtjkqSr)r FIFFV_STIM_CHrr+r+r,rsrcSs$g|]}|dtjtjtjfkqSr)rrrrrr+r+r,rsrcsg|] }|qSr+r+)r)rm)rr+r,rsz:Currently interlacing EEG and PNS channelsis not supported kind_boundsnchan)onsetduration descriptionrB pns_filepathrCr2z$This file has the EGI PSG sample bugrrrZ BAD_EGI_PSGz/PNS samples (%d) did not match EEG samples (%d)Zsingle)rZ orig_format filenamesr/r.Z raw_extrasrr/g?Z BAD_ACQ_SKIPZ orig_time)Grrr rrVwhererWtolistrrr[rUrZrdrLrRuntimeWarningextendsortr^ isinstance ValueErrorreprrKrrMr]Zevent_idvstackshaperT astimezoner|timezoneutcrrr concatenaterepeatrFIFFV_COIL_EEGrrrNZFIFF_UNITM_NONErrrrZ _unlockedZ_update_redundant set_montagerZin1drREFERENCE_NAMESrb_unlockZFIFFV_MNE_CUSTOM_REF_ONrJrecumsumrQ _filenames _raw_extrassuperrrZset_annotationsr)+rrrrrrrrrrrrZ more_excludesiieventkkvrZ events_idsZ meas_dt_utcr rrch_coilrrZ sti_ch_idxZref_idxZ ref_coordsZchanZis_eegZ is_not_refZfile_binkeysrmr/r.ZannotrwZ eeg_samplesrtZ prev_lastgap __class__)rrrrrrrr,rsz                   &                 zRawMff.__init__c* Cstd|dd|ddd}|j|} t| dd||f} | d} | d } | d} t|trvt|j|j }t || d kd }||tj f}||}t || d k|| d k@d }||}||| d }t || d k|| d k@d }||| d }||tj f}~~~| d|||f| |ddf<| d||}t |dkd }t |st || |||dS||d }||dd }t |||kstt| }t||k}||d kr||d nd }||}t|j|dd dx}d }d}d } ||krRt|}!|!dk r6|!}||dd |d 7}q|d kr|td|t|}!|d 7}|!dk r|!}|d|d}"t|||"}#|#j| ddd}#|#jd }$td|$dtd|d|d kr|#dd|df}#|$|}$d }||$kr2|#ddd|f}#|}$td|$d| }%|%|$}&|#|| |||%|&f<||$}| |$} qRW5QRXt |d krx| d}'| d}(|(d} |(d } t| }t||k}||d kr||d nd }||}t|'dd dp}|d d |})|d d }d}d } ||krjt|}!|!dk rN|!}||dd |d 7}q.|d krn|d kr||)krd | |df<qnt|}!|!dk r|!}|d|d}"t|||"}#|#j| ddd}#|#jd }$|d kr|#dd|df}#|$|}$d }||$kr6|#ddd|f}#|}$| }%|%|$}&|#|| |||%|&f<||$}| |$} qjW5QRXt || |||dS)zRead a chunk of data.z Reading MFF Z6dz ...  ...z.r) _import_mffpyendswithrReaderZ mff_flavorAttributeErrorflavor directorylistdir categoriesr[rrrr rL) rh conditionrrrmffpymffr"Z return_listr%outputr+rr,read_evokeds_mffs04     r*c Csddl}t|}||}|jj}t|trHtd||dd}|j|} nZt|trz|j|} Wn0t k rt d|dt |jdYnX| j }nt d || } | d \} } d | kr| d \} } t| | f}|d |d }n | }|d }|d9}t|||\}}||dt||jd |}|*t|dd|d<t|j|d<W5QRXt||}t|tdt |dg}tj}tj}t ||||dddd}t!|||}|||d<W5QRX|dk r|j"|dd||d<z||dd}Wn.t#k r<t$d|dt%dd}YnXg}|r|D]z}|ddkrL|d dkr|d!D]}|&||dqtn4|d d"krL|d!D]}|&|d|dqqL||d#<|d$|d%<W5QRXz|j'(d&}Wnt t)fk rYn^X||j*+|}W5QRX|j,D]8}|d'd(kr:d)|d*krht-|\}}nd+|d%<q:z||dd,d-d.}Wn.t#k rt$d/|d0t%dd}YnXt.||d1|||d2S)3zRead evoked data from MFF file.rNr&zprovided as category name)extraz"condition" parameter (zB), provided as epoch index, is out of range for available epochs (z).z0"condition" parameter must be either int or str.rZPNSDatarGr@rrDr6rrrrr+rrrrZ channelStatusz,Channel status data not found for condition z$. No channels will be marked as bad.)categoryZ exclusionZ badChannelsZ signalBinZchannelsr badsFrhistorymethodzMontage Operations ToolzAverage ReferencesettingsTrz#segraz2Number of averaged epochs not found for condition z. nave will default to 1.g)Ztmincommentnaver)/r'rr r%rstrrrirb IndexErrorrrLr5 TypeErrorZget_physical_samples_from_epochrVrrrrZsampling_ratesrrMrZZ num_channelsvaluesrrrrrrrrrKeyErrorr UserWarningrdr#Z filepointerFileNotFoundErrorXML from_fileentriesrr)rhr&rrr'rr(r%r,epochraZeeg_datat0Zpns_dataZall_dataZch_typesrrr rrrrZchannel_statusr-entryrfpr.Z projectorr2r+r+r,rMs    "                  rread averaged .mff filesc CsLz ddl}Wn:tk rF}zd|d|}t|W5d}~XYnX|S)zImport and return module mffpy.rNzmffpy is required to z, got: )r' ImportError)whyr'expmsgr+r+r,rs  r)NNNNFrN)NrNN)rN)rA):r collectionsrr|rXos.pathpathrJrOZxml.dom.minidomrnumpyrVeventsrrZgeneralrrr r r r baser constantsrZ meas_inforrrZprojrutilsrr annotationsrrrrrrZevokedrrrytzinforzrrrrrrr*rrr+r+r+r,sV         .1 @s R t