Select Git revision
MPyramid.cpp
Forked from
gmsh / gmsh
Source project has a limited visibility.
MPyramid.cpp 10.95 KiB
// Gmsh - Copyright (C) 1997-2012 C. Geuzaine, J.-F. Remacle
//
// See the LICENSE.txt file for license information. Please report all
// bugs and problems to <gmsh@geuz.org>.
#include "MPyramid.h"
#include "Numeric.h"
#include "Context.h"
#include "BergotBasis.h"
#include "BasisFactory.h"
int MPyramid::getVolumeSign()
{
double mat[3][3];
mat[0][0] = _v[1]->x() - _v[0]->x();
mat[0][1] = _v[3]->x() - _v[0]->x();
mat[0][2] = _v[4]->x() - _v[0]->x();
mat[1][0] = _v[1]->y() - _v[0]->y();
mat[1][1] = _v[3]->y() - _v[0]->y();
mat[1][2] = _v[4]->y() - _v[0]->y();
mat[2][0] = _v[1]->z() - _v[0]->z();
mat[2][1] = _v[3]->z() - _v[0]->z();
mat[2][2] = _v[4]->z() - _v[0]->z();
double d = det3x3(mat);
if(d < 0.) return -1;
else if(d > 0.) return 1;
else return 0;
}
const nodalBasis* MPyramid::getFunctionSpace(int o) const
{
int order = (o == -1) ? getPolynomialOrder() : o;
int nv = getNumVolumeVertices();
if ((nv == 0) && (o == -1)) {
switch (order) {
case 1: return BasisFactory::create(MSH_PYR_5);
case 2: return BasisFactory::create(MSH_PYR_14);
case 0: return BasisFactory::create(MSH_PYR_1);
case 3: return BasisFactory::create(MSH_PYR_29);
case 4: return BasisFactory::create(MSH_PYR_50);
case 5: return BasisFactory::create(MSH_PYR_77);
case 6: return BasisFactory::create(MSH_PYR_110);
case 7: return BasisFactory::create(MSH_PYR_149);
case 8: return BasisFactory::create(MSH_PYR_194);
case 9: return BasisFactory::create(MSH_PYR_245);
default: Msg::Error("Order %d pyramid function space not implemented", order);
}
}
else {
switch (order) {
case 0: return BasisFactory::create(MSH_PYR_1);
case 1: return BasisFactory::create(MSH_PYR_5);
case 2: return BasisFactory::create(MSH_PYR_14);
case 3: return BasisFactory::create(MSH_PYR_30);
case 4: return BasisFactory::create(MSH_PYR_55);
case 5: return BasisFactory::create(MSH_PYR_91);
case 6: return BasisFactory::create(MSH_PYR_140);
case 7: return BasisFactory::create(MSH_PYR_204);
case 8: return BasisFactory::create(MSH_PYR_285);
case 9: return BasisFactory::create(MSH_PYR_385);
default: Msg::Error("Order %d pyramid function space not implemented", order);
}
}
return 0;
}
MPyramidN::~MPyramidN() {}
double MPyramidN::distoShapeMeasure()
{
#if defined(HAVE_MESH)
_disto = 0.0; //qmDistorsionOfMapping(this);
#else
_disto = 0.;
#endif
return _disto;
}
int MPyramidN::getNumEdgesRep(){ return 8 * CTX::instance()->mesh.numSubEdges; }
void MPyramidN::getEdgeRep(int num, double *x, double *y, double *z, SVector3 *n)
{
int numSubEdges = CTX::instance()->mesh.numSubEdges;
static double pp[5][3] = {{-1,-1,0},{1,-1,0},{1,1,0},{-1,1,0},{0,0,1}};
static int ed [8][2] = {{0,1},{0,3},{0,4},{1,2},{1,4},{2,3},{2,4},{3,4}};
int iEdge = num / numSubEdges;
int iSubEdge = num % numSubEdges;
int iVertex1 = ed [iEdge][0];
int iVertex2 = ed [iEdge][1];
double t1 = (double) iSubEdge / (double) numSubEdges;
double u1 = pp[iVertex1][0] * (1.-t1) + pp[iVertex2][0] * t1;
double v1 = pp[iVertex1][1] * (1.-t1) + pp[iVertex2][1] * t1;
double w1 = pp[iVertex1][2] * (1.-t1) + pp[iVertex2][2] * t1;
double t2 = (double) (iSubEdge+1) / (double) numSubEdges;
double u2 = pp[iVertex1][0] * (1.-t2) + pp[iVertex2][0] * t2;
double v2 = pp[iVertex1][1] * (1.-t2) + pp[iVertex2][1] * t2;
double w2 = pp[iVertex1][2] * (1.-t2) + pp[iVertex2][2] * t2;
SPoint3 pnt1, pnt2;
pnt(u1,v1,w1,pnt1);
pnt(u2,v2,w2,pnt2);
x[0] = pnt1.x(); x[1] = pnt2.x();
y[0] = pnt1.y(); y[1] = pnt2.y();
z[0] = pnt1.z(); z[1] = pnt2.z();
// not great, but better than nothing
static const int f[8] = {0, 1, 0, 2, 2, 3, 2, 3};
n[0] = n[1] = getFace(f[iEdge]).normal();
}
int MPyramidN::getNumFacesRep(){ return 6 * SQU(CTX::instance()->mesh.numSubEdges); }
static void _myGetFaceRep(MPyramid *pyr, int num, double *x, double *y, double *z,
SVector3 *n, int numSubEdges)
{
static double pp[5][3] = {
{-1,-1,0},
{1,-1,0},
{1,1,0},
{-1,1,0},
{0,0,1}
};
int iFace = num / (numSubEdges * numSubEdges);
int iSubFace = num % (numSubEdges * numSubEdges);
if (iFace > 3) {
iFace = 4;
iSubFace = num % (2*numSubEdges * numSubEdges);
}
int iVertex1, iVertex2, iVertex3, iVertex4;
if (iFace < 4) {
iVertex1 = pyr->faces_pyramid(iFace,0);
iVertex2 = pyr->faces_pyramid(iFace,1); iVertex3 = pyr->faces_pyramid(iFace,2);
} else {
iVertex1 = 0;
iVertex2 = 1;
iVertex3 = 2;
iVertex4 = 3;
}
if (iFace < 4) {
int ix = 0, iy = 0;
int nbt = 0;
for (int i = 0; i < numSubEdges; i++){
int nbl = (numSubEdges - i - 1) * 2 + 1;
nbt += nbl;
if (nbt > iSubFace){
iy = i;
ix = nbl - (nbt - iSubFace);
break;
}
}
const double d = 1. / numSubEdges;
SPoint3 pnt1, pnt2, pnt3;
double u1, v1, u2, v2, u3, v3;
if (ix % 2 == 0){
u1 = ix / 2 * d; v1= iy*d;
u2 = (ix / 2 + 1) * d ; v2 = iy * d;
u3 = ix / 2 * d ; v3 = (iy+1) * d;
}
else{
u1 = (ix / 2 + 1) * d; v1= iy * d;
u2 = (ix / 2 + 1) * d; v2= (iy + 1) * d;
u3 = ix / 2 * d ; v3 = (iy + 1) * d;
}
double U1 = pp[iVertex1][0] * (1.-u1-v1) + pp[iVertex2][0] * u1 + pp[iVertex3][0] * v1;
double U2 = pp[iVertex1][0] * (1.-u2-v2) + pp[iVertex2][0] * u2 + pp[iVertex3][0] * v2;
double U3 = pp[iVertex1][0] * (1.-u3-v3) + pp[iVertex2][0] * u3 + pp[iVertex3][0] * v3;
double V1 = pp[iVertex1][1] * (1.-u1-v1) + pp[iVertex2][1] * u1 + pp[iVertex3][1] * v1;
double V2 = pp[iVertex1][1] * (1.-u2-v2) + pp[iVertex2][1] * u2 + pp[iVertex3][1] * v2;
double V3 = pp[iVertex1][1] * (1.-u3-v3) + pp[iVertex2][1] * u3 + pp[iVertex3][1] * v3;
double W1 = pp[iVertex1][2] * (1.-u1-v1) + pp[iVertex2][2] * u1 + pp[iVertex3][2] * v1;
double W2 = pp[iVertex1][2] * (1.-u2-v2) + pp[iVertex2][2] * u2 + pp[iVertex3][2] * v2;
double W3 = pp[iVertex1][2] * (1.-u3-v3) + pp[iVertex2][2] * u3 + pp[iVertex3][2] * v3;
pyr->pnt(U1, V1, W1, pnt1);
pyr->pnt(U2, V2, W2, pnt2);
pyr->pnt(U3, V3, W3, pnt3);
x[0] = pnt1.x(); x[1] = pnt2.x(); x[2] = pnt3.x();
y[0] = pnt1.y(); y[1] = pnt2.y(); y[2] = pnt3.y();
z[0] = pnt1.z(); z[1] = pnt2.z(); z[2] = pnt3.z();
SVector3 d1(x[1] - x[0], y[1] - y[0], z[1] - z[0]);
SVector3 d2(x[2] - x[0], y[2] - y[0], z[2] - z[0]);
n[0] = crossprod(d1, d2);
n[0].normalize();
n[1] = n[0];
n[2] = n[0];
}
else {
SPoint3 pnt1, pnt2, pnt3;
/*
0 0 1
0 1 2
0 1 2 3
0 1 2 3 4
0 1 2 3 4 5
*/
// on each layer, we have (numSubEdges) * 2 triangles
// ix and iy are the coordinates of the sub-quadrangle
int io = iSubFace%2;
int ix = (iSubFace/2)/numSubEdges;
int iy = (iSubFace/2)%numSubEdges;
const double d = 2. / numSubEdges;
double ox = -1. + d*ix;
double oy = -1. + d*iy;
if (io == 0){
double U1 =
pp[iVertex1][0] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][0] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][0] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][0] * (1.-ox)*(1+oy)*.25;
double V1 =
pp[iVertex1][1] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][1] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][1] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][1] * (1.-ox)*(1+oy)*.25;
double W1 =
pp[iVertex1][2] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][2] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][2] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][2] * (1.-ox)*(1+oy)*.25;
ox += d;
double U2 =
pp[iVertex1][0] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][0] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][0] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][0] * (1.-ox)*(1+oy)*.25;
double V2 =
pp[iVertex1][1] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][1] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][1] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][1] * (1.-ox)*(1+oy)*.25;
double W2 =
pp[iVertex1][2] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][2] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][2] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][2] * (1.-ox)*(1+oy)*.25;
oy += d;
double U3 =
pp[iVertex1][0] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][0] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][0] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][0] * (1.-ox)*(1+oy)*.25;
double V3 =
pp[iVertex1][1] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][1] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][1] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][1] * (1.-ox)*(1+oy)*.25;
double W3 =
pp[iVertex1][2] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][2] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][2] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][2] * (1.-ox)*(1+oy)*.25;
pyr->pnt(U1,V1,W1, pnt1);
pyr->pnt(U2,V2,W2, pnt2);
pyr->pnt(U3,V3,W3, pnt3);
}
else{
double U1 =
pp[iVertex1][0] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][0] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][0] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][0] * (1.-ox)*(1+oy)*.25;
double V1 =
pp[iVertex1][1] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][1] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][1] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][1] * (1.-ox)*(1+oy)*.25;
double W1 =
pp[iVertex1][2] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][2] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][2] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][2] * (1.-ox)*(1+oy)*.25;
ox += d;
oy += d;
double U2 =
pp[iVertex1][0] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][0] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][0] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][0] * (1.-ox)*(1+oy)*.25;
double V2 =
pp[iVertex1][1] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][1] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][1] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][1] * (1.-ox)*(1+oy)*.25;
double W2 =
pp[iVertex1][2] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][2] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][2] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][2] * (1.-ox)*(1+oy)*.25;
ox -= d;
double U3 =
pp[iVertex1][0] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][0] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][0] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][0] * (1.-ox)*(1+oy)*.25;
double V3 =
pp[iVertex1][1] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][1] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][1] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][1] * (1.-ox)*(1+oy)*.25;
double W3 =
pp[iVertex1][2] * (1.-ox)*(1-oy)*.25 +
pp[iVertex2][2] * (1.+ox)*(1-oy)*.25 +
pp[iVertex3][2] * (1.+ox)*(1+oy)*.25 +
pp[iVertex4][2] * (1.-ox)*(1+oy)*.25;
pyr->pnt(U1,V1,W1, pnt1);
pyr->pnt(U2,V2,W2, pnt2);
pyr->pnt(U3,V3,W3, pnt3);
}
n[0] = 1;
n[1] = 1;
n[2] = 1;
x[0] = pnt1.x(); x[1] = pnt2.x(); x[2] = pnt3.x();
y[0] = pnt1.y(); y[1] = pnt2.y(); y[2] = pnt3.y();
z[0] = pnt1.z(); z[1] = pnt2.z(); z[2] = pnt3.z(); }
}
void MPyramidN::getFaceRep(int num, double *x, double *y, double *z, SVector3 *n)
{
_myGetFaceRep(this, num, x, y, z, n, CTX::instance()->mesh.numSubEdges);
}