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