mosesdecoder/moses/LMList.h
2012-11-12 19:56:18 +00:00

65 lines
1.4 KiB
C++

#ifndef moses_LMList_h
#define moses_LMList_h
#include <list>
#include "LM/Base.h"
namespace Moses
{
class Phrase;
class ScoreColl;
class ScoreComponentCollection;
//! List of language models and function to calc scores from each LM, given a phrase
class LMList
{
protected:
typedef std::list < LanguageModel* > CollType;
CollType m_coll;
public:
typedef CollType::iterator iterator;
typedef CollType::const_iterator const_iterator;
//! iterators
const_iterator begin() const {
return m_coll.begin();
}
const_iterator end() const {
return m_coll.end();
}
size_t size() const {
return m_coll.size();
}
LMList()
{}
void CleanUp();
~LMList();
void CalcScore(const Phrase &phrase, float &retFullScore, float &retNGramScore, float &retOOVScore, ScoreComponentCollection* breakdown) const;
void InitializeBeforeSentenceProcessing() {
std::list<LanguageModel*>::iterator lm_iter;
for (lm_iter = m_coll.begin();
lm_iter != m_coll.end();
++lm_iter) {
(*lm_iter)->InitializeBeforeSentenceProcessing();
}
}
void CleanUpAfterSentenceProcessing(const InputType& source) {
std::list<LanguageModel*>::iterator lm_iter;
for (lm_iter = m_coll.begin();
lm_iter != m_coll.end();
++lm_iter) {
(*lm_iter)->CleanUpAfterSentenceProcessing(source);
}
}
void Add(LanguageModel *lm);
};
}
#endif