Add MeshWorker::ScratchData::n_dofs_per_cell()

diff --git a/include/deal.II/meshworker/scratch_data.h b/include/deal.II/meshworker/scratch_data.h
index 0c97cadd73a7d55729fc1e7f153f1ab0aa83a572..1b7971987c9d51799e9793d6804eec7ed7c91c69 100644 (file)
--- 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 208acece0b39b8edf32c43db85f72ce0ef8094a7..6d5394f341aa6ea6cd2c10f31d2062eccef182f4 100644 (file)
--- 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()