def writeDataSet(dataset, filename, verbose=0): mypy.my_print(verbose, "*** writeDataSet: " + filename + " ***") file_ext = filename[-3:] if (file_ext == "vti") or ((file_ext == "vtk") and dataset.IsA("vtkImageData")): myvtk.writeImage(image=dataset, filename=filename, verbose=verbose - 1) elif (file_ext == "vtp") or ((file_ext == "vtk") and dataset.IsA("vtkPolyData")): myvtk.writePData(pdata=dataset, filename=filename, verbose=verbose - 1) elif (file_ext == "stl"): myvtk.writeSTL(pdata=dataset, filename=filename, verbose=verbose - 1) elif (file_ext == "vts") or ((file_ext == "vtk") and dataset.IsA("vtkStructuredGrid")): myvtk.writeSGrid(sgrid=dataset, filename=filename, verbose=verbose - 1) elif (file_ext == "vtu") or ((file_ext == "vtk") and dataset.IsA("vtkUnstructuredGrid")): myvtk.writeUGrid(ugrid=dataset, filename=filename, verbose=verbose - 1) else: assert 0, "Wrong extention (" + file_ext + "). Extention must be vtk, vti, vtp, stl, vts, vtu. Aborting."
pdata_epi = myVTK.readSTL( filename=args.epi_filename, verbose=args.verbose) (clipped_pdata_endLV, clipped_pdata_endRV, clipped_pdata_epi, clipped_valM, clipped_valP) = myVTK.clipSurfacesForFullBiVMesh( pdata_endLV=pdata_endLV, pdata_endRV=pdata_endRV, pdata_epi=pdata_epi, verbose=args.verbose) myVTK.writeSTL( pdata=clipped_pdata_endLV, filename="endLV.stl", verbose=args.verbose) myVTK.writeSTL( pdata=clipped_pdata_endRV, filename="endRV.stl", verbose=args.verbose) myVTK.writeSTL( pdata=clipped_pdata_epi, filename="epi.stl", verbose=args.verbose) myVTK.writeSTL( pdata=clipped_pdata_valM, filename="valM.stl", verbose=args.verbose) myVTK.writeSTL( pdata=clipped_pdata_valP,
parser.add_argument("--r" , type=float, default=0.99) parser.add_argument("--endLV_filename", type=str , default=None) parser.add_argument("--epiLV_filename", type=str , default=None) parser.add_argument("--verbose", "-v" , type=int , default=1 ) args = parser.parse_args() if (args.endLV_filename is None): args.endLV_filename = args.domain_filename.replace("LV", "EndLV") if (args.epiLV_filename is None): args.epiLV_filename = args.domain_filename.replace("LV", "EpiLV") pdata_domain = myVTK.readSTL( filename=args.domain_filename, verbose=args.verbose) (pdata_end, pdata_epi) = myVTK.splitDomainBetweenEndoAndEpi( pdata_domain=pdata_domain, r=args.r, verbose=args.verbose) myVTK.writeSTL( pdata=pdata_end, filename=args.endLV_filename, verbose=args.verbose) myVTK.writeSTL( pdata=pdata_epi, filename=args.epiLV_filename, verbose=args.verbose)
mesh = myvtk.readUGrid( filename=args.mesh_in_filename, verbose=args.verbose) elif (args.mesh_in_filename.endswith(".vtp") or args.mesh_in_filename.endswith(".stl")): mesh = myvtk.readPData( filename=args.mesh_in_filename, verbose=args.verbose) else: assert (0), "Input file must be .vtk, .vtu, or .stl. Aborting." myvtk.moveMeshWithWorldMatrix( mesh=mesh, M=I2I, verbose=args.verbose) if (args.mesh_out_filename == None): args.mesh_out_filename = args.mesh_in_filename.replace(args.image_in, args.image_out) if (args.mesh_out_filename.endswith(".vtk") or args.mesh_out_filename.endswith(".vtu")): myvtk.writeUGrid( ugrid=mesh, filename=args.mesh_out_filename, verbose=args.verbose) elif (args.mesh_out_filename.endswith(".stl")): myvtk.writeSTL( pdata=mesh, filename=args.mesh_out_filename, verbose=args.verbose) else: assert (0), "Output file must be .vtk, .vtu, or .stl. Aborting."
######################################################################## ### ### ### Created by Martin Genet, 2012-2020 ### ### ### ### University of California at San Francisco (UCSF), USA ### ### Swiss Federal Institute of Technology (ETH), Zurich, Switzerland ### ### École Polytechnique, Palaiseau, France ### ### ### ######################################################################## from builtins import range import argparse import myPythonLibrary as mypy import myVTKPythonLibrary as myvtk ######################################################################## if (__name__ == "__main__"): parser = argparse.ArgumentParser() parser.add_argument("vtk_filename", type=str) args = parser.parse_args() mesh = myvtk.readPData( filename=args.vtk_filename) myvtk.writeSTL( pdata=mesh, filename=args.vtk_filename.replace(".vtk", ".stl").replace(".vtp", ".stl"))
parser.add_argument("--r" , type=float, default=0.99) parser.add_argument("--endLV_filename", type=str , default=None) parser.add_argument("--epiLV_filename", type=str , default=None) parser.add_argument("--verbose", "-v" , type=int , default=1 ) args = parser.parse_args() if (args.endLV_filename is None): args.endLV_filename = args.domain_filename.replace("LV", "EndLV") if (args.epiLV_filename is None): args.epiLV_filename = args.domain_filename.replace("LV", "EpiLV") pdata_domain = myVTK.readSTL( filename=args.domain_filename, verbose=args.verbose) (pdata_end, pdata_epi) = myVTK.splitDomainBetweenEndoAndEpi( pdata_domain=pdata_domain, r=args.r, verbose=args.verbose) myVTK.writeSTL( pdata=pdata_end, filename=args.endLV_filename, verbose=args.verbose) myVTK.writeSTL( pdata=pdata_epi, filename=args.epiLV_filename, verbose=args.verbose)
parser.add_argument('epi_filename', type=str) parser.add_argument('-v', '--verbose', type=int, default=1) args = parser.parse_args() pdata_endLV = myVTK.readSTL(filename=args.endLV_filename, verbose=args.verbose) pdata_endRV = myVTK.readSTL(filename=endRV_filename, verbose=args.verbose) pdata_epi = myVTK.readSTL(filename=args.epi_filename, verbose=args.verbose) (clipped_pdata_endLV, clipped_pdata_endRV, clipped_pdata_epi, clipped_valM, clipped_valP) = myVTK.clipSurfacesForFullBiVMesh(pdata_endLV=pdata_endLV, pdata_endRV=pdata_endRV, pdata_epi=pdata_epi, verbose=args.verbose) myVTK.writeSTL(pdata=clipped_pdata_endLV, filename="endLV.stl", verbose=args.verbose) myVTK.writeSTL(pdata=clipped_pdata_endRV, filename="endRV.stl", verbose=args.verbose) myVTK.writeSTL(pdata=clipped_pdata_epi, filename="epi.stl", verbose=args.verbose) myVTK.writeSTL(pdata=clipped_pdata_valM, filename="valM.stl", verbose=args.verbose) myVTK.writeSTL(pdata=clipped_pdata_valP, filename="valP.stl", verbose=args.verbose)
mesh = myvtk.readUGrid( filename=args.mesh_in_filename, verbose=args.verbose) elif (args.mesh_in_filename.endswith(".vtp") or args.mesh_in_filename.endswith(".stl")): mesh = myvtk.readPData( filename=args.mesh_in_filename, verbose=args.verbose) else: assert (0), "Input file must be .vtk, .vtu, or .stl. Aborting." myvtk.moveMeshWithWorldMatrix( mesh=mesh, M=I2I, verbose=args.verbose) if (args.mesh_out_filename == None): args.mesh_out_filename = args.mesh_in_filename.replace(args.image_in+"Coords", args.image_out+"Coords") if (args.mesh_out_filename.endswith(".vtk") or args.mesh_out_filename.endswith(".vtu")): myvtk.writeUGrid( ugrid=mesh, filename=args.mesh_out_filename, verbose=args.verbose) elif (args.mesh_out_filename.endswith(".stl")): myvtk.writeSTL( pdata=mesh, filename=args.mesh_out_filename, verbose=args.verbose) else: assert (0), "Output file must be .vtk, .vtu, or .stl. Aborting."