[u/mrichter/AliRoot.git] / HLT / hough / AliL3Hough.h

#ifndef ALIL3_HOUGH
#define ALIL3_HOUGH

#include "AliL3RootTypes.h"

class AliL3HoughMaxFinder;
class AliL3HoughBaseTransformer;
class AliL3Histogram;
class AliL3MemHandler;
class AliL3FileHandler;
class AliL3HoughEval;
class AliL3TrackArray;
class AliL3HoughMerger;
class AliL3HoughIntMerger;
class AliL3HoughGlobalMerger;

class AliL3Hough {
  
 private:
  Char_t fPath[256];
  Bool_t fBinary;
  Bool_t fAddHistograms;
  Bool_t fDoIterative;
  Bool_t fWriteDigits;
  Int_t fNEtaSegments;
  Int_t fNPatches;
  Int_t fPeakThreshold;
#ifdef use_aliroot
  AliL3FileHandler **fMemHandler; //!
#else
  AliL3MemHandler **fMemHandler; //!
#endif
  AliL3HoughBaseTransformer **fHoughTransformer; //!
  AliL3HoughEval **fEval; //!
  AliL3HoughMaxFinder *fPeakFinder; //!
  AliL3TrackArray **fTracks; //!
  AliL3HoughMerger *fMerger; //!
  AliL3HoughIntMerger *fInterMerger; //!
  AliL3HoughGlobalMerger *fGlobalMerger; //!

  void CleanUp();
  void Init();
  
 public:
  
  AliL3Hough(); 
  AliL3Hough(Char_t *path,Bool_t binary,Int_t n_eta_segments=100);
  virtual ~AliL3Hough();
  
  void Process(Int_t minslice,Int_t maxslice);
  void ReadData(Int_t slice);
  void Transform(Int_t row_range = -1);
  void ProcessSliceIter();
  void ProcessPatchIter(Int_t patch);
  void MergePatches();
  void MergeInternally();

  void FindTrackCandidates();
  void AddAllHistograms();
  void Evaluate(Int_t road_width=1);
  void EvaluateWithEta();
  void WriteTracks(Char_t *path="./");
#ifdef use_aliroot
  void WriteDigits(Char_t *outfile="output_digits.root");
#endif

  //Setters
  void SetNEtaSegments(Int_t i) {fNEtaSegments = i;}
  void SetAddHistograms() {fAddHistograms = kTRUE;}
  void DoIterative() {fDoIterative = kTRUE;}
  void SetWriteDigits() {fWriteDigits = kTRUE;}
  void SetPeakThreshold(Int_t i) {fPeakThreshold = i;}
  
  //Getters
  AliL3HoughBaseTransformer *GetTransformer(Int_t i) {if(!fHoughTransformer[i]) return 0; return fHoughTransformer[i];}
  AliL3TrackArray *GetTracks(Int_t i) {if(!fTracks[i]) return 0; return fTracks[i];}
  AliL3HoughEval *GetEval(Int_t i) {if(!fEval[i]) return 0; return fEval[i];}
  AliL3HoughMerger *GetMerger() {if(!fMerger) return 0; return fMerger;}
  AliL3HoughIntMerger *GetInterMerger() {if(!fInterMerger) return 0; return fInterMerger;}
#ifdef use_aliroot
  AliL3FileHandler *GetMemHandler(Int_t i) {if(!fMemHandler[i]) return 0; return fMemHandler[i];}
#else
  AliL3MemHandler *GetMemHandler(Int_t i) {if(!fMemHandler[i]) return 0; return fMemHandler[i];}
#endif

  ClassDef(AliL3Hough,1) //Hough transform base class

};

#endif
Commit	Line	Data
f000f8a5	1	#ifndef ALIL3_HOUGH
	2	#define ALIL3_HOUGH
	3
	4	#include "AliL3RootTypes.h"
	5
	6	class AliL3HoughMaxFinder;
237d3f5c	7	class AliL3HoughBaseTransformer;
f80b98cb	8	class AliL3Histogram;
95a00d93	9	class AliL3MemHandler;
f80b98cb	10	class AliL3FileHandler;
f80b98cb	11	class AliL3HoughEval;
f80b98cb	12	class AliL3TrackArray;
b1886074	13	class AliL3HoughMerger;
b1886074	14	class AliL3HoughIntMerger;
1c404dd5	15	class AliL3HoughGlobalMerger;
f000f8a5	16
b1886074	17	class AliL3Hough {
f000f8a5	18
f000f8a5	19	private:
f80b98cb	20	Char_t fPath[256];
4fc9a6a4	21	Bool_t fBinary;
a6008206	22	Bool_t fAddHistograms;
b1886074	23	Bool_t fDoIterative;
a6008206	24	Bool_t fWriteDigits;
f80b98cb	25	Int_t fNEtaSegments;
1c404dd5	26	Int_t fNPatches;
237d3f5c	27	Int_t fPeakThreshold;
95a00d93	28	#ifdef use_aliroot
4fc9a6a4	29	AliL3FileHandler **fMemHandler; //!
95a00d93	30	#else
	31	AliL3MemHandler **fMemHandler; //!
	32	#endif
237d3f5c	33	AliL3HoughBaseTransformer **fHoughTransformer; //!
a6008206	34	AliL3HoughEval **fEval; //!
a6008206	35	AliL3HoughMaxFinder *fPeakFinder; //!
b1886074	36	AliL3TrackArray **fTracks; //!
	37	AliL3HoughMerger *fMerger; //!
	38	AliL3HoughIntMerger *fInterMerger; //!
1c404dd5	39	AliL3HoughGlobalMerger *fGlobalMerger; //!
1c404dd5	40
b1886074	41	void CleanUp();
4fc9a6a4	42	void Init();
4fc9a6a4	43
f000f8a5	44	public:
4fc9a6a4	45
f000f8a5	46	AliL3Hough();
4fc9a6a4	47	AliL3Hough(Char_t *path,Bool_t binary,Int_t n_eta_segments=100);
f000f8a5	48	virtual ~AliL3Hough();
f000f8a5	49
a6008206	50	void Process(Int_t minslice,Int_t maxslice);
b1886074	51	void ReadData(Int_t slice);
237d3f5c	52	void Transform(Int_t row_range = -1);
b1886074	53	void ProcessSliceIter();
	54	void ProcessPatchIter(Int_t patch);
	55	void MergePatches();
	56	void MergeInternally();
	57
a6008206	58	void FindTrackCandidates();
a6008206	59	void AddAllHistograms();
b1886074	60	void Evaluate(Int_t road_width=1);
b1886074	61	void EvaluateWithEta();
95a00d93	62	void WriteTracks(Char_t *path="./");
95a00d93	63	#ifdef use_aliroot
a6008206	64	void WriteDigits(Char_t *outfile="output_digits.root");
95a00d93	65	#endif
95a00d93	66
4fc9a6a4	67	//Setters
4fc9a6a4	68	void SetNEtaSegments(Int_t i) {fNEtaSegments = i;}
a6008206	69	void SetAddHistograms() {fAddHistograms = kTRUE;}
b1886074	70	void DoIterative() {fDoIterative = kTRUE;}
a6008206	71	void SetWriteDigits() {fWriteDigits = kTRUE;}
237d3f5c	72	void SetPeakThreshold(Int_t i) {fPeakThreshold = i;}
f80b98cb	73
4fc9a6a4	74	//Getters
237d3f5c	75	AliL3HoughBaseTransformer *GetTransformer(Int_t i) {if(!fHoughTransformer[i]) return 0; return fHoughTransformer[i];}
b1886074	76	AliL3TrackArray *GetTracks(Int_t i) {if(!fTracks[i]) return 0; return fTracks[i];}
	77	AliL3HoughEval *GetEval(Int_t i) {if(!fEval[i]) return 0; return fEval[i];}
	78	AliL3HoughMerger *GetMerger() {if(!fMerger) return 0; return fMerger;}
	79	AliL3HoughIntMerger *GetInterMerger() {if(!fInterMerger) return 0; return fInterMerger;}
95a00d93	80	#ifdef use_aliroot
b1886074	81	AliL3FileHandler *GetMemHandler(Int_t i) {if(!fMemHandler[i]) return 0; return fMemHandler[i];}
95a00d93	82	#else
	83	AliL3MemHandler *GetMemHandler(Int_t i) {if(!fMemHandler[i]) return 0; return fMemHandler[i];}
	84	#endif
f000f8a5	85
b1886074	86	ClassDef(AliL3Hough,1) //Hough transform base class
f000f8a5	87
	88	};
	89
	90	#endif