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()