Add Elem::n_vertices_per_side()

Author: loganharbour

include/geom/cell_hex.h

@@ -77,6 +77,12 @@ class Hex : public Cell
    */
   virtual unsigned int n_vertices() const override final { return 8; }
 
+  /**
+   * \returns 4. Every side has four vertices.
+   */
+  virtual unsigned int n_vertices_on_side(const unsigned short libmesh_dbg_var(s)) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 4; }
+
   /**
    * \returns 12.  All hexahedra have 12 edges.
    */

include/geom/cell_inf_hex.h

@@ -90,6 +90,12 @@ class InfHex : public InfCell
    */
   virtual unsigned int n_vertices() const override final { return 8; }
 
+  /**
+   * \returns 4. Every side has four vertices.
+   */
+  virtual unsigned int n_vertices_on_side(const unsigned short libmesh_dbg_var(s)) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 4; }
+
   /**
    * \returns \p true if the specified (local) node number is a
    * "mid-edge" node on an infinite element edge.

include/geom/cell_inf_prism.h

@@ -86,6 +86,9 @@ class InfPrism : public InfCell
    */
   virtual unsigned int n_vertices() const override final { return 6; }
 
+  virtual unsigned int n_vertices_on_side(const unsigned short s) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 4 - (s == 0 || s == 4); }
+
   /**
    * \returns 6.  All infinite prisms have 6 edges,
    * 3 lying in the base, and 3 perpendicular to the base.

include/geom/cell_prism.h

@@ -83,6 +83,9 @@ class Prism : public Cell
    */
   virtual unsigned int n_vertices() const override final { return 6; }
 
+  virtual unsigned int n_vertices_on_side(const unsigned short s) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 4 - (s == 0 || s == 4); }
+
   /**
    * \returns 9.  All prisms have 9 edges.
    */

include/geom/cell_pyramid.h

@@ -87,6 +87,9 @@ class Pyramid : public Cell
    */
   virtual unsigned int n_vertices() const override { return 5; }
 
+  virtual unsigned int n_vertices_on_side(const unsigned short s) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 4 - (s != 4); }
+
   /**
    * \returns 8.  All pyramids have 8 edges.
    */

include/geom/cell_tet.h

@@ -78,6 +78,12 @@ class Tet : public Cell
    */
   virtual unsigned int n_vertices() const override final { return 4; }
 
+  /**
+   * \returns 3. Every side has three vertices.
+   */
+  virtual unsigned int n_vertices_on_side(const unsigned short libmesh_dbg_var(s)) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 3; }
+
   /**
    * \returns 6.  All tetrahedra have 6 edges.
    */

include/geom/edge.h

@@ -78,6 +78,12 @@ class Edge : public Elem
    */
   virtual unsigned int n_vertices() const override final { return 2; }
 
+  /**
+   * \returns 1. Every side has one vertex.
+   */
+  virtual unsigned int n_vertices_on_side(const unsigned short libmesh_dbg_var(s)) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 1; }
+
   /**
    * \returns 0.  All 1D elements have no edges.
    */

include/geom/elem.h

@@ -641,6 +641,12 @@ class Elem : public ReferenceCountedObject<Elem>,
    */
   IntRange<unsigned short> side_index_range () const;
 
+  /**
+   * \returns An integer range from 0 up to (but not including)
+   * the number of vertices this element has.
+   */
+  IntRange<unsigned short> vertex_index_range () const;
+
   /**
    * \returns The number of neighbors the element that has been derived
    * from this class has.
@@ -659,6 +665,11 @@ class Elem : public ReferenceCountedObject<Elem>,
    */
   virtual unsigned int n_vertices () const = 0;
 
+  /**
+   * \returns The number of verticies on the side with index \p s.
+   */
+  virtual unsigned int n_vertices_on_side (const unsigned short s) const = 0;
+
   /**
    * \returns The number of edges the element that has been derived
    * from this class has.
@@ -2427,6 +2438,13 @@ Elem::side_index_range() const
 
 
 
+inline
+IntRange<unsigned short>
+Elem::vertex_index_range() const
+{
+  return {0, cast_int<unsigned short>(this->n_vertices())};
+}
+
 
 inline
 std::unique_ptr<const Elem> Elem::side_ptr (unsigned int i) const

include/geom/face.h

@@ -66,6 +66,12 @@ class Face : public Elem
    */
   virtual unsigned int n_faces() const override final { return 0; }
 
+  /**
+   * \returns 2. Every side has two vertices.
+   */
+  virtual unsigned int n_vertices_on_side(const unsigned short libmesh_dbg_var(s)) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 2; }
+
   /**
    * build_side and build_edge are identical for faces.
    */

include/geom/face_inf_quad.h

@@ -106,6 +106,12 @@ class InfQuad : public Elem
    */
   virtual unsigned int n_vertices() const override final { return 4; }
 
+  /**
+   * \returns 2. Every side has two vertices.
+   */
+  virtual unsigned int n_vertices_on_side(const unsigned short libmesh_dbg_var(s)) const override final
+  { libmesh_assert_less(s, this->n_sides()); return 2; }
+
   /**
    * \returns 3.  All infinite quads have 1 edge in the
    * base, and 2 perpendicular to the base.