mosesdecoder/moses/FF/PhraseBoundaryFeature.h

71 lines
1.8 KiB
C
Raw Normal View History

#ifndef moses_PhraseBoundaryFeature_h
#define moses_PhraseBoundaryFeature_h
#include <stdexcept>
#include <sstream>
#include <string>
2013-05-24 22:11:15 +04:00
#include "StatefulFeatureFunction.h"
2013-05-24 21:02:49 +04:00
#include "moses/FF/FFState.h"
#include "moses/Word.h"
namespace Moses
{
2013-05-29 21:16:15 +04:00
class PhraseBoundaryState : public FFState
{
public:
PhraseBoundaryState(const Word* sourceWord, const Word* targetWord) :
2013-05-29 21:16:15 +04:00
m_sourceWord(sourceWord), m_targetWord(targetWord) {}
const Word* GetSourceWord() const {
return m_sourceWord;
}
const Word* GetTargetWord() const {
return m_targetWord;
}
2015-10-14 17:21:37 +03:00
virtual size_t hash() const;
virtual bool operator==(const FFState& other) const;
private:
const Word* m_sourceWord;
const Word* m_targetWord;
};
/**
* Concatenations of factors on boundaries of phrases.
**/
2013-05-29 21:16:15 +04:00
class PhraseBoundaryFeature : public StatefulFeatureFunction
{
public:
PhraseBoundaryFeature(const std::string &line);
bool IsUseable(const FactorMask &mask) const;
virtual const FFState* EmptyHypothesisState(const InputType &) const;
virtual FFState* EvaluateWhenApplied(const Hypothesis& cur_hypo, const FFState* prev_state,
2015-01-14 14:07:42 +03:00
ScoreComponentCollection* accumulator) const;
virtual FFState* EvaluateWhenApplied( const ChartHypothesis& /* cur_hypo */,
2015-01-14 14:07:42 +03:00
int /* featureID */,
ScoreComponentCollection* ) const {
throw std::logic_error("PhraseBoundaryState not supported in chart decoder, yet");
}
void SetParameter(const std::string& key, const std::string& value);
private:
void AddFeatures(
2013-05-29 21:16:15 +04:00
const Word* leftWord, const Word* rightWord, const FactorList& factors,
const std::string& side, ScoreComponentCollection* scores) const ;
FactorList m_sourceFactors;
FactorList m_targetFactors;
};
}
#endif