1 files changed, 34 insertions, 0 deletions
diff --git a/extractor/grammar.h b/extractor/grammar.h
new file mode 100644
index 00000000..fed41b16
--- /dev/null
+++ b/extractor/grammar.h
@@ -0,0 +1,34 @@
+#ifndef _GRAMMAR_H_
+#define _GRAMMAR_H_
+
+#include <iostream>
+#include <string>
+#include <vector>
+
+using namespace std;
+
+namespace extractor {
+
+class Rule;
+
+/**
+ * Grammar class wrapping the set of rules to be extracted.
+ */
+class Grammar {
+ public:
+  Grammar(const vector<Rule>& rules, const vector<string>& feature_names);
+
+  vector<Rule> GetRules() const;
+
+  vector<string> GetFeatureNames() const;
+
+  friend ostream& operator<<(ostream& os, const Grammar& grammar);
+
+ private:
+  vector<Rule> rules;
+  vector<string> feature_names;
+};
+
+} // namespace extractor
+
+#endif