ConnectionData.hpp

/*
 * ConnectionData.hpp
 *
 *  Created on: Nov 17, 2017
 *      Author: pschultz
 */

#ifndef CONNECTIONDATA_HPP_
#define CONNECTIONDATA_HPP_

#include "columns/BaseObject.hpp"
#include "layers/HyPerLayer.hpp"

namespace PV {

class ConnectionData : public BaseObject {
  protected:
   virtual void ioParam_preLayerName(enum ParamsIOFlag ioFlag);

   virtual void ioParam_postLayerName(enum ParamsIOFlag ioFlag); // end of ConnectionData parameters

  public:
   ConnectionData(char const *name, HyPerCol *hc);
   virtual ~ConnectionData();

   char const *getPreLayerName() const { return mPreLayerName; }

   char const *getPostLayerName() const { return mPostLayerName; }

   HyPerLayer *getPre() { return mPre; }

   HyPerLayer *getPost() { return mPost; }

  protected:
   ConnectionData();

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

   virtual void setObjectType() override;

   virtual int ioParamsFillGroup(enum ParamsIOFlag ioFlag) override;

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

   static void inferPreAndPostFromConnName(
         const char *name,
         int rank,
         std::string &preLayerNameString,
         std::string &postLayerNameString);

  protected:
   char *mPreLayerName  = nullptr;
   char *mPostLayerName = nullptr;
   HyPerLayer *mPre     = nullptr;
   HyPerLayer *mPost    = nullptr;

}; // class ConnectionData

} // namespace PV

#endif // CONNECTIONDATA_HPP_