Format_Post_Med.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_Med.h>
#include <Ecrire_MED.h>
#include <EcrFicPartage.h>
#include <EFichier.h>
#include <Param.h>
#include <med++.h>
 
Implemente_instanciable_sans_constructeur(Format_Post_Med,"Format_Post_Med",Format_Post_base);
 
 
/////////////////////////////////////////////////////////////////////////////
//Parameter: med_basename_
// Meaning: beginning of the file name
//Extension med
 
//Specifics of this format
//Generation of a format.med.data file usable to rewrite data in another format
//-Methods completer_post_med(...) and preparer_post_med(...) to generate the files
//-Method finir_med(...) to concatenate the created files
///////////////////////////////////////////////////////////////////////////////////////
 
/*! @brief Default constructor:
 *
 */
 
//Specify in comment what is fixed by default
Format_Post_Med::Format_Post_Med()
{
  reset();
}
 
/*! @brief Resets the object to the state obtained by the default constructor.
 *
 */
void Format_Post_Med::reset()
{
  med_basename_ = "??";
}
 
/*! @brief erreur => exit
 *
 */
Sortie& Format_Post_Med::printOn(Sortie& os) const
{
  Cerr << "Format_Post_Med::printOn : error" << finl;
  exit();
  return os;
}
 
/*! @brief Reads the post-processing parameters in "data set" format. The expected format is:
 *
 *   {
 *     nom_fichier filename_sans_extension
 *   }
 *
 */
Entree& Format_Post_Med::readOn(Entree& is)
{
  Format_Post_base::readOn(is);
  return is;
}
 
void Format_Post_Med::set_param(Param& param) const
{
  param.ajouter("nom_fichier",&med_basename_,Param::REQUIRED);
}
 
/*! @brief Initializes the class with default parameters.
 */
int Format_Post_Med::initialize_by_default(const Nom& file_basename)
{
  med_basename_= file_basename;
  return 1;
}
 
int Format_Post_Med::initialize(const Nom& file_basename, const int format, const Nom& option_para)
{
  med_basename_= file_basename;
  return 1;
}
 
int Format_Post_Med::ecrire_entete(const double temps_courant,const int reprise,const int est_le_premier_post)
{
 
  Nom nom_fich(med_basename_);
  nom_fich +=".";
  Nom format="med";
  nom_fich += format;
  nom_fich += ".index";
 
  ecrire_entete_med(nom_fich,est_le_premier_post);
  return 1;
}
 
int Format_Post_Med::finir(const int est_le_dernier_post)
{
  Nom nom_fic_base(med_basename_);
  nom_fic_base += ".med";
  Nom nom_fic=nom_fic_base.nom_me(Process::me());
  Cerr << "MED file " << nom_fic << " closed. See .index file for more infos." << finl;
  return 1;
}
 
 
int Format_Post_Med::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)
{
  return 1;
}
 
int Format_Post_Med::preparer_post(const Nom& id_du_domaine,const int est_le_premier_post,
                                   const int reprise,
                                   const double t_init)
{
  return 1;
}
 
int Format_Post_Med::ecrire_domaine(const Domaine& domaine,const int est_le_premier_post)
{
  const OBS_PTR(Domaine_dis_base) domaine_dis_base;
  return ecrire_domaine_dis(domaine, domaine_dis_base, est_le_premier_post);
}
 
void Format_Post_Med::ecrire_domaine_dual(const Domaine& domaine, const int est_le_premier_post)
{
  Nom nom_fich(med_basename_);
  nom_fich +="_dual.";
  Nom format="med";
  nom_fich += format;
 
  bool append = !est_le_premier_post;
 
  assert(domaine_dis_);
  ecr_med_.set_file_name_and_dom(nom_fich, domaine, &domaine_dis_.valeur());
  ecr_med_.ecrire_domaine_dual(append);
}
 
/*! @brief voir Format_Post_base::ecrire_domaine
 */
int Format_Post_Med::ecrire_domaine_dis(const Domaine& domaine,const OBS_PTR(Domaine_dis_base)& domaine_dis_base,const int est_le_premier_post)
{
  domaine_dis_ = domaine_dis_base; // as in base class
 
  Nom nom_fich(med_basename_);
  nom_fich +=".";
  Nom format="med";
  nom_fich += format;
  nom_fich += ".index";
 
  Nom nom_fic_base(med_basename_);
  nom_fic_base += ".";
  nom_fic_base += format;
  Nom nom_fic=nom_fic_base.nom_me(Process::me());
 
  ecrire_domaine_med(domaine,nom_fic,est_le_premier_post,nom_fich);
 
  return 1; // ok all is well
}
 
/*! @brief Starts writing a new time step.
 *
 * Opens the master file in APPEND mode and adds a line
 *    "TEMPS xxxxx" if this time has not yet been written.
 */
int Format_Post_Med::ecrire_temps(const double temps)
{
  Nom nom_fich(med_basename_);
  nom_fich +=".";
  Nom format="med";
  nom_fich += format;
  nom_fich += ".index";
 
  ecrire_temps_med(temps,nom_fich);
 
  return 1;
}
 
/*! @brief voir Format_Post_base::ecrire_champ
 *
 */
int Format_Post_Med::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)
{
  Nom nom_fich_index(med_basename_);
  nom_fich_index +=".";
  Nom format="med";
  nom_fich_index += format;
  nom_fich_index += ".index";
 
  Nom nom_fich_ecrit(med_basename_);
  nom_fich_ecrit += ".";
  nom_fich_ecrit += format;
 
  ecrire_champ_med(domaine,unite_,noms_compo,ncomp,temps_,nom_fich_ecrit,id_champ,id_du_domaine,localisation,valeurs,nom_fich_index);
 
  return 1;
}
 
/*! @brief
 *
 */
 
int Format_Post_Med::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)
{
 
  return 1;
}
 
int Format_Post_Med::ecrire_entete_med(Nom& nom_fich,const int est_le_premier_post)
{
 
  if (est_le_premier_post && Process::je_suis_maitre())
    {
      SFichier s(nom_fich);
      s << "TRUST Version "  << TRUST_VERSION << finl;
      s << nom_du_cas() << finl;
      s << "TRUST" << finl;
    }
  return 1;
}
 
int Format_Post_Med::finir_med(Nom& nom_fich,int& est_le_dernier_post)
{
  ////Nom nom_fichier_base = nom_fich.prefix(".med.index");
  Nom nom_fichier_base = nom_fich;
  nom_fichier_base.prefix(".med.index");
  Nom nom_fichier(nom_fichier_base);
  Nom format = "med";
  nom_fichier += ".";
  nom_fichier += format;
 
  if (!est_le_dernier_post || !Process::je_suis_maitre()) return 1;
 
  // we want to concatenate the 3 files
  SFichier file3(nom_fichier+".data");
  SFichier file("postmed.data");
  for (int i=0; i<3; i++)
    {
      Nom num(i);
      Nom fic2(nom_fichier);
      fic2+=".med";
      fic2+=num;
      {
        EFichier file2(fic2);
        Nom motlu;
        file2>>motlu;
        while (!file2.eof())
          {
            file<<motlu<<" "<<finl;
            if (motlu!="#")
              file3<<motlu<<" "<<finl;
            file2>>motlu;
          }
      }
    }
 
  SFichier s;
  s.ouvrir(nom_fich,ios::app);
  s << "FIN" << finl;
 
  return 1;
}
 
int Format_Post_Med::completer_post_med(const Nom& nom_fich2,const Nom& nom1,const Nom& nom2)
{
  if (je_suis_maitre())
    {
      SFichier file;
      file.ouvrir(nom_fich2,ios::app);
      file <<nom1<<" "<<nom2<<finl;
    }
  return 1;
}
 
int Format_Post_Med::preparer_post_med(const Nom& nom_fich1,const Nom& nom_fich2,const Nom& nom_fich3,
                                       const Nom& id_du_domaine,const int est_le_premier_post)
{
  Nom nom_fich(nom_fich1);
  nom_fich.prefix(".med0");
 
  if (!je_suis_maitre()) return 1;
 
  if (est_le_premier_post)
    {
      SFichier file(nom_fich1);
      file<<"{ Dimension "<<dimension<<finl<<"# export Domaine "<<id_du_domaine<<finl;
    }
  else
    {
      SFichier file(nom_fich1,ios::app);
      file<<" export Domaine "<<id_du_domaine<<finl;
    }
 
  SFichier file;
  if (est_le_premier_post)
    {
      file.ouvrir(nom_fich2);
      file<<"# \n Pbc_MED pbmed\n Lire pbmed { "<<finl;
    }
  else
    {
      file.ouvrir(nom_fich2,ios::app);
      file<<"} } } , "<<finl;
    }
  file<<nom_fich<<" "<<id_du_domaine<<"  { Postraitement { Champs dt_post 1e-9 {"<<finl;
 
  if (est_le_premier_post)
    {
      SFichier file2(nom_fich3);
      file2<<"} } } } }"<<finl;
    }
 
  return 1;
}
 
int Format_Post_Med::ecrire_domaine_med(const Domaine& domaine,const Nom& nom_fic,const int est_le_premier_post,Nom& nom_fich)
{
  int dim = domaine.les_sommets().dimension(1);
  bool append = !est_le_premier_post;
  if (je_suis_maitre())
    {
      SFichier s;
      s.ouvrir(nom_fich, ios::app);
      s << "format MED: " << ecr_med_.version() << finl;
      s << "dimension: " << dim << finl;
      s << "domaine: " << domaine.le_nom() << finl;
      s << "nb_proc: " << Process::nproc() << finl;
      Cerr << "Opening MED file " << nom_fic << " with " << ecr_med_.version() << " format. ";
      Cerr << finl;
    }
  ecr_med_.set_file_name_and_dom(nom_fic, domaine, domaine_dis_ ? &domaine_dis_.valeur() : nullptr);
  ecr_med_.ecrire_domaine_dis(append);
  return 1;
 
}
 
int Format_Post_Med::ecrire_temps_med(const double temps,Nom& nom_fich)
{
  if (je_suis_maitre())
    {
      SFichier s;
      s.ouvrir(nom_fich,ios::app);
      s << "TEMPS " << temps << finl;
    }
  return 1;
}
 
int Format_Post_Med::ecrire_champ_med(const Domaine& dom,const Noms& unite_, const Noms& noms_compo,
                                      int ncomp, double temps_,const Nom& nom_pdb,
                                      const Nom&   id_du_champ,
                                      const Nom&   id_du_domaine,
                                      const Nom&   loc_post,
                                      const DoubleTab& valeurs,Nom& nom_fich)
{
  Nom fic = nom_pdb.nom_me(me());
  ecr_med_.set_file_name_and_dom(fic, dom, domaine_dis_ ? &domaine_dis_.valeur() : nullptr);
 
  Nom nom_post(id_du_champ);
  Noms noms_compo_courts(noms_compo);
  if (ncomp != -1)
    {
      nom_post = noms_compo[ncomp];
    }
  nom_post.prefix(dom.le_nom());
  Nom nom_dom="";
  for (int i = 0; i < noms_compo.size(); ++i)
    noms_compo_courts[i] = Motcle(noms_compo_courts[i]).getPrefix(dom.le_nom());
  if (loc_post == "SOM")
    {
      nom_post.prefix("_som_");
      nom_post.prefix("_SOM_");
      nom_dom = dom.le_nom();
      for (int i = 0; i < noms_compo.size(); ++i)
        noms_compo_courts[i] = Motcle(noms_compo_courts[i]).getPrefix("_SOM_");
    }
  else if (loc_post == "ELEM")
    {
      nom_post.prefix("_ELEM_");
      nom_post.prefix("_elem_");
      nom_dom = dom.le_nom();
      for (int i = 0; i < noms_compo.size(); ++i)
        noms_compo_courts[i] = Motcle(noms_compo_courts[i]).getPrefix("_ELEM_");
    }
  else if (loc_post == "FACES")
    {
#ifdef MEDCOUPLING_
      nom_post.prefix("_FACES_");
      nom_post.prefix("_faces_");
      nom_dom = dom.le_nom();
      for (int i = 0; i < noms_compo.size(); ++i)
        noms_compo_courts[i] = Motcle(noms_compo_courts[i]).getPrefix("_FACES_");
#else
      Cerr << "Post-processing in MED format on faces needs MEDCoupling" << finl;
#endif
    }
  if (je_suis_maitre())
    {
      //Opening the file
      SFichier os;
      os.ouvrir(nom_fich, ios::app);
      os << "champ: " << nom_post << " " << nom_dom << " " << loc_post << finl;
    }
 
  Nom type_elem = dom.type_elem()->que_suis_je();
 
  // modify component names
  for (int i = 0; i < noms_compo.size(); ++i)
    noms_compo_courts[i] = Motcle(noms_compo_courts[i]).getSuffix(nom_post);
 
  if (loc_post == "SOM")
    ecr_med_.ecrire_champ("CHAMPPOINT", id_du_champ, valeurs, unite_, noms_compo_courts, type_elem, temps_);
  else if (loc_post == "ELEM")
    ecr_med_.ecrire_champ("CHAMPMAILLE", id_du_champ, valeurs, unite_, noms_compo_courts, type_elem, temps_);
  else if (loc_post == "FACES")
    ecr_med_.ecrire_champ("CHAMPFACES", id_du_champ, valeurs, unite_, noms_compo_courts, type_elem, temps_);
  else
    {
      Cerr << "We do not know to postprocess " << id_du_champ
           << " with the keyword " << loc_post << finl;
      return -1;
    }
  return 1;
}