Layer.h

Go to the documentation of this file.

#ifndef _LAYER_H
#define _LAYER_H

#include "Neuron.h"
#include "Link.h"

#include <vector>

namespace annie
{
class Layer
{
protected:
    int _label;

    uint _size;

    std::vector<Neuron *> _neurons;

public:
    static const int MAX_LAYER_SIZE;

    Layer(int label);

    virtual ~Layer();

    virtual int getLabel() const;

    virtual uint getSize() const;

    uint size() const { return getSize(); }

    virtual void addNeuron(Neuron *nrn);

    virtual Neuron& getNeuron(uint i);
    virtual const Neuron& getNeuron(uint i) const;

    Neuron& operator[](uint i)  { return getNeuron(i); }
    const Neuron& operator[](uint i)    const { return getNeuron(i); }

    virtual Vector getActivation();

    virtual Vector getOutput();

    virtual const char *getClassName();
};

template <class T>
class TLayer : public Layer {
  public:
    TLayer(int label) : Layer(label) {}
    T& operator[](uint i)   { return (T &) getNeuron(i); }
    const T& operator[](uint i) const { return (const T &) getNeuron(i); }
};

}; //namespace annie
#endif // define _LAYER_H

---