mosesdecoder/moses/src/ScoreComponentCollection.cpp

// $Id$

#include "ScoreComponentCollection.h"
#include "StaticData.h"

namespace Moses
{
ScoreComponentCollection::ScoreComponentCollection()
  : m_scores(StaticData::Instance().GetTotalScoreComponents(), 0.0f)
  , m_sim(&StaticData::Instance().GetScoreIndexManager())
{}

float ScoreComponentCollection::GetWeightedScore() const
{
	float ret = InnerProduct(StaticData::Instance().GetAllWeights());
	return ret;
}
		
void ScoreComponentCollection::ZeroAllLM(const LMList& lmList)
{
	
	for (size_t ind = lmList.GetMinIndex(); ind <= lmList.GetMaxIndex(); ++ind)
	{
		m_scores[m_sim->GetFeatureName(ind)] = 0;
	}
}

void ScoreComponentCollection::PlusEqualsAllLM(const LMList& lmList, const ScoreComponentCollection& rhs)
{
	
	for (size_t ind = lmList.GetMinIndex(); ind <= lmList.GetMaxIndex(); ++ind)
	{
		m_scores[m_sim->GetFeatureName(ind)] += rhs.m_scores[m_sim->GetFeatureName(ind)];
	}
	
}

void ScoreComponentCollection::PlusEquals(const ScoreProducer* sp, const std::vector<float>& scores)
{
	assert(scores.size() == sp->GetNumScoreComponents());
	size_t i = m_sim->GetBeginIndex(sp->GetScoreBookkeepingID());
	for (std::vector<float>::const_iterator vi = scores.begin();
			 vi != scores.end(); ++vi)
	{
		const std::string &name = m_sim->GetFeatureName(i++);
		m_scores[name] += *vi;
	}  
}
	
std::ostream& operator<<(std::ostream& os, const ScoreComponentCollection& rhs)
{
	os << "<<" << rhs.m_scores[rhs.m_sim->GetFeatureName(0)];
	for (size_t i=1; i<rhs.m_scores.size(); i++)
		os << ", " << rhs.m_scores[rhs.m_sim->GetFeatureName(i)];
	return os << ">>";
}
	
}