U f{j"@sddlmZddlmZddlZddlmZmZm Z ddl m Z ddl m Z ee dd d Ze d d Ze d dZddZddZdS)) OrderedDict)PathN)loggerfill_docverbose) create_info) EvokedArraycCs<t|}|jdkrt||S|jdkr0t||StddS)a[Reader function for BESA .avr or .mul files. When a .elp sidecar file is present, it will be used to determine electrode information. Parameters ---------- fname : str | Path Path to the .avr or .mul file. %(verbose)s Returns ------- ev : Evoked The evoked data in the .avr or .mul file. z.avrz.mulz(Filename must end in either .avr or .mulN)rsuffix_read_evoked_besa_avr_read_evoked_besa_mul ValueError)fnamerrO/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/besa/besa.pyread_evoked_besa s     rc st|4}|}d|k}|r4|}nd}W5QRXt|}tj||rXdnddd}t||rt|t|krt dt|dt|dnddk rt }t|t|krt d t|d t|dn t d d d tt|D}dkr&t ddgt|nfdd |Dd|krt|d|d<|d|jdkrt d|jd|ddd|krt|d|d<|d|jdkrt d|jd|ddd|krt|d|d<nt dd|krt|d|d<nd|d<d|kr4t|d|d<nd|d<d|krNd|d<t|d|ddd}t||dd ||dd|d|d!S)"z)Create EvokedArray from a BESA .avr file.zNchan=NrZskiprowsZndminzGMismatch between the number of channel names defined in the .avr file (z-) and the number of rows in the data matrix ().zBMismatch between the number of channels defined in the .avr file (z) and .elp file (zaNo .elp file found and no channel names present in the .avr file. Falling back to generic names. cSsg|]}d|ddqS)ZCHr02dr).0irrr Ksz)_read_evoked_besa_avr..Marking all channels as EEG.eegcsg|] }|qSrrrchch_typesrrrRsZNptsThe size of the data matrix (z#) does not match the "Npts" field (ZNchanrz$) does not match the "Nchan" field (ZDIz>No "DI" field present. Could not determine sampling frequency.ZTSBZSBg? SegmentNameZsfreqr.AZtmincommentr)openreadlinestripsplit _parse_headernploadtxt_read_elp_sidecarlen RuntimeErrorlistkeysrinforangeintshapewarnfloatr r ) rrfheader new_stylech_namesfieldsdatar4rrrr &sb              r c s t|"}|}|}W5QRXt|}tj|ddd}t||jdkrzt dt|d|jddt |dkrt dd gt|nfd d |Dd |krt |d |d <|d |jd krt d|jd|d dd|krJt |d|d<|d|jdkrJt d|jd|ddd|krft|d|d<nt dd|krt|d|d<nd|d<d|krt|d|d<nd|d<d|krd|d<t|d|dd}t|j|dd||dd|d|dS)z)Create EvokedArray from a BESA .mul file.rrrzGMismatch between the number of channel names defined in the .mul file (z0) and the number of columns in the data matrix (rNrrcsg|] }|qSrrrrrrrsz)_read_evoked_besa_mul..Z TimePointsrr z)) does not match the "TimePoints" field (ZChannelsz') does not match the "Channels" field (zSamplingInterval[ms]zPNo "SamplingInterval[ms]" field present. Could not determine sampling frequency.zBeginSweep[ms]gzBins/uVr!r"r#r$r%r&)r(r)r*r+r,r-r.r0r7r1r/rr4r6r8r9r r T)rrr:r;r=r>r?r4rrrr wsR          r cCs8|}t|ddd|ddd}tdd|DS)aParse an .avr or .mul header string into name/val pairs. The header line looks like: Npts= 256 TSB= 0.000 DI= 4.000000 SB= 1.000 SC= 200.0 Nchan= 27 No consistent use of separation chars, so parsing this is a bit iffy. Parameters ---------- header : str The first line of the file. Returns ------- fields : dict The parsed header fields Nrrcss"|]\}}|dd|fVqdS)=r"N)replace)rnamevalrrr sz _parse_header..)r+zipdict)r;partsZname_val_pairsrrrr,sr,c Cs|j|jd}|s.td|ddStd|t}t|z}|}t|d dkr|D]$}| dd\}}| ||<qnn,td |D]}| dd }d ||<qW5QRX|S) aDRead a possible .elp sidecar file with electrode information. The reason we don't use the read_custom_montage for this is that we are interested in the channel types, which a DigMontage object does not provide us. Parameters ---------- fname : Path The path of the .avr or .mul file. The corresponding .elp file will be derived from this path. Returns ------- ch_type : OrderedDict | None If the sidecar file exists, return a dictionary mapping channel names to channel types. Otherwise returns ``None``. z.elpzNo z/ file present containing electrode information.Nz'Reading electrode names and types from rrrzCNo channel types present in .elp file. Marking all channels as EEG.rr) parentstemexistsrr4rr( readlinesr0r+lower)rZ fname_elprr:lineslineZch_typeZch_namerrrr/s"  r/)N) collectionsrpathlibrnumpyr-utilsrrrZ meas_infor Zevokedr rr r r,r/rrrrs     P <