def test(self): tmpdir = vtkGetTempDir() fname = tmpdir + "/testxmlpartitioneddatasetcollection.vtpc" source = vtkPartitionedDataSetCollectionSource() writer = vtkXMLPartitionedDataSetCollectionWriter() writer.SetInputConnection(source.GetOutputPort()) writer.SetFileName(fname) writer.Write() reader = vtkXMLPartitionedDataSetCollectionReader() reader.SetFileName(fname) reader.Update() validData = source.GetOutputDataObject(0) data = reader.GetOutputDataObject(0) assert data.IsA("vtkPartitionedDataSetCollection") assert validData.GetNumberOfPartitionedDataSets() == data.GetNumberOfPartitionedDataSets() for i in range(validData.GetNumberOfPartitionedDataSets()): assert validData.GetNumberOfPartitions(i) == data.GetNumberOfPartitions(i) assert validData.GetMetaData(i).Get(validData.NAME()) == data.GetMetaData(i).Get(data.NAME()) assert validData.GetDataAssembly() is not None and data.GetDataAssembly() is not None os.remove(fname) shutil.rmtree(tmpdir + "/testxmlpartitioneddatasetcollection")
def test(self): p = dm.vtkPartitionedDataSet() s = ic.vtkRTAnalyticSource() s.SetWholeExtent(0, 10, 0, 10, 0, 5) s.Update() p1 = dm.vtkImageData() p1.ShallowCopy(s.GetOutput()) s.SetWholeExtent(0, 10, 0, 10, 5, 10) s.Update() p2 = dm.vtkImageData() p2.ShallowCopy(s.GetOutput()) p.SetPartition(0, p1) p.SetPartition(1, p2) p2 = dm.vtkPartitionedDataSet() p2.ShallowCopy(p) c = dm.vtkPartitionedDataSetCollection() c.SetPartitionedDataSet(0, p) c.SetPartitionedDataSet(1, p2) tmpdir = vtkGetTempDir() fname = tmpdir + "/testcompowriread.vtk" w = il.vtkCompositeDataWriter() w.SetInputData(c) w.SetFileName(fname) w.Write() r = il.vtkCompositeDataReader() r.SetFileName(fname) r.Update() o = r.GetOutputDataObject(0) self.assertTrue(o.IsA("vtkPartitionedDataSetCollection")) nd = o.GetNumberOfPartitionedDataSets() self.assertEqual(nd, 2) for i in range(nd): p = o.GetPartitionedDataSet(i) p2 = c.GetPartitionedDataSet(i) self.assertTrue(p.IsA("vtkPartitionedDataSet")) self.assertEqual(p.GetNumberOfPartitions(), 2) self.assertEqual( p.GetPartition(0).GetNumberOfCells(), p.GetPartition(0).GetNumberOfCells()) del (r) import gc gc.collect() os.remove(fname)
def test(self): p = dm.vtkPartitionedDataSet() s = ic.vtkRTAnalyticSource() s.SetWholeExtent(0, 10, 0, 10, 0, 5) s.Update() p1 = dm.vtkImageData() p1.ShallowCopy(s.GetOutput()) s.SetWholeExtent(0, 10, 0, 10, 5, 10) s.Update() p2 = dm.vtkImageData() p2.ShallowCopy(s.GetOutput()) p.SetPartition(0, p1) p.SetPartition(1, p2) p2 = dm.vtkPartitionedDataSet() p2.ShallowCopy(p) c = dm.vtkPartitionedDataSetCollection() c.SetPartitionedDataSet(0, p) c.SetPartitionedDataSet(1, p2) tmpdir = vtkGetTempDir() fname = tmpdir+"/testcompowriread.vtk" w = il.vtkCompositeDataWriter() w.SetInputData(c) w.SetFileName(fname) w.Write() r = il.vtkCompositeDataReader() r.SetFileName(fname) r.Update() o = r.GetOutputDataObject(0) self.assertTrue(o.IsA("vtkPartitionedDataSetCollection")) nd = o.GetNumberOfPartitionedDataSets() self.assertEqual(nd, 2) for i in range(nd): p = o.GetPartitionedDataSet(i) p2 = c.GetPartitionedDataSet(i) self.assertTrue(p.IsA("vtkPartitionedDataSet")) self.assertEqual(p.GetNumberOfPartitions(), 2) self.assertEqual(p.GetPartition(0).GetNumberOfCells(), p.GetPartition(0).GetNumberOfCells()) del(r) import gc gc.collect() os.remove(fname)
def test(self): p = dm.vtkPartitionedDataSet() s = ic.vtkRTAnalyticSource() s.SetWholeExtent(0, 10, 0, 10, 0, 5) s.Update() p1 = dm.vtkImageData() p1.ShallowCopy(s.GetOutput()) s.SetWholeExtent(0, 10, 0, 10, 5, 10) s.Update() p2 = dm.vtkImageData() p2.ShallowCopy(s.GetOutput()) p.SetPartition(0, p1) p.SetPartition(1, p2) tmpdir = vtkGetTempDir() fname = tmpdir+"/testxmlpartds.vtpd" w = ixml.vtkXMLPartitionedDataSetWriter() w.SetInputData(p) w.SetFileName(fname) w.Write() r = ixml.vtkXMLPartitionedDataSetReader() r.SetFileName(fname) r.Update() o = r.GetOutputDataObject(0) print(o.IsA("vtkPartitionedDataSet")) np = o.GetNumberOfPartitions() self.assertEqual(np, 2) for i in range(np): d = o.GetPartition(i) d2 = p.GetPartition(i) self.assertTrue(d.IsA("vtkImageData")) self.assertEqual(d.GetNumberOfCells(), d2.GetNumberOfCells()) os.remove(fname)
#!/usr/bin/env python import os import vtk from vtk.util.misc import vtkGetDataRoot from vtk.util.misc import vtkGetTempDir VTK_DATA_ROOT = vtkGetDataRoot() VTK_TEMP_DIR = vtkGetTempDir() file0 = VTK_TEMP_DIR + '/idFile0.vtp' file1 = VTK_TEMP_DIR + '/idFile1.vtp' file2 = VTK_TEMP_DIR + '/idFile2.vtp' # read in some poly data pdReader = vtk.vtkPolyDataReader() pdReader.SetFileName(VTK_DATA_ROOT + "/Data/fran_cut.vtk") pdReader.Update() extract = vtk.vtkExtractPolyDataPiece() extract.SetInputConnection(pdReader.GetOutputPort()) # write various versions pdWriter = vtk.vtkXMLPolyDataWriter() pdWriter.SetFileName(file0) pdWriter.SetDataModeToAscii() pdWriter.SetInputConnection(pdReader.GetOutputPort()) pdWriter.Write() pdWriter.SetFileName(file1) pdWriter.SetInputConnection(extract.GetOutputPort())
#!/usr/bin/env python import os import vtk from vtk.util.misc import vtkGetDataRoot from vtk.util.misc import vtkGetTempDir VTK_DATA_ROOT = vtkGetDataRoot() VTK_TEMP_DIR = vtkGetTempDir() rt = vtk.vtkRTAnalyticSource() rt.Update() inp = rt.GetOutput() inp.GetInformation().Set(vtk.vtkDataObject.DATA_TIME_STEP(), 11) file_root = VTK_TEMP_DIR + '/testxmlfield' file0 = file_root + ".vti" w = vtk.vtkXMLImageDataWriter() w.SetInputData(inp) w.SetFileName(file0) w.Write() r = vtk.vtkXMLImageDataReader() r.SetFileName(file0) r.UpdateInformation() assert(r.GetOutputInformation(0).Get(vtk.vtkStreamingDemandDrivenPipeline.TIME_STEPS(), 0) == 11.0) os.remove(file0)
def test(self): mdgTree = dm.vtkMutableDirectedGraph() mug = dm.vtkMutableUndirectedGraph() for i in range(10): mdgTree.AddVertex() mug.AddVertex() mdgTree.AddEdge(0, 1) mdgTree.AddEdge(0, 2) mdgTree.AddEdge(0, 3) mdgTree.AddEdge(1, 4) mdgTree.AddEdge(1, 5) mdgTree.AddEdge(2, 6) mdgTree.AddEdge(2, 7) mdgTree.AddEdge(3, 8) mdgTree.AddEdge(3, 9) # Undirected graph with parallel edges # and self-loops. mug.AddEdge(0, 0) mug.AddEdge(0, 1) mug.AddEdge(1, 0) mug.AddEdge(1, 2) mug.AddEdge(1, 3) mug.AddEdge(4, 5) mug.AddEdge(4, 5) mug.AddEdge(6, 7) mug.AddEdge(7, 7) w = il.vtkGraphWriter() tmpdir = vtkGetTempDir() fname = tmpdir + "/testgrapwriread.vtk" w.SetFileName(fname) w.SetInputData(mdgTree) w.Write() r = il.vtkGraphReader() r.SetFileName(fname) r.Update() g = r.GetOutput() self.assertTrue(g.IsA("vtkDirectedGraph")) self.assertTrue(g.GetNumberOfVertices() == 10) self.assertTrue(g.GetNumberOfEdges() == 9) self.assertTrue(g.GetTargetVertex(0) == 1) del (r) import gc gc.collect() os.remove(fname) w.SetFileName(fname) w.SetInputData(mug) w.Write() r = il.vtkGraphReader() r.SetFileName(fname) r.Update() g = r.GetOutput() self.assertTrue(g.IsA("vtkUndirectedGraph")) self.assertTrue(g.GetNumberOfVertices() == 10) self.assertTrue(g.GetNumberOfEdges() == 9) self.assertTrue(g.GetTargetVertex(0) == 0) del (r) import gc gc.collect() os.remove(fname)
def test(self): mdgTree = dm.vtkMutableDirectedGraph() mug = dm.vtkMutableUndirectedGraph() for i in range(10): mdgTree.AddVertex() mug.AddVertex() mdgTree.AddEdge(0, 1) mdgTree.AddEdge(0, 2) mdgTree.AddEdge(0, 3) mdgTree.AddEdge(1, 4) mdgTree.AddEdge(1, 5) mdgTree.AddEdge(2, 6) mdgTree.AddEdge(2, 7) mdgTree.AddEdge(3, 8) mdgTree.AddEdge(3, 9) # Undirected graph with parallel edges # and self-loops. mug.AddEdge(0, 0) mug.AddEdge(0, 1) mug.AddEdge(1, 0) mug.AddEdge(1, 2) mug.AddEdge(1, 3) mug.AddEdge(4, 5) mug.AddEdge(4, 5) mug.AddEdge(6, 7) mug.AddEdge(7, 7) w = il.vtkGraphWriter() tmpdir = vtkGetTempDir() fname = tmpdir+"/testgrapwriread.vtk" w.SetFileName(fname) w.SetInputData(mdgTree) w.Write() r = il.vtkGraphReader() r.SetFileName(fname) r.Update() g = r.GetOutput() self.assertTrue(g.IsA("vtkDirectedGraph")) self.assertTrue(g.GetNumberOfVertices() == 10) self.assertTrue(g.GetNumberOfEdges() == 9) self.assertTrue(g.GetTargetVertex(0) == 1) del(r) import gc gc.collect() os.remove(fname) w.SetFileName(fname) w.SetInputData(mug) w.Write() r = il.vtkGraphReader() r.SetFileName(fname) r.Update() g = r.GetOutput() self.assertTrue(g.IsA("vtkUndirectedGraph")) self.assertTrue(g.GetNumberOfVertices() == 10) self.assertTrue(g.GetNumberOfEdges() == 9) self.assertTrue(g.GetTargetVertex(0) == 0) del(r) import gc gc.collect() os.remove(fname)
out.SetMask(outMask) return 1 myAlgo = MyAlgorithm() myAlgo.SetSelectValue(1) # 1 or -1 ex = vtk.vtkPythonAlgorithm() ex.SetPythonObject(myAlgo) ex.SetInputConnection(depth.GetOutputPort()) ex.Update() from vtk.util.misc import vtkGetTempDir nrep = vtkGetTempDir() withAscii = False if withAscii: filename = nrep + 'toto_1_ascii.htg' else: filename = nrep + 'toto_1_binary.htg' #Avant writer default (1.0) writer = vtk.vtkXMLHyperTreeGridWriter() writer.SetInputConnection(ex.GetOutputPort()) writer.SetFileName(filename) #The default is in appended, data at the end of the file if withAscii: writer.SetDataModeToAscii() writer.Write() #print('Write Defaut(1.0) '+filename)