vtkXMLPMultiBlockDataWriter.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 vtkXMLPMultiBlockDataWriter_h
#define vtkXMLPMultiBlockDataWriter_h

#include "vtkIOParallelXMLModule.h" // For export macro
#include "vtkXMLMultiBlockDataWriter.h"

VTK_ABI_NAMESPACE_BEGIN
class vtkCompositeDataSet;
class vtkMultiProcessController;

class VTKIOPARALLELXML_EXPORT vtkXMLPMultiBlockDataWriter : public vtkXMLMultiBlockDataWriter
{
public:
  static vtkXMLPMultiBlockDataWriter* New();
  vtkTypeMacro(vtkXMLPMultiBlockDataWriter, vtkXMLMultiBlockDataWriter);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkSetMacro(NumberOfPieces, int);
  vtkGetMacro(NumberOfPieces, int);


  vtkSetMacro(StartPiece, int);
  vtkGetMacro(StartPiece, int);


  virtual void SetController(vtkMultiProcessController*);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);

  void SetWriteMetaFile(int flag) override;

  // See the vtkAlgorithm for a description of what these do
  vtkTypeBool ProcessRequest(
    vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;

protected:
  vtkXMLPMultiBlockDataWriter();
  ~vtkXMLPMultiBlockDataWriter() override;

  void FillDataTypes(vtkCompositeDataSet*) override;

  vtkMultiProcessController* Controller;

  int WriteComposite(
    vtkCompositeDataSet* compositeData, vtkXMLDataElement* parent, int& currentFileIndex) override;

  int ParallelWriteNonCompositeData(
    vtkDataObject* dObj, vtkXMLDataElement* parentXML, int currentFileIndex);

  virtual vtkStdString CreatePieceFileName(int currentFileIndex, int procId, int dataSetType);

  void MakeDirectory(const char* name) override;

  void RemoveWrittenFiles(const char* subDirectory) override;


  int StartPiece;
  int NumberOfPieces;

private:
  vtkXMLPMultiBlockDataWriter(const vtkXMLPMultiBlockDataWriter&) = delete;
  void operator=(const vtkXMLPMultiBlockDataWriter&) = delete;

  class vtkInternal;
  vtkInternal* XMLPMultiBlockDataWriterInternal;
};

VTK_ABI_NAMESPACE_END
#endif