summaryrefslogtreecommitdiff
path: root/2015-spacebot/include/brain/neural_network.h
blob: 7fcf5f4ff6c862d5f3f040a848818119d22c9b2f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#pragma once

#include <memory>
#include <istream>
#include <vector>
#include <map>
#include <string>

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

class NeuralNetwork
{
public:
    NeuralNetwork(std::istream &&networkConfigFile, unsigned int numberOfSensors, unsigned int numberOfOutputs);
    NeuralNetwork(std::istream &&networkConfigFile, std::vector<bool> sensorInitialValues, unsigned int numberOfOutputs);
    
    void setInput(unsigned int inputIndex, double activation);
    unsigned int findMaxOutputIndex() const;

    unsigned int numberOfSensors() const { return _sensors.size(); }
    unsigned int numberOfOutputs() const { return _outputs.size(); }
    unsigned int numberOfNeurons() const { return _neurons.size(); }

    bool linkExists(std::string srcIdentifier, std::string destIdentifier, double weight) const;
	
private:
    std::vector<std::shared_ptr<Sensor>> _sensors;
    std::shared_ptr<BiasNode> _biasNode;
    std::vector<std::shared_ptr<Neuron>> _neurons;
    std::vector<std::shared_ptr<Neuron>> _outputs;

    void parseFile(std::istream &&file);
    
    void addLink(std::shared_ptr<NeuralNode> source, std::shared_ptr<Neuron> destination, double weight);
    std::shared_ptr<Sensor> findOrAddSensor(unsigned int id);
    std::shared_ptr<Neuron> findOrAddNeuron(unsigned int id);
};