U f{j-@sddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZdd lmZd d l mZmZmZmZmZd d ZddZedddZGddde ZdS)N)_read_raw_egi_mff)_combine_triggers)BaseRaw)_read_segments_file _create_chs) _empty_info)FIFF)verboseloggerwarn_validate_type _check_fnamecCst|ddd}|dt|d@kr:|tj}ntddd}t|||dd||dd||dd||dd||dd||dd||d d||dd||dd||dd||dd||ddd }t|ddkrdnd}t|d}|dkrtd |r| tddt|d ddt|dddgggdd t |d D].}d t|ddd}|d |qNnt d||d<dddd|\|d<|d<t|d|d<|S)zRead EGI binary header.z.my_fread>i2z>i4) versionyearmonthdayhourminutesecondZ millisecond samp_rate n_channelsZgainbits value_rangez&Floating point precision is undefined.)Z n_categories n_segments n_samplesn_events event_codesZcategory_namesZcategory_lengthsZ pre_baseliner'ZS1ZU1r(z#Only continuous files are supported unsegmented)rshort)z>f4float)z>f8double)rr*rdtype orig_format)rrZ bitwise_andbyteswapZastypeZuint32 ValueErrordict RuntimeErrorupdaterangejoinappendNotImplementedErrorr/)fidrrinfor+ precisioneventr(rrr _read_headers^              r>cCst|d|d|dg}|d|dddt|dD]@}||d|djd t||d|d|d d |f<qB|S) z Read events.r'r%r&$r*rr"r/rN)rZzerosseekr6itemsizer)r:r;eventssirrr _read_eventsLs $rDFE%dc CsPt|ddt|}|ddr:t||||||||St||||||||S)aRead EGI simple binary as raw object. .. note:: This function attempts to create a synthetic trigger channel. See the Notes section below. Parameters ---------- input_fname : path-like Path to the raw file. Files with an extension .mff are automatically considered to be EGI's native MFF format files. 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 .. versionadded:: 0.11 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'. .. versionadded:: 0.14.0 %(verbose)s Returns ------- raw : instance of RawEGI A Raw object containing EGI data. See Also -------- mne.io.Raw : Documentation of attribute and methods. 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. z path-like input_fnamez/\z.mff)rstrrstripendswithrRawEGI)rFeogmiscincludeexcludepreloadchannel_namingr rrr read_raw_egiYs@  rQcs.eZdZdZed fdd ZddZZS) rJz'Raw object from EGI simple binary file.NFrEc st|ddd}|dkrg}|dkr&g}t|db} td|t| } tdt| | } | ddkr| d dkr| dd | d } nd } W5QRXtd g| d dkrt| ddkr|dkrddgn|fddtDg} |dkr8t| D],\}}|dkr |r | |q t t | t kr^t dt n  | fddt| d DfddtDnfddtDdfd|ffD]P\}}t|tr|D]}|krtd|qn|dk rtd|qtt d}tdtddfddtDt| |d | d!<ttfd"dD||_nd|_d| d!<t| d#}t| d$| d%| d&| d'| d(| d)}t|}|df|d*<fd+dt| d,D}| t| d| d!dk r:| d-t |}t| |}tj }tj!}t"|||||d.d.|}fd/dt|D}|D](}||#tj$d0tj%tj&tj'd1q||d2<d3|_(|)| d4d5kr| d4nd6}t*t+|j,||||g| d7dg| g|d8dS)9NreadTrFrbzReading EGI header from %s...z Reading events ...r$rr#g@gư>z# Assembling measurement info ...r'r(syncZTREVcsg|]\}}|kr|qSrr.0ik) exclude_listrr s z#RawEGI.__init__..rz5Did not find any event code with more than one event.csg|]}|kr|qSrrrVrW) exclude_indsrrrZscsg|]\}}|kr|qSrrrUinclude_rrrZscsg|]\}}|kr|qSrrrU)rMrrrZsrMrNzCould find event named "%s"z!`%s` must be None or of type listz. Synthesizing trigger channel "STI 014" ...z Excluding events {%s} ...z, csg|]\}}|kr|qSrrrUr]rrrZs)Z remapping new_triggercsg|]}|kr|qSrr)rVe) include_namesrrrZs r!rrrrrr Z meas_datecsg|]}|dqS)rrr[)rPrrrZsr"zSTI 014rcs&g|]\}}|ds|kr|qS)ZSTI) startswith)rVrWname)r(rrrZs g?)Zunit_mulcalkindZ coil_typeunitchsFr0r-Zsingler&)r0 filenamesZ last_sampsZ raw_extrasr )-ropenr r;r>rDlist enumeratesumr8lenrRuntimeWarningextendsortrZarange isinstancer2r7rr3zipZevent_idr datetimetimemktime timetupler6repeatr ZFIFFV_COIL_EEGZ FIFFV_EEG_CHrr5ZFIFF_UNITM_NONEZ FIFFV_STIM_CHZFIFFV_COIL_NONEZFIFF_UNIT_NONEZ _unlockedZ_update_redundantsuperrJ__init__) selfrFrKrLrMrNrOrPr r:egi_infoZ egi_eventsrdZ more_excludesiir=kkvrXZ event_idsr;Zmy_timeZ my_timestampZch_namesZnchancalsZch_coilZch_kindrgZ sti_ch_idxidxr0 __class__)rPr(r\rYrMr^rarrys                        zRawEGI.__init__c Cs^|j|}|d} |d|d} d|dd} |d} t||||||||| | | | d dS) z#Read a segment of data from a file.r/r"r'r?r*r_)r/r"offset trigger_chN)Z _raw_extrasr) rzdatarfistartstoprZmultr{r/Z n_chan_readrrrrr_read_segment_file s zRawEGI._read_segment_file)NNNNFrEN)__name__ __module__ __qualname____doc__r ryr __classcell__rrrrrJsfrJ)NNNNFrEN)rsrtnumpyrZegimffrrBrbaserutilsrrZ meas_infor constantsr r r rrrr>rDrQrJrrrrs(     8  H