PetaVision  Alpha
MomentumConn.hpp
1 /* MomentumConn.cpp
2  *
3  * Created on: Feburary 27, 2014
4  * Author: slundquist
5  */
6 
7 #ifndef MOMENTUMCONN_HPP_
8 #define MOMENTUMCONN_HPP_
9 
10 #include "connections/HyPerConn.hpp"
11 
12 namespace PV {
13 
14 class HyPerCol;
15 
16 class MomentumConn : public HyPerConn {
17  public:
18  MomentumConn(char const *name, HyPerCol *hc);
19 
20  virtual ~MomentumConn();
21 
22  char const *getMomentumMethod() const;
23  float getTimeConstantTau() const;
24 
25  protected:
26  MomentumConn();
27 
28  int initialize(char const *name, HyPerCol *hc);
29 
30  virtual BaseWeightUpdater *createWeightUpdater() override;
31 }; // class MomentumConn
32 
33 } // namespace PV
34 
35 #endif // MOMENTUMCONN_HPP_