]>
Commit | Line | Data |
---|---|---|
3e87ef69 | 1 | // @(#) $Id$ |
2 | ||
b2a02bce | 3 | #ifndef ALIL3HOUGH |
4 | #define ALIL3HOUGH | |
f000f8a5 | 5 | |
6 | #include "AliL3RootTypes.h" | |
7 | ||
8 | class AliL3HoughMaxFinder; | |
237d3f5c | 9 | class AliL3HoughBaseTransformer; |
f80b98cb | 10 | class AliL3Histogram; |
95a00d93 | 11 | class AliL3MemHandler; |
f80b98cb | 12 | class AliL3FileHandler; |
13 | class AliL3HoughEval; | |
f80b98cb | 14 | class AliL3TrackArray; |
b1886074 | 15 | class AliL3HoughMerger; |
16 | class AliL3HoughIntMerger; | |
1c404dd5 | 17 | class AliL3HoughGlobalMerger; |
3e87ef69 | 18 | class AliL3Benchmark; |
f000f8a5 | 19 | |
b1886074 | 20 | class AliL3Hough { |
f000f8a5 | 21 | |
22 | private: | |
b2a02bce | 23 | Char_t *fInputFile;//! |
24 | ||
25 | Char_t fPath[1024]; | |
4fc9a6a4 | 26 | Bool_t fBinary; |
a6008206 | 27 | Bool_t fAddHistograms; |
b1886074 | 28 | Bool_t fDoIterative; |
a6008206 | 29 | Bool_t fWriteDigits; |
d96f6a4a | 30 | Bool_t fUse8bits; |
f80b98cb | 31 | Int_t fNEtaSegments; |
1c404dd5 | 32 | Int_t fNPatches; |
636080ea | 33 | Int_t fVersion; //which HoughTransformer to use |
3bb06991 | 34 | Int_t fCurrentSlice; |
b2a02bce | 35 | Int_t fEvent; |
36 | ||
37 | Int_t fPeakThreshold[6]; | |
38 | Float_t fLowPt[6]; | |
39 | Float_t fUpperPt[6]; | |
40 | Float_t fPtRes[6]; | |
41 | Float_t fPhi[6]; | |
42 | Int_t fNBinX[6]; | |
43 | Int_t fNBinY[6]; | |
44 | Int_t fThreshold[6]; | |
45 | Int_t fNSaveIterations; //for HoughtransformerVhdl | |
636080ea | 46 | |
95a00d93 | 47 | AliL3MemHandler **fMemHandler; //! |
237d3f5c | 48 | AliL3HoughBaseTransformer **fHoughTransformer; //! |
a6008206 | 49 | AliL3HoughEval **fEval; //! |
50 | AliL3HoughMaxFinder *fPeakFinder; //! | |
b1886074 | 51 | AliL3TrackArray **fTracks; //! |
3e87ef69 | 52 | AliL3TrackArray *fGlobalTracks; //! |
b1886074 | 53 | AliL3HoughMerger *fMerger; //! |
54 | AliL3HoughIntMerger *fInterMerger; //! | |
1c404dd5 | 55 | AliL3HoughGlobalMerger *fGlobalMerger; //! |
3e87ef69 | 56 | AliL3Benchmark *fBenchmark; //! |
1c404dd5 | 57 | |
b1886074 | 58 | void CleanUp(); |
0309a5ee | 59 | Double_t GetCpuTime(); |
4fc9a6a4 | 60 | |
f000f8a5 | 61 | public: |
4fc9a6a4 | 62 | |
f000f8a5 | 63 | AliL3Hough(); |
b2a02bce | 64 | AliL3Hough(Char_t *path,Bool_t binary,Int_t n_eta_segments=100,Bool_t bit8=kFALSE,Int_t tv=0,Char_t *infile=0); |
f000f8a5 | 65 | virtual ~AliL3Hough(); |
66 | ||
b2a02bce | 67 | void Init(Char_t *path,Bool_t binary,Int_t n_eta_segments=100,Bool_t bit8=kFALSE,Int_t tv=0,Char_t *infile=0); |
6dbc57b4 | 68 | void Init(Bool_t doit=kFALSE, Bool_t addhists=kFALSE); |
69 | ||
a6008206 | 70 | void Process(Int_t minslice,Int_t maxslice); |
208b54c5 | 71 | void ReadData(Int_t slice,Int_t eventnr=0); |
3e87ef69 | 72 | void Transform(Int_t *row_range = 0); |
b1886074 | 73 | void ProcessSliceIter(); |
74 | void ProcessPatchIter(Int_t patch); | |
75 | void MergePatches(); | |
76 | void MergeInternally(); | |
44c7f8de | 77 | void MergeEtaSlices(); |
78 | ||
a6008206 | 79 | void FindTrackCandidates(); |
a6008206 | 80 | void AddAllHistograms(); |
0309a5ee | 81 | Int_t Evaluate(Int_t road_width=1,Int_t nrowstomiss=1); |
a4639de2 | 82 | void EvaluatePatch(Int_t i,Int_t road_width,Int_t nrowstomiss); |
3fe49b5b | 83 | void WriteTracks(Int_t slice,Char_t *path="./"); |
b2a02bce | 84 | void WriteTracks(Char_t *path); |
a6008206 | 85 | void WriteDigits(Char_t *outfile="output_digits.root"); |
3fe49b5b | 86 | void InitEvaluate(); |
3e87ef69 | 87 | void DoBench(Char_t *filename); |
b2a02bce | 88 | void AddTracks(); |
3fe49b5b | 89 | |
4fc9a6a4 | 90 | //Setters |
91 | void SetNEtaSegments(Int_t i) {fNEtaSegments = i;} | |
a6008206 | 92 | void SetAddHistograms() {fAddHistograms = kTRUE;} |
b1886074 | 93 | void DoIterative() {fDoIterative = kTRUE;} |
a6008206 | 94 | void SetWriteDigits() {fWriteDigits = kTRUE;} |
b2a02bce | 95 | void SetTransformerParams(Float_t ptres=0,Float_t ptmin=0,Float_t ptmax=0,Int_t ny=0,Int_t patch=-1); |
96 | //{fPtRes=ptres;fNBinY=ny;fLowPt=ptmin;fUpperPt=ptmax;fPhi=psi;} | |
97 | void SetTransformerParams(Int_t nx,Int_t ny,Float_t lpt,Int_t patch); | |
98 | void SetTransformerParams(Int_t nx,Int_t ny,Float_t lpt,Float_t phi); | |
99 | //{fNBinX=nx;fNBinY=ny;fLowPt=lpt;fPhi=phi;} | |
100 | void SetThreshold(Int_t t=3,Int_t patch=-1); | |
636080ea | 101 | void SetNSaveIterations(Int_t t=10) {fNSaveIterations=t;} |
b2a02bce | 102 | void SetPeakThreshold(Int_t threshold=0,Int_t patch=-1); |
3e87ef69 | 103 | |
4fc9a6a4 | 104 | //Getters |
237d3f5c | 105 | AliL3HoughBaseTransformer *GetTransformer(Int_t i) {if(!fHoughTransformer[i]) return 0; return fHoughTransformer[i];} |
b1886074 | 106 | AliL3TrackArray *GetTracks(Int_t i) {if(!fTracks[i]) return 0; return fTracks[i];} |
107 | AliL3HoughEval *GetEval(Int_t i) {if(!fEval[i]) return 0; return fEval[i];} | |
108 | AliL3HoughMerger *GetMerger() {if(!fMerger) return 0; return fMerger;} | |
109 | AliL3HoughIntMerger *GetInterMerger() {if(!fInterMerger) return 0; return fInterMerger;} | |
95a00d93 | 110 | AliL3MemHandler *GetMemHandler(Int_t i) {if(!fMemHandler[i]) return 0; return fMemHandler[i];} |
208b54c5 | 111 | AliL3HoughMaxFinder *GetMaxFinder() {return fPeakFinder;} |
f000f8a5 | 112 | |
b1886074 | 113 | ClassDef(AliL3Hough,1) //Hough transform base class |
f000f8a5 | 114 | }; |
115 | ||
116 | #endif | |
6dbc57b4 | 117 | |
118 | ||
119 | ||
120 | ||
121 | ||
122 | ||
123 |