/*
 * UnknownWordPenalty.h
 *
 *  Created on: 28 Oct 2015
 *      Author: hieu
 */

#pragma once

#include "PhraseTable.h"

namespace Moses2
{
class Sentence;
class InputPaths;
class Range;

class UnknownWordPenalty: public PhraseTable
{
public:
  UnknownWordPenalty(size_t startInd, const std::string &line);
  virtual ~UnknownWordPenalty();

  virtual void SetParameter(const std::string& key, const std::string& value);

  void Lookup(const Manager &mgr, InputPathsBase &inputPaths) const;
  virtual TargetPhrases *Lookup(const Manager &mgr, MemPool &pool,
                                InputPath &inputPath) const;

  void ProcessXML(
    const Manager &mgr,
    MemPool &pool,
    const Sentence &sentence,
    InputPaths &inputPaths) const;

  virtual void
  EvaluateInIsolation(const System &system, const Phrase<Moses2::Word> &source,
                      const TargetPhraseImpl &targetPhrase, Scores &scores,
                      SCORE &estimatedScore) const;

  virtual void InitActiveChart(
    MemPool &pool,
    const SCFG::Manager &mgr,
    SCFG::InputPath &path) const;

  void Lookup(MemPool &pool,
              const SCFG::Manager &mgr,
              size_t maxChartSpan,
              const SCFG::Stacks &stacks,
              SCFG::InputPath &path) const;

  void LookupUnary(MemPool &pool,
                   const SCFG::Manager &mgr,
                   const SCFG::Stacks &stacks,
                   SCFG::InputPath &path) const;

protected:
  virtual void LookupNT(
    MemPool &pool,
    const SCFG::Manager &mgr,
    const Moses2::Range &subPhraseRange,
    const SCFG::InputPath &prevPath,
    const SCFG::Stacks &stacks,
    SCFG::InputPath &outPath) const;

  virtual void LookupGivenWord(
    MemPool &pool,
    const SCFG::Manager &mgr,
    const SCFG::InputPath &prevPath,
    const SCFG::Word &wordSought,
    const Moses2::Hypotheses *hypos,
    const Moses2::Range &subPhraseRange,
    SCFG::InputPath &outPath) const;

  virtual void LookupGivenNode(
    MemPool &pool,
    const SCFG::Manager &mgr,
    const SCFG::ActiveChartEntry &prevEntry,
    const SCFG::Word &wordSought,
    const Moses2::Hypotheses *hypos,
    const Moses2::Range &subPhraseRange,
    SCFG::InputPath &outPath) const;
protected:
  bool m_drop;
  std::string m_prefix, m_suffix;
};

}