mosesdecoder/contrib/other-builds/moses2/Phrase.h

/*
 * Phrase.h
 *
 *  Created on: 23 Oct 2015
 *      Author: hieu
 */

#pragma once

#include <cstddef>
#include <string>
#include <iostream>
#include "Word.h"
#include "util/pool.hh"
#include "moses/FactorCollection.h"

class PhraseBase
{
public:
  virtual const Word& operator[](size_t pos) const = 0;
  virtual size_t GetSize() const = 0;
};

class SubPhrase;

class Phrase : public PhraseBase
{
	  friend std::ostream& operator<<(std::ostream &, const Phrase &);
public:
  static Phrase *CreateFromString(util::Pool &pool, Moses::FactorCollection &vocab, const std::string &str);

  Phrase(util::Pool &pool, size_t size);
  virtual ~Phrase();

  const Word& operator[](size_t pos) const {
	return m_words[pos];
  }

  Word& operator[](size_t pos) {
	return m_words[pos];
  }

  size_t GetSize() const
  { return m_size; }

  SubPhrase GetSubPhrase(size_t start, size_t end) const;
protected:
  size_t m_size;
  Word *m_words;

  void CreateFromString(Moses::FactorCollection &vocab, const std::vector<std::string> &toks);

};

class SubPhrase : public PhraseBase
{
  friend std::ostream& operator<<(std::ostream &, const SubPhrase &);
public:
  SubPhrase(const Phrase &origPhrase, size_t start, size_t size);
  virtual const Word& operator[](size_t pos) const
  { return m_origPhrase[pos + m_start]; }

  virtual size_t GetSize() const
  { return m_end - m_start + 1; }

protected:
  const Phrase &m_origPhrase;
  size_t m_start, m_end;
};