linDG3D/element.hh

Go to the documentation of this file.

/* Classes for DG elements
 */

#ifndef dgElement_hh
#define dgElement_hh

#include "basics/typedefs.hh"
#include "space/element.hh"
#include "geometry/cell3D.hh"
#include "geometry/topology3D.hh"
#include "function/vector.hh"
#include "space/tmatrix.hh"
#include "toolbox/elementGraphics.hh"
#include "space/postProcess.hh"
#include "graphics/dataDX.hh"
 
namespace linDG3D {

  using concepts::Real;
  
  // Forward declarations
  class FvdgP0TetGraphics;
  class FvdgP1TetGraphics;

  // ************************************************************** Elements **
  class FvdgElement : public concepts::ElementWithCell<Real> {
  public:
    FvdgElement(const concepts::Tetrahedron3d& cell, uint idx[], uint m);
    const concepts::Tetrahedron& support() const { return cell_.connector(); }
    const concepts::TMatrixBase<Real>& T() const { return *T_; }
    const concepts::Real3d vertex(const uint i) const { 
      return cell_.vertex(i);
    }
    virtual const concepts::Tetrahedron3d& cell() const { return cell_; }
    virtual Real shapeFct(const uint fctIdx, const concepts::Real3d xi) const 
      = 0;
    virtual Real shapeFctEdgeCenter(const uint fctIdx, const uint edgeIdx) 
      const = 0;
    virtual Real shapeFctVertex(const uint fctIdx, const uint vtxIdx) 
      const = 0;
    virtual concepts::Real3d shapeFctGradient(const uint fctIdx) const = 0;
    Real solution(const concepts::Vector<Real>& sol,
      const concepts::Real3d xi) const;
    virtual Real solutionEdgeCenter(const concepts::Vector<Real>& sol,
            const uint edgeIdx) const = 0;
    virtual Real solutionVertex(const concepts::Vector<Real>& sol,
        const uint vertexIdx) const = 0;
    concepts::Real3d edgeCenter(uint edgeIdx) const;
    virtual const concepts::ElementGraphics<Real>* graphics() const = 0;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
    
    std::auto_ptr<concepts::TMatrixBase<Real> > T_;
    const concepts::Tetrahedron3d& cell_;
    static Real eta_[6][3];
  };

  //********************************************************** FvdgP0TetElem **
  class FvdgP0TetElem : public FvdgElement {
  public:
    FvdgP0TetElem(const concepts::Tetrahedron3d& cell, uint idx[]);
    Real shapeFct(const uint fctIdx, const concepts::Real3d xi) const {
      return 1.0;
    }
    Real shapeFctEdgeCenter(const uint fctIdx, const uint edgeIdx) const {
      return 1.0;
    }
    Real shapeFctVertex(const uint fctIdx, const uint vtxIdx) const {
      return 1.0;
    }
    concepts::Real3d shapeFctGradient(const uint fctIdx) const {
      return concepts::Real3d(0.0, 0.0, 0.0);
    }
    Real solutionEdgeCenter(const concepts::Vector<Real>& sol,
          const uint edgeIdx) const {
      return sol(T_->index(0));
    }
    Real solutionVertex(const concepts::Vector<Real>& sol,
      const uint vertexIdx) const {
      return sol(T_->index(0));
    }
    const concepts::ElementGraphics<Real>* graphics() const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
    static std::auto_ptr<FvdgP0TetGraphics> graphics_;
  };
  
  //********************************************************** FvdgP1TetElem **
  class FvdgP1TetElem : public FvdgElement {
  public:
    FvdgP1TetElem(const concepts::Tetrahedron3d& cell, uint idx[]);
    Real shapeFct(const uint fctIdx, const concepts::Real3d xi) const;
    Real shapeFctEdgeCenter(const uint fctIdx, const uint edgeIdx) const {
      conceptsAssert(edgeIdx < 6, concepts::Assertion());
      conceptsAssert(fctIdx < 4, concepts::Assertion());
      return shapeFctEdgeCenter_[edgeIdx][fctIdx];
    }
    Real shapeFctVertex(const uint fctIdx, const uint vtxIdx) const {
      conceptsAssert(vtxIdx < 4, concepts::Assertion());
      conceptsAssert(fctIdx < 4, concepts::Assertion());
      return shapeFctVertex_[vtxIdx][fctIdx];
    }
    concepts::Real3d shapeFctGradient(const uint fctIdx) const;
    Real solutionEdgeCenter(const concepts::Vector<Real>& sol,
          const uint edgeIdx) const;
    Real solutionVertex(const concepts::Vector<Real>& sol,
      const uint vertexIdx) const;
    const concepts::ElementGraphics<Real>* graphics() const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
    static Real shapeFctEdgeCenter_[6][4];
    static Real shapeFctGradient_[3][4];
    static uint nonVanishingNbr_[6];
    static Real nonVanishingSign_[6];
    static Real shapeFctVertex_[4][4];
    static std::auto_ptr<FvdgP1TetGraphics> graphics_;
  };

  // *************************************************** TetrahedronGraphics **

  class FvdgP0TetGraphics : public concepts::ElementGraphics<Real> {
  public:
    virtual void operator()(const concepts::Element<Real>& elm,
          enum graphicsType type,
          concepts::CellPostprocess<Real>& post) const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
  };

  class FvdgP1TetGraphics : public concepts::ElementGraphics<Real> {
  public:
    virtual void operator()(const concepts::Element<Real>& elm,
          enum graphicsType type,
          concepts::CellPostprocess<Real>& post) const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
  };  
} // namespace linDG3D

#endif