Moved from AliTransbit to AliL3Transbit.
[u/mrichter/AliRoot.git] / HLT / comp / AliL3Modeller.h
1 #ifndef AliL3_Modeller
2 #define AliL3_Modeller
3
4
5 #include "AliL3RootTypes.h"
6
7 class AliL3TrackArray;
8 class AliL3MemHandler;
9 class AliL3DigitRowData;
10 class AliL3ModelTrack;
11
12 struct Cluster {
13   UInt_t fCharge;
14   UInt_t fPad;
15   UInt_t fTime;
16   UInt_t fSigmaY2;
17   UInt_t fSigmaZ2;
18 };
19
20 struct Digit {
21   Short_t fCharge;
22   Bool_t fUsed;
23 };
24
25 class AliL3Modeller {
26   
27  private:
28   
29   AliL3TrackArray *fTracks; //!
30   AliL3MemHandler *fMemHandler; //!
31   AliL3DigitRowData *fRowData;//!
32   
33   Int_t fNClusters;
34   Int_t fMaxClusters;
35   Int_t fCurrentPadRow;
36   
37   Float_t fPadOverlap;
38   Float_t fTimeOverlap;
39   Int_t fTrackThreshold; //minimum weigth track need in order to be included.(=Nhits/weight)
40   
41   Int_t fSlice;
42   Int_t fPatch;
43   Char_t fPath[100];
44
45   void FillCluster(AliL3ModelTrack *track,Cluster *cluster,Int_t row,Int_t npads);
46   void CalcClusterWidth(Cluster *cl,Float_t &sigmaY2,Float_t &sigmaZ2);
47   void FillZeros(AliL3DigitRowData *digPt,Digit *row);
48 #ifdef do_mc
49   void GetTrackID(Int_t pad,Int_t time,Int_t *trackID);
50 #endif
51     
52  public:
53   
54   AliL3Modeller();
55   virtual ~AliL3Modeller();
56   
57   void Init(Int_t slice,Int_t patch,Char_t *trackdata,Char_t *path,Bool_t houghtracks,Bool_t binary=kTRUE);
58   void FindClusters();
59   void CheckForOverlaps();
60   void CalculateCrossingPoints();
61   void WriteRemaining();
62   
63   void SetInputData(AliL3DigitRowData *digits) {fRowData = digits;}
64   void SetTrackThreshold(Int_t i) {fTrackThreshold=i;}
65
66   AliL3TrackArray *GetTracks() {return fTracks;}
67     
68   ClassDef(AliL3Modeller,1) //Modeller class
69     
70 };
71
72 #endif