Add run number in ntuple (Gian Michele)
[u/mrichter/AliRoot.git] / TRD / AliTRDcluster.h
CommitLineData
46d29e70 1#ifndef ALITRDCLUSTER_H
2#define ALITRDCLUSTER_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
6d50f529 8///////////////////////////////////////////////////////////////////////////////
9// //
10// TRD cluster //
11// //
12///////////////////////////////////////////////////////////////////////////////
5443e65e 13
5843e420 14#include "AliCluster.h"
46d29e70 15
a5b99acd 16class AliTRDtrackletWord;
17
5443e65e 18class AliTRDcluster : public AliCluster {
3044dfe5 19 friend class AliHLTTRDCluster;
9dcc64cc 20 friend class AliTRDtrackletOflHelper;
762571f0 21
3044dfe5 22public:
5843e420 23 enum ETRDclusterStatus {
24 kInChamber = BIT(16) // Out of fiducial volume of chamber (signal tails)
25 ,kFivePad = BIT(17) // Deconvoluted clusters
b72f4eaf 26 ,kLUT = BIT(18)
27 ,kGAUS = BIT(19)
28 ,kCOG = BIT(20)
8fc736d7 29 ,kXcorr = BIT(21) // steer efficient vd corrections
5843e420 30 };
31 enum ETRDclusterMask {
32 kMaskedLeft = 0
33 ,kMaskedCenter = 1
34 ,kMaskedRight = 2
f5375dcb 35 };
0e09df31 36
bdb68f8c 37 AliTRDcluster();
b72f4eaf 38 AliTRDcluster(Int_t det, UChar_t col, UChar_t row, UChar_t time, const Short_t *sig, UShort_t volid);
04e58504 39 AliTRDcluster(Int_t det, Float_t q, const Float_t *pos, const Float_t *sig
40 , const Int_t *tracks, Char_t npads, Short_t * const signals
762571f0 41 , UChar_t col, UChar_t row, UChar_t time
42 , Char_t timebin, Float_t center, UShort_t volid);
a5b99acd 43 AliTRDcluster(const AliTRDtrackletWord *const tracklet, Int_t det, UShort_t volid);
a2b90f83 44 AliTRDcluster(const AliTRDcluster &c);
acc49af9 45 virtual ~AliTRDcluster() {};
762571f0 46 AliTRDcluster &operator=(const AliTRDcluster &c);
a2b90f83 47
762571f0 48 virtual void AddTrackIndex(const Int_t * const i);
49 void Clear(Option_t *o="");
203967fc 50
762571f0 51 Bool_t IsEqual(const TObject *otherCluster) const;
52 Bool_t IsInChamber() const { return TestBit(kInChamber); }
53 Bool_t IsMasked() const { return fClusterMasking ? kTRUE : kFALSE; }
54 Bool_t IsShared() const { return IsClusterShared(); }
55 Bool_t IsUsed() const { return IsClusterUsed(); }
56 Bool_t IsFivePad() const { return TestBit(kFivePad); }
57 inline Bool_t IsRPhiMethod(ETRDclusterStatus m) const;
8fc736d7 58 Bool_t HasXcorr() const { return TestBit(kXcorr); }
762571f0 59
60 UChar_t GetPadMaskedPosition() const { return fClusterMasking & 7; }
61 UChar_t GetPadMaskedStatus() const { return fClusterMasking >> 3; }
62 Int_t GetDetector() const { return fDetector; }
63 Int_t GetLocalTimeBin() const { return fLocalTimeBin; }
64 Float_t GetQ() const { return fQ; }
65 Int_t GetNPads() const { return fNPads; }
66 Float_t GetCenter() const { return fCenter; }
67 Int_t GetPadCol() const { return fPadCol; }
68 Int_t GetPadRow() const { return fPadRow; }
69 Int_t GetPadTime() const { return fPadTime; }
70 Short_t *GetSignals() { return fSignals; }
71 Float_t GetSumS() const;
72
73 static Double_t GetSX(Int_t tb, Double_t z=-1);
74 static Double_t GetSYdrift(Int_t tb, Int_t ly=0, Double_t z=-1);
75 static Double_t GetSYcharge(Float_t q);
76 static Double_t GetSYprf(Int_t ly, Double_t center, Double_t s2);
77 static Double_t GetXcorr(Int_t tb, Double_t z=-1);
78 static Double_t GetYcorr(Int_t ly, Float_t y);
79 Float_t GetXloc(Double_t t0, Double_t vd
80 , const Double_t *const q = 0x0
81 , const Double_t *const xq = 0x0
82 , Double_t z = 0.2);
83 Float_t GetYloc(Double_t y0, Double_t s2, Double_t W, Double_t *const y1=0x0, Double_t *const y2=0x0);
84
85 void Print(Option_t* o="") const;
86
87 void SetLocalTimeBin(Char_t t) { fLocalTimeBin = t; }
88 void SetNPads(Int_t n) { fNPads = n; }
89 void SetPadCol(UChar_t inPadCol) { fPadCol = inPadCol; }
90 void SetPadRow(UChar_t inPadRow) { fPadRow = inPadRow; }
91 void SetPadTime(UChar_t inPadTime) { fPadTime = inPadTime; }
92 void SetDetector(Short_t inDetector) { fDetector = inDetector; }
93 void SetQ(Float_t inQ) { fQ = inQ; }
94 void SetClusterMasking(UChar_t inClusterMasking) { fClusterMasking = inClusterMasking; }
95 void SetShared(Bool_t sh = kTRUE) { SetBit(AliCluster::kShared,sh); }
96 void SetFivePad(Bool_t b = kTRUE) { SetBit(kFivePad,b); }
97 void SetInChamber(Bool_t in = kTRUE) { SetBit(kInChamber,in); }
98 void SetPadMaskedPosition(UChar_t position);
99 void SetPadMaskedStatus(UChar_t status);
100 void SetSigmaY2(Float_t s2, Float_t dt, Float_t exb, Float_t x0, Float_t z=-1., Float_t tgp=0.);
b72f4eaf 101 inline void SetRPhiMethod(ETRDclusterStatus m);
8fc736d7 102 void SetXcorr(Bool_t xc = kTRUE) { SetBit(kXcorr,xc); }
762571f0 103
104 void Use(Int_t u = 1) { SetBit(AliCluster::kUsed, u ? kTRUE : kFALSE); }
af26ce80 105
5843e420 106protected:
762571f0 107
5843e420 108 UChar_t fPadCol; // Central pad number in column direction
109 UChar_t fPadRow; // Central pad number in row direction
110 UChar_t fPadTime; // Uncalibrated time bin number
111 Char_t fLocalTimeBin; // T0-calibrated time bin number
112 UChar_t fNPads; // Number of pads in cluster
113 UChar_t fClusterMasking; // Bit field containing cluster status information;
114 Short_t fDetector; // TRD detector number
115 Short_t fSignals[7]; // Signals in the cluster
116 Float_t fQ; // Amplitude
117 Float_t fCenter; // Center of the cluster relative to the pad
0e09df31 118
9dcc64cc 119 inline void Update(Short_t adc[7]);
120
b72f4eaf 121private:
762571f0 122
123 Float_t GetDYcog(const Double_t *const y1=0x0, const Double_t *const y2=0x0);
124 Float_t GetDYlut(const Double_t *const y1=0x0, const Double_t *const y2=0x0);
125 Float_t GetDYgauss(Double_t sw, Double_t *const y1=0x0, Double_t *const y2=0x0);
b72f4eaf 126 static void FillLUT();
127
128 static const Int_t fgkNlut; //! Number of bins of the LUT
129 static Double_t *fgLUT; //! The lookup table
130
762571f0 131 ClassDef(AliTRDcluster, 7) // Cluster for the TRD
132
46d29e70 133};
b72f4eaf 134
9dcc64cc 135//___________________________________________________
136inline void AliTRDcluster::Update(Short_t adc[7])
137{
138 memcpy(fSignals, adc, 7*sizeof(Short_t));
139 fQ = Float_t(adc[2]+adc[3]+adc[4]);
140 if(IsRPhiMethod(AliTRDcluster::kLUT)) GetDYlut();
141 else GetDYcog();
142}
143
b72f4eaf 144//________________________________________________
145inline Bool_t AliTRDcluster::IsRPhiMethod(ETRDclusterStatus m) const
146{
147 if(m==kLUT && TestBit(kLUT)) return kTRUE;
148 else if(m==kGAUS && TestBit(kGAUS)) return kTRUE;
149 else if(m==kCOG && (!TestBit(kLUT)&&!TestBit(kGAUS))) return kTRUE;
150
151 return kFALSE;
152}
153
154//________________________________________________
155inline void AliTRDcluster::SetRPhiMethod(ETRDclusterStatus m)
156{
a8d5c885 157 SetBit(kCOG,0);SetBit(kLUT,0);SetBit(kGAUS,0);
b72f4eaf 158 switch(m){
159 case kCOG: SetBit(kCOG); break;
160 case kLUT: SetBit(kLUT); break;
161 case kGAUS: SetBit(kGAUS); break;
a8d5c885 162 default: SetBit(kLUT); break;
b72f4eaf 163 }
164}
165
46d29e70 166#endif