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