connect the nodes (no segfault edition)

1 files changed, 8 insertions, 2 deletions

diff --git a/Circuit.cpp b/Circuit.cpp
index 393060f..7da9384 100644
--- a/Circuit.cpp
+++ b/Circuit.cpp
@@ -25,16 +25,22 @@ void Circuit::new_net(string src, vector<string> dests) {
 	nets.push_back(net);
 
 	for (auto dest : dests) {
-		Node * node = nodes.find(dest)->second;
+		Node * node = find_node(dest);
 		if (node == nullptr) continue; // TODO: exception!
 		node->addInput(net);
 	}
 
-	Node * node = nodes.find(src)->second;
+	Node * node = find_node(src);
 	if (node == nullptr) return; // TODO: exception!
 	node->setOutput(net);
 }
 
+Node * Circuit::find_node(string label) {
+	auto map_index = this->nodes.find(label);
+	if (map_index == nodes.end()) return nullptr;
+	return map_index->second;
+}
+
 Circuit::~Circuit() {
 	for (auto & n : nodes)
 		delete n.second;