vtkAMRInformation.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Visualization Toolkit
  Module:    vtkAMRInformation.h
 
  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
 
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
#ifndef vtkAMRInformation_h
#define vtkAMRInformation_h
 
#include "vtkCommonDataModelModule.h" // For export macro
#include "vtkObject.h"
#include "vtkAMRBox.h" //for storing AMR Boxes
#include "vtkSmartPointer.h" //for ivars
#include <vector> //for storing AMR Boxes
 
 
typedef std::vector<vtkAMRBox> vtkAMRBoxList;
 
class vtkUnsignedIntArray;
class vtkIntArray;
class vtkDoubleArray;
class vtkAMRIndexIterator;
 
class VTKCOMMONDATAMODEL_EXPORT vtkAMRInformation : public vtkObject
{
public:
  static vtkAMRInformation* New();
  vtkTypeMacro(vtkAMRInformation, vtkObject);
 
  void PrintSelf(ostream& os, vtkIndent indent);
 
  bool operator==(const vtkAMRInformation& other);
 
  void Initialize(int numLevels, const int* blocksPerLevel);
 
 
  vtkGetMacro( GridDescription, int );
  void SetGridDescription(int description);
 
 
  void GetOrigin( double origin[3] );
  double* GetOrigin();
  void SetOrigin(const double* origin);
 
 
  unsigned int GetNumberOfLevels() const
  { return static_cast<unsigned int>(this->NumBlocks.size()-1);}
 
  unsigned int GetNumberOfDataSets(unsigned int level) const;
 
 
  unsigned int GetTotalNumberOfBlocks()
  { return this->NumBlocks.back();}
 
 
  int GetIndex(unsigned int level, unsigned int id) const
  { return this->NumBlocks[level] + id;}
 
  void ComputeIndexPair(unsigned int index, unsigned int& level, unsigned int& id);
 
  const double* GetBounds();
 
  void GetBounds(unsigned int level, unsigned int id, double* bb);
 
  bool GetOrigin(unsigned int level, unsigned int id, double* origin);
 
  void GetSpacing(unsigned int level, double spacing[3]);
 
  bool HasSpacing(unsigned int level);
 
 
  void SetAMRBox(unsigned int level, unsigned int id, const vtkAMRBox& box);
  const vtkAMRBox& GetAMRBox(unsigned int level, unsigned int id) const;
 
  bool GetCoarsenedAMRBox(unsigned int level, unsigned int id, vtkAMRBox& box) const;
 
 
  int GetAMRBlockSourceIndex(int index);
  void SetAMRBlockSourceIndex(int index, int sourceId);
 
  void GenerateRefinementRatio();
 
  bool HasRefinementRatio();
 
  void SetRefinementRatio(unsigned int level, int ratio);
 
  int GetRefinementRatio(unsigned int level) const;
 
  void SetSpacing(unsigned int level,const double* h);
 
  bool HasChildrenInformation();
 
  unsigned int *GetParents(unsigned int level, unsigned int index, unsigned int& numParents);
 
  unsigned int *GetChildren(unsigned int level, unsigned int index, unsigned int& numChildren);
 
  void PrintParentChildInfo(unsigned int level, unsigned int index);
 
  void GenerateParentChildInformation();
 
  bool Audit();
 
  bool FindCell(double q[3],unsigned int level, unsigned int index,int &cellIdx);
 
  bool FindGrid(double q[3], int level, unsigned int& gridId);
 
  bool FindGrid(double q[3], unsigned int& level, unsigned int& gridId);
 
 
  const std::vector<int>& GetNumBlocks() const
  { return this->NumBlocks;}
 
  std::vector<std::vector<unsigned int> >& GetChildrenAtLevel(unsigned int i)
  { return this->AllChildren[i];}
 
  void DeepCopy(vtkAMRInformation *other);
 
 private:
  vtkAMRInformation();
  ~vtkAMRInformation();
  vtkAMRInformation(const vtkAMRInformation&); // Not implemented.
  void operator=(const vtkAMRInformation&); // Not implemented.
 
  bool HasValidOrigin();
  bool HasValidBounds();
  void UpdateBounds(const int level, const int id);
  void AllocateBoxes(unsigned int n);
  void GenerateBlockLevel();
  void CalculateParentChildRelationShip( unsigned int level,
                                        std::vector<std::vector<unsigned int> >& children,
                                        std::vector<std::vector<unsigned int> >& parents );
 
  //-------------------------------------------------------------------------
  // Essential information that determines an AMR structure. Must be copied
  //-------------------------------------------------------------------------
  int GridDescription; //example: VTK_XYZ_GRID
  double Origin[3]; //the origin of the whole data set
  vtkAMRBoxList Boxes; // vtkAMRBoxes, one per data set
  std::vector<int> NumBlocks; //NumBlocks[i] stores the total number of blocks from level 0 to level i-1
 
  vtkSmartPointer<vtkIntArray> SourceIndex; //Typically, this maps to a file block index used by the reader
  vtkSmartPointer<vtkDoubleArray> Spacing; //The grid spacing for all levels
  double Bounds[6]; //the bounds of the entire domain
 
  //-------------------------------------------------------------------------
  // Auxillary information that be computed
  //-------------------------------------------------------------------------
  vtkSmartPointer<vtkIntArray> Refinement; //refinement ratio between two adjacent levels
  vtkSmartPointer<vtkUnsignedIntArray> BlockLevel; //only necessary if need to call ComputeIndexPair
 
  //parent child information
  std::vector<std::vector<std::vector<unsigned int> > > AllChildren;
  std::vector<std::vector<std::vector<unsigned int> > > AllParents;
};
 
#endif