U f{j-@sddlZddlZddlmZmZddlmZmZm Z e dZ e dZ ed!ddZ ed"d d Zd d ZddZdddddZddZddddddddZddZddZddZdd ZdS)#N) _picks_to_idx pick_types)fill_doc _check_option_validate_typezS(\d+)_D(\d+) (\d+\.?\d*)zS(\d+)_D(\d+) (\w+)cs$tfddt|gdDtS)a4Determine the distance between NIRS source and detectors. Parameters ---------- %(info_not_none)s %(picks_all)s Returns ------- dists : array of float Array containing distances in meters. Of shape equal to number of channels, or shape of picks if supplied. c sBg|]:}tjtjd|ddddddddqS)chslocr r)Zaxis)npZlinalgZnormdiffZreshape.0pickinfoZ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/nirs/nirs.py "s z-source_detector_distances..)exclude)r arrayrfloat)rpicksrrrsource_detector_distancess  r{Gz?cCs t||kS)aDetermine which NIRS channels are short. Channels with a source to detector distance of less than ``threshold`` are reported as short. The default threshold is 0.01 m. Parameters ---------- %(info_not_none)s threshold : float The threshold distance for what is considered short in meters. Returns ------- short : array of bool Array indicating which channels are short. Of shape equal to number of channels. )r)r thresholdrrrshort_channels(src CsXt|ddggdd}t}|D],}|ttt|d|dqt |t S)z,Return the light frequency for each channel.fnirs_cw_amplitudefnirs_odTr allow_emptych_namesr ) rlistappendroundr _S_D_F_REmatchgroupsr rint)rrfreqsrrrr_channel_frequencies>s  r+cCsBt|ddggdd}g}|D] }||d|ddq|S)z'Return the chromophore of each channel.hbohbrTr r" )rr$split)rrZchromaiirrr_channel_chromophoreJs r2T throw_errors check_badscstddggdd}tddggdd}t|dkt|dk@rJtd|}t|r`d }t}|}n d }t}|}t|}|jd krtd |d t ||j j dk} t |}| r|ddgkrtd|d|dn(t t ||std|d|dt|ddkr&tdtjd|}fdd|D} tt| r\td| d|}|D]} |d| d} t| stdtj| |}q| d} t|r| } n,| dkr`tdd| d|}qq`|tfdd|D}t|d d d|d!d dD]\}}d|d}d|d}||}||}|d d"\}}}|d d"\}}}t|rt|t|}}||ks||ks||dks||d!kr*td#|d$|dd%|d!d&|d'|d(t|d)t||}q.q*|rt|d d d|d!d dD]V\}}j|j|g}t td*|}t|d!krRtd+|d,|qR|S)-aiCheck channels follow expected fNIRS format. If the channels are correctly ordered then an array of valid picks will be returned. If throw_errors is True then any errors in fNIRS formatting will be thrown to inform the user. If throw_errors is False then an empty array will be returned if the channels are not sufficiently formatted. rrTr r,r-rzqMNE does not support a combination of amplitude, optical density, and haemoglobin data in the same raw structure.Z frequencies chromophore)r z Exactly two z must exist in info, got UzThe z) in info must be ["hbo", "hbr"], but got z insteadz0 in info must be unique and sorted, but got got r zRNIRS channels not ordered correctly. An even number of NIRS channels is required. z channels wereprovidedcs g|]}d|ddqS)rr r r)rr1rrrrsz+_check_channels_ordered..zjNIRS channels is missing wavelength information in the info["chs"] structure. The encoded wavelengths are .rZch_namez^NIRS channels have specified naming conventions. The provided channel name can not be parsed: )r,r-z{NIRS channels have specified naming conventions.Chromophore data must be labeled either hbo or hbr. The failing channel is csg|]}d|qSr"rrrrrrsNr/rzhNIRS channels not ordered correctly. Channels must be ordered as source detector pairs with alternating r.z & z , but got SZ_Dz pair z and badsz,NIRS bad labelling is not consistent, found z but needed )rlen_throw_or_return_emptyr& _S_D_H_REr rshape ValueErrorr#ZdtypekindZ array_equaluniquer"anyisnanr'boolreprr(Zargsortziprset intersection RuntimeError)r pair_valsr4r5 picks_wave picks_chromarZ error_wordZuse_REZis_strZ all_freqsrZ ch_name_infovaluer1jjch1_namech2_nameZch1_reZch2_reZch1_SZch1_DZ ch1_valueZch2_SZch2_DZ ch2_valueZwantgotrrr_check_channels_orderedTs            (     < &rRcCs|rt|ngSdSN)r?)msgr4rrrr<s r<)r4fnirswhichr5r!c Cst|dtfdtdddd}td|dt||dk r||}|dkrRd d gnd |}tt||d st|d |dt t |}|j dkr|} nt t |} t || ||d} | S)zCApply all checks to fNIRS info. Works on all continuous wave types.NrUzoptical densityzcontinuous waver6)ZodZ cw_amplitudehbrSrWr,r-Zfnirs_)rUz must operate on z data, but none was found.rr3)rstrdictrtupler;rrIr rAr+sizer2rR) rr4rUrVr5r!kindsr@r*rJoutrrr_validate_nirs_infos2 r^cCst|dd}t|d}t|ddd|dddD]D\}}|j||j|}}||krh||q6||kr6||q6t||d<|S)z*Spread bad labeling across fnirs channels.F)r5r:Nr r/)r^rGrFr"addsorted)rrZnew_badsr1rNrOrPrrr_fnirs_spread_badss  &   racst|ddggdd}t|ddggdd}t|dkr:tnt|dkrLtnggfStfdd |jD}tfd d |jD}d d |D}d d |D}||fS) z#Return list of unique optode names.rrTr r,r-rcs"g|]}t|dqS)rr)r'r(rchregexrrrsz'_fnirs_optode_names..cs"g|]}t|dqS)r/rbrcrerrrscSsg|]}d|qS)Sr)rsrrrrscSsg|]}d|qS)Dr)rdrrrrs)rr;r&r=r rAr")rrKrLsourcesZ detectorsZ src_namesZ det_namesrrer_fnirs_optode_names s,    rlcsTfdd|jDd}dkr.tdd}ndkr@tdd}|d |d |S) zFind the position of an optode.csg|] }|kqSrr)raoptoderrr$sz$_optode_position..Trgrrir rr )r"indexrange)rroidxZloc_idxrrnr_optode_position"s   rtcsZ|jtdgd}fdd|Dfdd|D}t|fddd}||dS) NT)rUrcs g|]}d|dqS)r"r)r0rrrrr3sz!_reorder_nirx..csg|]}d|qSr9rrrrrr4scs"|d|jdddfS)Nrr/)maxsplit)rqr0)name)prefixesrr6sz_reorder_nirx..)key)rrr`Zreorder_channels)rawrZ nirs_namesZ nirs_sortedr)rrwr _reorder_nirx.s r{)N)r)renumpyr Zio.pickrrutilsrrrcompiler&r=rrr+r2rRr<r^rarlrtr{rrrrs0