File size: 1,632 Bytes
158b61b |
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 |
/*
* 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_ */
|