gmWedgeShape.h

Go to the documentation of this file.

/************************************************************************
**
** Copyright (C) 2014 by Carlos Augusto Teixera Mendes
** All rights reserved.
**
** This file is part of the "GeMA" software. It's use should respect
** the terms in the license agreement that can be found together 
** with this source code.
** It is provided AS IS, with NO WARRANTY OF ANY KIND, 
** INCLUDING THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR 
** A PARTICULAR PURPOSE.
**
************************************************************************/

#ifndef _GEMA_WEDGE_SHAPE_H_
#define _GEMA_WEDGE_SHAPE_H_

#include "gmShape.h"


//--------------------------------------------------------------------------------------------------------------------------
//      Wedge Elements
//--------------------------------------------------------------------------------------------------------------------------

class GMC_API_EXPORT GmWedgeShape : public GmShape
{
public:
  // See comments on the base class
  virtual int numNaturalCoord() const { return 3; }

  // See comments on the base class
  virtual int numCartesianCoord() const { return 3; }

  // See comments on the base class
  virtual void naturalCoordLimits(int coord, double* min, double* max) const 
  { 
    *min = 0.0; *max = 1.0;
    if (coord == 2) { *min = -1.0; }
  }

  // See comments on the base class
  virtual void naturalCenter(GmVector& coord) const { coord.set_size(3); coord[0] = coord[1] = 1/3.0; coord[2] = 0.0; } // Center = (1/3, 1/3, 0) for all wedges

  virtual bool translateEdgePoint(int edge, const GmVector& srcEdgeCoord, GmVector& elementCoord) const;
  virtual bool translateFacePoint(int face, const GmVector& srcFaceCoord, GmVector& elementCoord) const;

  virtual double shapeCartesianPartialsFromPJ(const GmMatrix& P, const GmMatrix& J, GmMatrix& dN, bool transposed = false) const;

  // See comments on the base class
  virtual double scaledJacobianDet(const GmMatrix& J) const { return 0.5* arma::det(J); } 

  // See comments on the base class
  virtual void jacobianAndPartials(const GmVector& ncoord, const GmMatrix& X, GmMatrix& J, GmMatrix& P, bool transposed = false) const
  {
    jacIndependentNatCoord(ncoord, X, J, P, transposed);
  }

  // See comments on the base class
  virtual double borderScalingFactor(int border, const GmVector& borderCoord, const GmVector& elementCoord,
                                     const GmMatrix& X, bool transposed = false) const
  {
    return faceScalingFactor(border, borderCoord, elementCoord, X, transposed);
  }

  virtual double edgeScalingFactor(int border, const GmVector& borderCoord, const GmVector& elementCoord,
                                   const GmMatrix& X, bool transposed = false) const;

  virtual double faceScalingFactor(int border, const GmVector& borderCoord, const GmVector& elementCoord,
                                   const GmMatrix& X, bool transposed = false) const;

protected:
  const int*    fixedEdgeCoord(int border) const;
  const double* fixedEdgeValue(int border) const;
  int           fixedFaceCoord(int border) const;
  double        fixedFaceValue(int border) const;

  // See comments on the base class. Calls the default algorithm for Wedge elements
  virtual bool gradientBasedCartesianToNatural(const GmVector& coord, const GmMatrix& X, double tol,
                                               int maxIter, double natTol, GmVector& ncoord, bool* inside) const
  {
    return gradientBasedCartesianToNaturalWedge(coord, X, tol, maxIter, natTol, ncoord, inside);
  }
};

class GMC_API_EXPORT GmWedge6Shape : public GmWedgeShape
{
public:
  // See comments on the base class
  virtual GmCellType elemType() const { return GM_WEDGE6; }

  // See comments on the base class
  virtual int numFunctions() const { return 6; }

  virtual void nodeNaturalCoord(int node, GmVector& coord) const;

  virtual void shapeValues(const GmVector& ncoord, GmVector& N) const;
  virtual void shapePartials(const GmVector& ncoord, GmMatrix& dN, bool transposed = false) const;
};


class GMC_API_EXPORT GmWedge15Shape : public GmWedgeShape
{
public:
  // See comments on the base class
  virtual GmCellType elemType() const { return GM_WEDGE15; }

  // See comments on the base class
  virtual int numFunctions() const { return 15; }

  virtual void nodeNaturalCoord(int node, GmVector& coord) const;

  virtual void shapeValues(const GmVector& ncoord, GmVector& N) const;
  virtual void shapePartials(const GmVector& ncoord, GmMatrix& dN, bool transposed = false) const;
};

#endif