mosesdecoder/moses/FF/RuleScope.cpp
2015-05-02 11:45:24 +01:00

109 lines
2.6 KiB
C++

#include "RuleScope.h"
#include "moses/StaticData.h"
#include "moses/Word.h"
using namespace std;
namespace Moses
{
RuleScope::RuleScope(const std::string &line)
:StatelessFeatureFunction(1, line)
,m_sourceSyntax(true)
,m_perScope(false)
,m_futureCostOnly(false)
{
}
bool IsAmbiguous(const Word &word, bool sourceSyntax)
{
const Word &inputDefaultNonTerminal = StaticData::Instance().GetInputDefaultNonTerminal();
return word.IsNonTerminal() && (!sourceSyntax || word == inputDefaultNonTerminal);
}
void RuleScope::EvaluateInIsolation(const Phrase &source
, const TargetPhrase &targetPhrase
, ScoreComponentCollection &scoreBreakdown
, ScoreComponentCollection &estimatedFutureScore) const
{
if (IsGlueRule(source)) {
return;
}
float score = 0;
if (source.GetSize() > 0 && source.Front().IsNonTerminal()) {
++score;
}
if (source.GetSize() > 1 && source.Back().IsNonTerminal()) {
++score;
}
/*
int count = 0;
for (size_t i = 0; i < source.GetSize(); ++i) {
const Word &word = source.GetWord(i);
bool ambiguous = IsAmbiguous(word, m_sourceSyntax);
if (ambiguous) {
++count;
} else {
if (count > 0) {
score += count;
}
count = -1;
}
}
// 1st & last always adjacent to ambiguity
++count;
if (count > 0) {
score += count;
}
*/
if (m_perScope) {
UTIL_THROW_IF2(m_numScoreComponents <= score,
"Insufficient number of score components. Scope=" << score << ". NUmber of score components=" << score);
vector<float> scores(m_numScoreComponents, 0);
scores[score] = 1;
if (m_futureCostOnly) {
estimatedFutureScore.PlusEquals(this, scores);
} else {
scoreBreakdown.PlusEquals(this, scores);
}
} else if (m_futureCostOnly) {
estimatedFutureScore.PlusEquals(this, score);
} else {
scoreBreakdown.PlusEquals(this, score);
}
}
void RuleScope::SetParameter(const std::string& key, const std::string& value)
{
if (key == "source-syntax") {
m_sourceSyntax = Scan<bool>(value);
} else if (key == "per-scope") {
m_perScope = Scan<bool>(value);
} else if ("future-cost-only") {
m_futureCostOnly = Scan<bool>(value);
} else {
StatelessFeatureFunction::SetParameter(key, value);
}
}
bool RuleScope::IsGlueRule(const Phrase &source) const
{
string sourceStr = source.ToString();
if (sourceStr == "<s> " || sourceStr == "X </s> " || sourceStr == "X X ") {
// don't score glue rule
//cerr << "sourceStr=" << sourceStr << endl;
return true;
} else {
return false;
}
}
}