mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2025-01-06 19:49:41 +03:00
90 lines
2.1 KiB
C++
90 lines
2.1 KiB
C++
/*
|
|
* System.h
|
|
*
|
|
* Created on: 23 Oct 2015
|
|
* Author: hieu
|
|
*/
|
|
|
|
#pragma once
|
|
#include <vector>
|
|
#include <boost/thread/tss.hpp>
|
|
#include <boost/pool/object_pool.hpp>
|
|
#include "FF/FeatureFunctions.h"
|
|
#include "Weights.h"
|
|
#include "MemPool.h"
|
|
#include "Recycler.h"
|
|
#include "legacy/FactorCollection.h"
|
|
#include "legacy/Parameter.h"
|
|
#include "TypeDef.h"
|
|
#include "Search/CubePruning/Misc.h"
|
|
#include "legacy/Bitmaps.h"
|
|
#include "legacy/OutputCollector.h"
|
|
|
|
namespace Moses2
|
|
{
|
|
namespace NSCubePruning
|
|
{
|
|
class Stack;
|
|
}
|
|
|
|
class FeatureFunction;
|
|
class StatefulFeatureFunction;
|
|
class PhraseTable;
|
|
class Hypothesis;
|
|
|
|
class System {
|
|
public:
|
|
const Parameter ¶ms;
|
|
FeatureFunctions featureFunctions;
|
|
Weights weights;
|
|
std::vector<const PhraseTable*> mappings;
|
|
mutable MemPool systemPool;
|
|
mutable OutputCollector bestCollector;
|
|
|
|
// moses.ini params
|
|
size_t stackSize;
|
|
int maxDistortion;
|
|
size_t maxPhraseLength;
|
|
int numThreads;
|
|
|
|
SearchAlgorithm searchAlgorithm;
|
|
size_t popLimit;
|
|
|
|
std::string outputFilePath;
|
|
size_t nbestSize;
|
|
bool onlyDistinct;
|
|
|
|
bool outputHypoScore;
|
|
|
|
System(const Parameter ¶msArg);
|
|
virtual ~System();
|
|
|
|
MemPool &GetManagerPool() const;
|
|
FactorCollection &GetVocab() const;
|
|
|
|
Recycler<Hypothesis*> &GetHypoRecycler() const;
|
|
ObjectPoolContiguous<Hypothesis*> &GetBatchForEval() const;
|
|
Bitmaps &GetBitmaps() const;
|
|
NSCubePruning::CubeEdge::Queue &GetQueue() const;
|
|
NSCubePruning::CubeEdge::SeenPositions &GetSeenPositions() const;
|
|
|
|
protected:
|
|
mutable FactorCollection m_vocab;
|
|
mutable boost::thread_specific_ptr<MemPool> m_managerPool;
|
|
mutable boost::thread_specific_ptr< Recycler<Hypothesis*> > m_hypoRecycler;
|
|
mutable boost::thread_specific_ptr< ObjectPoolContiguous<Hypothesis*> > m_batchForEval;
|
|
|
|
mutable boost::thread_specific_ptr< NSCubePruning::CubeEdge::Queue> m_queue;
|
|
mutable boost::thread_specific_ptr< NSCubePruning::CubeEdge::SeenPositions> m_seenPositions;
|
|
|
|
mutable boost::thread_specific_ptr<Bitmaps> m_bitmaps;
|
|
|
|
void LoadWeights();
|
|
void LoadMappings();
|
|
|
|
void ini_performance_options();
|
|
};
|
|
|
|
}
|
|
|