src/checkpoint.h

#ifndef CHECKPOINT_H
#define CHECKPOINT_H
 
#include "hila.h"
#include "parameters.h"
 
template <typename group>
void checkpoint(const GaugeField<group> &U, int iteration, const parameters &p) {
 
    double t = hila::gettime();
    // save config
    U.config_write(p.config_file);
 
    // write run_status file
    if (hila::myrank() == 0) {
        std::ofstream outf;
        outf.open("run_status", std::ios::out | std::ios::trunc);
        outf << "iteration   " << iteration + 1 << '\n';
        outf << "seed        " << static_cast<uint64_t>(hila::random() * (1UL << 61)) << '\n';
        outf << "time        " << hila::gettime() << '\n';
        outf.close();
    }
 
    std::stringstream msg;
    msg << "Checkpointing, time " << hila::gettime() - t;
    hila::timestamp(msg.str().c_str());
}
 
template <typename group>
bool restore_checkpoint(GaugeField<group> &U, int &trajectory, parameters &p) {
 
    uint64_t seed;
    bool ok = true;
    p.time_offset = 0;
 
    hila::input status;
    if (status.open("run_status", false, false)) {
 
        hila::out0 << "RESTORING FROM CHECKPOINT:\n";
 
        trajectory = status.get("iteration");
        seed = status.get("seed");
        p.time_offset = status.get("time");
        status.close();
 
        hila::seed_random(seed);
 
        U.config_read(p.config_file);
 
        ok = true;
    } else {
 
        std::ifstream in;
        in.open(p.config_file, std::ios::in | std::ios::binary);
 
        if (in.is_open()) {
            in.close();
 
            hila::out0 << "READING initial config\n";
 
            U.config_read(p.config_file);
 
            ok = true;
        } else {
 
            ok = false;
        }
    }
 
    return ok;
}
 
#endif