From 63b30ed9c8510da8c8e2f6a456576424fddacc0e Mon Sep 17 00:00:00 2001
From: Paul Baltescu <pauldb89@gmail.com>
Date: Thu, 14 Feb 2013 23:17:15 +0000
Subject: Working version of the grammar extractor.

---
 extractor/grammar_extractor.h | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'extractor/grammar_extractor.h')

diff --git a/extractor/grammar_extractor.h b/extractor/grammar_extractor.h
index 243f33cf..5f87faa7 100644
--- a/extractor/grammar_extractor.h
+++ b/extractor/grammar_extractor.h
@@ -32,13 +32,19 @@ class GrammarExtractor {
       bool use_baeza_yates,
       bool require_tight_phrases);
 
+  // For testing only.
+  GrammarExtractor(shared_ptr<Vocabulary> vocabulary,
+                   shared_ptr<HieroCachingRuleFactory> rule_factory);
+
   Grammar GetGrammar(const string& sentence);
 
  private:
+  vector<string> TokenizeSentence(const string& sentence);
+
   vector<int> AnnotateWords(const vector<string>& words);
 
   shared_ptr<Vocabulary> vocabulary;
-  HieroCachingRuleFactory rule_factory;
+  shared_ptr<HieroCachingRuleFactory> rule_factory;
 };
 
 #endif
-- 
cgit v1.2.3