TranslationTaskChart --> TranslationTask

1 files changed, 9 insertions, 0 deletions

diff --git a/moses/TranslationTask.h b/moses/TranslationTask.h
index f0f237c35..698c7009c 100644
--- a/moses/TranslationTask.h
+++ b/moses/TranslationTask.h
@@ -4,6 +4,8 @@
 #include "moses/ThreadPool.h"
 #include "moses/Manager.h"
 #include "moses/HypergraphOutput.h"
+#include "moses/Manager.h"
+#include "moses/ChartManager.h"
 
 namespace Moses
 {
@@ -26,6 +28,9 @@ public:
                   bool outputSearchGraphSLF,
                   boost::shared_ptr<Moses::HypergraphOutput<Moses::Manager> > hypergraphOutput);
 
+  TranslationTask(Moses::InputType *source, IOWrapper &ioWrapper,
+    boost::shared_ptr<Moses::HypergraphOutput<Moses::ChartManager> > hypergraphOutputChart);
+
   ~TranslationTask();
 
   /** Translate one sentence
@@ -34,12 +39,16 @@ public:
 
 
 private:
+  int m_pbOrChart; // 1=pb. 2=chart
   Moses::InputType* m_source;
   Moses::IOWrapper &m_ioWrapper;
 
   bool m_outputSearchGraphSLF;
   boost::shared_ptr<Moses::HypergraphOutput<Moses::Manager> > m_hypergraphOutput;
+  boost::shared_ptr<Moses::HypergraphOutput<Moses::ChartManager> > m_hypergraphOutputChart;
 
+  void RunPb();
+  void RunChart();
 
 };