3 //___________________________________________________________________________
4 /////////////////////////////////////////////////////////////////////////////
6 // File reader for HLT tracks ESD //
8 // loizides@ikf.uni-frankfurt.de //
9 /////////////////////////////////////////////////////////////////////////////
11 #include <Riostream.h>
14 #include <TObjString.h>
18 #include <AliESDtrack.h>
19 #include <AliESDHLTtrack.h>
20 #include <AliL3Track.h>
21 #include <AliL3Vertex.h>
22 #include <AliKalmanTrack.h>
23 #include <AliJetEventParticles.h>
24 #include "AliJetParticlesReaderHLT.h"
26 ClassImp(AliJetParticlesReaderHLT)
28 AliJetParticlesReaderHLT::AliJetParticlesReaderHLT(Bool_t bMapper, const Char_t* esdfilename) :
29 AliJetParticlesReaderESD(0,esdfilename),
30 fTrackerType(bMapper),
37 /********************************************************************/
39 AliJetParticlesReaderHLT::AliJetParticlesReaderHLT(
42 const Char_t* esdfilename) :
43 AliJetParticlesReaderESD(0,dirs,esdfilename),
44 fTrackerType(bMapper),
52 /********************************************************************/
54 AliJetParticlesReaderHLT::~AliJetParticlesReaderHLT()
59 Int_t AliJetParticlesReaderHLT::ReadESD(AliESD* esd)
65 Error("ReadESD","ESD is NULL");
69 Float_t mf = esd->GetMagneticField();
72 Error("ReadESD","Magnetic Field is 0. Skipping to next event.");
75 AliKalmanTrack::SetMagneticField(mf/10.);
77 Info("ReadESD","Reading Event %d",fCurrentDir*1000+fCurrentEvent);
78 if((!fOwner) || (fEventParticles==0))
79 fEventParticles = new AliJetEventParticles();
83 ntr =esd->GetNumberOfHLTHoughTracks();
84 Info("ReadESD","Found %d conformal tracks.",ntr);
86 ntr=esd->GetNumberOfHLTHoughTracks();
87 Info("ReadESD","Found %d hough tracks.",ntr);
89 fEventParticles->Reset(ntr);
93 headdesc+=esd->GetRunNumber();
95 headdesc+=esd->GetEventNumber();
96 fEventParticles->SetHeader(headdesc);
98 Double_t vertexpos[3];//vertex position
99 const AliESDVertex* kvertex = esd->GetVertex();
102 Info("ReadESD","ESD returned NULL pointer to vertex - assuming (0.0,0.0,0.0)");
109 kvertex->GetXYZ(vertexpos);
112 fEventParticles->SetVertex(vertexpos[0],vertexpos[1],vertexpos[2]);
113 //cout << vertexpos[0] << " " << vertexpos[1] << " " << vertexpos[2] << endl;
115 v.SetX(vertexpos[0]);
116 v.SetY(vertexpos[1]);
117 v.SetZ(vertexpos[2]);
119 for (Int_t i = 0;i<ntr; i++) {
120 AliESDHLTtrack *kesdtrack;
122 kesdtrack=esd->GetHLTConfMapTrack(i);
124 kesdtrack=esd->GetHLTHoughTrack(i);
129 Error("ReadESD","Can not get track %d", i);
133 //const Float_t kpid=kesdtrack->GetPID();
134 const Int_t knhits=kesdtrack->GetNHits();
135 const Int_t kweight=kesdtrack->GetWeight();
136 //cout << i << " " << kweight << " " << knhits << endl;
137 if((fMinHits>0) && (knhits<fMinHits)) continue;
138 if(kweight>1000) continue; //avoid ghosts
139 if((fMinWeight>0) && (kweight<fMinWeight)) continue;
141 Float_t px=kesdtrack->GetPx();
142 Float_t py=kesdtrack->GetPy();
143 Float_t pz=kesdtrack->GetPz();
144 if(TMath::IsNaN(px)||TMath::IsNaN(py)||TMath::IsNaN(pz)) continue;
145 if(TMath::Abs(px)>1e3||TMath::Abs(py)>1e3||TMath::Abs(pz)>1e3) continue;
146 //cout << px << " " << py << " " << pz << " " << TMath::Sqrt(px*px+py*py) << endl;
150 //kesdtrack->SetCharge(-kesdtrack->GetCharge());
152 if(!kesdtrack->GetPxPyPzAt(0,mom)) continue;
158 //if(!kesdtrack->ComesFromMainVertex()) continue;
159 //cout << "Pos: " << kesdtrack->GetFirstPointX() << " " << kesdtrack->GetFirstPointY() << " " << kesdtrack->GetFirstPointZ() << endl;
160 l3.SetFirstPoint(kesdtrack->GetFirstPointX(),kesdtrack->GetFirstPointY(),kesdtrack->GetFirstPointZ());
161 l3.SetLastPoint(kesdtrack->GetLastPointX(),kesdtrack->GetLastPointY(),kesdtrack->GetLastPointZ());
162 l3.SetCharge(kesdtrack->GetCharge());
163 l3.SetPt(kesdtrack->GetPt());
164 l3.SetTgl(kesdtrack->GetTgl());
165 l3.SetPsi(kesdtrack->GetPsi());
167 Double_t xc=0.,yc=0.,zc=0.;
168 l3.GetClosestPoint(&v,xc,yc,zc);
169 if(TMath::Abs(zc)>10.) continue;
170 l3.SetFirstPoint(0,0,0);
171 //cout << "Pos: " << xc << " " << yc << " " << zc << endl;
172 l3.UpdateToFirstPoint();
178 const Float_t kpt=TMath::Sqrt(px*px+py*py);
179 //cout << px << " " << py << " " << pz << " " << kpt << endl;
181 const Float_t kp=TMath::Sqrt(pz*pz+kpt*kpt);
182 const Float_t keta=0.5*TMath::Log((kp+pz+1e-30)/(kp-pz+1e-30));
183 const Float_t kphi=TMath::Pi()+TMath::ATan2(-py,-px);
185 if(IsAcceptedParticle(kpt,kphi,keta))
186 fEventParticles->AddParticle(px,py,pz,kp,i,kesdtrack->GetMCid(),knhits,kpt,kphi,keta);