CheckpointerDataInterface.hpp

/*
 * CheckpointerDataInterface.hpp
 *
 *  Created on Feb 22, 2018
 *      Author: Pete Schultz
 */

#ifndef CHECKPOINTERDATAINTERFACE_HPP_
#define CHECKPOINTERDATAINTERFACE_HPP_

#include "observerpattern/Observer.hpp"

#include "Checkpointer.hpp"
#include "checkpointing/CheckpointingMessages.hpp"

namespace PV {

class CheckpointerDataInterface : public Observer {
  public:
   virtual Response::Status registerData(Checkpointer *checkpointer);

   virtual Response::Status respond(std::shared_ptr<BaseMessage const> message) override;

   virtual Response::Status readStateFromCheckpoint(Checkpointer *checkpointer) {
      return Response::NO_ACTION;
   }

   MPIBlock const *getMPIBlock() { return mMPIBlock; }

  protected:
   Response::Status
   respondRegisterData(std::shared_ptr<RegisterDataMessage<Checkpointer> const> message);
   Response::Status respondReadStateFromCheckpoint(
         std::shared_ptr<ReadStateFromCheckpointMessage<Checkpointer> const> message);

   Response::Status
   respondProcessCheckpointRead(std::shared_ptr<ProcessCheckpointReadMessage const> message);
   Response::Status
   respondPrepareCheckpointWrite(std::shared_ptr<PrepareCheckpointWriteMessage const> message);

   virtual Response::Status processCheckpointRead() { return Response::NO_ACTION; }
   virtual Response::Status prepareCheckpointWrite() { return Response::NO_ACTION; }

  private:
   MPIBlock const *mMPIBlock = nullptr;
};

} // namespace PV

#endif // CHECKPOINTERDATAINTERFACE_HPP_