mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2025-01-08 04:27:53 +03:00
89 lines
2.0 KiB
C++
89 lines
2.0 KiB
C++
/*
|
|
* TargetPhrase.h
|
|
*
|
|
* Created on: 26 Apr 2016
|
|
* Author: hieu
|
|
*/
|
|
|
|
#pragma once
|
|
#include "PhraseImplTemplate.h"
|
|
#include "System.h"
|
|
#include "Scores.h"
|
|
|
|
namespace Moses2
|
|
{
|
|
|
|
template<typename WORD>
|
|
class TargetPhrase: public PhraseImplTemplate<WORD>
|
|
{
|
|
friend std::ostream& operator<<(std::ostream &, const TargetPhrase &);
|
|
|
|
public:
|
|
const PhraseTable &pt;
|
|
mutable void **ffData;
|
|
SCORE *scoreProperties;
|
|
|
|
TargetPhrase(MemPool &pool, const PhraseTable &pt, const System &system, size_t size)
|
|
: PhraseImplTemplate<WORD>(pool, size)
|
|
, pt(pt)
|
|
, scoreProperties(NULL)
|
|
{
|
|
m_scores = new (pool.Allocate<Scores>()) Scores(system, pool,
|
|
system.featureFunctions.GetNumScores());
|
|
}
|
|
|
|
|
|
Scores &GetScores()
|
|
{ return *m_scores; }
|
|
|
|
const Scores &GetScores() const
|
|
{ return *m_scores; }
|
|
|
|
SCORE GetFutureScore() const
|
|
{ return m_scores->GetTotalScore() + m_estimatedScore; }
|
|
|
|
void SetEstimatedScore(const SCORE &value)
|
|
{ m_estimatedScore = value; }
|
|
|
|
SCORE *GetScoresProperty(int propertyInd) const
|
|
{ return scoreProperties ? scoreProperties + propertyInd : NULL; }
|
|
|
|
protected:
|
|
Scores *m_scores;
|
|
SCORE m_estimatedScore;
|
|
};
|
|
|
|
///////////////////////////////////////////////////////////////////////
|
|
inline std::ostream& operator<<(std::ostream &out, const TargetPhrase<Moses2::Word> &obj)
|
|
{
|
|
out << (const Phrase<Moses2::Word> &) obj << " SCORES:" << obj.GetScores();
|
|
return out;
|
|
}
|
|
/*
|
|
template<typename WORD>
|
|
std::ostream& operator<<(std::ostream &out, const TargetPhrase<WORD> &obj)
|
|
{
|
|
out << (const Phrase<WORD> &) obj << " SCORES:" << obj.GetScores();
|
|
return out;
|
|
}
|
|
*/
|
|
|
|
///////////////////////////////////////////////////////////////////////
|
|
template<typename WORD>
|
|
struct CompareFutureScore
|
|
{
|
|
bool operator()(const TargetPhrase<WORD> *a, const TargetPhrase<WORD> *b) const
|
|
{
|
|
return a->GetFutureScore() > b->GetFutureScore();
|
|
}
|
|
|
|
bool operator()(const TargetPhrase<WORD> &a, const TargetPhrase<WORD> &b) const
|
|
{
|
|
return a.GetFutureScore() > b.GetFutureScore();
|
|
}
|
|
};
|
|
|
|
|
|
} /* namespace Moses2a */
|
|
|