File size: 1,408 Bytes
158b61b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
#pragma once
#include <set>
#include <vector>
#include <boost/shared_ptr.hpp>
#include "moses/InputType.h"
#include "moses/Syntax/KBestExtractor.h"
#include "moses/Syntax/Manager.h"
#include "moses/Syntax/SVertexStack.h"
#include "moses/Word.h"
#include "OovHandler.h"
#include "ParserCallback.h"
#include "PChart.h"
#include "SChart.h"
namespace Moses
{
namespace Syntax
{
struct SHyperedge;
namespace S2T
{
template<typename Parser>
class Manager : public Syntax::Manager
{
public:
Manager(ttasksptr const& ttask);
void Decode();
// Get the SHyperedge for the 1-best derivation.
const SHyperedge *GetBestSHyperedge() const;
void ExtractKBest(
std::size_t k,
std::vector<boost::shared_ptr<KBestExtractor::Derivation> > &kBestList,
bool onlyDistinct=false) const;
void OutputDetailedTranslationReport(OutputCollector *collector) const;
private:
void FindOovs(const PChart &, boost::unordered_set<Word> &, std::size_t);
void InitializeCharts();
void InitializeParsers(PChart &, std::size_t);
void RecombineAndSort(const std::vector<SHyperedge*> &, SVertexStack &);
void PrunePChart(const SChart::Cell &, PChart::Cell &);
PChart m_pchart;
SChart m_schart;
boost::shared_ptr<typename Parser::RuleTrie> m_oovRuleTrie;
std::vector<boost::shared_ptr<Parser> > m_parsers;
};
} // S2T
} // Syntax
} // Moses
// Implementation
#include "Manager-inl.h"
|