app-radu/triangleP2.hh

Go to the documentation of this file.

// p=2 triangles for stochastic diffusion

#ifndef triangleP2_HH
#define triangleP2_HH

#include "basics/exceptions.hh"
#include "basics/typedefs.hh"
#include "linearFEM/element2D.hh"
#include "toolbox/elementGraphics.hh"
#include "toolbox/hashMap.hh"
#include "space/space.hh"

using concepts::Real;

namespace concepts {
  // forward declarations
  class Mesh2;
  class BoundaryConditions;
}

// ***************************************************** TrianglesP2Graphics **

class TriangleP2Graphics : 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;
};

// ************************************************************** TriangleP2 **

class TriangleP2 : public linearFEM::Element {
public:
  TriangleP2(const concepts::Triangle2d& cell, uint idx[]) :
    linearFEM::Element(6, idx), cell_(cell) {}
  TriangleP2(const concepts::Triangle2d& cell, concepts::TColumn<Real>* T0,
       concepts::TColumn<Real>* T1 = 0) :
    linearFEM::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 Real xi1, const Real xi2) const;
  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<TriangleP2Graphics> graphics_;
};

namespace concepts {
  template<>
  class Scan<TriangleP2> : public Scan<concepts::Element<Real> > {
  public:
    virtual TriangleP2& operator++(int) = 0;
  };
}

// ************************************************************* TrianglesP2 **

class TrianglesP2 : public concepts::Space<Real> {
public:
  typedef concepts::Scan<TriangleP2> Scan;
  typedef void (*SMap)(const concepts::TColumn<Real>&,
           concepts::TColumn<Real>&);
  TrianglesP2(concepts::Mesh2& msh, uint level,
        concepts::BoundaryConditions* bc = 0);
  virtual ~TrianglesP2();
  virtual uint dim() const { return dim_; }
  virtual uint nelm() const { return nelm_; }
  virtual TrianglesP2::Scan* scan() const;
  void rebuild();
protected:
  virtual std::ostream& info(std::ostream& os) const;
private:
  uint dim_;
  uint nelm_;
  bool rebuild_;
  concepts::Mesh2& msh_;
  concepts::BoundaryConditions* bc_;
  concepts::Joiner<TriangleP2*, 1>* elm_;
  __gnu_cxx::hash_map<uint, short> adj_;
  __gnu_cxx::hash_map<uint, short> ctrl0_;
  __gnu_cxx::hash_map<uint, short> ctrl0L_;
  __gnu_cxx::hash_map<uint, short> ctrl1_;
  __gnu_cxx::hash_map<uint, short> ctrl1L_;
  __gnu_cxx::hash_map<uint, short> ctrl2_;
  __gnu_cxx::hash_map<uint, uint> dofVtx_, dofEdg_;

  static SMap S_[4];

  void rebuild0_(concepts::Connector2& cntr, const int l, int& L);
  void rebuild1_(concepts::Triangle2d& cell,
     concepts::TColumn<Real>* T0 = 0);
  void deactivate_(concepts::Connector1& edg);
};

#endif // triangleP2_HH