From 0172721855098ca02b207231a654dffa5e4eb1c9 Mon Sep 17 00:00:00 2001
From: redpony <redpony@ec762483-ff6d-05da-a07a-a48fb63a330f>
Date: Tue, 22 Jun 2010 05:12:27 +0000
Subject: initial checkin

git-svn-id: https://ws10smt.googlecode.com/svn/trunk@2 ec762483-ff6d-05da-a07a-a48fb63a330f
---
 decoder/forest_writer.cc | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 decoder/forest_writer.cc

(limited to 'decoder/forest_writer.cc')

diff --git a/decoder/forest_writer.cc b/decoder/forest_writer.cc
new file mode 100644
index 00000000..a9117d18
--- /dev/null
+++ b/decoder/forest_writer.cc
@@ -0,0 +1,23 @@
+#include "forest_writer.h"
+
+#include <iostream>
+
+#include <boost/lexical_cast.hpp>
+
+#include "filelib.h"
+#include "hg_io.h"
+#include "hg.h"
+
+using namespace std;
+
+ForestWriter::ForestWriter(const std::string& path, int num) :
+  fname_(path + '/' + boost::lexical_cast<string>(num) + ".json.gz"), used_(false) {}
+
+bool ForestWriter::Write(const Hypergraph& forest, bool minimal_rules) {
+  assert(!used_);
+  used_ = true;
+  cerr << "  Writing forest to " << fname_ << endl;
+  WriteFile wf(fname_);
+  return HypergraphIO::WriteToJSON(forest, minimal_rules, wf.stream());
+}
+
-- 
cgit v1.2.3