Code reorganization with respect to hypergraph output.

1 files changed, 6 insertions, 1 deletions

diff --git a/moses/TranslationTask.cpp b/moses/TranslationTask.cpp
index 3794d35e7..0b425ef23 100644
--- a/moses/TranslationTask.cpp
+++ b/moses/TranslationTask.cpp
@@ -178,7 +178,12 @@ void TranslationTask::Run()
 
   // Output search graph in hypergraph format for Kenneth Heafield's
   // lazy hypergraph decoder; writes to stderr
-  manager->OutputSearchGraphHypergraph();
+  if (StaticData::Instance().GetOutputSearchGraphHypergraph()) 
+    {
+      size_t transId = manager->GetSource().GetTranslationId();
+      string fname = io->GetHypergraphOutputFileName(transId);
+      manager->OutputSearchGraphAsHypergraph(fname, PRECISION);
+    }
 
   additionalReportingTime.stop();