vtkEnSightWriter.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright (c) Sandia Corporation
// SPDX-License-Identifier: BSD-3-Clause

 
#ifndef vtkEnSightWriter_h
#define vtkEnSightWriter_h
 
#include "vtkIOParallelModule.h" // For export macro
#include "vtkWriter.h"
 
VTK_ABI_NAMESPACE_BEGIN
class vtkUnstructuredGrid;
 
class VTKIOPARALLEL_EXPORT vtkEnSightWriter : public vtkWriter
{
 
public:
  vtkTypeMacro(vtkEnSightWriter, vtkWriter);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  static vtkEnSightWriter* New();


  vtkSetMacro(ProcessNumber, int);
  vtkGetMacro(ProcessNumber, int);


  vtkSetFilePathMacro(Path);
  vtkGetFilePathMacro(Path);


  vtkSetStringMacro(BaseName);
  vtkGetStringMacro(BaseName);


  vtkSetFilePathMacro(FileName);
  vtkGetFilePathMacro(FileName);


  vtkSetMacro(TimeStep, int);
  vtkGetMacro(TimeStep, int);


  vtkSetMacro(GhostLevel, int);
  vtkGetMacro(GhostLevel, int);


  vtkSetMacro(TransientGeometry, bool);
  vtkGetMacro(TransientGeometry, bool);


  vtkSetMacro(NumberOfBlocks, int);
  vtkGetMacro(NumberOfBlocks, int);


  vtkBooleanMacro(WriteNodeIDs, bool);
  vtkSetMacro(WriteNodeIDs, bool);
  vtkGetMacro(WriteNodeIDs, bool);


  vtkBooleanMacro(WriteElementIDs, bool);
  vtkSetMacro(WriteElementIDs, bool);
  vtkGetMacro(WriteElementIDs, bool);


  virtual void SetBlockIDs(int* val) { BlockIDs = val; }
  virtual int* GetBlockIDs() { return BlockIDs; }


  virtual void SetInputData(vtkUnstructuredGrid* input);
  virtual vtkUnstructuredGrid* GetInput();


  virtual void WriteCaseFile(int TotalTimeSteps);
  virtual void WriteSOSCaseFile(int NumProcs);
 
protected:
  vtkEnSightWriter();
  ~vtkEnSightWriter() override;
 
  int FillInputPortInformation(int port, vtkInformation* info) override;
  int RequestData(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
 
  bool WriteDataAndReturn()
    override; // method to allow this class to be instantiated and delegated to
 
  virtual void WriteStringToFile(const char* string, FILE* file);
  virtual void WriteTerminatedStringToFile(const char* string, FILE* file);
  virtual void WriteIntToFile(int i, FILE* file);
  virtual void WriteFloatToFile(float f, FILE* file);
  virtual void WriteElementTypeToFile(int ElementType, FILE* fd);
 
  virtual bool ShouldWriteGeometry();
  virtual void SanitizeFileName(char* name);
  virtual FILE* OpenFile(char* name);
 
  void ComputeNames();
  void DefaultNames();
 
  int GetExodusModelIndex(int* ElementArray, int NumberElements, int PartID);
 
  static int GetDestinationComponent(int srcComponent, int numComponents);
 
  char* Path;
  char* BaseName;
  char* FileName;
  int TimeStep;
  int GhostLevelMultiplier;
  int ProcessNumber;
  int NumberOfProcesses;
  int NumberOfBlocks;
  int* BlockIDs;
  bool TransientGeometry;
  int GhostLevel;
  bool WriteNodeIDs;
  bool WriteElementIDs;
  vtkUnstructuredGrid* TmpInput;
 
  vtkEnSightWriter(const vtkEnSightWriter&) = delete;
  void operator=(const vtkEnSightWriter&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif