]>
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; |
11 | class AliL3HoughEval; | |
f80b98cb | 12 | class AliL3TrackArray; |
b1886074 | 13 | class AliL3HoughMerger; |
14 | class AliL3HoughIntMerger; | |
1c404dd5 | 15 | class AliL3HoughGlobalMerger; |
f000f8a5 | 16 | |
b1886074 | 17 | class AliL3Hough { |
f000f8a5 | 18 | |
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; |
d96f6a4a | 25 | Bool_t fUse8bits; |
f80b98cb | 26 | Int_t fNEtaSegments; |
1c404dd5 | 27 | Int_t fNPatches; |
b46b53c1 | 28 | Int_t fversion; |
3bb06991 | 29 | Int_t fCurrentSlice; |
30 | ||
95a00d93 | 31 | AliL3MemHandler **fMemHandler; //! |
237d3f5c | 32 | AliL3HoughBaseTransformer **fHoughTransformer; //! |
a6008206 | 33 | AliL3HoughEval **fEval; //! |
34 | AliL3HoughMaxFinder *fPeakFinder; //! | |
b1886074 | 35 | AliL3TrackArray **fTracks; //! |
36 | AliL3HoughMerger *fMerger; //! | |
37 | AliL3HoughIntMerger *fInterMerger; //! | |
1c404dd5 | 38 | AliL3HoughGlobalMerger *fGlobalMerger; //! |
39 | ||
b1886074 | 40 | void CleanUp(); |
0309a5ee | 41 | Double_t GetCpuTime(); |
4fc9a6a4 | 42 | |
f000f8a5 | 43 | public: |
4fc9a6a4 | 44 | |
f000f8a5 | 45 | AliL3Hough(); |
b46b53c1 | 46 | AliL3Hough(Char_t *path,Bool_t binary,Int_t n_eta_segments=100,Int_t tversion=0); |
f000f8a5 | 47 | virtual ~AliL3Hough(); |
48 | ||
60a3d829 | 49 | void Init(Char_t *path,Bool_t binary,Int_t n_eta_segments,Bool_t bit8=kFALSE,Int_t tv=0); |
a6008206 | 50 | void Process(Int_t minslice,Int_t maxslice); |
208b54c5 | 51 | void ReadData(Int_t slice,Int_t eventnr=0); |
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(); | |
3fe49b5b | 57 | |
a6008206 | 58 | void FindTrackCandidates(); |
a6008206 | 59 | void AddAllHistograms(); |
0309a5ee | 60 | Int_t Evaluate(Int_t road_width=1,Int_t nrowstomiss=1); |
b1886074 | 61 | void EvaluateWithEta(); |
3fe49b5b | 62 | void WriteTracks(Int_t slice,Char_t *path="./"); |
a6008206 | 63 | void WriteDigits(Char_t *outfile="output_digits.root"); |
3fe49b5b | 64 | void InitEvaluate(); |
65 | ||
4fc9a6a4 | 66 | //Setters |
67 | void SetNEtaSegments(Int_t i) {fNEtaSegments = i;} | |
a6008206 | 68 | void SetAddHistograms() {fAddHistograms = kTRUE;} |
b1886074 | 69 | void DoIterative() {fDoIterative = kTRUE;} |
a6008206 | 70 | void SetWriteDigits() {fWriteDigits = kTRUE;} |
f80b98cb | 71 | |
4fc9a6a4 | 72 | //Getters |
237d3f5c | 73 | AliL3HoughBaseTransformer *GetTransformer(Int_t i) {if(!fHoughTransformer[i]) return 0; return fHoughTransformer[i];} |
b1886074 | 74 | AliL3TrackArray *GetTracks(Int_t i) {if(!fTracks[i]) return 0; return fTracks[i];} |
75 | AliL3HoughEval *GetEval(Int_t i) {if(!fEval[i]) return 0; return fEval[i];} | |
76 | AliL3HoughMerger *GetMerger() {if(!fMerger) return 0; return fMerger;} | |
77 | AliL3HoughIntMerger *GetInterMerger() {if(!fInterMerger) return 0; return fInterMerger;} | |
95a00d93 | 78 | AliL3MemHandler *GetMemHandler(Int_t i) {if(!fMemHandler[i]) return 0; return fMemHandler[i];} |
208b54c5 | 79 | AliL3HoughMaxFinder *GetMaxFinder() {return fPeakFinder;} |
f000f8a5 | 80 | |
b1886074 | 81 | ClassDef(AliL3Hough,1) //Hough transform base class |
f000f8a5 | 82 | }; |
83 | ||
84 | #endif |