vtkExodusIIReaderParser.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright (c) Kitware, Inc.
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkExodusIIReaderParser_h
#define vtkExodusIIReaderParser_h
 
#include "vtkIOExodusModule.h" // For export macro
#include "vtkSmartPointer.h"   // for ivars
#include "vtkXMLParser.h"
 
#include <map>    // for std::map
#include <set>    // for std::set
#include <string> // for std::string
#include <vector> // for std::vector
 
VTK_ABI_NAMESPACE_BEGIN
class vtkMutableDirectedGraph;
class vtkStringArray;
class vtkUnsignedCharArray;
 
class VTKIOEXODUS_EXPORT vtkExodusIIReaderParser : public vtkXMLParser
{
public:
  static vtkExodusIIReaderParser* New();
  vtkTypeMacro(vtkExodusIIReaderParser, vtkXMLParser);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
 
  vtkGetObjectMacro(SIL, vtkMutableDirectedGraph);
 
  void Go(VTK_FILEPATH const char* filename);
 
  // Returns if the parser has some information about the block with given "id".
  // This is valid only after Go().
  bool HasInformationAboutBlock(int id)
  {
    return (this->BlockID_To_VertexID.find(id) != this->BlockID_To_VertexID.end());
  }
 
  std::string GetBlockName(int id);
 
 
  void GetBlockIds(std::set<int>& blockIdsSet)
  {
    std::map<int, vtkIdType>::iterator iter;
    for (iter = this->BlockID_To_VertexID.begin(); iter != this->BlockID_To_VertexID.end(); ++iter)
    {
      blockIdsSet.insert(iter->first);
    }
  }
 
protected:
  vtkExodusIIReaderParser();
  ~vtkExodusIIReaderParser() override;
 
  void StartElement(const char* tagName, const char** attrs) override;
  void EndElement(const char* tagName) override;
  void FinishedParsing();
 
  const char* GetValue(const char* attr, const char** attrs)
  {
    int i;
    for (i = 0; attrs[i]; i += 2)
    {
      const char* name = strrchr(attrs[i], ':');
      if (!name)
      {
        name = attrs[i];
      }
      else
      {
        name++;
      }
      if (strcmp(attr, name) == 0)
      {
        return attrs[i + 1];
      }
    }
    return nullptr;
  }
 
  // Convenience methods to add vertices/edges to the SIL.
  vtkIdType AddVertexToSIL(const char* name);
  vtkIdType AddChildEdgeToSIL(vtkIdType src, vtkIdType dst);
  vtkIdType AddCrossEdgeToSIL(vtkIdType src, vtkIdType dst);
 
  vtkIdType GetPartVertex(const char* part_number_instance_string);
 
  // For each of the blocks, this maps the "id" attribute in the XML to the
  // vertex id for the block in the SIL.
  std::map<int, vtkIdType> BlockID_To_VertexID;
 
  // Maps block "id"s to material names.
  std::map<int, std::string> BlockID_To_MaterialName;
 
  // Maps material name to vertex id.
  // This will be build only if <material-list> is present in the XML.
  std::map<std::string, vtkIdType> MaterialName_To_VertexID;
 
  std::map<vtkIdType, std::string> PartVertexID_To_Descriptions;
 
  // These save the values read from <material-specification /> element present
  // withint the <part /> elements.
  // key: part vertex id
  // value: material name = (desp + spec)
  std::map<vtkIdType, std::string> MaterialSpecifications;
 
  // Maps the "{part-number} Instance: {part-instance}" key for the vertex id
  // for the part vertex in the Assemblies hierarchy.
  std::map<std::string, vtkIdType> Part_To_VertexID;
 
  // Maps a block-id to the "{part-number} Instance: {part-instance}" string.
  std::map<int, std::string> BlockID_To_Part;
 
  vtkMutableDirectedGraph* SIL;
  vtkSmartPointer<vtkStringArray> NamesArray;
  vtkSmartPointer<vtkUnsignedCharArray> CrossEdgesArray;
 
  std::string BlockPartNumberString;
 
  vtkIdType RootVertex;
  vtkIdType BlocksVertex;
  vtkIdType AssembliesVertex;
  vtkIdType MaterialsVertex;
  std::vector<vtkIdType> CurrentVertex;
 
  bool InBlocks;
  bool InMaterialAssignments;
 
private:
  vtkExodusIIReaderParser(const vtkExodusIIReaderParser&) = delete;
  void operator=(const vtkExodusIIReaderParser&) = delete;
};
 
VTK_ABI_NAMESPACE_END
#endif