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
|
/*
* 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_ */
|