CheckpointManager.H

Go to the documentation of this file.

 
#ifndef _CHECKPOINTMANAGER_H_
#define _CHECKPOINTMANAGER_H_
 
#include <string>
#include "ParmParser.H"
#include "../Core/CRTree.H"
#include "../Losses/Losses.H"
#include "../Prob/Prob.H"
#include "../Propagation/Propagation.H"
#include "../Utils/RngThread.H"
#include "../Utils/Types.H"
#include "hdf5.h"
 
namespace criptic {
 
  class CheckpointManager {
 
  public:
 
    CheckpointManager(const ParmParser& pp,
              const propagation::Propagation& prop,
              const Losses& loss);
 
    void writeCheckpoint(const int step,
             const Real t,
             const Real dt,
             const CRTree& tree,
             Prob* prob);
 
    void readCheckpoint(const std::string& filename,
            int& step,
            Real& t,
            Real& dt,
            CRTree& tree,
            Prob* prob,
      bool restoreRng = true
    );
 
    int lastStep() const { return lastChkStep; }
    
    int nextStep() const { return nextChkStep; }
    
    Real lastTime() const { return lastChkTime; }
    
    Real nextTime() const { return nextChkTime; }
 
    static const int fWidth = 15;  
  private:
 
    void syncCounts(std::vector<IdxType>& nPkt,
            std::vector<IdxType>& nSrc,
            std::vector<IdxType>& nDel,
            const CRTree& tree) const;
    
#ifdef ENABLE_MPI
     void sendCheckpointData(const Real& t,
                 const CRTree& tree) const;
#endif
 
    void writePackets(hid_t& fileID,
              const std::vector<IdxType>& nPkt,
              const int& step,
              const Real& t,
              const Real& dt,
              const CRTree& tree) const;
 
    void writeSources(hid_t& fileID,
              const std::vector<IdxType>& nSrc,
              const CRTree& tree) const;
    
    void writeDelPackets(hid_t& fileID,
             const std::vector<IdxType>& nDel,
             const CRTree& tree) const;
    
    void writeFieldQty(hid_t& fileID,
               const std::vector<IdxType>& nPkt,
               const CRTree& tree) const;
    
    void writeLoss(hid_t& fileID,
           const std::vector<IdxType>& nPkt,
           const Real& t,
           const CRTree& tree) const;
 
    void writePhotonsFull(hid_t& fileID,
              const std::vector<Real>& photEn,
              const std::vector<IdxType>& nPkt,
              const Real& t,
              const CRTree& tree) const;
    
    void writePhotonsSummary(hid_t& fileID,
                 const std::vector<Real>& photEn,
                 const Real t, 
                 const CRTree& tree) const;
    
    void writeIonization(hid_t& fileID,
             const std::vector<IdxType>& nPkt,
             const Real& t,
             const CRTree& tree) const;
 
#ifndef TRACK_PITCH_ANGLE
    void writePdot(hid_t& fileID,
             const std::vector<IdxType>& nPkt,
             const Real& t,
             const CRTree& tree) const;
#endif
 
    void writeRng(hid_t& fileID, const CRTree& tree) const;
 
    // Internal data
    std::string baseChkName; 
    bool needFQ;             
    bool needFQG;            
    bool writeDelPackets_;   
    bool writeFieldQty_;     
    bool writeLoss_;         
    bool writePhotons_;      
    bool writePhotonsFull_;  
    bool writeIonization_;   
#ifndef TRACK_PITCH_ANGLE
    bool writePdot_;         
#endif
    std::vector<Real> photEn; 
    int verbosity;           
    int chkNum;              
    int chkInt;              
    int lastChkStep;         
    int nextChkStep;         
    Real chkDt;              
    Real lastChkTime;        
    Real nextChkTime;        
    hid_t RealVec_HDF5;      
    hid_t RealTensor2_HDF5;  
    hid_t CRPacket_HDF5;     
    hid_t CRSource_HDF5;     
    hid_t FieldQty_HDF5;     
    hid_t FieldQtyGrad_HDF5; 
    hid_t RngState_HDF5;     
  };
 
}
 
// _CHECKPOINTMANAGER_H_
#endif