mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-09-11 11:25:40 +03:00
58 lines
1.3 KiB
C++
58 lines
1.3 KiB
C++
#ifndef moses_Search_h
|
|
#define moses_Search_h
|
|
|
|
#include <vector>
|
|
#include "TypeDef.h"
|
|
#include "TranslationOption.h"
|
|
#include "Phrase.h"
|
|
#include "InputPath.h"
|
|
#include "Bitmaps.h"
|
|
#include "Timer.h"
|
|
|
|
namespace Moses
|
|
{
|
|
|
|
class HypothesisStack;
|
|
class Hypothesis;
|
|
class InputType;
|
|
class TranslationOptionCollection;
|
|
class Manager;
|
|
class Phrase;
|
|
|
|
/** Base search class used in the phrase-based decoder.
|
|
*
|
|
* Actual search class that implement the cube pruning algorithm (SearchCubePruning)
|
|
* or standard beam search (SearchNormal) should inherits from this class, and
|
|
* override pure virtual functions.
|
|
*/
|
|
class Search
|
|
{
|
|
public:
|
|
virtual const std::vector<HypothesisStack*>& GetHypothesisStacks() const = 0;
|
|
virtual const Hypothesis *GetBestHypothesis() const = 0;
|
|
|
|
//! Decode the sentence according to the specified search algorithm.
|
|
virtual void Decode() = 0;
|
|
|
|
explicit Search(Manager& manager);
|
|
virtual ~Search() {}
|
|
|
|
protected:
|
|
Manager& m_manager;
|
|
const InputType &m_source;
|
|
AllOptions const& m_options;
|
|
|
|
InputPath m_inputPath; // for initial hypo
|
|
TranslationOption m_initialTransOpt; /**< used to seed 1st hypo */
|
|
Bitmaps m_bitmaps;
|
|
|
|
/** flag indicating that decoder ran out of time (see switch -time-out) */
|
|
size_t interrupted_flag;
|
|
|
|
Timer m_timer;
|
|
bool out_of_time();
|
|
};
|
|
|
|
}
|
|
#endif
|