def toVTK(swc): vtkgen = swc2vtk.VtkGenerator() vtkgen.add_swc(swc) vtkgen.write_vtk(swc.strip('.swc') + '.vtk')
# -*- coding: utf-8 -*- import os import swc2vtk import swcfilelist # input_dir = '/home/nebula/git/LAL-VPCmapping/converted_swc' input_dir = './swc' output_dir = '/home/nebula/work/paraview/standardbrain20170131/' if not os.path.isdir(output_dir): os.mkdir(output_dir) filelist = swcfilelist.filelist_lalvpc for i, filename in enumerate(filelist): vtkgen = swc2vtk.VtkGenerator() vtkgen.add_swc(os.path.join(input_dir, filename + '.swc')) vtkgen.write_vtk(os.path.join(output_dir, filename + '.vtk'), radius_data=True, normalize_diam=True) for i, filename in enumerate(filelist): vtkgen = swc2vtk.VtkGenerator() vtkgen.add_swc(os.path.join(input_dir, filename + '.swc'), inv_x=True, shift_x=1024.0) vtkgen.write_vtk(os.path.join(output_dir, filename + '_flip.vtk'), radius_data=True, normalize_diam=True)
def setUp(self): self.swc_file_path = os.path.join('tests', 'simple.swc') self.data_file_path = os.path.join('tests', 'simple.dat') self.output_file_path = os.path.join('output.vtk') self.vtk_generator = swc2vtk.VtkGenerator()