]> 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
Updates (Chiara)
[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.)
6321ee46 83
49061176 84{
85 // Default constructor
9de8c723 86 for(Int_t i=0;i<5;i++) {fTotPosPairs[i]=0; fTotNegPairs[i]=0;}
49061176 87
88 // Output slot #1 writes into a TList container
89 DefineOutput(1,TList::Class()); //My private output
a4ae02cd 90 // Output slot #2 writes into a TList container
91 DefineOutput(2,TList::Class()); //My private output
92 // Output slot #3 writes into a TH1F container
93 DefineOutput(3,TH1F::Class()); //My private output
a41f6fad 94 // Output slot #4 writes into a TList container
95 DefineOutput(4,TList::Class()); //My private output
49061176 96}
97
98//________________________________________________________________________
99AliAnalysisTaskSED0Mass::~AliAnalysisTaskSED0Mass()
100{
9de8c723 101 if (fOutputPPR) {
102 delete fOutputPPR;
103 fOutputPPR = 0;
a4ae02cd 104 }
9de8c723 105 if (fVHFPPR) {
106 delete fVHFPPR;
107 fVHFPPR = 0;
a4ae02cd 108 }
527f330b 109 if (fOutputmycuts) {
110 delete fOutputmycuts;
111 fOutputmycuts = 0;
a4ae02cd 112 }
a41f6fad 113
114 if (fDistr) {
115 delete fDistr;
116 fDistr = 0;
117 }
118
527f330b 119 if (fVHFmycuts) {
120 delete fVHFmycuts;
121 fVHFmycuts = 0;
49061176 122 }
46c96ce5 123 if (fNentries){
124 delete fNentries;
125 fNentries = 0;
126 }
a41f6fad 127
128
49061176 129}
130
131//________________________________________________________________________
132void AliAnalysisTaskSED0Mass::Init()
133{
134 // Initialization
135
136 if(fDebug > 1) printf("AnalysisTaskSED0Mass::Init() \n");
137
138 gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/ConfigVertexingHF.C");
139
a4ae02cd 140 // 2 sets of dedidcated cuts -- defined in UserExec
9de8c723 141 fVHFPPR = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
527f330b 142 fVHFmycuts = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
a4ae02cd 143
49061176 144 return;
145}
146
147//________________________________________________________________________
148void AliAnalysisTaskSED0Mass::UserCreateOutputObjects()
149{
150
151 // Create the output container
152 //
153 if(fDebug > 1) printf("AnalysisTaskSED0Mass::UserCreateOutputObjects() \n");
154
155 // Several histograms are more conveniently managed in a TList
9de8c723 156 fOutputPPR = new TList();
157 fOutputPPR->SetOwner();
158 fOutputPPR->SetName("listPPR");
a4ae02cd 159
527f330b 160 fOutputmycuts = new TList();
161 fOutputmycuts->SetOwner();
162 fOutputmycuts->SetName("listloose");
49061176 163
a41f6fad 164 fDistr = new TList();
165 fDistr->SetOwner();
166 fDistr->SetName("distributionslist");
167
9de8c723 168 const Int_t nhist=5;
a4ae02cd 169
0108fa62 170 TString nameMass=" ",nameSgn27=" ",nameSgn=" ", nameBkg=" ", nameRfl=" ",nameMassNocutsS =" ",nameMassNocutsB =" ", namedistr=" ";
49061176 171
172 for(Int_t i=0;i<nhist;i++){
b272aebf 173
7646d6da 174 nameMass="histMass_";
49061176 175 nameMass+=i+1;
9de8c723 176 nameSgn27="histSgn27_";
177 nameSgn27+=i+1;
7646d6da 178 nameSgn="histSgn_";
49061176 179 nameSgn+=i+1;
7646d6da 180 nameBkg="histBkg_";
49061176 181 nameBkg+=i+1;
a4ae02cd 182 nameRfl="histRfl_";
183 nameRfl+=i+1;
0108fa62 184 nameMassNocutsS="hMassS_";
185 nameMassNocutsS+=i+1;
186 nameMassNocutsB="hMassB_";
187 nameMassNocutsB+=i+1;
7646d6da 188
b272aebf 189 //histograms of cut variable distributions
190
191 // pT
192 namedistr="hptpiS_";
193 namedistr+=i+1;
194 TH1F *hptpiS = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
195
196 namedistr="hptKS_";
197 namedistr+=i+1;
198 TH1F *hptKS = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
199
200 namedistr="hptB_";
201 namedistr+=i+1;
202 TH1F *hptB = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
203
204 // pT no mass cut
205 namedistr="hptpiSnoMcut_";
206 namedistr+=i+1;
207 TH1F *hptpiSnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
208
209 namedistr="hptKSnoMcut_";
210 namedistr+=i+1;
211 TH1F *hptKSnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
212
213 namedistr="hptB1prongnoMcut_";
214 namedistr+=i+1;
215 TH1F *hptB1pnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
216
217 namedistr="hptB2prongsnoMcut_";
218 namedistr+=i+1;
219 TH1F *hptB2pnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
220
221
222 // dca
223 namedistr="hdcaS_";
224 namedistr+=i+1;
225 TH1F *hdcaS = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
226 namedistr="hdcaB_";
227 namedistr+=i+1;
228 TH1F *hdcaB = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
229
230 // costhetastar
231 namedistr="hcosthetastarS_";
232 namedistr+=i+1;
233 TH1F *hcosthetastarS = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
234 namedistr="hcosthetastarB_";
235 namedistr+=i+1;
236 TH1F *hcosthetastarB = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
237
238 // impact parameter
239 namedistr="hd0piS_";
240 namedistr+=i+1;
241 TH1F *hd0piS = new TH1F(namedistr.Data(), "Impact parameter distribution (pions);d0(#pi) [cm]",200,-0.1,0.1);
242
243 namedistr="hd0KS_";
244 namedistr+=i+1;
245 TH1F *hd0KS = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons);d0(K) [cm]",200,-0.1,0.1);
246 namedistr="hd0B_";
247 namedistr+=i+1;
248 TH1F *hd0B = new TH1F(namedistr.Data(), "Impact parameter distribution;d0 [cm]",200,-0.1,0.1);
249
250 namedistr="hd0d0S_";
251 namedistr+=i+1;
252 TH1F *hd0d0S = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
253 namedistr="hd0d0B_";
254 namedistr+=i+1;
255 TH1F *hd0d0B = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
256
257 // costhetapoint
258 namedistr="hcosthetapointS_";
259 namedistr+=i+1;
260 TH1F *hcosthetapointS = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
261 namedistr="hcosthetapointB_";
262 namedistr+=i+1;
263 TH1F *hcosthetapointB = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
264
265 namedistr="hcosthpointd0d0S_";
266 namedistr+=i+1;
267 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);
268 namedistr="hcosthpointd0d0B_";
269 namedistr+=i+1;
270 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);
271
272 fDistr->Add(hptpiS);
273 fDistr->Add(hptKS);
274 fDistr->Add(hptB);
275
276 fDistr->Add(hptpiSnoMcut);
277 fDistr->Add(hptKSnoMcut);
278 fDistr->Add(hptB1pnoMcut);
279 fDistr->Add(hptB2pnoMcut);
280
281 fDistr->Add(hdcaS);
282 fDistr->Add(hdcaB);
283
284 fDistr->Add(hd0piS);
285 fDistr->Add(hd0KS);
286 fDistr->Add(hd0B);
287
288 fDistr->Add(hd0d0S);
289 fDistr->Add(hd0d0B);
290
291 fDistr->Add(hcosthetastarS);
292 fDistr->Add(hcosthetastarB);
293
294 fDistr->Add(hcosthetapointS);
295 fDistr->Add(hcosthetapointB);
296
297 fDistr->Add(hcosthpointd0d0S);
298 fDistr->Add(hcosthpointd0d0B);
299
300
a41f6fad 301 //histograms of invariant mass distributions
302
46c96ce5 303 TH1F* tmpMt = new TH1F(nameMass.Data(),"D^{0} invariant mass; M [GeV]; Entries",200,1.765,1.965);
304 TH1F *tmpMl=(TH1F*)tmpMt->Clone();
305 tmpMt->Sumw2();
306 tmpMl->Sumw2();
a4ae02cd 307
9de8c723 308 //to compare with AliAnalysisTaskCharmFraction
309 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);
310 TH1F *tmpS27l=(TH1F*)tmpS27t->Clone();
311 tmpS27t->Sumw2();
312 tmpS27l->Sumw2();
0108fa62 313
314 //distribution w/o cuts
6321ee46 315 // TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",300,0.7,3.);
316 TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",300,1.56484,2.16484); //range (MD0-300MeV, mD0 + 300MeV)
317 TH1F *tmpMB=(TH1F*)tmpMS->Clone();
0108fa62 318 tmpMB->SetName(nameMassNocutsB.Data());
319 tmpMS->Sumw2();
320 tmpMB->Sumw2();
321
322 //MC signal and background
46c96ce5 323 TH1F* tmpSt = new TH1F(nameSgn.Data(), "D^{0} invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
324 TH1F *tmpSl=(TH1F*)tmpSt->Clone();
325 tmpSt->Sumw2();
326 tmpSl->Sumw2();
a4ae02cd 327
46c96ce5 328 TH1F* tmpBt = new TH1F(nameBkg.Data(), "Background invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
329 TH1F *tmpBl=(TH1F*)tmpBt->Clone();
330 tmpBt->Sumw2();
331 tmpBl->Sumw2();
a4ae02cd 332
feb73eca 333 //Reflection: histo filled with D0Mass which pass the cut (also) as D0bar and with D0bar which pass (also) the cut as D0
46c96ce5 334 TH1F* tmpRt = new TH1F(nameRfl.Data(), "Reflected signal invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
335 TH1F *tmpRl=(TH1F*)tmpRt->Clone();
336 tmpRt->Sumw2();
337 tmpRl->Sumw2();
a4ae02cd 338 // printf("Created histograms %s\t%s\t%s\t%s\n",tmpM->GetName(),tmpS->GetName(),tmpB->GetName(),tmpR->GetName());
339
9de8c723 340 fOutputPPR->Add(tmpMt);
341 fOutputPPR->Add(tmpSt);
342 fOutputPPR->Add(tmpS27t);
343 fOutputPPR->Add(tmpBt);
344 fOutputPPR->Add(tmpRt);
a4ae02cd 345
527f330b 346 fOutputmycuts->Add(tmpMl);
347 fOutputmycuts->Add(tmpSl);
348 fOutputmycuts->Add(tmpS27l);
349 fOutputmycuts->Add(tmpBl);
350 fOutputmycuts->Add(tmpRl);
0108fa62 351
352 fDistr->Add(tmpMS);
353 fDistr->Add(tmpMB);
354
7646d6da 355 }
a4ae02cd 356
a41f6fad 357
9de8c723 358 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 359
49061176 360 return;
361}
362
363//________________________________________________________________________
364void AliAnalysisTaskSED0Mass::UserExec(Option_t */*option*/)
365{
366 // Execute analysis for current event:
367 // heavy flavor candidates association to MC truth
a4ae02cd 368 //cout<<"I'm in UserExec"<<endl;
49061176 369 AliAODEvent *aod = dynamic_cast<AliAODEvent*> (InputEvent());
feb73eca 370
b557eb43 371 TString bname;
feb73eca 372 if(fArray==0){ //D0 candidates
b557eb43 373 // load D0->Kpi candidates
feb73eca 374 //cout<<"D0 candidates"<<endl;
b557eb43 375 bname="D0toKpi";
feb73eca 376 } else { //LikeSign candidates
feb73eca 377 // load like sign candidates
b557eb43 378 //cout<<"LS candidates"<<endl;
379 bname="LikeSign2Prong";
380 }
381
382 TClonesArray *inputArray=0;
383
384 if(!aod && AODEvent() && IsStandardAOD()) {
385 // In case there is an AOD handler writing a standard AOD, use the AOD
386 // event in memory rather than the input (ESD) event.
387 aod = dynamic_cast<AliAODEvent*> (AODEvent());
388 // in this case the braches in the deltaAOD (AliAOD.VertexingHF.root)
389 // have to taken from the AOD event hold by the AliAODExtension
390 AliAODHandler* aodHandler = (AliAODHandler*)
391 ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
392 if(aodHandler->GetExtensions()) {
393 AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject("AliAOD.VertexingHF.root");
394 AliAODEvent *aodFromExt = ext->GetAOD();
395 inputArray=(TClonesArray*)aodFromExt->GetList()->FindObject(bname.Data());
feb73eca 396 }
b557eb43 397 } else {
398 inputArray=(TClonesArray*)aod->GetList()->FindObject(bname.Data());
399 }
feb73eca 400
b557eb43 401
402 if(!inputArray) {
403 printf("AliAnalysisTaskSED0Mass::UserExec: input branch not found!\n");
404 return;
49061176 405 }
406
407 // AOD primary vertex
408 AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
409 //vtx1->Print();
feb73eca 410
ce39f0ac 411 TClonesArray *mcArray = 0;
412 AliAODMCHeader *mcHeader = 0;
413
414 if(fReadMC) {
415 // load MC particles
416 mcArray = (TClonesArray*)aod->GetList()->FindObject(AliAODMCParticle::StdBranchName());
417 if(!mcArray) {
418 printf("AliAnalysisTaskSED0Mass::UserExec: MC particles branch not found!\n");
419 return;
420 }
49061176 421
ce39f0ac 422 // load MC header
423 mcHeader = (AliAODMCHeader*)aod->GetList()->FindObject(AliAODMCHeader::StdBranchName());
424 if(!mcHeader) {
425 printf("AliAnalysisTaskSED0Mass::UserExec: MC header branch not found!\n");
426 return;
427 }
49061176 428 }
429
b557eb43 430 //printf("VERTEX Z %f %f\n",vtx1->GetZ(),mcHeader->GetVtxZ());
a4ae02cd 431
432 //histogram filled with 1 for every AOD
433 fNentries->Fill(1);
434 PostData(3,fNentries);
9de8c723 435
feb73eca 436 // loop over candidates
437 Int_t nInD0toKpi = inputArray->GetEntriesFast();
4464ce7e 438 if(fDebug>1) printf("Number of D0->Kpi: %d\n",nInD0toKpi);
49061176 439
440 for (Int_t iD0toKpi = 0; iD0toKpi < nInD0toKpi; iD0toKpi++) {
feb73eca 441 //Int_t nPosPairs=0, nNegPairs=0;
a41f6fad 442 //cout<<"inside the loop"<<endl;
feb73eca 443 AliAODRecoDecayHF2Prong *d = (AliAODRecoDecayHF2Prong*)inputArray->UncheckedAt(iD0toKpi);
49061176 444 Bool_t unsetvtx=kFALSE;
445 if(!d->GetOwnPrimaryVtx()) {
446 d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
447 unsetvtx=kTRUE;
448 }
449
9de8c723 450 //check reco daughter in acceptance
451 Double_t eta0=d->EtaProng(0);
452 Double_t eta1=d->EtaProng(1);
453 Bool_t prongsinacc=kFALSE;
454 if (TMath::Abs(eta0) < 0.9 && TMath::Abs(eta1) < 0.9) prongsinacc=kTRUE;
a41f6fad 455
456 //add distr here
457 UInt_t pdgs[2];
49061176 458
a41f6fad 459 Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
460 pdgs[0]=211;
461 pdgs[1]=321;
462 Double_t minvD0 = d->InvMassD0();
463 pdgs[1]=211;
464 pdgs[0]=321;
465 Double_t minvD0bar = d->InvMassD0bar();
0108fa62 466 //cout<<"inside mass cut"<<endl;
467
468 Int_t pdgDgD0toKpi[2]={321,211};
469 Int_t lab=-9999;
470 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)
b272aebf 471 Double_t pt = d->Pt(); //mother pt
ce39f0ac 472
b272aebf 473 if(lab>=0 && fReadMC){ //signal
474
475 //check pdg of the prongs
476 AliAODTrack *prong0=(AliAODTrack*)d->GetDaughter(0);
477 AliAODTrack *prong1=(AliAODTrack*)d->GetDaughter(1);
478 Int_t labprong[2];
479 labprong[0]=prong0->GetLabel();
480 labprong[1]=prong1->GetLabel();
481 AliAODMCParticle *mcprong=0;
482 Int_t PDGprong[2]={0,0};
483 for (Int_t iprong=0;iprong<2;iprong++){
484 if(labprong[iprong]>=0) mcprong= (AliAODMCParticle*)mcArray->At(labprong[iprong]);
485 PDGprong[iprong]=mcprong->GetPdgCode();
486 }
487
488 //no mass cut ditributions: ptbis
489
490 if(pt>0. && pt<=1.) {
491 if (TMath::Abs(PDGprong[0]) == 211 && TMath::Abs(PDGprong[1]) == 321){
492 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_1"))->Fill(d->PtProng(0));
493 ((TH1F*)fDistr->FindObject("hptKSnoMcut_1"))->Fill(d->PtProng(1));
494 }else {
495 if (TMath::Abs(PDGprong[0]) == 321 && TMath::Abs(PDGprong[1]) == 211){
496 ((TH1F*)fDistr->FindObject("hptKSnoMcut_1"))->Fill(d->PtProng(0));
497 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_1"))->Fill(d->PtProng(1));
498 }
499 }
500 }
501 if(pt>1. && pt<=2.) {
502 if (TMath::Abs(PDGprong[0]) == 211 && TMath::Abs(PDGprong[1]) == 321){
503 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_2"))->Fill(d->PtProng(0));
504 ((TH1F*)fDistr->FindObject("hptKSnoMcut_2"))->Fill(d->PtProng(1));
505 }else {
506 if (TMath::Abs(PDGprong[0]) == 321 && TMath::Abs(PDGprong[1]) == 211){
507 ((TH1F*)fDistr->FindObject("hptKSnoMcut_2"))->Fill(d->PtProng(0));
508 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_2"))->Fill(d->PtProng(1));
509 }
510 }
511 }
512 if(pt>2. && pt<=3.) {
513 if (TMath::Abs(PDGprong[0]) == 211 && TMath::Abs(PDGprong[1]) == 321){
514 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_3"))->Fill(d->PtProng(0));
515 ((TH1F*)fDistr->FindObject("hptKSnoMcut_3"))->Fill(d->PtProng(1));
516 }else {
517 if (TMath::Abs(PDGprong[0]) == 321 && TMath::Abs(PDGprong[1]) == 211){
518 ((TH1F*)fDistr->FindObject("hptKSnoMcut_3"))->Fill(d->PtProng(0));
519 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_3"))->Fill(d->PtProng(1));
520 }
521 }
522 }
523 if(pt>3. && pt<=5.) {
524
525 if (TMath::Abs(PDGprong[0]) == 211 && TMath::Abs(PDGprong[1]) == 321){
526 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_4"))->Fill(d->PtProng(0));
527 ((TH1F*)fDistr->FindObject("hptKSnoMcut_4"))->Fill(d->PtProng(1));
528 }else {
529 if (TMath::Abs(PDGprong[0]) == 321 && TMath::Abs(PDGprong[1]) == 211){
530 ((TH1F*)fDistr->FindObject("hptKSnoMcut_4"))->Fill(d->PtProng(0));
531 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_4"))->Fill(d->PtProng(1));
532 }
533 }
534 }
535 if(pt>5.) {
536
537 if (TMath::Abs(PDGprong[0]) == 211 && TMath::Abs(PDGprong[1]) == 321){
538 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_5"))->Fill(d->PtProng(0));
539 ((TH1F*)fDistr->FindObject("hptKSnoMcut_5"))->Fill(d->PtProng(1));
540 }else {
541 if (TMath::Abs(PDGprong[0]) == 321 && TMath::Abs(PDGprong[1]) == 211){
542 ((TH1F*)fDistr->FindObject("hptKSnoMcut_5"))->Fill(d->PtProng(0));
543 ((TH1F*)fDistr->FindObject("hptpiSnoMcut_5"))->Fill(d->PtProng(1));
544 }
545 }
546 }
547
0108fa62 548 if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421){//D0
b272aebf 549
550 //no mass cut ditributions: mass, costhetastar
551
552 if(pt>0. && pt<=1.) {
553 ((TH1F*)fDistr->FindObject("hMassS_1"))->Fill(minvD0);
554 }
555 if(pt>1. && pt<=2.) {
556 ((TH1F*)fDistr->FindObject("hMassS_2"))->Fill(minvD0);
557
558 }
559 if(pt>2. && pt<=3.) {
560 ((TH1F*)fDistr->FindObject("hMassS_3"))->Fill(minvD0);
561
562 }
563 if(pt>3. && pt<=5.) {
564 ((TH1F*)fDistr->FindObject("hMassS_4"))->Fill(minvD0);
565
566 }
567 if(pt>5.) {
568 ((TH1F*)fDistr->FindObject("hMassS_5"))->Fill(minvD0);
569
0108fa62 570 }
0108fa62 571
572 }
573 else { //D0bar
b272aebf 574
575 //no mass cut ditributions: mass
576
577 if(pt>0. && pt<=1.) {
578 ((TH1F*)fDistr->FindObject("hMassS_1"))->Fill(minvD0bar);
579
580 }
581 if(pt>1. && pt<=2.) {
582 ((TH1F*)fDistr->FindObject("hMassS_2"))->Fill(minvD0bar);
583
0108fa62 584 }
b272aebf 585 if(pt>2. && pt<=3.) {
586 ((TH1F*)fDistr->FindObject("hMassS_3"))->Fill(minvD0bar);
0108fa62 587
b272aebf 588 }
589 if(pt>3. && pt<=5.) {
590 ((TH1F*)fDistr->FindObject("hMassS_4"))->Fill(minvD0bar);
591
592 }
593 if(pt>5.) {
594 ((TH1F*)fDistr->FindObject("hMassS_5"))->Fill(minvD0bar);
595 }
0108fa62 596
597 }
598
599 //apply cut on invariant mass on the pair
600 if(TMath::Abs(minvD0-mPDG)<0.03 || TMath::Abs(minvD0bar-mPDG)<0.03){
601
602 if(fArray==1) cout<<"LS signal: ERROR"<<endl;
603 for (Int_t iprong=0; iprong<2; iprong++){
604 AliAODTrack *prong=(AliAODTrack*)d->GetDaughter(iprong);
605 Int_t labprong=prong->GetLabel();
606
607 //cout<<"prong name = "<<prong->GetName()<<" label = "<<prong->GetLabel()<<endl;
608 AliAODMCParticle *mcprong=0;
609 if(labprong>=0) mcprong= (AliAODMCParticle*)mcArray->At(labprong);
610 Int_t pdgprong=mcprong->GetPdgCode();
611 if(TMath::Abs(pdgprong)==211) {
612 //cout<<"pi"<<endl;
b272aebf 613 if(pt>0. && pt<=1.){
614 ((TH1F*)fDistr->FindObject("hptpiS_1"))->Fill(d->PtProng(iprong));
615 ((TH1F*)fDistr->FindObject("hd0piS_1"))->Fill(d->Getd0Prong(iprong));
616 }
617 if(pt>1. && pt<=2.){
618 ((TH1F*)fDistr->FindObject("hptpiS_2"))->Fill(d->PtProng(iprong));
619 ((TH1F*)fDistr->FindObject("hd0piS_2"))->Fill(d->Getd0Prong(iprong));
620 }
621
622 if(pt>2. && pt<=3.){
623 ((TH1F*)fDistr->FindObject("hptpiS_3"))->Fill(d->PtProng(iprong));
624 ((TH1F*)fDistr->FindObject("hd0piS_3"))->Fill(d->Getd0Prong(iprong));
0108fa62 625
b272aebf 626 }
627 if(pt>3. && pt<=5.){
628 ((TH1F*)fDistr->FindObject("hptpiS_4"))->Fill(d->PtProng(iprong));
629 ((TH1F*)fDistr->FindObject("hd0piS_4"))->Fill(d->Getd0Prong(iprong));
630
631 }
632 if(pt>5.) {
633 ((TH1F*)fDistr->FindObject("hptpiS_5"))->Fill(d->PtProng(iprong));
634 ((TH1F*)fDistr->FindObject("hd0piS_5"))->Fill(d->Getd0Prong(iprong));
635
636 }
637
638
639 }
640
641 if(TMath::Abs(pdgprong)==321) {
0108fa62 642 //cout<<"kappa"<<endl;
b272aebf 643 if(pt>0. && pt<=1.){
644 ((TH1F*)fDistr->FindObject("hptKS_1"))->Fill(d->PtProng(iprong));
645 ((TH1F*)fDistr->FindObject("hd0KS_1"))->Fill(d->Getd0Prong(iprong));
646 }
647 if(pt>1. && pt<=2.){
648 ((TH1F*)fDistr->FindObject("hptKS_2"))->Fill(d->PtProng(iprong));
649 ((TH1F*)fDistr->FindObject("hd0KS_2"))->Fill(d->Getd0Prong(iprong));
650 }
651 if(pt>2. && pt<=3.){
652 ((TH1F*)fDistr->FindObject("hptKS_3"))->Fill(d->PtProng(iprong));
653 ((TH1F*)fDistr->FindObject("hd0KS_3"))->Fill(d->Getd0Prong(iprong));
654 }
655 if(pt>3. && pt<=5.){
656 ((TH1F*)fDistr->FindObject("hptKS_4"))->Fill(d->PtProng(iprong));
657 ((TH1F*)fDistr->FindObject("hd0KS_4"))->Fill(d->Getd0Prong(iprong));
658
659 }
660 if(pt>5.) {
661 ((TH1F*)fDistr->FindObject("hptKS_5"))->Fill(d->PtProng(iprong));
662 ((TH1F*)fDistr->FindObject("hd0KS_5"))->Fill(d->Getd0Prong(iprong));
663
664 }
665
0108fa62 666 }
0108fa62 667
b272aebf 668 if(pt>0. && pt<=1.){
669 ((TH1F*)fDistr->FindObject("hdcaS_1"))->Fill(d->GetDCA());
670
671 if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
672 ((TH1F*)fDistr->FindObject("hcosthetastarS_1"))->Fill(d->CosThetaStarD0());
673 else ((TH1F*)fDistr->FindObject("hcosthetastarS_1"))->Fill(d->CosThetaStarD0bar());
674
675 }
676 if(pt>1. && pt<=2.){
677 ((TH1F*)fDistr->FindObject("hdcaS_2"))->Fill(d->GetDCA());
678
679 if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
680 ((TH1F*)fDistr->FindObject("hcosthetastarS_2"))->Fill(d->CosThetaStarD0());
681 else ((TH1F*)fDistr->FindObject("hcosthetastarS_2"))->Fill(d->CosThetaStarD0bar());
682
683 }
684 if(pt>2. && pt<=3.){
685 ((TH1F*)fDistr->FindObject("hdcaS_3"))->Fill(d->GetDCA());
0108fa62 686
b272aebf 687 if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
688 ((TH1F*)fDistr->FindObject("hcosthetastarS_3"))->Fill(d->CosThetaStarD0());
689 else ((TH1F*)fDistr->FindObject("hcosthetastarS_3"))->Fill(d->CosThetaStarD0bar());
0108fa62 690
b272aebf 691 }
692 if(pt>3. && pt<=5.){
693 ((TH1F*)fDistr->FindObject("hdcaS_4"))->Fill(d->GetDCA());
694
695 if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
696 ((TH1F*)fDistr->FindObject("hcosthetastarS_4"))->Fill(d->CosThetaStarD0());
697 else ((TH1F*)fDistr->FindObject("hcosthetastarS_4"))->Fill(d->CosThetaStarD0bar());
698
699 }
700 if(pt>5.) {
701 ((TH1F*)fDistr->FindObject("hdcaS_5"))->Fill(d->GetDCA());
0108fa62 702
b272aebf 703 if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
704 ((TH1F*)fDistr->FindObject("hcosthetastarS_5"))->Fill(d->CosThetaStarD0());
705 else ((TH1F*)fDistr->FindObject("hcosthetastarS_5"))->Fill(d->CosThetaStarD0bar());
706
707 }
708
6321ee46 709 }
0108fa62 710
b272aebf 711 if(pt>0. && pt<=1.){
712 ((TH1F*)fDistr->FindObject("hd0d0S_1"))->Fill(d->Prodd0d0());
713 ((TH1F*)fDistr->FindObject("hcosthetapointS_1"))->Fill(d->CosPointingAngle());
714 ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_1"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
715 }
716 if(pt>1. && pt<=2.){
717 ((TH1F*)fDistr->FindObject("hd0d0S_2"))->Fill(d->Prodd0d0());
718 ((TH1F*)fDistr->FindObject("hcosthetapointS_2"))->Fill(d->CosPointingAngle());
719 ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_2"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
720 }
721 if(pt>2. && pt<=3.){
722 ((TH1F*)fDistr->FindObject("hd0d0S_3"))->Fill(d->Prodd0d0());
723 ((TH1F*)fDistr->FindObject("hcosthetapointS_3"))->Fill(d->CosPointingAngle());
724 ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_3"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
725 }
726 if(pt>3. && pt<=5.){
727 ((TH1F*)fDistr->FindObject("hd0d0S_4"))->Fill(d->Prodd0d0());
728 ((TH1F*)fDistr->FindObject("hcosthetapointS_4"))->Fill(d->CosPointingAngle());
729 ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_4"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
730 }
731 if(pt>5.) {
732 ((TH1F*)fDistr->FindObject("hd0d0S_5"))->Fill(d->Prodd0d0());
733 ((TH1F*)fDistr->FindObject("hcosthetapointS_5"))->Fill(d->CosPointingAngle());
734 ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_5"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
735 }
0108fa62 736
737 } //invmass cut
738
739 } else{ //Background or LS
740 //cout<<"is background"<<endl;
741 Double_t pt = d->Pt();
b272aebf 742
743 //no mass cut distributions: mass, ptbis
0108fa62 744 if(pt>0. && pt<=1.) {
745 ((TH1F*)fDistr->FindObject("hMassB_1"))->Fill(minvD0);
746 ((TH1F*)fDistr->FindObject("hMassB_1"))->Fill(minvD0bar);
b272aebf 747
748 ((TH1F*)fDistr->FindObject("hptB1prongnoMcut_1"))->Fill(d->PtProng(0));
749 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_1"))->Fill(d->PtProng(0));
750 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_1"))->Fill(d->PtProng(1));
751
0108fa62 752 }
753 if(pt>1. && pt<=2.) {
754 ((TH1F*)fDistr->FindObject("hMassB_2"))->Fill(minvD0);
755 ((TH1F*)fDistr->FindObject("hMassB_2"))->Fill(minvD0bar);
b272aebf 756
757 ((TH1F*)fDistr->FindObject("hptB1prongnoMcut_2"))->Fill(d->PtProng(0));
758 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_2"))->Fill(d->PtProng(0));
759 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_2"))->Fill(d->PtProng(1));
760
0108fa62 761 }
762 if(pt>2. && pt<=3.) {
763 ((TH1F*)fDistr->FindObject("hMassB_3"))->Fill(minvD0);
764 ((TH1F*)fDistr->FindObject("hMassB_3"))->Fill(minvD0bar);
b272aebf 765
766 ((TH1F*)fDistr->FindObject("hptB1prongnoMcut_3"))->Fill(d->PtProng(0));
767 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_3"))->Fill(d->PtProng(0));
768 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_3"))->Fill(d->PtProng(1));
769
0108fa62 770 }
771 if(pt>3. && pt<=5.) {
772 ((TH1F*)fDistr->FindObject("hMassB_4"))->Fill(minvD0);
773 ((TH1F*)fDistr->FindObject("hMassB_4"))->Fill(minvD0bar);
b272aebf 774
775 ((TH1F*)fDistr->FindObject("hptB1prongnoMcut_4"))->Fill(d->PtProng(0));
776 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_4"))->Fill(d->PtProng(0));
777 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_4"))->Fill(d->PtProng(1));
778
0108fa62 779 }
780 if(pt>5.) {
781 ((TH1F*)fDistr->FindObject("hMassB_5"))->Fill(minvD0);
782 ((TH1F*)fDistr->FindObject("hMassB_5"))->Fill(minvD0bar);
b272aebf 783
784 ((TH1F*)fDistr->FindObject("hptB1prongnoMcut_5"))->Fill(d->PtProng(0));
785 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_5"))->Fill(d->PtProng(0));
786 ((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_5"))->Fill(d->PtProng(1));
787
0108fa62 788 }
789
790 //apply cut on invariant mass on the pair
791 if(TMath::Abs(minvD0-mPDG)<0.03 || TMath::Abs(minvD0bar-mPDG)<0.03){
792
793
794 AliAODTrack *prong=(AliAODTrack*)d->GetDaughter(0);
795 if(!prong) cout<<"No daughter found";
796 else{
797 if(prong->Charge()==1) {fTotPosPairs[4]++;} else {fTotNegPairs[4]++;}
798 }
b272aebf 799 if(pt>0. && pt<=1.){
800 //normalise pt distr to half afterwards
801 ((TH1F*)fDistr->FindObject("hptB_1"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_1"))->Fill(d->PtProng(1));
802 ((TH1F*)fDistr->FindObject("hd0B_1"))->Fill(d->Getd0Prong(0));
803 ((TH1F*)fDistr->FindObject("hdcaB_1"))->Fill(d->GetDCA());
804 ((TH1F*)fDistr->FindObject("hcosthetastarB_1"))->Fill(d->CosThetaStarD0());
805 ((TH1F*)fDistr->FindObject("hcosthetastarB_1"))->Fill(d->CosThetaStarD0bar());
806 ((TH1F*)fDistr->FindObject("hd0d0B_1"))->Fill(d->Prodd0d0());
807 ((TH1F*)fDistr->FindObject("hcosthetapointB_1"))->Fill(d->CosPointingAngle());
808 ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_1"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
809 }
810 if(pt>1. && pt<=2.){
811 ((TH1F*)fDistr->FindObject("hptB_2"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_2"))->Fill(d->PtProng(1));
812 ((TH1F*)fDistr->FindObject("hd0B_2"))->Fill(d->Getd0Prong(0));
813 ((TH1F*)fDistr->FindObject("hdcaB_2"))->Fill(d->GetDCA());
814 ((TH1F*)fDistr->FindObject("hcosthetastarB_2"))->Fill(d->CosThetaStarD0());
815 ((TH1F*)fDistr->FindObject("hcosthetastarB_2"))->Fill(d->CosThetaStarD0bar());
816 ((TH1F*)fDistr->FindObject("hd0d0B_2"))->Fill(d->Prodd0d0());
817 ((TH1F*)fDistr->FindObject("hcosthetapointB_2"))->Fill(d->CosPointingAngle());
818 ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_2"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
819 }
820 if(pt>2. && pt<=3.){
821 ((TH1F*)fDistr->FindObject("hptB_3"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_3"))->Fill(d->PtProng(1));
822 ((TH1F*)fDistr->FindObject("hd0B_3"))->Fill(d->Getd0Prong(0));
823 ((TH1F*)fDistr->FindObject("hdcaB_3"))->Fill(d->GetDCA());
824 ((TH1F*)fDistr->FindObject("hcosthetastarB_3"))->Fill(d->CosThetaStarD0());
825 ((TH1F*)fDistr->FindObject("hcosthetastarB_3"))->Fill(d->CosThetaStarD0bar());
826 ((TH1F*)fDistr->FindObject("hd0d0B_3"))->Fill(d->Prodd0d0());
827 ((TH1F*)fDistr->FindObject("hcosthetapointB_3"))->Fill(d->CosPointingAngle());
828 ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_3"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
829 }
830 if(pt>3. && pt<=5.){
831 ((TH1F*)fDistr->FindObject("hptB_4"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_4"))->Fill(d->PtProng(1));
832 ((TH1F*)fDistr->FindObject("hd0B_4"))->Fill(d->Getd0Prong(0));
833 ((TH1F*)fDistr->FindObject("hdcaB_4"))->Fill(d->GetDCA());
834 ((TH1F*)fDistr->FindObject("hcosthetastarB_4"))->Fill(d->CosThetaStarD0());
835 ((TH1F*)fDistr->FindObject("hcosthetastarB_4"))->Fill(d->CosThetaStarD0bar());
836 ((TH1F*)fDistr->FindObject("hd0d0B_4"))->Fill(d->Prodd0d0());
837 ((TH1F*)fDistr->FindObject("hcosthetapointB_4"))->Fill(d->CosPointingAngle());
838 ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_4"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
839 }
840 if(pt>5.) {
841 ((TH1F*)fDistr->FindObject("hptB_5"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_5"))->Fill(d->PtProng(1));
842 ((TH1F*)fDistr->FindObject("hd0B_5"))->Fill(d->Getd0Prong(0));
843 ((TH1F*)fDistr->FindObject("hdcaB_5"))->Fill(d->GetDCA());
844 ((TH1F*)fDistr->FindObject("hcosthetastarB_5"))->Fill(d->CosThetaStarD0());
845 ((TH1F*)fDistr->FindObject("hcosthetastarB_5"))->Fill(d->CosThetaStarD0bar());
846 ((TH1F*)fDistr->FindObject("hd0d0B_5"))->Fill(d->Prodd0d0());
847 ((TH1F*)fDistr->FindObject("hcosthetapointB_5"))->Fill(d->CosPointingAngle());
848 ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_5"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
849 }
0108fa62 850
851
852 }// end if inv mass cut
853 }//end if background
ce39f0ac 854
0108fa62 855
a41f6fad 856
9de8c723 857 //cuts order
858// printf(" |M-MD0| [GeV] < %f\n",fD0toKpiCuts[0]);
859// printf(" dca [cm] < %f\n",fD0toKpiCuts[1]);
860// printf(" cosThetaStar < %f\n",fD0toKpiCuts[2]);
861// printf(" pTK [GeV/c] > %f\n",fD0toKpiCuts[3]);
862// printf(" pTpi [GeV/c] > %f\n",fD0toKpiCuts[4]);
863// printf(" |d0K| [cm] < %f\n",fD0toKpiCuts[5]);
864// printf(" |d0pi| [cm] < %f\n",fD0toKpiCuts[6]);
865// printf(" d0d0 [cm^2] < %f\n",fD0toKpiCuts[7]);
866// printf(" cosThetaPoint > %f\n",fD0toKpiCuts[8]);
0108fa62 867
49061176 868 Int_t ptbin=0;
6321ee46 869 Bool_t isInRange=kFALSE;
49061176 870
871 //cout<<"P_t = "<<pt<<endl;
872 if (pt>0. && pt<=1.) {
a4ae02cd 873 ptbin=1;
6321ee46 874 isInRange=kTRUE;
11ecfd50 875 /*
876 //test d0 cut
877 fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.1,-0.0002,0.5);
878 fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.1,-0.00025,0.7);
879 */
880 //original
9de8c723 881 fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.0002,0.5);
527f330b 882 fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.00025,0.7);
11ecfd50 883
9de8c723 884// fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.0002,0.7);
527f330b 885// fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,1,1,-0.00015,0.5);
49061176 886 //printf("I'm in the bin %d\n",ptbin);
a4ae02cd 887
49061176 888 }
11ecfd50 889
46c96ce5 890 if(pt>1. && pt<=3.) {
9de8c723 891 if(pt>1. && pt<=2.) ptbin=2;
892 if(pt>2. && pt<=3.) ptbin=3;
6321ee46 893 isInRange=kTRUE;
11ecfd50 894 /*
895 //test d0 cut
896 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.0002,0.6);
897 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,1,0.1,-0.00025,0.8);
898 */
899 //original
900 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.0002,0.6);
527f330b 901 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,1,1,-0.00025,0.8);
11ecfd50 902
46c96ce5 903 //printf("I'm in the bin %d\n",ptbin);
904 }
905
906 if(pt>3. && pt<=5.){
6321ee46 907 ptbin=4;
908 isInRange=kTRUE;
909 /*
910 //test d0 cut
911 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.0001,0.8);
912 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00015,0.8);
913 */
914 //original
915 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.0001,0.8);
916 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00015,0.8);
917
918 //printf("I'm in the bin %d\n",ptbin);
49061176 919 }
6321ee46 920 if(pt>5.&& pt<=10.){ //additional upper limit to compare with Correction Framework
9de8c723 921 ptbin=5;
6321ee46 922 isInRange=kTRUE;
11ecfd50 923 /*
924 //test d0 cut
925 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00005,0.8);
926 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00015,0.9);
927 */
928 //original
9de8c723 929 fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00005,0.8);
527f330b 930 fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00015,0.9);
11ecfd50 931
932 }//if(pt>5) if (pt>0. && pt<=1.) {
a4ae02cd 933 //printf("I'm in the bin %d\n",ptbin);
934 //old
935 //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
6321ee46 936 if(prongsinacc && isInRange){
9de8c723 937 FillHists(ptbin,d,mcArray,fVHFPPR,fOutputPPR);
527f330b 938 FillHists(ptbin,d,mcArray,fVHFmycuts,fOutputmycuts);
9de8c723 939 }
49061176 940 if(unsetvtx) d->UnsetOwnPrimaryVtx();
0108fa62 941 } //end for prongs
46c96ce5 942
49061176 943
944
945
946 // Post the data
9de8c723 947 PostData(1,fOutputPPR);
527f330b 948 PostData(2,fOutputmycuts);
a41f6fad 949 PostData(4,fDistr);
49061176 950 return;
951}
952//____________________________________________________________________________*
a4ae02cd 953void AliAnalysisTaskSED0Mass::FillHists(Int_t ptbin, AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliAnalysisVertexingHF *vhf, TList *listout){
49061176 954 //
955 // function used in UserExec:
956 //
9de8c723 957
958
959 Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
960
49061176 961 Int_t okD0=0,okD0bar=0;
feb73eca 962 //cout<<"inside FillHist"<<endl;
a4ae02cd 963 if(part->SelectD0(vhf->GetD0toKpiCuts(),okD0,okD0bar)) {//selected
49061176 964 Double_t invmassD0 = part->InvMassD0(), invmassD0bar = part->InvMassD0bar();
965 //printf("SELECTED\n");
feb73eca 966
9de8c723 967
feb73eca 968 AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(0);
969 if(!prong) cout<<"No daughter found";
970 else{
9de8c723 971 if(prong->Charge()==1) {fTotPosPairs[ptbin-1]++;} else {fTotNegPairs[ptbin-1]++;}
feb73eca 972 }
973
974 TString fillthis="";
c8112d2f 975 Int_t pdgDgD0toKpi[2]={321,211};
ce39f0ac 976 Int_t labD0=-1;
977 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 978
9de8c723 979 //count candidates selected by cuts
980 fNentries->Fill(2);
981 //count true D0 selected by cuts
6321ee46 982 if (fReadMC && labD0>=0) fNentries->Fill(3);
9de8c723 983 PostData(3,fNentries);
984
a4ae02cd 985 if (okD0==1) {
986 fillthis="histMass_";
7646d6da 987 fillthis+=ptbin;
988 //cout<<"Filling "<<fillthis<<endl;
989
49061176 990 //printf("Fill mass with D0");
a4ae02cd 991 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
feb73eca 992
a4ae02cd 993 if(labD0>=0) {
feb73eca 994 if(fArray==1) cout<<"LS signal ERROR"<<endl;
995
a4ae02cd 996 AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
997 Int_t pdgD0 = partD0->GetPdgCode();
998 //cout<<"pdg = "<<pdgD0<<endl;
a4ae02cd 999 if (pdgD0==421){ //D0
1000 //cout<<"Fill S with D0"<<endl;
46c96ce5 1001 fillthis="histSgn_";
1002 fillthis+=ptbin;
a4ae02cd 1003 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
9de8c723 1004 if(TMath::Abs(invmassD0 - mPDG) < 0.027){
1005 fillthis="histSgn27_";
1006 fillthis+=ptbin;
1007 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
1008 }
a4ae02cd 1009 } else{ //it was a D0bar
46c96ce5 1010 fillthis="histRfl_";
1011 fillthis+=ptbin;
1012 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
a4ae02cd 1013 }
1014 } else {//background
1015 fillthis="histBkg_";
1016 fillthis+=ptbin;
1017 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
1018 }
feb73eca 1019
49061176 1020 }
1021 if (okD0bar==1) {
a4ae02cd 1022 fillthis="histMass_";
1023 fillthis+=ptbin;
49061176 1024 //printf("Fill mass with D0bar");
a4ae02cd 1025 ((TH1F*)listout->FindObject(fillthis))->Fill(invmassD0bar);
feb73eca 1026
a4ae02cd 1027 if(labD0>=0) {
feb73eca 1028 if(fArray==1) cout<<"LS signal ERROR"<<endl;
a4ae02cd 1029 AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
1030 Int_t pdgD0 = partD0->GetPdgCode();
1031 //cout<<" pdg = "<<pdgD0<<endl;
1032 if (pdgD0==-421){ //D0bar
1033 fillthis="histSgn_";
1034 fillthis+=ptbin;
1035 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
9de8c723 1036 if (TMath::Abs(invmassD0bar - mPDG) < 0.027){
1037 fillthis="histSgn27_";
1038 fillthis+=ptbin;
1039 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
1040 }
1041
a4ae02cd 1042
1043 } else{
46c96ce5 1044 fillthis="histRfl_";
1045 fillthis+=ptbin;
1046 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
a4ae02cd 1047 }
feb73eca 1048 } else {//background or LS
a4ae02cd 1049 fillthis="histBkg_";
1050 fillthis+=ptbin;
1051 ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
1052 }
1053 }
1054
1055
49061176 1056 } //else cout<<"NOT SELECTED"<<endl;
1057
1058}
1059//________________________________________________________________________
1060void AliAnalysisTaskSED0Mass::Terminate(Option_t */*option*/)
1061{
1062 // Terminate analysis
1063 //
1064 if(fDebug > 1) printf("AnalysisTaskSED0Mass: Terminate() \n");
1065
6321ee46 1066
9de8c723 1067 fOutputPPR = dynamic_cast<TList*> (GetOutputData(1));
1068 if (!fOutputPPR) {
a41f6fad 1069 printf("ERROR: fOutputthight not available\n");
a4ae02cd 1070 return;
1071 }
527f330b 1072 fOutputmycuts = dynamic_cast<TList*> (GetOutputData(2));
1073 if (!fOutputmycuts) {
1074 printf("ERROR: fOutputmycuts not available\n");
a41f6fad 1075 return;
1076 }
9de8c723 1077 fDistr = dynamic_cast<TList*> (GetOutputData(4));
a41f6fad 1078 if (!fDistr) {
1079 printf("ERROR: fDistr not available\n");
49061176 1080 return;
1081 }
1082
feb73eca 1083 if(fArray==1){
1084 for(Int_t ipt=0;ipt<4;ipt++){
1085 fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[ipt]*fTotNegPairs[ipt]);
1086
1087
1088 if(fLsNormalization>0) {
9de8c723 1089
feb73eca 1090 TString massName="histMass_";
1091 massName+=ipt+1;
9de8c723 1092 ((TH1F*)fOutputPPR->FindObject(massName))->Scale((1/fLsNormalization)*((TH1F*)fOutputPPR->FindObject(massName))->GetEntries());
527f330b 1093 ((TH1F*)fOutputmycuts->FindObject(massName))->Scale((1/fLsNormalization)*((TH1F*)fOutputmycuts->FindObject(massName))->GetEntries());
feb73eca 1094 }
1095 }
1096
1097 fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[4]*fTotNegPairs[4]);
1098
1099 if(fLsNormalization>0) {
1100
1101 TString nameDistr="hptB";
1102 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1103 nameDistr="hdcaB";
1104 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1105 nameDistr="hcosthetastarB";
1106 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1107 nameDistr="hd0B";
1108 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1109 nameDistr="hd0d0B";
1110 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1111 nameDistr="hcosthetapointB";
1112 ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1113 nameDistr="hcosthpointd0d0B";
1114 ((TH2F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH2F*)fDistr->FindObject(nameDistr))->GetEntries());
1115
1116 }
1117 }
6ce5994f 1118
527f330b 1119 TString cvname;
1120
1121 if (fArray==0){
1122 cvname="D0invmass";
1123 } else cvname="LSinvmass";
1124
1125 TCanvas *c1=new TCanvas(cvname,cvname);
1126 c1->cd();
1127 ((TH1F*)fOutputPPR->FindObject("histMass_4"))->Draw();
1128
1129
49061176 1130 return;
1131}
1132