U f{jZ$@sddlZddlmZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZmZmZmZmZmZddlmZmZmZmZmZmZdd lmZmZmZm Z e dd d Z!d dZ"ddZ#ddZ$ddZ%ddZ&dS)N)FIFF) fiff_open)find_tag) dir_tree_find) start_block end_block write_stringstart_and_end_filewrite_float_sparse_rcs write_int) read_surface_triangle_neighbors_compute_nearest_normalize_vectors_get_tri_supp_geom_find_nearest_tri_pts)get_subjects_dirwarnloggerverboseFc Cs:t|dd}t|d}t|sRzt|Wn tk rPtd|YnX|rp||krftdd}d}nd}d }|||f|||fg}d } |D].} t|d | } t | rt | ||Sqt d | t |||ft ||||} ||krd } n t |||ft ||||} t| ||| | | S) a1Read morph map. Morph maps can be generated with mne_make_morph_maps. If one isn't available, it will be generated automatically and saved to the ``subjects_dir/morph_maps`` directory. Parameters ---------- subject_from : str Name of the original subject as named in the SUBJECTS_DIR. subject_to : str Name of the subject on which to morph as named in the SUBJECTS_DIR. subjects_dir : str Path to SUBJECTS_DIR is not set in the environment. xhemi : bool Morph across hemisphere. Currently only implemented for ``subject_to == subject_from``. See notes of :func:`mne.compute_source_morph`. %(verbose)s Returns ------- left_map, right_map : ~scipy.sparse.csr_matrix The morph maps for the 2 hemispheres. T)Z raise_errorz morph-mapsz/Could not find or make morph map directory "%s"z\Morph-maps between hemispheres are currently only implemented for subject_to == subject_fromz %s-%s-xhemiz!Creating morph map %s -> %s xhemiz%s-%szCreating morph map %s -> %sNz %s-morph.fifz@Morph map "%s" does not exist, creating it and saving it to disk)ropjoinisdirosmkdir ExceptionrNotImplementedErrorexists_read_morph_maprinfo_make_morph_map_write_morph_map) subject_from subject_to subjects_dirxhemirZmmap_dirZ map_name_tempZlog_msgZ map_namesfnameZmap_namemmap_1mmap_2r*L/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/morph_map.pyread_morph_mapsJ       r,c Cs t|\}}}|}t|tj}t|dkr4tdd}d} |D]} t|| tj} | j|kr@t|| tj } | j|kr@t|| tj } | jtj krt|| tj } | j}t dq@| jtjkr@t|| tj } | j} t dq@W5QRX|dks| dkrtd||| fS)zRead a morph map from disk.rzMorphing map data not foundNz Left-hemisphere map read.z Right-hemisphere map read.z%Could not find both hemispheres in %s)rrrFIFFB_MNE_MORPH_MAPlen ValueErrorrFIFF_MNE_MORPH_MAP_FROMdataFIFF_MNE_MORPH_MAP_TO FIFF_MNE_HEMIFIFFV_MNE_SURF_LEFT_HEMIFIFF_MNE_MORPH_MAPrr FIFFV_MNE_SURF_RIGHT_HEMI) r'r#r$ftree_fidmapsZleft_mapZ right_mapmtagr*r*r+rfs0        rc Cs`z(t|}t|||||W5QRXWn2tk rZ}ztd||fW5d}~XYnXdS)zWrite a morph map to disk.z/Could not write morph-map file "%s" (error: %s)N)r _write_morph_map_rr)r'r#r$r(r)r:expr*r*r+r"s r"cCst|dksttjtjg}t||D]X\}}t|tjt|tj |t|tj |t |tj |t |tj|t|tjq&|dk rt|dkstt||D]X\}}t|tjt|tj |t|tj |t |tj |t |tj|t|tjqdS)N)r.AssertionErrorrr4r6ziprr-r r0r2r r3r r5r)r:r#r$r(r)hemisr<Zhemir*r*r+r>s$   r>cs6t|rdd}ndd}fdd|DS)aConstruct morph map from one subject to another. Note that this is close, but not exactly like the C version. For example, parts are more accurate due to double precision, so expect some small morph-map differences! Note: This seems easily parallelizable, but the overhead of pickling all the data structures makes it less efficient than just running on a single core :( z%s.sphere.left_right))lhrh)rErDz %s.sphere.reg))rDrD)rErEc s(g|] \}}t||qSr*)_make_morph_map_hemi).0Z hemi_fromZhemi_toregr#r$r%r*r+ s z#_make_morph_map..)r)r#r$r%r&rCr*rHr+r!s r!csddlm}m}||krR||krRt||d|}tt|ddd}|||ddSt||d|}t|dd\} } t||d|}t|ddd} t| t| t| | dd } t | t| fd d | Dt dgd d D} t j d kst| dtkstg}g}tt| | d}t| | fddd|\}}| |}t |}t t t| d}||||fft| t| fd}|S)z'Construct morph map for one hemisphere.r) csr_matrixeyeZsurfF)rZcsr)formatZcKDTree)methodcsg|]}|tqSr*)Zastypeint)rGZpt_idxZ from_pt_trisr*r+rJsz(_make_morph_map_hemi..cSsg|] }t|qSr*)r.)rGxr*r*r+rJsr)ZrrZtris)Zrun_allZ reproject)shape)Z scipy.sparserKrLrrr.r rrrnpZcumsumZ concatenatendimrArdictrarrayrepeatZarangeZravel)r#r$r%Zreg_fromZreg_torKZspeyer'Zn_ptsZfrom_rrZfrom_triZto_rrZ nn_pts_idxZ from_pt_lensZtri_indsweightsZtri_geomZnn_idxZrow_indZthis_mapr*rPr+rFsL   rF)NFN)'ros.pathpathrnumpyrUZ io.constantsrZio.openrZio.tagrZio.treerZio.writerrr r r r Zsurfacer rrrrrutilsrrrrr,rr"r>r!rFr*r*r*r+ s$        I"