/** 
* @file SemSegSTF.h
* @brief Localization system
* @author Erik Rodner
* @date 02/11/2008

*/
#ifndef SemSegSTFINCLUDE
#define SemSegSTFINCLUDE

#ifdef NOVISUAL
#include <objrec/nice_nonvis.h>
#else
#include <objrec/nice.h>
#endif
 
#include "SemanticSegmentation.h"
#include "objrec/classifier/fpclassifier/randomforest/DecisionNode.h"
#include "objrec/classifier/fpclassifier/randomforest/FPCRandomForests.h"


namespace OBJREC {

/** Localization system */
class SemSegSTF : public SemanticSegmentation
{
    protected:
 
	int numClasses;
	int maxdepthSegmentationForest;
	int maxdepthSegmentationForestScores;

	bool use_semantic_features;
	bool use_pixelpair_features;

	int subsamplex;
	int subsampley;

	bool read_seg_cache;
	bool read_pixel_cache;
	// refactor-nice.pl: check this substitution
	// old: string cacheseg;
	std::string cacheseg;
	// refactor-nice.pl: check this substitution
	// old: string cachepixel;
	std::string cachepixel;

	// refactor-nice.pl: check this substitution
	// old: string resultsdatadir;
	std::string resultsdatadir;

	double alphaDetectionPrior;
	double alphaImagePrior;
	
	int imagePriorMethod;
	int priorK;

	int detectionPriorMethod;

	/** 
	    assign all pixels with
	    normalized entropy $E/log(n)$ above
	    this threshold to the
	    background class 
	*/
	double entropyThreshold;

	int backgroundModelType;

	map<DecisionNode *, pair<long, int> > index;
	FPCRandomForests *fpcSeg;
	FPCRandomForests *fpcPixel;

	map<string, LocalizationResult *> detresults;

	void fillCachePixel ( CachedExample *ce );
	void fillCacheSegmentation ( CachedExample *ce );

    public:
	/** simple constructor */
	SemSegSTF( const Config *conf,
  	              const MultiDataset *md );
      
	/** simple destructor */
	virtual ~SemSegSTF();

	void semanticseg ( CachedExample *ce, 
			   // refactor-nice.pl: check this substitution
			   // old: Image & segresult,
			   NICE::Image & segresult,
			   GenericImage<double> & probabilities );

};


} // namespace

#endif