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

#pragma once
#include <vector>
#include <deque>
#include <boost/thread/tss.hpp>
#include <boost/pool/object_pool.hpp>
#include <boost/shared_ptr.hpp>
#include "FF/FeatureFunctions.h"
#include "Weights.h"
#include "MemPool.h"
#include "Recycler.h"
#include "legacy/FactorCollection.h"
#include "legacy/Parameter.h"
#include "TypeDef.h"
#include "legacy/Bitmaps.h"
#include "legacy/OutputCollector.h"
#include "parameters/AllOptions.h"

namespace Moses2
{
namespace NSCubePruning
{
class Stack;
}

class FeatureFunction;
class StatefulFeatureFunction;
class PhraseTable;
class HypothesisBase;

class System
{
public:
  const Parameter &params;
  AllOptions options;
  FeatureFunctions featureFunctions;
  Weights weights;
  std::vector<const PhraseTable*> mappings;
  bool isPb;

  mutable boost::shared_ptr<OutputCollector> bestCollector, nbestCollector;

  // moses.ini params
  int cpuAffinityOffset;
  int cpuAffinityOffsetIncr;

  System(const Parameter &paramsArg);
  virtual ~System();

  MemPool &GetSystemPool() const;
  MemPool &GetManagerPool() const;
  FactorCollection &GetVocab() const;

  Recycler<HypothesisBase*> &GetHypoRecycler() const;

  Batch &GetBatch(MemPool &pool) const;

protected:
  mutable FactorCollection m_vocab;
  mutable boost::thread_specific_ptr<MemPool> m_managerPool;
  mutable boost::thread_specific_ptr<MemPool> m_systemPool;

  mutable boost::thread_specific_ptr<Recycler<HypothesisBase*> > m_hypoRecycler;

  mutable boost::thread_specific_ptr<Batch> m_batch;

  void LoadWeights();
  void LoadMappings();

  void IsPb();

};

}