Working version of the grammar extractor.

1 files changed, 11 insertions, 3 deletions

diff --git a/extractor/matchings_trie.cc b/extractor/matchings_trie.cc
index 851d4596..921ec582 100644
--- a/extractor/matchings_trie.cc
+++ b/extractor/matchings_trie.cc
@@ -1,11 +1,19 @@
 #include "matchings_trie.h"
 
 void MatchingsTrie::Reset() {
-  // TODO(pauldb): This is probably memory leaking because of the suffix links.
-  // Check if it's true and free the memory properly.
-  root.reset(new TrieNode());
+  ResetTree(root);
+  root = make_shared<TrieNode>();
 }
 
 shared_ptr<TrieNode> MatchingsTrie::GetRoot() const {
   return root;
 }
+
+void MatchingsTrie::ResetTree(shared_ptr<TrieNode> root) {
+  if (root != NULL) {
+    for (auto child: root->children) {
+      ResetTree(child.second);
+    }
+    root.reset();
+  }
+}