mosesdecoder/moses/src/ScoreComponentCollection.cpp

// $Id$
#include <vector>
#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;
	}  
}
	
	
void ScoreComponentCollection::MultiplyEquals(float scalar)
{
	m_scores *= scalar;
}
	
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 << ">>";
}
	
ScoreComponentCollection StaticData::GetAllWeightsScoreComponentCollection() const
{
	Moses::ScoreComponentCollection ret;
	
	const std::vector<const ScoreProducer*> &producers = m_scoreIndexManager.GetProducers();
	std::vector<const ScoreProducer*>::const_iterator iter;
	
	for (iter = producers.begin(); iter != producers.end(); ++iter)
	{
		const ScoreProducer *producer = *iter;
		unsigned int bookId = producer->GetScoreBookkeepingID();
		size_t startInd = m_scoreIndexManager.GetBeginIndex(bookId);
		size_t endInd = m_scoreIndexManager.GetEndIndex(bookId);
		
		std::cerr << producer->GetScoreProducerDescription();
		
		std::vector<float> weights;
		copy(m_allWeights.begin() + startInd, m_allWeights.begin() + endInd, inserter(weights,weights.end()));
		
		ret.Assign(producer, weights);
		
	}
	
	return ret;
}
	
void StaticData::SetAllWeightsScoreComponentCollection(const ScoreComponentCollection &weightsScoreComponentCollection)
{	
	const std::vector<const ScoreProducer*> &producers = m_scoreIndexManager.GetProducers();
	std::vector<const ScoreProducer*>::const_iterator iter;
	
	for (iter = producers.begin(); iter != producers.end(); ++iter)
	{
		const ScoreProducer *producer = *iter;
		unsigned int bookId = producer->GetScoreBookkeepingID();
		size_t startInd = m_scoreIndexManager.GetBeginIndex(bookId);
		size_t endInd = m_scoreIndexManager.GetEndIndex(bookId);
		
		std::cerr << producer->GetScoreProducerDescription();
		
		std::vector<float> weights = weightsScoreComponentCollection.GetScoresForProducer(producer);

		size_t allInd = startInd;
		for (size_t ind = 0; ind < weights.size(); ++ind)
		{
			m_allWeights[allInd] = weights[ind];
			allInd++;
		}
		assert(allInd == endInd);
	}
	
}

}