CheckpointableFileStream.hpp

#ifndef __CHECKPOINTABLEFILESTREAM_HPP__
#define __CHECKPOINTABLEFILESTREAM_HPP__

#include "CheckpointerDataInterface.hpp"
#include "io/FileStream.hpp"

#include <string>

using std::string;

namespace PV {

class CheckpointableFileStream : public FileStream, public CheckpointerDataInterface {

  public:
   CheckpointableFileStream(
         string const &path,
         bool newFile,
         Checkpointer *checkpointer,
         string const &objName,
         bool verifyWrites);

   CheckpointableFileStream(
         string const &path,
         bool newFile,
         Checkpointer *checkpointer,
         string const &objName);
   virtual Response::Status respond(std::shared_ptr<BaseMessage const> message) override;
   virtual void write(void const *data, long length) override;
   virtual void read(void *data, long length) override;
   virtual void setOutPos(long pos, bool fromBeginning) override;
   virtual void setInPos(long pos, bool fromBeginning) override;

  private:
   void initialize(
         string const &path,
         bool newFile,
         Checkpointer *checkpointer,
         string const &objName,
         bool verifyWrites);
   void setDescription();
   virtual Response::Status registerData(Checkpointer *checkpointer) override;
   Response::Status respondProcessCheckpointRead(ProcessCheckpointReadMessage const *message);
   void syncFilePos();
   void updateFilePos();
   long mFileReadPos  = 0;
   long mFileWritePos = 0;
   string mObjName; // Used for CheckpointerDataInterface
};
}

#endif