mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-09-11 11:25:40 +03:00
82 lines
1.7 KiB
C++
82 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include <string>
|
|
#include <cstring>
|
|
#include <cstdlib>
|
|
#include "StatefulFeatureFunction.h"
|
|
#include "FFState.h"
|
|
|
|
namespace Moses
|
|
{
|
|
class CdecFF;
|
|
|
|
class ExternalFeatureState : public FFState
|
|
{
|
|
protected:
|
|
int m_stateSize;
|
|
void *m_data;
|
|
public:
|
|
ExternalFeatureState(int stateSize)
|
|
:m_stateSize(stateSize)
|
|
,m_data(NULL) {
|
|
}
|
|
ExternalFeatureState(int stateSize, void *data);
|
|
|
|
~ExternalFeatureState() {
|
|
free(m_data);
|
|
}
|
|
|
|
int Compare(const FFState& other) const {
|
|
const ExternalFeatureState &otherFF = static_cast<const ExternalFeatureState&>(other);
|
|
int ret = memcmp(m_data, otherFF.m_data, m_stateSize);
|
|
return ret;
|
|
}
|
|
};
|
|
|
|
// copied from cdec
|
|
class ExternalFeature : public StatefulFeatureFunction
|
|
{
|
|
public:
|
|
ExternalFeature(const std::string &line)
|
|
:StatefulFeatureFunction(line) {
|
|
ReadParameters();
|
|
}
|
|
~ExternalFeature();
|
|
|
|
void Load(AllOptions const& opts);
|
|
|
|
bool IsUseable(const FactorMask &mask) const {
|
|
return true;
|
|
}
|
|
|
|
void SetParameter(const std::string& key, const std::string& value);
|
|
|
|
FFState* EvaluateWhenApplied(const Hypothesis& cur_hypo, const FFState* prev_state,
|
|
ScoreComponentCollection* accumulator) const;
|
|
|
|
FFState* EvaluateWhenApplied(
|
|
const ChartHypothesis& /* cur_hypo */,
|
|
int /* featureID - used to index the state in the previous hypotheses */,
|
|
ScoreComponentCollection* accumulator) const;
|
|
|
|
virtual const FFState* EmptyHypothesisState(const InputType &input) const {
|
|
return new ExternalFeatureState(m_stateSize);
|
|
}
|
|
|
|
protected:
|
|
std::string m_path;
|
|
void* lib_handle;
|
|
CdecFF *ff_ext;
|
|
int m_stateSize;
|
|
};
|
|
|
|
class CdecFF
|
|
{
|
|
public:
|
|
virtual ~CdecFF() {}
|
|
virtual int StateSize() const = 0;
|
|
};
|
|
|
|
}
|
|
|