3 #include "AlidNdEtaSystematicsSelector.h"
13 #include <TParticle.h>
14 #include <TParticlePDG.h>
19 #include <AliHeader.h>
20 #include <AliGenEventHeader.h>
21 #include <../STEER/AliGenPythiaEventHeader.h>
22 #include <../STEER/AliGenCocktailEventHeader.h>
24 #include "esdTrackCuts/AliESDtrackCuts.h"
25 #include "AliPWG0Helper.h"
26 #include "dNdEta/AlidNdEtaCorrection.h"
28 ClassImp(AlidNdEtaSystematicsSelector)
30 AlidNdEtaSystematicsSelector::AlidNdEtaSystematicsSelector() :
41 // Constructor. Initialization of pointers
44 for (Int_t i=0; i<4; ++i)
45 fdNdEtaCorrectionSpecies[i] = 0;
47 for (Int_t i=0; i<3; ++i) {
48 fdNdEtaCorrectionVertexReco[i] = 0;
49 fdNdEtaCorrectionTriggerBias[i] = 0;
53 AlidNdEtaSystematicsSelector::~AlidNdEtaSystematicsSelector()
60 void AlidNdEtaSystematicsSelector::Begin(TTree* tree)
64 AliSelectorRL::Begin(tree);
66 ReadUserObjects(tree);
69 void AlidNdEtaSystematicsSelector::ReadUserObjects(TTree* tree)
71 // read the user objects, called from slavebegin and begin
73 if (!fEsdTrackCuts && fInput)
74 fEsdTrackCuts = dynamic_cast<AliESDtrackCuts*> (fInput->FindObject("AliESDtrackCuts"));
76 if (!fEsdTrackCuts && tree)
77 fEsdTrackCuts = dynamic_cast<AliESDtrackCuts*> (tree->GetUserInfo()->FindObject("AliESDtrackCuts"));
80 AliDebug(AliLog::kError, "ERROR: Could not read EsdTrackCuts from input list.");
83 void AlidNdEtaSystematicsSelector::SlaveBegin(TTree* tree)
85 // The SlaveBegin() function is called after the Begin() function.
86 // When running with PROOF SlaveBegin() is called on each slave server.
87 // The tree argument is deprecated (on PROOF 0 is passed).
89 AliSelector::SlaveBegin(tree);
91 ReadUserObjects(tree);
93 TString option(GetOption());
95 printf("Running AlidNdEtaSystematicsSelector with options %s\n", option.Data());
97 // Options: secondaries particle-composition sigma-vertex vertexreco triggerbias
99 if (option.Contains("secondaries"))
101 fSecondaries = new TH2F("fSecondaries", "fSecondaries;Case;N", 8, 0.5, 8.5, 2001, -0.5, 2000.5);
102 fSecondaries->GetXaxis()->SetBinLabel(1, "All Primaries");
103 fSecondaries->GetXaxis()->SetBinLabel(2, "All Secondaries");
104 fSecondaries->GetXaxis()->SetBinLabel(3, "Primaries pT > 0.3 GeV/c");
105 fSecondaries->GetXaxis()->SetBinLabel(4, "Secondaries pT > 0.3 GeV/c");
106 fSecondaries->GetXaxis()->SetBinLabel(5, "Tracks from Primaries");
107 fSecondaries->GetXaxis()->SetBinLabel(6, "Tracks from Secondaries");
108 fSecondaries->GetXaxis()->SetBinLabel(7, "Accepted Tracks from Primaries");
109 fSecondaries->GetXaxis()->SetBinLabel(8, "Accepted Tracks from Secondaries");
112 if (option.Contains("particle-composition"))
114 for (Int_t i=0; i<4; ++i)
117 name.Form("correction_%d", i);
118 fdNdEtaCorrectionSpecies[i] = new AlidNdEtaCorrection(name, name);
122 if (option.Contains("sigma-vertex"))
124 fSigmaVertex = new TH1F("fSigmaVertex", "fSigmaVertex;Nsigma2vertex;NacceptedTracks", 50, 0.05, 5.05);
125 printf("WARNING: sigma-vertex analysis enabled. This will produce weird results in the AliESDtrackCuts histograms\n");
128 if (option.Contains("vertexreco")) {
129 fdNdEtaCorrectionVertexReco[0] = new AlidNdEtaCorrection("vertexRecoND", "vertexRecoND");
130 fdNdEtaCorrectionVertexReco[1] = new AlidNdEtaCorrection("vertexRecoSD", "vertexRecoSD");
131 fdNdEtaCorrectionVertexReco[2] = new AlidNdEtaCorrection("vertexRecoDD", "vertexRecoDD");
134 if (option.Contains("triggerbias")) {
135 fdNdEtaCorrectionTriggerBias[0] = new AlidNdEtaCorrection("triggerBiasND", "triggerBiasND");
136 fdNdEtaCorrectionTriggerBias[1] = new AlidNdEtaCorrection("triggerBiasSD", "triggerBiasSD");
137 fdNdEtaCorrectionTriggerBias[2] = new AlidNdEtaCorrection("triggerBiasDD", "triggerBiasDD");
140 fPIDParticles = new TH1F("pid_particles", "PID of generated primary particles", 10001, -5000.5, 5000.5);
141 fPIDTracks = new TH1F("pid_tracks", "MC PID of reconstructed tracks", 10001, -5000.5, 5000.5);
143 if (option.Contains("only-positive"))
145 AliInfo("Processing only positive particles.");
148 else if (option.Contains("only-negative"))
150 AliInfo("Processing only negative particles.");
154 if (option.Contains("low-multiplicity"))
156 AliInfo("Processing only events with low multiplicity.");
157 fMultiplicityMode = 1;
159 else if (option.Contains("high-multiplicity"))
161 AliInfo("Processing only events with high multiplicity.");
162 fMultiplicityMode = 2;
166 Bool_t AlidNdEtaSystematicsSelector::Process(Long64_t entry)
168 // The Process() function is called for each entry in the tree (or possibly
169 // keyed object in the case of PROOF) to be processed. The entry argument
170 // specifies which entry in the currently loaded tree is to be processed.
171 // It can be passed to either TTree::GetEntry() or TBranch::GetEntry()
172 // to read either all or the required parts of the data. When processing
173 // keyed objects with PROOF, the object is already loaded and is available
174 // via the fObject pointer.
176 // This function should contain the "body" of the analysis. It can contain
177 // simple or elaborate selection criteria, run algorithms on the data
178 // of the event and typically fill histograms.
180 // WARNING when a selector is used with a TChain, you must use
181 // the pointer to the current TTree to call GetEntry(entry).
182 // The entry is always the local entry number in the current tree.
183 // Assuming that fTree is the pointer to the TChain being processed,
184 // use fTree->GetTree()->GetEntry(entry).
186 if (AliSelectorRL::Process(entry) == kFALSE)
189 // Check prerequisites
192 AliDebug(AliLog::kError, "ESD branch not available");
198 AliDebug(AliLog::kError, "fESDTrackCuts not available");
202 AliStack* stack = GetStack();
205 AliDebug(AliLog::kError, "Stack not available");
209 // --------------------------------------------------------------
210 // related to events:
212 // stuff for vertex reconstruction correction systematics
213 Bool_t vertexRecoStudy = kFALSE;
214 Bool_t triggerBiasStudy = kFALSE;
215 if (fdNdEtaCorrectionVertexReco[0]) vertexRecoStudy = kTRUE;
216 if (fdNdEtaCorrectionTriggerBias[0]) triggerBiasStudy = kTRUE;
218 AliHeader* header = GetHeader();
220 AliDebug(AliLog::kError, "Header not available");
224 // getting process information NB: this only works for Pythia !!!
225 Int_t processtype = AliPWG0Helper::GetPythiaEventProcessType(header);
227 // can only read pythia headers, either directly or from cocktalil header
228 AliGenEventHeader* genHeader = (AliGenEventHeader*)(header->GenEventHeader());
231 AliDebug(AliLog::kError, "Gen header not available");
237 genHeader->PrimaryVertex(vtxMC);
239 TObjArray* listOfTracks = fEsdTrackCuts->GetAcceptedTracks(fESD);
240 Int_t nGoodTracks = listOfTracks->GetEntries();
242 Bool_t eventTriggered = AliPWG0Helper::IsEventTriggered(fESD);
243 Bool_t vertexReconstructed = AliPWG0Helper::IsVertexReconstructed(fESD);
246 if (processtype!=92 && processtype!=93 && processtype!=94) {
247 // NB: passing the wrong process type here (1), since the process type is defined by the index in the array (here non-diffractive)
248 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[0]->FillEvent(vtxMC[2], nGoodTracks, eventTriggered, vertexReconstructed, 1);
249 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[0] ->FillEvent(vtxMC[2], nGoodTracks, eventTriggered, vertexReconstructed, 1);
252 // single diffractive
253 if (processtype==92 || processtype==93) {
254 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[1]->FillEvent(vtxMC[2], nGoodTracks, eventTriggered, vertexReconstructed, 1);
255 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[1] ->FillEvent(vtxMC[2], nGoodTracks, eventTriggered, vertexReconstructed, 1);
258 // double diffractive
259 if (processtype==94) {
260 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[2]->FillEvent(vtxMC[2], nGoodTracks, eventTriggered, vertexReconstructed, 1);
261 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[2] ->FillEvent(vtxMC[2], nGoodTracks, eventTriggered, vertexReconstructed, 1);
264 for (Int_t i=0; i<4; ++i) {
265 if (fdNdEtaCorrectionSpecies[i])
266 fdNdEtaCorrectionSpecies[i]->FillEvent(vtxMC[2], nGoodTracks, eventTriggered, vertexReconstructed, 1);
269 // --------------------------------------------------------------
273 Int_t nPrim = stack->GetNprimary();
275 for (Int_t iMc = 0; iMc < nPrim; ++iMc)
277 TParticle* particle = stack->Particle(iMc);
281 AliDebug(AliLog::kError, Form("UNEXPECTED: particle with label %d not found in stack (mc loop).", iMc));
285 if (AliPWG0Helper::IsPrimaryCharged(particle, nPrim) == kFALSE)
288 if (SignOK(particle->GetPDG()) == kFALSE)
291 Float_t eta = particle->Eta();
292 Float_t pt = particle->Pt();
295 switch (TMath::Abs(particle->GetPdgCode()))
297 case 211: id = 0; break;
298 case 321: id = 1; break;
299 case 2212: id = 2; break;
302 /*if (pt < 0.1 && particle->GetMother(0) > -1)
304 TParticle* mother = stack->Particle(particle->GetMother(0));
305 printf("Mother pdg code is %d\n", mother->GetPdgCode());
308 //if (particle->GetUniqueID() == 1)
310 //printf("Found illegal particle mc id = %d file = %s event = %d\n", iMc, fTree->GetCurrentFile()->GetName(), fTree->GetTree()->GetReadEntry());
312 default: id = 3; break;
315 if (eventTriggered & vertexReconstructed) {
316 if (fdNdEtaCorrectionSpecies[id])
317 fdNdEtaCorrectionSpecies[id]->FillMCParticle(vtxMC[2], eta, pt, eventTriggered, vertexReconstructed, 1);
320 fPIDParticles->Fill(particle->GetPdgCode());
324 if (processtype!=92 && processtype!=93 && processtype!=94) {
325 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[0]->FillMCParticle(vtxMC[2], eta, pt, eventTriggered, vertexReconstructed, 1);
326 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[0] ->FillMCParticle(vtxMC[2], eta, pt, eventTriggered, vertexReconstructed, 1);
328 // single diffractive
329 if (processtype==92 || processtype==93) {
330 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[1]->FillMCParticle(vtxMC[2], eta, pt, eventTriggered, vertexReconstructed, 1);
331 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[1] ->FillMCParticle(vtxMC[2], eta, pt, eventTriggered, vertexReconstructed, 1);
333 // double diffractive
334 if (processtype==94) {
335 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[2]->FillMCParticle(vtxMC[2], eta, pt, eventTriggered, vertexReconstructed, 1);
336 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[2] ->FillMCParticle(vtxMC[2], eta, pt, eventTriggered, vertexReconstructed, 1);
339 }// end of mc particle
342 // --------------------------------------------------------------
346 if (fMultiplicityMode == 1 && listOfTracks->GetEntries() > 20 ||
347 fMultiplicityMode == 2 && listOfTracks->GetEntries() < 40)
354 // loop over esd tracks
355 TIterator* iter = listOfTracks->MakeIterator();
357 while ((obj = iter->Next()) != 0)
359 AliESDtrack* esdTrack = dynamic_cast<AliESDtrack*> (obj);
363 // using the properties of the mc particle
364 Int_t label = TMath::Abs(esdTrack->GetLabel());
365 TParticle* particle = stack->Particle(label);
368 AliDebug(AliLog::kError, Form("UNEXPECTED: particle with label %d not found in stack (track loop).", label));
372 Float_t eta = particle->Eta();
373 Float_t pt = particle->Pt();
376 if (processtype!=92 && processtype!=93 && processtype!=94) {
377 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[0]->FillTrackedParticle(vtxMC[2], eta, pt);
378 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[0] ->FillTrackedParticle(vtxMC[2], eta, pt);
381 // single diffractive
382 if (processtype==92 || processtype==93) {
383 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[1]->FillTrackedParticle(vtxMC[2], eta, pt);
384 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[1] ->FillTrackedParticle(vtxMC[2], eta, pt);
387 // double diffractive
388 if (processtype==94) {
389 if (triggerBiasStudy) fdNdEtaCorrectionTriggerBias[2]->FillTrackedParticle(vtxMC[2], eta, pt);
390 if (vertexRecoStudy) fdNdEtaCorrectionVertexReco[2] ->FillTrackedParticle(vtxMC[2], eta, pt);
393 // find primary particle that created this particle
394 TParticle* mother = AliPWG0Helper::FindPrimaryMother(stack, label);
398 if (SignOK(mother->GetPDG()) == kFALSE)
401 //printf("We continue with particle %d (pdg %d)\n", label, mother->GetPdgCode());
404 switch (TMath::Abs(mother->GetPdgCode()))
406 case 211: id = 0; break;
407 case 321: id = 1; break;
408 case 2212: id = 2; break;
409 default: id = 3; break;
412 if (vertexReconstructed && eventTriggered) {
413 if (fdNdEtaCorrectionSpecies[id])
414 fdNdEtaCorrectionSpecies[id]->FillTrackedParticle(vtxMC[2], eta, pt);
417 fPIDTracks->Fill(particle->GetPdgCode());
419 } // end of track loop
438 void AlidNdEtaSystematicsSelector::FillSecondaries()
440 // fills the secondary histograms
442 AliStack* stack = GetStack();
444 Int_t particlesPrimaries = 0;
445 Int_t particlesSecondaries = 0;
446 Int_t particlesPrimariesPtCut = 0;
447 Int_t particlesSecondariesPtCut = 0;
449 for (Int_t iParticle = 0; iParticle < stack->GetNtrack(); iParticle++)
451 TParticle* particle = stack->Particle(iParticle);
454 AliDebug(AliLog::kError, Form("UNEXPECTED: particle with label %d not found in stack (particle loop).", iParticle));
458 if (TMath::Abs(particle->Eta()) > 0.9)
461 Bool_t isPrimary = kFALSE;
462 if (iParticle < stack->GetNprimary())
464 if (AliPWG0Helper::IsPrimaryCharged(particle, stack->GetNprimary()) == kFALSE)
471 particlesPrimaries++;
473 particlesSecondaries++;
475 if (particle->Pt() < 0.3)
479 particlesPrimariesPtCut++;
481 particlesSecondariesPtCut++;
484 fSecondaries->Fill(1, particlesPrimaries);
485 fSecondaries->Fill(2, particlesSecondaries);
486 fSecondaries->Fill(3, particlesPrimariesPtCut);
487 fSecondaries->Fill(4, particlesSecondariesPtCut);
489 Int_t allTracksPrimaries = 0;
490 Int_t allTracksSecondaries = 0;
491 Int_t acceptedTracksPrimaries = 0;
492 Int_t acceptedTracksSecondaries = 0;
494 for (Int_t iTrack = 0; iTrack < fESD->GetNumberOfTracks(); iTrack++)
496 AliESDtrack* esdTrack = fESD->GetTrack(iTrack);
499 AliDebug(AliLog::kError, Form("UNEXPECTED: Could not get track %d.", iTrack));
503 Int_t label = TMath::Abs(esdTrack->GetLabel());
504 TParticle* particle = stack->Particle(label);
507 AliDebug(AliLog::kError, Form("UNEXPECTED: particle with label %d not found in stack (track loop).", label));
511 if (label < stack->GetNprimary())
512 allTracksPrimaries++;
514 allTracksSecondaries++;
516 if (!fEsdTrackCuts->AcceptTrack(esdTrack))
519 if (label < stack->GetNprimary())
520 acceptedTracksPrimaries++;
522 acceptedTracksSecondaries++;
525 fSecondaries->Fill(5, allTracksPrimaries);
526 fSecondaries->Fill(6, allTracksSecondaries);
527 fSecondaries->Fill(7, acceptedTracksPrimaries);
528 fSecondaries->Fill(8, acceptedTracksSecondaries);
530 //printf("P = %d, S = %d, P_pt = %d, S_pt = %d, T_P = %d, T_S = %d, T_P_acc = %d, T_S_acc = %d\n", particlesPrimaries, particlesSecondaries, particlesPrimariesPtCut, particlesSecondariesPtCut, allTracksPrimaries, allTracksSecondaries, acceptedTracksPrimaries, acceptedTracksSecondaries);
533 void AlidNdEtaSystematicsSelector::FillSigmaVertex()
535 // fills the fSigmaVertex histogram
537 // save the old value
538 Float_t oldSigmaVertex = fEsdTrackCuts->GetMinNsigmaToVertex();
541 fEsdTrackCuts->SetMinNsigmaToVertex(5);
543 TObjArray* list = fEsdTrackCuts->GetAcceptedTracks(fESD);
545 TIterator* iter = list->MakeIterator();
547 while ((obj = iter->Next()) != 0)
549 AliESDtrack* esdTrack = dynamic_cast<AliESDtrack*> (obj);
553 Float_t sigma2Vertex = fEsdTrackCuts->GetSigmaToVertex(esdTrack);
555 for (Double_t nSigma = 0.1; nSigma < 5.05; nSigma += 0.1)
557 if (sigma2Vertex < nSigma)
558 fSigmaVertex->Fill(nSigma);
568 // set back the old value
569 fEsdTrackCuts->SetMinNsigmaToVertex(oldSigmaVertex);
572 void AlidNdEtaSystematicsSelector::SlaveTerminate()
574 // The SlaveTerminate() function is called after all entries or objects
575 // have been processed. When running with PROOF SlaveTerminate() is called
576 // on each slave server.
578 AliSelector::SlaveTerminate();
580 // Add the histograms to the output on each slave server
583 AliDebug(AliLog::kError, Form("ERROR: Output list not initialized."));
588 fOutput->Add(fSecondaries);
590 for (Int_t i=0; i<4; ++i)
591 if (fdNdEtaCorrectionSpecies[i])
592 fOutput->Add(fdNdEtaCorrectionSpecies[i]);
595 fOutput->Add(fSigmaVertex);
597 for (Int_t i=0; i<3; ++i) {
598 if (fdNdEtaCorrectionVertexReco[i])
599 fOutput->Add(fdNdEtaCorrectionVertexReco[i]);
601 if (fdNdEtaCorrectionTriggerBias[i])
602 fOutput->Add(fdNdEtaCorrectionTriggerBias[i]);
607 void AlidNdEtaSystematicsSelector::Terminate()
609 // The Terminate() function is the last function to be called during
610 // a query. It always runs on the client, it can be used to present
611 // the results graphically or save the results to file.
613 AliSelector::Terminate();
615 fSecondaries = dynamic_cast<TH2F*> (fOutput->FindObject("fSecondaries"));
616 for (Int_t i=0; i<4; ++i)
617 fdNdEtaCorrectionSpecies[i] = dynamic_cast<AlidNdEtaCorrection*> (fOutput->FindObject(Form("correction_%d", i)));
618 fSigmaVertex = dynamic_cast<TH1F*> (fOutput->FindObject("fSigmaVertex"));
620 fdNdEtaCorrectionVertexReco[0] = dynamic_cast<AlidNdEtaCorrection*> (fOutput->FindObject("vertexRecoND"));
621 fdNdEtaCorrectionVertexReco[1] = dynamic_cast<AlidNdEtaCorrection*> (fOutput->FindObject("vertexRecoSD"));
622 fdNdEtaCorrectionVertexReco[2] = dynamic_cast<AlidNdEtaCorrection*> (fOutput->FindObject("vertexRecoDD"));
624 fdNdEtaCorrectionTriggerBias[0] = dynamic_cast<AlidNdEtaCorrection*> (fOutput->FindObject("triggerBiasND"));
625 fdNdEtaCorrectionTriggerBias[1] = dynamic_cast<AlidNdEtaCorrection*> (fOutput->FindObject("triggerBiasSD"));
626 fdNdEtaCorrectionTriggerBias[2] = dynamic_cast<AlidNdEtaCorrection*> (fOutput->FindObject("triggerBiasDD"));
630 TDatabasePDG* pdgDB = new TDatabasePDG;
632 for (Int_t i=0; i <= fPIDParticles->GetNbinsX()+1; ++i)
633 if (fPIDParticles->GetBinContent(i) > 0)
634 printf("PDG = %d (%s): generated: %d, reconstructed: %d, ratio: %f\n",
635 (Int_t) fPIDParticles->GetBinCenter(i),
636 pdgDB->GetParticle((Int_t) fPIDParticles->GetBinCenter(i))->GetName(),
637 (Int_t) fPIDParticles->GetBinContent(i),
638 (Int_t) fPIDTracks->GetBinContent(i),
639 ((fPIDTracks->GetBinContent(i) > 0) ? fPIDParticles->GetBinContent(i) / fPIDTracks->GetBinContent(i) : -1));
645 for (Int_t i=0; i<3; ++i) {
646 if (fdNdEtaCorrectionVertexReco[i])
647 fdNdEtaCorrectionVertexReco[i]->Finish();
649 if (fdNdEtaCorrectionTriggerBias[i])
650 fdNdEtaCorrectionTriggerBias[i]->Finish();
653 TFile* fout = TFile::Open("systematics.root", "RECREATE");
656 fEsdTrackCuts->SaveHistograms("esd_track_cuts");
659 fSecondaries->Write();
662 fSigmaVertex->Write();
664 for (Int_t i=0; i<4; ++i)
665 if (fdNdEtaCorrectionSpecies[i])
666 fdNdEtaCorrectionSpecies[i]->SaveHistograms();
668 for (Int_t i=0; i<3; ++i) {
669 if (fdNdEtaCorrectionVertexReco[i])
670 fdNdEtaCorrectionVertexReco[i]->SaveHistograms();
672 if (fdNdEtaCorrectionTriggerBias[i])
673 fdNdEtaCorrectionTriggerBias[i]->SaveHistograms();
682 fSecondaries->Draw("COLZ");
688 fSigmaVertex->Draw();
692 Bool_t AlidNdEtaSystematicsSelector::SignOK(TParticlePDG* particle)
694 // returns if a particle with this sign should be counted
695 // this is determined by the value of fSignMode, which should have the same sign
697 // if fSignMode is 0 all particles are counted
704 printf("WARNING: not counting a particle that does not have a pdg particle\n");
708 Double_t charge = particle->Charge();