def __init__(self, module_manager):
SimpleVTKClassModuleBase.__init__(self,
module_manager,
vtk.vtkXMLPRectilinearGridWriter(),
'Writing vtkXMLPRectilinearGrid.',
('vtkXMLPRectilinearGrid', ), (),
replaceDoc=True,
inputFunctions=None,
outputFunctions=None)
def __init__(self, module_manager):
SimpleVTKClassModuleBase.__init__(
self,
module_manager,
vtk.vtkXMLPRectilinearGridWriter(),
"Writing vtkXMLPRectilinearGrid.",
("vtkXMLPRectilinearGrid",),
(),
replaceDoc=True,
inputFunctions=None,
outputFunctions=None,
)
import os
os.remove(filename)
for i in range(npieces):
if not useSubdir:
os.remove(VTK_TEMP_DIR + "/%s_%d.%s" % (dataType, i, ext))
else:
os.remove(VTK_TEMP_DIR + "/%s/%s_%d.%s" %
(dataType, dataType, i, ext))
assert da2.GetValue(0) == numTris
TestDataType('ImageData', vtk.vtkXMLPImageDataReader(),
vtk.vtkXMLPImageDataWriter(), 'vti', 4924)
TestDataType('RectilinearGrid', vtk.vtkXMLPRectilinearGridReader(),
vtk.vtkXMLPRectilinearGridWriter(), 'vtr', 4924)
TestDataType('StructuredGrid', vtk.vtkXMLPStructuredGridReader(),
vtk.vtkXMLPStructuredGridWriter(), 'vts', 4924)
TestDataType('UnstructuredGrid', vtk.vtkXMLPUnstructuredGridReader(),
vtk.vtkXMLPUnstructuredGridWriter(), 'vtu', 11856)
# Test writers with UseSubdirectory on
TestDataType('ImageData',
vtk.vtkXMLPImageDataReader(),
vtk.vtkXMLPImageDataWriter(),
'vti',
4924,
useSubdir=True)
TestDataType('RectilinearGrid',
vtk.vtkXMLPRectilinearGridReader(),
vtk.vtkXMLPRectilinearGridWriter(),
da.InsertNextValue(ntris)
da2 = vtk.vtkIntArray()
da2.SetNumberOfTuples(1)
contr.AllReduce(da, da2, vtk.vtkCommunicator.SUM_OP)
if rank == 0:
print(da2.GetValue(0))
import os
os.remove(filename)
for i in range(npieces):
if not useSubdir:
os.remove(VTK_TEMP_DIR + "/%s_%d.%s" % (dataType, i, ext))
else:
os.remove(VTK_TEMP_DIR + "/%s/%s_%d.%s" %(dataType, dataType, i, ext))
assert da2.GetValue(0) == numTris
TestDataType('ImageData', vtk.vtkXMLPImageDataReader(), vtk.vtkXMLPImageDataWriter(), 'vti', 4924)
TestDataType('RectilinearGrid', vtk.vtkXMLPRectilinearGridReader(), vtk.vtkXMLPRectilinearGridWriter(), 'vtr', 4924)
TestDataType('StructuredGrid', vtk.vtkXMLPStructuredGridReader(), vtk.vtkXMLPStructuredGridWriter(), 'vts', 4924)
TestDataType('UnstructuredGrid', vtk.vtkXMLPUnstructuredGridReader(), vtk.vtkXMLPUnstructuredGridWriter(), 'vtu', 11856)
TestDataType('Table', vtk.vtkXMLPTableReader(), vtk.vtkXMLPTableWriter(), 'vtt', 18522)
# Test writers with UseSubdirectory on
TestDataType('ImageData', vtk.vtkXMLPImageDataReader(), vtk.vtkXMLPImageDataWriter(), 'vti', 4924, useSubdir=True)
TestDataType('RectilinearGrid', vtk.vtkXMLPRectilinearGridReader(), vtk.vtkXMLPRectilinearGridWriter(), 'vtr', 4924, useSubdir=True)
TestDataType('StructuredGrid', vtk.vtkXMLPStructuredGridReader(), vtk.vtkXMLPStructuredGridWriter(), 'vts', 4924, useSubdir=True)
TestDataType('UnstructuredGrid', vtk.vtkXMLPUnstructuredGridReader(), vtk.vtkXMLPUnstructuredGridWriter(), 'vtu', 11856, useSubdir=True)
TestDataType('Table', vtk.vtkXMLPTableReader(), vtk.vtkXMLPTableWriter(), 'vtt', 18522, useSubdir=True)
#from vtk.util import numpy_support # NEW
#s=sliced.flatten() # NEW: need to keep copy of s, otherwise get segfault below, because the VTK module points to the actual data stored in the numpy array
#vtkVar=numpy_support.numpy_to_vtk(s) # NEW
vtkVar.SetName(var)
if centered == 'point':
grid.GetPointData().AddArray(vtkVar)
else:
grid.GetCellData().AddArray(vtkVar)
#grid = vtk.vtkStructuredGrid()
#sg = vtk.vtkXMLStructuredGridWriter()
# WRITING FILES
import os, os.path
outdir = movie_file.split('.')[0] + '.' + movie_file.split('.')[2] + '.vtk'
if not os.path.exists(outdir): os.mkdir(outdir)
if not os.path.exists(outdir + '/' + centered + '_centered'):
os.mkdir(outdir + '/' + centered + '_centered')
outvardir = outdir + '/' + centered + '_centered/' + var
if not os.path.exists(outvardir): os.mkdir(outvardir)
outfile = outvardir + '/' + movie_file + '.vtk'
sg = vtk.vtkXMLPRectilinearGridWriter()
sg.SetFileName(outfile)
sg.SetInput(grid)
sg.SetDataModeToBinary()
sg.Write()
cf.SetValue(0, 130)
cf.SetComputeNormals(0)
cf.SetComputeGradients(0)
cf.SetInputConnection(reader.GetOutputPort())
cf.UpdateInformation()
cf.GetOutputInformation(0).Set(vtk.vtkStreamingDemandDrivenPipeline.UPDATE_NUMBER_OF_PIECES(), nranks)
cf.GetOutputInformation(0).Set(vtk.vtkStreamingDemandDrivenPipeline.UPDATE_PIECE_NUMBER(), rank)
cf.Update()
ntris = cf.GetOutput().GetNumberOfCells()
da = vtk.vtkIntArray()
da.InsertNextValue(ntris)
da2 = vtk.vtkIntArray()
da2.SetNumberOfTuples(1)
contr.AllReduce(da, da2, vtk.vtkCommunicator.SUM_OP)
if rank == 0:
print(da2.GetValue(0))
import os
os.remove(filename)
for i in range(npieces):
os.remove(VTK_TEMP_DIR + "/%s_%d.%s" % (dataType, i, ext))
assert da2.GetValue(0) == numTris
TestDataType('ImageData', vtk.vtkXMLPImageDataReader(), vtk.vtkXMLPImageDataWriter(), 'vti', 4924)
TestDataType('RectilinearGrid', vtk.vtkXMLPRectilinearGridReader(), vtk.vtkXMLPRectilinearGridWriter(), 'vtr', 4924)
TestDataType('StructuredGrid', vtk.vtkXMLPStructuredGridReader(), vtk.vtkXMLPStructuredGridWriter(), 'vts', 4924)
TestDataType('UnstructuredGrid', vtk.vtkXMLPUnstructuredGridReader(), vtk.vtkXMLPUnstructuredGridWriter(), 'vtu', 11856)
