// $Id$

#ifndef PHRASEDICTIONARYTREEADAPTOR_H_
#define PHRASEDICTIONARYTREEADAPTOR_H_
#include <vector>
#include "TypeDef.h"
#include "PhraseDictionaryMemory.h"
#include "TargetPhraseCollection.h"

class Phrase;
class PDTAimp;
class WordsRange;
class InputType;

/*** Implementation of a phrase table in a trie that is binarized and
 * stored on disk.
 */
class PhraseDictionaryTreeAdaptor : public PhraseDictionary {
	typedef PhraseDictionary MyBase;
	PDTAimp *imp;
	friend class PDTAimp;
	PhraseDictionaryTreeAdaptor();
	PhraseDictionaryTreeAdaptor(const PhraseDictionaryTreeAdaptor&);
	void operator=(const PhraseDictionaryTreeAdaptor&);
	
 public:
	PhraseDictionaryTreeAdaptor(size_t numScoreComponent,unsigned numInputScores);
	virtual ~PhraseDictionaryTreeAdaptor();

	// enable/disable caching
	// you enable caching if you request the target candidates for a source phrase multiple times
	// if you do caching somewhere else, disable it
	// good settings for current Moses: disable for first factor, enable for other factors
	// default: enable	

	void EnableCache();
	void DisableCache();

	// initialize ...
	bool Load(const std::vector<FactorType> &input
							, const std::vector<FactorType> &output
							, const std::string &filePath
							, const std::vector<float> &weight
							, size_t tableLimit
							, const LMList &languageModels
							, float weightWP

							);

	// get translation candidates for a given source phrase
	// returns null pointer if nothing found
	TargetPhraseCollection const* GetTargetPhraseCollection(Phrase const &src) const;
	TargetPhraseCollection const* GetTargetPhraseCollection(InputType const& src,WordsRange const & srcRange) const;

	// clean up temporary memory etc.
	void CleanUp();

	void InitializeForInput(InputType const& source);

	// change model scaling factors
	void SetWeightTransModel(const std::vector<float> &weightT);

	// this function can be only used for UNKNOWN source phrases
	void AddEquivPhrase(const Phrase &source, const TargetPhrase &targetPhrase);
		
	std::string GetScoreProducerDescription() const;
	
	size_t GetNumInputScores() const;
	
};
#endif