U f{jR@sddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl m Z mZmZmZmZmZdd lmZdd lmZdd lmZmZdd lmZdd lmZed ddZddZeGdddeZ ddZ!ddZ"ddZ#ddZ$ddZ%dS)!N)BaseRaw) create_info_format_dig_points) _mult_cal_one) Annotations)loggerverbosefill_docwarn _check_fname _import_h5py)FIFF)_make_dig_points) _frame_to_str apply_trans)_convert_fnirs_to_head)get_mni_fiducialsunknownFcCst||||S)aReader for a continuous wave SNIRF data. .. note:: This reader supports the .snirf file type only, not the .jnirs version. Files with either 3D or 2D locations can be read. However, we strongly recommend using 3D positions. If 2D positions are used the behaviour of MNE functions can not be guaranteed. Parameters ---------- fname : str Path to the SNIRF data file. optode_frame : str Coordinate frame used for the optode positions. The default is unknown, in which case the positions are not modified. If a known coordinate frame is provided (head, meg, mri), then the positions are transformed in to the Neuromag head coordinate frame (head). %(preload)s %(verbose)s Returns ------- raw : instance of RawSNIRF A Raw object containing fNIRS data. See Also -------- mne.io.Raw : Documentation of attribute and methods. )RawSNIRF)fname optode_framepreloadr rR/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/snirf/_snirf.pyread_raw_snirfs rcCst|dddS)Nrzlatin-1)encoding)open)rrrr_open9sr cs.eZdZdZed fdd ZddZZS) ra5Raw object from a continuous wave SNIRF file. Parameters ---------- fname : str Path to the SNIRF data file. optode_frame : str Coordinate frame used for the optode positions. The default is unknown, in which case the positions are not modified. If a known coordinate frame is provided (head, meg, mri), then the positions are transformed in to the Neuromag head coordinate frame (head). %(preload)s %(verbose)s See Also -------- mne.io.Raw : Documentation of attribute and methods. rFNcC sf ddlm}t}t|ddd}td|||d dd krRtd td }|d k|d k@rpd}t  d }|dkrt d| djdd} t} | dkrtdt  d} dd| D} t| dkr t dt| d| dddfdd } t d!dd d"D} t d#d| D}| |} t| | d$} d%d&krt  d'}d(d|D}n&t fd)d| D}d*d|D}d+d&krt  d,}d-d|D}n&t fd.d| D}d/d|D}d0d&kd1d&k@rLt  d2}t  d3}nd4d&kd5d&k@rtd6t  d7}t  d8}t j|t |jddfdd9}t j|t |jddfdd9}nt d:t||jdkstt||jdkstg}g}| D]}ttt  d;|d<d}ttt  d;|d=d}|dkrttt  d;|d>d}||dd?||dd@t| |d}|||dAn|dBkrtt  d;|dCddD}|dEdF}||dd?||d}|dFkrvttt  d;|d>d}d@t| |d}n d@|}||}||||qt|| |dGi}t  dH}t|ddD|dI<dJ dKkr dLddD} | |dM<dN dKkr2 dOddD}!|!|dP<dQ dKkr dRddD}"|"dSkrltj|dQ<n*|"dTkrtj |dQ<n|"dUkrtj!|dQ<j"|dVtdW}#t#|#}$||$}||$}|dXkrt$dY|d||g\}}}%}&n t %dZ}&|d[krtj&}'n*d\ dKkr*t d]d}'ntj'}'t(| D]`\}(}ttt  d;|d<d}ttt  d;|d=d}||dddfd^|(d_dd`<||dddfd^|(d_d`da<d^|(d_dd`d^|(d_d`dad})|)d^|(d_dd<|'d^|(db<|dcksX|dBkr8||(dFkr8ttt  d;|d>d}| |dd^|(d_da<q8dd dekrt  df}*|*|$}*t  dg}+dh\},}-}.}/t)}0t(|+D]\}(}1|1}1|1dikr|*|(ddf}-n\|1djkr6|*|(ddf},n@|1dkkrR|*|(ddf}.n$|*|(ddf|0dlt|0ddm<qt*|,|-|.|/|0t+|'dn}2nfdodt,t| D}3t-dYdpdq}2|2D]"}4t.|&|4d|4d<tj&|4db<qt(|3dD]$\}5}6|2t)tj/|6|5tj&drqt0|2}2~&1|2ds<W5QRXtt  dtddD}7tt  duddD}8|7|8}9dvD]6}:zt2j23|9|:};Wnt4k rYnX qڐ qtdw|9t22dxddddd};|;jt2j5j6dy};1|;dz<W5QRXd{ dKk rt  d|d}1|>d~d<W5QRXt:t;|j<||g| g|dt=ggg}?d D]z}@d|@k rt >t  d|@d}A|Aj?dk rtt  d|@dd}B|?|Adddfd|BdD q|j@|?dpdW5QRX||jdS)Nr)_validate_nirs_inforeadTrz Loading %srZdata2ZnirszoFile contains multiple recordings. MNE does not support this feature. Only the first dataset will be processed.ZManufacturerNamerZ Gowerlabsheadz$nirs/data1/measurementList1/dataType)韆zMNE only supports reading continuous wave amplitude and processed haemoglobin SNIRF files. Expected type code 1 or 99999 but received type code /nirs/data1/dataTimeSeriesrr$z.Unable to extract sample rate from SNIRF file.znirs/probe/wavelengthscSsg|] }t|qSrint).0wrrr {sz%RawSNIRF.__init__..rzThe data contains z wavelengths: zW. MNE only supports reading continuous wave amplitude SNIRF files with two wavelengths.cSs|rt|S|S)N)isdigitr(textrrratoiszRawSNIRF.__init__..atoicsfddtd|DS)Ncsg|] }|qSrrr)cr/rrr+sz;RawSNIRF.__init__..natural_keys..z(\d+))resplitr-r2rr natural_keyssz'RawSNIRF.__init__..natural_keyscSsg|]}|qSrr)r)namerrrr+sZdata1cSsg|] }d|kqS)ZmeasurementListr)r)nrrrr+s)keyZsourceLabels_disabledz nirs/probeznirs/probe/sourceLabelscSsg|]}|dqSUTF-8decoder)srrrr+sc s,g|]$}ttd|ddqS) nirs/data1/ /sourceIndexr_correct_shapenparraygetr0datrrr+s   cSsg|]}dt|qS)Sr'r=rrrr+sZdetectorLabels_disabledznirs/probe/detectorLabelscSsg|]}|dqSr9r;r)drrrr+sc s,g|]$}ttd|ddqS)r?/detectorIndexrrAr0rFrrr+s   cSsg|]}dt|qS)Dr'rIrrrr+sZ detectorPos3DZ sourcePos3Dznirs/probe/detectorPos3Dznirs/probe/sourcePos3DZ detectorPos2DZ sourcePos2DaThe data only contains 2D location information for the optode positions. It is highly recommended that data is used which contains 3D location information for the optode positions. With only 2D locations it can not be guaranteed that MNE functions will behave correctly and produce accurate results. If it is not possible to include 3D positions in your data, please consider using the set_montage() function.znirs/probe/detectorPos2Dznirs/probe/sourcePos2D)ZaxiszYNo optode location information is provided. MNE requires at least 2D location informationr?r@rKz/wavelengthIndex_ Zfnirs_cw_amplituder%z/dataTypeLabelr:ZdodZfnirs_od)ch_typesznirs/metaDataTags/SubjectIDZ first_nameZlastNameznirs/metaDataTags/z/nirs/metaDataTags/lastName last_nameZ middleNamez/nirs/metaDataTags/middleNameZ middle_nameZsexz/nirs/metaDataTags/sex>m1MMale>FFemalef2>0u) subject_infoZ LengthUnit)mrimegZ fsaverage)r#r\r]ZMNE_coordFramez!/nirs/metaDataTags/MNE_coordFramechsloc coord_frame)r$Z landmarkPos3Dz nirs/probe/z/nirs/probe/landmarkPos3Dz/nirs/probe/landmarkLabels)NNNN)slpasal)snasion)srpasarZEEGZ03d)nasionlparpahpiZ dig_ch_posrccs$g|]}d|dddqS)r_r`rrr)r)idx)inforrr+UsF)r )kindridentrcdigz"/nirs/metaDataTags/MeasurementDatez"/nirs/metaDataTags/MeasurementTime)z%Y-%m-%d%H:%M:%SZz%Y-%m-%d%H:%M:%SznExtraction of measurement date from SNIRF file failed. The date is being set to January 1st, 2000, instead of i)tzinfo meas_dateZ DateOfBirthz/nirs/metaDataTags/DateOfBirthz(\d+)-(\d+)-(\d+)r[birthday) filenames last_sampsr Zstimz/nirs/z/dataz/name?)Z emit_warning)AZpreprocessing.nirsr!rr r riFiler _get_metadata_strrCrDrEitem RuntimeErrorshape_extract_sampling_ratelenkeyssorteduniqueappendZzerosAssertionErrorr(rBstrr<lowerreplacerrZFIFFV_SUBJ_SEX_MALEZFIFFV_SUBJ_SEX_FEMALEZFIFFV_SUBJ_SEX_UNKNOWNupdate_get_lengthunit_scalingreyeZFIFFV_COORD_HEADZFIFFV_COORD_UNKNOWN enumeratedictrrrangerrZFIFFV_POINT_EEGrZ_unlockdatetimestrptime ValueErrortimezoneutcr3 fullmatchgroupssuperr__init__rZ atleast_2dsizeZset_annotations)Cselfrrrr r!h5pyZ manafacturerZsnirf_data_typerq sampling_rateZfnirs_wavelengthsr5ZchannelsZ channels_idxsourcesZ detectorsZdetPos3DZsrcPos3DZdetPos2DZsrcPos2DZchnamesrOZchanZsrc_idxZdet_idxZwve_idxZch_nameZdt_idsuffixr[nameslnrQr> length_unitZlength_scalingrMZhead_trcrhZmidpointZdiglocsZdignamerdrerfrgZextra_psZdignrlZch_locsZfidiiZch_locZstr_dateZstr_timeZ str_datetimeZdt_codernZ str_birthZ birth_matchedroZannotr8datadesc __class__)r/rGrirrRsb                                   ((                           $zRawSNIRF.__init__c CsLddl}||jdd} | d||j} W5QRXt|| |||dS)z#Read a segment of data from a file.rNrr&)rrsZ _filenamesTr) rrrhfistartstopZcalsZmultrrGonerrr_read_segment_fileszRawSNIRF._read_segment_file)rFN)__name__ __module__ __qualname____doc__r rr __classcell__rrrrr=sNrcCs|jdkr|tj}|S)Nr)rwrCZnewaxis)ZarrrrrrBs  rBcCs0dddd}||kr||Std|ddS)z5MNE expects time in seconds, return required scaling.r$)msr>rzThe time unit ^ is not supported by MNE. Please report this error as a GitHub issue to inform the developers.Nrv) time_unitscalingsrrr_get_timeunit_scalings rcCs0dddd}||kr||Std|ddS)z3MNE expects distance in m, return required scaling.r$dr)rQcmmmzThe length unit rNr)rrrrrrs rcCst|d}d}t|dkr6d|d|d}n>tjt|dd}tt|dkrldt|}ntdt|d }t |}||9}|S) z,Extract the sample rate from the time field.znirs/data1/timerrrrr$r^)ZdecimalszQMNE does not currently support reading SNIRF files with non-uniform sampled data.ZTimeUnit) rCrDrEryarounddiffr|r rtr)rGZ time_datarZfs_diffrZtime_unit_scalingrrrrxs  rxcCsH|t|dkrdS|d|}tt|}t|dd}|S)Nznirs/metaDataTagsz/nirs/metaDataTags/rzutf-8)rCrDrErBr)rGfieldrrrrrts rt)rFN)&r3numpyrCrbaserZ meas_inforrutilsr annotationsrr r r r r r constantsrZ _digitizationrZ transformsrrZ nirx.nirxrZ _freesurferrrr rrBrrrxrtrrrrs0         "n