]> git.uio.no Git - u/mrichter/AliRoot.git/blob - TRD/AliTRDclusterizer.h
- prepare train for merging with TPC.
[u/mrichter/AliRoot.git] / TRD / AliTRDclusterizer.h
1 #ifndef ALITRDCLUSTERIZER_H
2 #define ALITRDCLUSTERIZER_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 ////////////////////////////////////////////////////////////////////////////
9 //                                                                        //
10 //  TRD cluster finder                                                    //
11 //                                                                        //
12 ////////////////////////////////////////////////////////////////////////////
13
14 #include <TNamed.h>
15
16 class TFile;
17 class TTree;
18 class TClonesArray;
19
20 class AliRunLoader;
21 class AliRawReader;
22
23 class AliTRD;
24 class AliTRDcluster;
25
26 class AliTRDarrayADC;
27 class AliTRDarraySignal;
28 class AliTRDdigitsManager;
29 class AliTRDSignalIndex;
30 class AliTRDtransform;
31 class AliTRDCalROC;
32 class AliTRDReconstructor;
33 class AliTRDCalSingleChamberStatus;
34
35 class AliTRDclusterizer : public TNamed 
36 {
37
38  public:
39
40   // steering flags
41   enum{
42     kOwner  = BIT(14)  //  toggle cluster ownership
43    ,kLabels = BIT(15)  //  toggle MC labels for clusters
44    ,kHLT    = BIT(16)  //  HLT mode
45    ,kLUT    = BIT(17)  //  using look up table for cluster's r-phi position
46    ,kGAUS   = BIT(18)  //  using gauss approx. for cluster's r-phi position
47   };
48
49   struct MaxStruct
50   {
51     Int_t       Row;
52     Int_t       Col;
53     Int_t       Time;
54     UChar_t     padStatus;
55     Short_t     Signals[3];
56     Bool_t      FivePad;
57   MaxStruct():Row(-1),Col(0),Time(0),padStatus(0),FivePad(kFALSE)
58       {memset(Signals, 0, 3*sizeof(Short_t));}
59     MaxStruct &operator=(const MaxStruct &a)
60     {Row=a.Row; Col=a.Col; Time=a.Time; padStatus=a.padStatus; FivePad=a.FivePad;
61      memcpy(Signals, a.Signals, 3*sizeof(Short_t)); return *this;}
62   };
63   
64   AliTRDclusterizer(const AliTRDReconstructor *const rec = 0x0);
65   AliTRDclusterizer(const Text_t* name, const Text_t* title, const AliTRDReconstructor *const rec = 0x0);
66   AliTRDclusterizer(const AliTRDclusterizer &c);
67   virtual         ~AliTRDclusterizer();
68   AliTRDclusterizer &operator=(const AliTRDclusterizer &c);
69
70   void     Copy(TObject &c) const;
71
72   Bool_t   Open(const Char_t *name, Int_t nEvent = 0);
73   Bool_t   OpenInput(Int_t nEvent = 0);
74   Bool_t   OpenOutput();
75   Bool_t   OpenOutput(TTree *clusterTree);
76
77   Bool_t   ReadDigits();
78   Bool_t   ReadDigits(AliRawReader *rawReader);
79   Bool_t   ReadDigits(TTree *digitsTree);
80
81   Bool_t   WriteClusters(Int_t det);
82   void     ResetRecPoints();
83   virtual TClonesArray    *RecPoints();
84   Bool_t   WriteTracklets(Int_t det);
85
86   Bool_t   Raw2Clusters(AliRawReader *rawReader);
87   Bool_t   Raw2ClustersChamber(AliRawReader *rawReader);
88
89   Bool_t   MakeClusters();
90   Bool_t   MakeClusters(Int_t det);
91
92   Bool_t   AddLabels();
93   Bool_t   SetUseLabels(const Bool_t kset) { SetBit(kLabels, kset); return TestBit(kLabels);  } // should we assign labels to clusters
94   void     SetRawVersion(const Int_t iver) { fRawVersion = iver; } // set the expected raw data version
95   void             SetReconstructor(const AliTRDReconstructor *rec) {fReconstructor = rec;}
96   static UChar_t   GetStatus(Short_t &signal);
97   Int_t            GetAddedClusters() {return fNoOfClusters;}
98
99   Bool_t   IsClustersOwner() const {return TestBit(kOwner);}
100   virtual void     SetClustersOwner(Bool_t own=kTRUE) {SetBit(kOwner, own); if(!own) {fRecPoints = 0x0; fNoOfClusters=0;} }
101
102 protected:
103
104   void             DeConvExp (const Double_t *const source, Double_t *const target
105                              ,const Int_t nTimeTotal, const Int_t nexp);
106   void             TailCancelation();
107
108   Float_t  Unfold(Double_t eps, Int_t layer, Double_t *padSignal) const;
109   //Double_t GetCOG(Double_t signal[5]) const; 
110   //Double_t LUTposition(Int_t ilayer, Double_t ampL, Double_t ampC, Double_t ampR) const;
111   
112   void             SetPadStatus(const UChar_t status, UChar_t &encoding);
113   UChar_t          GetPadStatus(UChar_t encoding) const;
114   Int_t            GetCorruption(UChar_t encoding) const;
115
116   Bool_t           IsMaximum(const MaxStruct &Max, UChar_t &padStatus, Short_t *const Signals);       //for const correctness reasons not const parameters are given separately
117   Bool_t           FivePadCluster(MaxStruct &ThisMax, MaxStruct &NeighbourMax);
118   void             CreateCluster(const MaxStruct &Max); 
119   inline void      CalcAdditionalInfo(const MaxStruct &Max, Short_t *const signals, Int_t &nPadCount);
120   virtual void     AddClusterToArray(AliTRDcluster *cluster);
121
122   const AliTRDReconstructor *fReconstructor; //! reconstructor
123   AliRunLoader        *fRunLoader;           //! Run Loader
124   TTree               *fClusterTree;         //! Tree with the cluster
125   TClonesArray        *fRecPoints;           //! Array of clusters
126
127   TTree               *fTrackletTree;        //! Tree for tracklets
128
129   AliTRDdigitsManager *fDigitsManager;       //! TRD digits manager
130
131   UInt_t              **fTrackletContainer;  //! tracklet container
132
133   Int_t                fRawVersion;          //  Expected raw version of the data - default is 2
134
135   AliTRDtransform     *fTransform;           //! Transforms the reconstructed space points
136
137   AliTRDarrayADC      *fDigits;               // Array holding the digits
138   AliTRDSignalIndex   *fIndexes;              // Array holding the indexes to the digits
139   Float_t              fADCthresh;            // ADC thresholds: There is no ADC threshold anymore, and simParam should not be used in clusterizer. KO
140   Float_t              fMaxThresh;            // Threshold value for the maximum
141   Float_t              fSigThresh;            // Threshold value for the digit signal
142   Float_t              fMinMaxCutSigma;       // Threshold value for the maximum (cut noise)
143   Float_t              fMinLeftRightCutSigma; // Threshold value for the sum pad (cut noise)
144   Int_t                fLayer;                // Current layer of the detector
145   Int_t                fDet;                  // Current detecor
146   UShort_t             fVolid;                // Volume ID
147   Int_t                fColMax;               // Number of Colums in one detector
148   Int_t                fTimeTotal;            // Number of time bins
149   AliTRDCalROC        *fCalGainFactorROC;     // Calibration object with pad wise values for the gain factors
150   Float_t              fCalGainFactorDetValue;// Calibration value for chamber wise noise
151   AliTRDCalROC        *fCalNoiseROC;          // Calibration object with pad wise values for the noise
152   Float_t              fCalNoiseDetValue;     // Calibration value for chamber wise noise
153   AliTRDCalSingleChamberStatus *fCalPadStatusROC; // Calibration object with the pad status
154   Int_t                fClusterROC;           // The index to the first cluster of a given ROC
155   Int_t                firstClusterROC;       // The number of cluster in a given ROC
156   Int_t                fNoOfClusters;         // Number of Clusters already processed and still owned by the clusterizer
157
158   ClassDef(AliTRDclusterizer,6)               //  TRD clusterfinder
159
160 };
161
162 #endif