mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-09-11 11:25:40 +03:00
strong them locally where appropriate, so that compontents can become independent of StaticData once instantiated.
93 lines
2.2 KiB
C++
93 lines
2.2 KiB
C++
// -*- c++ -*-
|
|
#ifndef moses_ForestInput_h
|
|
#define moses_ForestInput_h
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <boost/shared_ptr.hpp>
|
|
|
|
#include <util/string_piece.hh>
|
|
|
|
#include "moses/Syntax/F2S/Forest.h"
|
|
|
|
#include "Sentence.h"
|
|
|
|
namespace Moses
|
|
{
|
|
class TranslationTask;
|
|
class ForestInput : public Sentence
|
|
{
|
|
public:
|
|
friend std::ostream &operator<<(std::ostream&, const ForestInput &);
|
|
|
|
ForestInput(AllOptions::ptr const& opts) : Sentence(opts), m_rootVertex(NULL) {}
|
|
|
|
InputTypeEnum GetType() const {
|
|
return ForestInputType;
|
|
}
|
|
|
|
//! populate this InputType with data from in stream
|
|
virtual int
|
|
Read(std::istream& in,
|
|
std::vector<FactorType> const& factorOrder,
|
|
AllOptions const& opts);
|
|
|
|
//! Output debugging info to stream out
|
|
virtual void Print(std::ostream&) const;
|
|
|
|
//! create trans options specific to this InputType
|
|
virtual TranslationOptionCollection*
|
|
CreateTranslationOptionCollection() const;
|
|
|
|
boost::shared_ptr<const Syntax::F2S::Forest> GetForest() const {
|
|
return m_forest;
|
|
}
|
|
|
|
const Syntax::F2S::Forest::Vertex *GetRootVertex() const {
|
|
return m_rootVertex;
|
|
}
|
|
|
|
private:
|
|
typedef Syntax::F2S::Forest Forest;
|
|
|
|
struct VertexSetHash {
|
|
std::size_t operator()(const Forest::Vertex *v) const {
|
|
std::size_t seed = 0;
|
|
boost::hash_combine(seed, v->pvertex.symbol);
|
|
boost::hash_combine(seed, v->pvertex.span.GetStartPos());
|
|
boost::hash_combine(seed, v->pvertex.span.GetEndPos());
|
|
return seed;
|
|
}
|
|
};
|
|
|
|
struct VertexSetPred {
|
|
bool operator()(const Forest::Vertex *v, const Forest::Vertex *w) const {
|
|
return v->pvertex == w->pvertex;
|
|
}
|
|
};
|
|
|
|
typedef boost::unordered_set<Forest::Vertex *, VertexSetHash,
|
|
VertexSetPred> VertexSet;
|
|
|
|
Forest::Vertex *AddOrDeleteVertex(Forest::Vertex *);
|
|
|
|
std::size_t FindMaxEnd(const Forest &);
|
|
|
|
void FindTopVertices(Forest &, std::vector<Forest::Vertex *> &);
|
|
|
|
void ParseHyperedgeLine(const std::string &,
|
|
const std::vector<FactorType> &);
|
|
|
|
Forest::Vertex *ParseVertex(const StringPiece &,
|
|
const std::vector<FactorType> &);
|
|
|
|
boost::shared_ptr<Forest> m_forest;
|
|
Forest::Vertex *m_rootVertex;
|
|
VertexSet m_vertexSet;
|
|
};
|
|
|
|
} // namespace Moses
|
|
|
|
#endif
|