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

StatefulFeatureFunction.h « FF « moses2 - github.com/moses-smt/mosesdecoder.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 7cb3eaae97c23373c4236b9842fd7d8ec58205f7 (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
/*
 * StatefulFeatureFunction.h
 *
 *  Created on: 24 Oct 2015
 *      Author: hieu
 */

#ifndef STATEFULFEATUREFUNCTION_H_
#define STATEFULFEATUREFUNCTION_H_

#include "FeatureFunction.h"
#include "FFState.h"
#include "../MemPool.h"

namespace Moses2
{

class Hypothesis;
class InputType;

namespace SCFG
{
class Hypothesis;
class Manager;
}

class StatefulFeatureFunction: public FeatureFunction
{
public:
  StatefulFeatureFunction(size_t startInd, const std::string &line);
  virtual ~StatefulFeatureFunction();

  void SetStatefulInd(size_t ind) {
    m_statefulInd = ind;
  }
  size_t GetStatefulInd() const {
    return m_statefulInd;
  }

  //! return uninitialise state
  virtual FFState* BlankState(MemPool &pool, const System &sys) const = 0;

  //! return the state associated with the empty hypothesis for a given sentence
  virtual void EmptyHypothesisState(FFState &state, const ManagerBase &mgr,
                                    const InputType &input, const Hypothesis &hypo) const = 0;

  virtual void EvaluateWhenApplied(const ManagerBase &mgr,
                                   const Hypothesis &hypo, const FFState &prevState, Scores &scores,
                                   FFState &state) const = 0;

  virtual void EvaluateWhenApplied(const SCFG::Manager &mgr,
                                   const SCFG::Hypothesis &hypo, int featureID, Scores &scores,
                                   FFState &state) const = 0;

  virtual void EvaluateWhenAppliedBatch(
    const System &system,
    const Batch &batch) const;

protected:
  size_t m_statefulInd;

};

}

#endif /* STATEFULFEATUREFUNCTION_H_ */