local-dof-table.hpp

Go to the documentation of this file.

#ifndef LOCALDOFTABLE_HPP
#define LOCALDOFTABLE_HPP
 
#include "discrete-space-descriptor.hpp"
 
namespace HArDCore2D {
 
  namespace DSL {
 
    class LocalDOFTable {
    public:
      LocalDOFTable(const DiscreteSpaceDescriptor & descriptor);
 
      const Mesh & mesh() const
      {
        return m_descriptor.mesh();
      }
 
      DiscreteSpaceDescriptor & descriptor() {
        return m_descriptor;
      }
      
      const DiscreteSpaceDescriptor & descriptor() const {
        return m_descriptor;
      }
      
      size_t numberOfLocalVertexDofs() const
      {
        return m_n_local_vertex_dofs;
      }
 
      size_t numberOfLocalVertexDofs(const std::string & dof_name) const
      {
        return m_descriptor.numberOfLocalVertexDofs(dof_name);
      }
 
      size_t numberOfLocalEdgeDofs() const
      {
        return m_n_local_edge_dofs;
      }
 
      size_t numberOfLocalEdgeDofs(const std::string & dof_name) const
      {
        return m_descriptor.numberOfLocalEdgeDofs(dof_name);
      }
 
      size_t numberOfLocalCellDofs() const
      {
        return m_n_local_cell_dofs;
      }
 
      size_t numberOfLocalCellDofs(const std::string & dof_name) const
      {
        return m_descriptor.numberOfLocalCellDofs(dof_name);
      }
 
      inline size_t dimension() const
      {
        return mesh().n_vertices() * m_n_local_vertex_dofs
          + mesh().n_edges() * m_n_local_edge_dofs
          + mesh().n_cells() * m_n_local_cell_dofs;
      }
 
      inline size_t numTotalDofsVertices() const
      {
        return mesh().n_vertices() * m_n_local_vertex_dofs;
      }
 
      inline size_t numTotalDofsEdges() const
      {
        return mesh().n_edges() * m_n_local_edge_dofs;
      }
 
      inline size_t numTotalDofsCells() const
      {
        return mesh().n_cells() * m_n_local_cell_dofs;
      }
    
      inline size_t dimensionVertex(const Vertex & V) const
      {
        return m_n_local_vertex_dofs;
      }
 
      inline size_t dimensionVertex(size_t iV) const
      {
        return dimensionVertex(*mesh().vertex(iV));
      }
 
      inline size_t dimensionEdge(const Edge & E) const
      {
        return 2 * m_n_local_vertex_dofs
          + m_n_local_edge_dofs;
      }
 
      inline size_t dimensionEdge(size_t iE) const
      {
        return dimensionEdge(*mesh().edge(iE));
      }
 
      inline size_t dimensionCell(const Cell & T) const
      {
        return T.n_vertices() * m_n_local_vertex_dofs
          + T.n_edges() * m_n_local_edge_dofs
          + m_n_local_cell_dofs;
      }
 
      inline size_t dimensionCellBoundary(const Cell & T) const
      {
        return T.n_vertices() * m_n_local_vertex_dofs
          + T.n_edges() * m_n_local_edge_dofs;
      }
    
      inline size_t dimensionCell(size_t iT) const
      {
        return dimensionCell(*mesh().cell(iT));
      }
 
      inline size_t dimensionCellBoundary(size_t iT) const
      {
        return dimensionCellBoundary(*mesh().cell(iT));
      }
 
      size_t localOffset(const Edge & E, const Vertex & V) const;
      size_t localOffset(const Edge & E, const Vertex & V, const std::string & vertex_dof_name) const;
      size_t localOffset(const Edge & E) const;
      size_t localOffset(const Edge & E, const std::string & edge_dof_name) const;
      size_t localOffset(const Cell & T, const Vertex & V) const;
      size_t localOffset(const Cell & T, const Vertex & V, const std::string & vertex_dof_name) const;
      size_t localOffset(const Cell & T, const Edge & E) const;
      size_t localOffset(const Cell & T, const Edge & E, const std::string & edge_dof_name) const;
      size_t localOffset(const Cell & T) const;
      size_t localOffset(const Cell & T, const std::string & cell_dof_name) const;
      
    protected:
      DiscreteSpaceDescriptor m_descriptor;
 
      size_t m_n_local_cell_dofs;
      size_t m_n_local_edge_dofs;
      size_t m_n_local_vertex_dofs;
 
      std::map<std::string, size_t> m_cell_dofs_offsets;
      std::map<std::string, size_t> m_edge_dofs_offsets;
      std::map<std::string, size_t> m_vertex_dofs_offsets;
    };    
    
  } // namespace DSL
 
} // namespace HArDCore2D
 
#endif