mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-11-10 10:59:21 +03:00
119 lines
4.1 KiB
C++
119 lines
4.1 KiB
C++
#include "TranslationTaskChart.h"
|
|
#include "moses/Sentence.h"
|
|
#include "moses/StaticData.h"
|
|
#include "moses/Incremental.h"
|
|
#include "moses/IOWrapperChart.h"
|
|
#include "moses/OutputCollector.h"
|
|
|
|
using namespace std;
|
|
using namespace Moses;
|
|
using namespace MosesChartCmd;
|
|
|
|
TranslationTaskChart::TranslationTaskChart(InputType *source, IOWrapperChart &ioWrapper,
|
|
boost::shared_ptr<HypergraphOutput<ChartManager> > hypergraphOutput)
|
|
: m_source(source)
|
|
, m_ioWrapper(ioWrapper)
|
|
, m_hypergraphOutput(hypergraphOutput)
|
|
{}
|
|
|
|
TranslationTaskChart::~TranslationTaskChart() {
|
|
delete m_source;
|
|
}
|
|
|
|
void TranslationTaskChart::Run() {
|
|
const StaticData &staticData = StaticData::Instance();
|
|
const size_t translationId = m_source->GetTranslationId();
|
|
|
|
VERBOSE(2,"\nTRANSLATING(" << translationId << "): " << *m_source);
|
|
|
|
if (staticData.GetSearchAlgorithm() == ChartIncremental) {
|
|
Incremental::Manager manager(*m_source);
|
|
const std::vector<search::Applied> &nbest = manager.ProcessSentence();
|
|
if (!nbest.empty()) {
|
|
m_ioWrapper.OutputBestHypo(nbest[0], translationId);
|
|
if (staticData.IsDetailedTranslationReportingEnabled()) {
|
|
const Sentence &sentence = dynamic_cast<const Sentence &>(*m_source);
|
|
m_ioWrapper.OutputDetailedTranslationReport(&nbest[0], sentence, translationId);
|
|
}
|
|
if (staticData.IsDetailedTreeFragmentsTranslationReportingEnabled()) {
|
|
const Sentence &sentence = dynamic_cast<const Sentence &>(*m_source);
|
|
m_ioWrapper.OutputDetailedTreeFragmentsTranslationReport(&nbest[0], sentence, translationId);
|
|
}
|
|
} else {
|
|
m_ioWrapper.OutputBestNone(translationId);
|
|
}
|
|
if (staticData.GetNBestSize() > 0)
|
|
m_ioWrapper.OutputNBestList(nbest, translationId);
|
|
return;
|
|
}
|
|
|
|
ChartManager manager(translationId,*m_source);
|
|
manager.ProcessSentence();
|
|
|
|
UTIL_THROW_IF2(staticData.UseMBR(), "Cannot use MBR");
|
|
|
|
// Output search graph in hypergraph format for Kenneth Heafield's lazy hypergraph decoder
|
|
if (m_hypergraphOutput.get()) {
|
|
m_hypergraphOutput->Write(manager);
|
|
}
|
|
|
|
|
|
// 1-best
|
|
const ChartHypothesis *bestHypo = manager.GetBestHypothesis();
|
|
m_ioWrapper.OutputBestHypo(bestHypo, translationId);
|
|
IFVERBOSE(2) {
|
|
PrintUserTime("Best Hypothesis Generation Time:");
|
|
}
|
|
|
|
if (!staticData.GetAlignmentOutputFile().empty()) {
|
|
m_ioWrapper.OutputAlignment(translationId, bestHypo);
|
|
}
|
|
|
|
if (staticData.IsDetailedTranslationReportingEnabled()) {
|
|
const Sentence &sentence = dynamic_cast<const Sentence &>(*m_source);
|
|
m_ioWrapper.OutputDetailedTranslationReport(bestHypo, sentence, translationId);
|
|
}
|
|
if (staticData.IsDetailedTreeFragmentsTranslationReportingEnabled()) {
|
|
const Sentence &sentence = dynamic_cast<const Sentence &>(*m_source);
|
|
m_ioWrapper.OutputDetailedTreeFragmentsTranslationReport(bestHypo, sentence, translationId);
|
|
}
|
|
if (!staticData.GetOutputUnknownsFile().empty()) {
|
|
m_ioWrapper.OutputUnknowns(manager.GetParser().GetUnknownSources(),
|
|
translationId);
|
|
}
|
|
|
|
//DIMw
|
|
if (staticData.IsDetailedAllTranslationReportingEnabled()) {
|
|
const Sentence &sentence = dynamic_cast<const Sentence &>(*m_source);
|
|
size_t nBestSize = staticData.GetNBestSize();
|
|
std::vector<boost::shared_ptr<ChartKBestExtractor::Derivation> > nBestList;
|
|
manager.CalcNBest(nBestSize, nBestList, staticData.GetDistinctNBest());
|
|
m_ioWrapper.OutputDetailedAllTranslationReport(nBestList, manager, sentence, translationId);
|
|
}
|
|
|
|
// n-best
|
|
size_t nBestSize = staticData.GetNBestSize();
|
|
if (nBestSize > 0) {
|
|
VERBOSE(2,"WRITING " << nBestSize << " TRANSLATION ALTERNATIVES TO " << staticData.GetNBestFilePath() << endl);
|
|
std::vector<boost::shared_ptr<ChartKBestExtractor::Derivation> > nBestList;
|
|
manager.CalcNBest(nBestSize, nBestList,staticData.GetDistinctNBest());
|
|
m_ioWrapper.OutputNBestList(nBestList, translationId);
|
|
IFVERBOSE(2) {
|
|
PrintUserTime("N-Best Hypotheses Generation Time:");
|
|
}
|
|
}
|
|
|
|
if (staticData.GetOutputSearchGraph()) {
|
|
std::ostringstream out;
|
|
manager.OutputSearchGraphMoses( out);
|
|
OutputCollector *oc = m_ioWrapper.GetSearchGraphOutputCollector();
|
|
UTIL_THROW_IF2(oc == NULL, "File for search graph output not specified");
|
|
oc->Write(translationId, out.str());
|
|
}
|
|
|
|
IFVERBOSE(2) {
|
|
PrintUserTime("Sentence Decoding Time:");
|
|
}
|
|
manager.CalcDecoderStatistics();
|
|
}
|