mosesdecoder/moses2/EstimatedScores.cpp
2017-02-01 00:10:16 +00:00

118 lines
3.7 KiB
C++

// $Id$
// vim:tabstop=2
/***********************************************************************
Moses - factored phrase-based language decoder
Copyright (C) 2006 University of Edinburgh
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
***********************************************************************/
#include <string>
#include <iostream>
#include "EstimatedScores.h"
using namespace std;
namespace Moses2
{
/**
* Calculate future score estimate for a given coverage bitmap
*
* /param bitmap coverage bitmap
*/
float EstimatedScores::CalcEstimatedScore(Bitmap const &bitmap) const
{
const size_t notInGap = numeric_limits<size_t>::max();
size_t startGap = notInGap;
float estimatedScore = 0.0f;
for (size_t currPos = 0; currPos < bitmap.GetSize(); currPos++) {
// start of a new gap?
if (bitmap.GetValue(currPos) == false && startGap == notInGap) {
startGap = currPos;
}
// end of a gap?
else if (bitmap.GetValue(currPos) == true && startGap != notInGap) {
estimatedScore += GetValue(startGap, currPos - 1);
startGap = notInGap;
}
}
// coverage ending with gap?
if (startGap != notInGap) {
estimatedScore += GetValue(startGap, bitmap.GetSize() - 1);
}
return estimatedScore;
}
/**
* Calculare future score estimate for a given coverage bitmap
* and an additional span that is also covered. This function is used
* to compute future score estimates for hypotheses that we may want
* build, but first want to check.
*
* Note: this function is implemented a bit more complex than
* the basic one (w/o additional phrase) for speed reasons,
* which is probably overkill.
*
* /param bitmap coverage bitmap
* /param startPos start of the span that is added to the coverage
* /param endPos end of the span that is added to the coverage
*/
float EstimatedScores::CalcEstimatedScore(Bitmap const &bitmap, size_t startPos,
size_t endPos) const
{
const size_t notInGap = numeric_limits<size_t>::max();
float estimatedScore = 0.0f;
size_t startGap = bitmap.GetFirstGapPos();
if (startGap == NOT_FOUND) return estimatedScore; // everything filled
// start loop at first gap
size_t startLoop = startGap + 1;
if (startPos == startGap) { // unless covered by phrase
startGap = notInGap;
startLoop = endPos + 1; // -> postpone start
}
size_t lastCovered = bitmap.GetLastPos();
if (endPos > lastCovered || lastCovered == NOT_FOUND) lastCovered = endPos;
for (size_t currPos = startLoop; currPos <= lastCovered; currPos++) {
// start of a new gap?
if (startGap == notInGap && bitmap.GetValue(currPos) == false
&& (currPos < startPos || currPos > endPos)) {
startGap = currPos;
}
// end of a gap?
else if (startGap != notInGap
&& (bitmap.GetValue(currPos) == true
|| (startPos <= currPos && currPos <= endPos))) {
estimatedScore += GetValue(startGap, currPos - 1);
startGap = notInGap;
}
}
// coverage ending with gap?
if (lastCovered != bitmap.GetSize() - 1) {
estimatedScore += GetValue(lastCovered + 1, bitmap.GetSize() - 1);
}
return estimatedScore;
}
}