/* =============================================================================
**  This file is part of the mmg software package for the tetrahedral
**  mesh modification.
**  Copyright (c) Bx INP/Inria/UBordeaux/UPMC, 2004- .
**
**  mmg is free software: you can redistribute it and/or modify it
**  under the terms of the GNU Lesser General Public License as published
**  by the Free Software Foundation, either version 3 of the License, or
**  (at your option) any later version.
**
**  mmg is distributed in the hope that it will be useful, but WITHOUT
**  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
**  FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
**  License for more details.
**
**  You should have received a copy of the GNU Lesser General Public
**  License and of the GNU General Public License along with mmg (in
**  files COPYING.LESSER and COPYING). If not, see
**  <http://www.gnu.org/licenses/>. Please read their terms carefully and
**  use this copy of the mmg distribution only if you accept them.
** =============================================================================
*/

/**
 * Example of use of the mmg3dmov function of the mmg3d library (basic use of
 * lagrangian motion option)
 *
 * \author Luca Cirrottola (Inria)
 * \author Charles Dapogny (LJLL, UPMC)
 * \author Cécile Dobrzynski (Inria / IMB, Université de Bordeaux)
 * \author Pascal Frey (LJLL, UPMC)
 * \author Algiane Froehly (Inria / IMB, Université de Bordeaux)
 * \version 5
 * \copyright GNU Lesser General Public License.
 */

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

/** Include the mmg3d library hader file */
// 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"

int main(int argc,char *argv[]) {
  MMG5_pMesh      mmgMesh;
  MMG5_pSol       mmgSol,mmgDisp;
  int             ier;
  char            *inname,*outname;

  fprintf(stdout,"  -- TEST MMG3DMOV \n");

  if ( argc != 3 ) {
    printf(" Usage: %s filein fileout \n",argv[0]);
    return(1);
  }

  /* Name and path of the mesh files */
  inname = (char *) calloc(strlen(argv[1]) + 1, sizeof(char));
  if ( inname == NULL ) {
    perror("  ## Memory problem: calloc");
    exit(EXIT_FAILURE);
  }
  strcpy(inname,argv[1]);

  outname = (char *) calloc(strlen(argv[2]) + 1, sizeof(char));
  if ( outname == NULL ) {
    perror("  ## Memory problem: calloc");
    exit(EXIT_FAILURE);
  }
  strcpy(outname,argv[2]);

  /** 1) Initialisation of mesh and sol structures */
  /* 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;
  mmgDisp = NULL;
  MMG3D_Init_mesh(MMG5_ARG_start,
                  MMG5_ARG_ppMesh,&mmgMesh,MMG5_ARG_ppMet,&mmgSol,
                  MMG5_ARG_ppDisp,&mmgDisp,
                  MMG5_ARG_end);

  /** 2) Build mesh in MMG5 format */
  /** Two solutions: just use the MMG3D_loadMesh function that will read a .mesh(b)
     file formatted or manually set your mesh using the MMG3D_Set* functions */

  /** with MMG3D_loadMesh function */
  if ( MMG3D_loadMesh(mmgMesh,inname) != 1 )  exit(EXIT_FAILURE);

  /** 3) Build displacement in MMG5 format */
  /** Two solutions: just use the MMG3D_loadSol function that will read a .sol(b)
      file formatted or manually set your sol using the MMG3D_Set* functions */

  /**------------------- Lagrangian motion option ----------------------------*/
  /* Ask for lagrangian motion (mode 1) */
  if ( MMG3D_Set_iparameter(mmgMesh,mmgDisp,MMG3D_IPARAM_lag, 2) != 1 )
    exit(EXIT_FAILURE);

  /* Get mesh size */
  int np,ne,nprism,nt,nquad,na;
  if( MMG3D_Get_meshSize(mmgMesh,&np,&ne,&nprism,&nt,&nquad,&na) != 1 )
    exit(EXIT_FAILURE);

  /* Initialize displacement structure */
  if( MMG3D_Set_solSize(mmgMesh,mmgDisp,MMG5_Vertex,np,MMG5_Vector) != 1 )
    exit(EXIT_FAILURE);

  /* Fill the structure, here constant displacement in the z-direction */
  double delta = 0.08;
  int k,ia,ib,ic,ref,isRequired;
  for( k = 1; k <= nt; k++ ) {
    /* Get triangle */
    if( MMG3D_Get_triangle(mmgMesh, &ia, &ib, &ic, &ref, &isRequired) != 1 )
      exit(EXIT_FAILURE);
    /* Apply displacement only on ref 10 */
    if( ref == 10 ) {
      if( MMG3D_Set_vectorSol(mmgDisp, 0.0, 0.0, delta, ia) != 1 )
        exit(EXIT_FAILURE);
      if( MMG3D_Set_vectorSol(mmgDisp, 0.0, 0.0, delta, ib) != 1 )
        exit(EXIT_FAILURE);
      if( MMG3D_Set_vectorSol(mmgDisp, 0.0, 0.0, delta, ic) != 1 )
        exit(EXIT_FAILURE);
    }
  }

  /**------------------- Lagrangian motion computation ---------------------*/
  /** 5) (not mandatory): set your global parameters using the
      MMG3D_Set_iparameter and MMG3D_Set_dparameter function
      (resp. for integer parameters and double param)*/


  /* Don't remesh (1) or remesh (0) boundary surfaces */
  if ( MMG3D_Set_iparameter(mmgMesh,mmgDisp,MMG3D_IPARAM_nosurf, 0) != 1 )
    exit(EXIT_FAILURE);

  /* Verbosity */
  if ( MMG3D_Set_iparameter(mmgMesh,mmgDisp,MMG3D_IPARAM_verbose, 1) != 1 )
    exit(EXIT_FAILURE);

  /* Prescribe a large Hausdorff on outer boundaries */
  if ( MMG3D_Set_dparameter(mmgMesh,mmgDisp,MMG3D_DPARAM_hausd, 10) != 1 )
    exit(EXIT_FAILURE);

  /* Set the number of local parameters */
  if ( MMG3D_Set_iparameter(mmgMesh,mmgDisp,MMG3D_IPARAM_numberOfLocalParam,2) != 1 )
    exit(EXIT_FAILURE);

  /* Set local parameters on surfaces indentified by their reference in order
   * to assign a smaller Hausdorff (min and max edge size won't be applied with
   * Lagrangian motion anyway... */
  double hmindummy = 0.001;
  double hmaxdummy = 10;
  double hausd = 0.007;
  if ( MMG3D_Set_localParameter(mmgMesh,mmgDisp,MMG5_Triangle,2,hmindummy,hmaxdummy,hausd) !=1 )
    exit(EXIT_FAILURE);
  if ( MMG3D_Set_localParameter(mmgMesh,mmgDisp,MMG5_Triangle,10,hmindummy,hmaxdummy,hausd) != 1 )
    exit(EXIT_FAILURE);


  /** remesh function */
  ier = MMG3D_mmg3dmov(mmgMesh,mmgSol,mmgDisp);

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

  /* (Not mandatory) Automatically save the mesh */
  if ( MMG3D_saveMesh(mmgMesh,outname) != 1 )
    exit(EXIT_FAILURE);

  /* 9) free the MMG3D5 structures */
  MMG3D_Free_all(MMG5_ARG_start,
                 MMG5_ARG_ppMesh,&mmgMesh,MMG5_ARG_ppMet,&mmgSol,
                 MMG5_ARG_ppDisp,&mmgDisp,
                 MMG5_ARG_end);
  free(inname);
  inname = NULL;

  free(outname);
  outname = NULL;

  return(ier);
}