TRUSTChamp_Don_generique.tpp

/****************************************************************************
* Copyright (c) 2024, 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_Don_generique_TPP_included
#define TRUSTChamp_Don_generique_TPP_included
 
template <Champ_Don_Type _TYPE_>
Sortie& TRUSTChamp_Don_generique<_TYPE_>::printOn(Sortie& os) const
{
  const DoubleTab& tab = valeurs();
  os << tab.size() << " ";
  for (int i = 0; i < tab.size(); i++) os << tab(0, i);
  return os;
}
 
template <Champ_Don_Type _TYPE_>
void TRUSTChamp_Don_generique<_TYPE_>::mettre_a_jour(double t)
{
  static constexpr bool IS_TXYZ = (_TYPE_ == Champ_Don_Type::TXYZ);
 
  if (!IS_TXYZ) /* XYZ ou LU */
    Champ_Don_base::mettre_a_jour(t);
  else
    {
      changer_temps(t);
      int nb_elems = mon_domaine->nb_elem();
      DoubleTab& mes_val = valeurs();
      DoubleTab positions(nb_elems, dimension);
      mettre_a_jour_positions(positions);
      eval_fct(positions, temps(), mes_val);
      mes_val.echange_espace_virtuel();
    }
}
 
template <Champ_Don_Type _TYPE_>
double TRUSTChamp_Don_generique<_TYPE_>::valeur_a_compo(const DoubleVect& x, int ncomp) const
{
  static constexpr bool IS_XYZ = (_TYPE_ == Champ_Don_Type::XYZ);
  if (!IS_XYZ)
    return Champ_Don_base::valeur_a_compo(x, ncomp); // simple call for TXYZ or LU !
  else
    {
      if (ncomp > nb_compo_) erreur_champ_(__func__); // asking ncomp > nb_compo_ ?????
 
      DoubleTab positions(1, dimension), val_fct(1);
      for (int d = 0; d < dimension; d++) positions(0, d) = x(d);
      eval_fct(positions, val_fct, ncomp);
      return val_fct(0);
    }
}
 
template <Champ_Don_Type _TYPE_>
void TRUSTChamp_Don_generique<_TYPE_>::mettre_a_jour_positions(DoubleTab& positions)
{
  positions = 0.0;
  const int nb_elems = mon_domaine->nb_elem(), D = dimension;
  const IntTab& les_Polys = mon_domaine->les_elems();
  for (int num_elem = 0; num_elem < nb_elems; num_elem++)
    {
      int nb_som = 0;
      for (int s = 0, num_som; s < les_Polys.dimension(1); s++)
        if ((num_som = les_Polys(num_elem, s)) >= 0)
          {
            for (int d = 0; d < D; d++)
              positions(num_elem, d) += mon_domaine->coord(num_som, d);
            nb_som++;
          }
      for (int d = 0; d < D; d++)
        positions(num_elem, d) /= nb_som;
    }
}
 
/*! @brief Returns the value of the field at the point specified by its coordinates.
 *
 * @param (DoubleVect& x) the coordinates of the computation point
 * @param (DoubleVect& val) the value of the field at the specified point
 */
template <Champ_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T != Champ_Don_Type::LU, DoubleVect&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_a_(const DoubleVect& x, DoubleVect& val) const
{
  static constexpr bool IS_XYZ = (_TYPE_ == Champ_Don_Type::XYZ);
  if (val.size() != nb_compo_) erreur_champ_(__func__); // DoubleVect val doesn't have the correct size
 
  IS_XYZ ? eval_fct(x,val) : eval_fct(x,temps(),val);
  return val;
}
 
template <Champ_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T == Champ_Don_Type::LU, DoubleVect&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_a_(const DoubleVect& x, DoubleVect& val) const
{
  IntVect le_poly(1);
  mon_domaine->chercher_elements(x,le_poly);
  return valeur_a_elem(x,val,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_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T != Champ_Don_Type::LU, DoubleVect&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_a_elem_(const DoubleVect& x, DoubleVect& val, int ) const
{
  return valeur_a(x,val);
}
 
template <Champ_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T == Champ_Don_Type::LU, DoubleVect&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_a_elem_(const DoubleVect& , DoubleVect& val, int le_poly) const
{
  if (val.size() != nb_compo_) erreur_champ_(__func__); // DoubleVect val doesn't have the correct size
 
  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_Don_Type _TYPE_> template <Champ_Don_Type T>
std::enable_if_t<T != Champ_Don_Type::TXYZ, double> /* XYZ ou LU */
TRUSTChamp_Don_generique<_TYPE_>::valeur_a_elem_compo_(const DoubleVect& x, int le_poly, int ncomp) const
{
  static constexpr bool IS_XYZ = (_TYPE_ == Champ_Don_Type::XYZ);
  if (ncomp > nb_compo_) erreur_champ_(__func__); // asking ncomp > nb_compo_ ?????
 
  return IS_XYZ ? valeur_a_compo(x,ncomp) : valeurs()(le_poly,ncomp) /* LU */;
}
 
template <Champ_Don_Type _TYPE_> template <Champ_Don_Type T>
std::enable_if_t<T == Champ_Don_Type::TXYZ, double>
TRUSTChamp_Don_generique<_TYPE_>::valeur_a_elem_compo_(const DoubleVect& x, int , int ncomp) const
{
  DoubleTab positions(1, dimension);
  for (int d = 0; d < dimension; d++) positions(0, d) = x(d);
 
  DoubleVect val_fct(1);
  eval_fct(positions, temps(), val_fct, ncomp);
 
  return val_fct(0);
}
 
/*! @brief Returns the values of the field at the points specified by their coordinates.
 *
 * @param (DoubleTab& x) the array of coordinates of the computation points
 * @param (DoubleTab& val) the array of field values at the specified points
 */
template <Champ_Don_Type _TYPE_> template <Champ_Don_Type T>
std::enable_if_t<T != Champ_Don_Type::LU, DoubleTab&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_aux_(const DoubleTab& x, DoubleTab& val) const
{
  static constexpr bool IS_XYZ = (_TYPE_ == Champ_Don_Type::XYZ);
 
  if (val.nb_dim() == 1) assert(nb_compo_ == 1);
  else if (val.nb_dim() == 2) assert(val.dimension(1) == nb_compo_);
  else erreur_champ_(__func__); // DoubleTab val has more than 2 entries !
 
  IS_XYZ ? eval_fct(x,val) : eval_fct(x,temps(),val);
  return val;
}
 
template <Champ_Don_Type _TYPE_> template <Champ_Don_Type T>
std::enable_if_t<T == Champ_Don_Type::LU, DoubleTab&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_aux_(const DoubleTab& x, DoubleTab& val) const
{
  IntVect les_polys(mon_domaine->nb_elem());
  mon_domaine->chercher_elements(x,les_polys);
  return valeur_aux_elems(x,les_polys,val); // VTABLE for Champ_som_lu
}
 
/*! @brief Returns the values of one component of the field at the points specified by their coordinates.
 *
 * @param (DoubleVect& x) 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_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T != Champ_Don_Type::LU, DoubleVect&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_aux_compo_(const DoubleTab& x, DoubleVect& val, int ncomp) const
{
  static constexpr bool IS_XYZ = (_TYPE_ == Champ_Don_Type::XYZ);
  IS_XYZ ? eval_fct(x,val,ncomp) : eval_fct(x,temps(),val,ncomp);
  return val;
}
 
template <Champ_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T == Champ_Don_Type::LU, DoubleVect&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_aux_compo_(const DoubleTab& x, DoubleVect& val, int ncomp) const
{
  IntVect les_polys(mon_domaine->nb_elem());
  mon_domaine->chercher_elements(x,les_polys);
  return valeur_aux_elems_compo(x,les_polys,val,ncomp); // VTABLE for Champ_som_lu
}
 
/*! @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 (unused)
 * @param (DoubleTab& val) the array of field values at the specified points
 */
template <Champ_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T != Champ_Don_Type::LU, DoubleTab&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_aux_elems_(const DoubleTab& x, const IntVect&, DoubleTab& val) const
{
  return valeur_aux(x, val);
}
 
template <Champ_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T == Champ_Don_Type::LU, DoubleTab&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_aux_elems_(const DoubleTab&, const IntVect& les_polys, DoubleTab& val) const
{
  if (val.nb_dim() > 2) erreur_champ_(__func__); // DoubleTab val don't have 2 entries
 
  int p;
  const DoubleTab& ch = valeurs();
  val = 0.;
 
  for (int rang_poly = 0; rang_poly < les_polys.size(); rang_poly++)
    if ((p = les_polys(rang_poly)) != -1)
      for (int n = 0; n < nb_compo_; n++) val(rang_poly, n) = ch(p, n);
 
  return 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& x) 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_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T != Champ_Don_Type::LU, DoubleVect&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_aux_elems_compo_(const DoubleTab& x, const IntVect&, DoubleVect& val, int ncomp) const
{
  return valeur_aux_compo(x, val, ncomp);
}
 
template <Champ_Don_Type _TYPE_> template<Champ_Don_Type T>
std::enable_if_t<T == Champ_Don_Type::LU, DoubleVect&>
TRUSTChamp_Don_generique<_TYPE_>::valeur_aux_elems_compo_(const DoubleTab& , const IntVect& les_polys, DoubleVect& val, int ncomp) const
{
  assert(val.size_totale() >= les_polys.size());
  const DoubleTab& ch = valeurs();
 
  for (int rang_poly = 0; rang_poly < les_polys.size(); rang_poly++)
    {
      int le_poly = les_polys(rang_poly);
      if (le_poly == -1) val(rang_poly) = 0;
      else val(rang_poly) = ch(le_poly, ncomp);
    }
  return val;
}
 
#endif /* TRUSTChamp_Don_generique_TPP_included */