ArcaneCheckpointModule.cc

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2026 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* ArcaneCheckpointModule.cc                                   (C) 2000-2026 */
/*                                                                           */
/* Module gérant les protections/reprises.                                   */
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/ArcanePrecomp.h"
 
#include "arcane/utils/ScopedPtr.h"
#include "arcane/utils/StringBuilder.h"
#include "arcane/utils/Convert.h"
 
#include "arcane/core/ISubDomain.h"
#include "arcane/core/EntryPoint.h"
#include "arcane/core/Timer.h"
#include "arcane/core/ITimeHistoryMng.h"
#include "arcane/core/ModuleFactory.h"
#include "arcane/core/ServiceUtils.h"
#include "arcane/core/ICheckpointWriter.h"
#include "arcane/core/ICheckpointMng.h"
#include "arcane/core/Directory.h"
#include "arcane/core/IParallelMng.h"
 
#include "arcane/core/OutputChecker.h"
#include "arcane/std/ArcaneCheckpoint_axl.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

class ArcaneCheckpointModule
: public ArcaneArcaneCheckpointObject
{
 public:
 
  ArcaneCheckpointModule(const ModuleBuilder& cb);
  ~ArcaneCheckpointModule();
 
 public:
 
  virtual VersionInfo versionInfo() const { return VersionInfo(0, 9, 1); }
 
 public:
 
  virtual void checkpointCheckAndWriteData();
  virtual void checkpointStartInit();
  virtual void checkpointInit();
  virtual void checkpointExit();
 
 private:
 
  OutputChecker m_output_checker;
  Timer* m_checkpoint_timer;
  ICheckpointWriter* m_checkpoint_writer;
  String m_checkpoint_dirname;
 
 private:
 
  void _doCheckpoint(bool save_history);
  void _dumpStats();
  void _getCheckpointService();
  void _setDirectoryName();
  bool _checkHasOutput();
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ARCANE_REGISTER_MODULE_ARCANECHECKPOINT(ArcaneCheckpointModule);
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ArcaneCheckpointModule::
ArcaneCheckpointModule(const ModuleBuildInfo& mb)
: ArcaneArcaneCheckpointObject(mb)
, m_output_checker(mb.subDomain(), "CheckpointRestart")
, m_checkpoint_timer(0)
, m_checkpoint_writer(0)
, m_checkpoint_dirname(".")
{
  m_checkpoint_timer = new Timer(mb.subDomain(), "Checkpoint", Timer::TimerReal);
  m_output_checker.assignIteration(&m_next_iteration, &options()->period);
  m_output_checker.assignGlobalTime(&m_next_global_time, &options()->frequency);
  m_output_checker.assignCPUTime(&m_next_cpu_time, &options()->frequencyCpu);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ArcaneCheckpointModule::
~ArcaneCheckpointModule()
{
  delete m_checkpoint_timer;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void ArcaneCheckpointModule::
checkpointInit()
{
  // Contrairement aux autres types de sortie, il faut remettre le temps
  // CPU de la prochaine sortie à zéro car le temps CPU utilisé est de
  // l'exécution courante.
  m_next_cpu_time = options()->frequencyCpu();
  info() << " -------------------------------------------";
  info() << "|            PROTECTION-REPRISE             |";
  info() << " -------------------------------------------";
  info() << " ";
  //info() << " Utilise le service '" << options()->checkpointServiceName()
  //     << "' pour les protections";
  info() << " ";
  m_output_checker.initialize();
  info() << " ";
  // Si les protections sont actives, vérifie que le service spécifié
  // existe bien
  if (options()->doDumpAtEnd()) {
    info() << "Protection requise à la fin des calculs";
    _getCheckpointService();
  }
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
bool ArcaneCheckpointModule::
_checkHasOutput()
{
  Real old_time = m_global_old_time();
  Real current_time = m_global_time();
  Integer iteration = m_global_iteration();
  Integer cpu_used = Convert::toInteger(m_global_cpu_time());
 
  bool do_output = m_output_checker.check(old_time, current_time, iteration, cpu_used);
  return do_output;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void ArcaneCheckpointModule::
checkpointStartInit()
{
  m_next_global_time = 0.;
  m_next_iteration = 0;
  m_next_cpu_time = 0;
 
  // Initialise le vérificateur de sortie. Il faut le faire ici si on
  // veut sauver des choses à l'itération 1.
  _checkHasOutput();
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
void ArcaneCheckpointModule::
checkpointExit()
{
  if (!options()->doDumpAtEnd())
    return;
 
  _doCheckpoint(false);
  _dumpStats();
 
  if (m_checkpoint_writer)
    m_checkpoint_writer->close();
  m_checkpoint_writer = 0;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void ArcaneCheckpointModule::
_dumpStats()
{
  // Affiche statistiques d'exécutions
  Real total_time = m_checkpoint_timer->totalTime();
  info() << "Temps total passé dans la sortie de protection (secondes) : " << total_time;
  Integer nb_time = m_checkpoint_timer->nbActivated();
  if (nb_time != 0)
    info() << "Temps moyen par sortie (secondes) : " << total_time / nb_time
           << " (pour " << nb_time << " sorties";
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void ArcaneCheckpointModule::
_setDirectoryName()
{
  Directory export_dir = subDomain()->storageDirectory();
  if (export_dir.path().null())
    export_dir = subDomain()->exportDirectory();
 
  Directory output_directory = Directory(export_dir, "protection");
  IParallelMng* pm = parallelMng();
  if (pm->isMasterIO())
    output_directory.createDirectory();
  pm->barrier();
 
  m_checkpoint_dirname = output_directory.path();
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

void ArcaneCheckpointModule::
_doCheckpoint(bool save_history)
{
  {
    Timer::Sentry sentry(m_checkpoint_timer);
    Timer::Phase tp(subDomain(), TP_InputOutput);
    if (!m_checkpoint_writer)
      _getCheckpointService();
    if (m_checkpoint_writer) {
      Integer nb_checkpoint = m_checkpoints_time.size();
      m_checkpoints_time.resize(nb_checkpoint + 1);
      Real checkpoint_time = m_global_time();
      m_checkpoints_time[nb_checkpoint] = checkpoint_time;
      m_checkpoint_writer->setCheckpointTimes(m_checkpoints_time);
      m_checkpoint_writer->setBaseDirectoryName(m_checkpoint_dirname);
      info() << "****  Protection active au temps " << checkpoint_time
             << " répertoire=" << m_checkpoint_dirname
             << " numéro " << nb_checkpoint << " ******";
      subDomain()->checkpointMng()->writeDefaultCheckpoint(m_checkpoint_writer);
    }
  }
  info() << "Temps d'écriture du point de contrôle (secondes) : "
         << m_checkpoint_timer->lastActivationTime();
 
  // Sauve les historiques
  if (save_history)
    subDomain()->timeHistoryMng()->dumpHistory(true);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
void ArcaneCheckpointModule::
_getCheckpointService()
{
  ICheckpointWriter* checkpoint = options()->checkpointService();
  if (!checkpoint) {
    StringUniqueArray valid_values;
    options()->checkpointService.getAvailableNames(valid_values);
    pfatal() << "Les protections sont requises mais le service de protection/restauration sélectionné ("
             << options()->checkpointService.serviceName() << ") n'est pas disponible "
             << "(valeurs valides : " << String::join(", ", valid_values) << ")";
  }
  m_checkpoint_writer = checkpoint;
  _setDirectoryName();
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

void ArcaneCheckpointModule::
checkpointCheckAndWriteData()
{
  bool do_output = _checkHasOutput();
  if (!do_output)
    return;
 
  info() << "Protection requise.";
  // TEMPORAIRE:
  // Contrairement a la protection en fin d'execution, celle-ci a lieu
  // dans à la fin de la boucle en temps, mais avant que le numéro
  // d'itération ne soit incrémenté. Si on reprend à ce temps,
  // le numéro d'itération ne sera pas bon. Pour corriger ce problème,
  // on incrémente ce nombre avant la protection et on le remet
  // à sa bonne valeur après.
  // SDC : ce probleme n'est pas constaté. Il me semble que c'est bien le numérod
  // de l'itération courante qu'il faut sauver. Changé pour un problème de restart 
  // pour une application IFPEN (en interne Bugzilla 778.
  //  m_global_iteration = m_global_iteration() + 1;
  _doCheckpoint(true);
  //  m_global_iteration = m_global_iteration() - 1;
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/