dealfewrapper.h

Go to the documentation of this file.

#ifndef  _MY_OrthoMeshEqVar_
#define _MY_OrthoMeshEqVar_
#include "globals.h"
#include <base/quadrature_lib.h>
#include <fe/fe.h>
#include "orthomesh.h"


typedef std::vector<FEFaceValues<3>* > FaceValuesVec;

class DealFEWrapper 
{
 private:
  

  
  FEValues<3> *m_fev;
  Triangulation<3> *m_pTria;
  DoFHandler<3> *m_pDof;
  DoFHandler<DIM>::active_cell_iterator *m_pDoFCell;
  OrthoMesh::Cell_It *m_pCell;
  FaceDirection3D m_faceDir;
  FaceValuesVec m_facesValues;
  const FiniteElement<3> *m_fe;
  OrthoMesh *m_pMesh;
 protected:
  //void updateFE_Values(OrthoMesh &mesh,FEValues<3> &feValues);
  //void updateFE_FaceValues(OrthoMesh &mesh,FaceValuesVec &vFaceValues); 

 public:
  DealFEWrapper();
  void  initFEValues(OrthoMesh &mesh,const FiniteElement<3> &fe, const Quadrature<3> &quadrature, const Quadrature<2> &face_quadrature,const UpdateFlags update_flags,const UpdateFlags face_update_flags);  
  void clear();
  OrthoMesh& getMesh(){assert(m_pMesh);return *m_pMesh;}
  const FiniteElement<3>* getFE(){return m_fe;}
  void setCell(OrthoMesh::Cell_It &cell);
  void setFace(OrthoMesh::Face_It &face);

  FaceDirection3D getFaceAtBoundaryDir(){return m_faceDir;}

  OrthoMesh::Cell_It& getCell(){assert(m_pCell);return *m_pCell;} 
  ~DealFEWrapper();
  FEValues<3>& getFEValues(){assert(m_fev);return *m_fev;}
  
  //Info functions about the Deal finite element 
  unsigned n_dofs_per_face();
  unsigned n_dofs_per_cell();
  unsigned n_quadrature_points();
  unsigned n_face_quadrature_points(); 
  unsigned n_dofs(); 
  //Info about local indices mapping in the parametric element
  Index local_dof_component(Index cDof){return m_fe->system_to_component_index(cDof).first;} 
  Index local_face_to_local_cell_map(unsigned face_no, unsigned local_face_dof); 
  void printDealFEInfo();

  
  //Values at cells
  double                  shape_value(unsigned i,unsigned qPoint) {assert(m_fev);return m_fev->shape_value(i,qPoint);} 
  const  Tensor<1,DIM>&   shape_grad(unsigned i,unsigned qPoint)  {assert(m_fev);return m_fev->shape_grad(i,qPoint);} 
  double                  shape_value(unsigned i,unsigned component,unsigned qPoint) {assert(m_fev);return m_fev->shape_value_component(i,qPoint,component);}
  Tensor<1,DIM>           shape_grad (unsigned i,unsigned component,unsigned qPoint) {assert(m_fev);return m_fev->shape_grad_component(i,qPoint,component);}
  double                  JxW(unsigned qPoint)  {assert(m_fev);return m_fev->JxW(qPoint);}
  Point3D                 get_qpoint(unsigned qPoint);
  //Values at faces
  double                  JxWFace(unsigned qPoint)                     {return m_facesValues[m_faceDir]->JxW(qPoint);}
  const Tensor<1,DIM>&    normal_vector(unsigned face_no,unsigned qPoint)               {return m_facesValues[face_no]->normal_vector(qPoint);}
  const  Tensor<1,DIM>&   shape_grad_face (unsigned face_no,unsigned i,unsigned qPoint) {return m_facesValues[face_no]->shape_grad(i,qPoint);}
  double                  shape_value_face(unsigned face_no,unsigned i,unsigned qPoint) {return m_facesValues[face_no]->shape_value(i,qPoint);}
  Point3D                 get_face_qpoint(unsigned qPoint);

  /* The events functions */
  void changedCellEvent(){}
};


inline unsigned DealFEWrapper::n_dofs_per_face() 
{
  assert(m_fe);
  return m_fe->n_dofs_per_face();
}

inline unsigned DealFEWrapper::n_dofs_per_cell() 
{
  assert(m_fe);
  return m_fe->n_dofs_per_cell();
}


inline unsigned DealFEWrapper::n_dofs() 
{
  assert(m_fe);
  return   m_fe->n_dofs_per_face()*getMesh().numFaces() +
           m_fe->n_dofs_per_vertex()*getMesh().numVertices() +
           m_fe->n_dofs_per_hex()*getMesh().numCells();
}

inline unsigned DealFEWrapper::n_quadrature_points() 
{
  assert(m_fe);
  return m_fev->n_quadrature_points;
}


inline unsigned DealFEWrapper::n_face_quadrature_points() 
{
  assert(m_fe);
  assert(m_facesValues.size() != 0);
  return m_facesValues[0]->n_quadrature_points;
}

inline Index DealFEWrapper::local_face_to_local_cell_map(unsigned face_no, unsigned local_face_dof) 
{
  assert(m_fe); //Assert m_fe is initialized
  return m_fe->face_to_cell_index(local_face_dof,face_no);
}

#endif

---