U f»{j+ã@sJddlZddlmZddlmZdd„Zdd „Zd d „Zedd d„ƒZ dS)éNé)ÚFIFFé)Úfill_doccCs^d}d}t|dƒD]D\}}|dtjkrt|dƒd?}|dkrH|}q||krtdƒ‚q|S) z3Get the current compensation in effect in the data.NéÿÿÿÿÚchsÚkindÚ coil_typeérz3Compensation is not set equally on all MEG channels)Ú enumeraterÚ FIFFV_MEG_CHÚintÚ ValueError)ÚinfoÚcompZ first_compÚkÚchan©rúQ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/io/compensator.pyÚget_current_comps rcCsTt|ƒ}t|dƒD]:\}}|dtjkr|d|d>}t||d>ƒ|d<qdS)z3Set the current compensation in effect in the data.rrr r N)rr rr r )rrZcomp_nowrrÚremrrrÚset_current_comps rc srtt|dƒƒD]N}|d|d|kr|d|d}t |d|df¡}t|dƒD]d\}‰‡fdd„t|d ƒDƒ}t|ƒd kr˜td ˆƒ‚nt|ƒd kr°td ˆƒ‚d|||d f<q\t |d|df¡}t|d ƒD]Z\}‰‡fdd„t|dƒDƒ}t|ƒd kr"td ˆƒ‚qät|ƒd kräd|||d f<qät |t |d|¡¡} | Sqtd|ƒ‚dS)z(Auxiliary function for make_compensator.ÚcompsrÚdataZncolÚnchanZ col_namescsg|]\}}|ˆkr|‘qSrr©Ú.0rÚch)Úcol_namerrÚ (sÿz%_make_compensator..Zch_namesrz#Channel %s is not available in datarzAmbiguous channel %sgð?Znrowcsg|]\}}|ˆkr|‘qSrrr)Úch_namerrr4sÿZ row_namesz2Desired compensation matrix (grade = %d) not foundN)ÚrangeÚlenÚnpZzerosr rÚdot) rZgraderZ this_dataZpreselÚcolÚindZpostselÚcZ this_compr)r rrÚ_make_compensators2 ÿ    ÿr(Fc CsÆ||kr dS|dkr8t||ƒ}tj t |d¡|¡}|dkr\t||ƒ}t |d¡|}|dkr€|dkrzt ||¡}q„|}n|}|rÂdd„t|dƒDƒ} t| ƒdkr²tdƒ‚|| dd…f}|S)a Return compensation matrix eg. for CTF system. Create a compensation matrix to bring the data from one compensation state to another. Parameters ---------- %(info_not_none)s from_ : int Compensation in the input data. to : int Desired compensation in the output. exclude_comp_chs : bool Exclude compensation channels from the output. Returns ------- comp : array | None. The compensation matrix. Might be None if no compensation is needed (from == to). NrrcSs"g|]\}}|dtjkr|‘qS)r)rZFIFFV_REF_MEG_CH)rrr'rrrrrsÿz$make_compensator..rz9Nothing remains after excluding the compensation channels) r(r#ZlinalgÚinvÚeyer$r r"r) rZfrom_ÚtoZexclude_comp_chsZC1Z comp_from_0ZC2Z comp_0_torÚpickrrrÚmake_compensatorBs&   r-)F) Únumpyr#Ú constantsrÚutilsrrrr(r-rrrrÚs   #