WeightsPairInterface.hpp

/*
 * WeightsPairInterface.hpp
 *
 *  Created on: Jan 8, 2018
 *      Author: Pete Schultz
 */

#ifndef WEIGHTSPAIRINTERFACE_HPP_
#define WEIGHTSPAIRINTERFACE_HPP_

#include "columns/BaseObject.hpp"
#include "components/ConnectionData.hpp"
#include "components/PatchSize.hpp"
#include "components/Weights.hpp"

namespace PV {

class WeightsPairInterface : public BaseObject {
  public:
   WeightsPairInterface(char const *name, HyPerCol *hc);

   virtual ~WeightsPairInterface();

   void needPre();

   void needPost();

   Weights *getPreWeights() { return mPreWeights; }
   Weights *getPostWeights() { return mPostWeights; }

   ConnectionData const *getConnectionData() { return mConnectionData; }

  protected:
   WeightsPairInterface() {}

   int initialize(char const *name, HyPerCol *hc);

   virtual void setObjectType() override;

   virtual Response::Status
   communicateInitInfo(std::shared_ptr<CommunicateInitInfoMessage const> message) override;

   virtual void createPreWeights(std::string const &weightsName) = 0;

   virtual void createPostWeights(std::string const &weightsName) = 0;

   virtual Response::Status allocateDataStructures() override;

   virtual void allocatePreWeights();

   virtual void allocatePostWeights();

  protected:
   PatchSize *mPatchSize           = nullptr;
   ConnectionData *mConnectionData = nullptr;

   Weights *mPreWeights  = nullptr;
   Weights *mPostWeights = nullptr;
};

} // namespace PV

#endif // WEIGHTSPAIRINTERFACE_HPP_