Coverity fixes (A. Shabetai)
[u/mrichter/AliRoot.git] / JETAN / DEV / AliJetFillCalTrkTrack.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 /* $Id$ */
17
18 //--------------------------------------------------
19 // Filling of CalTrkTrack objects in the reader task
20 //
21 // Author: magali.estienne@subatech.in2p3.fr
22 //         alexandre.shabetai@cern.ch
23 //-------------------------------------------------
24
25 // --- AliRoot header files ---
26 #include "AliJetFillCalTrkTrack.h"
27 #include "AliJetReaderHeader.h"
28 #include "AliVEvent.h"
29 #include "AliESDtrack.h"
30 #include "AliAODTrack.h"
31
32 ClassImp(AliJetFillCalTrkTrack)
33
34 /////////////////////////////////////////////////////////////////////
35
36 AliJetFillCalTrkTrack::AliJetFillCalTrkTrack(): 
37   AliJetFillCalTrkEvent(),
38   fHadCorr(0),
39   fApplyMIPCorrection(kTRUE),
40   fVEvt(0x0)
41 {
42   // constructor
43 }
44
45 //-----------------------------------------------------------------------
46 AliJetFillCalTrkTrack::AliJetFillCalTrkTrack(AliVEvent* evt): 
47   AliJetFillCalTrkEvent(),
48   fHadCorr(0),
49   fApplyMIPCorrection(kTRUE),
50   fVEvt(evt)
51 {
52   // constructor
53 }
54
55 //-----------------------------------------------------------------------
56 AliJetFillCalTrkTrack::AliJetFillCalTrkTrack(const AliJetFillCalTrkTrack &det): 
57   AliJetFillCalTrkEvent(det),
58   fHadCorr(det.fHadCorr),
59   fApplyMIPCorrection(det.fApplyMIPCorrection),
60   fVEvt(det.fVEvt)
61 {
62   // Copy constructor
63 }
64
65 //-----------------------------------------------------------------------
66 AliJetFillCalTrkTrack& AliJetFillCalTrkTrack::operator=(const AliJetFillCalTrkTrack& other)
67 {
68   // Assignment
69   if (this != &other) {
70    fHadCorr = other.fHadCorr;
71    fApplyMIPCorrection = other.fApplyMIPCorrection;
72    fVEvt = other.fVEvt;
73  }
74
75   return (*this);
76
77 }
78
79 //-----------------------------------------------------------------------
80 AliJetFillCalTrkTrack::~AliJetFillCalTrkTrack()
81 {
82   // destructor
83 }
84
85 //-----------------------------------------------------------------------
86 void AliJetFillCalTrkTrack::Exec(Option_t* const /*option*/)
87 {
88   // Main method.
89   // Fill AliJetFillCalTrkTrack the with the charged particle information
90
91   fDebug = fReaderHeader->GetDebug();
92   fOpt = fReaderHeader->GetDetector();
93   TString datatype = fReaderHeader->GetDataType();
94   datatype.ToUpper();
95
96   Int_t type=0; 
97
98   if      (datatype.Contains("AODEXTRAONLY")) { type=kTrackAODextraonly;} 
99   else if (datatype.Contains("AODEXTRA")) { type=kTrackAODextra; }
100   else if (datatype.Contains("AOD")) { type=kTrackAOD; }
101   else if (datatype.Contains("ESD")) { type=kTrackESD; }
102   else { cout<< "Unknown Data type !" << endl; }
103
104   // temporary storage of signal and pt cut flag
105   Bool_t sflag = 0;
106   Bool_t cflag = 0;
107
108   // get cuts set by user
109   Float_t ptMin  = fReaderHeader->GetPtCut();
110   Float_t etaMin = fReaderHeader->GetFiducialEtaMin();
111   Float_t etaMax = fReaderHeader->GetFiducialEtaMax();
112   Float_t phiMin = fReaderHeader->GetFiducialPhiMin();
113   Float_t phiMax = fReaderHeader->GetFiducialPhiMax();
114   UInt_t filterMask = ((AliJetReaderHeader*)fReaderHeader)->GetTestFilterMask();
115   UInt_t filterType = ((AliJetReaderHeader*)fReaderHeader)->GetFilterType();
116
117   if(fDebug>2)Printf("%s:%d Selecting tracks with %d",(char*)__FILE__,__LINE__,type);
118
119   Int_t goodTrackStd = 0;
120   Int_t goodTrackNonStd = 0;
121   if(type==kTrackESD || type==kTrackAOD || type==kTrackAODextra || type==kTrackAODextraonly){
122     if(type!=kTrackAODextraonly) {
123       if(!fVEvt){
124         if(fDebug>2)Printf("%s:%d No AOD or ESD input event",(char*)__FILE__,__LINE__);
125         return;
126       }
127       for(int it = 0;it < fVEvt->GetNumberOfTracks();++it){
128         AliVTrack *tr = (AliVTrack*)fVEvt->GetTrack(it);
129         cflag=sflag=0;
130         if(fVEvt->InheritsFrom("AliESDEvent")){ 
131           // ESD case ....
132           if(tr->GetStatus()==0){ 
133             if(fDebug>10)Printf("%s:%d Not matching status %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks());
134             continue;
135           }
136           if ((tr->GetStatus() & AliESDtrack::kTPCrefit) == 0)    continue;      // quality check
137           if ((tr->GetStatus() & AliESDtrack::kITSrefit) == 0)    continue;      // quality check
138           if (((AliJetReaderHeader*) fReaderHeader)->ReadSignalOnly() 
139               && TMath::Abs(tr->GetLabel()) > 10000)  continue;      // quality check
140           if (((AliJetReaderHeader*) fReaderHeader)->ReadBkgdOnly() 
141               && TMath::Abs(tr->GetLabel()) < 10000)  continue;      // quality check
142         }
143         if((fVEvt->InheritsFrom("AliAODEvent"))){
144           // AOD case ....
145           AliAODTrack *trAOD = dynamic_cast<AliAODTrack*> (tr);
146           if(!trAOD) continue;
147           Bool_t bGood = false;
148           if(filterType == 0)bGood = true;
149           else if(filterType == 1)bGood = trAOD->IsHybridTPCConstrainedGlobal();
150           else if(filterType == 2)bGood = trAOD->IsHybridGlobalConstrainedGlobal();
151           if((filterMask>0)&&((!trAOD->TestFilterBit(filterMask)||(!bGood)))){
152             if(fDebug>10)Printf("%s:%d Not matching filter %d/%d %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks(),filterMask,trAOD->GetFilterMap());
153             continue;
154           }
155           // Should we also check the GetStatus ?
156         }
157         if((tr->Eta()> etaMax) || (tr->Eta() < etaMin)){
158           if(fDebug>10)Printf("%s:%d Not matching eta %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks());
159           continue;
160         }
161         Float_t phi = ( (tr->Phi()) < 0) ? (tr->Phi()) + 2. * TMath::Pi() : (tr->Phi());
162         if((phi > phiMax) || (phi < phiMin)){
163           if(fDebug>10)Printf("%s:%d Not matching eta %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks());
164           continue;
165         }
166         if(tr->Pt() <= ptMin){
167           if(fDebug>10)Printf("%s:%d Not matching pt %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks());
168         }
169         else {cflag=1;}
170         sflag=(TMath::Abs(tr->GetLabel()) < 10000) ? 1 : 0;
171
172         if(fDebug>10)Printf("%s:%d MATCHED %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks());
173
174         fCalTrkEvent->AddCalTrkTrack(tr,cflag,sflag); 
175         goodTrackStd++;
176       }
177     }
178     if(type==kTrackAODextra || type==kTrackAODextraonly) {
179       if(!fVEvt){
180         return;   
181       }
182       TClonesArray *aodExtraTracks = dynamic_cast<TClonesArray*>(fVEvt->FindListObject("aodExtraTracks"));
183       if(!aodExtraTracks) return; 
184       for(int it =0; it<aodExtraTracks->GetEntries(); it++) {
185         cflag=sflag=0;
186         AliVParticle *track = dynamic_cast<AliVParticle*> ((*aodExtraTracks)[it]);
187         if (!track) continue; 
188         AliAODTrack *trackAOD = dynamic_cast<AliAODTrack*> (track);
189         if(!trackAOD) continue;
190         if (trackAOD->GetStatus() == 0) continue; 
191         Bool_t bGood = false;
192         if(filterType == 0)bGood = true;
193         else if(filterType == 1)bGood = trackAOD->IsHybridTPCConstrainedGlobal();
194         else if(filterType == 2)bGood = trackAOD->IsHybridGlobalConstrainedGlobal();
195         if((filterMask>0)&&((!trackAOD->TestFilterBit(filterMask)||(!bGood))))continue;
196         if((trackAOD->Eta()> etaMax) || (trackAOD->Eta() < etaMin)){
197           if(fDebug>10)Printf("%s:%d Not matching eta %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks());
198           continue;
199         }
200         Float_t phi = ( (trackAOD->Phi()) < 0) ? (trackAOD->Phi()) + 2. * TMath::Pi() : (trackAOD->Phi());
201         if((phi > phiMax) || (phi < phiMin)){
202           if(fDebug>10)Printf("%s:%d Not matching eta %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks());
203           continue;
204         }
205         if(trackAOD->Pt() <= ptMin){
206           if(fDebug>10)Printf("%s:%d Not matching pt %d/%d",(char*)__FILE__,__LINE__,it,fVEvt->GetNumberOfTracks());
207         }
208         else {cflag=1;}
209         sflag=(TMath::Abs(trackAOD->GetLabel()) < 10000) ? 1 : 0; 
210         fCalTrkEvent->AddCalTrkTrack(trackAOD,cflag,sflag);
211         goodTrackNonStd++;
212       }
213     }
214   }
215
216   if(fDebug>0)printf("Number of good tracks selected: %5d \n", goodTrackStd+goodTrackNonStd);
217
218 }
219
220
221
222
223
224
225
226
227
228
229
230
231