| #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; |
| } |
| }; |
|
|
| |
| 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& , |
| int , |
| 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; |
| }; |
|
|
| } |
|
|
|
|
