U f{jgE@sddlmZddlmZmZddlmZddlZddlm Z m Z m Z m Z m Z ddlmZdd lmZdd lmZdd lmZd d Ze dZddZdddddddddg ZddZddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5gZed6d7ed8d9D7Zed:d7ed8d;D7Zedd?d@dAg7ZedBd7ed;dCD7ZedDdEdFdGg7ZedHd7edIdJD7ZedKdLg7ZdMZdNdOZdPdQZdRdSZdTdUZdVdWZ e GdXdYdYeZ!dS)[) OrderedDict)datetimetimezone)PathN)fill_docloggerverbosewarn _check_fname)BaseRaw) create_info) Annotations) _mult_cal_onecCs|}t|tst|}|SN) isinstancer)fnameoutrQ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/nihon/nihon.py _ensure_paths rFcCs t|||S)aReader for an Nihon Kohden EEG file. Parameters ---------- fname : str Path to the Nihon Kohden data file (``.EEG``). preload : bool If True, all data are loaded at initialization. %(verbose)s Returns ------- raw : instance of RawNihon A Raw object containing Nihon Kohden data. See Also -------- mne.io.Raw : Documentation of attribute and methods. )RawNihon)rpreloadr rrrread_raw_nihonsrzEEG-1100A V01.00zEEG-1100B V01.00zEEG-1100C V01.00zQI-403A V01.00zQI-403A V02.00zEEG-2100 V01.00zEEG-2100 V02.00zDAE-2100D V01.30zDAE-2100D V02.00c Csi}t|}|d}|s*td|Stdt|d}t|dd dd}|t krpt d |d ||d <| d t|d d dd}t |d}|jtjd}||d<W5QRX|S)Nz.PNTz*No PNT file exists. Metadata will be blankz!Found PNT file, reading metadata.r|S16U16rz#Not a valid Nihon Kohden PNT file ()version@z|S14ZU14z %Y%m%d%H%M%S)tzinfo meas_date)r with_suffixexistsr rinfoopennpfromfileastype_valid_headers ValueErrorseekrstrptimereplacerutc)rmetadataZ pnt_fnamefidr Zmeas_strr#rrr_read_nihon_metadata?s$     r3ZFP1ZFP2ZF3ZF4ZC3ZC4ZP3ZP4ZO1ZO2ZF7ZF8ZT3ZT4ZT5ZT6ZFZZCZZPZEZPG1ZPG2A1A2ZT1ZT2cCsg|]}d|qSXr.0irrr \sr<r cCsg|]}d|qSNArr9rrrr<]scCsg|]}d|dqS)DCZ02rr9rrrr<^s!ZBN1ZBN2ZMark1ZMark2cCsg|]}d|qSr>rr9rrrr<`szX12/BP1zX13/BP2zX14/BP3zX15/BP4cCsg|]}d|qSr7rr9rrrr<bsZNA28Z)zutf-8latin1c Cst|}|d}ddtD}|rtdtD]}zt|d|d}d}|D]}|drd |kstd |krzd }qd}qV|d krV| d \}}t |}|t |kr|t |d } | dg| | ||<qVW5QRXWntk rYq8Xqq8tdtd|S)Nz.21EcSsg|]}|qSrrr:xrrrr<ksz"_read_21e_file..z&Found 21E file, reading channel names.r)encodingF[Z ELECTRODEZ REFERENCET=rZUNKz$Could not decode 21E file as one of z#; Default channel names are chosen.)rr$_default_chan_labelsr%rr& _encodingsr' startswithsplitintlenextendstripUnicodeDecodeErrorr ) rZe_fname _chan_labelsencr2Z keep_parsinglineidxnamenrrr_read_21e_filehs4     r\c Cs:t|}t|}i}td|t|d}t|dddd}|tkr`t d || dt|dddd}|tkrt d || d t|tj dd}|dkrt d ||d <| d t|tj dd}||d<g}t |D]} i} | d| dt|tjdd} | | d<| | dt|tj dd} | | d<g| d<t | D]4} i}| | | ddt|tjdd}||d<| |dt|tj dd}||d<g}t |D]>}| |d|dt|tj dd}|||q||d<| |dt|tjdd}||d<| |dt|tjddd@}||d<t||d|d <| d|qr|| q||d!<W5QRXg}g}g}|d!D]@}|dD]0}||d||d||d qqt dt|D]:}|d||krLt d"|d||kr.t d#q.|d|d$<|d|d<t||d <|ddkrtd%|d!dddkr6|d!dd}|d}|ddD]T}|d|dkrt d&|d|dkrt d'|d|dkrt d(q|S))NzReading header from rrrrrz&Not a valid Nihon Kohden EEG file ({})z4Not a valid Nihon Kohden EEG file (control block {})iz2Not a valid Nihon Kohden EEG file (waveform block)r n_ctlblocksaddress n_datablocks datablocks& n_channels' ZchannelsrCdurationi?sfreq n_samples controlblocksz(Channel names in datablocks do not matchz+Sample frequency in datablocks do not matchch_nameszLI dont know how to read more than one control block for this type of file :(zGCannot read NK file with different number of channels in each datablockz=Cannot read NK file with different channels in each datablockz:Cannot read NK file with different sfreq in each datablock)rr\rr&r'r(r)r*r+r,formatr-uint8rangeuint32appendZuint16rQrRsumNotImplementedError)rrVheaderr2r Z control_blockZ waveform_signr_roZ i_ctl_blockZt_controlblockZ t_ctl_addressrdZ i_data_blockZ t_datablockZt_data_addressZ t_n_channelsZ t_channelsi_chZt_idxZt_record_durationrmZchansZsfreqsZnsamplesZt_ctlZt_dtbZi_elemreZblock_0t_blockrrr_read_nihon_headers         r{c Cst|}|d}|s0tdtgggdStdt|d}t |dd dd }|t krvt d ||d t |tjdd }g}g}t|D]}|d |d t |tjdd }||dt |tjdd } ||d t |d| } | D]} tD]2} z| | } Wntk rBYnXq`qtdtq| dd d} t| d dd}|jd|jd|j}|||| qqt|d gt||d}W5QRX|S)Nz.LOGz0No LOG file exists. Annotations will not be read)onsetrk descriptionzFound LOG file, reading events.rrrrrz&Not a valid Nihon Kohden LOG file ({})r^r`rarfz|S45zCould not decode log as one of rlz%H%M%Si<)rr$r%r dictrr&r'r(r)r*r+r,rqr-rrrsrtrNdecoderUrTrr.hourminutesecondrurR)rZ log_fnamer2r Z n_logblocksZ all_onsetsZall_descriptionsrzZ t_blk_addressZn_logsZt_logsZt_logrWZt_descZt_onsetannotsrrr_read_nihon_annotationssZ      rcsBtdddg}|D]&\}}tfdd|Dr|SqdS)N)Zstim)ZMark)misc)rAr?rF$)Zbior7c3s|]}|kVqdSrr)r:kindch_namerr 1sz"_map_ch_to_type..Zeeg)ritemsany)rZch_type_patternkeykindsrrr_map_ch_to_type,s rc Csd}d}d}d}|tkrNt|}|dks:|dkrN|dkrNd}d }d }||}|d }|||}t||||||d } | S) NgMbP?g3333sqgzGq@i*I)LMgư>ig@i)unitphys_minphys_maxdig_mincaloffset)upperrMindexr) rZ unit_multrrrrYt_rangerrrrrr_map_ch_to_specs6s&  rcs.eZdZdZedfdd ZddZZS) ra>Raw object from a Nihon Kohden EEG file. Parameters ---------- fname : str Path to the Nihon Kohden data file (.eeg). preload : bool If True, all data are loaded at initialization. %(verbose)s See Also -------- mne.io.Raw : Documentation of attribute and methods. FNc st|ddd}t|}|j}td|t|}t|}|d}|d}dd|D} t||| } |d } d |kr| |d | d <W5QRXd d | dDt fd d| dDt ddt j f} t fdd| dDt ddt j f} t fdd| dDt ddt j f}t | | ||d}||_t| dD]F\}}|d|d}|| d|d<d|| d|d<qLtt|j| || df|gd|gdt|}|jdd}d}|ddkrzt|ddD]}|d|}||d }|d|d}|d||dd |d!d"|d||dd |d!d#qtf|d$| d i}||dS)%NreadTrz Loading %srmrpcSsg|] }t|qSr)rrHrrrr<isz%RawNihon.__init__..rnr#cSsi|]}|t|qSr)rrHrrr qsz%RawNihon.__init__..csg|]}|dqS)rrrHchsrrr<tscsg|]}|dqS)rrrHrrrr<vscsg|]}|dqS)rrrHrrrr<xs)roffsetsgainsrxrrrrsrrshort)rZ last_samps filenamesZ orig_format raw_extrasrorrdreg?r|rkgr}z BAD boundaryz EDGE boundaryZ orig_time)r rrZrr&r{r3rZ_unlockr(arrayfloatZnewaxisrZ_header enumeratesuperr__init__as_posixrrsrurZset_annotations)selfrrr Z data_namerxr1rmrpZch_typesr&rnrrrrryrrrZ controlblockZ cur_sampleZi_blockrzZ cur_tpoint annotations __class__rrr[s     zRawNihon.__init__c Cs|j|d}|j|d} |j|d} |j|d} |ddd} tdd | D} t|| kdd}t|| kdd}||kr|}d}t||d D]d}| |}||kr|}n |d |}||}||}||d d ||f|||||||}|}qn| |}|d d }|dd|d d}|}|dkrZ|| |d }|||d}t|j|dz}|||}||t |d|d}| tj }|j |ddd}|d d| }|| 7}|| 9}t |||||W5QRXd S)zRead a chunk of raw data.rxrrrrorrecSsg|] }|dqS)rnr)r:trrrr<sz/RawNihon._read_segment_file..rrnNrhrbrirjr rbzs        #p1