mymethod.cpp

Go to the documentation of this file.

/* ./bridging/mymethod.cpp 
**********************************
Copyright INRIA and CEA 

author : Guillaume ANCIAUX (anciaux@labri.fr, g.anciaux@laposte.net)

The LibMultiScale is a C++ parallel framework for the multiscale
coupling methods dedicated to material simulations. This framework
provides an API which makes it possible to program coupled simulations
and integration of already existing codes.

This Project is done in a collaboration between INRIA Futurs Bordeaux
within ScAlApplix team and CEA/DPTA Ile de France. 

This software is governed by the CeCILL-C license under French law and
abiding by the rules of distribution of free software.  You can  use, 
modify and/ or redistribute the software under the terms of the CeCILL-C
license as circulated by CEA, CNRS and INRIA at the following URL
"http://www.cecill.info". 

As a counterpart to the access to the source code and  rights to copy,
modify and redistribute granted by the license, users are provided only
with a limited warranty  and the software's author,  the holder of the
economic rights,  and the successive licensors  have only  limited
liability. 

In this respect, the user's attention is drawn to the risks associated
with loading,  using,  modifying and/or developing or reproducing the
software by the user in light of its specific status of free software,
that may mean  that it is complicated to manipulate,  and  that  also
therefore means  that it is reserved for developers  and  experienced
professionals having in-depth computer knowledge. Users are therefore
encouraged to load and test the software's suitability as regards their
requirements in conditions enabling the security of their systems and/or 
data to be ensured and,  more generally, to use and operate it in the 
same conditions as regards security. 

The fact that you are presently reading this means that you have had
knowledge of the CeCILL-C license and that you accept its terms.
***********************************/

#define NEED_STAMP
#define NEED_LAMMPS
#define NEED_LIBMESH
#define NEED_MD1D

#define ORDRE LINEAR
#define LOCAL_MODULE MOD_COUPLING

template <typename DomainA, typename DomainC,unsigned int Dim>
void MyMethodFull<DomainA,DomainC,Dim>::Init(){

  DUMP("Computing freedom degrees weight",DBG_INFO);
  Parent::BuildAtomsWeight();
  Parent::BuildContinuWeight();  

  Parent::BuildPositions();
  Parent::BuildShapeMatrix(Parent::positions,Parent::atomes_rec.nbElem());
  delete [] Parent::positions;

  Parent::BuildFictifsPositions();
  BuildShapeMatrixFictifs(Parent::positions,Parent::atomes_fictifs.nbElem());
  delete [] Parent::positions;

  Parent::CorrectSurfaceEffect();

  unsigned int taille = Parent::atomes_rec.nbElem();
  dist = new double[taille];
  orig_FE = new double[taille];
  orig_atom = new double[taille];

  DUMP("Allocating matrix (" << taille << ")",DBG_INFO);  
  A = new FullMatrix(taille,taille);
  //  A_sav = new FullMatrix(taille,taille);
  lagrange2 = new double[taille];


  DUMP("Allocating vector",DBG_INFO);  
  rhs = new Vector(taille);
  rhs->zero();
  A->zero();
  //A_sav->zero();
  
  DUMP("BuildConstraintMatrices",DBG_INFO);
  BuildConstraintMatrix();
  DUMP("Init done",DBG_INFO);
}

template <typename DomainA, typename DomainC,unsigned int Dim>
void MyMethodFull<DomainA,DomainC,Dim>::Stucking(){
  /* c l'implementation de la methode generique qui
     colle les deux domaines ensemble sur la zone de 
     recouvrement */

  STARTTIMER("Correct Surface Effect");
  Parent::CorrectSurfaceEffect();
  STOPTIMER("Correct Surface Effect");

  STARTTIMER();
  BuildRHS();
  STOPTIMER("BuildRHS");
  
  STARTTIMER();
  SolveConstraint();
  STOPTIMER("Solving constraint");
  
  STARTTIMER();
  ApplyCorrection();
  STOPTIMER("Correcting");
  

}

template <typename DomainA, typename DomainC,unsigned int Dim>
void MyMethodFull<DomainA,DomainC,Dim>::BuildConstraintMatrix(){
  /* ici demarre l'algo de recoupement des éléments 

  On associe dans un tableau de taille nbElementRec 
  des petites matrices de taille nb_atome_par_element x connectivite 
  
  qui vont conenir un bloc de la  ShapeMatrix 

  pour cela on commence par creer un tableau qui donne l'element conteneur 
  pour chaque atome

  ensuite il nous faut le nombre de d'atomes par elements

  ensuite on parcours les elements et on creer la matrice
  aux bonnes dimensions et on la remplie !!

  ZONE : Continu
  */

  MechaContLibMesh<Dim> & model = static_cast<MechaContLibMesh<Dim>&>(Parent::domC.getModel());  
  double xI,xJ;
  std::vector<unsigned int> nodesI;
  std::vector<unsigned int> nodesJ;
  std::vector<double> shapesI;
  std::vector<double> shapesJ;

  A->zero();

  int taille = Parent::atomes_rec.nbElem();

  for (int i = 0; i < taille ; ++i)
    for (int j = 0; j < taille ; ++j){

      xI = Parent::atomes_rec.Get(i).position0(0);
      xJ = Parent::atomes_rec.Get(j).position0(0);

      model.GlobalIndexes(Parent::elems_rec.Get(Parent::assoc[i]),nodesI);
      model.GlobalIndexes(Parent::elems_rec.Get(Parent::assoc[j]),nodesJ);

      model.ComputeShapes(Parent::elems_rec.Get(Parent::assoc[i]),shapesI,xI);
      model.ComputeShapes(Parent::elems_rec.Get(Parent::assoc[j]),shapesJ,xJ);

      if (nodesI[0] == nodesJ[0]){
        (*A)(i,j) += shapesI[0]*shapesJ[0]/model.Mass(nodesI[0]);
        //(*A)(i,j) += shapesI[0]*shapesJ[0];
      }

      if (nodesI[1] == nodesJ[1]){
        (*A)(i,j) += shapesI[1]*shapesJ[1]/model.Mass(nodesI[1]);
        //(*A)(i,j) += shapesI[1]*shapesJ[1];
      }

      if (nodesI[0] == nodesJ[1]){
        (*A)(i,j) += shapesI[0]*shapesJ[1]/model.Mass(nodesI[0]);
        //(*A)(i,j) += shapesI[0]*shapesJ[1];
      }

      if (nodesI[1] == nodesJ[0]){
        (*A)(i,j) += shapesI[1]*shapesJ[0]/model.Mass(nodesI[1]);
        //(*A)(i,j) += shapesI[1]*shapesJ[0];
      }
    }

  A->PrintToFile("constraint_matrix.mat"); 
  //la factoLU
  A->FactoLU();
}

static std::ofstream out_rhs("rhs_matrix.mat");
static std::ofstream out_dist("dist.plot");

template <typename DomainA, typename DomainC,unsigned int Dim>
void MyMethodFull<DomainA,DomainC,Dim>::BuildRHS(){

  std::vector<unsigned int> nodesI;
  std::vector<double> shapesI;

  double xI;
  MechaContLibMesh<Dim> & model = static_cast<MechaContLibMesh<Dim>&>(Parent::domC.getModel());  
  typename Parent::_Vec_ & v = Parent::domC.getDofs().V();

  rhs->zero();
  int taille = Parent::atomes_rec.nbElem();

  for (int i = 0 ; i < taille ; ++i){
    xI = Parent::atomes_rec.Get(i).position0(0);      

    model.GlobalIndexes(Parent::elems_rec.Get(Parent::assoc[i]),nodesI);
    model.ComputeShapes(Parent::elems_rec.Get(Parent::assoc[i]),shapesI,xI);

    (*rhs)(i) = v(nodesI[0])*shapesI[0] + v(nodesI[1])*shapesI[1] - Parent::atomes_rec.Get(i).vitesse(0);
    lagrange2[i] = (*rhs)(i);
    (*rhs)(i) *= Parent::lambdasA[i];

    dist[i] = (1-Parent::lambdasA[i])*(v(nodesI[0])*shapesI[0] + v(nodesI[1])*shapesI[1]) + Parent::lambdasA[i]*Parent::atomes_rec.Get(i).vitesse(0);
    orig_FE[i] = v(nodesI[0])*shapesI[0] + v(nodesI[1])*shapesI[1];
    orig_atom[i] = Parent::atomes_rec.Get(i).vitesse(0);
  }

  for (int i=0; i<taille; ++i)
    {
      out_rhs << i+1 << "\t" << current_step  << "\t" << lagrange2[i] << std::endl;
    }
}
  
template <typename DomainA, typename DomainC,unsigned int Dim>
void MyMethodFull<DomainA,DomainC,Dim>::SolveConstraint(){
  int taille = Parent::atomes_rec.nbElem();

  for (int i=0; i<taille; ++i)
    {
      DUMP("rhs[" << i << "] = " << (*rhs)(i),DBG_ALL);
    }  

  A->Solve(rhs);

  for (int i=0; i<taille; ++i)
    {
      DUMP("multL[" << i << "] = " << (*rhs)(i),DBG_ALL);
    }  
}
  

template <typename DomainA, typename DomainC,unsigned int Dim>
void MyMethodFull<DomainA,DomainC,Dim>::ApplyCorrection(){

  double xI;

  typename Parent::_Vec_ & v = Parent::domC.getDofs().V();
  std::vector<unsigned int> nodesI;
  std::vector<double> shapesI;
  MechaContLibMesh<Dim> & model = static_cast<MechaContLibMesh<Dim>&>(Parent::domC.getModel());  
  int taille = Parent::atomes_rec.nbElem();
  
  for (int i = 0 ; i < taille ; ++i)
    {
      xI = Parent::atomes_rec.Get(i).position0(0);
      //correction des atomes
      Parent::atomes_rec.Get(i).vitesse(0) += (1-Parent::lambdasA[i])*lagrange2[i];

      model.GlobalIndexes(Parent::elems_rec.Get(Parent::assoc[i]),nodesI);
      
      model.ComputeShapes(Parent::elems_rec.Get(Parent::assoc[i]),shapesI,xI);
      DUMP(xI << " node1 " << nodesI[0] << " nodes2 " << nodesI[1],DBG_ALL);
      DUMP("pour " << xI << " correction = " << -1.0*(*rhs)(i)*shapesI[0]/model.Mass(nodesI[0]) << " et " << 
           -1.0*(*rhs)(i)*shapesI[1]/model.Mass(nodesI[1]),DBG_ALL);
      v.add(nodesI[0],-1.0*(*rhs)(i)*shapesI[0]/model.Mass(nodesI[0]));
      v.add(nodesI[1],-1.0*(*rhs)(i)*shapesI[1]/model.Mass(nodesI[1]));
//      v.add(nodesI[0],-1.0*(*rhs)(i)*shapesI[0]);
//      v.add(nodesI[1],-1.0*(*rhs)(i)*shapesI[1]);
    }

  v.close();
  for (int i = 0 ; i < taille ; ++i){
    xI = Parent::atomes_rec.Get(i).position0(0);
    model.GlobalIndexes(Parent::elems_rec.Get(Parent::assoc[i]),nodesI);
    model.ComputeShapes(Parent::elems_rec.Get(Parent::assoc[i]),shapesI,xI);
    double new_dist = (1-Parent::lambdasA[i])*(v(nodesI[0])*shapesI[0] + v(nodesI[1])*shapesI[1]) + 
      Parent::lambdasA[i]*Parent::atomes_rec.Get(i).vitesse(0);
    double contrainte = v(nodesI[0])*shapesI[0] + v(nodesI[1])*shapesI[1] - Parent::atomes_rec.Get(i).vitesse(0);
    out_dist << current_step << "\t"  << xI << "\t" << dist[i] << "\t" <<  new_dist << "\t" << orig_FE[i] <<  "\t" << orig_atom[i] << "\t" << Parent::atomes_rec.Get(i).vitesse(0) << "\t" << v(nodesI[0])*shapesI[0] + v(nodesI[1])*shapesI[1] << "\t" << (*rhs)(i) << "\t" << Parent::lambdasA[i]*lagrange2[i] << "\t" << contrainte << std::endl;
  }



}

#ifdef USING_STAMP
template class MyMethodFull<DomainStamp,DomainLibMesh<1>,1>;
template class MyMethodFull<DomainStamp,DomainLibMesh<2>,2>;
#endif
#ifdef USING_LAMMPS
template class MyMethodFull<DomainLammps,DomainLibMesh<2>,2>;
template class MyMethodFull<DomainLammps,DomainLibMesh<3>,3>;
#endif
template class MyMethodFull<DomainMD1D,DomainLibMesh<1>,1>;

---