#pragma once
#include "../FFState.h"
#include "LRModel.h"

namespace Moses2
{
template<typename WORD>
class TargetPhrase;

class LexicalReordering;
class Hypothesis;
class System;
class Scores;
class Bitmap;
class ManagerBase;
class InputType;
class InputPathBase;
class Word;

class LRState: public FFState
{
public:
  typedef LRModel::ReorderingType ReorderingType;
  const TargetPhrase<Moses2::Word> *prevTP;

  LRState(const LRModel &config, LRModel::Direction dir, size_t offset);

  virtual void Init(const LRState *prev, const TargetPhrase<Moses2::Word> &topt,
                    const InputPathBase &path, bool first, const Bitmap *coverage) = 0;

  virtual void Expand(const ManagerBase &mgr, const LexicalReordering &ff,
                      const Hypothesis &hypo, size_t phraseTableInd, Scores &scores,
                      FFState &state) const = 0;

  void CopyScores(const System &system, Scores &accum, const TargetPhrase<Moses2::Word> &topt,
                  ReorderingType reoType) const;

protected:
  const LRModel& m_configuration;
  LRModel::Direction m_direction;
  size_t m_offset;

  int
  ComparePrevScores(const TargetPhrase<Moses2::Word> *other) const;

};

}