]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG3/vertexingHF/AliAnalysisTaskSED0Mass.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Bugfix: wrong logic in determining which v0 reader to use
[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSED0Mass.cxx
CommitLineData
0108fa62 1
49061176 2/**************************************************************************
3 * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
4 * *
5 * Author: The ALICE Off-line Project. *
6 * Contributors are mentioned in the code where appropriate. *
7 * *
8 * Permission to use, copy, modify and distribute this software and its *
9 * documentation strictly for non-commercial purposes is hereby granted *
10 * without fee, provided that the above copyright notice appears in all *
11 * copies and that both the copyright notice and this permission notice *
12 * appear in the supporting documentation. The authors make no claims *
13 * about the suitability of this software for any purpose. It is *
14 * provided "as is" without express or implied warranty. *
15 **************************************************************************/
16
17/////////////////////////////////////////////////////////////
18//
19// AliAnalysisTaskSE for D0 candidates invariant mass histogram
a41f6fad 20// and comparison with the MC truth and cut variables distributions.
49061176 21//
22// Authors: A.Dainese, andrea.dainese@lnl.infn.it
feb73eca 23// Chiara Bianchin, chiara.bianchin@pd.infn.it (invariant mass)
24// Carmelo Di Giglio, carmelo.digiglio@ba.infn.it (like sign)
49061176 25/////////////////////////////////////////////////////////////
26
27#include <Riostream.h>
28#include <TClonesArray.h>
527f330b 29#include <TCanvas.h>
49061176 30#include <TNtuple.h>
31#include <TList.h>
32#include <TH1F.h>
a41f6fad 33#include <TH2F.h>
34#include <TDatabasePDG.h>
49061176 35
b557eb43 36#include "AliAnalysisManager.h"
37#include "AliAODHandler.h"
49061176 38#include "AliAODEvent.h"
39#include "AliAODVertex.h"
40#include "AliAODTrack.h"
41#include "AliAODMCHeader.h"
42#include "AliAODMCParticle.h"
43#include "AliAODRecoDecayHF2Prong.h"
44#include "AliAODRecoCascadeHF.h"
45#include "AliAnalysisVertexingHF.h"
46#include "AliAnalysisTaskSE.h"
47#include "AliAnalysisTaskSED0Mass.h"
48
b557eb43 49
49061176 50ClassImp(AliAnalysisTaskSED0Mass)
51
52
53//________________________________________________________________________
54AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass():
55AliAnalysisTaskSE(),
9de8c723 56fOutputPPR(0),
527f330b 57fOutputmycuts(0),
a41f6fad 58fDistr(0),
feb73eca 59fNentries(0),
9de8c723 60fVHFPPR(0),
527f330b 61fVHFmycuts(0),
feb73eca 62fArray(0),
ce39f0ac 63fReadMC(0),
feb73eca 64fLsNormalization(1.)
65
49061176 66{
67 // Default constructor
9de8c723 68 for(Int_t i=0;i<5;i++) {fTotPosPairs[i]=0; fTotNegPairs[i]=0;}
49061176 69}
70
71//________________________________________________________________________
72AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass(const char *name):
73AliAnalysisTaskSE(name),
9de8c723 74fOutputPPR(0),
527f330b 75fOutputmycuts(0),
a41f6fad 76fDistr(0),
feb73eca 77fNentries(0),
9de8c723 78fVHFPPR(0),
527f330b 79fVHFmycuts(0),
feb73eca 80fArray(0),
ce39f0ac 81fReadMC(0),
feb73eca 82fLsNormalization(1.)
49061176 83{
84 // Default constructor
9de8c723 85 for(Int_t i=0;i<5;i++) {fTotPosPairs[i]=0; fTotNegPairs[i]=0;}
49061176 86
87 // Output slot #1 writes into a TList container
88 DefineOutput(1,TList::Class()); //My private output
a4ae02cd 89 // Output slot #2 writes into a TList container
90 DefineOutput(2,TList::Class()); //My private output
91 // Output slot #3 writes into a TH1F container
92 DefineOutput(3,TH1F::Class()); //My private output
a41f6fad 93 // Output slot #4 writes into a TList container
94 DefineOutput(4,TList::Class()); //My private output
49061176 95}
96
97//________________________________________________________________________
98AliAnalysisTaskSED0Mass::~AliAnalysisTaskSED0Mass()
99{
9de8c723 100 if (fOutputPPR) {
101 delete fOutputPPR;
102 fOutputPPR = 0;
a4ae02cd 103 }
9de8c723 104 if (fVHFPPR) {
105 delete fVHFPPR;
106 fVHFPPR = 0;
a4ae02cd 107 }
527f330b 108 if (fOutputmycuts) {
109 delete fOutputmycuts;
110 fOutputmycuts = 0;
a4ae02cd 111 }
a41f6fad 112
113 if (fDistr) {
114 delete fDistr;
115 fDistr = 0;
116 }
117
527f330b 118 if (fVHFmycuts) {
119 delete fVHFmycuts;
120 fVHFmycuts = 0;
49061176 121 }
46c96ce5 122 if (fNentries){
123 delete fNentries;
124 fNentries = 0;
125 }
a41f6fad 126
127
49061176 128}
129
130//________________________________________________________________________
131void AliAnalysisTaskSED0Mass::Init()
132{
133 // Initialization
134
135 if(fDebug > 1) printf("AnalysisTaskSED0Mass::Init() \n");
136
137 gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/ConfigVertexingHF.C");
138
a4ae02cd 139 // 2 sets of dedidcated cuts -- defined in UserExec
9de8c723 140 fVHFPPR = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
527f330b 141 fVHFmycuts = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
a4ae02cd 142
49061176 143 return;
144}
145
146//________________________________________________________________________
147void AliAnalysisTaskSED0Mass::UserCreateOutputObjects()
148{
149
150 // Create the output container
151 //
152 if(fDebug > 1) printf("AnalysisTaskSED0Mass::UserCreateOutputObjects() \n");
153
154 // Several histograms are more conveniently managed in a TList
9de8c723 155 fOutputPPR = new TList();
156 fOutputPPR->SetOwner();
157 fOutputPPR->SetName("listPPR");
a4ae02cd 158
527f330b 159 fOutputmycuts = new TList();
160 fOutputmycuts->SetOwner();
161 fOutputmycuts->SetName("listloose");
49061176 162
a41f6fad 163 fDistr = new TList();
164 fDistr->SetOwner();
165 fDistr->SetName("distributionslist");
166
9de8c723 167 const Int_t nhist=5;
a4ae02cd 168
0108fa62 169 TString nameMass=" ",nameSgn27=" ",nameSgn=" ", nameBkg=" ", nameRfl=" ",nameMassNocutsS =" ",nameMassNocutsB =" ", namedistr=" ";
49061176 170
171 for(Int_t i=0;i<nhist;i++){
7646d6da 172 nameMass="histMass_";
49061176 173 nameMass+=i+1;
9de8c723 174 nameSgn27="histSgn27_";
175 nameSgn27+=i+1;
7646d6da 176 nameSgn="histSgn_";
49061176 177 nameSgn+=i+1;
7646d6da 178 nameBkg="histBkg_";
49061176 179 nameBkg+=i+1;
a4ae02cd 180 nameRfl="histRfl_";
181 nameRfl+=i+1;
0108fa62 182 nameMassNocutsS="hMassS_";
183 nameMassNocutsS+=i+1;
184 nameMassNocutsB="hMassB_";
185 nameMassNocutsB+=i+1;
7646d6da 186
a41f6fad 187 //histograms of invariant mass distributions
188
46c96ce5 189 TH1F* tmpMt = new TH1F(nameMass.Data(),"D^{0} invariant mass; M [GeV]; Entries",200,1.765,1.965);
190 TH1F *tmpMl=(TH1F*)tmpMt->Clone();
191 tmpMt->Sumw2();
192 tmpMl->Sumw2();
a4ae02cd 193
9de8c723 194 //to compare with AliAnalysisTaskCharmFraction
195 TH1F* tmpS27t = new TH1F(nameSgn27.Data(),"D^{0} invariant mass in M(D^{0}) +/- 27 MeV - MC; M [GeV]; Entries",200,1.765,1.965);
196 TH1F *tmpS27l=(TH1F*)tmpS27t->Clone();
197 tmpS27t->Sumw2();
198 tmpS27l->Sumw2();
0108fa62 199
200 //distribution w/o cuts
201 TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",200,1.765,1.965);
202 TH1F *tmpMB=(TH1F*)tmpMt->Clone();
203 tmpMB->SetName(nameMassNocutsB.Data());
204 tmpMS->Sumw2();
205 tmpMB->Sumw2();
206
207 //MC signal and background
46c96ce5 208 TH1F* tmpSt = new TH1F(nameSgn.Data(), "D^{0} invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
209 TH1F *tmpSl=(TH1F*)tmpSt->Clone();
210 tmpSt->Sumw2();
211 tmpSl->Sumw2();
a4ae02cd 212
46c96ce5 213 TH1F* tmpBt = new TH1F(nameBkg.Data(), "Background invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
214 TH1F *tmpBl=(TH1F*)tmpBt->Clone();
215 tmpBt->Sumw2();
216 tmpBl->Sumw2();
a4ae02cd 217
feb73eca 218 //Reflection: histo filled with D0Mass which pass the cut (also) as D0bar and with D0bar which pass (also) the cut as D0
46c96ce5 219 TH1F* tmpRt = new TH1F(nameRfl.Data(), "Reflected signal invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
220 TH1F *tmpRl=(TH1F*)tmpRt->Clone();
221 tmpRt->Sumw2();
222 tmpRl->Sumw2();
a4ae02cd 223 // printf("Created histograms %s\t%s\t%s\t%s\n",tmpM->GetName(),tmpS->GetName(),tmpB->GetName(),tmpR->GetName());
224
9de8c723 225 fOutputPPR->Add(tmpMt);
226 fOutputPPR->Add(tmpSt);
227 fOutputPPR->Add(tmpS27t);
228 fOutputPPR->Add(tmpBt);
229 fOutputPPR->Add(tmpRt);
a4ae02cd 230
527f330b 231 fOutputmycuts->Add(tmpMl);
232 fOutputmycuts->Add(tmpSl);
233 fOutputmycuts->Add(tmpS27l);
234 fOutputmycuts->Add(tmpBl);
235 fOutputmycuts->Add(tmpRl);
0108fa62 236
237 fDistr->Add(tmpMS);
238 fDistr->Add(tmpMB);
239
7646d6da 240 }
a4ae02cd 241
a41f6fad 242 //histograms of cut variable distributions
243 // pT
244 namedistr="hptpiS";
245 TH1F *hptpiS = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
a41f6fad 246
247 namedistr="hptKS";
248 TH1F *hptKS = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
249
250 namedistr="hptB";
251 TH1F *hptB = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
a41f6fad 252
253 // dca
254 namedistr="hdcaS";
255 TH1F *hdcaS = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
256 namedistr="hdcaB";
257 TH1F *hdcaB = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
a41f6fad 258
259 // costhetastar
feb73eca 260 namedistr="hcosthetastarS";
a41f6fad 261 TH1F *hcosthetastarS = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
feb73eca 262 namedistr="hcosthetastarB";
a41f6fad 263 TH1F *hcosthetastarB = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
a41f6fad 264
265 // impact parameter
266 namedistr="hd0piS";
267 TH1F *hd0piS = new TH1F(namedistr.Data(), "Impact parameter distribution (pions);d0(#pi) [cm]",200,-0.1,0.1);
a41f6fad 268
269 namedistr="hd0KS";
270 TH1F *hd0KS = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons);d0(K) [cm]",200,-0.1,0.1);
271 namedistr="hd0B";
272 TH1F *hd0B = new TH1F(namedistr.Data(), "Impact parameter distribution;d0 [cm]",200,-0.1,0.1);
a41f6fad 273
274 namedistr="hd0d0S";
275 TH1F *hd0d0S = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
276 namedistr="hd0d0B";
277 TH1F *hd0d0B = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
a41f6fad 278
279 // costhetapoint
280 namedistr="hcosthetapointS";
281 TH1F *hcosthetapointS = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
282 namedistr="hcosthetapointB";
283 TH1F *hcosthetapointB = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
a41f6fad 284
285 namedistr="hcosthpointd0d0S";
286 TH2F *hcosthpointd0d0S= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0}#timesd_{0};cos#theta_{Point};d_{0}#timesd_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
287 namedistr="hcosthpointd0d0B";
288 TH2F *hcosthpointd0d0B= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0}#timesd_{0};cos#theta_{Point};d_{0}#timesd_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
289
290 fDistr->Add(hptpiS);
a41f6fad 291 fDistr->Add(hptKS);
292 fDistr->Add(hptB);
a41f6fad 293
294 fDistr->Add(hdcaS);
295 fDistr->Add(hdcaB);
a41f6fad 296
297 fDistr->Add(hd0piS);
a41f6fad 298 fDistr->Add(hd0KS);
299 fDistr->Add(hd0B);
a41f6fad 300
301 fDistr->Add(hd0d0S);
302 fDistr->Add(hd0d0B);
a41f6fad 303
304 fDistr->Add(hcosthetastarS);
305 fDistr->Add(hcosthetastarB);
a41f6fad 306
307 fDistr->Add(hcosthetapointS);
308 fDistr->Add(hcosthetapointB);
a41f6fad 309
310 fDistr->Add(hcosthpointd0d0S);
311 fDistr->Add(hcosthpointd0d0B);
312
9de8c723 313 fNentries=new TH1F("nentriesD0", "nentriesD0->Integral(1,2) = number of AODs *** nentriesD0->Integral(3,4) = number of candidates selected with cuts *** nentriesD0->Integral(5,6) = number of D0 selected with cuts", 6,1.,4.);
a4ae02cd 314
49061176 315 return;
316}
317
318//________________________________________________________________________
319void AliAnalysisTaskSED0Mass::UserExec(Option_t */*option*/)
320{
321 // Execute analysis for current event:
322 // heavy flavor candidates association to MC truth
a4ae02cd 323 //cout<<"I'm in UserExec"<<endl;
49061176 324 AliAODEvent *aod = dynamic_cast<AliAODEvent*> (InputEvent());
feb73eca 325
b557eb43 326 TString bname;
feb73eca 327 if(fArray==0){ //D0 candidates
b557eb43 328 // load D0->Kpi candidates
feb73eca 329 //cout<<"D0 candidates"<<endl;
b557eb43 330 bname="D0toKpi";
feb73eca 331 } else { //LikeSign candidates
feb73eca 332 // load like sign candidates
b557eb43 333 //cout<<"LS candidates"<<endl;
334 bname="LikeSign2Prong";
335 }
336
337 TClonesArray *inputArray=0;
338
339 if(!aod && AODEvent() && IsStandardAOD()) {
340 // In case there is an AOD handler writing a standard AOD, use the AOD
341 // event in memory rather than the input (ESD) event.
342 aod = dynamic_cast<AliAODEvent*> (AODEvent());
343 // in this case the braches in the deltaAOD (AliAOD.VertexingHF.root)
344 // have to taken from the AOD event hold by the AliAODExtension
345 AliAODHandler* aodHandler = (AliAODHandler*)
346 ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
347 if(aodHandler->GetExtensions()) {
348 AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject("AliAOD.VertexingHF.root");
349 AliAODEvent *aodFromExt = ext->GetAOD();
350 inputArray=(TClonesArray*)aodFromExt->GetList()->FindObject(bname.Data());
feb73eca 351 }
b557eb43 352 } else {
353 inputArray=(TClonesArray*)aod->GetList()->FindObject(bname.Data());
354 }
feb73eca 355
b557eb43 356
357 if(!inputArray) {
358 printf("AliAnalysisTaskSED0Mass::UserExec: input branch not found!\n");
359 return;
49061176 360 }
361
362 // AOD primary vertex
363 AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
364 //vtx1->Print();
feb73eca 365
ce39f0ac 366 TClonesArray *mcArray = 0;
367 AliAODMCHeader *mcHeader = 0;
368
369 if(fReadMC) {
370 // load MC particles
371 mcArray = (TClonesArray*)aod->GetList()->FindObject(AliAODMCParticle::StdBranchName());
372 if(!mcArray) {
373 printf("AliAnalysisTaskSED0Mass::UserExec: MC particles branch not found!\n");
374 return;
375 }
49061176 376
ce39f0ac 377 // load MC header
378 mcHeader = (AliAODMCHeader*)aod->GetList()->FindObject(AliAODMCHeader::StdBranchName());
379 if(!mcHeader) {
380 printf("AliAnalysisTaskSED0Mass::UserExec: MC header branch not found!\n");
381 return;
382 }
49061176 383 }
384
b557eb43 385 //printf("VERTEX Z %f %f\n",vtx1->GetZ(),mcHeader->GetVtxZ());
a4ae02cd 386
387 //histogram filled with 1 for every AOD
388 fNentries->Fill(1);
389 PostData(3,fNentries);
9de8c723 390
feb73eca 391 // loop over candidates
392 Int_t nInD0toKpi = inputArray->GetEntriesFast();
4464ce7e 393 if(fDebug>1) printf("Number of D0->Kpi: %d\n",nInD0toKpi);
49061176 394
395 for (Int_t iD0toKpi = 0; iD0toKpi < nInD0toKpi; iD0toKpi++) {
feb73eca 396 //Int_t nPosPairs=0, nNegPairs=0;
a41f6fad 397 //cout<<"inside the loop"<<endl;
feb73eca 398 AliAODRecoDecayHF2Prong *d = (AliAODRecoDecayHF2Prong*)inputArray->UncheckedAt(iD0toKpi);
49061176 399 Bool_t unsetvtx=kFALSE;
400 if(!d->GetOwnPrimaryVtx()) {
401 d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
402 unsetvtx=kTRUE;
403 }
404
9de8c723 405 //check reco daughter in acceptance
406 Double_t eta0=d->EtaProng(0);
407 Double_t eta1=d->EtaProng(1);
408 Bool_t prongsinacc=kFALSE;
409 if (TMath::Abs(eta0) < 0.9 && TMath::Abs(eta1) < 0.9) prongsinacc=kTRUE;
a41f6fad 410
411 //add distr here
412 UInt_t pdgs[2];
49061176 413
a41f6fad 414 Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
415 pdgs[0]=211;
416 pdgs[1]=321;
417 Double_t minvD0 = d->InvMassD0();
418 pdgs[1]=211;
419 pdgs[0]=321;
420 Double_t minvD0bar = d->InvMassD0bar();
0108fa62 421 //cout<<"inside mass cut"<<endl;
422
423 Int_t pdgDgD0toKpi[2]={321,211};
424 Int_t lab=-9999;
425 if(fReadMC) lab=d->MatchToMC(421,mcArray,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
426 Double_t pt = d->Pt();
ce39f0ac 427
0108fa62 428 if(lab>=0 && fReadMC){ //signal
429 if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421){//D0
430 if(TMath::Abs(minvD0-mPDG)<0.03 || TMath::Abs(minvD0bar-mPDG)<0.03){
431 ((TH1F*)fDistr->FindObject("hcosthetastarS"))->Fill(d->CosThetaStarD0());
432 }
433
434 //no mass cut on mass distribution
435 if(pt>0. && pt<=1.) ((TH1F*)fDistr->FindObject("hMassS_1"))->Fill(minvD0);
436 if(pt>1. && pt<=2.) ((TH1F*)fDistr->FindObject("hMassS_2"))->Fill(minvD0);
437 if(pt>2. && pt<=3.) ((TH1F*)fDistr->FindObject("hMassS_3"))->Fill(minvD0);
438 if(pt>3. && pt<=5.) ((TH1F*)fDistr->FindObject("hMassS_4"))->Fill(minvD0);
439 if(pt>5.) ((TH1F*)fDistr->FindObject("hMassS_5"))->Fill(minvD0);
440
441 }
442 else { //D0bar
443 if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421){
444 ((TH1F*)fDistr->FindObject("hcosthetastarS"))->Fill(d->CosThetaStarD0bar());
445 }
446
447 //no mass cut on mass distribution
448 if(pt>0. && pt<=1.) ((TH1F*)fDistr->FindObject("hMassS_1"))->Fill(minvD0bar);
449 if(pt>1. && pt<=2.) ((TH1F*)fDistr->FindObject("hMassS_2"))->Fill(minvD0bar);
450 if(pt>2. && pt<=3.) ((TH1F*)fDistr->FindObject("hMassS_3"))->Fill(minvD0bar);
451 if(pt>3. && pt<=5.) ((TH1F*)fDistr->FindObject("hMassS_4"))->Fill(minvD0bar);
452 if(pt>5.) ((TH1F*)fDistr->FindObject("hMassS_5"))->Fill(minvD0bar);
453
454 }
455
456 //apply cut on invariant mass on the pair
457 if(TMath::Abs(minvD0-mPDG)<0.03 || TMath::Abs(minvD0bar-mPDG)<0.03){
458
459 if(fArray==1) cout<<"LS signal: ERROR"<<endl;
460 for (Int_t iprong=0; iprong<2; iprong++){
461 AliAODTrack *prong=(AliAODTrack*)d->GetDaughter(iprong);
462 Int_t labprong=prong->GetLabel();
463
464 //cout<<"prong name = "<<prong->GetName()<<" label = "<<prong->GetLabel()<<endl;
465 AliAODMCParticle *mcprong=0;
466 if(labprong>=0) mcprong= (AliAODMCParticle*)mcArray->At(labprong);
467 Int_t pdgprong=mcprong->GetPdgCode();
468 if(TMath::Abs(pdgprong)==211) {
469 //cout<<"pi"<<endl;
470 ((TH1F*)fDistr->FindObject("hptpiS"))->Fill(d->PtProng(iprong));
471 ((TH1F*)fDistr->FindObject("hd0piS"))->Fill(d->Getd0Prong(iprong));
472 }
473
474 if(TMath::Abs(pdgprong)==321) {
475 //cout<<"kappa"<<endl;
476 ((TH1F*)fDistr->FindObject("hptKS"))->Fill(d->PtProng(iprong));
477 ((TH1F*)fDistr->FindObject("hd0KS"))->Fill(d->Getd0Prong(iprong));
478 }
479 ((TH1F*)fDistr->FindObject("hdcaS"))->Fill(d->GetDCA());
480
481 }
482
483
484 ((TH1F*)fDistr->FindObject("hd0d0S"))->Fill(d->Prodd0d0());
485
486 ((TH1F*)fDistr->FindObject("hcosthetapointS"))->Fill(d->CosPointingAngle());
487 ((TH1F*)fDistr->FindObject("hcosthpointd0d0S"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
488
489 //cout<<"ok point"<<endl;
490
491 } //invmass cut
492
493 } else{ //Background or LS
494 //cout<<"is background"<<endl;
495 Double_t pt = d->Pt();
496 if(pt>0. && pt<=1.) {
497 ((TH1F*)fDistr->FindObject("hMassB_1"))->Fill(minvD0);
498 ((TH1F*)fDistr->FindObject("hMassB_1"))->Fill(minvD0bar);
499 }
500 if(pt>1. && pt<=2.) {
501 ((TH1F*)fDistr->FindObject("hMassB_2"))->Fill(minvD0);
502 ((TH1F*)fDistr->FindObject("hMassB_2"))->Fill(minvD0bar);
503 }
504 if(pt>2. && pt<=3.) {
505 ((TH1F*)fDistr->FindObject("hMassB_3"))->Fill(minvD0);
506 ((TH1F*)fDistr->FindObject("hMassB_3"))->Fill(minvD0bar);
507 }
508 if(pt>3. && pt<=5.) {
509 ((TH1F*)fDistr->FindObject("hMassB_4"))->Fill(minvD0);
510 ((TH1F*)fDistr->FindObject("hMassB_4"))->Fill(minvD0bar);
511 }
512 if(pt>5.) {
513 ((TH1F*)fDistr->FindObject("hMassB_5"))->Fill(minvD0);
514 ((TH1F*)fDistr->FindObject("hMassB_5"))->Fill(minvD0bar);
515 }
516
517 //apply cut on invariant mass on the pair
518 if(TMath::Abs(minvD0-mPDG)<0.03 || TMath::Abs(minvD0bar-mPDG)<0.03){
519
520
521 AliAODTrack *prong=(AliAODTrack*)d->GetDaughter(0);
522 if(!prong) cout<<"No daughter found";
523 else{
524 if(prong->Charge()==1) {fTotPosPairs[4]++;} else {fTotNegPairs[4]++;}
525 }
526 ((TH1F*)fDistr->FindObject("hptB"))->Fill(d->PtProng(0));
527 //cout<<"ptok"<<endl;
528 ((TH1F*)fDistr->FindObject("hd0B"))->Fill(d->Getd0Prong(0));
529 //cout<<"d0ok"<<endl;
530 ((TH1F*)fDistr->FindObject("hdcaB"))->Fill(d->GetDCA());
531 //cout<<"dcaok"<<endl;
532 ((TH1F*)fDistr->FindObject("hcosthetastarB"))->Fill(d->CosThetaStarD0());
533 ((TH1F*)fDistr->FindObject("hcosthetastarB"))->Fill(d->CosThetaStarD0bar());
534 ((TH1F*)fDistr->FindObject("hd0d0B"))->Fill(d->Prodd0d0());
535 //cout<<"d0d0ok"<<endl;
536 ((TH1F*)fDistr->FindObject("hcosthetapointB"))->Fill(d->CosPointingAngle());
537 ((TH1F*)fDistr->FindObject("hcosthpointd0d0B"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
538
539
540 }// end if inv mass cut
541 }//end if background
ce39f0ac 542
0108fa62 543
a41f6fad 544
9de8c723 545 //cuts order
546// printf(" |M-MD0| [GeV] < %f\n",fD0toKpiCuts[0]);
547// printf(" dca [cm] < %f\n",fD0toKpiCuts[1]);
548// printf(" cosThetaStar < %f\n",fD0toKpiCuts[2]);
549// printf(" pTK [GeV/c] > %f\n",fD0toKpiCuts[3]);
550// printf(" pTpi [GeV/c] > %f\n",fD0toKpiCuts[4]);
551// printf(" |d0K| [cm] < %f\n",fD0toKpiCuts[5]);
552// printf(" |d0pi| [cm] < %f\n",fD0toKpiCuts[6]);
553// printf(" d0d0 [cm^2] < %f\n",fD0toKpiCuts[7]);
554// printf(" cosThetaPoint > %f\n",fD0toKpiCuts[8]);
0108fa62 555
49061176 556 Int_t ptbin=0;
557
558 //cout<<"P_t = "<<pt<<endl;
559 if (pt>0. && pt<=1.) {
a4ae02cd 560 ptbin=1;
11ecfd50 561 /*
562 //test d0 cut
563 fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.1,-0.0002,0.5);
564 fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.1,-0.00025,0.7);
565 */
566 //original
9de8c723 567 fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.0002,0.5);
527f330b 568 fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.00025,0.7);
11ecfd50 569
9de8c723 570// fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.0002,0.7);
527f330b 571// fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,1,1,-0.00015,0.5);
49061176 572 //printf("I'm in the bin %d\n",ptbin);
a4ae02cd 573
49061176 574 }
11ecfd50 575
46c96ce5 576 if(pt>1. && pt<=3.) {
9de8c723 577 if(pt>1. && pt<=2.) ptbin=2;
578 if(pt>2. && pt<=3.) ptbin=3;
11ecfd50 579 /*
580 //test d0 cut
581 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.0002,0.6);
582 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,1,0.1,-0.00025,0.8);
583 */
584 //original
585 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.0002,0.6);
527f330b 586 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,1,1,-0.00025,0.8);
11ecfd50 587
46c96ce5 588 //printf("I'm in the bin %d\n",ptbin);
589 }
590
591 if(pt>3. && pt<=5.){
9de8c723 592 ptbin=4;
11ecfd50 593 /*
594 //test d0 cut
595 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.0001,0.8);
596 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00015,0.8);
597 */
598 //original
9de8c723 599 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.0001,0.8);
527f330b 600 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00015,0.8);
11ecfd50 601
49061176 602 //printf("I'm in the bin %d\n",ptbin);
49061176 603 }
46c96ce5 604 if(pt>5.){
9de8c723 605 ptbin=5;
11ecfd50 606 /*
607 //test d0 cut
608 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00005,0.8);
609 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00015,0.9);
610 */
611 //original
9de8c723 612 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00005,0.8);
527f330b 613 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00015,0.9);
11ecfd50 614
615 }//if(pt>5) if (pt>0. && pt<=1.) {
a4ae02cd 616 //printf("I'm in the bin %d\n",ptbin);
617 //old
618 //fVHF->SetD0toKpiCuts(0.7,0.03,0.8,0.06,0.06,0.05,0.05,-0.0002,0.6); //2.p-p vertex reconstructed
9de8c723 619 if(prongsinacc){
620 FillHists(ptbin,d,mcArray,fVHFPPR,fOutputPPR);
527f330b 621 FillHists(ptbin,d,mcArray,fVHFmycuts,fOutputmycuts);
9de8c723 622 }
49061176 623 if(unsetvtx) d->UnsetOwnPrimaryVtx();
0108fa62 624 } //end for prongs
46c96ce5 625
49061176 626
627
628
629 // Post the data
9de8c723 630 PostData(1,fOutputPPR);
527f330b 631 PostData(2,fOutputmycuts);
a41f6fad 632 PostData(4,fDistr);
49061176 633 return;
634}
635//____________________________________________________________________________*
a4ae02cd 636void AliAnalysisTaskSED0Mass::FillHists(Int_t ptbin, AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliAnalysisVertexingHF *vhf, TList *listout){
49061176 637 //
638 // function used in UserExec:
639 //
9de8c723 640
641
642 Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
643
49061176 644 Int_t okD0=0,okD0bar=0;
feb73eca 645 //cout<<"inside FillHist"<<endl;
a4ae02cd 646 if(part->SelectD0(vhf->GetD0toKpiCuts(),okD0,okD0bar)) {//selected
49061176 647 Double_t invmassD0 = part->InvMassD0(), invmassD0bar = part->InvMassD0bar();
648 //printf("SELECTED\n");
feb73eca 649
9de8c723 650
feb73eca 651 AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(0);
652 if(!prong) cout<<"No daughter found";
653 else{
9de8c723 654 if(prong->Charge()==1) {fTotPosPairs[ptbin-1]++;} else {fTotNegPairs[ptbin-1]++;}
feb73eca 655 }
656
657 TString fillthis="";
c8112d2f 658 Int_t pdgDgD0toKpi[2]={321,211};
ce39f0ac 659 Int_t labD0=-1;
660 if (fReadMC) labD0 = part->MatchToMC(421,arrMC,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
a41f6fad 661
9de8c723 662 //count candidates selected by cuts
663 fNentries->Fill(2);
664 //count true D0 selected by cuts
665 if (labD0>=0) fNentries->Fill(3);
666 PostData(3,fNentries);
667
a4ae02cd 668 if (okD0==1) {
669 fillthis="histMass_";
7646d6da 670 fillthis+=ptbin;
671 //cout<<"Filling "<<fillthis<<endl;
672
49061176 673 //printf("Fill mass with D0");
a4ae02cd 674 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
feb73eca 675
a4ae02cd 676 if(labD0>=0) {
feb73eca 677 if(fArray==1) cout<<"LS signal ERROR"<<endl;
678
a4ae02cd 679 AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
680 Int_t pdgD0 = partD0->GetPdgCode();
681 //cout<<"pdg = "<<pdgD0<<endl;
a4ae02cd 682 if (pdgD0==421){ //D0
683 //cout<<"Fill S with D0"<<endl;
46c96ce5 684 fillthis="histSgn_";
685 fillthis+=ptbin;
a4ae02cd 686 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
9de8c723 687 if(TMath::Abs(invmassD0 - mPDG) < 0.027){
688 fillthis="histSgn27_";
689 fillthis+=ptbin;
690 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
691 }
a4ae02cd 692 } else{ //it was a D0bar
46c96ce5 693 fillthis="histRfl_";
694 fillthis+=ptbin;
695 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
a4ae02cd 696 }
697 } else {//background
698 fillthis="histBkg_";
699 fillthis+=ptbin;
700 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
701 }
feb73eca 702
49061176 703 }
704 if (okD0bar==1) {
a4ae02cd 705 fillthis="histMass_";
706 fillthis+=ptbin;
49061176 707 //printf("Fill mass with D0bar");
a4ae02cd 708 ((TH1F*)listout->FindObject(fillthis))->Fill(invmassD0bar);
feb73eca 709
a4ae02cd 710 if(labD0>=0) {
feb73eca 711 if(fArray==1) cout<<"LS signal ERROR"<<endl;
a4ae02cd 712 AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
713 Int_t pdgD0 = partD0->GetPdgCode();
714 //cout<<" pdg = "<<pdgD0<<endl;
715 if (pdgD0==-421){ //D0bar
716 fillthis="histSgn_";
717 fillthis+=ptbin;
718 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
9de8c723 719 if (TMath::Abs(invmassD0bar - mPDG) < 0.027){
720 fillthis="histSgn27_";
721 fillthis+=ptbin;
722 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
723 }
724
a4ae02cd 725
726 } else{
46c96ce5 727 fillthis="histRfl_";
728 fillthis+=ptbin;
729 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
a4ae02cd 730 }
feb73eca 731 } else {//background or LS
a4ae02cd 732 fillthis="histBkg_";
733 fillthis+=ptbin;
734 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
735 }
736 }
737
738
49061176 739 } //else cout<<"NOT SELECTED"<<endl;
740
741}
742//________________________________________________________________________
743void AliAnalysisTaskSED0Mass::Terminate(Option_t */*option*/)
744{
745 // Terminate analysis
746 //
747 if(fDebug > 1) printf("AnalysisTaskSED0Mass: Terminate() \n");
748
9de8c723 749 fOutputPPR = dynamic_cast<TList*> (GetOutputData(1));
750 if (!fOutputPPR) {
a41f6fad 751 printf("ERROR: fOutputthight not available\n");
a4ae02cd 752 return;
753 }
527f330b 754 fOutputmycuts = dynamic_cast<TList*> (GetOutputData(2));
755 if (!fOutputmycuts) {
756 printf("ERROR: fOutputmycuts not available\n");
a41f6fad 757 return;
758 }
9de8c723 759 fDistr = dynamic_cast<TList*> (GetOutputData(4));
a41f6fad 760 if (!fDistr) {
761 printf("ERROR: fDistr not available\n");
49061176 762 return;
763 }
764
feb73eca 765 if(fArray==1){
766 for(Int_t ipt=0;ipt<4;ipt++){
767 fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[ipt]*fTotNegPairs[ipt]);
768
769
770 if(fLsNormalization>0) {
9de8c723 771
feb73eca 772 TString massName="histMass_";
773 massName+=ipt+1;
9de8c723 774 ((TH1F*)fOutputPPR->FindObject(massName))->Scale((1/fLsNormalization)*((TH1F*)fOutputPPR->FindObject(massName))->GetEntries());
527f330b 775 ((TH1F*)fOutputmycuts->FindObject(massName))->Scale((1/fLsNormalization)*((TH1F*)fOutputmycuts->FindObject(massName))->GetEntries());
feb73eca 776 }
777 }
778
779 fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[4]*fTotNegPairs[4]);
780
781 if(fLsNormalization>0) {
782
783 TString nameDistr="hptB";
784 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
785 nameDistr="hdcaB";
786 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
787 nameDistr="hcosthetastarB";
788 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
789 nameDistr="hd0B";
790 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
791 nameDistr="hd0d0B";
792 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
793 nameDistr="hcosthetapointB";
794 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
795 nameDistr="hcosthpointd0d0B";
796 ((TH2F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH2F*)fDistr->FindObject(nameDistr))->GetEntries());
797
798 }
799 }
6ce5994f 800
527f330b 801 TString cvname;
802
803 if (fArray==0){
804 cvname="D0invmass";
805 } else cvname="LSinvmass";
806
807 TCanvas *c1=new TCanvas(cvname,cvname);
808 c1->cd();
809 ((TH1F*)fOutputPPR->FindObject("histMass_4"))->Draw();
810
811
49061176 812 return;
813}
814