1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 // Class AliHFEtaggedTrackAnalysis
17 // Analyses tracks with an apriori PID information (i.e. using the daugther
18 // tracks from well-identified decays of neutral charged particles). Tracks
19 // are processed in the Process function, where given tracks are filtered
20 // via the track cuts and used for PID later. The plugin fills Correction
21 // Framework containers and additional PID QA containers
24 // Markus Fasel <M.Fasel@gsi.de>
26 #include "AliCFCutBase.h"
27 #include "AliCFContainer.h"
28 #include "AliCFManager.h"
29 #include "AliESDtrack.h"
32 #include "AliPIDResponse.h"
33 #include "AliExternalTrackParam.h"
35 #include "AliHFEcollection.h"
36 #include "AliHFEcontainer.h"
37 #include "AliHFEcuts.h"
38 #include "AliHFEpid.h"
39 #include "AliHFEpidTPC.h"
40 #include "AliHFEpidQAmanager.h"
41 #include "AliHFEtaggedTrackAnalysis.h"
42 #include "AliHFEvarManager.h"
44 ClassImp(AliHFEtaggedTrackAnalysis)
45 //____________________________________________________________
46 AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis():
58 , fVariablesTRD(kFALSE)
67 //____________________________________________________________
68 AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis(const char *name):
80 , fVariablesTRD(kFALSE)
85 // Default constructor
87 fVarManager = new AliHFEvarManager("taggedTrackVarManager");
88 //fVarManager->AddVariable("pt");
89 //fVarManager->AddVariable("eta");
90 //fVarManager->AddVariable("phi");
91 //fVarManager->AddVariable("charge");
92 //fVarManager->AddVariable("species");
93 fPIDqa = new AliHFEpidQAmanager;
94 fCFM = new AliCFManager;
95 SetBit(kIsOwner, kTRUE);
98 //____________________________________________________________
99 AliHFEtaggedTrackAnalysis::AliHFEtaggedTrackAnalysis(const AliHFEtaggedTrackAnalysis &ref):
101 , fVarManager(ref.fVarManager)
107 , fQAhistos(ref.fQAhistos)
108 , fCentralityF(ref.fCentralityF)
110 , fMagneticField(ref.fMagneticField)
111 , fVariablesTRD(ref.fVariablesTRD)
112 , fIsPbPb(ref.fIsPbPb)
121 SetBit(kIsOwner, kFALSE);
124 //____________________________________________________________
125 AliHFEtaggedTrackAnalysis &AliHFEtaggedTrackAnalysis::operator=(const AliHFEtaggedTrackAnalysis &ref){
127 // Assignment operator
129 TNamed::operator=(ref);
131 fVarManager = ref.fVarManager;
136 fQAhistos = ref.fQAhistos;
137 fCentralityF = ref.fCentralityF;
139 fMagneticField = ref.fMagneticField;
140 fVariablesTRD = ref.fVariablesTRD;
141 fIsPbPb = ref.fIsPbPb;
144 if(ref.fContainer) InitContainer();
146 SetBit(kIsOwner, kFALSE);
147 SetBit(kIsOwnerCuts, kFALSE);
152 //____________________________________________________________
153 AliHFEtaggedTrackAnalysis::~AliHFEtaggedTrackAnalysis(){
157 if(TestBit(kIsOwner)){
158 if(fVarManager) delete fVarManager;
159 if(fPIDqa) delete fPIDqa;
161 if(TestBit(kIsOwnerCuts)) delete fCuts;
162 if(fContainer) delete fContainer;
165 //____________________________________________________________
166 void AliHFEtaggedTrackAnalysis::InitContainer(){
168 // Initialize output container
170 if(fContainer) return;
173 AliError("No PID set - defining container without PID steps");
175 nStepPID = fPID->GetNumberOfPIDdetectors();
177 fContainer = new AliHFEcontainer("containerV0");
178 fVarManager->DefineVariables(fContainer);
179 fContainer->CreateContainer("taggedTrackContainerReco", "Container for Tagged Tracks", AliHFEcuts::kNcutStepsRecTrack + nStepPID);
181 // Set the step titles
182 for(Int_t istep = 0; istep < AliHFEcuts::kNcutStepsRecTrack; istep++)
183 fContainer->SetStepTitle("taggedTrackContainerReco", AliHFEcuts::RecoCutName(istep), istep);
184 for(Int_t ipid = 0; ipid < nStepPID; ipid++){
185 fContainer->SetStepTitle("taggedTrackContainerReco", fPID->SortedDetectorName(ipid), ipid + AliHFEcuts::kNcutStepsRecTrack);
187 fCFM->SetParticleContainer(fContainer->GetCFContainer("taggedTrackContainerReco"));
189 // temporarily special QA
190 fQAhistos = new AliHFEcollection("taggedTrackQA", "Special QA for the TaggedTrackAnalysis");
191 fQAhistos->CreateTH2F("TPCclusters2_1", "TPCclusterInfo for findable clusters for 2 neighbors", 30, 0.1, 10., 162, 0., 161.);
192 fQAhistos->CreateTH2F("TPCclusters2_0", "TPCclusterInfo for the ratio for 2 neighbors", 30, 0.1, 10., 110, 0., 1.1);
193 fQAhistos->CreateTH2F("TPCncls", "TPC number of clusters", 30, 0.1, 10., 162, 0., 161.);
194 fQAhistos->CreateTH2F("TPCclr", "TPC cluster ratio", 30, 0.1, 10., 110, 0., 1.1);
195 fQAhistos->BinLogAxis("TPCclusters2_1", 0); // pt axis in logarithmic binning
196 fQAhistos->BinLogAxis("TPCclusters2_0", 0); // pt axis in logarithmic binning
197 fQAhistos->BinLogAxis("TPCncls", 0); // pt axis in logarithmic binning
198 fQAhistos->BinLogAxis("TPCclr", 0); // pt axis in logarithmic binning
201 //____________________________________________________________
202 void AliHFEtaggedTrackAnalysis::ProcessTrack(AliVTrack *track, Int_t abinitioPID){
204 // Filter tracks tagged by V0 PID class
207 fVarManager->NewTrack(track, NULL, fCentralityF, abinitioPID, kTRUE);
213 AliESDtrack *esdtrackc = dynamic_cast<AliESDtrack *>(track);
216 const AliExternalTrackParam *trueparam = NULL;
217 if(esdtrackc->GetOuterParam()) {
218 trueparam = esdtrackc->GetOuterParam();
219 fVarManager->NewTrack((AliVParticle *)trueparam, NULL, fCentralityF, abinitioPID, kTRUE);
226 // Try a loose cut to reject pion contamination
228 if(abinitioPID == AliPID::kElectron){
229 AliHFEpidTPC *pidTPC = (AliHFEpidTPC *) fPID->GetDetPID(AliHFEpid::kTPCpid);
231 Double_t numberOfSigmaTPC = pidTPC->GetPIDResponse()->NumberOfSigmasTPC(track,AliPID::kElectron);
232 if(numberOfSigmaTPC < -5) return;
236 // temporarily monitoring of the number of TPC clusters
237 AliESDtrack *esdtrack = dynamic_cast<AliESDtrack *>(track);
238 if(esdtrack && abinitioPID == AliPID::kElectron){
239 if((esdtrack->GetITSClusterMap() & (BIT(0) | BIT(1))) && (TMath::Abs(esdtrack->Eta()) < 0.8)){ // Only select quasi-primary tracks
240 fQAhistos->Fill("TPCclusters2_1", track->Pt(), esdtrack->GetTPCClusterInfo(2,1));
241 fQAhistos->Fill("TPCclusters2_0", track->Pt(), esdtrack->GetTPCNclsF() > 0 ? esdtrack->GetTPCClusterInfo(2,1)/esdtrack->GetTPCNclsF() : 0.);
242 fQAhistos->Fill("TPCncls", track->Pt(), esdtrack->GetTPCNcls());
243 fQAhistos->Fill("TPCclr", track->Pt(), esdtrack->GetTPCNclsF() > 0 ? static_cast<Double_t>(esdtrack->GetTPCNcls())/static_cast<Double_t>(esdtrack->GetTPCNclsF()) : 0.);
247 Int_t offset = AliHFEcuts::kStepRecKineITSTPC;
248 fVarManager->FillContainer(fCFM->GetParticleContainer(), 0); // Fill Container without filtering
250 Bool_t survived = kTRUE;
251 for(Int_t icut = AliHFEcuts::kStepRecKineITSTPC; icut <= AliHFEcuts::kStepHFEcutsTRD; icut++){
252 AliDebug(2, Form("Checking cut %d for species %s", icut + AliHFEcuts::kNcutStepsMCTrack, AliPID::ParticleName(abinitioPID)));
254 TObjArray *cutlist = fCFM->GetParticleCutsList(icut + AliHFEcuts::kNcutStepsMCTrack);
256 AliDebug(2, Form("No cuts for step %d set", icut + AliHFEcuts::kNcutStepsMCTrack));
258 AliDebug(2, Form("Cut Collection %s", cutlist->GetName()));
259 TIter cutiter(cutlist);
261 while((cut = dynamic_cast<AliCFCutBase *>(cutiter()))){
262 AliDebug(2, Form("Cut object %s, QA on? %s", cut->GetName(), cut->IsQAOn() ? "yes" : "no"));
266 //if(!fCFM->CheckParticleCuts(icut + AliHFEcuts::kNcutStepsMCTrack, (TObject *)track)){
267 if(!fCuts->CheckParticleCuts(icut + AliHFEcuts::kNcutStepsMCTrack, (TObject *)track)){
268 AliDebug(2, Form("Track didn' survive cut %d", icut + AliHFEcuts::kNcutStepsMCTrack));
272 AliDebug(2, Form("Cut passed, filling container %d", icut - offset + 1));
273 fVarManager->FillContainer(fCFM->GetParticleContainer(), icut - offset + 1);
277 AliDebug(2, "Use track in the PID");
279 AliHFEpidObject hfetrack;
280 if(track->IsA() == AliESDtrack::Class())
281 hfetrack.SetAnalysisType(AliHFEpidObject::kESDanalysis);
283 hfetrack.SetAnalysisType(AliHFEpidObject::kAODanalysis);
284 hfetrack.SetRecTrack(track);
285 hfetrack.SetAbInitioPID(abinitioPID);
286 hfetrack.SetCentrality(fCentralityF);
287 if(fIsPbPb) hfetrack.SetPbPb();
289 if(fIspPb) hfetrack.SetpPb();
290 else hfetrack.SetPP();
292 fPID->SetVarManager(fVarManager);
293 fPID->IsSelected(&hfetrack, fContainer, "taggedTrackContainer", fPIDqa);
297 //____________________________________________________________
298 void AliHFEtaggedTrackAnalysis::SetCuts(AliHFEcuts *cuts){
300 // Set HFE cuts to be used to filter the tagged tracks
303 AliWarning("Nob cuts provided - Using standard cuts");
304 fCuts = new AliHFEcuts("cutsTagged", "HFE Cuts for the V0 tagged tracks");
305 fCuts->CreateStandardCuts();
307 SetBit(kIsOwnerCuts);
309 AliDebug(1, "Setting single track cuts");
312 const Int_t kNcutSteps = AliHFEcuts::kNcutStepsMCTrack + AliHFEcuts::kNcutStepsRecTrack + AliHFEcuts::kNcutStepsDETrack;
313 printf("Setting Number of cut steps %d\n", kNcutSteps);
314 fCFM->SetNStepParticle(kNcutSteps);
315 for(Int_t istep = 0; istep < kNcutSteps; istep++)
316 fCFM->SetParticleCutsList(istep, NULL);
318 fCuts->Initialize(fCFM);
321 //____________________________________________________________
322 void AliHFEtaggedTrackAnalysis::SetPID(AliHFEpid *pid){
324 // Set the PID and initialize the the QA manager
327 fPIDqa->Initialize(fPID);
330 //____________________________________________________________
331 TList *AliHFEtaggedTrackAnalysis::GetPIDQA() const {
335 return fPIDqa->MakeList("PIDqa_taggedTracks");
338 //____________________________________________________________
339 TList *AliHFEtaggedTrackAnalysis::GetCutQA() const {
343 return fCuts->GetQAhistograms();