vtkCellGridIOQuery.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 vtkCellGridIOQuery_h
#define vtkCellGridIOQuery_h

#include "vtkCellAttribute.h" // For vtkCellAttribute::CellTypeInfo
#include "vtkCellGridQuery.h"
#include "vtkIOCellGridModule.h" // For export macro

#include <vtk_nlohmannjson.h>        // For API.
#include VTK_NLOHMANN_JSON(json.hpp) // For API.

VTK_ABI_NAMESPACE_BEGIN
class vtkDataArraySelection;
class vtkFieldData;
class vtkImageData;
class vtkStringArray;

class VTKIOCELLGRID_EXPORT vtkCellGridIOQuery : public vtkCellGridQuery
{
public:
  static vtkCellGridIOQuery* New();
  vtkTypeMacro(vtkCellGridIOQuery, vtkCellGridQuery);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  void PrepareToDeserialize(const nlohmann::json& sourceData, const nlohmann::json& attributeData,
    std::vector<vtkCellAttribute*>& attributeList);


  void PrepareToSerialize(nlohmann::json& destination, nlohmann::json& attributeData,
    const std::unordered_map<vtkAbstractArray*, vtkStringToken>& arrayLocations);

  nlohmann::json* GetData() { return this->Data; }

  nlohmann::json* GetAttributeData() { return this->AttributeData; }

  const std::unordered_map<vtkAbstractArray*, vtkStringToken>& GetArrayLocations() const
  {
    return this->ArrayLocations;
  }
  std::unordered_map<vtkAbstractArray*, vtkStringToken>& GetArrayLocations()
  {
    return this->ArrayLocations;
  }

  const std::vector<vtkCellAttribute*>& GetAttributeList() const { return *this->AttributeList; }

  bool IsSerializing() const { return this->Serializing; }

  bool ExtractCellTypeAttributeInfo(vtkCellGrid* grid, vtkCellAttribute::CellTypeInfo& cellTypeInfo,
    const nlohmann::json& jsonInfo, vtkStringToken cellTypeName);

  bool InsertCellTypeAttributeInfo(vtkCellGrid* grid,
    const vtkCellAttribute::CellTypeInfo& cellTypeInfo, nlohmann::json& jsonInfo,
    vtkStringToken cellTypeName);

  nlohmann::json& AddMetadataEntry(vtkStringToken cellTypeName);

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

  nlohmann::json* Data{ nullptr };
  nlohmann::json* AttributeData{ nullptr };
  std::vector<vtkCellAttribute*>* AttributeList;
  std::unordered_map<vtkAbstractArray*, vtkStringToken> ArrayLocations;
  bool Serializing{ true };

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

VTK_ABI_NAMESPACE_END
#endif