/***********************************************************************
Moses - factored phrase-based language decoder
Copyright (C) 2014- University of Edinburgh

This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.

This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
***********************************************************************/
#pragma once

#include <vector>

#include <boost/scoped_ptr.hpp>
#include <boost/shared_ptr.hpp>

#include "ForestRescore.h"
#include "Hypergraph.h"
#include "HypPackEnumerator.h"
#include "MiraFeatureVector.h"
#include "MiraWeightVector.h"

//
// Used by batch mira to get the hope, fear and model hypothesis. This wraps
// the n-best list and lattice/hypergraph implementations
//

namespace MosesTuning
{


/** Initialise weights from files. Returns weight vector and number of dense features */
std::pair<MiraWeightVector*,size_t>
InitialiseWeights(const std::string& denseInitFile, const std::string& sparseInitFile,
                  const std::string& type, bool verbose);

class Scorer;

/** To be filled in by the decoder */
struct HopeFearData {
  MiraFeatureVector modelFeatures;
  MiraFeatureVector hopeFeatures;
  MiraFeatureVector fearFeatures;

  std::vector<float> modelStats;
  std::vector<float> hopeStats;

  ValType hopeBleu;
  ValType fearBleu;

  bool hopeFearEqual;
};

//Abstract base class
class HopeFearDecoder
{
public:
  //iterator methods
  virtual void reset() = 0;
  virtual void next() = 0;
  virtual bool finished() = 0;

  virtual ~HopeFearDecoder() {};

  /**
    * Calculate hope, fear and model hypotheses
    **/
  virtual void HopeFear(
    const std::vector<ValType>& backgroundBleu,
    const MiraWeightVector& wv,
    HopeFearData* hopeFear
  ) = 0;

  /** Max score decoding */
  virtual void MaxModel(const AvgWeightVector& wv, std::vector<ValType>* stats)
  = 0;

  /** Calculate bleu on training set */
  ValType Evaluate(const AvgWeightVector& wv);

protected:
  Scorer* scorer_;
};


/** Gets hope-fear from nbest lists */
class NbestHopeFearDecoder : public virtual HopeFearDecoder
{
public:
  NbestHopeFearDecoder(const std::vector<std::string>& featureFiles,
                       const std::vector<std::string>&  scoreFiles,
                       bool streaming,
                       bool  no_shuffle,
                       bool safe_hope,
                       Scorer* scorer
                      );

  virtual void reset();
  virtual void next();
  virtual bool finished();

  virtual void HopeFear(
    const std::vector<ValType>& backgroundBleu,
    const MiraWeightVector& wv,
    HopeFearData* hopeFear
  );

  virtual void MaxModel(const AvgWeightVector& wv, std::vector<ValType>* stats);

private:
  boost::scoped_ptr<HypPackEnumerator> train_;
  bool safe_hope_;

};



/** Gets hope-fear from hypergraphs */
class HypergraphHopeFearDecoder : public virtual HopeFearDecoder
{
public:
  HypergraphHopeFearDecoder(
    const std::string& hypergraphDir,
    const std::vector<std::string>& referenceFiles,
    size_t num_dense,
    bool streaming,
    bool no_shuffle,
    bool safe_hope,
    size_t hg_pruning,
    const MiraWeightVector& wv,
    Scorer* scorer_
  );

  virtual void reset();
  virtual void next();
  virtual bool finished();

  virtual void HopeFear(
    const std::vector<ValType>& backgroundBleu,
    const MiraWeightVector& wv,
    HopeFearData* hopeFear
  );

  virtual void MaxModel(const AvgWeightVector& wv, std::vector<ValType>* stats);

private:
  size_t num_dense_;
  //maps sentence Id to graph ptr
  typedef std::map<size_t, boost::shared_ptr<Graph> > GraphColl;
  GraphColl graphs_;
  std::vector<size_t> sentenceIds_;
  std::vector<size_t>::const_iterator sentenceIdIter_;
  ReferenceSet references_;
  Vocab vocab_;
};

};