TVector.h missing
[u/mrichter/AliRoot.git] / RICH / AliRICH.h
2f614988 1#ifndef AliRICH_h
2#define AliRICH_h
2e5f0f7b 3
3da30618 4/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
1c349ae5 7#include <Riostream.h>
853634d3 8#include <TClonesArray.h>
1c349ae5 9#include <TObjArray.h>
10#include <TVector.h>
d4c94996 11#include <TVector3.h>
1c349ae5 12
3582c1f9 13#include "AliRICHDigitizer.h"
cb8189e7 14#include "AliRICHParam.h"
1c349ae5 15#include <AliDetector.h>
16#include <AliDigit.h>
17#include <AliHit.h>
cb8189e7 18
c021cb15 19//__________________AliRICHhit______________________________________________________________________
853634d3 20class AliRICHhit : public AliHit
e42a7b46 23 AliRICHhit():AliHit(),fChamber(kBad),fEloss(kBad) {fInX3.SetXYZ(0,0,0);fOutX3.SetXYZ(0,0,0);}
d4c94996 24 AliRICHhit(Int_t c,Int_t tid,TVector3 in,TVector3 out,Double_t e):AliHit(0,tid)
25 {fChamber=c;fInX3=in; fOutX3=out;fEloss=e; fX=out.X();fY=out.Y();fZ=out.Z();}
26 virtual ~AliRICHhit() {;}
af3d25a6 27
d4c94996 28 Int_t C() const{return fChamber;} //chamber number
29 Int_t Chamber() const{return fChamber;} //chamber number
30 Float_t Eloss() const{return fEloss;} //energy lost by track inside amplification gap
31 TVector3 InX3() const{return fInX3;} //track position at the faceplane of the gap
32 TVector3 OutX3() const{return fOutX3;} //track position at the backplane of the gap
33 Double_t Length() const{return (fOutX3-fInX3).Mag();} //track length inside the amplification gap
34 void Print(Option_t *option="")const; //virtual
853634d3 35protected:
c021cb15 36 Int_t fChamber; //chamber number
d4c94996 37 Double_t fEloss; //ionisation energy lost in GAP
3582c1f9 38 TVector3 fInX3; //position at the entrance of the GAP
d4c94996 39 TVector3 fOutX3; //position at the exit of the GAP
ed3ceb24 40 ClassDef(AliRICHhit,2) //RICH hit class
853634d3 41};//class AliRICHhit
c021cb15 42
c021cb15 43//__________________AliRICHdigit____________________________________________________________________
853634d3 44class AliRICHdigit :public AliDigit
e42a7b46 47 AliRICHdigit():AliDigit(),fCFM(0),fChamber(0),fPadX(0),fPadY(0),fQdc(kBad){fTracks[0]=fTracks[1]=fTracks[2]=kBad;}
48 AliRICHdigit(Int_t c,TVector pad,Double_t q,Int_t cfm,Int_t tid0,Int_t tid1,Int_t tid2)
49 {fPadX=(Int_t)pad[0];fPadY=(Int_t)pad[1];fQdc=q;fChamber=10*c+AliRICHParam::Pad2Sec(pad);fCFM=cfm;fTracks[0]=tid0;fTracks[1]=tid1;fTracks[2]=tid2;}
853634d3 50 virtual ~AliRICHdigit() {;}
d4c94996 51 Int_t Compare(const TObject *pObj) const
52 {if(Id()==((AliRICHdigit*)pObj)->Id())return 0;else if(Id()>((AliRICHdigit*)pObj)->Id())return 1;else return -1;} //virtual
e42a7b46 53 virtual Bool_t IsSortable() const{return kTRUE;} //sort interface
54 virtual void Print(Option_t *option="") const; //virtual
55 Int_t ChFbMi() const{return fCFM;} //particle mixture for this digit
d4c94996 56 Int_t C() const{return fChamber/10;} //chamber number
57 Int_t S() const{return fChamber-(fChamber/10)*10;} //sector number
58 Int_t X() const{return fPadX;} //x position of the pad
59 Int_t Y() const{return fPadY;} //y postion of the pad
e42a7b46 60 TVector Pad() const{Float_t v[2]={fPadX,fPadY}; return TVector(2,v);}
d4c94996 61 Int_t Id() const{return fChamber*10000000+fPadX*1000+fPadY;} //absolute id of this pad
62 Double_t Q() const{return fQdc;} //charge in terms of ADC channels
89924db8 63 void AddTidOffset(Int_t offset)
64 {for (Int_t i=0; i<3; i++) if (fTracks[i]>0) fTracks[i]+=offset;};
853634d3 65protected:
e42a7b46 66 Int_t fCFM; //1000000*Ncerenkovs+1000*Nfeedbacks+Nmips
ed3ceb24 67 Int_t fChamber; //10*chamber number+ sector number
c1863e3c 68 Int_t fPadX; //pad number along X
69 Int_t fPadY; //pad number along Y
ed3ceb24 70 Double_t fQdc; //QDC value, fractions are permitted for summable procedure
d4c94996 71 ClassDef(AliRICHdigit,3) //RICH digit class
853634d3 72};//class AliRICHdigit
cbaf35fb 73
ed3ceb24 74//__________________AliRICHcluster__________________________________________________________________
c1863e3c 75class AliRICHcluster :public TObject
e42a7b46 78 enum ClusterStatus {kEdge,kShape,kSize,kRaw,kResolved,kEmpty=kBad};
79 AliRICHcluster():TObject(),fCFM(0),fSize(0),fShape(0),fQdc(0),fChamber(0),fX(0),fY(0),fStatus(kEmpty),fDigits(0) {}
80 virtual ~AliRICHcluster() {Reset();}
81 void Reset() {DeleteDigits();fCFM=fSize=fShape=fQdc=fChamber=0;fX=fY=0;fStatus=kEmpty;} //cleans the cluster
82 void DeleteDigits() {if(fDigits) delete fDigits; fDigits=0;} //deletes the list of digits
83 AliRICHcluster& operator=(const AliRICHcluster&) {return *this;}
84 Int_t Nlocals() const{return fSize-10000*(fSize/10000);} //number of local maximums
85 Int_t Size() const{return fSize/10000;} //number of digits in cluster
86 Int_t Fsize() const{return fSize;} //
87 Int_t Shape() const{return fShape;} //cluster shape rectangulare
88 Int_t C() const{return fChamber/10;} //chamber number
89 Int_t S() const{return fChamber-(fChamber/10)*10;} //sector number
90 Int_t Fchamber() const{return fChamber;} //
91 Int_t Q() const{return fQdc;} //cluster charge in QDC channels
92 Double_t X() const{return fX;} //cluster x position in LRS
93 Double_t Y() const{return fY;} //cluster y position in LRS
94 Int_t Status() const{return fStatus;} //
95 void SetStatus(Int_t status) {fStatus=status;} //
96 Int_t Nmips() const{return fCFM-1000000*Ncerenkovs()-1000*Nfeedbacks();} //
97 Int_t Ncerenkovs() const{return fCFM/1000000;} //
98 Int_t Nfeedbacks() const{return (fCFM-1000000*Ncerenkovs())/1000;} //
99 Bool_t IsPureMip() const{return fCFM<1000;} //
100 Bool_t IsPureCerenkov() const{return Nmips()==0&&Nfeedbacks()==0;} //
101 Bool_t IsPureFeedback() const{return Nmips()==0&&Ncerenkovs()==0;} //
102 Bool_t IsSingleMip() const{return Nmips()==1&&Ncerenkovs()==0&&Nfeedbacks()==0;} //
103 Bool_t IsSingleCerenkov() const{return Nmips()==0&&Ncerenkovs()==1&&Nfeedbacks()==0;} //
104 Bool_t IsSingleFeedback() const{return Nmips()==0&&Ncerenkovs()==0&&Nfeedbacks()==1;} //
105 Bool_t IsMip() const{return Nmips()!=0;} //
106 Bool_t IsCerenkov() const{return Ncerenkovs()!=0;} //
107 Bool_t IsFeedback() const{return Nfeedbacks()!=0;} //
108 Int_t CombiPid() const{return fCFM;} //
109 void CFM(Int_t c,Int_t f,Int_t m) {fCFM=1000000*c+1000*f+m;} //cluster contributors
110 TObjArray* Digits() const{return fDigits;} //
111 virtual void Print(Option_t *option="")const; //
112 inline void AddDigit(AliRICHdigit *pDig); //
113 inline void CoG(Int_t nLocals); //calculates center of gravity
114 void Fill(AliRICHcluster *pRaw,Double_t x,Double_t y,Double_t q,Int_t cfm) //form new resolved cluster from raw one
115 {fCFM=cfm;fChamber=pRaw->Fchamber();fSize=pRaw->Fsize();fQdc=(Int_t)(q*pRaw->Q());fX=x;fY=y;fStatus=kResolved;} //
c1863e3c 116protected:
e42a7b46 117 Int_t fCFM; //1000000*Ncerenkovs+1000*Nfeedbacks+Nmips
6865fb8d 118 Int_t fSize; //10000*(how many digits belong to this cluster) + nLocalMaxima
e42a7b46 119 Int_t fShape; //100*xdim+ydim box containing the cluster
c1863e3c 120 Int_t fQdc; //QDC value
121 Int_t fChamber; //10*module number+sector number
122 Double_t fX; //local x postion
123 Double_t fY; //local y postion
124 Int_t fStatus; //flag to mark the quality of the cluster
125 TObjArray *fDigits; //! list of digits forming this cluster
ed3ceb24 126 ClassDef(AliRICHcluster,2) //RICH cluster class
c1863e3c 127};//class AliRICHcluster
129void AliRICHcluster::AddDigit(AliRICHdigit *pDig)
e42a7b46 130{
131// Adds a given digit to the list of digits belonging to this cluster
132 if(!fDigits) {fQdc=fSize=fCFM=0;fDigits = new TObjArray;}
cbaf35fb 133 fQdc+=(Int_t)pDig->Q(); fDigits->Add(pDig);
cb8189e7 134 fChamber=10*pDig->C()+pDig->S();
6865fb8d 135 fSize+=10000;
e42a7b46 136 fStatus=kRaw;
cbaf35fb 137}
6865fb8d 139void AliRICHcluster::CoG(Int_t nLocals)
e42a7b46 140{
141// Calculates naive cluster position as a center of gravity of its digits.
142 Float_t xmin=999,ymin=999,xmax=0,ymax=0;
cb8189e7 143 fX=fY=0;
144 for(Int_t iDig=0;iDig<Size();iDig++) {
145 AliRICHdigit *pDig=(AliRICHdigit*)fDigits->At(iDig);
e42a7b46 146 TVector pad=pDig->Pad(); Double_t q=pDig->Q();
147 TVector2 x2=AliRICHParam::Pad2Loc(pad);
3582c1f9 148 fX += x2.X()*q;fY +=x2.Y()*q;
e42a7b46 149 if(pad[0]<xmin)xmin=pad[0];if(pad[0]>xmax)xmax=pad[0];if(pad[1]<ymin)ymin=pad[1];if(pad[1]>ymax)ymax=pad[1];
cbaf35fb 150 }
cb8189e7 151 fX/=fQdc;fY/=fQdc;//Center of Gravity
e42a7b46 152 fShape=Int_t(100*(xmax-xmin+1)+ymax-ymin+1);//find box containing cluster
6865fb8d 153 fSize+=nLocals;
cb8189e7 154 fStatus=kRaw;
d4c94996 156
c021cb15 157//__________________AliRICH_________________________________________________________________________
e42a7b46 158class AliESD;
3ea9cb08 159
2f614988 160class AliRICH : public AliDetector
2f614988 162public:
853634d3 163 AliRICH();
af3d25a6 164 AliRICH(const char *name, const char *title);
d4c94996 165 AliRICH(const AliRICH& RICH):AliDetector(RICH) {;} //copy ctor
853634d3 166 virtual ~AliRICH();
dfb4e77d 167
942194a4 168 AliRICH& operator=(const AliRICH&) {return *this;}
d4c94996 169//framework part
e42a7b46 170 virtual Int_t IsVersion() const =0; //interface from
171 virtual void StepManager() =0; //interface from AliMC
172 virtual void Hits2SDigits(); //interface from AliSimulation
173 virtual AliDigitizer* CreateDigitizer(AliRunDigitizer* man) const {return new AliRICHDigitizer(man);} //interface from AliSimulation
174 virtual void Reconstruct() const; //interface from AliReconstruction
175 virtual void FillESD(AliESD *pESD) const; //interface from AliReconstruction
176 virtual void Print(Option_t *option) const; //prints current RICH status
177 virtual void SetTreeAddress(); //interface from AliLoader
178 virtual void MakeBranch(Option_t *opt=" "); //interface from AliLoader
179 virtual void CreateMaterials(); //interface from AliMC
180 virtual void CreateGeometry(); //interface from AliMC
181 virtual void BuildGeometry(); //interface
d4c94996 182//private part
183 Float_t AbsoCH4(Float_t x)const; //calculates absorption length for methane
184 Float_t Fresnel(Float_t ene,Float_t pdoti, Bool_t pola)const; //deals with Fresnel absorption
d4c94996 185 inline void CreateHits(); //create hits container as a simple list
186 inline void CreateSDigits(); //create sdigits container as a simple list
187 inline void CreateDigits(); //create digits container as 7 lists, one per chamber
188 inline void CreateClusters(); //create clusters container as 7 lists, one per chamber
d4c94996 189// void ResetHits() {AliDetector::ResetHits();} //virtual
190 void ResetSDigits() {fNsdigits=0; if(fSdigits) fSdigits ->Clear();}
e42a7b46 191 void ResetDigits() {if(fDigitsNew)for(int i=0;i<kNchambers;i++){fDigitsNew->At(i)->Clear();fNdigitsNew[i]=0;}} //virtual
192 void ResetClusters() {if(fClusters) for(int i=0;i<kNchambers;i++){fClusters ->At(i)->Clear();fNclusters[i]=0;}}
c1863e3c 193 TClonesArray* SDigits() const{return fSdigits;}
543d5224 194 TClonesArray* Digits(Int_t iC) const{if(fDigitsNew) return (TClonesArray *)fDigitsNew->At(iC-1);else return 0;}
c60862bf 195 TClonesArray* Clusters(Int_t iC) const{if(fClusters) return (TClonesArray *)fClusters->At(iC-1);else return 0;}
e42a7b46 196 AliRICHChamber* C(Int_t iC) const{return fpParam->C(iC);} //provides pointer to a given chamber
197 AliRICHParam* P() const{return fpParam;} //provides pointer to a RICH params
198 AliRICH* R() {return this;} //provides pointer to RICH main object
199 TVector Counters() const{return fCounters;} //provides a set of counters
200 void ControlPlots(); //utility
201 void PrintHits (Int_t iEvent=0); //utility
202 void PrintSDigits (Int_t iEvent=0); //utility
203 void PrintDigits (Int_t iEvent=0); //utility
204 void PrintClusters(Int_t iEvent=0); //utility
c60862bf 205
e42a7b46 206 void AddHit(Int_t c,Int_t tid,TVector3 i3,TVector3 o3,Double_t eloss=0){TClonesArray &tmp=*fHits;new(tmp[fNhits++])AliRICHhit(c,tid,i3,o3,eloss);}
207 inline void AddSDigit(Int_t c,TVector pad,Double_t q,Int_t pid,Int_t tid);
208 void AddDigit(int c,TVector pad,int q,int cfm,int *tid)//Add simulated digit
209 {TClonesArray &tmp=*((TClonesArray*)fDigitsNew->At(c-1));new(tmp[fNdigitsNew[c-1]++])AliRICHdigit(c,pad,q,cfm,tid[0],tid[1],tid[2]);}
210 void AddDigit(Int_t c,TVector pad,Int_t q)//for real data digits
211 {TClonesArray &tmp=*((TClonesArray*)fDigitsNew->At(0));new(tmp[fNdigitsNew[0]++])AliRICHdigit(c,pad,q,0,-1,-1,-1);}
d4c94996 212 void AddCluster(AliRICHcluster &cl)
e42a7b46 213 {Int_t c=cl.C()-1;TClonesArray &tmp=*((TClonesArray*)fClusters->At(c));new(tmp[fNclusters[c]++])AliRICHcluster(cl);}
214 AliRICHhit* Hit(Int_t tid); //returns pointer ot RICH hit for a given tid
c60862bf 215protected:
e42a7b46 216 enum {kAir=1,kCSI=6,kGAP=9,kAl=10,kCH4=5,kSteel=15,kPerpex=16,kSr90=17};
c021cb15 217 AliRICHParam *fpParam; //main RICH parametrization
d4c94996 218 //fHits and fDigits belong to AliDetector
e42a7b46 219 TClonesArray *fSdigits; //! list of sdigits
220 Int_t fNsdigits; //! current number of sdigits
84365c70 221
e42a7b46 222 TObjArray *fDigitsNew; //! each chamber holds it's one lists of digits
223 Int_t fNdigitsNew[kNchambers]; //! array of current numbers of digits
84365c70 224
e42a7b46 225 TObjArray *fClusters; //! each chamber holds it's one lists of clusters
226 Int_t fNclusters[kNchambers]; //! array of current numbers of raw clusters
84365c70 227
e42a7b46 228 TVector fCounters; //Photon history conters, explanation in StepManager()
229 ClassDef(AliRICH,7) //Main RICH class
c60862bf 230};//class AliRICH
ed3ceb24 231
c1863e3c 232//__________________________________________________________________________________________________
c60862bf 233void AliRICH::CreateHits()
235 if(fHits) return;
236 if(GetDebug())Info("CreateHits","creating hits container.");
237 fHits=new TClonesArray("AliRICHhit",10000); fNhits=0;
543d5224 238}
c60862bf 239//__________________________________________________________________________________________________
c1863e3c 240void AliRICH::CreateSDigits()
c60862bf 241{
242 if(fSdigits) return;
c1863e3c 243 if(GetDebug())Info("CreateSDigits","creating sdigits container.");
c60862bf 244 fSdigits=new TClonesArray("AliRICHdigit",10000); fNsdigits=0;
543d5224 245}
c60862bf 246//__________________________________________________________________________________________________
247void AliRICH::CreateDigits()
249 if(fDigitsNew) return;
250 if(GetDebug())Info("CreateDigits","creating digits containers.");
e42a7b46 251 fDigitsNew = new TObjArray(kNchambers);
252 for(Int_t i=0;i<kNchambers;i++) {fDigitsNew->AddAt(new TClonesArray("AliRICHdigit",10000), i); fNdigitsNew[i]=0;}
543d5224 253}
c60862bf 254//__________________________________________________________________________________________________
255void AliRICH::CreateClusters()
257 if(fClusters) return;
258 if(GetDebug())Info("CreateClusters","creating clusters containers.");
e42a7b46 259 fClusters = new TObjArray(kNchambers);
260 for(Int_t i=0;i<kNchambers;i++) {fClusters->AddAt(new TClonesArray("AliRICHcluster",10000), i); fNclusters[i]=0;}
543d5224 261}
ed3ceb24 262//__________________________________________________________________________________________________
e42a7b46 263void AliRICH::AddSDigit(Int_t c,TVector pad,Double_t q,Int_t pid,Int_t tid)
265 Int_t cfm;
ed3ceb24 266 switch(pid){
e42a7b46 267 case 50000050: cfm=1000000;break;//cerenkov
268 case 50000051: cfm=1000; break;//feedback
269 default: cfm=1; break;//mip
ed3ceb24 270 }
e42a7b46 271 TClonesArray &tmp=*fSdigits; new(tmp[fNsdigits++])AliRICHdigit(c,pad,q,cfm,tid,kBad,kBad);
d4c94996 273//__________________________________________________________________________________________________
c60862bf 274#endif//#ifndef AliRICH_h