P13DElement.h

Go to the documentation of this file.

#ifndef P13DELEMENT
#define P13DELEMENT

#include "P1nDElement.h"
#include "P12DElement.h"
#include "ShapesEvaluatedP13D.h"
#include "Tetrahedron.h"
#include "ElementGeometry.h"


#include <iostream>
#include <vector>
#include <cassert>

template<size_t NF>
class P13DElement: public P1nDElement<NF> {
public:
  P13DElement (const Tetrahedron& _elemGeom): P1nDElement<NF> (_elemGeom) {
    ShapesP13D::Bulk modelShape (_elemGeom);
    Element_::addBasisFunctions (modelShape);
  }


  P13DElement (const P13DElement<NF>& that): P1nDElement<NF> (that) {

  }

  virtual P13DElement<NF>* clone () const {
    return new P13DElement<NF> (*this);
  }

};


// Class P13DTrace:
template<size_t NF>
class P13DTrace: public P1nDTrace<NF> {
public:

  P13DTrace (const P13DElement<NF> &BaseElement, const typename P1nDTrace<NF>::FaceLabel& FaceName,
      const typename P1nDTrace<NF>::ShapeType& Type);

  virtual ~P13DTrace () {
  }

  P13DTrace (const P13DTrace<NF> &OldElement_) :
    P1nDTrace<NF> (OldElement_) {
  }

  virtual P13DTrace<NF>* clone () const {
    return new P13DTrace<NF> (*this);
  }

protected:
  static const size_t FaceNodes[];

private:
  void checkArgs (const typename P1nDTrace<NF>::FaceLabel& flabel, const typename P1nDTrace<NF>::ShapeType& shType) {
    assert (shType != P1nDTrace<NF>::TwoDofs);
  }

};

template<size_t NF>
const size_t P13DTrace<NF>::FaceNodes[] = { 2, 1, 0, 2, 0, 3, 2, 3, 1, 0, 1, 3 };
// Implementation of class P13DTrace:

template<size_t NF>
P13DTrace<NF>::P13DTrace (const P13DElement<NF> &BaseElement,
    const typename P1nDTrace<NF>::FaceLabel& FaceName, const typename P1nDTrace<NF>::ShapeType& Type) :
  P1nDTrace<NF> (BaseElement) {

  checkArgs (FaceName, Type);

  const ElementGeometry& tetGeom = Element::getGeometry ();

  Triangle<3> FaceGeom (tetGeom.getConnectivity ()[FaceNodes[FaceName * 3 + 0]],
      tetGeom.getConnectivity ()[FaceNodes[FaceName * 3 + 1]],
      tetGeom.getConnectivity ()[FaceNodes[FaceName * 3 + 2]]);

  if (Type == P1nDTrace<NF>::ThreeDofs) {
    ShapesP13D::Faces ModelShape (&FaceGeom);
    Element_::addBasisFunctions (ModelShape);
  } else {
    //Type == FourDofs
    switch (FaceName) {
      case P1nDTrace<NF>::FaceOne: {
        ShapesP13D::FaceOne ModelShape(FaceGeom);
        Element_::addBasisFunctions(ModelShape);
        break;
      }

      case P1nDTrace<NF>::FaceTwo: {
        ShapesP13D::FaceTwo ModelShape(FaceGeom);
        Element_::addBasisFunctions(ModelShape);
        break;
      }

      case P1nDTrace<NF>::FaceThree: {
        ShapesP13D::FaceThree ModelShape(FaceGeom);
        Element_::addBasisFunctions(ModelShape);
        break;
      }

      case P1nDTrace<NF>::FaceFour: {
        ShapesP13D::FaceFour ModelShape(FaceGeom);
        Element_::addBasisFunctions(ModelShape);
        break;
      }

    }
  }
}

// Class for ElementBoundaryTraces:

template<size_t NF>
class P13DElementBoundaryTraces: public P1nDBoundaryTraces<NF> {
public:

  P13DElementBoundaryTraces (const P13DElement<NF> &BaseElement,
      const std::vector<typename P1nDTrace<NF>::FaceLabel>& faceLabels,
      typename P13DTrace<NF>::ShapeType Type):
        P1nDBoundaryTraces<NF> (BaseElement, faceLabels, Type) {

  }

  virtual ~P13DElementBoundaryTraces () {
  }

  P13DElementBoundaryTraces (const P13DElementBoundaryTraces<NF> &OldElem) :
    P1nDBoundaryTraces<NF> (OldElem) {
  }

  P13DElementBoundaryTraces<NF> * clone () const {
    return new P13DElementBoundaryTraces<NF> (*this);
  }

  size_t dofMap (size_t FaceIndex, size_t field, size_t dof) const;
};

// Implementation of class P13DElementBoundaryTraces
template<size_t NF>
size_t P13DElementBoundaryTraces<NF>::dofMap (size_t FaceIndex, size_t field, size_t dof) const {
  size_t val;

  if (ElementBoundaryTraces::getTrace (FaceIndex).getDof (field) == 4) {
    val = dof;
  } else { // Three dofs per face.
  
    // size_t FaceNodes[] = { 2, 1, 0, 2, 0, 3, 2, 3, 1, 0, 1, 3 };
    const size_t* FaceNodes = P13DTrace<NF>::FaceNodes;
    size_t facenum = ElementBoundaryTraces::getTraceFaceIds ()[FaceIndex];
    val = FaceNodes[3 * facenum + dof];
  }
  return val;
}

template<size_t NF>
class P13D: public SpecificElementFamily {

public:
  typedef P13DElement<NF> Bulk;

  typedef P13DTrace<NF> Face;

  typedef P13DElementBoundaryTraces<NF> Traces;

};



#endif

---