def write(obj, fileoutput): ''' Write 3D object to file. Possile extensions are: .vtk, .ply, .obj, .stl, .byu, .vtp ''' fr = fileoutput.lower() if '.vtk' in fr: w = vtk.vtkPolyDataWriter() elif '.ply' in fr: w = vtk.vtkPLYWriter() elif '.obj' in fr: w = vtk.vtkOBJExporter() w.SetFilePrefix(fileoutput.replace('.obj', '')) vc.printc('Please use write(vp.renderWin)', 3) w.SetInput(obj) w.Update() vc.printc("Saved file: " + fileoutput, 'g') return elif '.stl' in fr: w = vtk.vtkSTLWriter() elif '.byu' in fr or '.g' in fr: w = vtk.vtkBYUWriter() elif '.vtp' in fr: w = vtk.vtkXMLPolyDataWriter() else: vc.printc('Unavailable format in file ' + fileoutput, c='r') exit(1) try: vu.setInput(w, vu.polydata(obj, True)) w.SetFileName(fileoutput) w.Write() vc.printc("Saved file: " + fileoutput, 'g') except: vc.printc("Error saving: " + fileoutput, 'r')
def __init__(self, module_manager): SimpleVTKClassModuleBase.__init__( self, module_manager, vtk.vtkBYUWriter(), 'Writing vtkBYU.', ('vtkBYU',), (), replaceDoc=True, inputFunctions=None, outputFunctions=None)
def __init__(self, module_manager): SimpleVTKClassModuleBase.__init__(self, module_manager, vtk.vtkBYUWriter(), 'Writing vtkBYU.', ('vtkBYU', ), (), replaceDoc=True, inputFunctions=None, outputFunctions=None)
def write(obj, fileoutput): ''' Write 3D object to file. Possile extensions are: .vtk, .ply, .obj, .stl, .byu, .vtp, .xyz, .tif ''' fr = fileoutput.lower() if '.vtk' in fr: w = vtk.vtkPolyDataWriter() elif '.ply' in fr: w = vtk.vtkPLYWriter() elif '.stl' in fr: w = vtk.vtkSTLWriter() elif '.vtp' in fr: w = vtk.vtkXMLPolyDataWriter() elif '.xyz' in fr: w = vtk.vtkSimplePointsWriter() elif '.byu' in fr or fr.endswith('.g'): w = vtk.vtkBYUWriter() elif '.obj' in fr: obj = vu.polydata(obj, True) w = vtk.vtkOBJExporter() w.SetFilePrefix(fileoutput.replace('.obj', '')) vc.printc('Please use write(vp.renderWin)', c=3) w.SetInput(obj) w.Update() vc.printc("Saved file: " + fileoutput, c='g') return elif '.tif' in fr: wr = vtk.vtkTIFFWriter() wr.SetFileDimensionality(len(obj.GetDimensions())) vu.setInput(wr, obj) wr.SetFileName(fileoutput) wr.Write() vc.printc("TIFF stack saved as: " + fileoutput, c='g') return else: vc.printc('Unavailable format in file ' + fileoutput, c='r') exit(1) try: obj = vu.polydata(obj, True) vu.setInput(w, vu.polydata(obj, True)) w.SetFileName(fileoutput) w.Write() vc.printc("Saved file: " + fileoutput, c='g') except: vc.printc("Error saving: " + fileoutput, c='r')
iv2 = vtk.vtkIVWriter() iv2.SetInputConnection(triangles.GetOutputPort()) iv2.SetFileName("brain2.iv") iv2.Write() file.delete("-force", "brain2.iv") pass if (info.command(globals(), locals(), "vtkIVWriter") != ""): edges = vtk.vtkExtractEdges() edges.SetInputConnection(triangles.GetOutputPort()) iv3 = vtk.vtkIVWriter() iv3.SetInputConnection(edges.GetOutputPort()) iv3.SetFileName("brain3.iv") iv3.Write() file.delete("-force", "brain3.iv") pass byu = vtk.vtkBYUWriter() byu.SetGeometryFileName("brain.g") byu.SetScalarFileName("brain.s") byu.SetDisplacementFileName("brain.d") byu.SetInputConnection(triangles.GetOutputPort()) byu.Write() file.delete("-force", "brain.g") file.delete("-force", "brain.s") file.delete("-force", "brain.d") mcubes = vtk.vtkMCubesWriter() mcubes.SetInputConnection(triangles.GetOutputPort()) mcubes.SetFileName("brain.tri") mcubes.SetLimitsFileName("brain.lim") mcubes.Write() file.delete("-force", "brain.lim") file.delete("-force", "brain.tri")
def write(obj, fileoutput, binary=True): ''' Write 3D object to file. Possile extensions are: vtk, vti, ply, obj, stl, byu, vtp, xyz, tif, png, bmp ''' if isinstance(obj, Actor): obj = obj.polydata(True) elif isinstance(obj, vtk.vtkActor): obj = obj.GetMapper().GetInput() fr = fileoutput.lower() if '.vtk' in fr: w = vtk.vtkPolyDataWriter() elif '.ply' in fr: w = vtk.vtkPLYWriter() elif '.stl' in fr: w = vtk.vtkSTLWriter() elif '.vtp' in fr: w = vtk.vtkXMLPolyDataWriter() elif '.xyz' in fr: w = vtk.vtkSimplePointsWriter() elif '.byu' in fr or fr.endswith('.g'): w = vtk.vtkBYUWriter() elif '.obj' in fr: obj = obj.polydata(True) w = vtk.vtkOBJExporter() w.SetFilePrefix(fileoutput.replace('.obj', '')) colors.printc('Please use write(vp.renderWin)', c=3) w.SetInputData(obj) w.Update() colors.printc("Saved file: " + fileoutput, c='g') return elif '.tif' in fr: w = vtk.vtkTIFFWriter() w.SetFileDimensionality(len(obj.GetDimensions())) elif '.vti' in fr: w = vtk.vtkXMLImageDataWriter() elif '.png' in fr: w = vtk.vtkPNGWriter() elif '.jpg' in fr: w = vtk.vtkJPEGWriter() elif '.bmp' in fr: w = vtk.vtkBMPWriter() else: colors.printc('Unknown format', fileoutput, 'file not saved.', c='r') return try: if binary: w.SetFileTypeToBinary() else: w.SetFileTypeToASCII() w.SetInputData(obj) w.SetFileName(fileoutput) w.Write() colors.printc("Saved file: " + fileoutput, c='g') except Exception as e: colors.printc("Error saving: " + fileoutput, '\n', e, c='r') return
polyData_t.Modified() outputPath_idx = regression_output_folder_path + "CMRep_EGReg_" + anatomy + "_Subjects_" + subj_i + "_t_" + str( age_i) + ".vtk" writer_t = vtk.vtkPolyDataWriter() writer_t.SetFileName(outputPath_idx) writer_t.SetInputData(polyData_t) writer_t.Update() writer_t.Write() byuOutputPath_idx = regression_output_folder_path + "CMRep_EGReg_" + anatomy + "_Subjects_" + subj_i + "_t_" + str( age_i) + ".byu" byuWriter_t = vtk.vtkBYUWriter() byuWriter_t.SetGeometryFileName(byuOutputPath_idx) byuWriter_t.SetInputData(polyData_t) byuWriter_t.Update() byuWriter_t.Write() stlOutputPath_idx = regression_output_folder_path + "CMRep_EGReg_" + anatomy + "_Subjects_" + subj_i + "_t_" + str( age_i) + ".stl" stlWriter_t = vtk.vtkSTLWriter() stlWriter_t.SetFileName(stlOutputPath_idx) stlWriter_t.SetInputData(polyData_t) stlWriter_t.Update() stlWriter_t.Write() # for n in range( nTimePt ):
est_vtk_i.Modified() filePath_est_i = outFolderPath + "Surface_ShapeWork_CTRL_Complex_" + anatomy + "_LinearModel_" + subject_i + "_" + str( age_i) + ".vtk" writer_i = vtk.vtkPolyDataWriter() writer_i.SetFileName(filePath_est_i) writer_i.SetInputData(est_vtk_i) writer_i.Update() writer_i.Write() byuFilePath_est_i = outFolderPath + "Surface_ShapeWork_CTRL_Complex_" + anatomy + "_LinearModel_" + subject_i + "_" + str( age_i) + ".byu" byuWriter_i = vtk.vtkBYUWriter() byuWriter_i.SetGeometryFileName(byuFilePath_est_i) byuWriter_i.SetInputData(est_vtk_i) byuWriter_i.Update() byuWriter_i.Write() stlFilePath_est_i = outFolderPath + "Surface_ShapeWork_CTRL_Complex_" + anatomy + "_LinearModel_" + subject_i + "_" + str( age_i) + ".stl" stlWriter_i = vtk.vtkSTLWriter() stlWriter_i.SetFileName(stlFilePath_est_i) stlWriter_i.SetInputData(est_vtk_i) stlWriter_i.Update() stlWriter_i.Write() # plt.show()