|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#include "InputType.h" |
|
|
#include "System.h" |
|
|
#include <iostream> |
|
|
|
|
|
using namespace std; |
|
|
|
|
|
namespace Moses2 |
|
|
{ |
|
|
|
|
|
InputType::XMLOption::XMLOption(MemPool &pool, const std::string &nodeName, size_t vStartPos) |
|
|
:startPos(vStartPos) |
|
|
,prob(0) |
|
|
,m_entity(NULL) |
|
|
{ |
|
|
m_nodeName = pool.Allocate<char>(nodeName.size() + 1); |
|
|
strcpy(m_nodeName, nodeName.c_str()); |
|
|
} |
|
|
|
|
|
void InputType::XMLOption::SetTranslation(MemPool &pool, const std::string &val) |
|
|
{ |
|
|
m_translation = pool.Allocate<char>(val.size() + 1); |
|
|
strcpy(m_translation, val.c_str()); |
|
|
} |
|
|
|
|
|
void InputType::XMLOption::SetEntity(MemPool &pool, const std::string &val) |
|
|
{ |
|
|
m_entity = pool.Allocate<char>(val.size() + 1); |
|
|
strcpy(m_entity, val.c_str()); |
|
|
} |
|
|
|
|
|
std::string InputType::XMLOption::Debug(const System &system) const |
|
|
{ |
|
|
std::stringstream out; |
|
|
out << "[" << startPos << "," << phraseSize << "]=" |
|
|
<< m_nodeName << "," |
|
|
<< m_translation << "," |
|
|
<< prob; |
|
|
if (m_entity) { |
|
|
out << "," << m_entity; |
|
|
} |
|
|
return out.str(); |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
InputType::InputType(MemPool &pool) |
|
|
:m_reorderingConstraint(pool) |
|
|
,m_xmlOptions(pool) |
|
|
,m_xmlCoverageMap(pool) |
|
|
{ |
|
|
} |
|
|
|
|
|
InputType::~InputType() |
|
|
{ |
|
|
|
|
|
} |
|
|
|
|
|
void InputType::Init(const System &system, size_t size, int max_distortion) |
|
|
{ |
|
|
m_reorderingConstraint.InitializeWalls(size, max_distortion); |
|
|
|
|
|
if (system.options.input.xml_policy != XmlPassThrough) { |
|
|
m_xmlCoverageMap.assign(size, false); |
|
|
} |
|
|
} |
|
|
|
|
|
void InputType::AddXMLOption(const System &system, const XMLOption *xmlOption) |
|
|
{ |
|
|
m_xmlOptions.push_back(xmlOption); |
|
|
|
|
|
if (system.options.input.xml_policy != XmlPassThrough) { |
|
|
for(size_t j = xmlOption->startPos; j < xmlOption->startPos + xmlOption->phraseSize; ++j) { |
|
|
m_xmlCoverageMap[j]=true; |
|
|
} |
|
|
} |
|
|
} |
|
|
|
|
|
bool InputType::XmlOverlap(size_t startPos, size_t endPos) const |
|
|
{ |
|
|
for (size_t pos = startPos; pos <= endPos ; pos++) { |
|
|
if (pos < m_xmlCoverageMap.size() && m_xmlCoverageMap[pos]) { |
|
|
return true; |
|
|
} |
|
|
} |
|
|
return false; |
|
|
} |
|
|
|
|
|
std::string InputType::Debug(const System &system) const |
|
|
{ |
|
|
cerr << "InputType::Debug" << endl; |
|
|
return ""; |
|
|
} |
|
|
|
|
|
} |
|
|
|
