product_dofhandler.h

#ifndef PROJECTS_DPG_PRODUCT_DOFHANDLER
#define PROJECTS_DPG_PRODUCT_DOFHANDLER
 
#include <lf/assemble/assemble.h>
#include <lf/mesh/mesh.h>
 
#include "dpg.h"
 
namespace projects::dpg {
 
using dof_map_t = lf::assemble::UniformFEDofHandler::dof_map_t;
 
class ProductUniformFEDofHandler : public lf::assemble::DofHandler {
 public:
  ProductUniformFEDofHandler(std::shared_ptr<const lf::mesh::Mesh> mesh_p,
                             std::vector<dof_map_t> dofmaps);
 
  [[nodiscard]] size_type NumDofs() const override;
 
  [[nodiscard]] size_type NumDofs(size_type component) const;
 
  [[nodiscard]] size_type NumLocalDofs(
      const lf::mesh::Entity& entity) const override;
 
  [[nodiscard]] size_type NumLocalDofs(const lf::mesh::Entity& entity,
                                       size_type component) const;
 
  [[nodiscard]] size_type NumInteriorDofs(
      const lf::mesh::Entity& entity) const override;
 
  [[nodiscard]] size_type NumInteriorDofs(const lf::mesh::Entity& entity,
                                          size_type component) const;
 
  [[nodiscard]] nonstd::span<const gdof_idx_t> GlobalDofIndices(
      const lf::mesh::Entity& entity) const override;
 
  [[nodiscard]] nonstd::span<const gdof_idx_t> InteriorGlobalDofIndices(
      const lf::mesh::Entity& entity) const override;
 
  [[nodiscard]] const lf::mesh::Entity& Entity(
      gdof_idx_t dofnum) const override;
 
  [[nodiscard]] size_type Component(glb_idx_t dofnum) const;
 
  [[nodiscard]] std::shared_ptr<const lf::mesh::Mesh> Mesh() const override;
 
  [[nodiscard]] const lf::assemble::UniformFEDofHandler& ComponentDofHandler(
      size_type component) const {
    return *component_dof_handlers_[component];
  }
 
  [[nodiscard]] ldof_idx_t LocalStartIndex(const lf::mesh::Entity& entity,
                                           size_type component) const;
 
  [[nodiscard]] size_type NumComponents() const { return num_components_; }
 
  [[nodiscard]] size_type Offset(size_type component) const {
    return offsets_[component];
  }
 
 private:
  void initIndexArrays();
 
  std::shared_ptr<const lf::mesh::Mesh> mesh_p_;
 
  std::vector<std::shared_ptr<lf::assemble::UniformFEDofHandler>>
      component_dof_handlers_;
 
  size_type num_components_;
 
  // access via  dof_[codim][entity_idx][local_idx].
  std::array<std::vector<std::vector<gdof_idx_t>>, 3> dofs_;
 
  // access via  dof_[codim][entity_idx][local_idx].
  std::array<std::vector<std::vector<gdof_idx_t>>, 3> internal_dofs_;
 
  std::vector<glb_idx_t> offsets_;
};
 
}  // namespace projects::dpg
 
#endif  // PROJECTS_DPG_PRODUCT_DOFHANDLER