#pragma once

#include <memory>
#include <istream>
#include <vector>

#include "brain/neural_node.h"
#include "brain/sensor.h"

class NeuralNetwork
{
public:
    NeuralNetwork(std::istream &&networkConfigFile, int numberOfSensors, int numberOfOutputs);

    void setInput(int inputIndex, double activation);
    int findMaxOutputIndex() const;

    int numberOfSensors() const { return _sensors.size(); }
    int numberOfOutputs() const { return _outputs.size(); }
	
private:
    std::vector<std::shared_ptr<NeuralNode>> _nodes;
    std::vector<std::shared_ptr<Sensor>> _sensors;
    std::vector<std::shared_ptr<NeuralNode>> _outputs;
};