Moses server now accepts context weights as translation request parameters.

1 files changed, 28 insertions, 0 deletions

diff --git a/moses/ContextScope.h b/moses/ContextScope.h
index e1173cbdc..fd869ef0e 100644
--- a/moses/ContextScope.h
+++ b/moses/ContextScope.h
@@ -12,6 +12,15 @@
 #include <boost/foreach.hpp>
 #endif
 
+// for some reason, the xmlrpc_c headers must be included AFTER the 
+// boost thread-related ones ...
+#ifdef HAVE_XMLRPC_C
+#include <xmlrpc-c/base.hpp>
+#include <xmlrpc-c/registry.hpp>
+#include <xmlrpc-c/server_abyss.hpp>
+#endif
+
+
 #include <map>
 #include <boost/shared_ptr.hpp>
 // #include "thread_safe_container.h"
@@ -29,6 +38,7 @@ protected:
 #ifdef WITH_THREADS
   mutable boost::shared_mutex m_lock;
 #endif
+  SPTR<std::map<std::string,float> > m_context_weights;
 public:
   // class write_access
   // {
@@ -99,6 +109,24 @@ public:
     m_scratchpad = other.m_scratchpad;
   }
 
+#ifdef HAVE_XMLRPC_C
+  SPTR<std::map<std::string,float> >
+  GetContextWeights(xmlrpc_c::value const* spec = NULL)
+  {
+    if (spec && m_context_weights == NULL)
+      { 
+	boost::unique_lock<boost::shared_mutex> lock(m_lock);
+	m_context_weights.reset(new std::map<std::string, float>);
+
+	typedef std::map<std::string,xmlrpc_c::value> tmap;
+	tmap const tmp = static_cast<tmap>(xmlrpc_c::value_struct(*spec));
+	for(tmap::const_iterator m = tmp.begin(); m != tmp.end(); ++m)
+	  (*m_context_weights)[m->first] = xmlrpc_c::value_double(m->second);
+      }
+    return m_context_weights;
+  }
+#endif
+
 };
 
 };