main.c

Go to the documentation of this file.

 
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <string.h>
#include <ctype.h>
#include <math.h>
#include <float.h>

// if the header file is in the "include" directory
// #include "libmmg3d.h"
// if the header file is in "include/mmg/mmg3d"
#include "mmg/mmg3d/libmmg3d.h"
 
#define MAX0(a,b)     (((a) > (b)) ? (a) : (b))
#define MAX4(a,b,c,d)  (((MAX0(a,b)) > (MAX0(c,d))) ? (MAX0(a,b)) : (MAX0(c,d)))
 
int main(int argc,char *argv[]) {
  MMG5_pMesh      mmgMesh;
  MMG5_pSol       mmgSol;
  int             ier;
  /* To save final mesh in a file */
  FILE*           inm;
  /* To manually recover the mesh */
  MMG5_int        k,np,ne,nt,na,nc,nr,nreq,ref,Tetra[4],Tria[3],Edge[2];
  MMG5_int        ktet[2];
  int             typEntity, typSol,iface[2];
  int             *corner, *required, *ridge;
  double          Point[3],Sol;
  char            *fileout, *solout;
 
  fprintf(stdout,"  -- TEST MMG3DLIB \n");
 
  if ( argc != 2 ) {
    printf(" Usage: %s fileout\n",argv[0]);
    return(1);
  }
 
  /* Name and path of the mesh file */
  fileout = (char *) calloc(strlen(argv[1]) + 6, sizeof(char));
  if ( fileout == NULL ) {
    perror("  ## Memory problem: calloc");
    exit(EXIT_FAILURE);
  }
  strcpy(fileout,argv[1]);
  strcat(fileout,".mesh");
 
  solout = (char *) calloc(strlen(argv[1]) + 5, sizeof(char));
  if ( solout == NULL ) {
    perror("  ## Memory problem: calloc");
    exit(EXIT_FAILURE);
  }
  strcpy(solout,argv[1]);
  strcat(solout,".sol");
 
 

  /* args of InitMesh:
   * MMG5_ARG_start: we start to give the args of a variadic func
   * MMG5_ARG_ppMesh: next arg will be a pointer over a MMG5_pMesh
   * &mmgMesh: pointer toward your MMG5_pMesh (that store your mesh)
   * MMG5_ARG_ppMet: next arg will be a pointer over a MMG5_pSol storing a metric
   * &mmgSol: pointer toward your MMG5_pSol (that store your metric) */
  mmgMesh = NULL;
  mmgSol  = NULL;
 
  MMG3D_Init_mesh(MMG5_ARG_start,
                  MMG5_ARG_ppMesh,&mmgMesh,MMG5_ARG_ppMet,&mmgSol,
                  MMG5_ARG_end);


  if ( MMG3D_Set_meshSize(mmgMesh,12,12,0,20,0,0) != 1 )  exit(EXIT_FAILURE);

  if ( MMG3D_Set_vertex(mmgMesh,0  ,0  ,0  ,0,  1) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,0.5,0  ,0  ,0,  2) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,0.5,0  ,1  ,0,  3) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,0  ,0  ,1  ,0,  4) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,0  ,1  ,0  ,0,  5) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,0.5,1  ,0  ,0,  6) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,0.5,1  ,1  ,0,  7) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,0  ,1  ,1  ,0,  8) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,1  ,0  ,0  ,0,  9) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,1  ,1  ,0  ,0, 10) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,1  ,0  ,1  ,0, 11) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_vertex(mmgMesh,1  ,1  ,1  ,0, 12) != 1 )  exit(EXIT_FAILURE);

  if ( MMG3D_Set_tetrahedron(mmgMesh,  1,  4,  2,  8,1, 1) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  8,  3,  2,  7,1, 2) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  5,  2,  6,  8,1, 3) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  5,  8,  1,  2,1, 4) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  7,  2,  8,  6,1, 5) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  2,  4,  3,  8,1, 6) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  9,  2,  3,  7,2, 7) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  7, 11,  9, 12,2, 8) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  6,  9, 10,  7,2, 9) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  6,  7,  2,  9,2,10) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh, 12,  9,  7, 10,2,11) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_tetrahedron(mmgMesh,  9,  3, 11,  7,2,12) != 1 )  exit(EXIT_FAILURE);

  if ( MMG3D_Set_triangle(mmgMesh,  1,  4,  8, 3, 1) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  1,  2,  4, 3, 2) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  8,  3,  7, 3, 3) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  5,  8,  6, 3, 4) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  5,  6,  2, 3, 5) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  5,  2,  1, 3, 6) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  5,  1,  8, 3, 7) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  7,  6,  8, 3, 8) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  4,  3,  8, 3, 9) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  2,  3,  4, 3,10) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  9,  3,  2, 4,11) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh, 11,  9, 12, 4,12) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  7, 11, 12, 4,13) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  6,  7, 10, 4,14) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  6, 10,  9, 4,15) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  6,  9,  2, 4,16) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh, 12, 10,  7, 4,17) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh, 12,  9, 10, 4,18) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  3, 11,  7, 4,19) != 1 )  exit(EXIT_FAILURE);
  if ( MMG3D_Set_triangle(mmgMesh,  9, 11,  3, 4,20) != 1 )  exit(EXIT_FAILURE);
 


  if ( MMG3D_Set_solSize(mmgMesh,mmgSol,MMG5_Vertex,12,MMG5_Scalar) != 1 )
    exit(EXIT_FAILURE);

  for(k=1 ; k<=12 ; k++) {
    if ( MMG3D_Set_scalarSol(mmgSol,0.5,k) != 1 ) exit(EXIT_FAILURE);
  }

  if ( MMG3D_Chk_meshData(mmgMesh,mmgSol) != 1 ) exit(EXIT_FAILURE);

  ier = MMG3D_mmg3dlib(mmgMesh,mmgSol);
 
  if ( ier == MMG5_STRONGFAILURE ) {
    fprintf(stdout,"BAD ENDING OF MMG3DLIB: UNABLE TO SAVE MESH\n");
    return(ier);
  } else if ( ier == MMG5_LOWFAILURE )
    fprintf(stdout,"BAD ENDING OF MMG3DLIB\n");
 


  if( !(inm = fopen(fileout,"w")) ) {
    fprintf(stderr,"  ** UNABLE TO OPEN OUTPUT MESH FILE.\n");
    exit(EXIT_FAILURE);
  }
  fprintf(inm,"MeshVersionFormatted 2\n");
  fprintf(inm,"\nDimension 3\n");

  if ( MMG3D_Get_meshSize(mmgMesh,&np,&ne,NULL,&nt,NULL,&na) !=1 )  exit(EXIT_FAILURE);
 
  /* Table to know if a vertex is corner */
  corner = (int*)calloc(np+1,sizeof(int));
  if ( !corner ) {
    perror("  ## Memory problem: calloc");
    exit(EXIT_FAILURE);
  }
  /* Table to know if a vertex/tetra/tria/edge is required */
  required = (int*)calloc(MAX4(np,ne,nt,na)+1 ,sizeof(int));
  if ( !required ) {
    perror("  ## Memory problem: calloc");
    exit(EXIT_FAILURE);
  }
  /* Table to know if an edge delimits a sharp angle */
  ridge = (int*)calloc(na+1 ,sizeof(int));
  if ( !ridge ) {
    perror("  ## Memory problem: calloc");
    exit(EXIT_FAILURE);
  }
 
  nreq = 0; nc = 0;
  fprintf(inm,"\nVertices\n%"MMG5_PRId"\n",np);
  for(k=1; k<=np; k++) {
    if ( MMG3D_Get_vertex(mmgMesh,&(Point[0]),&(Point[1]),&(Point[2]),
                          &ref,&(corner[k]),&(required[k])) != 1 )
      exit(EXIT_FAILURE);
    fprintf(inm,"%.15lg %.15lg %.15lg %"MMG5_PRId" \n",Point[0],Point[1],Point[2],ref);
    if ( corner[k] )  nc++;
    if ( required[k] )  nreq++;
  }
  fprintf(inm,"\nCorners\n%"MMG5_PRId"\n",nc);
  for(k=1; k<=np; k++) {
    if ( corner[k] )  fprintf(inm,"%"MMG5_PRId" \n",k);
  }
  fprintf(inm,"\nRequiredVertices\n%"MMG5_PRId"\n",nreq);
  for(k=1; k<=np; k++) {
    if ( required[k] )  fprintf(inm,"%"MMG5_PRId" \n",k);
  }
  free(corner);
  corner = NULL;
 
  nreq = 0;
  fprintf(inm,"\nTriangles\n%"MMG5_PRId"\n",nt);
  for(k=1; k<=nt; k++) {
    if ( MMG3D_Get_triangle(mmgMesh,&(Tria[0]),&(Tria[1]),&(Tria[2]),
                            &ref,&(required[k])) != 1 )
      exit(EXIT_FAILURE);
    fprintf(inm,"%"MMG5_PRId" %"MMG5_PRId" %"MMG5_PRId" %"MMG5_PRId" \n",Tria[0],Tria[1],Tria[2],ref);
    if ( required[k] )  nreq++;
  }
  fprintf(inm,"\nRequiredTriangles\n%"MMG5_PRId"\n",nreq);
  for(k=1; k<=nt; k++) {
    if ( required[k] )  fprintf(inm,"%"MMG5_PRId" \n",k);
  }
 
  /* Facultative step : if you want to know with which tetrahedra a triangle is
   * connected */
  for(k=1; k<=nt; k++) {
    ktet[0]  = ktet[1]  = 0;
    iface[0] = iface[1] = 0;
 
    if (! MMG3D_Get_tetFromTria(mmgMesh,k,ktet,iface) ) {
      printf("Get tet from tria fail.\n");
      return 0;
    }
    printf("Tria %"MMG5_PRId" is connected with tet %"MMG5_PRId" "
           "(face %d) and %"MMG5_PRId" (face %d) \n",
           k,ktet[0],iface[0],ktet[1],iface[1]);
  }
 
  nreq = 0;nr = 0;
  fprintf(inm,"\nEdges\n%"MMG5_PRId"\n",na);
  for(k=1; k<=na; k++) {
    if ( MMG3D_Get_edge(mmgMesh,&(Edge[0]),&(Edge[1]),&ref,
                        &(ridge[k]),&(required[k])) != 1 )  exit(EXIT_FAILURE);
    fprintf(inm,"%"MMG5_PRId" %"MMG5_PRId" %"MMG5_PRId" \n",Edge[0],Edge[1],ref);
    if ( ridge[k] )  nr++;
    if ( required[k] )  nreq++;
  }
  fprintf(inm,"\nRequiredEdges\n%"MMG5_PRId"\n",nreq);
  for(k=1; k<=na; k++) {
    if ( required[k] )  fprintf(inm,"%"MMG5_PRId" \n",k);
  }
  fprintf(inm,"\nRidges\n%"MMG5_PRId"\n",nr);
  for(k=1; k<=na; k++) {
    if ( ridge[k] )  fprintf(inm,"%"MMG5_PRId" \n",k);
  }
 
  nreq = 0;
  fprintf(inm,"\nTetrahedra\n%"MMG5_PRId"\n",ne);
  for(k=1; k<=ne; k++) {
    if ( MMG3D_Get_tetrahedron(mmgMesh,&(Tetra[0]),&(Tetra[1]),&(Tetra[2]),&(Tetra[3]),
                               &ref,&(required[k])) != 1 )  exit(EXIT_FAILURE);
    fprintf(inm,"%"MMG5_PRId" %"MMG5_PRId" %"MMG5_PRId" %"MMG5_PRId" %"MMG5_PRId" \n",
            Tetra[0],Tetra[1],Tetra[2],Tetra[3],ref);
    if ( required[k] )  nreq++;
  }
  fprintf(inm,"\nRequiredTetrahedra\n%"MMG5_PRId"\n",nreq);
  for(k=1; k<=ne; k++) {
    if ( required[k] )  fprintf(inm,"%"MMG5_PRId" \n",k);
  }
 
  fprintf(inm,"\nEnd\n");
  fclose(inm);
 
  free(required);
  required = NULL;
  free(ridge);
  ridge    = NULL;

  if( !(inm = fopen(solout,"w")) ) {
    fprintf(stderr,"  ** UNABLE TO OPEN OUTPUT FILE.\n");
    exit(EXIT_FAILURE);
  }
  fprintf(inm,"MeshVersionFormatted 2\n");
  fprintf(inm,"\nDimension 3\n");

  if ( MMG3D_Get_solSize(mmgMesh,mmgSol,&typEntity,&np,&typSol) != 1 )
    exit(EXIT_FAILURE);
 
  if ( ( typEntity != MMG5_Vertex )  || ( typSol != MMG5_Scalar ) )
    exit(EXIT_FAILURE);
 
  fprintf(inm,"\nSolAtVertices\n%"MMG5_PRId"\n",np);
  fprintf(inm,"1 1 \n\n");
  for(k=1; k<=np; k++) {
    if ( MMG3D_Get_scalarSol(mmgSol,&Sol) != 1 )  exit(EXIT_FAILURE);
    fprintf(inm,"%.15lg \n",Sol);
  }
  fprintf(inm,"\nEnd\n");
  fclose(inm);

  MMG3D_Free_all(MMG5_ARG_start,
                 MMG5_ARG_ppMesh,&mmgMesh,MMG5_ARG_ppMet,&mmgSol,
                 MMG5_ARG_end);
 
  free(fileout);
  fileout = NULL;
 
  free(solout);
  solout = NULL;
 
 
  return(ier);
}