checkpoint.h

Go to the documentation of this file.

/*
 * Copyright 2006-2009 Savarese Software Research Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     https://www.savarese.com/software/ApacheLicense-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef __SSRC_WSPR_UTILITY_CHECKPOINT_H
#define __SSRC_WSPR_UTILITY_CHECKPOINT_H

#include <ssrc/wispers-packages.h>

#include <boost/archive/binary_iarchive.hpp>
#include <boost/archive/binary_oarchive.hpp>
#include <boost/filesystem/convenience.hpp>
#include <fstream>

__BEGIN_NS_SSRC_WSPR_UTILITY

template<typename T>
void save_checkpoint(const std::string & filename, const T & data)
  SSRC_DECL_THROW(std::runtime_error, boost::archive::archive_exception)
{
  std::ofstream ofs(filename.c_str(), std::ios_base::out |
                    std::ios_base::binary | std::ios_base::trunc);
  if(ofs.is_open()) {
    boost::archive::binary_oarchive oa(ofs);
    oa << data;
  } else {
    throw std::runtime_error("checkpoint file cannot be opened for writing");
  }
}

template<typename T>
void load_checkpoint(const std::string & filename, T & data)
  SSRC_DECL_THROW(std::runtime_error, boost::archive::archive_exception)
{
  if(boost::filesystem::exists(filename)) {
    std::ifstream ifs(filename.c_str(),
                      std::ios_base::in | std::ios_base::binary);

    if(ifs.is_open()) {
      boost::archive::binary_iarchive ia(ifs);
      ia >> data;
    } else {
      throw std::runtime_error("checkpoint file exists but cannot be opened");
    }
  }
}

__END_NS_SSRC_WSPR_UTILITY

#endif