FilenameParsingGroundTruthLayer.hpp

/*
 * FilenameParsingGroundTruthLayer.hpp
 *
 *  Created on: Nov 10, 2014
 *      Author: wchavez
 */
#ifndef FILENAMEPARSINGGROUNDTRUTHLAYER_HPP_
#define FILENAMEPARSINGGROUNDTRUTHLAYER_HPP_

#include "HyPerLayer.hpp"
#include "InputLayer.hpp"
#include <string>

namespace PV {

class FilenameParsingGroundTruthLayer : public HyPerLayer {

  public:
   FilenameParsingGroundTruthLayer(const char *name, HyPerCol *hc);
   virtual ~FilenameParsingGroundTruthLayer();
   virtual Response::Status
   communicateInitInfo(std::shared_ptr<CommunicateInitInfoMessage const> message) override;
   virtual Response::Status updateState(double timef, double dt) override;
   virtual bool needUpdate(double time, double dt) override;
   int ioParamsFillGroup(enum ParamsIOFlag ioFlag) override;

  private:
   std::vector<std::string> mClasses;
   char *mInputLayerName    = nullptr;
   char *mClassListFileName = nullptr;
   InputLayer *mInputLayer  = nullptr;
   float mGtClassTrueValue  = 1.0f;
   float mGtClassFalseValue = 0.0f;

  protected:
   virtual Response::Status registerData(Checkpointer *checkpointer) override;

   virtual void ioParam_classList(enum ParamsIOFlag ioFlag);

   virtual void ioParam_inputLayerName(enum ParamsIOFlag ioFlag);

   virtual void ioParam_gtClassTrueValue(enum ParamsIOFlag ioFlag);

   virtual void ioParam_gtClassFalseValue(enum ParamsIOFlag ioFlag);

};
}

#endif // FILENAMEPARSINGGROUNDTRUTHLAYER_HPP_