def write(self):
""" Write the staged vtkPolyData to a file."""
self.poly_data.SetPoints(self.pnts)
self.poly_data.Allocate(len(self.cell_ids))
for cell_id in self.cell_ids.values():
self.poly_data.InsertNextCell(vtk.VTK_LINE, cell_id)
writer = WRITER[vtk.VTK_POLY_DATA]()
writer.SetFileName(self.filename)
if Parallel.is_parallel():
writer.SetNumberOfPieces(Parallel.get_size())
writer.SetStartPiece(Parallel.get_rank())
writer.SetEndPiece(Parallel.get_rank())
if vtk.vtkVersion.GetVTKMajorVersion() <= 6:
writer.SetWriteSummaryFile(Parallel.get_rank() == 0)
else:
controller = vtk.vtkMPIController()
controller.SetCommunicator(
vtk.vtkMPICommunicator.GetWorldCommunicator())
writer.SetController(controller)
if vtk.vtkVersion.GetVTKMajorVersion() < 6:
writer.SetInput(self.poly_data)
else:
writer.SetInputData(self.poly_data)
writer.Write()
if Parallel.is_parallel():
make_subdirectory(self.filename)
def write_to_file(vtk_data, outfile):
""" Wrapper around the various VTK writer routines"""
writer = WRITER[vtk_data.GetDataObjectType()]()
writer.SetFileName(outfile)
if Parallel.is_parallel():
writer.SetNumberOfPieces(Parallel.get_size())
writer.SetStartPiece(Parallel.get_rank())
writer.SetEndPiece(Parallel.get_rank())
writer.SetWriteSummaryFile(Parallel.get_rank()==0)
if vtk.vtkVersion.GetVTKMajorVersion()<6:
writer.SetInput(vtk_data)
else:
writer.SetInputData(vtk_data)
writer.Write()
def write_to_file(vtk_data, outfile):
""" Wrapper around the various VTK writer routines"""
writer = WRITER[vtk_data.GetDataObjectType()]()
writer.SetFileName(outfile)
if Parallel.is_parallel():
writer.SetNumberOfPieces(Parallel.get_size())
writer.SetStartPiece(Parallel.get_rank())
writer.SetEndPiece(Parallel.get_rank())
if vtk.vtkVersion.GetVTKMajorVersion() <= 6:
writer.SetWriteSummaryFile(Parallel.get_rank() == 0)
else:
controller = vtk.vtkMPIController()
controller.SetCommunicator(
vtk.vtkMPICommunicator.GetWorldCommunicator())
writer.SetController(controller)
if vtk.vtkVersion.GetVTKMajorVersion() < 6:
writer.SetInput(vtk_data)
else:
writer.SetInputData(vtk_data)
writer.Write()
if Parallel.is_parallel():
make_subdirectory(outfile)
""" Test parallel execution."""
from particle_model import Parallel
import pytest
@pytest.mark.skipif(Parallel.get_size() > 1, reason='Serial test')
def test_serial():
assert not Parallel.is_parallel()
@pytest.mark.skipif(Parallel.get_size() == 1, reason='Parallel test')
def test_paralle():
assert Parallel.is_parallel()
""" Test parallel execution."""
from particle_model import Parallel
import pytest
@pytest.mark.skipif(Parallel.get_size() > 1, reason="Serial test")
def test_serial():
assert not Parallel.is_parallel()
@pytest.mark.skipif(Parallel.get_size() == 1, reason="Parallel test")
def test_paralle():
assert Parallel.is_parallel()
