vtkCellGridCopyQuery.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 vtkCellGridCopyQuery_h
#define vtkCellGridCopyQuery_h

#include "vtkCellGridQuery.h"
#include "vtkStringToken.h" // For API.

#include <map> // For ArrayMap ivar.
#include <set> // For CellAttributeIds ivar.

VTK_ABI_NAMESPACE_BEGIN

class vtkAbstractArray;
class vtkCellAttribute;
class vtkCellGrid;
class vtkIdList;

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

  bool Initialize() override;

  bool Finalize() override;


  virtual void SetSource(vtkCellGrid* source);
  vtkGetObjectMacro(Source, vtkCellGrid);

  virtual void SetTarget(vtkCellGrid* target);
  vtkGetObjectMacro(Target, vtkCellGrid);



  vtkGetMacro(CopyCellTypes, int);
  vtkSetMacro(CopyCellTypes, int);
  vtkBooleanMacro(CopyCellTypes, int);

  vtkGetMacro(CopyCells, int);
  vtkSetMacro(CopyCells, int);
  vtkBooleanMacro(CopyCells, int);

  vtkGetMacro(CopyOnlyShape, int);
  vtkSetMacro(CopyOnlyShape, int);
  vtkBooleanMacro(CopyOnlyShape, int);

  vtkGetMacro(CopyArrays, int);
  vtkSetMacro(CopyArrays, int);
  vtkBooleanMacro(CopyArrays, int);

  vtkGetMacro(CopyArrayValues, int);
  vtkSetMacro(CopyArrayValues, int);
  vtkBooleanMacro(CopyArrayValues, int);

  vtkGetMacro(DeepCopyArrays, int);
  vtkSetMacro(DeepCopyArrays, int);
  vtkBooleanMacro(DeepCopyArrays, int);

  vtkGetMacro(CopySchema, int);
  vtkSetMacro(CopySchema, int);
  vtkBooleanMacro(CopySchema, int);



  virtual bool AddSourceCellAttributeId(int attributeId);

  virtual bool RemoveSourceCellAttributeId(int attributeId);

  virtual bool AddAllSourceCellAttributeIds();

  const std::set<int>& GetCellAttributeIds() const { return this->CellAttributeIds; }

  void GetCellAttributeIds(vtkIdList* ids) const;

  virtual void ResetCellAttributeIds();



  const std::map<vtkAbstractArray*, vtkAbstractArray*>& GetArrayMap() const
  {
    return this->ArrayMap;
  }
  std::map<vtkAbstractArray*, vtkAbstractArray*>& GetArrayMap() { return this->ArrayMap; }

  const std::map<vtkCellAttribute*, vtkCellAttribute*>& GetAttributeMap() const
  {
    return this->AttributeMap;
  }
  std::map<vtkCellAttribute*, vtkCellAttribute*>& GetAttributeMap() { return this->AttributeMap; }



  void CopyAttributeArrays(vtkCellAttribute* srcAtt, vtkStringToken cellType);

  vtkCellAttribute* CopyOrUpdateAttributeRecord(vtkCellAttribute* srcAtt, vtkStringToken cellType);


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

  vtkCellGrid* Source{ nullptr };
  vtkCellGrid* Target{ nullptr };
  std::set<int> CellAttributeIds;
  std::map<vtkAbstractArray*, vtkAbstractArray*> ArrayMap;
  std::map<vtkCellAttribute*, vtkCellAttribute*> AttributeMap;
  int CopyCellTypes{ 1 };
  int CopyCells{ 1 };
  int CopyOnlyShape{ 1 };
  int CopyArrays{ 1 };
  int CopyArrayValues{ 1 };
  int DeepCopyArrays{ 0 };
  int CopySchema{ 1 };

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

VTK_ABI_NAMESPACE_END
#endif // vtkCellGridCopyQuery_h