From 3696373fd78d6ce87c9a17b0020f9ac6938530b5 Mon Sep 17 00:00:00 2001
From: Jean-Paul Pelteret <jppelteret@gmail.com>
Date: Fri, 7 Jan 2022 22:26:48 +0100
Subject: [PATCH] Add MeshWorker::ScratchData::n_dofs_per_cell()

---
 include/deal.II/meshworker/scratch_data.h | 14 ++++++++++++++
 source/meshworker/scratch_data.cc         | 19 ++++++++++++++++++-
 2 files changed, 32 insertions(+), 1 deletion(-)

diff --git a/include/deal.II/meshworker/scratch_data.h b/include/deal.II/meshworker/scratch_data.h
index 0c97cadd73..1b7971987c 100644
--- a/include/deal.II/meshworker/scratch_data.h
+++ b/include/deal.II/meshworker/scratch_data.h
@@ -527,6 +527,13 @@ namespace MeshWorker
     const std::vector<types::global_dof_index> &
     get_local_dof_indices() const;
 
+    /**
+     * Return the number of local dof indices for the cell passed the last time
+     * the reinit() function was called.
+     */
+    unsigned int
+    n_dofs_per_cell() const;
+
     /** @} */ // CurrentCellMethods
 
     /**
@@ -606,6 +613,13 @@ namespace MeshWorker
     const std::vector<types::global_dof_index> &
     get_neighbor_dof_indices() const;
 
+    /**
+     * Return the number of local dof indices for the neighbor passed the last
+     * time the reinit_neighbor() function was called.
+     */
+    unsigned int
+    n_neighbor_dofs_per_cell() const;
+
     /** @} */ // NeighborCellMethods
 
     /**
diff --git a/source/meshworker/scratch_data.cc b/source/meshworker/scratch_data.cc
index 208acece0b..6d5394f341 100644
--- a/source/meshworker/scratch_data.cc
+++ b/source/meshworker/scratch_data.cc
@@ -516,7 +516,6 @@ namespace MeshWorker
 
             neighbor_dof_indices.resize(neighbor_fe.n_dofs_per_cell());
             cell->get_dof_indices(neighbor_dof_indices);
-            cell->get_dof_indices(neighbor_dof_indices);
 
             current_neighbor_fe_values = &neighbor_fe_subface_values;
             return neighbor_fe_subface_values;
@@ -618,6 +617,15 @@ namespace MeshWorker
 
 
 
+  template <int dim, int spacedim>
+  unsigned int
+  ScratchData<dim, spacedim>::n_dofs_per_cell() const
+  {
+    return local_dof_indices.size();
+  }
+
+
+
   template <int dim, int spacedim>
   const std::vector<types::global_dof_index> &
   ScratchData<dim, spacedim>::get_neighbor_dof_indices() const
@@ -627,6 +635,15 @@ namespace MeshWorker
 
 
 
+  template <int dim, int spacedim>
+  unsigned int
+  ScratchData<dim, spacedim>::n_neighbor_dofs_per_cell() const
+  {
+    return neighbor_dof_indices.size();
+  }
+
+
+
   template <int dim, int spacedim>
   GeneralDataStorage &
   ScratchData<dim, spacedim>::get_general_data_storage()
-- 
2.39.5