HLT/hough/AliL3Hough.h

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