U f{jKi@s`ddlmZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZdd lmZd d l mZmZmZmZmZmZd d lmZd d lmZd dlmZd dlmZm Z dZ!ddZ"ddZ#ddZ$ddZ%ddZ&d3ddZ'ddZ(ed4d!d"Z)ed5d#d$Z*eGd%d&d&eZ+Gd'd(d(eZ,d)d*Z-d+d,Z.d-d.Z/d6d/d0Z0d1d2Z1dS)7N)_readmat)_PICK_TYPES_KEYS)_read_segments_file_find_channels)FIFF) create_info)BaseRaw)loggerverbosewarnfill_docBunch _check_fname)make_dig_montage) BaseEpochs) read_events) Annotationsread_annotationsgư>cCstt|d}|dkr$tdn|dkr8td|t|}t||}t|st|dd}t|r|}d}t| |t |n||ksd}t | |||S) aCheck whether the filename is valid. Check if the file extension is ``.fdt`` (older ``.dat`` being invalid) or whether the ``EEG.data`` filename exists. If ``EEG.data`` file is absent the set file name with .set changed to .fdt is checked. z.datzcOld data format .dat detected. Please update your EEGLAB version and resave the data in .fdt formatz.fdtz*Expected .fdt file format. Found %s formatrzxData file name in EEG.data ({}) is incorrect, the file name must have changed on disk, using the correct file name ({}).z3Could not find the .fdt data file, tried {} and {}.) stropsplitextNotImplementedErrorIOErrordirnamejoinexistsrformatbasenameFileNotFoundError)fnameZdatanamefmtZbasedir data_fnameZfdt_from_set_fnamemsgr'S/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/eeglab/eeglab.py_check_eeglab_fnames&     r)cCsjt||d}d|krtdd|kr,|d}|d|}tf|}t|j|_t|j|_t|j|_|S)z'Check if the mat struct contains 'EEG'. uint16_codecZALLEEGzcLoading an ALLEEG array is not supported. Please contactmne-python developers for more information.ZEEG)rrgetrinttrialsnbchanpnts)r#r+eegr'r'r(_check_load_mat:s      r2cCs(t|ttfst|dkr|StjSdS)zCheck if location exists.rN) isinstancer-floatlennpnan)Zllr'r'r(_to_locKsr8cszddlm}Wn tk r0ddlm}YnXtjsBd}ndddg}tjd|rttfdd|D}nVtjdtj rtfd d|D}n,tjdt rtfd d|D}nd}|S) Nr) mat_structFXYZc3s|]}tjd|VqdSrN)hasattrchanlocs.0Zfldr1r'r( ]sz/_eeg_has_montage_information..c3s |]}|jdjjkVqdSr=)r?dtypenamesr@rBr'r(rCasc3s|]}|jdkVqdSr=)r?r@rBr'r(rCes) Zscipy.io.matlabr9 ImportErrorZscipy.io.matlab.mio5_paramsr5r?r3allr6ndarraydict)r1r9has_posZ pos_fieldsr'rBr(_eeg_has_montage_informationSs&      rKcsbttttf\}}}}t|jD]}||dd}|dd}t|tr|}|t krn|}n*|kr||dn|dg|<|||r(t |d} t |d} t |d} t j | | | f} ||d|| q(t r$tdd fd d tDd \} }}t|d r*t |jdgr*tt|jdD]}tt|jd|}|dddkrqdn|dddkrt |d|d|dg}n^|dddkrt |d|d|dg}n.|dddkrdt |d|d|dg} qd|rTttt|t |d| ||d}nd}|||fS)zAGet channel name, type and montage information from ['chanlocs'].labelsr1typeNr:r;r<z*Unknown types found, setting as type EEG:  cs"g|]}|dt|qS)z: )sortedrAkeyZ unknown_typesr'r( sz,_get_montage_information..)NNNchaninfoZ nodatchansZFID descriptionZNasionzRight periauricular pointzLeft periauricular pointhead)ch_pos coord_framelparpanasion)listrIr?appendr,r3rstriplowerrr8r6Zr_r5rrrOr>rTzipvalueskeysarrayr)r1Zget_posch_namesch_typesZ pos_ch_namesposZchanlocch_typeZtry_typeZloc_xZloc_yZloc_zZlocsrYrZr[itemdmontager'rRr(_get_montage_informationlsb             rkr'c Cst|jtjs"|jdkr"|jg|_t|jtr:t|j|_t|jdk}|rjt|}t ||\}}}d}n ddt |jD}d}d}d}t ||j |d }|d krt |d d n|}t|d D]8\} } t| d<| d|ks| |krtj| d<tj| d<q|||fS)zGet measurement info.rrFcSsg|] }d|qS)zEEG %03dr')rAiir'r'r(rSsz_get_info..r1NT)sfreqreautoZEOG)rgZchscalZch_nameZ coil_typekind)r3r?r6rHr/rI _dol_to_lodr5rKrkranger srater enumerateCALrZFIFFV_COIL_NONEZ FIFFV_EOG_CH) r1eogZeeg_has_ch_names_inforJrdre eeg_montageZupdate_ch_namesinfoidxchr'r'r( _get_infos,     r{cCsf|dkr|dnNt|jt|j}ttt|tt|dftj }||t |dddS)zSet EEG sensor configuration and head digitization from when init. This is done from the information within fname when read_raw_eeglab(fname) or read_epochs_eeglab(fname). Nr rV)rWrX) Z set_montagesetrdrIr`r\r6fullr5r7r)selfrjZmissing_channelsrWr'r'r(_set_dig_montage_in_inits rFcCst|||||dS)aRead an EEGLAB .set file. Parameters ---------- input_fname : str Path to the .set file. If the data is stored in a separate .fdt file, it is expected to be in the same folder as the .set file. eog : list | tuple | 'auto' Names or indices of channels that should be designated EOG channels. If 'auto', the channel names containing ``EOG`` or ``EYE`` are used. Defaults to empty tuple. %(preload)s Note that preload=False will be effective only if the data is stored in a separate binary file. uint16_codec : str | None If your \*.set file contains non-ascii characters, sometimes reading it may fail and give rise to error message stating that "buffer is too small". ``uint16_codec`` allows to specify what codec (for example: 'latin1' or 'utf-8') should be used when reading character arrays and can therefore help you solve this problem. %(verbose)s Returns ------- raw : instance of RawEEGLAB A Raw object containing EEGLAB .set data. See Also -------- mne.io.Raw : Documentation of attribute and methods. Notes ----- .. versionadded:: 0.11.0 ) input_fnamepreloadrvr r+) RawEEGLAB)rrvrr+r r'r'r(read_raw_eeglabs &rcCst||||||d}|S)aReader function for EEGLAB epochs files. Parameters ---------- input_fname : str Path to the .set file. If the data is stored in a separate .fdt file, it is expected to be in the same folder as the .set file. events : str | array, shape (n_events, 3) | None Path to events file. If array, it is the events typically returned by the read_events function. If some events don't match the events of interest as specified by event_id, they will be marked as 'IGNORED' in the drop log. If None, it is constructed from the EEGLAB (.set) file with each unique event encoded with a different integer. event_id : int | list of int | dict | None The id of the event to consider. If dict, the keys can later be used to access associated events. Example:: {"auditory":1, "visual":3} If int, a dict will be created with the id as string. If a list, all events with the IDs specified in the list are used. If None, the event_id is constructed from the EEGLAB (.set) file with each descriptions copied from ``eventtype``. eog : list | tuple | 'auto' Names or indices of channels that should be designated EOG channels. If 'auto', the channel names containing ``EOG`` or ``EYE`` are used. Defaults to empty tuple. %(verbose)s uint16_codec : str | None If your \*.set file contains non-ascii characters, sometimes reading it may fail and give rise to error message stating that "buffer is too small". ``uint16_codec`` allows to specify what codec (for example: 'latin1' or 'utf-8') should be used when reading character arrays and can therefore help you solve this problem. Returns ------- epochs : instance of Epochs The epochs. See Also -------- mne.Epochs : Documentation of attribute and methods. Notes ----- .. versionadded:: 0.11.0 )reventsrvevent_idr r+) EpochsEEGLAB)rrrrvr r+epochsr'r'r(read_epochs_eeglabs 4rcs.eZdZdZed fdd ZddZZS) racRaw object from EEGLAB .set file. Parameters ---------- input_fname : str Path to the .set file. If the data is stored in a separate .fdt file, it is expected to be in the same folder as the .set file. eog : list | tuple | 'auto' Names or indices of channels that should be designated EOG channels. If 'auto', the channel names containing ``EOG`` or ``EYE`` are used. Defaults to empty tuple. %(preload)s Note that preload=False will be effective only if the data is stored in a separate binary file. uint16_codec : str | None If your \*.set file contains non-ascii characters, sometimes reading it may fail and give rise to error message stating that "buffer is too small". ``uint16_codec`` allows to specify what codec (for example: 'latin1' or 'utf-8') should be used when reading character arrays and can therefore help you solve this problem. %(verbose)s See Also -------- mne.io.Raw : Documentation of attribute and methods. Notes ----- .. versionadded:: 0.11.0 r'FNcsxt|ddd}t||}|jdkr0td|j|jdg}t||d\}} } t|jtrt ||j} t d| t t |j||| g|d|d n|d kst|trtd |jdkrt|jjdkrd|jjd g\} } n |jj\} } tj| | ftd }|j|d| <|t9}t t |j|||g|d|d t|}||t|dt|| t|j|j d}t|dS)NreadTrrzThe number of trials is %d. It must be 1 for raw files. Please use `mne.io.read_epochs_eeglab` if the .set file contains epochs.rvz Reading %sdouble) filenames last_sampsZ orig_formatr FzsData will be preloaded. preload=False or a string preload is not supported when the data is stored in the .set filerrDrm)rr2r. TypeErrorr0r{r3datarr)r rxsuperr__init__rr/r5shaper6emptyr4rurZset_annotations_check_boundaryrroundonset_check_latencies)r~rrvrr+r r1rrxrw_r%Zn_chanZn_timesrannot latencies __class__r'r(rbsR           zRawEEGLAB.__init__c Cst||||||||dd dS)zRead a chunk of raw data.z.cSsi|]\}}||dqSrr')rAryZevr'r'r( sz)EpochsEEGLAB.__init__..z]At least one epoch has multiple events. Only the latency of the first event will be retained.r rrz Extracting parameters from %s...rz-No matching events found for %s (event id %i)rbF)order)rrrr)rejectflat reject_tmin reject_tmaxfilenamer zReady.)0rr2 ValueErrorr.r\ _bunchifyepocheventr3Z eventtyper-r4rrr]latencyr5rtrr6zerosrrrr rxr{itemsrr)openfromfileZfloat32Zreshaper/r0rZnewaxisZ transposeZastyperuAssertionErrorZxminZxmaxrrrZ _bad_droppedr)!r~rrrZtminZbaselinerrrrrvr r+r1Z event_nameZevent_latenciesZ unique_evZev_idxZwarn_multiple_eventsrepZ event_typeryZ prev_stimrxrwrrQvalr%Zdata_fidrZtmaxrr'r(rs                  zEpochsEEGLAB.__init__) NNrNNNNNr'NN)rrrrr rrr'r'rr(rsJrcCs,|dkrt}d|jkr(d|kr(tddS)NboundaryzThe data contains 'boundary' events, indicating data discontinuities. Be cautious of filtering and epoching around these events.)rIrUr)rrr'r'r(rLsrcCs,|dkrtd|dkr(tddS)NrzaAt least one event sample index is negative. Please check if EEG.event.sample values are correct.aSAt least one event has a sample index of -1. This usually is a consequence of how eeglab handles event latency after resampling - especially when you had a boundary event at the beginning of the file. Please make sure that the events at the very beginning of your EEGLAB file can be safely dropped (e.g., because they are boundary events).)anyrr)rr'r'r(rUs  rcCs>t|trt|}t|dkr:t|dtr:dd|D}|S)NrcSsg|]}tf|qSr')r)rArhr'r'r(rSfsz_bunchify..)r3rIrqr5)rr'r'r(rbs  rc Cs*t|trt||d}t|ds&g}nPt|jtrTt|jdjdkrTt |j}n"t|jtj t fsp|jg}n|j}t |}dd|D}dd|D}t t|}t|dkrt|ddrt|D]8\}}t|jtj ot|jdk}|rtjn|j||<qtt||j||j|d d S) a Create Annotations from EEGLAB file. This function reads the event attribute from the EEGLAB structure and makes an :class:`mne.Annotations` object. Parameters ---------- eeg : object | str 'EEG' struct or the path to the (EEGLAB) .set file. uint16_codec : str | None If your \*.set file contains non-ascii characters, sometimes reading it may fail and give rise to error message stating that "buffer is too small". ``uint16_codec`` allows to specify what codec (for example: 'latin1' or 'utf-8') should be used when reading character arrays and can therefore help you solve this problem. Returns ------- annotations : instance of Annotations The annotations present in the file. r*rrrcSsg|]}t|jqSr')rrMrArr'r'r(rSsz,_read_annotations_eeglab..cSsg|]}|jdqSr)rrr'r'r(rSsdurationN)rrrUZ orig_time)r3rr2r>rrIr6rcndimrqrHr\rrr5rtrr7rrs) r1r+rrUrrryrZis_empty_arrayr'r'r(_read_annotations_eeglabjs4       rcs*fddtttdDS)z+Convert a dict of lists to a list of dicts.cs$g|]fddDqS)csi|]}||qSr'r'rP)dolrlr'r(rsz*_dol_to_lod...)rb)rAr)rlr(rSsz_dol_to_lod..r)rrr5r\rbrr'rr(rqs rq)r')r'FNN)NNr'NN)N)2os.pathpathrnumpyr6Z_eeglabrpickrutilsrr constantsrZ meas_infor baser r r rrrrZchannelsrrrrr annotationsrrrur)r2r8rKrkr{rrrrrrrrrrqr'r'r'r(sP          < " ) 9W4   3