extract-ghkm: add --STSG option

1 files changed, 41 insertions, 0 deletions

diff --git a/phrase-extract/extract-ghkm/StsgRuleWriter.h b/phrase-extract/extract-ghkm/StsgRuleWriter.h
new file mode 100644
index 000000000..efba44d2c
--- /dev/null
+++ b/phrase-extract/extract-ghkm/StsgRuleWriter.h
@@ -0,0 +1,41 @@
+#pragma once
+#ifndef EXTRACT_GHKM_STSG_RULE_WRITER_H_
+#define EXTRACT_GHKM_STSG_RULE_WRITER_H_
+
+#include "Subgraph.h"
+
+#include <ostream>
+
+namespace Moses
+{
+namespace GHKM
+{
+
+struct Options;
+class StsgRule;
+class Symbol;
+
+class StsgRuleWriter
+{
+public:
+  StsgRuleWriter(std::ostream &fwd, std::ostream &inv, const Options &options)
+    : m_fwd(fwd)
+    , m_inv(inv)
+    , m_options(options) {}
+
+  void Write(const StsgRule &rule);
+
+private:
+  // Disallow copying
+  StsgRuleWriter(const StsgRuleWriter &);
+  StsgRuleWriter &operator=(const StsgRuleWriter &);
+
+  std::ostream &m_fwd;
+  std::ostream &m_inv;
+  const Options &m_options;
+};
+
+}  // namespace GHKM
+}  // namespace Moses
+
+#endif