vtkEnSightReader.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 vtkEnSightReader_h
#define vtkEnSightReader_h
 
#include "vtkGenericEnSightReader.h"
#include "vtkIOEnSightModule.h" // For export macro
#include "vtkSmartPointer.h"    // for vtkSmartPointer
 
#include <map>    // for std::map
#include <vector> // for std::vector
 
VTK_ABI_NAMESPACE_BEGIN
class vtkDataSet;
class vtkDataSetCollection;
class vtkDoubleArray;
class vtkEnSightReaderCellIdsType;
class vtkIdList;
class vtkMultiBlockDataSet;
class vtkTransform;
 
class VTKIOENSIGHT_EXPORT vtkEnSightReader : public vtkGenericEnSightReader
{
public:
  vtkTypeMacro(vtkEnSightReader, vtkGenericEnSightReader);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  enum ElementTypesList
  {
    POINT = 0,
    BAR2 = 1,
    BAR3 = 2,
    NSIDED = 3,
    TRIA3 = 4,
    TRIA6 = 5,
    QUAD4 = 6,
    QUAD8 = 7,
    NFACED = 8,
    TETRA4 = 9,
    TETRA10 = 10,
    PYRAMID5 = 11,
    PYRAMID13 = 12,
    HEXA8 = 13,
    HEXA20 = 14,
    PENTA6 = 15,
    PENTA15 = 16,
    NUMBER_OF_ELEMENT_TYPES = 17
  };
 
  enum VariableTypesList
  {
    SCALAR_PER_NODE = 0,
    VECTOR_PER_NODE = 1,
    TENSOR_SYMM_PER_NODE = 2,
    SCALAR_PER_ELEMENT = 3,
    VECTOR_PER_ELEMENT = 4,
    TENSOR_SYMM_PER_ELEMENT = 5,
    SCALAR_PER_MEASURED_NODE = 6,
    VECTOR_PER_MEASURED_NODE = 7,
    COMPLEX_SCALAR_PER_NODE = 8,
    COMPLEX_VECTOR_PER_NODE = 9,
    COMPLEX_SCALAR_PER_ELEMENT = 10,
    COMPLEX_VECTOR_PER_ELEMENT = 11,
    TENSOR_ASYM_PER_NODE = 12,
    TENSOR_ASYM_PER_ELEMENT = 13
  };
 
  enum SectionTypeList
  {
    COORDINATES = 0,
    BLOCK = 1,
    ELEMENT = 2
  };
 
 
  vtkGetFilePathMacro(MeasuredFileName);
 
 
  vtkGetFilePathMacro(MatchFileName);
 
 
  vtkGetFilePathMacro(RigidBodyFileName);
 
protected:
  vtkEnSightReader();
  ~vtkEnSightReader() override;
 
  int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
 
  void ClearForNewCaseFileName() override;
 
 
  vtkSetFilePathMacro(MeasuredFileName);
 
 
  vtkSetFilePathMacro(MatchFileName);
 
 
  vtkSetFilePathMacro(RigidBodyFileName);
 
 
  int ReadCaseFile();
  int ReadCaseFileGeometry(char* line);
  int ReadCaseFileVariable(char* line);
  int ReadCaseFileTime(char* line);
  int ReadCaseFileFile(char* line);
  int ReadCaseFileScripts(char* line);
 
 
  // set in UpdateInformation to value returned from ReadCaseFile
  int CaseFileRead;
 
  virtual int ReadGeometryFile(
    const char* fileName, int timeStep, vtkMultiBlockDataSet* output) = 0;
 
  virtual int ReadMeasuredGeometryFile(
    const char* fileName, int timeStep, vtkMultiBlockDataSet* output) = 0;
 
  int ReadRigidBodyGeometryFile();
 
  int ReadRigidBodyEulerParameterFile(const char* path);
 
  int ReadRigidBodyMatrixLines(char* line, vtkTransform* transform, bool& applyToVectors);
 
  int ApplyRigidBodyTransforms(int partId, const char* name, vtkDataSet* output);
 
  int ReadVariableFiles(vtkMultiBlockDataSet* output);
 
  virtual int ReadScalarsPerNode(const char* fileName, const char* description, int timeStep,
    vtkMultiBlockDataSet* output, int measured = 0, int numberOfComponents = 1,
    int component = 0) = 0;
 
  virtual int ReadVectorsPerNode(const char* fileName, const char* description, int timeStep,
    vtkMultiBlockDataSet* output, int measured = 0) = 0;
 
  virtual int ReadAsymmetricTensorsPerNode(
    const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet* output) = 0;
 
  virtual int ReadTensorsPerNode(
    const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet* output) = 0;
 
  virtual int ReadScalarsPerElement(const char* fileName, const char* description, int timeStep,
    vtkMultiBlockDataSet* output, int numberOfComponents = 1, int component = 0) = 0;
 
  virtual int ReadVectorsPerElement(
    const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet* output) = 0;
 
  virtual int ReadAsymmetricTensorsPerElement(
    const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet* output) = 0;
 
  virtual int ReadTensorsPerElement(
    const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet* output) = 0;
 
  virtual int CreateUnstructuredGridOutput(
    int partId, char line[80], const char* name, vtkMultiBlockDataSet* output) = 0;
 
  virtual int CreateStructuredGridOutput(
    int partId, char line[80], const char* name, vtkMultiBlockDataSet* output) = 0;
 
  void AddVariableFileName(const char* fileName1, const char* fileName2 = nullptr);
 
  void AddVariableDescription(const char* description);
 
  void AddVariableType();
 
  int GetElementType(const char* line);
 
  int GetSectionType(const char* line);
 
  void RemoveLeadingBlanks(char* line);
 
  // Get the vtkIdList for the given output index and cell type.
  vtkIdList* GetCellIds(int index, int cellType);
 
  void AddToBlock(vtkMultiBlockDataSet* output, unsigned int blockNo, vtkDataSet* dataset);
 
  vtkDataSet* GetDataSetFromBlock(vtkMultiBlockDataSet* output, unsigned int blockNo);
 
  void SetBlockName(vtkMultiBlockDataSet* output, unsigned int blockNo, const char* name);
 
  char* MeasuredFileName;
  char* MatchFileName; // may not actually be necessary to read this file
  char* RigidBodyFileName;
 
  // pointer to lists of vtkIdLists (cell ids per element type per part)
  vtkEnSightReaderCellIdsType* CellIds;
 
  // part ids of unstructured outputs
  vtkIdList* UnstructuredPartIds;
 
  int VariableMode;
 
  // pointers to lists of filenames
  char** VariableFileNames; // non-complex
  char** ComplexVariableFileNames;
 
  // array of time sets
  vtkIdList* VariableTimeSetIds;
  vtkIdList* ComplexVariableTimeSetIds;
 
  // array of file sets
  vtkIdList* VariableFileSetIds;
  vtkIdList* ComplexVariableFileSetIds;
 
  // collection of filename numbers per time set
  vtkIdListCollection* TimeSetFileNameNumbers;
  vtkIdList* TimeSetsWithFilenameNumbers;
 
  // collection of filename numbers per file set
  vtkIdListCollection* FileSetFileNameNumbers;
  vtkIdList* FileSetsWithFilenameNumbers;
 
  // collection of number of steps per file per file set
  vtkIdListCollection* FileSetNumberOfSteps;
 
  // ids of the time and file sets
  vtkIdList* TimeSetIds;
  vtkIdList* FileSets;
 
  int GeometryTimeSet;
  int GeometryFileSet;
  int MeasuredTimeSet;
  int MeasuredFileSet;
 
  double GeometryTimeValue;
  double MeasuredTimeValue;
 
  vtkTypeBool UseTimeSets;
  vtkSetMacro(UseTimeSets, vtkTypeBool);
  vtkGetMacro(UseTimeSets, vtkTypeBool);
  vtkBooleanMacro(UseTimeSets, vtkTypeBool);
 
  vtkTypeBool UseFileSets;
  vtkSetMacro(UseFileSets, vtkTypeBool);
  vtkGetMacro(UseFileSets, vtkTypeBool);
  vtkBooleanMacro(UseFileSets, vtkTypeBool);
 
  int NumberOfGeometryParts;
 
  // global list of points for measured geometry
  int NumberOfMeasuredPoints;
 
  int NumberOfNewOutputs;
  int InitialRead;
 
  int CheckOutputConsistency();
 
  double ActualTimeValue;
 
  // We support only version 2 of rigid body transform files for only ensight gold files,
  // but it's implemented here, so we don't need to duplicate implementation for ASCII
  // and binary readers (the erb and eet files are always in ASCII).
  // For rigid body transforms, we need to track per part:
  // 1. transforms to be applied before the Euler transformation
  // 2. Information about which data to use in the Euler Transform file (eet file)
  // 3. transforms to be applied after the Euler transformation
  struct PartTransforms
  {
    // Pre and post transforms do not change over time
    // We have to track each transform separately, because some transforms need to be
    // applied to geometry and vectors, while others should only be applied to the geometry
    std::vector<vtkSmartPointer<vtkTransform>> PreTransforms;
    std::vector<bool> PreTransformsApplyToVectors;
    std::vector<vtkSmartPointer<vtkTransform>> PostTransforms;
    std::vector<bool> PostTransformsApplyToVectors;
 
    // EnSight format requires specifying the eet file per part, but according to the user manual
    // use of different eet files for the same dataset is not actually allowed
    std::string EETFilename;
 
    // title is related to, but not necessarily a part name. for instance, if you have 4 wheel parts
    // there may only be a single "wheel" title that all wheel parts use, applying the same Euler
    // rotation to all wheels
    std::string EETTransTitle;
  };
 
  // rigid body files allows for using either part names or part Ids to specify
  // transforms for parts;
  bool UsePartNamesRB;
 
  // keeps track of all transforms for each part
  // if UsePartNamesRB == true, the key is the part name
  // otherwise, the key name is the partId converted to a string
  std::map<std::string, PartTransforms> RigidBodyTransforms;
 
  // map time step to the Euler transform for a part
  using TimeToEulerTransMapType = std::map<double, vtkSmartPointer<vtkTransform>>;
 
  // map a title to all of its Euler transforms
  using TitleToTimeStepMapType = std::map<std::string, TimeToEulerTransMapType>;
 
  TitleToTimeStepMapType EulerTransformsMap;
 
  // It's possible for an EnSight dataset to not contain transient data, except for the
  // Euler transforms. In this case, we will populate EulerTimeSteps so we can use it for
  // time information, instead of the usual time set
  bool UseEulerTimeSteps;
  vtkSmartPointer<vtkDoubleArray> EulerTimeSteps;
 
private:
  vtkEnSightReader(const vtkEnSightReader&) = delete;
  void operator=(const vtkEnSightReader&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif