mappedfevaluesbase.h

Go to the documentation of this file.

#ifndef  _MY_MappedFEValuesBase_
#define _MY_MappedFEValuesBase_
#include "finiteelementinterface.h"
#include "mappingorthomesh.h"

enum COMPUTE_DEF {COMPUTE_VALUES=1,COMPUTE_GRADS=2,COMPUTE_PTS=4,COMPUTE_J=8,COMPUTE_JxW=16};


class MappedFEValuesBase 
{

   protected:
  FiniteElementInterface &_fe;
  ArrayOfVecDouble _pts;  
  VecDouble _weights;
  unsigned m_n_dofs; 
  Matrix _Values;  
  std::vector<VecDouble> _Grad; 
  std::vector<VecDouble> _Mapped_Points;
  std::vector<Matrix> _TGrad;  
  VecDouble _J; 
  VecDouble _JxW; 
  unsigned m_compute_flags;
  
  void setPoints(const ArrayOfVecDouble &v,const VecDouble &weights);

  
 public:
  
 MappedFEValuesBase(FiniteElementInterface &fe,Index n_dofs)
   :_fe(fe),m_n_dofs(n_dofs),m_compute_flags(0){}


  unsigned n_pts(){return _pts.size();}

  double function_value(Index pt, Index local_dof) const 
  {
    //assert(m_compute_flags | COMPUTE_VALUES);
    return _Values(pt,local_dof);
  }
  


  const VecDouble& function_grad(Index pt, Index local_dof) const
  {
    //assert(m_compute_flags | COMPUTE_GRADS);
    return _Grad[pt*m_n_dofs + local_dof];
  }

  double J(unsigned ptIndex)
  {
    //assert(m_compute_flags | COMPUTE_J);
    return _J(ptIndex);
  }

  double JxW(unsigned ptIndex)
  {
    //assert(m_compute_flags | COMPUTE_JxW);

    return _JxW(ptIndex);
  }

  void compute_points(MappingOrthoMesh &map); 

  const VecDouble& mapped_pts(unsigned ptIndex)
  {
    return _Mapped_Points[ptIndex];
  }

  ~MappedFEValuesBase(){}
  
  void print(std::ostream &out=std::cout);
};

#endif

---