#include "brain/neural_network.h"
#include "brain/neuron.h"

NeuralNetwork::NeuralNetwork(std::istream &&networkConfigFile, int numberOfSensors, int numberOfOutputs)
{
    for (int i=0; i<numberOfSensors; ++i)
    {
	_sensors.push_back(std::make_shared<Sensor>(i));
    }
    for (int i=0; i<numberOfOutputs; ++i)
    {
	_outputs.push_back(std::make_shared<Neuron>(i));
    }
}

void NeuralNetwork::setInput(int inputIndex, double activation)
{
    for (auto sensor : _sensors)
    {
	if (sensor->id() == inputIndex)
	{
	    sensor->setActivation(activation);
	}
    }
}

int NeuralNetwork::findMaxOutputIndex() const
{
    double currentMaxActivation = 0;
    int currentMaxIndex = 0;
    for (auto output : _outputs)
    {
	double activation = output->activation();
	if (activation >= currentMaxActivation)
	{
	    currentMaxActivation = activation;
	    currentMaxIndex = output->id();
	}
    }
    return currentMaxIndex;
}