quadrature_data.hpp

Go to the documentation of this file.

// Copyright (c) 2019-2024, Lawrence Livermore National Security, LLC and
// other Serac Project Developers. See the top-level LICENSE file for
// details.
//
// SPDX-License-Identifier: (BSD-3-Clause)
 
#pragma once
 
#include "mfem.hpp"
 
#include "axom/core.hpp"
 
#include "serac/serac_config.hpp"
 
#include "serac/infrastructure/accelerator.hpp"
 
namespace serac {
 
struct Nothing {};
 
struct Empty {};
 
template <typename T>
struct QuadratureData;
 
}  // namespace serac
 
// we define these specializations to make it so that materials
// without state variables can use the same interface, without
// actually storing/accessing any data
namespace axom {
 
template <>
class Array<serac::Nothing, 2, MemorySpace::Dynamic> {
public:
  Array() {}
  Array(uint32_t, uint32_t) {}
};
 
template <>
class ArrayView<serac::Nothing, 2, MemorySpace::Dynamic> {
public:
  ArrayView(Array<serac::Nothing, 2, MemorySpace::Dynamic> /* unused */) {}
 
  SERAC_HOST_DEVICE serac::Nothing& operator()(const size_t, const size_t) { return data; }
 
  SERAC_HOST_DEVICE const serac::Nothing& operator()(const size_t, const size_t) const { return data; }
 
  serac::Nothing data;
};
 
template <>
class Array<serac::Empty, 2, MemorySpace::Dynamic> {
public:
  Array() {}
  Array(uint32_t, uint32_t) {}
};
 
template <>
class ArrayView<serac::Empty, 2, MemorySpace::Dynamic> {
public:
  ArrayView(Array<serac::Empty, 2, MemorySpace::Dynamic> /* unused */) {}
 
  SERAC_HOST_DEVICE serac::Empty& operator()(const size_t, const size_t) { return data; }
 
  SERAC_HOST_DEVICE const serac::Empty& operator()(const size_t, const size_t) const { return data; }
 
  serac::Empty data;
};
 
}  // namespace axom
 
namespace serac {
 
template <typename T>
struct QuadratureData {
  using geom_array_t = std::array<uint32_t, mfem::Geometry::NUM_GEOMETRIES>;
 
  QuadratureData(geom_array_t elements, geom_array_t qpts_per_element, T value = T{})
  {
    constexpr std::array geometries = {mfem::Geometry::SEGMENT, mfem::Geometry::TRIANGLE, mfem::Geometry::SQUARE,
                                       mfem::Geometry::TETRAHEDRON, mfem::Geometry::CUBE};
 
    for (auto geom : geometries) {
      if (elements[uint32_t(geom)] > 0) {
        data[geom] = axom::Array<T, 2>(elements[uint32_t(geom)], qpts_per_element[uint32_t(geom)]);
        data[geom].fill(value);
      }
    }
  }
 
  axom::ArrayView<T, 2> operator[](mfem::Geometry::Type geom) { return axom::ArrayView<T, 2>(data.at(geom)); }
 
  std::map<mfem::Geometry::Type, axom::Array<T, 2> > data;
};
 
template <>
struct QuadratureData<Nothing> {
  using geom_array_t = std::array<uint32_t, mfem::Geometry::NUM_GEOMETRIES>;
 
  QuadratureData() {}
 
  axom::ArrayView<Nothing, 2> operator[](mfem::Geometry::Type) { return axom::ArrayView<Nothing, 2>(data); }
 
  axom::Array<Nothing, 2, axom::MemorySpace::Dynamic> data;
};
 
template <>
struct QuadratureData<Empty> {
  using geom_array_t = std::array<uint32_t, mfem::Geometry::NUM_GEOMETRIES>;
 
  QuadratureData() {}
 
  axom::ArrayView<Empty, 2> operator[](mfem::Geometry::Type) { return axom::ArrayView<Empty, 2>(data); }
 
  axom::Array<Empty, 2, axom::MemorySpace::Dynamic> data;
};
 
extern std::shared_ptr<QuadratureData<Nothing> > NoQData;
extern std::shared_ptr<QuadratureData<Empty> >   EmptyQData;
 
}  // namespace serac