moses/src/ScoreIndexManager.h

Go to the documentation of this file.

// $Id: ScoreIndexManager.h 3394 2010-08-10 13:12:00Z bhaddow $

#ifndef moses_ScoreIndexManager_h
#define moses_ScoreIndexManager_h

#include <iostream>
#include <vector>
#if HAVE_CONFIG_H
#include "config.h"
#endif
#ifdef HAVE_PROTOBUF
#include "hypergraph.pb.h"
#endif

namespace Moses
{

class ScoreProducer;
class ScoreComponentCollection;  // debugging only
class StatefulFeatureFunction;
class StatelessFeatureFunction;

class ScoreIndexManager
{
  friend std::ostream& operator<<(std::ostream& os, const ScoreIndexManager& sim);
public:
        ScoreIndexManager() : m_last(0) {}

        void AddScoreProducer(const ScoreProducer* producer);
        void InitFeatureNames();

        size_t GetBeginIndex(size_t scoreBookkeepingID) const { return m_begins[scoreBookkeepingID]; }
        size_t GetEndIndex(size_t scoreBookkeepingID) const { return m_ends[scoreBookkeepingID]; }
        size_t GetTotalNumberOfScores() const { return m_last; }
        void PrintLabeledScores(std::ostream& os, const ScoreComponentCollection& scc) const;
        void PrintLabeledWeightedScores(std::ostream& os, const ScoreComponentCollection& scc, const std::vector<float>& weights) const;
#ifdef HAVE_PROTOBUF
        void SerializeFeatureNamesToPB(hgmert::Hypergraph* hg) const;
#endif
        void InitWeightVectorFromFile(const std::string& fnam, std::vector<float>* m_allWeights) const;
private:
        ScoreIndexManager(const ScoreIndexManager&); // don't implement

        std::vector<size_t> m_begins;
        std::vector<size_t> m_ends;
        std::vector<const ScoreProducer*> m_producers; 
        std::vector<std::string> m_featureNames;
        std::vector<std::string> m_featureShortNames;
        size_t m_last;
};


}

#endif

---