vtkPDataSetWriter.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkPDataSetWriter_h
#define vtkPDataSetWriter_h
 
#include "vtkDataSetWriter.h"
#include "vtkIOParallelModule.h" // For export macro
 
#include <map>    // for keeping track of extents
#include <vector> // for keeping track of extents
 
VTK_ABI_NAMESPACE_BEGIN
class vtkImageData;
class vtkRectilinearGrid;
class vtkStructuredGrid;
class vtkMultiProcessController;
 
class VTKIOPARALLEL_EXPORT vtkPDataSetWriter : public vtkDataSetWriter
{
public:
  void PrintSelf(ostream& os, vtkIndent indent) override;
  vtkTypeMacro(vtkPDataSetWriter, vtkDataSetWriter);
  static vtkPDataSetWriter* New();
 
  int Write() override;
 
 
  void SetNumberOfPieces(int num);
  vtkGetMacro(NumberOfPieces, int);
 
 
  vtkSetMacro(GhostLevel, int);
  vtkGetMacro(GhostLevel, int);
 
 
  vtkSetMacro(StartPiece, int);
  vtkGetMacro(StartPiece, int);
  vtkSetMacro(EndPiece, int);
  vtkGetMacro(EndPiece, int);
 
 
  vtkSetFilePathMacro(FilePattern);
  vtkGetFilePathMacro(FilePattern);
 
 
  vtkSetMacro(UseRelativeFileNames, vtkTypeBool);
  vtkGetMacro(UseRelativeFileNames, vtkTypeBool);
  vtkBooleanMacro(UseRelativeFileNames, vtkTypeBool);
 
 
  virtual void SetController(vtkMultiProcessController*);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);
 
protected:
  vtkPDataSetWriter();
  ~vtkPDataSetWriter() override;
 
  ostream* OpenFile();
  int WriteUnstructuredMetaData(
    vtkDataSet* input, char* root, char* str, size_t strSize, ostream* fptr);
  int WriteImageMetaData(vtkImageData* input, char* root, char* str, size_t strSize, ostream* fptr);
  int WriteRectilinearGridMetaData(
    vtkRectilinearGrid* input, char* root, char* str, size_t strSize, ostream* fptr);
  int WriteStructuredGridMetaData(
    vtkStructuredGrid* input, char* root, char* str, size_t strSize, ostream* fptr);
 
  int StartPiece;
  int EndPiece;
  int NumberOfPieces;
  int GhostLevel;
 
  vtkTypeBool UseRelativeFileNames;
 
  char* FilePattern;
 
  void DeleteFiles();
 
  typedef std::map<int, std::vector<int>> ExtentsType;
  ExtentsType Extents;
 
  vtkMultiProcessController* Controller;
 
private:
  vtkPDataSetWriter(const vtkPDataSetWriter&) = delete;
  void operator=(const vtkPDataSetWriter&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif