/*
 * PndAhHoughTransform.h
 *
 *  Created on: May 14, 2013
 *      Author: Andreas Herten
 */

#ifndef PNDAHHOUGHTRANSFORM_H_
#define PNDAHHOUGHTRANSFORM_H_

#include "TH2D.h"

class PndAhHoughTransform {

public:
	PndAhHoughTransform();
	PndAhHoughTransform(std::vector<double>, std::vector<double>, std::vector<double>);
	PndAhHoughTransform(std::vector<double>, std::vector<double>, std::vector<double>, double __maxAngle = 360., double __everyXDegrees = 30., bool __verbose = false, bool __keepHitOriginInfo = false);

	void SetXData(std::vector<double> __x) {fVX = __x;};
	void SetYData(std::vector<double> __y) {fVY = __y;};
	void SetRData(std::vector<double> __r) {fVR = __r;};
	void SetMaxAngle(double __maxAngle = 360.) {fMaxAngle = __maxAngle;};
	// void SetAngleRange();
	void SetAngleGranularity(double __everyXDegrees = 30.) {fEveryXDegrees = __everyXDegrees;};
	void SetVerbose(bool __verbose = false) {fVerbose = __verbose;};
	void SetHoughTransformingFunction();

	void SetKeepHitOriginInfo(bool __keepInfo) {fKeepHitOriginInfo = __keepInfo;};
	
	void StartEverything();

	void SetManualHistLimits(double, double);
	TH2D * GetHistogram(bool useAutoLimits = true, TString histName = "HoughHist", TString histTitle = "HT histogram");
	std::vector<std::vector<double> > Getrs() { return fVVrs;};
	std::vector<std::pair<std::vector<double > > > GetrsWithOriginInfo() { return fVpVrs;};
	std::vector<double> GetAngles() {return fVAngles;};

	virtual ~PndAhHoughTransform();

protected:
	void DoEverything();
	void DoGenerateAngles();
	// void DoConformalMapping();
	void DoHoughTransform();

	// helper functions
	template<typename _InputType>
	std::vector<std::vector<_InputType> > generateAllrsForAllXYRs (std::vector<_InputType>, std::vector<_InputType>, std::vector<_InputType>, std::vector<_InputType>);
	std::vector<std::pair<std::vector<_InputType> > > generateAllrsForAllXYRsWithOrigin (std::vector<_InputType>, std::vector<_InputType>, std::vector<_InputType>, std::vector<_InputType>);
	template<typename _InputIterator, typename _DataType>
	std::vector<_DataType> generateAllrsForOneXYR(_InputIterator, _InputIterator, _DataType, _DataType, _DataType);
	template<typename _DataType>
	_DataType function_houghTrafo(_DataType (*trafoFunction)(_DataType, _DataType, _DataType, _DataType), _DataType, _DataType, _DataType, _DataType);
	template<typename _DataType>
	static _DataType function_generic_houghTrafo(_DataType, _DataType, _DataType, _DataType);

private:
	void DoSetAutoHistLimits();

	std::vector<double> fVX;
	std::vector<double> fVY;
	std::vector<double> fVR;
	std::vector<double> fVAngles;
	std::vector<std::vector<double> > fVVrs;
	std::vector<std::pair<std::vector<double>, unsigned int> > fVpVrs; // outer vector: per hit, pair: vector of r per alpha_i and hit id
	double fMaxAngle;
	double fEveryXDegrees;
	double fYMin, fYMax;
	bool fKeepHitOriginInfo;

	// template<typename _DataType>
	double (*functionPointer_houghTrafo)(double, double, double, double); // (x, y, r; alpha)

	bool fHistLimitsAreSet;
	bool fVerbose;
};

// class PndAhHoughOptions {
// public:
// 	PndAhHoughOptions();
	

// 	virtual ~PndAhHoughOptions();
// };


#endif /* PNDAHHOUGHTRANSFORM_H_ */