#ifndef HPYP_LEX_TRANS
#define HPYP_LEX_TRANS

#include <vector>
#include "wordid.h"
#include "prob.h"
#include "sampler.h"
#include "freqdict.h"
#include "poisson_uniform_word_model.h"

struct FreqBinner;
template <class B> struct PYPWordModel;
template <typename T, class B> struct ConditionalPYPWordModel;

struct HPYPLexicalTranslation {
  explicit HPYPLexicalTranslation(const std::vector<std::vector<WordID> >& lets,
                                 const unsigned vocab_size,
                                 const unsigned num_letters);

  prob_t Likelihood() const;

  void ResampleHyperparameters(MT19937* rng);
  prob_t Prob(WordID src, WordID trg) const;  // return p(trg | src)
  void Summary() const;
  void Increment(WordID src, WordID trg, MT19937* rng);
  void Decrement(WordID src, WordID trg, MT19937* rng);
  unsigned UniqueConditioningContexts() const;

 private:
  const std::vector<std::vector<WordID> >& letters;   // spelling dictionary
  PoissonUniformWordModel base;  // "generator" of English types
  PYPWordModel<PoissonUniformWordModel>* up0;  // model English lexicon
  ConditionalPYPWordModel<PYPWordModel<PoissonUniformWordModel>, FreqBinner>* tmodel;  // translation distributions
                      // (model English word | French word)
  const WordID kX;
};

#endif