Instances of InputType (and derived classes) now know which TranslationTask (if any) created them.

This is a first step towards providing phrase tables etc. access to context information etc.
associated with specific translation tasks.

1 files changed, 3 insertions, 2 deletions

diff --git a/moses/TreeInput.h b/moses/TreeInput.h
index d65f22ddf..d9ea2d0ba 100644
--- a/moses/TreeInput.h
+++ b/moses/TreeInput.h
@@ -1,3 +1,4 @@
+// -*- c++ -*-
 #ifndef moses_TreeInput_h
 #define moses_TreeInput_h
 
@@ -7,6 +8,7 @@
 
 namespace Moses
 {
+  class TranslationTask;
 //! @todo what is this?
 class XMLParseOutput
 {
@@ -44,8 +46,7 @@ protected:
   bool ProcessAndStripXMLTags(std::string &line, std::vector<XMLParseOutput> &sourceLabels, std::vector<XmlOption*> &res);
 
 public:
-  TreeInput() {
-  }
+  TreeInput(TranslationTask const* ttask) : Sentence(ttask) { }
 
   InputTypeEnum GetType() const {
     return TreeInputType;