1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: ALICE Offline. *
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 **************************************************************************/
17 //=========================================================================//
18 // AliEbyE Analysis for Particle Ratio Fluctuation //
19 // Deepika Rathee | Satyajit Jena //
20 // drathee@cern.ch | sjena@cern.ch //
21 // Date: Wed Jul 9 18:38:30 CEST 2014 //
22 // New approch to find particle ratio to reduce memory //
24 //=========================================================================//
29 #include "AliESDEvent.h"
31 #include "AliMCEvent.h"
32 #include "AliESDtrackCuts.h"
34 #include "AliAODEvent.h"
35 #include "AliAODMCParticle.h"
37 #include "AliEbyEPidRatioDCA.h"
41 ClassImp(AliEbyEPidRatioDCA)
43 //________________________________________________________________________
44 AliEbyEPidRatioDCA::AliEbyEPidRatioDCA() :
45 AliEbyEPidRatioBase("DCA", "DCA"),
46 fESDTrackCutsBkg(NULL),
51 AliLog::SetClassDebugLevel("AliEbyEPidRatioDCA",10);
54 //________________________________________________________________________
55 AliEbyEPidRatioDCA::~AliEbyEPidRatioDCA() {
59 //________________________________________________________________________
60 void AliEbyEPidRatioDCA::Process() {
62 fESDTrackCutsBkg->GetEtaRange(etaRange[0],etaRange[1]);
65 fESDTrackCutsBkg->GetPtRange(ptRange[0],ptRange[1]);
68 for (Int_t idxTrack = 0; idxTrack < fNTracks; ++idxTrack) {
70 AliVTrack *track = (fESD) ? static_cast<AliVTrack*>(fESD->GetTrack(idxTrack)) : static_cast<AliVTrack*>(fAOD->GetTrack(idxTrack));
72 // -- Check if track is accepted for basic parameters
73 if (!fHelper->IsTrackAcceptedBasicCharged(track))
76 // -- Check if accepted - ESD
77 if (fESD && !fESDTrackCutsBkg->AcceptTrack(dynamic_cast<AliESDtrack*>(track)))
80 // -- Check if accepted - AOD
82 AliAODTrack * trackAOD = dynamic_cast<AliAODTrack*>(track);
85 AliError("Pointer to dynamic_cast<AliAODTrack*>(track) = ZERO");
88 if (!trackAOD->TestFilterBit(fAODtrackCutBit))
91 // -- Check if in pT and eta range (is done in ESDTrackCuts for ESDs)
92 if(!(track->Pt() > ptRange[0] && track->Pt() <= ptRange[1] && TMath::Abs(track->Eta()) <= etaRange[1]))
99 if (fHelper->IsTrackAcceptedPID(track, pid, (AliPID::kPion))) { iPid = 1; gPdgCode = 211;}
100 else if (fHelper->IsTrackAcceptedPID(track, pid, (AliPID::kKaon))) { iPid = 2; gPdgCode = 321;}
101 else if (fHelper->IsTrackAcceptedPID(track, pid, (AliPID::kProton))){ iPid = 3; gPdgCode = 2212;}
102 else { iPid = 0; gPdgCode = 0;}
104 // cout << " --- DCA ---- " << iPid << " " << gPdgCode << endl;
106 if (!fHelper->IsTrackAcceptedRapidity(track, yP, iPid))
109 Bool_t isDCArAccepted = fHelper->IsTrackAcceptedDCA(track);
111 // -- Check if accepted with thighter DCA cuts
112 // ?!?!? How to mimic this in AODs?
113 if (fESD && !fESDTrackCuts->AcceptTrack(dynamic_cast<AliESDtrack*>(track)))
114 isDCArAccepted = kFALSE;
116 // -- Check for contamination
117 Int_t contIdx = (fIsMC) ? GetContIdxTrack(TMath::Abs(track->GetLabel()), track->Charge(), gPdgCode) : 1;
119 // -- Get DCAs (dca_r, dca_z, sigma_xy, sigma_xy_z, sigma_z)
120 Float_t dca[2], cov[3]; //
122 (dynamic_cast<AliESDtrack*>(track))->GetImpactParameters(dca, cov);
129 Double_t hnDCA[10] = {fCentralityBin,0, track->Charge(),
140 Double_t hnDCA[10] = {fCentralityBin, iPid, track->Charge(),
151 } // for (Int_t idxTrack = 0; idxTrack < fNTracks; ++idxTrack) {
156 //________________________________________________________________________
157 void AliEbyEPidRatioDCA::CreateHistograms() {
158 Int_t binHnDCA[10] = {AliEbyEPidRatioHelper::fgkfHistNBinsCent,4,
159 AliEbyEPidRatioHelper::fgkfHistNBinsSign,
160 AliEbyEPidRatioHelper::fgkfHistNBinsEta,
161 AliEbyEPidRatioHelper::fgkfHistNBinsRap,
162 AliEbyEPidRatioHelper::fgkfHistNBinsPhi,
163 AliEbyEPidRatioHelper::fgkfHistNBinsPt,
166 Double_t minHnDCA[10] = {AliEbyEPidRatioHelper::fgkfHistRangeCent[0],-0.5,
167 AliEbyEPidRatioHelper::fgkfHistRangeSign[0],
168 AliEbyEPidRatioHelper::fgkfHistRangeEta[0],
169 AliEbyEPidRatioHelper::fgkfHistRangeRap[0],
170 AliEbyEPidRatioHelper::fgkfHistRangePhi[0],
171 AliEbyEPidRatioHelper::fgkfHistRangePt[0],
174 Double_t maxHnDCA[10] = {AliEbyEPidRatioHelper::fgkfHistRangeCent[1],3.5,
175 AliEbyEPidRatioHelper::fgkfHistRangeSign[1],
176 AliEbyEPidRatioHelper::fgkfHistRangeEta[1],
177 AliEbyEPidRatioHelper::fgkfHistRangeRap[1],
178 AliEbyEPidRatioHelper::fgkfHistRangePhi[1],
179 AliEbyEPidRatioHelper::fgkfHistRangePt[1],
183 fHnDCA = new THnSparseD("hnDCA", "cent:pid:etaRec:yRec:phiRec:ptRec:sign:contPart:contSign:DCArAccepted:DCAr", 10, binHnDCA, minHnDCA, maxHnDCA);
187 fHnDCA->GetAxis(0)->SetTitle("centrality"); // 0-5|5-10|10-20|20-30|30-40|40-50|50-60|60-70|70-80|80-90 --> 10 bins
188 fHnDCA->GetAxis(1)->SetTitle("N_{ch}|N_{#pi}|N_{K}|N_{p}"); // 0 | 1 | 2 | 3
189 fHnDCA->GetAxis(2)->SetTitle("sign"); // -1 | 0 | +1
190 fHnDCA->GetAxis(3)->SetTitle("#eta_{Rec}"); // eta [-0.9, 0.9]
191 fHnDCA->GetAxis(4)->SetTitle("#it{y}_{Rec}"); // rapidity [-0.5, 0.5]
192 fHnDCA->GetAxis(5)->SetTitle("#varphi_{Rec} (rad)"); // phi [ 0. , 2Pi]
193 fHnDCA->GetAxis(6)->SetTitle("#it{p}_{T,Rec} (GeV/#it{c})"); // pT [ 0.2, 2.6]
194 fHnDCA->GetAxis(7)->SetTitle("contPart"); // 1 primary | 2 missId | 3 from WeakDecay | 4 p from Material
195 fHnDCA->GetAxis(8)->SetTitle("DCArAccepted"); // 0 not accepted | 1 accepted
196 fHnDCA->GetAxis(9)->SetTitle("DCAr"); // DCAr [-3, 3]
199 fHelper->BinLogAxis(fHnDCA, 6, fESDTrackCuts);
200 fHelper->BinLogAxis(fHnDCA, 6, fESDTrackCutsBkg);
202 // -- Set binning for DCAr
203 Double_t binsDCAr[77] = {-3.,-2.85,-2.7,-2.55,-2.4,-2.25,-2.1,-1.95,-1.8,-1.65,-1.5,-1.35,-1.2,-1.05,-0.9,-0.75,-0.6,-0.45,-0.3,-0.285,-0.27,-0.255,-0.24,-0.225,-0.21,-0.195,-0.18,-0.165,-0.15,-0.135,-0.12,-0.105,-0.09,-0.075,-0.06,-0.045,-0.03,-0.015,0.,0.015,0.03,0.045,0.06,0.075,0.09,0.105,0.12,0.135,0.15,0.165,0.18,0.195,0.21,0.225,0.24,0.255,0.27,0.285,0.3,0.45,0.6,0.75,0.9,1.05,1.2,1.35,1.5,1.65,1.8,1.95,2.1,2.25,2.4,2.55,2.7,2.85,3.};
204 fHnDCA->GetAxis(9)->Set(76, binsDCAr);
206 // ------------------------------------------------------------------
211 //________________________________________________________________________
212 Int_t AliEbyEPidRatioDCA::GetContIdxTrack(Int_t label, Int_t sign, Int_t gPdgCode) {
215 AliVParticle* particle = (fESD) ? fMCEvent->GetTrack(label) : static_cast<AliVParticle*>(fArrayMC->At(label));
219 Bool_t isPhysicalPrimary = (fESD) ? fStack->IsPhysicalPrimary(label): (static_cast<AliAODMCParticle*>(particle))->IsPhysicalPrimary();
220 Bool_t isSecondaryFromWeakDecay = (fESD) ? fStack->IsSecondaryFromWeakDecay(label) : (static_cast<AliAODMCParticle*>(particle))->IsSecondaryFromWeakDecay();
221 Bool_t isSecondaryFromMaterial = (fESD) ? fStack->IsSecondaryFromMaterial(label) : (static_cast<AliAODMCParticle*>(particle))->IsSecondaryFromMaterial();
223 if (isPhysicalPrimary) {
225 // -- Check if correctly identified
226 if (particle->PdgCode() == (sign*gPdgCode))
228 // -- MissIdentification
236 // -- Check if secondaries from material or weak decay
237 else if(isSecondaryFromWeakDecay)
239 else if (isSecondaryFromMaterial)