vtkAMRDataObject.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 vtkAMRDataObject_h
#define vtkAMRDataObject_h
 
#include "vtkCommonDataModelModule.h" // For export macro
#include "vtkNew.h"                   // for vtkNew
#include "vtkPartitionedDataSetCollection.h"
#include "vtkSmartPointer.h" // for vtkSmartPointer
 
VTK_ABI_NAMESPACE_BEGIN
class vtkAMRMetaData;
class vtkCartesianGrid;
class vtkCompositeDataIterator;
class vtkDataSet;
class vtkImageData;
class vtkRectilinearGrid;
class vtkUniformGrid;
class VTKCOMMONDATAMODEL_EXPORT vtkAMRDataObject : public vtkPartitionedDataSetCollection
{
public:
  static vtkAMRDataObject* New();
  vtkTypeMacro(vtkAMRDataObject, vtkPartitionedDataSetCollection);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  VTK_NEWINSTANCE vtkCompositeDataIterator* NewIterator() override;

  int GetDataObjectType() VTK_FUTURE_CONST override { return VTK_AMR_DATA_OBJECT; }

  void Initialize() override;

  virtual void Initialize(const std::vector<unsigned int>& blocksPerLevel);

  virtual void Initialize(vtkAMRMetaData* metadata);


  void SetGridDescription(int gridDescription);
  int GetGridDescription();

  [[nodiscard]] unsigned int GetNumberOfLevels() const;

  [[nodiscard]] unsigned int GetNumberOfBlocks() const;

  [[nodiscard]] unsigned int GetNumberOfBlocks(unsigned int level) const;


  void GetBounds(double bounds[6]);
  virtual const double* GetBounds();
  void GetMin(double min[3]);
  void GetMax(double max[3]);
 
  using Superclass::SetDataSet;
  virtual void SetDataSet(unsigned int level, unsigned int idx, vtkDataSet* grid);

  vtkCartesianGrid* GetDataSetAsCartesianGrid(unsigned int level, unsigned int idx);

  vtkImageData* GetDataSetAsImageData(unsigned int level, unsigned int idx);

  vtkRectilinearGrid* GetDataSetAsRectilinearGrid(unsigned int level, unsigned int idx);

  [[nodiscard]] int GetAbsoluteBlockIndex(unsigned int level, unsigned int index) const;

  void ComputeIndexPair(unsigned int index, unsigned int& level, unsigned int& id);


  void CompositeShallowCopy(vtkCompositeDataSet* src) override;
  void ShallowCopy(vtkDataObject* src) override;

  void DeepCopy(vtkDataObject* src) override;

  void CopyStructure(vtkCompositeDataSet* src) override;


  static vtkAMRDataObject* GetData(vtkInformation* info);
  static vtkAMRDataObject* GetData(vtkInformationVector* v, int i = 0);


  [[nodiscard]] vtkAMRMetaData* GetAMRMetaData();
  void SetAMRMetaData(vtkAMRMetaData* metadata);


  void SetPartition(unsigned int idx, unsigned int partition, vtkDataObject* object) override;
  vtkDataSet* GetPartition(unsigned int idx, unsigned int partition) override;
 
protected:
  vtkAMRDataObject();
  ~vtkAMRDataObject() override;

  virtual void InstantiateMetaData();
 
private:
  vtkAMRDataObject(const vtkAMRDataObject&) = delete;
  void operator=(const vtkAMRDataObject&) = delete;
 
  void InitializeInternal();
 
  double Bounds[6] = {
    VTK_DOUBLE_MAX,
    VTK_DOUBLE_MIN,
    VTK_DOUBLE_MAX,
    VTK_DOUBLE_MIN,
    VTK_DOUBLE_MAX,
    VTK_DOUBLE_MIN,
  };
  vtkSmartPointer<vtkAMRMetaData> AMRMetaData;
};
 
VTK_ABI_NAMESPACE_END
#endif