ISTALayer.hpp

/*
 * ISTALayer.hpp
 *
 *  Created on: Jan 24, 2013
 *      Author: garkenyon
 */

#ifndef ISTALAYER_HPP_
#define ISTALAYER_HPP_
// TODO: Take care of code duplication between ISTALayer and HyPerLCALayer.

#include "ANNLayer.hpp"
#include "probes/AdaptiveTimeScaleProbe.hpp"

namespace PV {

class ISTALayer : public PV::ANNLayer {
  public:
   ISTALayer(const char *name, HyPerCol *hc);
   virtual ~ISTALayer();
   virtual double getDeltaUpdateTime() override;
   virtual int requireChannel(int channelNeeded, int *numChannelsResult) override;

  protected:
   ISTALayer();
   int initialize(const char *name, HyPerCol *hc);
   virtual Response::Status allocateDataStructures() override;
   virtual int ioParamsFillGroup(enum ParamsIOFlag ioFlag) override;

   virtual void ioParam_timeConstantTau(enum ParamsIOFlag ioFlag);
   virtual void ioParam_selfInteract(enum ParamsIOFlag ioFlag);

   virtual void ioParam_adaptiveTimeScaleProbe(enum ParamsIOFlag ioFlag);
   virtual Response::Status updateState(double time, double dt) override;

#ifdef PV_USE_CUDA
   virtual Response::Status updateStateGpu(double time, double dt) override;
#endif

   virtual float getChannelTimeConst(enum ChannelType channel_type) override {
      return timeConstantTau;
   }

#ifdef PV_USE_CUDA
   virtual int allocateUpdateKernel() override;
#endif

   double *deltaTimes(); // TODO: make const-correct
   // Better name?  getDeltaTimes isn't good because it sounds like it's just the getter-method.

  private:
   int initialize_base();
#ifdef PV_USE_CUDA
   PVCuda::CudaBuffer *d_dtAdapt;
#endif

   // Data members
  protected:
   float timeConstantTau;
   bool selfInteract;
   char *mAdaptiveTimeScaleProbeName               = nullptr;
   AdaptiveTimeScaleProbe *mAdaptiveTimeScaleProbe = nullptr;
   std::vector<double> mDeltaTimes;
}; // class ISTALayer

} /* namespace PV */
#endif /* ISTALAYER_HPP_ */