#ifndef NLAYER_HPP
#define NLAYER_HPP 1

#include <iostream>
#include <vector>
#include "neuron.hpp"
#include "util.hpp"

using namespace std;

class NLayer
{
private:

    vector<Neuron*> neurons;

    // These layers are simple enough that we just store all necessary training
    // information in the layer
    vector<double> signals;
    vector<double> inputs;
    vector<double> outputs;

    // Learning rate, number of neurons, and size of input
    double eta;
    int layerSize;
    int inputVectorSize;

    // Too lazy to write copy constructor/assignment operator

    NLayer& operator=(const NLayer rhs);
    NLayer(NLayer& l);

public:

    // Constructor/Destructor
    NLayer(int ivs, int s);
    ~NLayer();
    int size();

    // Training/Operation functions
    vector<double> calcLayerOutput(vector<double> inputs);
    vector<double> calcSignals(vector<double> targets);
    vector<double> calcSignals(vector<double> err, vector< vector<double> > w);
    void updateWeights(double eta);

    // Returns input weights in terms of output weights from the previous layer
    vector< vector<double> > getInputWeights();

    ostream& print(ostream& out);
};

ostream& operator<<(ostream& out, NLayer& n);

#endif // NLAYER_HPP