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 Returns the value of the field at the point specified by its coordinates.
 *
 * @param (DoubleVect& positions) the coordinates of the computation point
 * @param (DoubleVect& valeurs) the value of the field at the specified point
 */
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 Returns the value of the field at the point specified by its coordinates, indicating that this point is located in a specified element.
 *
 * @param (DoubleVect&) the coordinates of the computation point
 * @param (DoubleVect& val) the value of the field at the specified point
 * @param (int le_poly) the element in which the computation point is located
 */
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 Returns the value of one component of the field at the point specified by its coordinates, indicating that this point is located in a specified element.
 *
 * @param (DoubleVect&) the coordinates of the computation point
 * @param (int le_poly) the element in which the computation point is located
 * @param (int ncomp) the index of the field component to compute
 */
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 Returns the values of the field at the points specified by their coordinates.
 *
 * @param (DoubleTab& positions) the array of coordinates of the computation points
 * @param (DoubleTab& valeurs) the array of field values at the specified points
 */
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 Returns the values of one component of the field at the points specified by their coordinates.
 *
 * @param (DoubleTab& positions) the array of coordinates of the computation points
 * @param (DoubleVect& valeurs) the array of values of the field component at the specified points
 * @param (int ncomp) the index of the field component to compute
 */
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 Returns the values of the field at the points specified by their coordinates, indicating that the computation points are located in the specified elements.
 *
 * @param (DoubleTab&) the array of coordinates of the computation points
 * @param (IntVect& les_polys) the array of elements in which the computation points are located
 * @param (DoubleTab& val) the array of field values at the specified points
 */
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 Returns the values of one component of the field at the points specified by their coordinates, indicating that the computation points are located in the specified elements.
 *
 * @param (DoubleTab&) the array of coordinates of the computation points
 * @param (IntVect& les_polys) the array of elements in which the computation points are located
 * @param (DoubleVect& val) the array of values of the field component at the specified points
 * @param (int ncomp) the index of the field component to compute
 */
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 : coordinates of the poly via barycenter of vertices -> not ideal */
        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;
 
        /* compute each component */
        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();
    }
 
  /* default parsers */
  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 */