U f{j/ @sddlZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z mZd gZd d d d d dd d dd ZdZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%ZdS)&N)DigPoint)FIFF) create_info) pick_info)rotation3d_align_z_axis)warn_check_pandas_installed neuromag306g{Gz?gMbP?gư>gY@) mcmmmVZmVZuVTzT/mzT/cmzImporting FieldTrip data without an info dict from the original file. Channel locations, orientations and types will be incorrect. The imported data cannot be used for source analysis, channel interpolation etc.cCst|trtddS)z*Run validation checks on the ft_structure.z/Loading of data in cell arrays is not supportedN) isinstancelist RuntimeError) ft_structrU/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/fieldtrip/utils.py_validate_ft_struct!s rc sN|dkrttt|}|d|r|ttdrtdtffddD}fddD}||d<d|krt|d||d<d |kr|d jd krtj |d |d d |d < |d <W5QRXfddD}t |ddn7sz _create_info..csg|]}|kr|qSrrr)missing_channelsrrr!8strialavgrrZaxissfreqcsg|]}d|qSr rr)inforrr!JsF)copyT)Zupdate_redundant)chsdig)r NOINFO_WARNING _set_sfreqr(setstr#_remove_missing_channels_from_trialndimnpdeleteZ_unlockrr_create_info_chs_digupdate)rZraw_infor&missing_chan_idxZnew_chsZch_idxr)r*r)rr'r"r _create_info'sF     r6cCsdt|tr0tt|D]}t|||||<qn0t|tjrX|jdkr`tj||dd}nt d|S)Nrrr%z?"trial" field of the FieldTrip structure has an unknown format.) rrrangelenr/r1Zndarrayr0r2 ValueError)r#r5Z idx_trialrrrr/Us    r/c Cs|d}ttjddtjtddddddddddddg tjd}z |d}Wntk rdd}YnXz |d}Wntk rd}YnX|dkr|dkrtd d |p|pd dikrt d t }t }d}t |D]4\}} | } | | d <|d| d <|d| d<|r\| |dkr\t | |} |d7}t| dddtjtj|d} || n|r|| |dkr|t| |} n| drtj| d<tj| d<nf| drtj| d<tj| d<nD| drtj| d<tj| d<n"td| ftj| d<tj| d<|| q||fS)z7Create the chs info field from the FieldTrip structure.rg?rr ) coord_framecalr7Zunit_mullocunitelecNgradzThe supplied FieldTrip structure does not have an elec or grad field. No channel locations will extracted and the kind of channel might be inaccurate.chanposzThis file was created with an old version of FieldTrip. You can convert the data to the new version by loading it into FieldTrip and applying ft_selectdata with an empty cfg structure on it. Otherwise you can supply the Info field.ch_nameZlognoZscannor<r)rr:kindidentZEOGrC coil_typeZECGZSTIzFCannot guess the correct type of channel %s. Making it a MISC channel.)dictrZFIFFV_COORD_UNKNOWNZFIFF_UNITM_NONEr1array FIFF_UNIT_VKeyErrorr rr enumerater(_process_channel_eegrFIFFV_COORD_HEADZFIFFV_POINT_EEGappend_process_channel_meg startswithZ FIFFV_EOG_CHFIFFV_COIL_EEGZ FIFFV_ECG_CHZFIFFV_COIL_EEG_BIPOLARZ FIFFV_STIM_CHZFIFFV_COIL_NONEZ FIFFV_MISC_CH) rZ all_channelsZ ch_defaultsr>r?r)r*counterZidx_chanZcur_channel_labelcur_chdrrrr3gst                      r3c Csz |d}Wndtk rpz |d}Wntk rDtdYn(Xt|dt|d}}d||}YnXz t|}Wn4tk rtd|dt|d}YnX|S)zSet the sample frequency.ZfsampletimezNo Source for sfreq foundrr zJFieldTrip structure contained multiple sample rates, trying the first of: z Hz)rIr9float TypeErrorr Zravel)rr&rTt1t2rrrr,s   r,cs@|dtfddtD}|r4dd}ntd|S)z?Set the start time before the event in evoked data if possible.rTc3s.|]&\}}|d|ddkVqdS)rr Nr)rixtimesrr sz_set_tmin..rz>Loading data with non-uniform times per epoch is not supported)allrJr)rZ time_checkZtminrr[r _set_tmins r_cCsd|kr dS|d}tt|}|dkr0tdd}|jdkrH|jd}||dkr\tdtt|}|jdkr|dd|f}tt|||g dj }|S) z4Create an event matrix from the FieldTrip structure. trialinfoNrz!trialinfo_column must be positiver rzBtrialinfo_column is higher than the amount ofcolumns in trialinfo.int) r7r8r9r0shaper1zerosZvstackrGZastyper)rZtrialinfo_columnZ event_typeZ event_numberZavailable_ti_colsZevent_trans_valeventsrrr_create_eventss(     recCs,tdd}|stddS||d}|S)z%Create event metadata from trialinfo.F)strictz]The Pandas library is not installed. Not returning the original trialinfo matrix as metadata.Nr`)r r Z DataFrame)rZpandasmetadatarrr_create_event_metadatas  rhcCst|d}t||dkdd}t|d|ddf}|d}|t|}t|tdf|d<tj|d<tj |d <tj |d <tj |d <|S) aAConvert EEG channel from FieldTrip to MNE. Parameters ---------- cur_ch: dict Channel specific dictionary to populate. elec: dict elec dict as loaded from the FieldTrip structure Returns ------- cur_ch: dict The original dict (cur_ch) with the added information rrArr@Nr=) r<rCrEr:) r1 asanyarraywheresqueeze _unit_dicthstackrcrrHZ FIFFV_EEG_CHrPrL)rRr> all_labelsZchan_idx_in_elecposition position_unitrrrrKs     rKcCsXt|d}t||dkdd}|d}|d|}|d}t|d|ddf}|t|}|d krd |krd |krt|d |ddfdkd}|d |ddf} | t|} |d |ddf} |d kr| dd| d}|dkr| d| } tj| dd}t|d|ddf} zt| j} Wn tk rft d} YnX| j dksxt| } t || f|d<t j|d<|d krt j|d<t j|d<n|dkrt j|d<t j|d<nn|dkrt j|d<t j|d<nN|dkrt j|d<t j|d<n.|dkrs8      .E