U f{j@s~ddlmZddlmZddlZddlmZddl m Z m Z ddl m Z mZmZmZddlmZmZmZmZed d d ZdS))path) combinationsN) _picks_to_idx)_read_mri_surface fast_cross_3d) apply_transinvert_transform _cart_to_sph _ensure_trans)verboseget_subjects_dir_validate_type _ensure_int Tc sdddlm}m} t|d}t|td|r2|dkrHtd|t|dd}zt t ||d d } Wn2t k r} zt | d d W5d } ~ XYnXt|d krd n|} tfdd| D} t|dd}t|| } | || }t| }t| d}t| jtj}t| D]\}}| t||d |d}ttjtt|d\}}}t||||}|||dkd9<tj|dd}|tj|}| dt |d d dd f||dd fj!ddtj"dk}tjjt|||||dd}|t#|||<qtt$||}t| |D]"\}}|d|dd d<q<S)a<Project sensors onto the brain surface. Parameters ---------- %(info_not_none)s %(trans_not_none)s %(subject)s %(subjects_dir)s %(picks_base)s only ``ecog`` channels. n_neighbors : int The number of neighbors to use to compute the normal vectors for the projection. Must be 2 or greater. More neighbors makes a normal vector with greater averaging which preserves the grid structure. Fewer neighbors has less averaging which better preserves contours in the grid. copy : bool If ``True``, return a new instance of ``info``, if ``False`` ``info`` is modified in place. %(verbose)s Returns ------- %(info_not_none)s Notes ----- This is useful in ECoG analysis for compensating for "brain shift" or shrinking of the brain away from the skull due to changes in pressure during the craniotomy. To use the brain surface, a BEM model must be created e.g. using :ref:`mne watershed_bem` using the T1 or :ref:`mne flash_bem` using a FLASH scan. r)pdist squareform n_neighborscopyz&n_neighbors must be 2 or greater, got T)Z raise_errorZbemz brain.surfz The brain surface requires generating a BEM using `mne flash_bem` (if you have the FLASH scan) or `mne watershed_bem` (to use the T1)NZecogcs$g|]}d|dddqS)chslocNr).0idxinfora/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/preprocessing/ieeg/_projection.py Jsz.project_sensors_onto_brain..headZmriZrrr)Zaxisrr)%Zscipy.spatial.distancerrrrboolr ValueErrorr ropjoinFileNotFoundError RuntimeErrorrnparrayr rr Zzerosshapenan enumerateZargsortmapziprrZmeanZlinalgZnormabssumpiZargminr )rZtranssubjectZ subjects_dirZpicksrrr rrZsurferrZ picks_idxZlocsdistsZanglesZ surf_anglesZ proj_locsirZ neighbor_ptsZpt1Zpt2Zpt3ZnormalsnormalZuse_rrZ surf_distsrrrrproject_sensors_onto_brainsh&         "r8)NNrTN)osrr% itertoolsrnumpyr)Zio.pickrZsurfacerrZ transformsrr r r utilsr r rrr8rrrrs