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)
