mosesdecoder/moses/src/TreeInput.h

#ifndef moses_TreeInput_h
#define moses_TreeInput_h


#include <vector>
#include "Sentence.h"
#include "ChartTranslationOption.h"

namespace Moses
{
class XMLParseOutput
{
public:
  std::string m_label;
  WordsRange m_range;

  XMLParseOutput(const std::string &label, const WordsRange &range)
    : m_label(label)
    , m_range(range)
  {}
};

class TreeInput : public Sentence
{
  friend std::ostream& operator<<(std::ostream&, const TreeInput&);

protected:
  std::vector<std::vector<NonTerminalSet> > m_sourceChart;
  std::vector <ChartTranslationOption*> m_xmlChartOptionsList;

  void AddChartLabel(size_t startPos, size_t endPos, const std::string &label
                     ,const std::vector<FactorType>& factorOrder);
  void AddChartLabel(size_t startPos, size_t endPos, const Word &label
                     ,const std::vector<FactorType>& factorOrder);
  NonTerminalSet &GetLabelSet(size_t startPos, size_t endPos) {
    return m_sourceChart[startPos][endPos - startPos];
  }

  bool ProcessAndStripXMLTags(std::string &line, std::vector<XMLParseOutput> &sourceLabels, std::vector<XmlOption*> &res);

public:
  TreeInput()
  {}

  InputTypeEnum GetType() const {
    return TreeInputType;
  }

  //! populate this InputType with data from in stream
  virtual int Read(std::istream& in,const std::vector<FactorType>& factorOrder);

  //! Output debugging info to stream out
  virtual void Print(std::ostream&) const;

  //! create trans options specific to this InputType
  virtual TranslationOptionCollection* CreateTranslationOptionCollection() const;

  virtual const NonTerminalSet &GetLabelSet(size_t startPos, size_t endPos) const {
    return m_sourceChart[startPos][endPos - startPos];
  }

  std::vector <ChartTranslationOption*> GetXmlChartTranslationOptions() const {
    return m_xmlChartOptionsList;
  };
};

}

#endif