vtkUnstructuredGridToCellGrid.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 vtkUnstructuredGridToCellGrid_h
#define vtkUnstructuredGridToCellGrid_h

#include "vtkPartitionedDataSetCollectionAlgorithm.h"

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

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

VTK_ABI_NAMESPACE_BEGIN

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

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

  virtual void Reset();
  virtual void AddPreferredOutputType(
    int inputCellType, vtkStringToken preferredOutputType, int priority);

  class TranscribeQuery : public vtkCellGridQuery
  {
  public:
    vtkTypeMacro(vtkUnstructuredGridToCellGrid::TranscribeQuery, vtkCellGridQuery);
    static TranscribeQuery* New();

    bool Initialize() override;
    bool Finalize() override;

    bool SumOutputCounts();

    void AddCellAttributes(vtkDataSetAttributes* attributes);

    struct Claim
    {
      Claim() = default;
      Claim(const Claim&) = default;
      Claim& operator=(const Claim&) = default;
      Claim(vtkIdType numberOfCells, int priority, vtkStringToken cellType)
        : NumberOfCells(numberOfCells)
        , CellTypePriority(priority)
        , CellType(cellType)
      {
      }

      vtkIdType NumberOfCells{ 0 };
      int CellTypePriority{ 0 };
      vtkStringToken CellType;
    };

    int Phase{ 0 };
    vtkUnstructuredGrid* Input{ nullptr };
    vtkCellGrid* Output{ nullptr };
    unsigned int FlatIndex{ 0 };
    vtkDataArray* Coordinates{ nullptr };
    std::map<int, Claim> CellTypeMap;
    std::unordered_map<vtkStringToken, vtkIdType> OutputAllocations;

    using BlockAttributesKey = vtkUnstructuredGridFieldAnnotations::BlockAttributesKey;

    using FieldGlom = vtkUnstructuredGridFieldAnnotations::FieldGlom;

    using BlockAttributesValue = vtkUnstructuredGridFieldAnnotations::BlockAttributesValue;

    vtkNew<vtkUnstructuredGridFieldAnnotations> Annotations;

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

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

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

  int FillInputPortInformation(int port, vtkInformation* info) override;

  int RequestData(
    vtkInformation* request, vtkInformationVector** inInfo, vtkInformationVector* ouInfo) override;

  void AddAnnotatedAttributes(vtkPartitionedDataSetCollection* input);

  virtual bool ProcessUnstructuredGrid(vtkUnstructuredGrid* input, vtkCellGrid* output);

  vtkNew<TranscribeQuery> Request;

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

VTK_ABI_NAMESPACE_END
#endif // vtkUnstructuredGridToCellGrid_h
// VTK-HeaderTest-Exclude: vtkUnstructuredGridToCellGrid.h