From 2e6ecf423c8228ac8de4badf4fc2d037a876b7ff Mon Sep 17 00:00:00 2001
From: Justin Worthe <justin.worthe@gmail.com>
Date: Sat, 1 Aug 2015 22:50:00 +0200
Subject: Reading network from file

---
 include/brain/neural_network.h | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

(limited to 'include/brain/neural_network.h')

diff --git a/include/brain/neural_network.h b/include/brain/neural_network.h
index 308cb56..b2c441f 100644
--- a/include/brain/neural_network.h
+++ b/include/brain/neural_network.h
@@ -3,9 +3,12 @@
 #include <memory>
 #include <istream>
 #include <vector>
+#include <string>
 
 #include "brain/neural_node.h"
 #include "brain/sensor.h"
+#include "brain/bias_node.h"
+#include "brain/neuron.h"
 
 class NeuralNetwork
 {
@@ -19,8 +22,15 @@ public:
     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;
+    std::shared_ptr<BiasNode> _biasNode;
+    std::vector<std::shared_ptr<Neuron>> _hiddenNodes;
+    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(int id);
+    std::shared_ptr<Neuron> findOrAddNeuron(int id);
 };
 
-- 
cgit v1.2.3