Masse_VEF_P1NC.cpp

/****************************************************************************
* Copyright (c) 2025, CEA
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
* 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
* 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
*****************************************************************************/
 
#include <Masse_VEF_P1NC.h>
#include <Domaine_Cl_VEF.h>
#include <Domaine_VEF.h>
#include <Dirichlet.h>
#include <Dirichlet_homogene.h>
#include <Symetrie.h>
#include <Equation_base.h>
#include <Matrice_Morse.h>
#include <Milieu_base.h>
#include <Device.h>
 
Implemente_instanciable(Masse_VEF_P1NC,"Masse_VEF_P1NC",Solveur_Masse_base);
 
 
//     printOn()
/////
 
Sortie& Masse_VEF_P1NC::printOn(Sortie& s) const
{
  return s << que_suis_je() << " " << le_nom();
}
 
//// readOn
//
 
Entree& Masse_VEF_P1NC::readOn(Entree& s)
{
  return s ;
}
 
 
 
///////////////////////////////////////////////////////////////
//
//  Implementation of the Masse_VEF_P1NC class member functions
//
//////////////////////////////////////////////////////////////
 
 
DoubleTab& Masse_VEF_P1NC::appliquer_impl(DoubleTab& tab_sm) const
{
  const Domaine_Cl_VEF& domaine_Cl_VEF = ref_cast(Domaine_Cl_VEF,le_dom_Cl_VEF.valeur());
  const Domaine_VEF& domaine_VEF = le_dom_VEF.valeur();
  const DoubleVect& tab_volumes_entrelaces = domaine_VEF.volumes_entrelaces();
 
  int nfa = domaine_VEF.nb_faces();
  int num_std = domaine_VEF.premiere_face_std();
  int num_int = domaine_VEF.premiere_face_int();
  int nbcomp = tab_sm.line_size();
 
  if (nfa != tab_sm.dimension(0))
    {
      Cerr << "Error in Masse_VEF_P1NC: ";
      Cerr << "number of faces: " << nfa
           << " size of right-hand side: " << tab_sm.dimension(0) << finl;
      exit();
    }
 
  // Process standard faces that carry no boundary conditions
  CDoubleArrView porosite_face = equation().milieu().porosite_face().view_ro();
  CDoubleArrView volumes_entrelaces = tab_volumes_entrelaces.view_ro();
  DoubleTabView sm = tab_sm.view_rw();
 
  Kokkos::parallel_for(start_gpu_timer(__KERNEL_NAME__),
                       Kokkos::MDRangePolicy<Kokkos::Rank<2>>({num_std,0}, {nfa,nbcomp}),
                       KOKKOS_LAMBDA(int face, int comp)
  {
    sm(face, comp) /= (volumes_entrelaces(face) * porosite_face(face));
  });
  end_gpu_timer(__KERNEL_NAME__);
 
  // Process non-standard faces
  // boundary faces are non-standard faces that may carry boundary conditions
  // internal non-standard faces carry no boundary conditions
 
  // Process boundary conditions
  CIntTabView face_voisins = domaine_VEF.face_voisins().view_ro();
  CDoubleTabView normales = domaine_VEF.face_normales().view_ro();
  CDoubleArrView volumes_entrelaces_Cl = domaine_Cl_VEF.volumes_entrelaces_Cl().view_ro();
  for (int n_bord = 0; n_bord < domaine_VEF.nb_front_Cl(); n_bord++)
    {
      const Cond_lim& la_cl = domaine_Cl_VEF.les_conditions_limites(n_bord);
      const Front_VF& le_bord = ref_cast(Front_VF,la_cl->frontiere_dis());
      int num1 = le_bord.num_premiere_face();
      int num2 = num1 + le_bord.nb_faces();
 
      if ((sub_type(Dirichlet,la_cl.valeur())) ||
          (sub_type(Dirichlet_homogene,la_cl.valeur())))
        {
          // For Dirichlet faces set sm to 0
          Kokkos::parallel_for(start_gpu_timer(__KERNEL_NAME__),
                               range_1D(num1, num2),
                               KOKKOS_LAMBDA(const int face)
          {
            for (int comp = 0; comp < nbcomp; comp++)
              sm(face, comp) = 0;
          });
          end_gpu_timer(__KERNEL_NAME__);
        }
      else if ((sub_type(Symetrie,la_cl.valeur())) && (domaine_Cl_VEF.equation().inconnue().nature_du_champ()==vectoriel))
        {
          Kokkos::parallel_for(start_gpu_timer(__KERNEL_NAME__),
                               range_1D(num1, num2),
                               KOKKOS_LAMBDA(const int face)
          {
            double psc = 0;
            double surf = 0;
            for (int comp = 0; comp < nbcomp; comp++)
              {
                psc += sm(face,comp) * normales(face,comp);
                surf += normales(face,comp) * normales(face,comp);
              }
            psc /= surf;
            for(int comp = 0; comp < nbcomp; comp++)
              {
                sm(face,comp) -= psc * normales(face,comp);
                sm(face,comp) /= (volumes_entrelaces_Cl(face) *
                                  porosite_face(face));
              }
          });
          end_gpu_timer(__KERNEL_NAME__);
        }
      else
        {
          Kokkos::parallel_for(start_gpu_timer(__KERNEL_NAME__),
                               range_1D(num1, num2),
                               KOKKOS_LAMBDA(
                                 const int face)
          {
            int elem = face_voisins(face, 0);
            if (elem == -1) elem = face_voisins(face, 1);
            for (int comp = 0; comp < nbcomp; comp++)
              sm(face, comp) /= (volumes_entrelaces_Cl(face) * porosite_face(face));
          });
          end_gpu_timer(__KERNEL_NAME__);
        }
    }
 
  Kokkos::parallel_for(start_gpu_timer(__KERNEL_NAME__),
                       range_2D({num_int,0}, {num_std,nbcomp}),
                       KOKKOS_LAMBDA(int face, int comp)
  {
    sm(face,comp) /= (volumes_entrelaces_Cl(face) * porosite_face(face));
  });
  end_gpu_timer(__KERNEL_NAME__);
 
  //tab_sm.echange_espace_virtuel();
  //Debog::verifier("Masse_VEF_P1NC::appliquer, tab_sm=",tab_sm);
  return tab_sm;
}
 
 
//
void Masse_VEF_P1NC::associer_domaine_dis_base(const Domaine_dis_base& le_dom_dis_base)
{
  le_dom_VEF = ref_cast(Domaine_VEF, le_dom_dis_base);
}
 
void Masse_VEF_P1NC::associer_domaine_cl_dis_base(const Domaine_Cl_dis_base& le_dom_Cl_dis_base)
{
  le_dom_Cl_VEF = ref_cast(Domaine_Cl_VEF, le_dom_Cl_dis_base);
}
 
 
Matrice_Base& Masse_VEF_P1NC::ajouter_masse(double dt, Matrice_Base& matrice, int penalisation) const
{
  if (penalisation||(le_dom_Cl_VEF->equation().inconnue().nature_du_champ()!=vectoriel))
    return Solveur_Masse_base::ajouter_masse(dt,matrice,penalisation);
  // Otherwise temporarily change the field nature so that appliquer_impl does not project onto n.
  Champ_Inc_base& inco=ref_cast_non_const( Champ_Inc_base,le_dom_Cl_VEF->equation().inconnue());
  inco.fixer_nature_du_champ(multi_scalaire);
  Solveur_Masse_base::ajouter_masse(dt,matrice,penalisation);
  inco.fixer_nature_du_champ(vectoriel);
  return matrice;
 
 
}
 
 
DoubleTab& Masse_VEF_P1NC::ajouter_masse(double dt, DoubleTab& x, const DoubleTab& y, int penalisation, bool use_old_volumes) const
{
  if (penalisation||(le_dom_Cl_VEF->equation().inconnue().nature_du_champ()!=vectoriel))
    return Solveur_Masse_base::ajouter_masse(dt, x, y, penalisation, use_old_volumes);
 
  // Otherwise temporarily change the field nature so that appliquer_impl does not project onto n.
  Champ_Inc_base& inco=ref_cast_non_const( Champ_Inc_base,le_dom_Cl_VEF->equation().inconnue());
  inco.fixer_nature_du_champ(multi_scalaire);
  Solveur_Masse_base::ajouter_masse(dt, x, y, penalisation, use_old_volumes);
  inco.fixer_nature_du_champ(vectoriel);
  return x;
}
 
DoubleTab& Masse_VEF_P1NC::corriger_solution(DoubleTab& x, const DoubleTab& y, int incr) const
{
  //  assert(penalisation_==1);
 
  return Solveur_Masse_base::corriger_solution( x,  y, incr) ;
}