]>
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 | |
5a31e9df | 46 | |
47 | //parameters for the peak finder: | |
48 | Int_t fKappaSpread; | |
49 | Float_t fPeakRatio; | |
636080ea | 50 | |
95a00d93 | 51 | AliL3MemHandler **fMemHandler; //! |
237d3f5c | 52 | AliL3HoughBaseTransformer **fHoughTransformer; //! |
a6008206 | 53 | AliL3HoughEval **fEval; //! |
54 | AliL3HoughMaxFinder *fPeakFinder; //! | |
b1886074 | 55 | AliL3TrackArray **fTracks; //! |
3e87ef69 | 56 | AliL3TrackArray *fGlobalTracks; //! |
b1886074 | 57 | AliL3HoughMerger *fMerger; //! |
58 | AliL3HoughIntMerger *fInterMerger; //! | |
1c404dd5 | 59 | AliL3HoughGlobalMerger *fGlobalMerger; //! |
3e87ef69 | 60 | AliL3Benchmark *fBenchmark; //! |
1c404dd5 | 61 | |
b1886074 | 62 | void CleanUp(); |
0309a5ee | 63 | Double_t GetCpuTime(); |
4fc9a6a4 | 64 | |
f000f8a5 | 65 | public: |
4fc9a6a4 | 66 | |
f000f8a5 | 67 | AliL3Hough(); |
b2a02bce | 68 | 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 | 69 | virtual ~AliL3Hough(); |
70 | ||
b2a02bce | 71 | 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 | 72 | void Init(Bool_t doit=kFALSE, Bool_t addhists=kFALSE); |
73 | ||
a6008206 | 74 | void Process(Int_t minslice,Int_t maxslice); |
208b54c5 | 75 | void ReadData(Int_t slice,Int_t eventnr=0); |
3e87ef69 | 76 | void Transform(Int_t *row_range = 0); |
b1886074 | 77 | void ProcessSliceIter(); |
78 | void ProcessPatchIter(Int_t patch); | |
79 | void MergePatches(); | |
80 | void MergeInternally(); | |
44c7f8de | 81 | void MergeEtaSlices(); |
82 | ||
a6008206 | 83 | void FindTrackCandidates(); |
a6008206 | 84 | void AddAllHistograms(); |
0309a5ee | 85 | Int_t Evaluate(Int_t road_width=1,Int_t nrowstomiss=1); |
a4639de2 | 86 | void EvaluatePatch(Int_t i,Int_t road_width,Int_t nrowstomiss); |
3fe49b5b | 87 | void WriteTracks(Int_t slice,Char_t *path="./"); |
b2a02bce | 88 | void WriteTracks(Char_t *path); |
a6008206 | 89 | void WriteDigits(Char_t *outfile="output_digits.root"); |
3fe49b5b | 90 | void InitEvaluate(); |
3e87ef69 | 91 | void DoBench(Char_t *filename); |
b2a02bce | 92 | void AddTracks(); |
3fe49b5b | 93 | |
4fc9a6a4 | 94 | //Setters |
95 | void SetNEtaSegments(Int_t i) {fNEtaSegments = i;} | |
a6008206 | 96 | void SetAddHistograms() {fAddHistograms = kTRUE;} |
b1886074 | 97 | void DoIterative() {fDoIterative = kTRUE;} |
a6008206 | 98 | void SetWriteDigits() {fWriteDigits = kTRUE;} |
b2a02bce | 99 | void SetTransformerParams(Float_t ptres=0,Float_t ptmin=0,Float_t ptmax=0,Int_t ny=0,Int_t patch=-1); |
100 | //{fPtRes=ptres;fNBinY=ny;fLowPt=ptmin;fUpperPt=ptmax;fPhi=psi;} | |
101 | void SetTransformerParams(Int_t nx,Int_t ny,Float_t lpt,Int_t patch); | |
102 | void SetTransformerParams(Int_t nx,Int_t ny,Float_t lpt,Float_t phi); | |
103 | //{fNBinX=nx;fNBinY=ny;fLowPt=lpt;fPhi=phi;} | |
104 | void SetThreshold(Int_t t=3,Int_t patch=-1); | |
636080ea | 105 | void SetNSaveIterations(Int_t t=10) {fNSaveIterations=t;} |
b2a02bce | 106 | void SetPeakThreshold(Int_t threshold=0,Int_t patch=-1); |
3e87ef69 | 107 | |
5a31e9df | 108 | void SetPeakParameters(Int_t kspread,Float_t pratio) {fKappaSpread=kspread; fPeakRatio=pratio;} |
109 | ||
4fc9a6a4 | 110 | //Getters |
237d3f5c | 111 | AliL3HoughBaseTransformer *GetTransformer(Int_t i) {if(!fHoughTransformer[i]) return 0; return fHoughTransformer[i];} |
b1886074 | 112 | AliL3TrackArray *GetTracks(Int_t i) {if(!fTracks[i]) return 0; return fTracks[i];} |
113 | AliL3HoughEval *GetEval(Int_t i) {if(!fEval[i]) return 0; return fEval[i];} | |
114 | AliL3HoughMerger *GetMerger() {if(!fMerger) return 0; return fMerger;} | |
115 | AliL3HoughIntMerger *GetInterMerger() {if(!fInterMerger) return 0; return fInterMerger;} | |
95a00d93 | 116 | AliL3MemHandler *GetMemHandler(Int_t i) {if(!fMemHandler[i]) return 0; return fMemHandler[i];} |
208b54c5 | 117 | AliL3HoughMaxFinder *GetMaxFinder() {return fPeakFinder;} |
f000f8a5 | 118 | |
b1886074 | 119 | ClassDef(AliL3Hough,1) //Hough transform base class |
f000f8a5 | 120 | }; |
121 | ||
122 | #endif | |
6dbc57b4 | 123 | |
124 | ||
125 | ||
126 | ||
127 | ||
128 | ||
129 |