vtkUnstructuredGridFieldAnnotations.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 vtkUnstructuredGridFieldAnnotations_h
#define vtkUnstructuredGridFieldAnnotations_h

#include "vtkCellGrid.h"              // for API + ivars
#include "vtkFiltersCellGridModule.h" // For export macro
#include "vtkNew.h"                   // for ivar
#include "vtkObject.h"
#include "vtkStringToken.h" // for API + ivars

#include <map>
#include <unordered_map>
#include <unordered_set>
#include <vector>

VTK_ABI_NAMESPACE_BEGIN

class vtkDataArray;
class vtkDataAssembly;
class vtkDataSetAttributes;
class vtkUnstructuredGrid;
class vtkPartitionedDataSetCollection;

class VTKFILTERSCELLGRID_EXPORT vtkUnstructuredGridFieldAnnotations : public vtkObject
{
public:
  static vtkUnstructuredGridFieldAnnotations* New();
  vtkTypeMacro(vtkUnstructuredGridFieldAnnotations, vtkObject);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  virtual void FetchAnnotations(vtkFieldData* fieldData, vtkDataAssembly* assembly);
  virtual void AddAnnotations(vtkFieldData* fieldData, vtkDataAssembly* assembly);
  virtual void Reset();

  struct BlockAttributesKey
  {
    vtkStringToken DOFSharing;
    vtkStringToken FunctionSpace;
    bool operator<(const BlockAttributesKey& other) const
    {
      return this->DOFSharing.GetId() < other.DOFSharing.GetId() ||
        (this->DOFSharing == other.DOFSharing &&
          (this->FunctionSpace.GetId() < other.FunctionSpace.GetId()));
    }
  };

  struct FieldGlom
  {
    std::vector<vtkStringToken> Members;
  };

  struct BlockAttributesValue
  {
    vtkStringToken BasisSource;      // Currently always "Intrepid2"
    vtkStringToken FunctionSpace;    // "HDIV", "HGRAD", "HCURL"
    vtkStringToken Shape;            // "HEX", "QUAD", etc.
    vtkStringToken QuadratureScheme; // "I1", "C2", etc.
    vtkStringToken Formulation;      // Currently always "FEM"
    std::set<unsigned int> NodeIds;  // Nodes in the vtkDataAssembly which reference this block.
    mutable std::unordered_set<vtkStringToken> FieldNames; // Special fields for this block.
    mutable std::unordered_map<vtkStringToken, FieldGlom>
      FieldGloms; // Special fields that are glommed.
  };

  std::unordered_map<unsigned int, std::map<BlockAttributesKey, BlockAttributesValue>> Data;

protected:
  vtkUnstructuredGridFieldAnnotations();
  ~vtkUnstructuredGridFieldAnnotations() override = default;

private:
  vtkUnstructuredGridFieldAnnotations(const vtkUnstructuredGridFieldAnnotations&) = delete;
  void operator=(const vtkUnstructuredGridFieldAnnotations&) = delete;
};

VTK_ABI_NAMESPACE_END

#endif // vtkUnstructuredGridFieldAnnotations_h
// VTK-HeaderTest-Exclude: vtkUnstructuredGridFieldAnnotations.h