PWG3/vertexingHF/AliAnalysisTaskSED0Mass.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   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  **************************************************************************/
  15
  16 /////////////////////////////////////////////////////////////
  17 //
  18 // AliAnalysisTaskSE for D0 candidates invariant mass histogram
  19 // and comparison with the MC truth and cut variables distributions.
  20 //
  21 // Authors: A.Dainese, andrea.dainese@lnl.infn.it
  22 // Chiara Bianchin, chiara.bianchin@pd.infn.it (invariant mass)
  23 // Carmelo Di Giglio, carmelo.digiglio@ba.infn.it (like sign)
  24 /////////////////////////////////////////////////////////////
  25
  26 #include <Riostream.h>
  27 #include <TClonesArray.h>
  28 #include <TCanvas.h>
  29 #include <TNtuple.h>
  30 #include <TList.h>
  31 #include <TH1F.h>
  32 #include <TH2F.h>
  33 #include <TDatabasePDG.h>
  34
  35 #include <AliAnalysisDataSlot.h>
  36 #include <AliAnalysisDataContainer.h>
  37 #include "AliAnalysisManager.h"
  38 #include "AliESDtrack.h"
  39 #include "AliVertexerTracks.h"
  40 #include "AliAODHandler.h"
  41 #include "AliAODEvent.h"
  42 #include "AliAODVertex.h"
  43 #include "AliAODTrack.h"
  44 #include "AliAODMCHeader.h"
  45 #include "AliAODMCParticle.h"
  46 #include "AliAODRecoDecayHF2Prong.h"
  47 #include "AliAODRecoCascadeHF.h"
  48 #include "AliAnalysisVertexingHF.h"
  49 #include "AliAnalysisTaskSE.h"
  50 #include "AliAnalysisTaskSED0Mass.h"
  51 #include "AliNormalizationCounter.h"
  52
  53 ClassImp(AliAnalysisTaskSED0Mass)
  54
  55
  56 //________________________________________________________________________
  57 AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass():
  58 AliAnalysisTaskSE(),
  59 fOutputMass(0),
  60 fDistr(0),
  61 fNentries(0),
  62 fCuts(0),
  63 fArray(0),
  64 fReadMC(0),
  65 fCutOnDistr(0),
  66 fUsePid4Distr(0),
  67 fCounter(0),
  68 fNPtBins(1),
  69 fLsNormalization(1.),
  70 fFillOnlyD0D0bar(0)
  71
  72
  73 {
  74   // Default constructor
  75 }
  76
  77 //________________________________________________________________________
  78 AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass(const char *name,AliRDHFCutsD0toKpi* cuts):
  79 AliAnalysisTaskSE(name),
  80 fOutputMass(0),
  81 fDistr(0),
  82 fNentries(0),
  83 fCuts(0),
  84 fArray(0),
  85 fReadMC(0),
  86 fCutOnDistr(0),
  87 fUsePid4Distr(0),
  88 fCounter(0),
  89 fNPtBins(1),
  90 fLsNormalization(1.),
  91 fFillOnlyD0D0bar(0)
  92
  93 {
  94   // Default constructor
  95
  96   fNPtBins=cuts->GetNPtBins();
  97   // fTotPosPairs=new Int_t[fNPtBins];
  98   // fTotNegPairs=new Int_t[fNPtBins];
  99   // for(Int_t i=0;i<fNPtBins;i++) {fTotPosPairs[i]=0; fTotNegPairs[i]=0;}
 100
 101   fCuts=cuts;
 102
 103   // Output slot #1 writes into a TList container (mass with cuts)
 104   DefineOutput(1,TList::Class());  //My private output
 105   // Output slot #2 writes into a TList container (distributions)
 106   DefineOutput(2,TList::Class());  //My private output
 107   // Output slot #3 writes into a TH1F container (number of events)
 108   DefineOutput(3,TH1F::Class());  //My private output
 109   // Output slot #4 writes into a TList container (cuts)
 110   DefineOutput(4,AliRDHFCutsD0toKpi::Class());  //My private output
 111   // Output slot #5 writes Normalization Counter
 112   DefineOutput(5,AliNormalizationCounter::Class());
 113 }
 114
 115 //________________________________________________________________________
 116 AliAnalysisTaskSED0Mass::~AliAnalysisTaskSED0Mass()
 117 {
 118    if (fOutputMass) {
 119     delete fOutputMass;
 120     fOutputMass = 0;
 121   }
 122   if (fDistr) {
 123     delete fDistr;
 124     fDistr = 0;
 125   }
 126   if (fCuts) {
 127     delete fCuts;
 128     fCuts = 0;
 129   }
 130   if (fNentries){
 131     delete fNentries;
 132     fNentries = 0;
 133   }
 134   if(fCounter){
 135     delete fCounter;
 136     fCounter=0;
 137   }
 138
 139 }
 140
 141 //________________________________________________________________________
 142 void AliAnalysisTaskSED0Mass::Init()
 143 {
 144   // Initialization
 145
 146   if(fDebug > 1) printf("AnalysisTaskSED0Mass::Init() \n");
 147
 148
 149   AliRDHFCutsD0toKpi* copyfCuts=new AliRDHFCutsD0toKpi(*fCuts);
 150   const char* nameoutput=GetOutputSlot(4)->GetContainer()->GetName();
 151   copyfCuts->SetName(nameoutput);
 152   // Post the data
 153   PostData(4,copyfCuts);
 154
 155
 156   return;
 157 }
 158
 159 //________________________________________________________________________
 160 void AliAnalysisTaskSED0Mass::UserCreateOutputObjects()
 161 {
 162
 163   // Create the output container
 164   //
 165   if(fDebug > 1) printf("AnalysisTaskSED0Mass::UserCreateOutputObjects() \n");
 166
 167   // Several histograms are more conveniently managed in a TList
 168   fOutputMass = new TList();
 169   fOutputMass->SetOwner();
 170   fOutputMass->SetName("listMass");
 171
 172   fDistr = new TList();
 173   fDistr->SetOwner();
 174   fDistr->SetName("distributionslist");
 175
 176   TString nameMass=" ",nameSgn27=" ",nameSgn=" ", nameBkg=" ", nameRfl=" ",nameMassNocutsS =" ",nameMassNocutsB =" ", namedistr=" ";
 177
 178   for(Int_t i=0;i<fCuts->GetNPtBins();i++){
 179
 180     nameMass="histMass_";
 181     nameMass+=i;
 182     nameSgn27="histSgn27_";
 183     nameSgn27+=i;
 184     nameSgn="histSgn_";
 185     nameSgn+=i;
 186     nameBkg="histBkg_";
 187     nameBkg+=i;
 188     nameRfl="histRfl_";
 189     nameRfl+=i;
 190     nameMassNocutsS="hMassS_";
 191     nameMassNocutsS+=i;
 192     nameMassNocutsB="hMassB_";
 193     nameMassNocutsB+=i;
 194
 195     //histograms of cut variable distributions
 196
 197     //  pT
 198     namedistr="hptpiS_";
 199     namedistr+=i;
 200     TH1F *hptpiS = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
 201
 202     namedistr="hptKS_";
 203     namedistr+=i;
 204     TH1F *hptKS = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
 205
 206     namedistr="hptB_";
 207     namedistr+=i;
 208     TH1F *hptB = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
 209
 210     //  pT no mass cut
 211     // namedistr="hptpiSnoMcut_";
 212     // namedistr+=i;
 213     // TH1F *hptpiSnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
 214
 215     // namedistr="hptKSnoMcut_";
 216     // namedistr+=i;
 217     // TH1F *hptKSnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
 218
 219     // namedistr="hptB1prongnoMcut_";
 220     // namedistr+=i;
 221     // TH1F *hptB1pnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
 222
 223     // namedistr="hptB2prongsnoMcut_";
 224     // namedistr+=i;
 225     // TH1F *hptB2pnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
 226
 227     // fDistr->Add(hptpiSnoMcut);
 228     // fDistr->Add(hptKSnoMcut);
 229     // fDistr->Add(hptB1pnoMcut);
 230     // fDistr->Add(hptB2pnoMcut);
 231
 232
 233     //  dca
 234     namedistr="hdcaS_";
 235     namedistr+=i;
 236     TH1F *hdcaS = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
 237     namedistr="hdcaB_";
 238     namedistr+=i;
 239     TH1F *hdcaB = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
 240
 241     //  costhetastar
 242     namedistr="hcosthetastarS_";
 243     namedistr+=i;
 244     TH1F *hcosthetastarS = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
 245     namedistr="hcosthetastarB_";
 246     namedistr+=i;
 247     TH1F *hcosthetastarB = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
 248
 249     // impact parameter
 250     namedistr="hd0piS_";
 251     namedistr+=i;
 252     TH1F *hd0piS = new TH1F(namedistr.Data(), "Impact parameter distribution (pions);d0(#pi) [cm]",200,-0.1,0.1);
 253
 254     namedistr="hd0KS_";
 255     namedistr+=i;
 256     TH1F *hd0KS = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons);d0(K) [cm]",200,-0.1,0.1);
 257
 258     namedistr="hd0B_";
 259     namedistr+=i;
 260     TH1F *hd0B = new TH1F(namedistr.Data(), "Impact parameter distribution (both);d0 [cm]",200,-0.1,0.1);
 261
 262     namedistr="hd0p0B_";
 263     namedistr+=i;
 264     TH1F *hd0p0B = new TH1F(namedistr.Data(), "Impact parameter distribution (prong +);d0 [cm]",200,-0.1,0.1);
 265
 266     namedistr="hd0p1B_";
 267     namedistr+=i;
 268     TH1F *hd0p1B = new TH1F(namedistr.Data(), "Impact parameter distribution (prong -);d0 [cm]",200,-0.1,0.1);
 269
 270     namedistr="hd0moresB_";
 271     namedistr+=i;
 272     TH1F *hd0moresB = new TH1F(namedistr.Data(), "Impact parameter distribution (both);d0 [cm]",200,-0.1,0.1);
 273
 274     namedistr="hd0d0moresB_";
 275     namedistr+=i;
 276     TH1F *hd0d0moresB = new TH1F(namedistr.Data(), "Impact parameter distribution (prong +);d0 [cm]",200,-0.001,0.001);
 277
 278     namedistr="hd0d0S_";
 279     namedistr+=i;
 280     TH1F *hd0d0S = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
 281     namedistr="hd0d0B_";
 282     namedistr+=i;
 283     TH1F *hd0d0B = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
 284
 285     //decay lenght
 286     namedistr="hdeclS_";
 287     namedistr+=i;
 288     TH1F *hdeclengthS = new TH1F(namedistr.Data(), "Decay Length distribution;Decay Length [cm]",200,0,0.15);
 289
 290     namedistr="hdeclB_";
 291     namedistr+=i;
 292     TH1F *hdeclengthB = new TH1F(namedistr.Data(), "Decay Length distribution;Decay Length [cm]",200,0,0.15);
 293
 294     namedistr="hnormdeclS_";
 295     namedistr+=i;
 296     TH1F *hnormdeclengthS = new TH1F(namedistr.Data(), "Normalized Decay Length distribution;Decay Length/Err ",200,0,6.);
 297
 298     namedistr="hnormdeclB_";
 299     namedistr+=i;
 300     TH1F *hnormdeclengthB = new TH1F(namedistr.Data(), "Normalized Decay Length distribution;Decay Length/Err ",200,0,6.);
 301
 302    //  costhetapoint
 303     namedistr="hcosthetapointS_";
 304     namedistr+=i;
 305     TH1F *hcosthetapointS = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
 306     namedistr="hcosthetapointB_";
 307     namedistr+=i;
 308     TH1F *hcosthetapointB = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
 309
 310     namedistr="hcosthetapointmoresB_";
 311     namedistr+=i;
 312     TH1F *hcosthetapointmoresB = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
 313
 314    //  costhetapoint vs d0 or d0d0
 315     namedistr="hcosthpointd0S_";
 316     namedistr+=i;
 317     TH2F *hcosthpointd0S= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0};cos#theta_{Point};d_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
 318
 319     namedistr="hcosthpointd0B_";
 320     namedistr+=i;
 321     TH2F *hcosthpointd0B= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0};cos#theta_{Point};d_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
 322
 323     namedistr="hcosthpointd0d0S_";
 324     namedistr+=i;
 325     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);
 326     namedistr="hcosthpointd0d0B_";
 327     namedistr+=i;
 328     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);
 329
 330     fDistr->Add(hptpiS);
 331     fDistr->Add(hptKS);
 332     fDistr->Add(hptB);
 333
 334     fDistr->Add(hdcaS);
 335     fDistr->Add(hdcaB);
 336
 337     fDistr->Add(hd0piS);
 338     fDistr->Add(hd0KS);
 339     fDistr->Add(hd0B);
 340     fDistr->Add(hd0p0B);
 341     fDistr->Add(hd0p1B);
 342     fDistr->Add(hd0moresB);
 343
 344     fDistr->Add(hd0d0S);
 345     fDistr->Add(hd0d0B);
 346     fDistr->Add(hd0d0moresB);
 347
 348     fDistr->Add(hcosthetastarS);
 349     fDistr->Add(hcosthetastarB);
 350
 351     fDistr->Add(hcosthetapointS);
 352     fDistr->Add(hcosthetapointB);
 353     fDistr->Add(hcosthetapointmoresB);
 354
 355     fDistr->Add(hdeclengthS);
 356     fDistr->Add(hdeclengthB);
 357
 358     fDistr->Add(hnormdeclengthS);
 359     fDistr->Add(hnormdeclengthB);
 360
 361     fDistr->Add(hcosthpointd0S);
 362     fDistr->Add(hcosthpointd0B);
 363
 364     fDistr->Add(hcosthpointd0d0S);
 365     fDistr->Add(hcosthpointd0d0B);
 366
 367     //histograms filled only when the secondary vertex is recalculated w/o the daughter tracks (as requested in the cut object)
 368
 369     if(fCuts->GetIsPrimaryWithoutDaughters()){
 370       namedistr="hd0vmoresB_";
 371       namedistr+=i;
 372       TH1F *hd0vmoresB = new TH1F(namedistr.Data(), "Impact parameter distribution (both);d0 [cm]",200,-0.1,0.1);
 373
 374       namedistr="hd0d0vmoresB_";
 375       namedistr+=i;
 376       TH1F *hd0d0vmoresB = new TH1F(namedistr.Data(), "Impact parameter distribution (prong +);d0 [cm]",200,-0.001,0.001);
 377
 378
 379       namedistr="hd0vpiS_";
 380       namedistr+=i;
 381       TH1F *hd0vpiS = new TH1F(namedistr.Data(), "Impact parameter distribution (pions)(vtx w/o these tracks);d0(#pi) [cm]",200,-0.1,0.1);
 382
 383       namedistr="hd0vKS_";
 384       namedistr+=i;
 385       TH1F *hd0vKS = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons) (vtx w/o these tracks);d0(K) [cm]",200,-0.1,0.1);
 386       namedistr="hd0vB_";
 387       namedistr+=i;
 388       TH1F *hd0vB = new TH1F(namedistr.Data(), "Impact parameter distribution (vtx w/o these tracks);d0 [cm]",200,-0.1,0.1);
 389
 390       namedistr="hd0vp0B_";
 391       namedistr+=i;
 392       TH1F *hd0vp0B = new TH1F(namedistr.Data(), "Impact parameter distribution (prong + ** vtx w/o these tracks);d0 [cm]",200,-0.1,0.1);
 393
 394       namedistr="hd0vp1B_";
 395       namedistr+=i;
 396       TH1F *hd0vp1B = new TH1F(namedistr.Data(), "Impact parameter distribution (prong - ** vtx w/o these tracks);d0 [cm]",200,-0.1,0.1);
 397
 398
 399       namedistr="hd0d0vS_";
 400       namedistr+=i;
 401       TH1F *hd0d0vS = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution (vtx w/o these tracks);d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
 402       namedistr="hd0d0vB_";
 403       namedistr+=i;
 404       TH1F *hd0d0vB = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution (vtx w/o these tracks);d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
 405
 406
 407       namedistr="hdeclvS_";
 408       namedistr+=i;
 409       TH1F *hdeclengthvS = new TH1F(namedistr.Data(), "Decay Length distribution (vtx w/o tracks);Decay Length [cm]",200,0,0.6);
 410
 411       namedistr="hdeclvB_";
 412       namedistr+=i;
 413       TH1F *hdeclengthvB = new TH1F(namedistr.Data(), "Decay Length distribution (vtx w/o tracks);Decay Length [cm]",200,0,0.6);
 414
 415       namedistr="hnormdeclvS_";
 416       namedistr+=i;
 417       TH1F *hnormdeclengthvS = new TH1F(namedistr.Data(), "Normalized Decay Length distribution (vtx w/o tracks);Decay Length/Err ",200,0,10.);
 418
 419       namedistr="hnormdeclvB_";
 420       namedistr+=i;
 421       TH1F *hnormdeclengthvB = new TH1F(namedistr.Data(), "Normalized Decay Length distribution (vtx w/o tracks);Decay Length/Err ",200,0,10.);
 422
 423
 424
 425       fDistr->Add(hd0vpiS);
 426       fDistr->Add(hd0vKS);
 427       fDistr->Add(hd0vB);
 428       fDistr->Add(hd0vp0B);
 429       fDistr->Add(hd0vp1B);
 430       fDistr->Add(hd0vmoresB);
 431
 432       fDistr->Add(hd0d0vS);
 433       fDistr->Add(hd0d0vB);
 434       fDistr->Add(hd0d0vmoresB);
 435
 436       fDistr->Add(hdeclengthvS);
 437       fDistr->Add(hdeclengthvB);
 438
 439       fDistr->Add(hnormdeclengthvS);
 440       fDistr->Add(hnormdeclengthvB);
 441     }
 442
 443     //histograms of invariant mass distributions
 444
 445     TH1F* tmpMt = new TH1F(nameMass.Data(),"D^{0} invariant mass; M [GeV]; Entries",200,1.5648,2.1648);
 446     TH1F *tmpMl=(TH1F*)tmpMt->Clone();
 447     tmpMt->Sumw2();
 448     tmpMl->Sumw2();
 449
 450     //to compare with AliAnalysisTaskCharmFraction
 451     // TH1F* tmpS27t = new TH1F(nameSgn27.Data(),"D^{0} invariant mass in M(D^{0}) +/- 27 MeV - MC; M [GeV]; Entries",200,1.5648,2.1648);
 452     // TH1F *tmpS27l=(TH1F*)tmpS27t->Clone();
 453     // tmpS27t->Sumw2();
 454     // tmpS27l->Sumw2();
 455
 456     //distribution w/o cuts
 457     //   TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",300,0.7,3.);
 458     TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",300,1.5648,2.1648); //range (MD0-300MeV, mD0 + 300MeV)
 459     TH1F *tmpMB=(TH1F*)tmpMS->Clone();
 460     tmpMB->SetName(nameMassNocutsB.Data());
 461     tmpMS->Sumw2();
 462     tmpMB->Sumw2();
 463
 464     //MC signal and background
 465     TH1F* tmpSt = new TH1F(nameSgn.Data(), "D^{0} invariant mass - MC; M [GeV]; Entries",200,1.5648,2.1648);
 466     TH1F *tmpSl=(TH1F*)tmpSt->Clone();
 467     tmpSt->Sumw2();
 468     tmpSl->Sumw2();
 469
 470     TH1F* tmpBt = new TH1F(nameBkg.Data(), "Background invariant mass - MC; M [GeV]; Entries",200,1.5648,2.1648);
 471     TH1F *tmpBl=(TH1F*)tmpBt->Clone();
 472     tmpBt->Sumw2();
 473     tmpBl->Sumw2();
 474
 475     //Reflection: histo filled with D0Mass which pass the cut (also) as D0bar and with D0bar which pass (also) the cut as D0
 476     TH1F* tmpRt = new TH1F(nameRfl.Data(), "Reflected signal invariant mass - MC; M [GeV]; Entries",200,1.5648,2.1648);
 477     TH1F *tmpRl=(TH1F*)tmpRt->Clone();
 478     tmpRt->Sumw2();
 479     tmpRl->Sumw2();
 480   //  printf("Created histograms %s\t%s\t%s\t%s\n",tmpM->GetName(),tmpS->GetName(),tmpB->GetName(),tmpR->GetName());
 481
 482     fOutputMass->Add(tmpMt);
 483     fOutputMass->Add(tmpSt);
 484     //    fOutputMass->Add(tmpS27t);
 485     fOutputMass->Add(tmpBt);
 486     fOutputMass->Add(tmpRt);
 487
 488     fDistr->Add(tmpMS);
 489     fDistr->Add(tmpMB);
 490
 491
 492   }
 493
 494   if(fArray==1){
 495     namedistr="hpospair";
 496     TH1F* hpospair=new TH1F(namedistr.Data(),"Number of positive pairs",2*fCuts->GetNPtBins(),-0.5,fCuts->GetNPtBins()-0.5);
 497     namedistr="hnegpair";
 498     TH1F* hnegpair=new TH1F(namedistr.Data(),"Number of negative pairs",fCuts->GetNPtBins(),-0.5,2*fCuts->GetNPtBins()-0.5);
 499     fDistr->Add(hpospair);
 500     fDistr->Add(hnegpair);
 501   }
 502
 503   const char* nameoutput=GetOutputSlot(3)->GetContainer()->GetName();
 504
 505   fNentries=new TH1F(nameoutput, "Integral(1,2) = number of AODs *** Integral(2,3) = number of candidates selected with cuts *** Integral(3,4) = number of D0 selected with cuts *** Integral(4,5) = events with good vertex ***  Integral(5,6) = pt out of bounds", 14,-0.5,13.5);
 506
 507   fNentries->GetXaxis()->SetBinLabel(1,"nEventsAnal");
 508   fNentries->GetXaxis()->SetBinLabel(2,"nCandSel(Cuts)");
 509   fNentries->GetXaxis()->SetBinLabel(3,"nD0Selected");
 510   fNentries->GetXaxis()->SetBinLabel(4,"nEventsGoodVtxS");
 511   fNentries->GetXaxis()->SetBinLabel(5,"ptbin = -1");
 512   fNentries->GetXaxis()->SetBinLabel(6,"no daughter");
 513   fNentries->GetXaxis()->SetBinLabel(7,"nCandSel(Tr)");
 514   fNentries->GetXaxis()->SetBinLabel(8,"PID=0");
 515   fNentries->GetXaxis()->SetBinLabel(9,"PID=1");
 516   fNentries->GetXaxis()->SetBinLabel(10,"PID=2");
 517   fNentries->GetXaxis()->SetBinLabel(11,"PID=3");
 518   fNentries->GetXaxis()->SetBinLabel(12,"K");
 519   fNentries->GetXaxis()->SetBinLabel(13,"Lambda");
 520   fNentries->GetXaxis()->SetBinLabel(14,"Pile-up Rej");
 521   fNentries->GetXaxis()->SetNdivisions(1,kFALSE);
 522
 523   fCounter = new AliNormalizationCounter(Form("%s",GetOutputSlot(5)->GetContainer()->GetName()));
 524
 525   // Post the data
 526   PostData(1,fOutputMass);
 527   PostData(2,fDistr);
 528   PostData(3,fNentries);
 529   PostData(5,fCounter);
 530   return;
 531 }
 532
 533 //________________________________________________________________________
 534 void AliAnalysisTaskSED0Mass::UserExec(Option_t */*option*/)
 535 {
 536   // Execute analysis for current event:
 537   // heavy flavor candidates association to MC truth
 538   //cout<<"I'm in UserExec"<<endl;
 539
 540
 541     //cuts order
 542     //       printf("    |M-MD0| [GeV]    < %f\n",fD0toKpiCuts[0]);
 543     //     printf("    dca    [cm]  < %f\n",fD0toKpiCuts[1]);
 544     //     printf("    cosThetaStar     < %f\n",fD0toKpiCuts[2]);
 545     //     printf("    pTK     [GeV/c]    > %f\n",fD0toKpiCuts[3]);
 546     //     printf("    pTpi    [GeV/c]    > %f\n",fD0toKpiCuts[4]);
 547     //     printf("    |d0K|  [cm]  < %f\n",fD0toKpiCuts[5]);
 548     //     printf("    |d0pi| [cm]  < %f\n",fD0toKpiCuts[6]);
 549     //     printf("    d0d0  [cm^2] < %f\n",fD0toKpiCuts[7]);
 550     //     printf("    cosThetaPoint    > %f\n",fD0toKpiCuts[8]);
 551
 552
 553   AliAODEvent *aod = dynamic_cast<AliAODEvent*> (InputEvent());
 554
 555   TString bname;
 556   if(fArray==0){ //D0 candidates
 557     // load D0->Kpi candidates
 558     //cout<<"D0 candidates"<<endl;
 559     bname="D0toKpi";
 560   } else { //LikeSign candidates
 561     // load like sign candidates
 562     //cout<<"LS candidates"<<endl;
 563     bname="LikeSign2Prong";
 564   }
 565
 566   TClonesArray *inputArray=0;
 567
 568   if(!aod && AODEvent() && IsStandardAOD()) {
 569     // In case there is an AOD handler writing a standard AOD, use the AOD
 570     // event in memory rather than the input (ESD) event.
 571     aod = dynamic_cast<AliAODEvent*> (AODEvent());
 572     // in this case the braches in the deltaAOD (AliAOD.VertexingHF.root)
 573     // have to taken from the AOD event hold by the AliAODExtension
 574     AliAODHandler* aodHandler = (AliAODHandler*)
 575       ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
 576
 577     if(aodHandler->GetExtensions()) {
 578       AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject("AliAOD.VertexingHF.root");
 579       AliAODEvent* aodFromExt = ext->GetAOD();
 580       inputArray=(TClonesArray*)aodFromExt->GetList()->FindObject(bname.Data());
 581     }
 582   } else {
 583     inputArray=(TClonesArray*)aod->GetList()->FindObject(bname.Data());
 584   }
 585
 586
 587   if(!inputArray) {
 588     printf("AliAnalysisTaskSED0Mass::UserExec: input branch not found!\n");
 589     return;
 590   }
 591
 592   // fix for temporary bug in ESDfilter
 593   // the AODs with null vertex pointer didn't pass the PhysSel
 594   if(!aod->GetPrimaryVertex() || TMath::Abs(aod->GetMagneticField())<0.001) return;
 595
 596
 597   TClonesArray *mcArray = 0;
 598   AliAODMCHeader *mcHeader = 0;
 599
 600   if(fReadMC) {
 601     // load MC particles
 602     mcArray = (TClonesArray*)aod->GetList()->FindObject(AliAODMCParticle::StdBranchName());
 603     if(!mcArray) {
 604       printf("AliAnalysisTaskSED0Mass::UserExec: MC particles branch not found!\n");
 605       return;
 606     }
 607
 608     // load MC header
 609     mcHeader = (AliAODMCHeader*)aod->GetList()->FindObject(AliAODMCHeader::StdBranchName());
 610     if(!mcHeader) {
 611       printf("AliAnalysisTaskSED0Mass::UserExec: MC header branch not found!\n");
 612       return;
 613     }
 614   }
 615
 616   //printf("VERTEX Z %f %f\n",vtx1->GetZ(),mcHeader->GetVtxZ());
 617
 618   //histogram filled with 1 for every AOD
 619   fNentries->Fill(0);
 620   fCounter->StoreEvent(aod,fReadMC);
 621   if(!fCuts->IsEventSelected(aod)) {
 622     if(fCuts->GetWhyRejection()==1) // rejected for pileup
 623       fNentries->Fill(13);
 624     return;
 625   }
 626
 627   // AOD primary vertex
 628   AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
 629
 630   Bool_t isGoodVtx=kFALSE;
 631
 632    //vtx1->Print();
 633   TString primTitle = vtx1->GetTitle();
 634   if(primTitle.Contains("VertexerTracks") && vtx1->GetNContributors()>0) {
 635     isGoodVtx=kTRUE;
 636     fNentries->Fill(3);
 637   }
 638
 639   // loop over candidates
 640   Int_t nInD0toKpi = inputArray->GetEntriesFast();
 641   if(fDebug>2) printf("Number of D0->Kpi: %d\n",nInD0toKpi);
 642
 643   // FILE *f=fopen("4display.txt","a");
 644   // fprintf(f,"Number of D0->Kpi: %d\n",nInD0toKpi);
 645   Int_t nSelectedloose=0,nSelectedtight=0;
 646   for (Int_t iD0toKpi = 0; iD0toKpi < nInD0toKpi; iD0toKpi++) {
 647     //Int_t nPosPairs=0, nNegPairs=0;
 648     //cout<<"inside the loop"<<endl;
 649     AliAODRecoDecayHF2Prong *d = (AliAODRecoDecayHF2Prong*)inputArray->UncheckedAt(iD0toKpi);
 650
 651     //check daughters
 652     if(!(d->GetDaughter(0) || d->GetDaughter(1))) {
 653       AliDebug(1,"at least one daughter not found!");
 654       fNentries->Fill(5);
 655       continue;
 656     }
 657
 658     // Bool_t unsetvtx=kFALSE;
 659     // if(!d->GetOwnPrimaryVtx()) {
 660     //   d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
 661     //   unsetvtx=kTRUE;
 662     // }
 663
 664
 665     //check reco daughter in acceptance
 666     /*
 667     Double_t eta0=d->EtaProng(0);
 668     Double_t eta1=d->EtaProng(1);
 669     */
 670     if ( fCuts->IsInFiducialAcceptance(d->Pt(),d->Y(421)) ) {
 671       //if( TMath::Abs(eta0)<0.9 && TMath::Abs(eta1)<0.9 ){
 672        //apply cuts on tracks
 673       nSelectedloose++;
 674       Int_t isSelected = fCuts->IsSelected(d,AliRDHFCuts::kTracks);
 675       if(((AliAODTrack*)d->GetDaughter(0))->GetTPCNcls() < 70 || ((AliAODTrack*)d->GetDaughter(1))->GetTPCNcls() < 70) isSelected=kFALSE;
 676       if (!isSelected) continue;
 677       nSelectedtight++;
 678       fNentries->Fill(6);
 679       if(fDebug>2) cout<<"tracks selected"<<endl;
 680
 681       Int_t ptbin=fCuts->PtBin(d->Pt());
 682       if(ptbin==-1) {fNentries->Fill(4); continue;} //out of bounds
 683       FillVarHists(aod,d,mcArray,fCuts,fDistr);
 684       FillMassHists(aod,d,mcArray,fCuts,fOutputMass);
 685     }
 686
 687     //if(unsetvtx) d->UnsetOwnPrimaryVtx();
 688   } //end for prongs
 689   fCounter->StoreCandidates(aod,nSelectedloose,kTRUE);
 690   fCounter->StoreCandidates(aod,nSelectedtight,kFALSE);
 691   // Post the data
 692   PostData(1,fOutputMass);
 693   PostData(2,fDistr);
 694   PostData(3,fNentries);
 695   PostData(5,fCounter);
 696   return;
 697 }
 698
 699 //____________________________________________________________________________
 700 void AliAnalysisTaskSED0Mass::FillVarHists(AliAODEvent* aod,AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout){
 701   //
 702   // function used in UserExec to fill variable histograms:
 703   //
 704
 705   //add distr here
 706   UInt_t pdgs[2];
 707
 708   Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
 709   pdgs[0]=211;
 710   pdgs[1]=321;
 711   Double_t minvD0 = part->InvMassD0();
 712   pdgs[1]=211;
 713   pdgs[0]=321;
 714   Double_t minvD0bar = part->InvMassD0bar();
 715   //cout<<"inside mass cut"<<endl;
 716
 717   Int_t pdgDgD0toKpi[2]={321,211};
 718   Int_t lab=-9999;
 719   if(fReadMC) lab=part->MatchToMC(421,arrMC,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
 720   //Double_t pt = d->Pt(); //mother pt
 721   Int_t isSelected=3;
 722   Int_t isSelectedPID=99;
 723
 724   //isSelectedPID = cuts->IsSelected(part,AliRDHFCuts::kPID); //0 rejected,1 D0,2 Dobar, 3 both
 725   isSelectedPID = cuts->IsSelectedPID(part); //0 rejected,1 D0,2 Dobar, 3 both
 726   if (isSelectedPID==0)fNentries->Fill(7);
 727   if (isSelectedPID==1)fNentries->Fill(8);
 728   if (isSelectedPID==2)fNentries->Fill(9);
 729   if (isSelectedPID==3)fNentries->Fill(10);
 730     //fNentries->Fill(8+isSelectedPID);
 731
 732   if(fCutOnDistr){
 733     isSelected = cuts->IsSelected(part,AliRDHFCuts::kCandidate,aod); //cuts with variables recalculated with new vertex (w/o daughters)
 734     if (!isSelected){
 735       //cout<<"Not Selected"<<endl;
 736       return;
 737     }
 738   }
 739
 740   Double_t invmasscut=0.03;
 741
 742   TString fillthispi="",fillthisK="",fillthis="";
 743
 744   Int_t ptbin=cuts->PtBin(part->Pt());
 745
 746   Double_t dz1[2],dz2[2],covar1[3],covar2[3];//,d0xd0proper,errd0xd0proper;
 747   dz1[0]=-99; dz2[0]=-99;
 748   Double_t d0[2];
 749   Double_t decl[2]={-99,-99};
 750   Bool_t recalcvtx=kFALSE;
 751
 752   if(fCuts->GetIsPrimaryWithoutDaughters()){
 753     recalcvtx=kTRUE;
 754     //recalculate vertex
 755     AliAODVertex *vtxProp=0x0;
 756     vtxProp=GetPrimaryVtxSkipped(aod,part);
 757     if(vtxProp) {
 758       part->SetOwnPrimaryVtx(vtxProp);
 759       //Bool_t unsetvtx=kTRUE;
 760       //Calculate d0 for daughter tracks with Vtx Skipped
 761       AliESDtrack *esdtrack1=new AliESDtrack((AliVTrack*)part->GetDaughter(0));
 762       AliESDtrack *esdtrack2=new AliESDtrack((AliVTrack*)part->GetDaughter(1));
 763       esdtrack1->PropagateToDCA(vtxProp,aod->GetMagneticField(),1.,dz1,covar1);
 764       esdtrack2->PropagateToDCA(vtxProp,aod->GetMagneticField(),1.,dz2,covar2);
 765       delete vtxProp; vtxProp=NULL;
 766       delete esdtrack1;
 767       delete esdtrack2;
 768     }
 769
 770     d0[0]=dz1[0];
 771     d0[1]=dz2[0];
 772
 773     decl[0]=part->DecayLength();
 774     decl[1]=part->NormalizedDecayLength();
 775     part->UnsetOwnPrimaryVtx();
 776
 777   }
 778
 779
 780   if(!fCutOnDistr || (fCutOnDistr && isSelected)){ //if no cuts or cuts passed
 781     //printf("\nif no cuts or cuts passed\n");
 782
 783     //disable the PID
 784     if(!fUsePid4Distr) isSelectedPID=0;
 785     if((lab>=0 && fReadMC) || (!fReadMC && isSelectedPID)){ //signal (from MC or PID)
 786
 787       //check pdg of the prongs
 788       AliAODTrack *prong0=(AliAODTrack*)part->GetDaughter(0);
 789       AliAODTrack *prong1=(AliAODTrack*)part->GetDaughter(1);
 790       if(!prong0 || !prong1) {
 791         return;
 792       }
 793       Int_t labprong[2];
 794       if(fReadMC){
 795         labprong[0]=prong0->GetLabel();
 796         labprong[1]=prong1->GetLabel();
 797       }
 798       AliAODMCParticle *mcprong=0;
 799       Int_t pdgProng[2]={0,0};
 800
 801       for (Int_t iprong=0;iprong<2;iprong++){
 802         if(fReadMC && labprong[iprong]>=0) {
 803           mcprong= (AliAODMCParticle*)arrMC->At(labprong[iprong]);
 804           pdgProng[iprong]=mcprong->GetPdgCode();
 805         }
 806       }
 807
 808
 809       //no mass cut ditributions: ptbis
 810
 811       // fillthispi="hptpiSnoMcut_";
 812       // fillthispi+=ptbin;
 813
 814       // fillthisK="hptKSnoMcut_";
 815       // fillthisK+=ptbin;
 816
 817       // if ((TMath::Abs(pdgProng[0]) == 211 && TMath::Abs(pdgProng[1]) == 321)
 818       //     || (isSelectedPID==1 || isSelectedPID==3)){
 819       //        ((TH1F*)listout->FindObject(fillthispi))->Fill(part->PtProng(0));
 820       //        ((TH1F*)listout->FindObject(fillthisK))->Fill(part->PtProng(1));
 821       // }
 822
 823       // if ((TMath::Abs(pdgProng[0]) == 321 && TMath::Abs(pdgProng[1]) == 211)
 824       //     || (isSelectedPID==2 || isSelectedPID==3)){
 825       //        ((TH1F*)listout->FindObject(fillthisK))->Fill(part->PtProng(0));
 826       //        ((TH1F*)listout->FindObject(fillthispi))->Fill(part->PtProng(1));
 827       // }
 828
 829       //no mass cut ditributions: mass
 830       fillthis="hMassS_";
 831       fillthis+=ptbin;
 832
 833       if ((fReadMC && ((AliAODMCParticle*)arrMC->At(lab))->GetPdgCode() == 421)
 834           || (!fReadMC && (isSelectedPID==1 || isSelectedPID==3))){//D0
 835         ((TH1F*)listout->FindObject(fillthis))->Fill(minvD0);
 836       }
 837       else { //D0bar
 838         if(fReadMC || (!fReadMC && isSelectedPID > 1))
 839         ((TH1F*)listout->FindObject(fillthis))->Fill(minvD0bar);
 840       }
 841
 842       //apply cut on invariant mass on the pair
 843       if(TMath::Abs(minvD0-mPDG)<invmasscut || TMath::Abs(minvD0bar-mPDG)<invmasscut){
 844
 845         if(fArray==1) cout<<"LS signal: ERROR"<<endl;
 846         for (Int_t iprong=0; iprong<2; iprong++){
 847           AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(iprong);
 848           if (fReadMC) labprong[iprong]=prong->GetLabel();
 849
 850           //cout<<"prong name = "<<prong->GetName()<<" label = "<<prong->GetLabel()<<endl;
 851           Int_t pdgprong=0;
 852           if(fReadMC && labprong[iprong]>=0) {
 853             mcprong= (AliAODMCParticle*)arrMC->At(labprong[iprong]);
 854             pdgprong=mcprong->GetPdgCode();
 855           }
 856
 857           Bool_t isPionHere[2]={(isSelectedPID==1 || isSelectedPID==3) ? kTRUE : kFALSE,(isSelectedPID==2 || isSelectedPID==3) ? kTRUE : kFALSE};
 858
 859           if(TMath::Abs(pdgprong)==211 || isPionHere[iprong]) {
 860             //cout<<"pi"<<endl;
 861
 862             fillthispi="hptpiS_";
 863             fillthispi+=ptbin;
 864             ((TH1F*)listout->FindObject(fillthispi))->Fill(part->PtProng(iprong));
 865             fillthispi="hd0piS_";
 866             fillthispi+=ptbin;
 867             ((TH1F*)listout->FindObject(fillthispi))->Fill(part->Getd0Prong(iprong));
 868             if(recalcvtx) {
 869
 870               fillthispi="hd0vpiS_";
 871               fillthispi+=ptbin;
 872               ((TH1F*)listout->FindObject(fillthispi))->Fill(d0[iprong]);
 873             }
 874
 875           }
 876
 877           if(TMath::Abs(pdgprong)==321 || !isPionHere[iprong]) {
 878             //cout<<"kappa"<<endl;
 879
 880             fillthisK="hptKS_";
 881             fillthisK+=ptbin;
 882             ((TH1F*)listout->FindObject(fillthisK))->Fill(part->PtProng(iprong));
 883             fillthisK="hd0KS_";
 884             fillthisK+=ptbin;
 885             ((TH1F*)listout->FindObject(fillthisK))->Fill(part->Getd0Prong(iprong));
 886             if (recalcvtx){
 887               fillthisK="hd0vKS_";
 888               fillthisK+=ptbin;
 889               ((TH1F*)listout->FindObject(fillthisK))->Fill(d0[iprong]);
 890             }
 891           }
 892
 893           fillthis="hcosthpointd0S_";
 894           fillthis+=ptbin;
 895           ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Getd0Prong(iprong));
 896
 897         } //end loop on prongs
 898
 899         fillthis="hdcaS_";
 900         fillthis+=ptbin;
 901         ((TH1F*)listout->FindObject(fillthis))->Fill(part->GetDCA());
 902
 903         fillthis="hcosthetapointS_";
 904         fillthis+=ptbin;
 905         ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle());
 906
 907         fillthis="hcosthpointd0d0S_";
 908         fillthis+=ptbin;
 909         ((TH2F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Prodd0d0());
 910
 911         fillthis="hcosthetastarS_";
 912         fillthis+=ptbin;
 913         if ((fReadMC && ((AliAODMCParticle*)arrMC->At(lab))->GetPdgCode() == 421)) ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0());
 914         else {
 915           if (fReadMC || isSelectedPID>1)((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0bar());
 916           if(isSelectedPID==1 || isSelectedPID==3)((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0());
 917         }
 918         fillthis="hd0d0S_";
 919         fillthis+=ptbin;
 920         ((TH1F*)listout->FindObject(fillthis))->Fill(part->Prodd0d0());
 921
 922         if(recalcvtx){
 923           fillthis="hd0d0vS_";
 924           fillthis+=ptbin;
 925           ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]*d0[1]);
 926         }
 927
 928         fillthis="hdeclS_";
 929         fillthis+=ptbin;
 930         ((TH1F*)listout->FindObject(fillthis))->Fill(part->DecayLength());
 931
 932         fillthis="hnormdeclS_";
 933         fillthis+=ptbin;
 934         ((TH1F*)listout->FindObject(fillthis))->Fill(part->NormalizedDecayLength());
 935
 936         if(recalcvtx) {
 937           fillthis="hdeclvS_";
 938           fillthis+=ptbin;
 939           ((TH1F*)listout->FindObject(fillthis))->Fill(decl[0]);
 940
 941           fillthis="hnormdeclvS_";
 942           fillthis+=ptbin;
 943           ((TH1F*)listout->FindObject(fillthis))->Fill(decl[1]);
 944         }
 945       } //end mass cut
 946
 947     } else{ //Background or LS
 948       //if(!fReadMC){
 949       //cout<<"is background"<<endl;
 950
 951       //no mass cut distributions: mass, ptbis
 952       fillthis="hMassB_";
 953       fillthis+=ptbin;
 954
 955       if (!fCutOnDistr || (fCutOnDistr && (isSelected==1 || isSelected==3))) ((TH1F*)listout->FindObject(fillthis))->Fill(minvD0);
 956       if (!fCutOnDistr || (fCutOnDistr && isSelected>1)) ((TH1F*)listout->FindObject(fillthis))->Fill(minvD0bar);
 957
 958       // fillthis="hptB1prongnoMcut_";
 959       // fillthis+=ptbin;
 960
 961       // ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(0));
 962
 963       // fillthis="hptB2prongsnoMcut_";
 964       // fillthis+=ptbin;
 965       // ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(0));
 966       // ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(1));
 967
 968       //apply cut on invariant mass on the pair
 969       if(TMath::Abs(minvD0-mPDG)<invmasscut || TMath::Abs(minvD0bar-mPDG)<invmasscut){
 970
 971
 972         AliAODTrack *prongg=(AliAODTrack*)part->GetDaughter(0);
 973         if(!prongg) {
 974           if(fDebug>2) cout<<"No daughter found";
 975           return;
 976         }
 977         else{
 978           if(fArray==1){
 979             if(prongg->Charge()==1) {
 980               //fTotPosPairs[ptbin]++;
 981               ((TH1F*)fDistr->FindObject("hpospair"))->Fill(ptbin);
 982             } else {
 983               //fTotNegPairs[ptbin]++;
 984               ((TH1F*)fDistr->FindObject("hnegpair"))->Fill(ptbin);
 985             }
 986           }
 987         }
 988
 989         //normalise pt distr to half afterwards
 990         fillthis="hptB_";
 991         fillthis+=ptbin;
 992         ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(0));
 993         ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(1));
 994
 995         fillthis="hd0p0B_";
 996         fillthis+=ptbin;
 997         ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(0));
 998         fillthis="hd0p1B_";
 999         fillthis+=ptbin;
1000         ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(1));
1001         fillthis="hd0B_";
1002         fillthis+=ptbin;
1003         ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(0));
1004         ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(1));
1005
1006         if(fReadMC){
1007           Int_t pdgMother[2]={0,0};
1008           Double_t factor[2]={1,1};
1009
1010           for(Int_t iprong=0;iprong<2;iprong++){
1011             AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(iprong);
1012             lab=prong->GetLabel();
1013             if(lab>=0){
1014               AliAODMCParticle* mcprong=(AliAODMCParticle*)arrMC->At(lab);
1015               if(mcprong){
1016                 Int_t labmom=mcprong->GetMother();
1017                 if(labmom>=0){
1018                   AliAODMCParticle* mcmother=(AliAODMCParticle*)arrMC->At(labmom);
1019                   if(mcmother) pdgMother[iprong]=mcmother->GetPdgCode();
1020                 }
1021               }
1022             }
1023
1024             fillthis="hd0moresB_";
1025             fillthis+=ptbin;
1026
1027             if(TMath::Abs(pdgMother[iprong])==310 || TMath::Abs(pdgMother[iprong])==130 || TMath::Abs(pdgMother[iprong])==321){ //K^0_S, K^0_L, K^+-
1028               if(part->PtProng(iprong)<=1)factor[iprong]=1./.7;
1029               else factor[iprong]=1./.6;
1030               fNentries->Fill(11);
1031             }
1032
1033             if(TMath::Abs(pdgMother[iprong])==3122) { //Lambda
1034               factor[iprong]=1./0.25;
1035               fNentries->Fill(12);
1036             }
1037             fillthis="hd0moresB_";
1038             fillthis+=ptbin;
1039
1040             ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(iprong),factor[iprong]);
1041
1042             if(recalcvtx){
1043               fillthis="hd0vmoresB_";
1044               fillthis+=ptbin;
1045               ((TH1F*)listout->FindObject(fillthis))->Fill(d0[iprong],factor[iprong]);
1046             }
1047
1048           }
1049
1050           fillthis="hd0d0moresB_";
1051           fillthis+=ptbin;
1052           ((TH1F*)listout->FindObject(fillthis))->Fill(part->Prodd0d0(),factor[0]*factor[1]);
1053
1054           if(recalcvtx){
1055             fillthis="hd0d0vmoresB_";
1056             fillthis+=ptbin;
1057             ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]*d0[1],factor[0]*factor[1]);
1058           }
1059           fillthis="hcosthetapointmoresB_";
1060           fillthis+=ptbin;
1061           ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),factor[0]*factor[1]);
1062         }
1063
1064         if(recalcvtx){
1065           fillthis="hd0vp0B_";
1066           fillthis+=ptbin;
1067           ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]);
1068           fillthis="hd0vp1B_";
1069           fillthis+=ptbin;
1070           ((TH1F*)listout->FindObject(fillthis))->Fill(d0[1]);
1071
1072           fillthis="hd0vB_";
1073           fillthis+=ptbin;
1074           ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]);
1075           ((TH1F*)listout->FindObject(fillthis))->Fill(d0[1]);
1076         }
1077
1078         fillthis="hcosthpointd0B_";
1079         fillthis+=ptbin;
1080         ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Getd0Prong(0));
1081         ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Getd0Prong(1));
1082
1083
1084         fillthis="hdcaB_";
1085         fillthis+=ptbin;
1086         ((TH1F*)listout->FindObject(fillthis))->Fill(part->GetDCA());
1087
1088         fillthis="hcosthetastarB_";
1089         fillthis+=ptbin;
1090         if (!fCutOnDistr || (fCutOnDistr && (isSelected==1 || isSelected==3)))((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0());
1091         if (!fCutOnDistr || (fCutOnDistr && isSelected>1))((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0bar());
1092
1093         fillthis="hd0d0B_";
1094         fillthis+=ptbin;
1095         ((TH1F*)listout->FindObject(fillthis))->Fill(part->Prodd0d0());
1096
1097         if(recalcvtx){
1098           fillthis="hd0d0vB_";
1099           fillthis+=ptbin;
1100           ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]*d0[1]);
1101         }
1102
1103         fillthis="hcosthetapointB_";
1104         fillthis+=ptbin;
1105         ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle());
1106
1107         fillthis="hcosthpointd0d0B_";
1108         fillthis+=ptbin;
1109         ((TH2F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Prodd0d0());
1110
1111         fillthis="hdeclB_";
1112         fillthis+=ptbin;
1113         ((TH1F*)listout->FindObject(fillthis))->Fill(part->DecayLength());
1114
1115         fillthis="hnormdeclB_";
1116         fillthis+=ptbin;
1117         ((TH1F*)listout->FindObject(fillthis))->Fill(part->NormalizedDecayLength());
1118
1119         if(recalcvtx) {
1120
1121           fillthis="hdeclvB_";
1122           fillthis+=ptbin;
1123           ((TH1F*)listout->FindObject(fillthis))->Fill(decl[0]);
1124
1125           fillthis="hnormdeclvB_";
1126           fillthis+=ptbin;
1127           ((TH1F*)listout->FindObject(fillthis))->Fill(decl[1]);
1128
1129         }
1130       }//mass cut
1131     }//else (background)
1132   }
1133   return;
1134 }
1135 //____________________________________________________________________________
1136
1137 void AliAnalysisTaskSED0Mass::FillMassHists(AliAODEvent* aod,AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi* cuts, TList *listout){
1138   //
1139   // function used in UserExec to fill mass histograms:
1140   //
1141
1142
1143   //  Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
1144
1145   Int_t isSelected=cuts->IsSelected(part,AliRDHFCuts::kCandidate,aod); //selected
1146
1147   //cout<<"is selected = "<<isSelected<<endl;
1148
1149   //cout<<"check cuts = "<<endl;
1150   //cuts->PrintAll();
1151   if (!isSelected){
1152     //cout<<"Not Selected"<<endl;
1153     //cout<<"Rejected because "<<cuts->GetWhy()<<endl;
1154     return;
1155   }
1156
1157
1158   if(fDebug>2)  cout<<"Candidate selected"<<endl;
1159
1160   Double_t invmassD0 = part->InvMassD0(), invmassD0bar = part->InvMassD0bar();
1161   //printf("SELECTED\n");
1162   Int_t ptbin=cuts->PtBin(part->Pt());
1163
1164   AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(0);
1165   if(!prong) {
1166     AliDebug(2,"No daughter found");
1167     return;
1168   }
1169   else{
1170     // if(prong->Charge()==1) {
1171     //   ((TH1F*)fDistr->FindObject("hpospair"))->Fill(fCuts->GetNPtBins()+ptbin);
1172     //   //fTotPosPairs[ptbin]++;
1173     // } else {
1174     //   ((TH1F*)fDistr->FindObject("hnegpair"))->Fill(fCuts->GetNPtBins()+ptbin);
1175     //   //fTotNegPairs[ptbin]++;
1176     // }
1177   }
1178
1179   // for(Int_t it=0;it<2;it++){
1180
1181   //    //request on spd points to be addes
1182   //   if(/*nSPD==2 && */part->Pt() > 5. && (TMath::Abs(invmassD0-mPDG)<0.01 || TMath::Abs(invmassD0bar-mPDG)<0.01)){
1183   //     FILE *f=fopen("4display.txt","a");
1184   //     fprintf(f,"pt: %f \n Rapidity: %f \t Period Number: %x \t Run Number: %d \t BunchCrossNumb: %d \t OrbitNumber: %d\n",part->Pt(),part->Y(421),aod->GetPeriodNumber(),aod->GetRunNumber(),aod->GetBunchCrossNumber(),aod->GetOrbitNumber());
1185   //     fclose(f);
1186   //     //printf("PrimVtx NContributors: %d \n Prongs Rel Angle: %f \n \n",ncont,relangle);
1187   //   }
1188   // }
1189
1190   TString fillthis="";
1191   Int_t pdgDgD0toKpi[2]={321,211};
1192   Int_t labD0=-1;
1193   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)
1194
1195   //count candidates selected by cuts
1196   fNentries->Fill(1);
1197   //count true D0 selected by cuts
1198   if (fReadMC && labD0>=0) fNentries->Fill(2);
1199
1200   if ((isSelected==1 || isSelected==3) && fFillOnlyD0D0bar<2) { //D0
1201     fillthis="histMass_";
1202     fillthis+=ptbin;
1203     //cout<<"Filling "<<fillthis<<endl;
1204
1205     //printf("Fill mass with D0");
1206     ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
1207
1208     if(labD0>=0) {
1209       if(fArray==1) cout<<"LS signal ERROR"<<endl;
1210
1211       AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
1212       Int_t pdgD0 = partD0->GetPdgCode();
1213       //cout<<"pdg = "<<pdgD0<<endl;
1214       if (pdgD0==421){ //D0
1215         //cout<<"Fill S with D0"<<endl;
1216         fillthis="histSgn_";
1217         fillthis+=ptbin;
1218         ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
1219         // if(TMath::Abs(invmassD0 - mPDG) < 0.027){
1220         //   fillthis="histSgn27_";
1221         //   fillthis+=ptbin;
1222         //   ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
1223         // }
1224       } else{ //it was a D0bar
1225         fillthis="histRfl_";
1226         fillthis+=ptbin;
1227         ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
1228       }
1229     } else {//background
1230       fillthis="histBkg_";
1231       fillthis+=ptbin;
1232       ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
1233     }
1234
1235   }
1236   if (isSelected>1 && (fFillOnlyD0D0bar==0 || fFillOnlyD0D0bar==2)) { //D0bar
1237     fillthis="histMass_";
1238     fillthis+=ptbin;
1239     //printf("Fill mass with D0bar");
1240     ((TH1F*)listout->FindObject(fillthis))->Fill(invmassD0bar);
1241
1242     if(labD0>=0) {
1243       if(fArray==1) cout<<"LS signal ERROR"<<endl;
1244       AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
1245       Int_t pdgD0 = partD0->GetPdgCode();
1246       //cout<<" pdg = "<<pdgD0<<endl;
1247       if (pdgD0==-421){ //D0bar
1248         fillthis="histSgn_";
1249         fillthis+=ptbin;
1250         ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
1251         // if (TMath::Abs(invmassD0bar - mPDG) < 0.027){
1252         //   fillthis="histSgn27_";
1253         //   fillthis+=ptbin;
1254         //   ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
1255         // }
1256
1257
1258       } else{
1259         fillthis="histRfl_";
1260         fillthis+=ptbin;
1261         ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
1262       }
1263     } else {//background or LS
1264       fillthis="histBkg_";
1265       fillthis+=ptbin;
1266       ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
1267     }
1268   }
1269
1270   return;
1271 }
1272
1273 //__________________________________________________________________________
1274 AliAODVertex* AliAnalysisTaskSED0Mass::GetPrimaryVtxSkipped(AliAODEvent *aodev,AliAODRecoDecayHF2Prong *d){
1275   //Calculate the primary vertex w/o the daughter tracks of the candidate
1276
1277   Int_t skipped[2];
1278   Int_t nTrksToSkip=2;
1279   AliAODTrack *dgTrack = (AliAODTrack*)d->GetDaughter(0);
1280   if(!dgTrack){
1281     AliDebug(2,"no daughter found!");
1282     return 0x0;
1283   }
1284   skipped[0]=dgTrack->GetID();
1285   dgTrack = (AliAODTrack*)d->GetDaughter(1);
1286   if(!dgTrack){
1287     AliDebug(2,"no daughter found!");
1288     return 0x0;
1289   }
1290   skipped[1]=dgTrack->GetID();
1291
1292   AliESDVertex *vertexESD=0x0;
1293   AliAODVertex *vertexAOD=0x0;
1294   AliVertexerTracks *vertexer = new AliVertexerTracks(aodev->GetMagneticField());
1295
1296   //
1297   vertexer->SetSkipTracks(nTrksToSkip,skipped);
1298   vertexer->SetMinClusters(4);
1299   vertexESD = (AliESDVertex*)vertexer->FindPrimaryVertex(aodev);
1300   if(!vertexESD) return vertexAOD;
1301   if(vertexESD->GetNContributors()<=0) {
1302     AliDebug(2,"vertexing failed");
1303     delete vertexESD; vertexESD=NULL;
1304     return vertexAOD;
1305   }
1306
1307   delete vertexer; vertexer=NULL;
1308
1309
1310   // convert to AliAODVertex
1311   Double_t pos[3],cov[6],chi2perNDF;
1312   vertexESD->GetXYZ(pos); // position
1313   vertexESD->GetCovMatrix(cov); //covariance matrix
1314   chi2perNDF = vertexESD->GetChi2toNDF();
1315   delete vertexESD; vertexESD=NULL;
1316
1317   vertexAOD = new AliAODVertex(pos,cov,chi2perNDF);
1318   return vertexAOD;
1319
1320 }
1321
1322
1323 //________________________________________________________________________
1324 void AliAnalysisTaskSED0Mass::Terminate(Option_t */*option*/)
1325 {
1326   // Terminate analysis
1327   //
1328   if(fDebug > 1) printf("AnalysisTaskSED0Mass: Terminate() \n");
1329
1330
1331   fOutputMass = dynamic_cast<TList*> (GetOutputData(1));
1332   if (!fOutputMass) {
1333     printf("ERROR: fOutputMass not available\n");
1334     return;
1335   }
1336   fDistr = dynamic_cast<TList*> (GetOutputData(2));
1337   if (!fDistr) {
1338     printf("ERROR: fDistr not available\n");
1339     return;
1340   }
1341
1342   fNentries = dynamic_cast<TH1F*>(GetOutputData(3));
1343
1344   if(!fNentries){
1345     printf("ERROR: fNEntries not available\n");
1346     return;
1347   }
1348   fCuts = dynamic_cast<AliRDHFCutsD0toKpi*>(GetOutputData(4));
1349   if(!fCuts){
1350     printf("ERROR: fCuts not available\n");
1351     return;
1352   }
1353   fCounter = dynamic_cast<AliNormalizationCounter*>(GetOutputData(5));
1354   if (!fCounter) {
1355     printf("ERROR: fCounter not available\n");
1356     return;
1357   }
1358   Int_t nptbins=fCuts->GetNPtBins();
1359   for(Int_t ipt=0;ipt<nptbins;ipt++){
1360
1361     if(fArray==1){
1362       fLsNormalization = 2.*TMath::Sqrt(((TH1F*)fDistr->FindObject("hpospair"))->Integral(nptbins+ipt+1,nptbins+ipt+2)*((TH1F*)fDistr->FindObject("hnegpair"))->Integral(nptbins+ipt+1,nptbins+ipt+2)); //after cuts
1363
1364
1365       if(fLsNormalization>1e-6) {
1366
1367         TString massName="histMass_";
1368         massName+=ipt;
1369         ((TH1F*)fOutputMass->FindObject(massName))->Scale((1/fLsNormalization)*((TH1F*)fOutputMass->FindObject(massName))->GetEntries());
1370
1371       }
1372
1373
1374       fLsNormalization = 2.*TMath::Sqrt(((TH1F*)fDistr->FindObject("hpospair"))->Integral(ipt+1,ipt+2)*((TH1F*)fDistr->FindObject("hnegpair"))->Integral(ipt+1,ipt+2));
1375       //fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[4]*fTotNegPairs[4]);
1376
1377       if(fLsNormalization>1e-6) {
1378
1379         TString nameDistr="hptB_";
1380         nameDistr+=ipt;
1381         ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1382         nameDistr="hdcaB_";
1383         nameDistr+=ipt;
1384         ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1385         nameDistr="hcosthetastarB_";
1386         nameDistr+=ipt;
1387         ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1388         nameDistr="hd0B_";
1389         nameDistr+=ipt;
1390         ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1391         nameDistr="hd0d0B_";
1392         nameDistr+=ipt;
1393         ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1394         nameDistr="hcosthetapointB_";
1395         nameDistr+=ipt;
1396         ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
1397         nameDistr="hcosthpointd0d0B_";
1398         nameDistr+=ipt;
1399         ((TH2F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH2F*)fDistr->FindObject(nameDistr))->GetEntries());
1400
1401       }
1402     }
1403   }
1404   TString cvname,cstname;
1405
1406   if (fArray==0){
1407     cvname="D0invmass";
1408     cstname="cstat0";
1409   } else {
1410     cvname="LSinvmass";
1411     cstname="cstat1";
1412   }
1413
1414   TCanvas *cMass=new TCanvas(cvname,cvname);
1415   cMass->cd();
1416   ((TH1F*)fOutputMass->FindObject("histMass_3"))->Draw();
1417
1418   TCanvas* cStat=new TCanvas(cstname,Form("Stat%s",fArray ? "LS" : "D0"));
1419   cStat->cd();
1420   cStat->SetGridy();
1421   fNentries->Draw("htext0");
1422
1423   // TCanvas *ccheck=new TCanvas(Form("cc%d",fArray),Form("cc%d",fArray));
1424   // ccheck->cd();
1425
1426   return;
1427 }
1428