LBPM/tests/test_dcel_tri_normal.cpp
2021-09-18 16:33:26 -04:00

142 lines
3.6 KiB
C++

#include <iostream>
#include <math.h>
#include "analysis/Minkowski.h"
#include "common/Domain.h"
#include "common/SpherePack.h"
using namespace std;
/*
* Compare the measured and analytical curvature for a sphere
*
*/
std::shared_ptr<Database> loadInputs( )
{
//auto db = std::make_shared<Database>( "Domain.in" );
auto db = std::make_shared<Database>();
db->putScalar<int>( "BC", 0 );
db->putVector<int>( "nproc", { 1, 1, 1 } );
db->putVector<int>( "n", { 16, 16, 16 } );
db->putScalar<int>( "nspheres", 1 );
db->putVector<double>( "L", { 1, 1, 1 } );
return db;
}
int main(int argc, char **argv)
{
Utilities::startup( argc, argv );
Utilities::MPI comm( MPI_COMM_WORLD );
int toReturn = 0;
{
int i,j,k;
// Load inputs
auto db = loadInputs( );
int Nx = db->getVector<int>( "n" )[0];
int Ny = db->getVector<int>( "n" )[1];
int Nz = db->getVector<int>( "n" )[2];
std::shared_ptr<Domain> Dm = std::shared_ptr<Domain>(new Domain(db,comm));
Nx+=2; Ny+=2; Nz+=2;
DoubleArray SDs(Nx,Ny,Nz);
DoubleArray SDs_x(Nx,Ny,Nz);
DoubleArray SDs_y(Nx,Ny,Nz);
DoubleArray SDs_z(Nx,Ny,Nz);
printf("Set distance map \n");
for (k=0; k<Nz; k++){
for (j=0; j<Ny; j++){
for (i=0; i<Nx; i++){
SDs(i,j,k) = sqrt((1.0*i-0.5*Nx)*(1.0*i-0.5*Nx)+(1.0*j-0.5*Ny)*(1.0*j-0.5*Ny)+(1.0*k-0.5*Nz)*(1.0*k-0.5*Nz))-0.3*Nx;
}
}
}
pmmc_MeshGradient(SDs,SDs_x,SDs_y,SDs_z,Nx,Ny,Nz);
DCEL object;
Point P1,P2,P3;
Point U,V,W;
int e1,e2,e3;
double nx,ny,nz;
double isovalue = 0.f;
int count_plus=0; int count_minus=0;
int count_check=0;
double dotprod;
for (int k=1; k<Nz-1; k++){
for (int j=1; j<Ny-1; j++){
for (int i=1; i<Nx-1; i++){
object.LocalIsosurface(SDs,isovalue,i,j,k);
for (int idx=0; idx<object.TriangleCount; idx++){
// normal from gradient
nx = SDs_x(i,j,k);
ny = SDs_y(i,j,k);
nz = SDs_z(i,j,k);
// triangle normals
e1 = object.Face(idx);
U = object.TriNormal(e1);
dotprod=U.x*nx + U.y*ny + U.z*nz;
if (dotprod < 0){
//printf("edge 1: negative %f \n",dotprod);
count_minus++;
}
else{
//printf("edge 1: positive %f \n",dotprod);
count_plus++;
}
// test that normal is independent of the edge
e2 = object.halfedge.next(e1);
V = object.TriNormal(e2);
dotprod=V.x*nx + V.y*ny + V.z*nz;
if (dotprod < 0){
//printf("negative %f \n",dotprod);
count_minus++;
}
else{
//printf("edge 2: positive %f \n",dotprod);
count_plus++;
}
// check third edge
e3 = object.halfedge.next(e2);
W = object.TriNormal(e3);
dotprod=W.x*nx + W.y*ny + W.z*nz;
if (dotprod < 0){
//printf("edge 3: negative %f \n",dotprod);
count_minus++;
}
else{
//printf("edge 3: positive %f \n",dotprod);
count_plus++;
}
if (object.halfedge.next(e3) != e1){
printf("Error in object.next \n");
count_check++;
}
dotprod=U.x*V.x+U.y*V.y+U.z*V.z;
if (dotprod < 0 ){
printf("normal for edge 1 / 2 disagree \n");
count_check++;
}
dotprod=U.x*W.x+U.y*W.y+U.z*W.z;
if (dotprod < 0 ){
printf("normal for edge 1 / 3 disagree \n");
count_check++;
}
dotprod=W.x*V.x+W.y*V.y+W.z*V.z;
if (dotprod < 0 ){
printf("normal for edge 2 / 3 disagree \n");
count_check++;
}
}
}
}
}
if (count_minus > 0 && count_plus>0) toReturn=1;
if (count_check > 0) toReturn=2;
else printf("Succeeded. \n");
}
Utilities::shutdown();
return toReturn;
}