globaldofspace.hpp

Go to the documentation of this file.

#ifndef GLOBALDOFSPACE_HPP
#define GLOBALDOFSPACE_HPP
 
#include <localdofspace.hpp>
 
namespace HArDCore2D
{
 
  class GlobalDOFSpace : public LocalDOFSpace {
  public:
    GlobalDOFSpace(
             const Mesh & mesh,
             size_t n_local_vertex_dofs,
             size_t n_local_edge_dofs,
             size_t n_local_cell_dofs
             );
 
    //------------------------------------------------------------------------------
    // Global offsets
    //------------------------------------------------------------------------------
    
    inline size_t globalOffset(const Vertex & V) const
    {
      return V.global_index() * m_n_local_vertex_dofs;
 
    }
 
    inline size_t globalOffset(const Edge & E) const
    {
      return m_mesh.n_vertices() * m_n_local_vertex_dofs
        + E.global_index() * m_n_local_edge_dofs;
    }
 
    inline size_t globalOffset(const Cell & T) const
    {
      return m_mesh.n_vertices() * m_n_local_vertex_dofs
        + m_mesh.n_edges() * m_n_local_edge_dofs
        + T.global_index() * m_n_local_cell_dofs;
    }
    
    //------------------------------------------------------------------------------
    // Restrictions
    //------------------------------------------------------------------------------
 
    Eigen::VectorXd restrictEdge(size_t iE, const Eigen::VectorXd & vh) const;
 
    Eigen::VectorXd restrictCell(size_t iT, const Eigen::VectorXd & vh) const;
 
    inline Eigen::VectorXd restrict(const Edge & E, const Eigen::VectorXd vh) const
    {
      return restrictEdge(E.global_index(), vh);
    }
 
    inline Eigen::VectorXd restrict(const Cell & T, const Eigen::VectorXd vh) const
    {
      return restrictCell(T.global_index(), vh);
    }
    
    //------------------------------------------------------------------------------
    // Extensions
    //------------------------------------------------------------------------------
 
    Eigen::MatrixXd extendOperator(const Cell & T, const Edge & E, const Eigen::MatrixXd & opE) const;
 
    //------------------------------------------------------------------------------
    // Global DOF indices for an element T
    //------------------------------------------------------------------------------
 
    std::vector<size_t> globalDOFIndices(const Cell & T) const;
 
    //------------------------------------------------------------------------------
    // Functions to handle labels
    //------------------------------------------------------------------------------
    inline void setLabelDOF(const size_t i, const int label)
    {
      m_labelDOF[i] = label;
    }
    
    inline int getLabelDOF(const size_t i)
    {
      return m_labelDOF[i];
    }
    
  private:
    std::vector<int> m_labelDOF;    
    
  };
 
} // namespace HArDCore2D
 
#endif