Moving lib*.pkg

[u/mrichter/AliRoot.git] / HLT / hough / AliL3Hough.h

diff --git a/HLT/hough/AliL3Hough.h b/HLT/hough/AliL3Hough.h
index 86d79829993f3207a03eb17dbfa0df6965389a7a..519e3a0b8496e65b4332455b6c9b903f985059cd 100644 (file)
--- a/HLT/hough/AliL3Hough.h
+++ b/HLT/hough/AliL3Hough.h
@@ -1,49 +1,132 @@
-#ifndef ALIL3_HOUGH
-#define ALIL3_HOUGH
+// @(#) $Id$
+
+#ifndef ALIL3HOUGH
+#define ALIL3HOUGH
 
 #include "AliL3RootTypes.h"
 
 class AliL3HoughMaxFinder;
-class AliL3HoughTransformer;
+class AliL3HoughBaseTransformer;
 class AliL3Histogram;
+class AliL3MemHandler;
 class AliL3FileHandler;
 class AliL3HoughEval;
-class AliL3Transform;
 class AliL3TrackArray;
-class TFile;
+class AliL3HoughMerger;
+class AliL3HoughIntMerger;
+class AliL3HoughGlobalMerger;
+class AliL3Benchmark;
 
-class AliL3Hough : public TObject {
+class AliL3Hough {
   
  private:
-  Char_t fPath[256];
+  Char_t *fInputFile;//!
+  Char_t *fInputPtr;//!
+  Char_t fPath[1024];
   Bool_t fBinary;
+  Bool_t fAddHistograms;
+  Bool_t fDoIterative;
+  Bool_t fWriteDigits;
+  Bool_t fUse8bits;
   Int_t fNEtaSegments;
-  AliL3FileHandler **fMemHandler; //!
-  AliL3HoughTransformer **fHoughTransformer; //!
-  TFile *fRootFile; //!
+  Int_t fNPatches;
+  Int_t fVersion; //which HoughTransformer to use
+  Int_t fCurrentSlice;
+  Int_t fEvent;
+
+  Int_t fPeakThreshold[6];
+  Float_t fLowPt[6];
+  Float_t fUpperPt[6];
+  Float_t fPtRes[6];
+  Float_t fPhi[6];
+  Int_t fNBinX[6];
+  Int_t fNBinY[6];
+  Int_t fThreshold[6];
+  Int_t fNSaveIterations; //for HoughtransformerVhdl
+  
+  //parameters for the peak finder:
+  Int_t fKappaSpread;
+  Float_t fPeakRatio;
+
+  Float_t fZVertex;
 
-  void DeleteTransformers();
-  void DeleteMemory();
-  void Init();
+  AliL3MemHandler **fMemHandler; //!
+  AliL3HoughBaseTransformer **fHoughTransformer; //!
+  AliL3HoughEval **fEval; //!
+  AliL3HoughMaxFinder *fPeakFinder; //!
+  AliL3TrackArray **fTracks; //!
+  AliL3TrackArray *fGlobalTracks; //!
+  AliL3HoughMerger *fMerger; //!
+  AliL3HoughIntMerger *fInterMerger; //!
+  AliL3HoughGlobalMerger *fGlobalMerger; //!
+  AliL3Benchmark *fBenchmark; //!
+
+  void CleanUp();
+  Double_t GetCpuTime();
   
  public:
   
   AliL3Hough(); 
-  AliL3Hough(Char_t *path,Bool_t binary,Int_t n_eta_segments=100);
+  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,Char_t *ptr=0);
   virtual ~AliL3Hough();
   
-  void TransformSlice(Int_t slice);
-  AliL3Histogram *AddHistograms();
-  void Evaluate(AliL3Histogram *hist);
+  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,Char_t *ptr=0,Float_t zvertex=0.0);
+  void Init(Bool_t doit=kFALSE, Bool_t addhists=kFALSE);
+
+  void Process(Int_t minslice,Int_t maxslice);
+  void ReadData(Int_t slice,Int_t eventnr=0);
+  void Transform(Int_t *row_range = 0);
+  void ProcessSliceIter();
+  void ProcessPatchIter(Int_t patch);
+  void MergePatches();
+  void MergeInternally();
+  void MergeEtaSlices();
 
+  void FindTrackCandidates();
+  void AddAllHistograms();
+  void AddAllHistogramsRows();
+  Int_t Evaluate(Int_t road_width=1,Int_t nrowstomiss=1);
+  void EvaluatePatch(Int_t i,Int_t road_width,Int_t nrowstomiss);
+  void WriteTracks(Int_t slice,Char_t *path="./");
+  void WriteTracks(Char_t *path);
+  void WriteDigits(Char_t *outfile="output_digits.root");
+  void InitEvaluate();
+  void DoBench(Char_t *filename);
+  void AddTracks();
+  
   //Setters
   void SetNEtaSegments(Int_t i) {fNEtaSegments = i;}
+  void SetAddHistograms() {fAddHistograms = kTRUE;}
+  void DoIterative() {fDoIterative = kTRUE;}
+  void SetWriteDigits() {fWriteDigits = kTRUE;}
+  void SetTransformerParams(Float_t ptres=0,Float_t ptmin=0,Float_t ptmax=0,Int_t ny=0,Int_t patch=-1);
+  //{fPtRes=ptres;fNBinY=ny;fLowPt=ptmin;fUpperPt=ptmax;fPhi=psi;}
+  void SetTransformerParams(Int_t nx,Int_t ny,Float_t lpt,Int_t patch);
+  void SetTransformerParams(Int_t nx,Int_t ny,Float_t lpt,Float_t phi);
+  //{fNBinX=nx;fNBinY=ny;fLowPt=lpt;fPhi=phi;}
+  void SetThreshold(Int_t t=3,Int_t patch=-1);
+  void SetNSaveIterations(Int_t t=10) {fNSaveIterations=t;}
+  void SetPeakThreshold(Int_t threshold=0,Int_t patch=-1);
+  
+  void SetPeakParameters(Int_t kspread,Float_t pratio) {fKappaSpread=kspread; fPeakRatio=pratio;}
   
   //Getters
-  AliL3HoughTransformer *GetTransformer(Int_t i) {if(!fHoughTransformer[i]) return 0; return fHoughTransformer[i];}
-
-  ClassDef(AliL3Hough,1)
+  AliL3HoughBaseTransformer *GetTransformer(Int_t i) {if(!fHoughTransformer[i]) return 0; return fHoughTransformer[i];}
+  AliL3TrackArray *GetTracks(Int_t i) {if(!fTracks[i]) return 0; return fTracks[i];}
+  AliL3HoughEval *GetEval(Int_t i) {if(!fEval[i]) return 0; return fEval[i];}
+  AliL3HoughMerger *GetMerger() {if(!fMerger) return 0; return fMerger;}
+  AliL3HoughIntMerger *GetInterMerger() {if(!fInterMerger) return 0; return fInterMerger;}
+  AliL3MemHandler *GetMemHandler(Int_t i) {if(!fMemHandler[i]) return 0; return fMemHandler[i];}
+  AliL3HoughMaxFinder *GetMaxFinder() {return fPeakFinder;}
 
+  ClassDef(AliL3Hough,1) //Hough transform base class
 };
 
 #endif
+
+
+
+
+
+
+