element2D.hh

Go to the documentation of this file.

 
#ifndef element2DLin_hh
#define element2DLin_hh
 
#include "element.hh"
#include "geometry/cell2D.hh"
 
namespace concepts {
  // forward declaration
  template<typename F>
  class ElementGraphics;
}
 
namespace linearFEM {
  // forward declaration
  class TriangleGraphics;
  class QuadGraphics;
 
  // ************************************************************** Triangle **
 
  class Triangle : public Element {
  public:
    Triangle(const concepts::Triangle2d& cell, uint idx[]) :
      Element(3, idx), cell_(cell) {}
    Triangle(const concepts::Triangle2d& cell, concepts::TColumn<Real>* T0,
       concepts::TColumn<Real>* T1 = 0) :
      Element(T0, T1), cell_(cell) {}
    virtual const concepts::Triangle& support() const {
      return cell_.connector(); }
    virtual const concepts::Triangle2d& cell() const { return cell_; }
    Real shapefct(const uint i, const Real xi1, const Real xi2) const;
    concepts::Real2d shapefctD(const uint i) const;
    virtual const concepts::ElementGraphics<Real>* graphics() const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
  private:
    const concepts::Triangle2d& cell_;
    static std::unique_ptr<TriangleGraphics> graphics_;
  };
 
  // ****************************************************************** Quad **
 
  class Quad : public Element {
  public:
    Quad(const concepts::Quad2d& cell, uint idx[]) :
      Element(4, idx), cell_(cell) {}
    Quad(const concepts::Quad2d& cell, concepts::TColumn<Real>* T0,
   concepts::TColumn<Real>* T1 = 0) : Element(T0, T1), cell_(cell) {}
    virtual const concepts::Quad& support() const { return cell_.connector(); }
    virtual const concepts::Quad2d& cell() const { return cell_; }
    void evaluate(Real* res, const Real* x, const uint n) const;
    void evaluateD(concepts::Real2d* res, const Real* x, const uint n) const;
    virtual const concepts::ElementGraphics<Real>* graphics() const;
  protected:
    virtual std::ostream& info(std::ostream& os) const;
  private:
    const concepts::Quad2d& cell_;
    static std::unique_ptr<QuadGraphics> graphics_;
  };
 
} // namespace linearFEM
 
#endif // element2DLin_hh