Welcome to mirror list, hosted at ThFree Co, Russian Federation.

PhraseTable.h « TranslationModel « moses2 « contrib - github.com/moses-smt/mosesdecoder.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 9237f5ba67575d5cea454aad5bcefc523cbdf569 (plain)
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*
 * PhraseTable.h
 *
 *  Created on: 23 Oct 2015
 *      Author: hieu
 */
#pragma once
#include <string>
#include <boost/unordered_map.hpp>
#include "../Word.h"
#include "../HypothesisColl.h"
#include "../FF/StatelessFeatureFunction.h"
#include "../legacy/Util2.h"

namespace Moses2
{

class System;
class InputPathsBase;
class InputPath;
class Manager;
class TargetPhrases;
class Range;

namespace SCFG
{
class InputPath;
class Stacks;
class Manager;
class ActiveChartEntry;
}

////////////////////////////////////////////////////////////////////////
class PhraseTable: public StatelessFeatureFunction
{
public:
  int decodeGraphBackoff;

  PhraseTable(size_t startInd, const std::string &line);
  virtual ~PhraseTable();

  virtual void SetParameter(const std::string& key, const std::string& value);
  virtual void Lookup(const Manager &mgr, InputPathsBase &inputPaths) const;
  virtual TargetPhrases *Lookup(const Manager &mgr, MemPool &pool,
      InputPath &inputPath) const;

  void SetPtInd(size_t ind)
  {  m_ptInd = ind; }

  size_t GetPtInd() const
  {  return m_ptInd; }

  bool SatisfyBackoff(const Manager &mgr, const InputPath &path) const;

  virtual void
  EvaluateInIsolation(MemPool &pool, const System &system, const Phrase<Moses2::Word> &source,
      const TargetPhraseImpl &targetPhrase, Scores &scores,
      SCORE &estimatedScore) const;

  virtual void
  EvaluateInIsolation(MemPool &pool, const System &system, const Phrase<SCFG::Word> &source,
      const TargetPhrase<SCFG::Word> &targetPhrase, Scores &scores,
      SCORE &estimatedScore) const;

  // scfg
  virtual void InitActiveChart(
      MemPool &pool,
      const SCFG::Manager &mgr,
      SCFG::InputPath &path) const = 0;

  virtual void Lookup(
      MemPool &pool,
      const SCFG::Manager &mgr,
      size_t maxChartSpan,
      const SCFG::Stacks &stacks,
      SCFG::InputPath &path) const = 0;

  virtual void LookupUnary(MemPool &pool,
      const SCFG::Manager &mgr,
      const SCFG::Stacks &stacks,
      SCFG::InputPath &path) const;

protected:
  std::string m_path;
  size_t m_ptInd; // in the order that it is list in [feature], NOT order of [mapping]
  size_t m_tableLimit;
  std::vector<FactorType> m_input, m_output;

  // cache
  size_t m_maxCacheSize; // 0 = no caching

  struct CacheCollEntry2
  {
    TargetPhrases *tpsPtr;
    clock_t clock;
  };

  // scfg
  virtual void LookupNT(
      MemPool &pool,
      const SCFG::Manager &mgr,
      const Moses2::Range &subPhraseRange,
      const SCFG::InputPath &prevPath,
      const SCFG::Stacks &stacks,
      SCFG::InputPath &outPath) const;

  virtual void LookupGivenWord(
      MemPool &pool,
      const SCFG::Manager &mgr,
      const SCFG::InputPath &prevPath,
      const SCFG::Word &wordSought,
      const Moses2::Hypotheses *hypos,
      const Moses2::Range &subPhraseRange,
      SCFG::InputPath &outPath) const;

  virtual void LookupGivenNode(
      MemPool &pool,
      const SCFG::Manager &mgr,
      const SCFG::ActiveChartEntry &prevEntry,
      const SCFG::Word &wordSought,
      const Moses2::Hypotheses *hypos,
      const Moses2::Range &subPhraseRange,
      SCFG::InputPath &outPath) const = 0;

};

}