TRUSTChamp_Morceaux_generique.tpp

/****************************************************************************
* Copyright (c) 2026, 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.
*
*****************************************************************************/
 
#ifndef TRUSTChamp_Morceaux_generique_TPP_included
#define TRUSTChamp_Morceaux_generique_TPP_included
 
#include <Sous_Domaine.h>
#include <Champ_Generique_base.h>
#include <ParserView.h>
 
template<Champ_Morceaux_Type _TYPE_>
Champ_base& TRUSTChamp_Morceaux_generique<_TYPE_>::affecter_(const Champ_base& ch)
{
  static constexpr bool IS_FONC = (_TYPE_ == Champ_Morceaux_Type::FONC);
 
  if (!IS_FONC) not_implemented_champ_<void>(__func__);
 
  if (sub_type(Champ_Uniforme, ch)) valeurs() = ch.valeurs()(0, 0);
  else valeurs() = ch.valeurs();
 
  return *this;
}
 
/*! @brief Renvoie la valeur du champ au point specifie par ses coordonnees.
 *
 * @param (DoubleVect& positions) les coordonnees du point de calcul
 * @param (DoubleVect& valeurs) la valeur du champ au point specifie
 */
template<Champ_Morceaux_Type _TYPE_>
DoubleVect& TRUSTChamp_Morceaux_generique<_TYPE_>::valeur_a(const DoubleVect& positions, DoubleVect& tab_valeurs) const
{
  const Domaine& le_dom = mon_domaine;
  IntVect le_poly(1);
  le_dom.chercher_elements(positions, le_poly);
  return valeur_a_elem(positions, tab_valeurs, le_poly[0]);
}
 
/*! @brief Renvoie la valeur du champ au point specifie par ses coordonnees, en indiquant que ce point est situe dans un element specifie.
 *
 * @param (DoubleVect&) les coordonnees du point de calcul
 * @param (DoubleVect& val) la valeur du champ au point specifie
 * @param (int le_poly) l'element dans lequel est situe le point de calcul
 */
template<Champ_Morceaux_Type _TYPE_>
DoubleVect& TRUSTChamp_Morceaux_generique<_TYPE_>::valeur_a_elem(const DoubleVect&, DoubleVect& val, int le_poly) const
{
  if (val.size() != nb_compo_) erreur_champ_(__func__);
 
  const DoubleTab& ch = valeurs();
  for (int k = 0; k < nb_compo_; k++) val(k) = ch(le_poly, k);
 
  return val;
}
 
/*! @brief Renvoie la valeur d'une composante du champ au point specifie par ses coordonnees, en indiquant que ce point est situe dans un element specifie.
 *
 * @param (DoubleVect&) les coordonnees du point de calcul
 * @param (int le_poly) l'element dans lequel est situe le point de calcul
 * @param (int ncomp) l'index de la composante du champ a calculer
 */
template<Champ_Morceaux_Type _TYPE_>
double TRUSTChamp_Morceaux_generique<_TYPE_>::valeur_a_elem_compo(const DoubleVect&, int le_poly, int ncomp) const
{
  if (ncomp > nb_compo_) erreur_champ_(__func__);
 
  const DoubleTab& ch = valeurs();
  double val = (nb_compo_ == 1) ? ch(le_poly, 0) : ch(le_poly, ncomp);
 
  return val;
}
 
/*! @brief Renvoie les valeurs du champ aux points specifies par leurs coordonnees.
 *
 * @param (DoubleTab& positions) le tableau des coordonnees des points de calcul
 * @param (DoubleTab& valeurs) le tableau des valeurs du champ aux points specifies
 */
template<Champ_Morceaux_Type _TYPE_>
DoubleTab& TRUSTChamp_Morceaux_generique<_TYPE_>::valeur_aux(const DoubleTab& positions, DoubleTab& tab_valeurs) const
{
  const Domaine& le_dom = mon_domaine;
  IntTrav les_polys;
  le_dom.chercher_elements(positions, les_polys);
  return valeur_aux_elems(positions, les_polys, tab_valeurs);
}
 
/*! @brief Renvoie les valeurs d'une composante du champ aux points specifies par leurs coordonnees.
 *
 * @param (DoubleTab& positions) le tableau des coordonnees des points de calcul
 * @param (DoubleVect& valeurs) le tableau des valeurs de la composante du champ aux points specifies
 * @param (int ncomp) l'index de la composante du champ a calculer
 */
template<Champ_Morceaux_Type _TYPE_>
DoubleVect& TRUSTChamp_Morceaux_generique<_TYPE_>::valeur_aux_compo(const DoubleTab& positions, DoubleVect& tab_valeurs, int ncomp) const
{
  const Domaine& le_dom = mon_domaine;
  IntVect les_polys(le_dom.nb_elem());
  le_dom.chercher_elements(positions, les_polys);
  return valeur_aux_elems_compo(positions, les_polys, tab_valeurs, ncomp);
}
 
/*! @brief Renvoie les valeurs du champ aux points specifies par leurs coordonnees, en indiquant que les points de calculs sont situes dans les elements indiques.
 *
 * @param (DoubleTab&) le tableau des coordonnees des points de calcul
 * @param (IntVect& les_polys) le tableau des elements dans lesquels sont situes les points de calcul
 * @param (DoubleTab& val) le tableau des valeurs du champ aux points specifies
 */
template<Champ_Morceaux_Type _TYPE_>
DoubleTab& TRUSTChamp_Morceaux_generique<_TYPE_>::valeur_aux_elems(const DoubleTab&, const IntVect& polys, DoubleTab& tab_val) const
{
  if (tab_val.nb_dim() == 2)
    {
      assert((tab_val.dimension(0) == polys.size()) || (tab_val.dimension_tot(0) == polys.size()));
      assert(tab_val.dimension(1) == nb_compo_);
    }
  else erreur_champ_(__func__); // (val.nb_dim() > 2)
 
  tab_val = 0.;
  CDoubleTabView ch = valeurs().view_ro();
  CIntArrView les_polys = polys.view_ro();
  DoubleTabView val = tab_val.view_rw();
  int nb_compo = nb_compo_;
  Kokkos::parallel_for(start_gpu_timer(__KERNEL_NAME__), les_polys.size(), KOKKOS_LAMBDA(const int rang_poly)
  {
    int p = les_polys(rang_poly);
    if (p != -1)
      for (int n = 0; n < nb_compo; n++)
        val(rang_poly, n) = ch(p, n);
  });
  end_gpu_timer(__KERNEL_NAME__);
 
  return tab_val;
}
 
/*! @brief Renvoie les valeurs d'une composante du champ aux points specifies par leurs coordonnees, en indiquant que les points de calculs sont situes dans les elements indiques.
 *
 * @param (DoubleTab&) le tableau des coordonnees des points de calcul
 * @param (IntVect& les_polys) le tableau des elements dans lesquels sont situes les points de calcul
 * @param (DoubleVect& val) le tableau des valeurs de la composante du champ aux points specifies
 * @param (int ncomp) l'index de la composante du champ a calculer
 */
template<Champ_Morceaux_Type _TYPE_>
DoubleVect& TRUSTChamp_Morceaux_generique<_TYPE_>::valeur_aux_elems_compo(const DoubleTab&, const IntVect& les_polys, DoubleVect& val, int ncomp) const
{
  assert(val.size_totale() >= les_polys.size());
  int le_poly;
  const DoubleTab& ch = valeurs();
  ToDo_Kokkos("critical");
  for (int rang_poly = 0; rang_poly < les_polys.size(); rang_poly++)
    {
      le_poly = les_polys(rang_poly);
      val(rang_poly) = (le_poly == -1) ? 0 : ch(le_poly, ncomp);
    }
 
  return val;
}
 
template<Champ_Morceaux_Type _TYPE_>
void TRUSTChamp_Morceaux_generique<_TYPE_>::mettre_a_jour(double time)
{
  static constexpr bool IS_UNIFORME = (_TYPE_ == Champ_Morceaux_Type::UNIFORME);
 
  if (IS_UNIFORME)
    {
      Champ_Don_base::mettre_a_jour(time);
      return;
    }
  else
    {
      OWN_PTR(Champ_base) espace_stockage;
      const Champ_base *ch = !ref_pb ? nullptr : ref_pb->has_champ(nom_champ_parametre_) ? &ref_pb->get_champ(nom_champ_parametre_) : &ref_pb->get_champ_post(nom_champ_parametre_).get_champ(espace_stockage);
      const int nb_som_elem = mon_domaine->nb_som_elem();
      int dim = dimension;
      bool pb = bool(ref_pb);
      const int max_parser_size = 200;
      if (parser.size()>max_parser_size)
        {
          Cerr << "Increase max_parser_size to " << parser.size() << " TRUSTChamp_Morceaux_generique<_TYPE_>::mettre_a_jour !" << finl;
          Process::exit();
        }
      Kokkos::Array<ParserView, max_parser_size> psr;
      for (int i=0; i<parser.size(); i++)
        {
          psr[i].set(parser[i]);
          psr[i].parseString();
        }
      bool has_champ = ch != nullptr;
      CDoubleTabView tab_ch;
      if (has_champ) tab_ch = ch->valeurs().view_ro();
      CDoubleTabView coord = mon_domaine->les_sommets().view_ro();
      CIntTabView les_elems = mon_domaine->les_elems().view_ro();
      CIntTabView parser_idx = parser_idx_.view_ro();
      DoubleTabView tab = valeurs().view_wo();
      const int nb_comp = (int)tab.extent(1);
      Kokkos::parallel_for(start_gpu_timer(__KERNEL_NAME__), mon_domaine->nb_elem_tot(), KOKKOS_LAMBDA(const int i)
      {
        /* xs : coordonnees du poly par barycentre des sommets -> pas top */
        double xs[3] = {0,0,0};
        int nb_som = 0, s, r;
        for (int j = 0; j < nb_som_elem && (s = les_elems(i, j)) >= 0; j++)
          for (r = 0, nb_som++; r < dim; r++)
            xs[r] += coord(s, r);
        for (r = 0; r < dim; r++)
          xs[r] /= nb_som;
 
        /* calcul de chaque composante */
        double val = has_champ ? tab_ch(i, 0) : 0;
        for (int k = 0; k < nb_comp; k++)
          {
            int idx = parser_idx(i, k);
            int threadId = psr[idx].acquire();
            psr[idx].setVar(0, xs[0], threadId); // x
            psr[idx].setVar(1, xs[1], threadId); // y
            psr[idx].setVar(2, xs[2], threadId); // z
            psr[idx].setVar(3, time, threadId);  // t
            if (pb) psr[idx].setVar(4, val, threadId); // val
            tab(i, k) = psr[idx].eval(threadId);
            psr[idx].release(threadId);
          }
      });
      end_gpu_timer(__KERNEL_NAME__);
    }
}
 
template<Champ_Morceaux_Type _TYPE_>
int TRUSTChamp_Morceaux_generique<_TYPE_>::initialiser(const double time)
{
  return Champ_Don_base::initialiser(time);
}
 
template<Champ_Morceaux_Type _TYPE_>
void TRUSTChamp_Morceaux_generique<_TYPE_>::interprete_get_domaine(const Nom& nom)
{
  static constexpr bool IS_FONC_TXYZ = (_TYPE_ == Champ_Morceaux_Type::FONC_TXYZ);
  if (IS_FONC_TXYZ || read_pb_instead_of_domain)
    {
      ref_pb = ref_cast(Probleme_base, Interprete::objet(nom));
      mon_domaine = ref_pb->domaine();
    }
  else mon_domaine = ref_cast(Domaine, Interprete::objet(nom));
}
 
template<Champ_Morceaux_Type _TYPE_>
void TRUSTChamp_Morceaux_generique<_TYPE_>::creer_tabs(const int dim)
{
  fixer_nb_comp(dim);
  parser_idx_.resize(0, dim);
  mon_domaine->creer_tableau_elements(parser_idx_);
  valeurs_.resize(0, dim);
  mon_domaine->creer_tableau_elements(valeurs_);
}
 
template<Champ_Morceaux_Type _TYPE_>
Entree& TRUSTChamp_Morceaux_generique<_TYPE_>::complete_readOn(const int dim, const Nom& qsj, Entree& is, Nom& nom)
{
  Motcle motlu;
  int k, poly;
 
  if (nom != "{")
    {
      Cerr << "Error while reading a " << qsj << " . We expected a { instead of " << nom << finl;
      Process::exit();
    }
  is >> motlu;
  if (motlu != Motcle("defaut"))
    {
      Cerr << "Error while reading a " << qsj << " . We expected defaut instead of " << nom << finl;
      Process::exit();
    }
 
  /* parsers par defaut */
  for (k = 0; k < dim; k++)
    {
      Parser_U psr;
      Nom tmp;
      is >> tmp;
      psr.setNbVar(5);
      psr.setString(tmp);
      psr.addVar("x"), psr.addVar("y"), psr.addVar("z"), psr.addVar("t");
      if (ref_pb) psr.addVar("val");
      psr.parseString();
 
      for (poly = 0; poly < mon_domaine->nb_elem_tot(); poly++)
        parser_idx_(poly, k) = parser.size();
 
      parser.add(psr);
    }
 
  is >> nom;
  while (nom != Nom("}"))
    {
      OBS_PTR(Sous_Domaine) refssz = les_sous_domaines.add(mon_domaine->ss_domaine(nom));
      Sous_Domaine& ssz = refssz.valeur();
      for (k = 0; k < dim; k++)
        {
          Parser_U psr;
          Nom tmp;
          is >> tmp;
          psr.setNbVar(5);
          psr.setString(tmp);
          psr.addVar("x"), psr.addVar("y"), psr.addVar("z"), psr.addVar("t");
          if (ref_pb) psr.addVar("val");
          psr.parseString();
 
          for (poly = 0; poly < ssz.nb_elem_tot(); poly++)
            parser_idx_(ssz(poly), k) = parser.size();
 
          parser.add(psr);
        }
      is >> nom;
    }
  return is;
}
 
#endif /* TRUSTChamp_Morceaux_generique_TPP_included */