linDG2D/triangle.hh

Go to the documentation of this file.

// triangle element for linear DGFEM

#ifndef DGTriangle_hh
#define DGTriangle_hh

#include "linearFEM/element.hh"
#include "geometry/cell2D.hh"

namespace concepts {
  // forward declaration
  template<typename F>
  class ElementGraphics;
}

namespace linDG2D {

  // forward declaration
  class TriangleGraphics;

  using concepts::Real;

  // ************************************************************** Triangle **

  class Triangle : public linearFEM::Element {
  public:
    Triangle(const concepts::Triangle2d& cell, uint nDoF, uint idx[])
      : linearFEM::Element(nDoF, idx), cell_(cell) {}
    virtual const concepts::Triangle& support() const {
      return cell_.connector();
    }
    virtual const concepts::Triangle2d& cell() const { return cell_; }
    virtual Real shapeFct(const uint i, const Real xi1, const Real xi2) const = 0;
    virtual Real solution(const concepts::Vector<Real>& coeffs,
                          const Real xi1, const Real xi2) const = 0;
    virtual const concepts::ElementGraphics<Real>* graphics() const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
  private:
    const concepts::Triangle2d& cell_;
    static std::auto_ptr<TriangleGraphics> graphics_;
  };

  // ************************************************************ TriangleP1 **

  class TriangleP1 : public Triangle {
  public:
    TriangleP1(const concepts::Triangle2d& cell, uint idx[])
      : Triangle(cell, 3, idx) {}
    virtual Real shapeFct(const uint i, const Real xi1, const Real xi2) const;
    concepts::Real2d shapeFctGradient(const uint i) const;
    virtual Real solution(const concepts::Vector<Real>& coeffs,
                          const Real xi1, const Real xi2) const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
  };

  // ************************************************************ TriangleP0 **

  class TriangleP0 : public Triangle {
  public:
    TriangleP0(const concepts::Triangle2d& cell, uint* idx)
      : Triangle(cell, 1, idx) {}
    virtual Real shapeFct(const uint i, const Real xi1, const Real xi2) const {
      return 1;
    }
    virtual Real solution(const concepts::Vector<Real>& coeffs,
                          const Real xi1, const Real xi2) const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
  };

} // namespace linDG2D

#endif // DGTriangle_hh