summaryrefslogtreecommitdiff
path: root/extractor/phrase.h
diff options
context:
space:
mode:
Diffstat (limited to 'extractor/phrase.h')
-rw-r--r--extractor/phrase.h12
1 files changed, 12 insertions, 0 deletions
diff --git a/extractor/phrase.h b/extractor/phrase.h
index 5a5124d9..f40a8169 100644
--- a/extractor/phrase.h
+++ b/extractor/phrase.h
@@ -1,6 +1,7 @@
#ifndef _PHRASE_H_
#define _PHRASE_H_
+#include <iostream>
#include <string>
#include <vector>
@@ -20,6 +21,17 @@ class Phrase {
int GetSymbol(int position) const;
+ //TODO(pauldb): Unit test this method.
+ int GetNumSymbols() const;
+
+ //TODO(pauldb): Add unit tests.
+ vector<string> GetWords() const;
+
+ //TODO(pauldb): Add unit tests.
+ int operator<(const Phrase& other) const;
+
+ friend ostream& operator<<(ostream& os, const Phrase& phrase);
+
private:
vector<int> symbols;
vector<int> var_pos;