adaptive_refinement_mesh_wrapper.h

Go to the documentation of this file.

/*
This file is part of the Ristra Wonton project.
Please see the license file at the root of this repository, or at:
    https://github.com/laristra/wonton/blob/master/LICENSE
*/

#ifndef WONTON_ADAPTIVE_REFINEMENT_MESH_WRAPPER_H_
#define WONTON_ADAPTIVE_REFINEMENT_MESH_WRAPPER_H_

#include "wonton/mesh/adaptive_refinement/adaptive_refinement_mesh.h"

#include <vector>
#include <algorithm>

#include "wonton/support/wonton.h"
#include "wonton/support/CoordinateSystem.h"
#include "wonton/support/Point.h"

namespace Wonton {

template<int D, class CoordSys=DefaultCoordSys>
class Adaptive_Refinement_Mesh_Wrapper {

 public:

  // ==========================================================================
  // Constructors

  Adaptive_Refinement_Mesh_Wrapper() = delete;

  explicit Adaptive_Refinement_Mesh_Wrapper(
      Adaptive_Refinement_Mesh<D,CoordSys> const & mesh);

  Adaptive_Refinement_Mesh_Wrapper(
      Adaptive_Refinement_Mesh_Wrapper<D,CoordSys> const &) = delete;

  Adaptive_Refinement_Mesh_Wrapper & operator=(
      Adaptive_Refinement_Mesh_Wrapper<D,CoordSys> const &) = delete;


  // ==========================================================================
  // Accessors

  int space_dimension() const;

  int num_owned_cells() const;

  int num_ghost_cells() const;

  void cell_get_bounds(const int id, Point<D> *plo, Point<D> *phi) const;

  counting_iterator begin(Entity_kind const entity,
      Entity_type const etype = Entity_type::ALL) const;

  counting_iterator end(Entity_kind const entity,
      Entity_type const etype = Entity_type::ALL) const;


 private:

  // ==========================================================================
  // Class data

  Adaptive_Refinement_Mesh<D,CoordSys> const & mesh_;

};  // class Adaptive_Refinement_Mesh_Wrapper


// ============================================================================
// Constructors

// ____________________________________________________________________________
// constructor
template<int D, class CoordSys>
Adaptive_Refinement_Mesh_Wrapper<D,CoordSys>::Adaptive_Refinement_Mesh_Wrapper(
    Adaptive_Refinement_Mesh<D,CoordSys> const & mesh) :
    mesh_(mesh) {
}


// ============================================================================
// Accessors

// ____________________________________________________________________________
// Get dimensionality of the mesh.
template<int D, class CoordSys>
int Adaptive_Refinement_Mesh_Wrapper<D,CoordSys>::space_dimension() const {
  return mesh_.space_dimension();
}

// ____________________________________________________________________________
// Get number of cells owned by this processing element.
template<int D, class CoordSys>
int Adaptive_Refinement_Mesh_Wrapper<D,CoordSys>::num_owned_cells() const {
  return mesh_.num_cells();
}

// ____________________________________________________________________________
// Get number of ghost cells on this processing element.
template<int D, class CoordSys>
int Adaptive_Refinement_Mesh_Wrapper<D,CoordSys>::num_ghost_cells() const {
  return 0;
}

// ____________________________________________________________________________
// Get lower and upper corners of cell bounding box
template<int D, class CoordSys>
void Adaptive_Refinement_Mesh_Wrapper<D,CoordSys>::cell_get_bounds(
    const int id, Point<D> *plo, Point<D> *phi) const {
  auto bounding_box = mesh_.cell_get_bounds(id);
  for (int d = 0; d < D; ++d) {
    (*plo)[d] = bounding_box[d][LO];
    (*phi)[d] = bounding_box[d][HI];
  }
}

// ____________________________________________________________________________
// Iterators on mesh entity - begin
template<int D, class CoordSys>
counting_iterator Adaptive_Refinement_Mesh_Wrapper<D,CoordSys>::begin(
    Entity_kind const entity, Entity_type const etype) const {
  // Currently only valid for cells
  assert(entity == Wonton::Entity_kind::CELL);
  // Currently only valid for owned cells
  assert(etype == Wonton::Entity_type::PARALLEL_OWNED);
  // Return iterator
  int start_index = 0;
  return(make_counting_iterator(start_index));
}

// ____________________________________________________________________________
// Iterator on mesh entity - end
template<int D, class CoordSys>
counting_iterator Adaptive_Refinement_Mesh_Wrapper<D,CoordSys>::end(
    Entity_kind const entity, Entity_type const etype) const {
  // Currently only valid for cells
  assert(entity == Wonton::Entity_kind::CELL);
  // Currently only valid for owned cells
  assert(etype == Wonton::Entity_type::PARALLEL_OWNED);
  // Return iterator
  int start_index = 0;
  return(make_counting_iterator(start_index) + num_owned_cells());
}

}  // namespace Wonton

#endif  // WONTON_ADAPTIVE_REFINEMENT_MESH_WRAPPER_H_