TargetPhrase: can store arbitrary data

1 files changed, 20 insertions, 0 deletions

diff --git a/moses/TargetPhrase.h b/moses/TargetPhrase.h
index db7da97c5..04211e66b 100644
--- a/moses/TargetPhrase.h
+++ b/moses/TargetPhrase.h
@@ -33,6 +33,7 @@ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 #include "util/string_piece.hh"
 
 #include <boost/shared_ptr.hpp>
+#include <boost/unordered_map.hpp>
 
 #ifdef HAVE_PROTOBUF
 #include "rule.pb.h"
@@ -65,6 +66,8 @@ private:
 
   const PhraseDictionary *m_container;
 
+  mutable boost::unordered_map<const std::string, boost::shared_ptr<void> > m_data;
+
 public:
   TargetPhrase(const PhraseDictionary *pt = NULL);
   TargetPhrase(std::string out_string, const PhraseDictionary *pt = NULL);
@@ -166,6 +169,23 @@ public:
     return m_container;
   }
 
+  bool SetData(const std::string& key, boost::shared_ptr<void> value) const {
+    std::pair< boost::unordered_map<const std::string, boost::shared_ptr<void> >::iterator, bool > inserted =
+      m_data.insert( std::pair<const std::string, boost::shared_ptr<void> >(key,value) );
+    if (!inserted.second) {
+      return false;
+    }
+    return true;
+  }
+
+  boost::shared_ptr<void> GetData(const std::string& key) const {
+    boost::unordered_map<const std::string, boost::shared_ptr<void> >::const_iterator found = m_data.find(key);
+    if (found == m_data.end()) {
+      return boost::shared_ptr<void>();
+    }
+    return found->second;
+  }
+
   // To be set by the FF that needs it, by default the rule source = NULL
   // make a copy of the source side of the rule
   void SetRuleSource(const Phrase &ruleSource) const;