Example #1
0
# [email protected]

import numpy as np
import nibabel as nib
from nibabel import trackvis
from trimesh_class import TriMesh
from trimesh_vtk import TriMesh_Vtk



# add normal direction
#file_name = "brain_mesh/lh_sim_tri_tau20_l3l10.ply"
file_name = "brain_mesh/lhwhitet.vtk"
mesh = TriMesh_Vtk(file_name, None)
vertices = mesh.get_vertices()

nb_step = 10
saved_flow = "testflow.dat"
trk_filename = "testtract_lhwhite2.trk"
lines = np.memmap(saved_flow, dtype=np.float64, mode='r', shape=(nb_step, vertices.shape[0], vertices.shape[1]))

# mesh.update()
mesh.update_normals()
normals = mesh.get_normals()


# load nifti for trk info
#subject_nii_file = "brain_mesh/s31_T1.nii.gz"
subject_nii_file = "brain_mesh/S1-A2_fa.nii.gz"
fa_file = nib.load(subject_nii_file)
img = fa_file.get_data()
 def update_polydata(self, vertices_flow_index=-1):
     TriMesh_Vtk.set_polydata_triangles(self, TriMeshFlow.get_triangles(self))
     TriMesh_Vtk.set_polydata_vertices(self, TriMeshFlow.get_vertices(self, vertices_flow_index=vertices_flow_index))