Format_Post_Lml.cpp

/****************************************************************************
* 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.
*
*****************************************************************************/
#include <Format_Post_Lml.h>
#include <communications.h>
#include <Param.h>
 
Implemente_instanciable_sans_constructeur(Format_Post_Lml,"Format_Post_Lml",Format_Post_base);
 
 
/////////////////////////////////////////////////////////////////////////////
//Parameter: lml_basename_
// Meaning: beginning of the file name
//Extension lml
 
//Specifics of this format
//Writing to a single file (sequential or parallel)
///////////////////////////////////////////////////////////////////////////
 
/*! @brief Default constructor: specify in comments what is set by default
 *
 */
 
Format_Post_Lml::Format_Post_Lml()
{
  reset();
}
 
/*! @brief Resets the object to the state obtained by the default constructor.
 *
 */
void Format_Post_Lml::reset()
{
  lml_basename_ = "??";
}
/*! @brief error => exit
 *
 */
Sortie& Format_Post_Lml::printOn(Sortie& os) const
{
  Cerr << "Format_Post_Lml::printOn : error" << finl;
  exit();
  return os;
}
 
/*! @brief Reads post-processing parameters in "dataset" format. The expected format is:
 *
 *   {
 *     nom_fichier basename_sans_extension
 *   }
 *
 */
Entree& Format_Post_Lml::readOn(Entree& is)
{
  Format_Post_base::readOn(is);
  return is;
}
 
void Format_Post_Lml::set_param(Param& param) const
{
  param.ajouter("nom_fichier",&lml_basename_,Param::REQUIRED);
}
 
/*! @brief Initializes the class with default parameters.
 *
 */
int Format_Post_Lml::initialize_by_default(const Nom& file_basename)
{
  lml_basename_= file_basename;
  return 1;
}
 
int Format_Post_Lml::initialize(const Nom& file_basename, const int format, const Nom& option_para)
{
 
  lml_basename_= file_basename;
  return 1;
}
 
int Format_Post_Lml::ecrire_entete(const double temps_courant,const int reprise,const int est_le_premier_post)
{
 
  Nom nom_fich(lml_basename_);
  nom_fich +=".";
  Nom format="lml";
  nom_fich += format;
 
  ecrire_entete_lml(nom_fich,est_le_premier_post);
  return 1;
 
}
 
int Format_Post_Lml::finir(const int est_le_dernier_post)
{
 
  Nom nom_fich(lml_basename_);
  nom_fich +=".";
  Nom format="lml";
  nom_fich += format;
 
  finir_lml(nom_fich,est_le_dernier_post);
  return 1;
}
 
 
int Format_Post_Lml::completer_post(const Domaine& dom,const int is_axi,
                                    const Nature_du_champ& nature,const int nb_compo,const Noms& noms_compo,
                                    const Motcle& loc_post,const Nom& le_nom_champ_post)
{
 
  completer_post_lml();
  return 1;
 
}
 
int Format_Post_Lml::preparer_post(const Nom& id_du_domaine,const int est_le_premier_post,
                                   const int reprise,
                                   const double t_init)
{
 
  preparer_post_lml();
  return 1;
 
}
 
/*! @brief see Format_Post_base::ecrire_domaine
 *
 */
int Format_Post_Lml::ecrire_domaine(const Domaine& domaine,const int est_le_premier_post)
{
  //Call to the static method specific to lml format
  Nom nom_fich(lml_basename_);
  nom_fich +=".";
  Nom format="lml";
  nom_fich += format;
  ecrire_domaine_lml(domaine,nom_fich);
  return 1; // ok all is well
}
 
/*! @brief Starts writing a new time step. For the LML format specifically:
 *
 *   Opens the master file in APPEND mode and adds a line
 *    "TEMPS xxxxx"
 *
 */
int Format_Post_Lml::ecrire_temps(const double temps)
{
 
  Nom nom_fich(lml_basename_);
  nom_fich +=".";
  Nom format="lml";
  nom_fich += format;
 
  ecrire_temps_lml(temps,nom_fich);
 
  return 1;
}
 
/*! @brief see Format_Post_base::ecrire_champ
 *
 */
int Format_Post_Lml::ecrire_champ(const Domaine& domaine,const Noms& unite_, const Noms& noms_compo,
                                  int ncomp,double temps_,
                                  const Nom& id_champ,
                                  const Nom& id_du_domaine,
                                  const Nom& localisation,
                                  const Nom&   nature,
                                  const DoubleTab& valeurs)
 
{
 
  //Call to the static method specific to lml format
  Nom nom_fich(lml_basename_);
  nom_fich +=".";
  Nom format="lml";
  nom_fich += format;
 
  ecrire_champ_lml(domaine,unite_,noms_compo,ncomp,temps_,id_champ,id_du_domaine,localisation,valeurs,nom_fich);
 
  return 1;
}
 
/*! @brief
 *
 */
int Format_Post_Lml::ecrire_item_int(const Nom&     id_item,
                                     const Nom&     id_du_domaine,
                                     const Nom&     id_domaine,
                                     const Nom&     localisation,
                                     const Nom&     reference,
                                     const IntVect& valeurs,
                                     const int   reference_size)
{
 
  Nom nom_fich(lml_basename_);
  nom_fich +=".";
  Nom format="lml";
  nom_fich += format;
  Nom id_domaine_dom(id_domaine);
  id_domaine_dom+="_";
  id_domaine_dom+=id_du_domaine;
  ecrire_item_int_lml(id_item,id_domaine,valeurs,nom_fich);
 
  return 1;
}
 
int Format_Post_Lml::ecrire_entete_lml(Nom& nom_fich,const int est_le_premier_post)
{
  if (est_le_premier_post)
    {
 
      EcrFicPartage s;
      s.ouvrir(nom_fich);
 
      if (Process::je_suis_maitre())
        {
          s << "TRUST" << " " << "Version " << TRUST_VERSION << finl;
          s << nom_du_cas() << finl;
          s << "TRUST" << finl;
        }
 
      s.flush();
      s.syncfile();
 
    }
  return 1;
}
 
int Format_Post_Lml::finir_lml(Nom& nom_fich, const int est_le_dernier_post)
{
 
  EcrFicPartage s;
  s.ouvrir(nom_fich,ios::app);
 
  if (est_le_dernier_post)
    {
      if (Process::je_suis_maitre())
        {
          s << "FIN" << finl;
        }
    }
 
  s.flush();
  s.syncfile();
 
  return 1;
}
 
int Format_Post_Lml::completer_post_lml()
{
  return 1;
}
 
int Format_Post_Lml::preparer_post_lml()
{
  return 1;
}
 
int Format_Post_Lml::ecrire_domaine_lml(const Domaine& domaine,Nom& nom_fich)
{
  //Currently an identical copy of the Domaine::postraiter_lml() method
 
  EcrFicPartage s;
  s.ouvrir(nom_fich,ios::app);
  Nom nom_grille("Grille_");
  nom_grille+= domaine.le_nom();
 
  if(je_suis_maitre())
    s << "GRILLE" << finl;
 
  trustIdType nb_som_tot0 = Process::mp_sum(domaine.nb_som());
  trustIdType nb_elem_tot0 = static_cast<int>(Process::mp_sum(domaine.nb_elem()));
  if (nb_som_tot0 >= std::numeric_limits<int>::max() || nb_elem_tot0 >= std::numeric_limits<int>::max())
    {
      Cerr << "Too many items in the domain to be written in LML (exceeds the 32b limit) - the support for this is not implemented." << finl;
      Cerr << "Please contact TRUST support!" << finl;
      Process::exit(-1);
    }
  int nb_som_tot = static_cast<int>(nb_som_tot0);
  int nb_elem_tot = static_cast<int>(nb_elem_tot0);
  int dim = domaine.les_sommets().dimension(1);
 
  // Writing node coordinates
  // All quantities appearing in the .lml file must
  // be expressed in the Cartesian coordinate frame.
  // If the Axi option was chosen, Cartesian coordinates
  // are recomputed from the cylindrical coordinates.
 
  if (axi)
    {
      // Writing mesh nodes
      if ( dim == 3)
        {
          if(je_suis_maitre())
            s << nom_grille << " " << "3 " << nb_som_tot << "" << finl;
          s.lockfile();
 
          for (int nsom=0; nsom<domaine.nb_som(); nsom++)
            {
              s << domaine.coord(nsom,0)*cos(domaine.coord(nsom,1)) << " "
                << domaine.coord(nsom,0)*sin(domaine.coord(nsom,1)) << " "
                << domaine.coord(nsom,2) << "" << finl;
            }
 
          s.unlockfile();
          s.syncfile();
        }
      else if ( dim == 2)
        {
          if(je_suis_maitre())
            s << nom_grille << " " << "3 " << 2*nb_som_tot << "" << finl;
          s.lockfile();
          // Writing mesh nodes
          int nsom;
 
          for (nsom=0; nsom<domaine.nb_som(); nsom++)
            {
              s << domaine.coord(nsom,0)*cos(domaine.coord(nsom,1)) << " "
                << domaine.coord(nsom,0)*sin(domaine.coord(nsom,1)) << " "
                <<  0. << "" << finl;
            }
          for (nsom=0; nsom<domaine.nb_som(); nsom++)
            {
              s << domaine.coord(nsom,0)*cos(domaine.coord(nsom,1)) << " "
                << domaine.coord(nsom,0)*sin(domaine.coord(nsom,1)) << " "
                << 1. << "" << finl;
            }
 
          s.unlockfile();
          s.syncfile();
        }
    }
  else
    {
      // Writing mesh nodes
      if ( dim == 3)
        {
          if(je_suis_maitre())
            s << nom_grille << " " << "3 " << nb_som_tot << "" << finl;
          s.lockfile();
 
          for (int nsom=0; nsom<domaine.nb_som(); nsom++)
            {
              for (int j=0; j<3; j++)
                s << domaine.coord(nsom, j) << " " ;
              s << "" << finl;
            }
 
          s.unlockfile();
          s.syncfile();
        }
      else if ( dim == 2)
        {
          if(je_suis_maitre())
            s << nom_grille << " " << "3 " << 2*nb_som_tot << "" << finl;
          s.lockfile();
          // Writing mesh nodes
          int nsom;
 
          for (nsom=0; nsom<domaine.nb_som(); nsom++)
            {
              s << domaine.coord(nsom,0) << " "
                << domaine.coord(nsom,1) << " "
                << 0. << "" << finl;
            }
          for (nsom=0; nsom<domaine.nb_som(); nsom++)
            {
              s << domaine.coord(nsom,0) << " "
                << domaine.coord(nsom,1) << " "
                << 1. << "" << finl;
            }
 
          s.unlockfile();
          s.syncfile();
        }
    }
 
  //Currently an identical copy of the Domaine::postraiter_lml() method
 
  //////////////////////////////////////////////////////////////
 
  int j;
  int nsom = domaine.nb_som();
  Nom nom_top("Topologie_");
  nom_top+= domaine.le_nom();
  nom_top+="_";
  nom_top+=domaine.le_nom();
  Nom type_maille=domaine.type_elem()->nom_lml();
 
 
  //Added for syntax modification below
  int nb_elem = domaine.nb_elem() ;
  int nb_som_elem = domaine.nb_som_elem();
  const IntTab& sommet_elem = domaine.les_elems();
  int nb_som_PE = static_cast<int>(mppartial_sum(nsom));
  if(je_suis_maitre())
    {
      s << "TOPOLOGIE" << finl;
      s << nom_top << " " << nom_grille << "" << finl;
      s <<  "MAILLE" << finl;
      s << nb_elem_tot << "" << finl;
    }
 
  // Writing mesh elements
  if (dim == 3 )
    {
      s.lockfile();
 
      for (int nelem=0; nelem<nb_elem; nelem++)
        {
          s << type_maille << " ";
          for (j=0; j<nb_som_elem; j++)
            s << sommet_elem(nelem,j)+1+nb_som_PE << " " ;
          s << "" << finl;
        }
      s.unlockfile();
      s.syncfile();
    }
  else if ( dim == 2)
    {
      s.lockfile();
      for (int nelem=0; nelem<nb_elem; nelem++)
        {
          s << type_maille << " ";
          int count = 0;
          for (j=0; j<nb_som_elem && (sommet_elem( nelem,j) > -1); j++)
            s << sommet_elem( nelem,j)+1+2*nb_som_PE << " ", count++;
          for (j=0; j<nb_som_elem && (sommet_elem( nelem,j) > -1); j++)
            s << sommet_elem( nelem,j)+1+2*nb_som_PE+nsom << " ", count++;
          for (int k = count; k < 2 * nb_som_elem; k++) s << (int)0 << " ";
          s << "" << finl;
          s.flush();
        }
      s.unlockfile();
      s.syncfile();
    }
  if(je_suis_maitre())
    {
      s <<  "FACE" << finl ;
      s << (int)0 << "" << finl;
    }
 
 
  //////////////////////////////////////////////////////////////
 
  s.flush();
  s.syncfile();
 
  return 1;
}
 
int Format_Post_Lml::ecrire_temps_lml(const double temps,Nom& nom_fich)
{
 
  EcrFicPartage s;
  s.ouvrir(nom_fich,ios::app);
  if (je_suis_maitre())
    s << "TEMPS " << temps << finl;
  s.flush();
  s.syncfile();
 
  return 1;
}
 
int Format_Post_Lml::ecrire_champ_lml(const Domaine& domaine,const Noms& unite_,const Noms& noms_compo,
                                      int ncomp, double temps_,
                                      const Nom&   id_du_champ,
                                      const Nom&   id_du_domaine,
                                      const Nom&   localisation,
                                      const DoubleTab& data,Nom& nom_fich)
 
{
  EcrFicPartage os;
  os.ouvrir(nom_fich,ios::app);
  Nom nom_top("Topologie_");
  nom_top+= domaine.le_nom();
  nom_top+="_";
  nom_top+=domaine.le_nom();
  int dim = domaine.les_sommets().dimension(1);
  Nom nom_post = id_du_champ;
  //Nom nom_dom= domaine.le_nom();
 
  if (ncomp==-1)
    {
      int nb_compo_ = data.nb_dim()>1 ? data.dimension(1) : 1;
      if(localisation=="SOM")
        {
          int nb_som = domaine.nb_som();
          int nb_som_tot = static_cast<int>(Process::mp_sum(nb_som));
          int nb_som_PE = static_cast<int>(mppartial_sum(nb_som));
          if (dim==2)
            {
              nb_som_tot*=2;
              nb_som_PE*=2;
            }
          int nb_compo_dv = (nb_compo_==2 ? 3 : nb_compo_);
 
          if(je_suis_maitre())
            {
              os << "CHAMPPOINT " << nom_post << " " << nom_top << " " << temps_ << finl;
              os << nom_post << " " << nb_compo_dv << " " << unite_[0] << " "
                 << "type0" << " " << nb_som_tot << finl;
            }
          os.lockfile();
          for (int i=0; i<4-dim; i++)
            for (int j = 0; j < nb_som; j++)
              {
                os << i * nb_som + j + 1 + nb_som_PE << " ";
                for (int comp = 0; comp < nb_compo_; comp++)
                  os << data(j, comp) << " ";
                if (nb_compo_ == 2) os << "0.";
                os << finl;
              }
          os.unlockfile();
          os.syncfile();
        }
      else if(localisation=="ELEM")
        {
          int nb_elem=domaine.nb_elem();
          int nb_elem_tot = static_cast<int>(Process::mp_sum(nb_elem));
          int nb_elem_PE = static_cast<int>(mppartial_sum(nb_elem));
          for (int comp=0; comp<nb_compo_; comp++)
            {
              if (je_suis_maitre())
                {
                  os << "CHAMPMAILLE ";
                  os << (nb_compo_ == 1 ? nom_post : noms_compo[comp]) << " " << nom_top << " " << temps_ << finl;
                  os << (nb_compo_ == 1 ? nom_post : noms_compo[comp]) << " " << (int)1 << " " << unite_[comp] << " " ;
                  os << "type0" << " " << nb_elem_tot << finl;
                }
              os.lockfile();
              for (int j=0; j<nb_elem; j++)
                os << j+1+nb_elem_PE << " " << data(j,comp) << finl;
              os.unlockfile();
              os.syncfile();
            }
        }
      else
        {
          Cerr << "We do not know to postprocess " <<  id_du_champ
               << " with the keyword " << localisation << finl;
          return -1;
        }
    }
  else
    {
      if(localisation=="SOM")
        {
          int nb_som=domaine.nb_som();
          int nb_som_tot = static_cast<int>(Process::mp_sum(nb_som));
          int nb_som_PE = static_cast<int>(mppartial_sum(nb_som));
          if (dim==2)
            {
              nb_som_tot*=2;
              nb_som_PE*=2;
            }
          if(je_suis_maitre())
            {
              os << "CHAMPPOINT ";
              os << nom_post << " " << nom_top << " " << temps_ << finl;
              os << nom_post << " " << (int)1 << " " << unite_[ncomp] << " "
                 << "type0" << " " << nb_som_tot << finl;
            }
          os.lockfile();
          for (int i=0; i<4-dim; i++)
            for (int j=0; j<nb_som; j++)
              os << i*nb_som+j+1+nb_som_PE << " " << data(j) << finl;
          os.unlockfile();
          os.syncfile();
        }
      else if(localisation=="ELEM")
        {
          int nb_elem=domaine.nb_elem();
          int nb_elem_tot = static_cast<int>(Process::mp_sum(nb_elem));
          int nb_elem_PE = static_cast<int>(mppartial_sum(nb_elem));
          if(je_suis_maitre())
            {
              os << "CHAMPMAILLE ";
              os << nom_post << " " << nom_top << " " << temps_ << finl;
              os << nom_post << " " << (int)1 << " " << unite_[ncomp] << " " ;
              os << "type0" << " " << nb_elem_tot << finl;
            }
          os.lockfile();
          for (int j=0; j<nb_elem; j++)
            os << j+1+nb_elem_PE << " " << data(j) << finl;
          os.unlockfile();
          os.syncfile();
        }
      else
        {
          Cerr << "We do not know to postprocess " << id_du_champ
               << " with the keyword " << localisation << finl;
          return -1;
        }
    }
  return 1;
}
 
 
int Format_Post_Lml::ecrire_item_int_lml(const Nom&   id_item,
                                         const Nom&   id_domaine,
                                         const IntVect& vect,
                                         const Nom& nom_fic)
{
  EcrFicPartage os;
  os.ouvrir(nom_fic,ios::app);
 
  Nom nom_vect(id_item);
  nom_vect += "_elem_dom";
 
  Nom nom_topologie("Topologie_");
  nom_topologie+= id_domaine;
 
  //Currently temps_courant is not passed as argument
  ////os << "CHAMPMAILLE " << nom_vect << " "<< nom_topologie << " " << temps_courant << finl;
 
  const IntTab* intT = dynamic_cast<const IntTab*>(&vect);
  if (intT)
    {
      const IntTab& tab = *intT;
      os.lockfile();
      os << nom_vect << tab.dimension(1) << " 1 " << " type0 " << tab.dimension(0) << finl;
      os.unlockfile();
      os.syncfile();
      {
        int tab_dimension0_opt=tab.dimension(0);
        os.lockfile();
        for (int i=0; i<tab_dimension0_opt; i++)
          {
            for (int j=0; j<tab.dimension(1); j++)
              {
                int tab_dimension1_opt=tab.dimension(1);
                for (int j2=0; j2<tab_dimension1_opt; j2++)
                  os << i+1 << " " << tab(i,j2) << " ";
              }
            os << finl;
          }
        os.unlockfile();
        os.syncfile();
      }
 
    }
  else
    {
      os.lockfile();
      os << nom_vect << " 1" << " 1 " << " type0 " << vect.size() << finl;
      for (int i=0; i<vect.size(); i++)
        os << i+1 << " " << vect(i) << finl;
      os.unlockfile();
      os.syncfile();
    }
 
  os.flush();
  os.syncfile();
 
  return 1;
}