PWG3/vertexingHF/AliAnalysisTaskSEHFQA.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 /* $Id$ */
  17
  18 /////////////////////////////////////////////////////////////
  19 //
  20 // AliAnalysisTaskSE for HF quality assurance
  21 //
  22 // Author: Chiara Bianchin, chiara.bianchin@pd.infn.it
  23 /////////////////////////////////////////////////////////////
  24
  25 #include <Riostream.h>
  26 #include <TClonesArray.h>
  27 #include <TCanvas.h>
  28 #include <TList.h>
  29 #include <TH1F.h>
  30 #include <TH2F.h>
  31 #include <TDatabasePDG.h>
  32
  33 #include <AliAnalysisDataSlot.h>
  34 #include <AliAnalysisDataContainer.h>
  35 #include "AliAnalysisManager.h"
  36 #include "AliESDtrack.h"
  37 #include "AliVertexerTracks.h"
  38 #include "AliPID.h"
  39 #include "AliTPCPIDResponse.h"
  40 #include "AliAODHandler.h"
  41 #include "AliAODEvent.h"
  42 #include "AliAODVertex.h"
  43 #include "AliAODTrack.h"
  44 #include "AliAODMCParticle.h"
  45 #include "AliAODMCHeader.h"
  46 #include "AliAODRecoDecayHF2Prong.h"
  47 #include "AliAODRecoCascadeHF.h"
  48 #include "AliAnalysisVertexingHF.h"
  49 #include "AliAnalysisTaskSE.h"
  50 #include "AliCounterCollection.h"
  51 #include "AliRDHFCuts.h"
  52 #include "AliRDHFCutsDplustoKpipi.h"
  53 #include "AliRDHFCutsD0toKpipipi.h"
  54 #include "AliRDHFCutsDstoKKpi.h"
  55 #include "AliRDHFCutsDStartoKpipi.h"
  56 #include "AliRDHFCutsD0toKpi.h"
  57 #include "AliRDHFCutsLctopKpi.h"
  58
  59 #include "AliAnalysisTaskSEHFQA.h"
  60
  61 ClassImp(AliAnalysisTaskSEHFQA)
  62
  63 //____________________________________________________________________________
  64
  65 AliAnalysisTaskSEHFQA::AliAnalysisTaskSEHFQA():AliAnalysisTaskSE(),
  66   fNEntries(0x0),
  67   fOutputPID(0x0),
  68   fOutputTrack(0x0),
  69   fOutputCounters(0x0),
  70   fOutputCheckCentrality(0x0),
  71   fDecayChannel(AliAnalysisTaskSEHFQA::kD0toKpi),
  72   fCuts(0x0),
  73   fEstimator(AliRDHFCuts::kCentTRK),
  74   fReadMC(kFALSE),
  75   fSimpleMode(kFALSE),
  76   fOnOff()
  77 {
  78   //default constructor
  79   fOnOff[0]=kTRUE;
  80   fOnOff[1]=kTRUE;
  81   fOnOff[2]=kTRUE;
  82 }
  83
  84 //____________________________________________________________________________
  85 AliAnalysisTaskSEHFQA::AliAnalysisTaskSEHFQA(const char *name, AliAnalysisTaskSEHFQA::DecChannel ch,AliRDHFCuts* cuts):
  86   AliAnalysisTaskSE(name),
  87   fNEntries(0x0),
  88   fOutputPID(0x0),
  89   fOutputTrack(0x0),
  90   fOutputCounters(0x0),
  91   fOutputCheckCentrality(0x0),
  92   fDecayChannel(ch),
  93   fCuts(0x0),
  94   fEstimator(AliRDHFCuts::kCentTRK),
  95   fReadMC(kFALSE),
  96   fSimpleMode(kFALSE),
  97   fOnOff()
  98 {
  99   //constructor
 100
 101   //SetCutObject(cuts);
 102   fCuts=cuts;
 103
 104   fOnOff[0]=kTRUE;
 105   fOnOff[1]=kTRUE;
 106   fOnOff[2]=kTRUE;
 107
 108   // Output slot #1 writes into a TH1F container (number of events)
 109   DefineOutput(1,TH1F::Class());  //My private output
 110   // Output slot #2 writes into a TList container (PID)
 111   if (fOnOff[1]) DefineOutput(2,TList::Class());  //My private output
 112   // Output slot #3 writes into a TList container (Tracks)
 113   if (fOnOff[0]) DefineOutput(3,TList::Class());  //My private output
 114   // Output slot #4 writes into a AliRDHFCuts container (cuts)
 115   switch(fDecayChannel){
 116   case 0:
 117     DefineOutput(4,AliRDHFCutsDplustoKpipi::Class());  //My private output
 118     break;
 119   case 1:
 120     DefineOutput(4,AliRDHFCutsD0toKpi::Class());  //My private output
 121     break;
 122   case 2:
 123     DefineOutput(4,AliRDHFCutsDStartoKpipi::Class());  //My private output
 124     break;
 125   case 3:
 126     DefineOutput(4,AliRDHFCutsDstoKKpi::Class());  //My private output
 127     break;
 128   case 4:
 129     DefineOutput(4,AliRDHFCutsD0toKpipipi::Class());  //My private output
 130     break;
 131   case 5:
 132     DefineOutput(4,AliRDHFCutsLctopKpi::Class());  //My private output
 133     break;
 134   }
 135   if (fOnOff[2]) {
 136     // Output slot #5 writes into a TList container (AliCounterCollection)
 137     DefineOutput(5,TList::Class());  //My private output
 138     // Output slot #6 writes into a TList container (TH1F)
 139     DefineOutput(6,TList::Class());  //My private output
 140   }
 141 }
 142
 143 //___________________________________________________________________________
 144 AliAnalysisTaskSEHFQA::~AliAnalysisTaskSEHFQA()
 145 {
 146   //destructor
 147   delete fNEntries;
 148
 149   delete fOutputPID;
 150
 151   delete fOutputTrack;
 152
 153   delete fOutputCounters;
 154
 155   delete fOutputCheckCentrality;
 156
 157 }
 158
 159 //___________________________________________________________________________
 160 void AliAnalysisTaskSEHFQA::Init(){
 161
 162   //initialization
 163   if(fDebug > 1) printf("AnalysisTaskSEHFQA::Init() \n");
 164
 165   switch(fDecayChannel){
 166   case 0:
 167     {
 168       AliRDHFCutsDplustoKpipi* copycut=new AliRDHFCutsDplustoKpipi(*(static_cast<AliRDHFCutsDplustoKpipi*>(fCuts)));
 169       // Post the data
 170       PostData(4,copycut);
 171     }
 172     break;
 173   case 1:
 174     {
 175       AliRDHFCutsD0toKpi* copycut=new AliRDHFCutsD0toKpi(*(static_cast<AliRDHFCutsD0toKpi*>(fCuts)));
 176       // Post the data
 177       PostData(4,copycut);
 178     }
 179     break;
 180   case 2:
 181     {
 182       AliRDHFCutsDStartoKpipi* copycut=new AliRDHFCutsDStartoKpipi(*(static_cast<AliRDHFCutsDStartoKpipi*>(fCuts)));
 183       // Post the data
 184       PostData(4,copycut);
 185     }
 186     break;
 187   case 3:
 188     {
 189       AliRDHFCutsDstoKKpi* copycut=new AliRDHFCutsDstoKKpi(*(static_cast<AliRDHFCutsDstoKKpi*>(fCuts)));
 190       // Post the data
 191       PostData(4,copycut);
 192     }
 193     break;
 194   case 4:
 195     {
 196       AliRDHFCutsD0toKpipipi* copycut=new AliRDHFCutsD0toKpipipi(*(static_cast<AliRDHFCutsD0toKpipipi*>(fCuts)));
 197       // Post the data
 198       PostData(4,copycut);
 199     }
 200     break;
 201   case 5:
 202     {
 203       AliRDHFCutsLctopKpi* copycut=new AliRDHFCutsLctopKpi(*(static_cast<AliRDHFCutsLctopKpi*>(fCuts)));
 204       // Post the data
 205       PostData(4,copycut);
 206     }
 207     break;
 208
 209   default:
 210     return;
 211   }
 212
 213
 214
 215 }
 216
 217 //___________________________________________________________________________
 218 void AliAnalysisTaskSEHFQA::UserCreateOutputObjects()
 219 {
 220
 221   //create the output container
 222   if(fDebug > 1) printf("AnalysisTaskSEHFQA::UserCreateOutputObjects() \n");
 223
 224   //count events
 225
 226   fNEntries=new TH1F(GetOutputSlot(1)->GetContainer()->GetName(), "Counts the number of events", 10,-0.5,9.5);
 227   fNEntries->GetXaxis()->SetBinLabel(1,"nEventsAnal");
 228   fNEntries->GetXaxis()->SetBinLabel(2,"Pile-up Rej");
 229   fNEntries->GetXaxis()->SetBinLabel(3,"No VertexingHF");
 230   fNEntries->GetXaxis()->SetBinLabel(4,"nCandidates(AnCuts)");
 231   fNEntries->GetXaxis()->SetBinLabel(5,"EventsWithGoodVtx");
 232   //fNEntries->GetXaxis()->SetBinLabel(6,"N. of 0SMH");
 233   fNEntries->GetXaxis()->SetBinLabel(6,"N. of CSH1");
 234   if(fReadMC){
 235     fNEntries->GetXaxis()->SetBinLabel(7,"MC Cand from c");
 236     fNEntries->GetXaxis()->SetBinLabel(8,"MC Cand from b");
 237     fNEntries->GetXaxis()->SetBinLabel(9,"N fake Trks");
 238     fNEntries->GetXaxis()->SetBinLabel(10,"N true Trks");
 239   } else{
 240     fNEntries->GetXaxis()->SetBinLabel(7,"N candidates");
 241   }
 242
 243   fNEntries->GetXaxis()->SetNdivisions(1,kFALSE);
 244
 245   //PID
 246   if(fOnOff[1]){
 247     fOutputPID=new TList();
 248     fOutputPID->SetOwner();
 249     fOutputPID->SetName(GetOutputSlot(2)->GetContainer()->GetName());
 250
 251     //TOF pid
 252     TString hname="hTOFsig";
 253     TH1F* hTOFsig=new TH1F(hname.Data(),"Distribution of TOF signal;TOF time [ps];Entries", 100, -2.e3,40.e3);
 254
 255     hname="hTOFtime";
 256     TH1F* hTOFtime=new TH1F(hname.Data(),"Distribution of TOF time Kaon;TOF time(Kaon) [ps];Entries", 1000, 0.,50000.);
 257
 258     hname="hTOFtimeKaonHyptime";
 259     TH2F* hTOFtimeKaonHyptime=new TH2F(hname.Data(),"TOFtime - timeHypothesisForKaon;p[GeV/c];TOFtime - timeHypothesisForKaon [ps]",200,0.,4.,1000,-20000.,20000.);
 260
 261     hname="hTOFtimeKaonHyptimeAC";
 262     TH2F* hTOFtimeKaonHyptimeAC=new TH2F(hname.Data(),"TOFtime - timeHypothesisForKaon;p[GeV/c];TOFtime - timeHypothesisForKaon [ps]",200,0.,4.,1000,-20000.,20000.);
 263
 264     hname="hTOFsigmaKSigPid";
 265     TH2F* hTOFsigmaKSigPid=new TH2F(hname.Data(),"(TOFsignal-timeK)/tofSigPid;p[GeV/c];(TOFsignal-timeK)/tofSigPid",200,0.,4.,100,-5,5);
 266
 267     hname="hTOFsigmaPionSigPid";
 268     TH2F* hTOFsigmaPionSigPid=new TH2F(hname.Data(),"(TOFsignal-time#pi)/tofSigPid;p[GeV/c];(TOFsignal-time#pi)/tofSigPid",200,0.,4.,100,-5,5);
 269
 270     hname="hTOFsigmaProtonSigPid";
 271     TH2F* hTOFsigmaProtonSigPid=new TH2F(hname.Data(),"(TOFsignal-timep)/tofSigPid;p[GeV/c];(TOFsignal-time p)/tofSigPid",200,0.,4.,100,-5,5);
 272
 273     hname="hTOFsigPid3sigPion";
 274     TH1F* hTOFsigPid3sigPion=new TH1F(hname.Data(),"TOF PID resolution (#pi) [ps]",500,0.,1000.);
 275
 276     hname="hTOFsigPid3sigKaon";
 277     TH1F* hTOFsigPid3sigKaon=new TH1F(hname.Data(),"TOF PID resolution (K) [ps]",500,0.,1000.);
 278
 279     hname="hTOFsigPid3sigProton";
 280     TH1F* hTOFsigPid3sigProton=new TH1F(hname.Data(),"TOF PID resolution (p) [ps]",500,0.,1000.);
 281
 282
 283     //TPC pid
 284     hname="hTPCsig";
 285     TH1F* hTPCsig=new TH1F(hname.Data(),"Distribution of TPC signal;TPC sig;Entries", 100, 35.,100.);
 286
 287     hname="hTPCsigvsp";
 288     TH2F* hTPCsigvsp=new TH2F(hname.Data(),"TPCsig vs p;TPC p[GeV/c];TPCsig",200,0.,4.,1000,35.,100.);
 289
 290     hname="hTPCsigvspAC";
 291     TH2F* hTPCsigvspAC=new TH2F(hname.Data(),"TPCsig vs p;TPCp[GeV/c];TPCsig",200,0.,4.,1000,35.,100.);
 292
 293     hname="hTPCsigmaK";
 294     TH2F* hTPCsigmaK=new TH2F(hname.Data(),"TPC Sigma for K as a function of momentum;p[GeV/c];Sigma Kaon",200,0.,4.,200,-5,5);
 295
 296     hname="hTPCsigmaPion";
 297     TH2F* hTPCsigmaPion=new TH2F(hname.Data(),"TPC Sigma for #pi as a function of momentum;p[GeV/c];Sigma #pi",200,0.,4.,200,-5,5);
 298
 299     hname="hTPCsigmaProton";
 300     TH2F* hTPCsigmaProton=new TH2F(hname.Data(),"TPC Sigma for proton as a function of momentum;p[GeV/c];Sigma Proton",200,0.,4.,200,-5,5);
 301
 302     fOutputPID->Add(hTOFsig);
 303     fOutputPID->Add(hTPCsig);
 304     fOutputPID->Add(hTOFtime);
 305     fOutputPID->Add(hTOFtimeKaonHyptime);
 306     fOutputPID->Add(hTOFtimeKaonHyptimeAC);
 307     fOutputPID->Add(hTOFsigmaKSigPid);
 308     fOutputPID->Add(hTOFsigmaPionSigPid);
 309     fOutputPID->Add(hTOFsigmaProtonSigPid);
 310     fOutputPID->Add(hTOFsigPid3sigPion);
 311     fOutputPID->Add(hTOFsigPid3sigKaon);
 312     fOutputPID->Add(hTOFsigPid3sigProton);
 313     fOutputPID->Add(hTPCsigvsp);
 314     fOutputPID->Add(hTPCsigvspAC);
 315     fOutputPID->Add(hTPCsigmaK);
 316     fOutputPID->Add(hTPCsigmaPion);
 317     fOutputPID->Add(hTPCsigmaProton);
 318   }
 319
 320   //quality of the tracks
 321   if(fOnOff[0]){
 322     fOutputTrack=new TList();
 323     fOutputTrack->SetOwner();
 324     fOutputTrack->SetName(GetOutputSlot(3)->GetContainer()->GetName());
 325
 326     TString hname="hnClsITS";
 327     TH1F* hnClsITS=new TH1F(hname.Data(),"Distribution of number of ITS clusters;nITScls;Entries",7,-0.5,6.5);
 328
 329     hname="hnClsITS-SA";
 330     TH1F* hnClsITSSA=new TH1F(hname.Data(),"Distribution of number of ITS clusters(ITS-SA);nITScls;Entries",7,-0.5,6.5);
 331
 332     hname="hnClsSPD";
 333     TH1F* hnClsSPD=new TH1F(hname.Data(),"Distribution of number of SPD clusters;nSPDcls;Entries",3,-0.5,2.5);
 334
 335     hname="hptGoodTr";
 336     TH1F* hptGoodTr=new TH1F(hname.Data(),"Pt distribution of 'good' tracks;p_{t}[GeV];Entries/0.05 GeV/c",400,0.,20.);
 337     hptGoodTr->SetTitleOffset(1.3,"Y");
 338
 339     hname="hdistrGoodTr";
 340     TH1F* hdistrGoodTr=new TH1F(hname.Data(),"Distribution of number of 'good' tracks per event;no.good-tracks/ev;Entries",4000,-0.5,3999.5);
 341     hdistrGoodTr->SetTitleOffset(1.3,"Y");
 342
 343     hname="hd0";
 344     TH1F* hd0=new TH1F(hname.Data(),"Impact parameter distribution of 'good' tracks;d_{0}[cm];Entries/10^{3} cm",200,-0.1,0.1);
 345
 346     fOutputTrack->Add(hnClsITS);
 347     fOutputTrack->Add(hnClsITSSA);
 348     fOutputTrack->Add(hnClsSPD);
 349     fOutputTrack->Add(hptGoodTr);
 350     fOutputTrack->Add(hdistrGoodTr);
 351     fOutputTrack->Add(hd0);
 352
 353     if(fReadMC){
 354       hname="hdistrFakeTr";
 355       TH1F* hdistrFakeTr=new TH1F(hname.Data(),"Distribution of number of fake tracks per event;no.fake-tracks/ev;Entries",4000,-0.5,3999.5);
 356       hdistrGoodTr->SetTitleOffset(1.3,"Y");
 357
 358       hname="hd0f";
 359       TH1F* hd0f=new TH1F(hname.Data(),"Impact parameter distribution of fake tracks;d_{0}[cm];Entries/10^{3} cm",200,-0.1,0.1);
 360
 361       hname="hptFakeTr";
 362       TH1F* hptFakeTr=new TH1F(hname.Data(),"Pt distribution of fake tracks;p_{t}[GeV];Entries/0.05 GeV/c",400,0.,20.);
 363       hptGoodTr->SetTitleOffset(1.3,"Y");
 364
 365       fOutputTrack->Add(hptFakeTr);
 366       fOutputTrack->Add(hdistrFakeTr);
 367       fOutputTrack->Add(hd0f);
 368
 369     }
 370   }
 371
 372
 373   if(fOnOff[2] && fCuts->GetUseCentrality()){
 374
 375     //Centrality (Counters)
 376     fOutputCounters=new TList();
 377     fOutputCounters->SetOwner();
 378     fOutputCounters->SetName(GetOutputSlot(5)->GetContainer()->GetName());
 379
 380     AliCounterCollection *stdEstimator=new AliCounterCollection("stdEstimator");
 381     stdEstimator->AddRubric("run",500000);
 382     stdEstimator->AddRubric("centralityclass","-10_0/0_10/10_20/20_30/30_40/40_50/50_60/60_70/70_80/80_90/90_100/-990_-980");
 383     stdEstimator->Init();
 384     AliCounterCollection *secondEstimator=new AliCounterCollection("secondEstimator");
 385     secondEstimator->AddRubric("run",500000);
 386     secondEstimator->AddRubric("centralityclass","-10_0/0_10/10_20/20_30/30_40/40_50/50_60/60_70/70_80/80_90/90_100/-990_-980");
 387     secondEstimator->Init();
 388
 389     fOutputCounters->Add(stdEstimator);
 390     fOutputCounters->Add(secondEstimator);
 391
 392     //Centrality (Checks)
 393     fOutputCheckCentrality=new TList();
 394     fOutputCheckCentrality->SetOwner();
 395     fOutputCheckCentrality->SetName(GetOutputSlot(6)->GetContainer()->GetName());
 396
 397     TString hname="hNtrackletsIn";
 398     TH1F* hNtrackletsIn=new TH1F(hname.Data(),"Number of tracklets in Centrality range;ntracklets;Entries",5000,-0.5,4999.5);
 399
 400     hname="hMultIn";
 401     TH1F* hMultIn=new TH1F(hname.Data(),"Multiplicity;multiplicity in Centrality range;Entries",10000,-0.5,9999.5);
 402
 403     hname="hNtrackletsOut";
 404     TH1F* hNtrackletsOut=new TH1F(hname.Data(),"Number of tracklets out of Centrality range;ntracklets;Entries",5000,-0.5,4999.5);
 405
 406     hname="hMultOut";
 407     TH1F* hMultOut=new TH1F(hname.Data(),"Multiplicity out of Centrality range;multiplicity;Entries",10000,-0.5,9999.5);
 408
 409     hname="hMultvsPercentile";
 410     TH2F* hMultvsPercentile=new TH2F(hname.Data(),"Multiplicity vs Percentile;multiplicity;percentile",10000,-0.5,9999.5,12,-10.,110);
 411
 412     fOutputCheckCentrality->Add(hNtrackletsIn);
 413     fOutputCheckCentrality->Add(hNtrackletsOut);
 414     fOutputCheckCentrality->Add(hMultIn);
 415     fOutputCheckCentrality->Add(hMultOut);
 416     fOutputCheckCentrality->Add(hMultvsPercentile);
 417
 418     PostData(6,fOutputCheckCentrality);
 419
 420   } else{
 421     if(fOnOff[0]){
 422       TString hname="hNtracklets";
 423       TH1F* hNtracklets=new TH1F(hname.Data(),"Number of tracklets;ntracklets;Entries",5000,-0.5,4999.5);
 424
 425       hname="hMult";
 426       TH1F* hMult=new TH1F(hname.Data(),"Multiplicity;multiplicity;Entries",10000,-0.5,9999.5);
 427       fOutputTrack->Add(hNtracklets);
 428       fOutputTrack->Add(hMult);
 429     }
 430   }
 431
 432   // Post the data
 433   PostData(1,fNEntries);
 434   if(fOnOff[1]) PostData(2,fOutputPID);
 435   if(fOnOff[0]) PostData(3,fOutputTrack);
 436   PostData(4,fCuts);
 437   if(fOnOff[2]) PostData(5,fOutputCounters);
 438
 439   if(!fOnOff[0] && !fOnOff[1] && !fOnOff[2]) AliError("Nothing will be filled!");
 440 }
 441
 442 //___________________________________________________________________________
 443 void AliAnalysisTaskSEHFQA::UserExec(Option_t */*option*/)
 444 {
 445   // Execute analysis for current event
 446
 447   AliAODEvent *aod = dynamic_cast<AliAODEvent*> (InputEvent());
 448   if(fDebug>2) printf("Analysing decay %d\n",fDecayChannel);
 449   // Post the data already here
 450   PostData(1,fNEntries);
 451   if(fOnOff[1]) PostData(2,fOutputPID);
 452   if(fOnOff[0]) PostData(3,fOutputTrack);
 453   PostData(4,fCuts);
 454   if(fOnOff[2]) {
 455     PostData(5,fOutputCounters);
 456     if(fCuts->GetUseCentrality()) PostData(6,fOutputCheckCentrality);
 457   }
 458
 459   TClonesArray *arrayProng =0;
 460   Int_t pdg=0;
 461   Int_t *pdgdaughters=0x0;
 462
 463   if(!aod && AODEvent() && IsStandardAOD()) {
 464     // In case there is an AOD handler writing a standard AOD, use the AOD
 465     // event in memory rather than the input (ESD) event.
 466     aod = dynamic_cast<AliAODEvent*> (AODEvent());
 467     // in this case the braches in the deltaAOD (AliAOD.VertexingHF.root)
 468     // have to taken from the AOD event hold by the AliAODExtension
 469     AliAODHandler* aodHandler = (AliAODHandler*)
 470       ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
 471     if(aodHandler->GetExtensions()) {
 472
 473       AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject("AliAOD.VertexingHF.root");
 474       AliAODEvent *aodFromExt = ext->GetAOD();
 475
 476
 477
 478       switch(fDecayChannel){
 479       case 0:
 480         arrayProng=(TClonesArray*)aodFromExt->GetList()->FindObject("Charm3Prong");
 481         pdg=411;
 482         if(fReadMC){
 483           pdgdaughters =new Int_t[3];
 484           pdgdaughters[0]=211;//pi
 485           pdgdaughters[1]=321;//K
 486           pdgdaughters[2]=211;//pi
 487         }
 488         break;
 489       case 1:
 490         arrayProng=(TClonesArray*)aodFromExt->GetList()->FindObject("D0toKpi");
 491         pdg=421;
 492         if(fReadMC){
 493           pdgdaughters =new Int_t[2];
 494           pdgdaughters[0]=211;//pi
 495           pdgdaughters[1]=321;//K
 496         }
 497         break;
 498       case 2:
 499         arrayProng=(TClonesArray*)aodFromExt->GetList()->FindObject("Dstar");
 500         pdg=413;
 501         if(fReadMC){
 502           pdgdaughters =new Int_t[3];
 503           pdgdaughters[1]=211;//pi
 504           pdgdaughters[0]=321;//K
 505           pdgdaughters[2]=211;//pi (soft?)
 506         }
 507         break;
 508       case 3:
 509         arrayProng=(TClonesArray*)aodFromExt->GetList()->FindObject("Charm3Prong");
 510         pdg=431;
 511         if(fReadMC){
 512           pdgdaughters =new Int_t[3];
 513           pdgdaughters[0]=321;//K
 514           pdgdaughters[1]=321;//K
 515           pdgdaughters[2]=211;//pi
 516         }
 517         break;
 518       case 4:
 519         arrayProng=(TClonesArray*)aodFromExt->GetList()->FindObject("Charm4Prong");
 520         pdg=421;
 521         if(fReadMC){
 522           pdgdaughters =new Int_t[4];
 523           pdgdaughters[0]=321;
 524           pdgdaughters[1]=211;
 525           pdgdaughters[2]=211;
 526           pdgdaughters[3]=211;
 527         }
 528         break;
 529       case 5:
 530         arrayProng=(TClonesArray*)aodFromExt->GetList()->FindObject("Charm3Prong");
 531         pdg=4122;
 532         if(fReadMC){
 533           pdgdaughters =new Int_t[3];
 534           pdgdaughters[0]=2212;//p
 535           pdgdaughters[1]=321;//K
 536           pdgdaughters[2]=211;//pi
 537         }
 538         break;
 539       }
 540     }
 541   } else if(aod) {
 542     switch(fDecayChannel){
 543     case 0:
 544       arrayProng=(TClonesArray*)aod->GetList()->FindObject("Charm3Prong");
 545       pdg=411;
 546       if(fReadMC){
 547         pdgdaughters =new Int_t[3];
 548         pdgdaughters[0]=211;//pi
 549         pdgdaughters[1]=321;//K
 550         pdgdaughters[2]=211;//pi
 551       }
 552       break;
 553     case 1:
 554       arrayProng=(TClonesArray*)aod->GetList()->FindObject("D0toKpi");
 555       pdg=421;
 556       if(fReadMC){
 557         pdgdaughters =new Int_t[2];
 558         pdgdaughters[0]=211;//pi
 559         pdgdaughters[1]=321;//K
 560       }
 561       break;
 562     case 2:
 563       arrayProng=(TClonesArray*)aod->GetList()->FindObject("Dstar");
 564       pdg=413;
 565       if(fReadMC){
 566         pdgdaughters =new Int_t[3];
 567         pdgdaughters[1]=211;//pi
 568         pdgdaughters[0]=321;//K
 569         pdgdaughters[2]=211;//pi (soft?)
 570       }
 571       break;
 572     case 3:
 573       arrayProng=(TClonesArray*)aod->GetList()->FindObject("Charm3Prong");
 574       pdg=431;
 575       if(fReadMC){
 576         pdgdaughters =new Int_t[3];
 577         pdgdaughters[0]=321;//K
 578         pdgdaughters[1]=321;//K
 579         pdgdaughters[2]=211;//pi
 580       }
 581       break;
 582     case 4:
 583       arrayProng=(TClonesArray*)aod->GetList()->FindObject("Charm4Prong");
 584       pdg=421;
 585       if(fReadMC){
 586         pdgdaughters =new Int_t[4];
 587         pdgdaughters[0]=321;
 588         pdgdaughters[1]=211;
 589         pdgdaughters[2]=211;
 590         pdgdaughters[3]=211;
 591       }
 592       break;
 593     case 5:
 594       arrayProng=(TClonesArray*)aod->GetList()->FindObject("Charm3Prong");
 595       pdg=4122;
 596       if(fReadMC){
 597         pdgdaughters =new Int_t[3];
 598         pdgdaughters[0]=2212;//p
 599         pdgdaughters[1]=321;//K
 600         pdgdaughters[2]=211;//pi
 601       }
 602       break;
 603     }
 604   }
 605   Bool_t isSimpleMode=fSimpleMode;
 606   if(!arrayProng) {
 607     AliInfo("Branch not found! The output will contain only trak related histograms\n");
 608     isSimpleMode=kTRUE;
 609     fNEntries->Fill(2);
 610   }
 611
 612   TClonesArray *mcArray = 0;
 613   AliAODMCHeader *mcHeader = 0;
 614
 615   //check if MC
 616   if(fReadMC) {
 617     // load MC particles
 618     mcArray = (TClonesArray*)aod->GetList()->FindObject(AliAODMCParticle::StdBranchName());
 619     if(!mcArray) {
 620       printf("AliAnalysisTaskSEHFQA::UserExec: MC particles branch not found!\n");
 621       delete [] pdgdaughters;
 622       return;
 623     }
 624
 625     // load MC header
 626     mcHeader = (AliAODMCHeader*)aod->GetList()->FindObject(AliAODMCHeader::StdBranchName());
 627     if(!mcHeader) {
 628       printf("AliAnalysisTaskSEHFQA::UserExec: MC header branch not found!\n");
 629       delete [] pdgdaughters;
 630       return;
 631     }
 632   }
 633   if(!aod) {delete [] pdgdaughters;return;}
 634   // fix for temporary bug in ESDfilter
 635   // the AODs with null vertex pointer didn't pass the PhysSel
 636   if(!aod->GetPrimaryVertex() || TMath::Abs(aod->GetMagneticField())<0.001) return;
 637
 638   // count event
 639   fNEntries->Fill(0);
 640
 641   //count events with good vertex
 642   // AOD primary vertex
 643   AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
 644   TString primTitle = vtx1->GetTitle();
 645   if(primTitle.Contains("VertexerTracks") && vtx1->GetNContributors()>0) fNEntries->Fill(4);
 646
 647   // trigger class for PbPb C0SMH-B-NOPF-ALLNOTRD, C0SMH-B-NOPF-ALL
 648   //TString trigclass=aod->GetFiredTriggerClasses();
 649   //if(trigclass.Contains("C0SMH-B-NOPF-ALLNOTRD") || trigclass.Contains("C0SMH-B-NOPF-ALL")) fNEntries->Fill(5); //tmp
 650
 651   Bool_t evSelbyCentrality=kTRUE,evSelected=kTRUE,evSelByVertex=kTRUE,evselByPileup=kFALSE;
 652   //select event
 653   if(!fCuts->IsEventSelected(aod)) {
 654     evSelected=kFALSE;
 655     if(fCuts->GetWhyRejection()==1) {fNEntries->Fill(1); evselByPileup=kTRUE;}// rejected for pileup
 656     if(fCuts->GetWhyRejection()==2 || fCuts->GetWhyRejection()==3) evSelbyCentrality=kFALSE; //rejected by centrality
 657     if(fCuts->GetWhyRejection()==4) evSelByVertex=kFALSE; //rejected by vertex
 658     if(fCuts->GetWhyRejection()==5) fNEntries->Fill(5);//tmp
 659   }
 660   if(evSelected || (!evSelected && !evSelbyCentrality && evSelByVertex && !evselByPileup)){ //events selected or not selected because of vtx or pileup
 661     if(fOnOff[2] && fCuts->GetUseCentrality()){
 662       Int_t runNumber = aod->GetRunNumber();
 663       Float_t stdCentf=fCuts->GetCentrality(aod);
 664       Int_t stdCent = (Int_t)(stdCentf+0.5);
 665       Float_t secondCentf =fCuts->GetCentrality(aod,fEstimator);
 666       Int_t secondCent = (Int_t)(secondCentf+0.5);
 667       Int_t mincent=stdCent-stdCent%10;
 668       if(stdCentf==-1) {
 669         mincent=-10;
 670         stdCent=-1;
 671       }
 672       ((AliCounterCollection*)fOutputCounters->FindObject("stdEstimator"))->Count(Form("centralityclass:%d_%d/Run:%d",mincent,mincent+10,runNumber));
 673
 674       mincent=secondCent-secondCent%10;
 675       if(secondCentf==-1) {
 676         mincent=-10;
 677         secondCent=-1;
 678       }
 679       ((AliCounterCollection*)fOutputCounters->FindObject("secondEstimator"))->Count(Form("centralityclass:%d_%d/Run:%d",mincent,mincent+10,runNumber));
 680
 681       if(stdCent<fCuts->GetMinCentrality() || stdCent>fCuts->GetMaxCentrality()){
 682         ((TH1F*)fOutputCheckCentrality->FindObject("hNtrackletsOut"))->Fill(aod->GetTracklets()->GetNumberOfTracklets());
 683         ((TH1F*)fOutputCheckCentrality->FindObject("hMultOut"))->Fill(aod->GetHeader()->GetRefMultiplicity());
 684       }else{
 685         ((TH1F*)fOutputCheckCentrality->FindObject("hNtrackletsIn"))->Fill(aod->GetTracklets()->GetNumberOfTracklets());
 686         ((TH1F*)fOutputCheckCentrality->FindObject("hMultIn"))->Fill(aod->GetHeader()->GetRefMultiplicity());
 687       }
 688       ((TH2F*)fOutputCheckCentrality->FindObject("hMultvsPercentile"))->Fill(aod->GetHeader()->GetRefMultiplicity(),stdCentf);
 689
 690       PostData(6,fOutputCheckCentrality);
 691
 692     } else{
 693       if(fOnOff[0]){
 694         ((TH1F*)fOutputTrack->FindObject("hNtracklets"))->Fill(aod->GetTracklets()->GetNumberOfTracklets());
 695         ((TH1F*)fOutputTrack->FindObject("hMult"))->Fill(aod->GetHeader()->GetRefMultiplicity());
 696       }
 697     }
 698   }
 699
 700   if(!evSelected) {
 701     delete [] pdgdaughters;
 702     return; //discard all events not selected (vtx and/or centrality)
 703   }
 704
 705   Int_t ntracks=0;
 706   Int_t isGoodTrack=0, isFakeTrack=0;
 707
 708   if(aod) ntracks=aod->GetNTracks();
 709
 710   if(fOnOff[0] || fOnOff[1]){
 711     //loop on tracks in the event
 712     for (Int_t k=0;k<ntracks;k++){
 713       AliAODTrack* track=aod->GetTrack(k);
 714       AliAODPidHF* pidHF=fCuts->GetPidHF();
 715       AliAODPid *pid = track->GetDetPid();
 716
 717
 718       if(fOnOff[1]){
 719         if(!pid)  {if (fDebug>1)cout<<"No AliAODPid found"<<endl; continue;}
 720         Double_t times[AliPID::kSPECIES];
 721         pid->GetIntegratedTimes(times);
 722
 723         Double_t tofRes[AliPID::kSPECIES];
 724         pid->GetTOFpidResolution(tofRes);
 725
 726         //check TOF
 727         if(pidHF && pidHF->CheckStatus(track,"TOF")){
 728           ((TH1F*)fOutputPID->FindObject("hTOFtime"))->Fill(times[AliPID::kProton]);
 729           ((TH2F*)fOutputPID->FindObject("hTOFtimeKaonHyptime"))->Fill(track->P(),pid->GetTOFsignal()-times[3]); //3 is kaon
 730           ((TH1F*)fOutputPID->FindObject("hTOFsig"))->Fill(pid->GetTOFsignal());
 731           if (pid->GetTOFsignal()< 0) ((TH1F*)fOutputPID->FindObject("hTOFsig"))->Fill(-1);
 732
 733
 734           // test TOF sigma PID
 735           if (tofRes[2] != 0.) {   // protection against 'old' AODs...
 736             ((TH2F*)fOutputPID->FindObject("hTOFsigmaKSigPid"))->Fill(track->P(),(pid->GetTOFsignal()-times[AliPID::kKaon])/tofRes[3]);
 737             ((TH2F*)fOutputPID->FindObject("hTOFsigmaPionSigPid"))->Fill(track->P(),(pid->GetTOFsignal()-times[AliPID::kPion])/tofRes[2]);
 738             ((TH2F*)fOutputPID->FindObject("hTOFsigmaProtonSigPid"))->Fill(track->P(),(pid->GetTOFsignal()-times[AliPID::kProton])/tofRes[4]);
 739             for (Int_t iS=2; iS<5; iS++){ //we plot TOF Pid resolution for 3-sigma identified particles
 740               if ( (TMath::Abs(times[iS]-pid->GetTOFsignal())/tofRes[iS])<3.){
 741                 switch (iS) {
 742                 case AliPID::kPion:
 743                   ((TH1F*)fOutputPID->FindObject("hTOFsigPid3sigPion"))->Fill(tofRes[iS]);
 744                   break;
 745                 case AliPID::kKaon:
 746                   ((TH1F*)fOutputPID->FindObject("hTOFsigPid3sigKaon"))->Fill(tofRes[iS]);
 747                   break;
 748                 case AliPID::kProton:
 749                   ((TH1F*)fOutputPID->FindObject("hTOFsigPid3sigProton"))->Fill(tofRes[iS]);
 750                   break;
 751                 default:
 752                   break;
 753                 }
 754               }
 755             }
 756           }
 757         }//if TOF status
 758
 759         if(pidHF && pidHF->CheckStatus(track,"TPC")){
 760           Double_t alephParameters[5];
 761           /*
 762           //this is recommended for LHC10d
 763           alephParameters[0] = 1.34490e+00/50.;
 764           alephParameters[1] =  2.69455e+01;
 765           alephParameters[2] =  TMath::Exp(-2.97552e+01);
 766           alephParameters[3] = 2.35339e+00;
 767           alephParameters[4] = 5.98079e+00;
 768           */
 769
 770           //this is recommended for PbPb (LHC10h)
 771           alephParameters[0] = 1.25202/50.;
 772           alephParameters[1] = 2.74992e+01;
 773           alephParameters[2] = TMath::Exp(-3.31517e+01);
 774           alephParameters[3] = 2.46246;
 775           alephParameters[4] = 6.78938;
 776
 777           /*
 778           //this is recommended for LHC10bc
 779           alephParameters[0] = 0.0283086/0.97;
 780           alephParameters[1] = 2.63394e+01;
 781           alephParameters[2] = 5.04114e-11;
 782           alephParameters[3] = 2.12543e+00;
 783           alephParameters[4] = 4.88663e+00;
 784           */
 785
 786           AliTPCPIDResponse* tpcres=new AliTPCPIDResponse();
 787           tpcres->SetBetheBlochParameters(alephParameters[0],alephParameters[1],alephParameters[2],alephParameters[3],alephParameters[4]);
 788           Double_t TPCp=pid->GetTPCmomentum();
 789           Double_t TPCsignal=pid->GetTPCsignal();
 790           ((TH1F*)fOutputPID->FindObject("hTPCsig"))->Fill(TPCsignal);
 791           ((TH1F*)fOutputPID->FindObject("hTPCsigvsp"))->Fill(TPCp,TPCsignal);
 792           //if (pidHF->IsKaonRaw(track, "TOF"))
 793           ((TH2F*)fOutputPID->FindObject("hTPCsigmaK"))->Fill(TPCp,tpcres->GetNumberOfSigmas(TPCp,TPCsignal,track->GetTPCNcls(),AliPID::kKaon));
 794           //if (pidHF->IsPionRaw(track, "TOF"))
 795           ((TH2F*)fOutputPID->FindObject("hTPCsigmaPion"))->Fill(TPCp,tpcres->GetNumberOfSigmas(TPCp,TPCsignal,track->GetTPCNcls(),AliPID::kPion));
 796           //if (pidHF->IsProtonRaw(track,"TOF"))
 797           ((TH2F*)fOutputPID->FindObject("hTPCsigmaProton"))->Fill(TPCp,tpcres->GetNumberOfSigmas(TPCp,TPCsignal,track->GetTPCNcls(),AliPID::kProton));
 798           delete tpcres;
 799         }//if TPC status
 800       } //end PID histograms
 801
 802       Int_t nclsTot=0,nclsSPD=0;
 803
 804       //check clusters of the tracks
 805       if(fOnOff[0]){
 806
 807         for(Int_t l=0;l<6;l++) {
 808           if(TESTBIT(track->GetITSClusterMap(),l)) {
 809             nclsTot++; if(l<2) nclsSPD++;
 810           }
 811         }
 812         ((TH1F*)fOutputTrack->FindObject("hnClsITS"))->Fill(nclsTot);
 813         ((TH1F*)fOutputTrack->FindObject("hnClsSPD"))->Fill(nclsSPD);
 814
 815         if(!(track->GetStatus()&AliESDtrack::kTPCin) && track->GetStatus()&AliESDtrack::kITSrefit && !(track->GetStatus()&AliESDtrack::kITSpureSA)){//tracks retrieved in the ITS and not reconstructed in the TPC
 816           ((TH1F*)fOutputTrack->FindObject("hnClsITS-SA"))->Fill(nclsTot);
 817         }
 818         Int_t label=0;
 819         if(fReadMC){
 820           label=track->GetLabel();
 821           if (label<0)fNEntries->Fill(8);
 822           else fNEntries->Fill(9);
 823         }
 824
 825         if(isSimpleMode){
 826
 827           if (track->Pt()>0.3 &&
 828               track->GetStatus()&AliESDtrack::kTPCrefit &&
 829               track->GetStatus()&AliESDtrack::kITSrefit &&
 830               /*nclsTot>3 &&*/
 831               nclsSPD>0) {//count good tracks
 832
 833
 834             if(fReadMC && label<0) {
 835               ((TH1F*)fOutputTrack->FindObject("hptFakeTr"))->Fill(track->Pt());
 836               isFakeTrack++;
 837             } else {
 838               ((TH1F*)fOutputTrack->FindObject("hptGoodTr"))->Fill(track->Pt());
 839               isGoodTrack++;
 840             }
 841           }//simple mode: no IsSelected on tracks: use "manual" cuts
 842         }
 843       }
 844     } //end loop on tracks
 845
 846       //fill once per event
 847     if(fOnOff[0]){
 848       if (fReadMC) ((TH1F*)fOutputTrack->FindObject("hdistrFakeTr"))->Fill(isFakeTrack);
 849       ((TH1F*)fOutputTrack->FindObject("hdistrGoodTr"))->Fill(isGoodTrack);
 850     }
 851
 852     if(!isSimpleMode){
 853       // loop over candidates
 854       Int_t nCand = arrayProng->GetEntriesFast();
 855       Int_t ndaugh=3;
 856       if(fDecayChannel==AliAnalysisTaskSEHFQA::kD0toKpi) ndaugh=2;
 857       if(fDecayChannel==AliAnalysisTaskSEHFQA::kD0toKpipipi) ndaugh=4;
 858
 859       for (Int_t iCand = 0; iCand < nCand; iCand++) {
 860         AliAODRecoDecayHF *d = (AliAODRecoDecayHF*)arrayProng->UncheckedAt(iCand);
 861         if(d->GetSelectionMap()) {
 862           if(fDecayChannel==AliAnalysisTaskSEHFQA::kD0toKpi && !d->HasSelectionBit(AliRDHFCuts::kD0toKpiCuts)) continue; //skip the D0 from Dstar
 863           if(fDecayChannel==AliAnalysisTaskSEHFQA::kDplustoKpipi && !d->HasSelectionBit(AliRDHFCuts::kDplusCuts)) continue; //skip the 3 prong !D+
 864         }
 865
 866         if(fReadMC){
 867           Int_t labD = d->MatchToMC(pdg,mcArray,ndaugh,pdgdaughters);
 868           if(labD>=0){
 869             AliAODMCParticle *partD = (AliAODMCParticle*)mcArray->At(labD);
 870             Int_t label=partD->GetMother();
 871             AliAODMCParticle *mot = (AliAODMCParticle*)mcArray->At(label);
 872             while(label>=0){//get first mother
 873               mot = (AliAODMCParticle*)mcArray->At(label);
 874               label=mot->GetMother();
 875             }
 876             Int_t pdgMotCode = mot->GetPdgCode();
 877
 878             if(TMath::Abs(pdgMotCode)==4) fNEntries->Fill(6); //from primary charm
 879             if(TMath::Abs(pdgMotCode)==5) fNEntries->Fill(7); //from beauty
 880
 881           }
 882         }//end MC
 883         else fNEntries->Fill(6); //count the candidates (data)
 884
 885         for(Int_t id=0;id<ndaugh;id++){
 886
 887           //other histograms to be filled when the cut object is given
 888           AliAODTrack* track=(AliAODTrack*)d->GetDaughter(id);
 889
 890           //track quality
 891
 892           if (fCuts->IsInFiducialAcceptance(d->Pt(),d->Y(pdg)) && fCuts->IsSelected(d,AliRDHFCuts::kTracks,aod)) {
 893
 894             Int_t label=0;
 895             if(fReadMC)label=track->GetLabel();
 896             if(fReadMC && label<0 && fOnOff[0]) {
 897               isFakeTrack++;
 898               ((TH1F*)fOutputTrack->FindObject("hptFakeTr"))->Fill(track->Pt());
 899
 900               ((TH1F*)fOutputTrack->FindObject("hd0f"))->Fill(d->Getd0Prong(id));
 901             } else {
 902               isGoodTrack++;
 903               ((TH1F*)fOutputTrack->FindObject("hptGoodTr"))->Fill(track->Pt());
 904
 905               ((TH1F*)fOutputTrack->FindObject("hd0"))->Fill(d->Getd0Prong(id));
 906             }
 907
 908             if (fCuts->IsSelected(d,AliRDHFCuts::kAll,aod) && fOnOff[1]){
 909
 910               AliAODPid *pid = track->GetDetPid();
 911               AliAODPidHF* pidHF=fCuts->GetPidHF();
 912               Double_t times[5];
 913               pid->GetIntegratedTimes(times);
 914               if(pidHF && pidHF->CheckStatus(track,"TOF")) ((TH2F*)fOutputPID->FindObject("hTOFtimeKaonHyptimeAC"))->Fill(track->P(),pid->GetTOFsignal()-times[AliPID::kKaon]);
 915               if(pidHF && pidHF->CheckStatus(track,"TPC")) ((TH2F*)fOutputPID->FindObject("hTPCsigvspAC"))->Fill(pid->GetTPCmomentum(),pid->GetTPCsignal());
 916
 917               fNEntries->Fill(3);
 918             } //end analysis cuts
 919           } //end acceptance and track cuts
 920         } //end loop on tracks in the candidate
 921       } //end loop on candidates
 922       if(fOnOff[0]){
 923         if(fReadMC) ((TH1F*)fOutputTrack->FindObject("hdistrFakeTr"))->Fill(isFakeTrack);
 924         ((TH1F*)fOutputTrack->FindObject("hdistrGoodTr"))->Fill(isGoodTrack);
 925       }
 926     }
 927   } //end if on pid or track histograms
 928
 929   delete [] pdgdaughters;
 930   PostData(1,fNEntries);
 931   if(fOnOff[1]) PostData(2,fOutputPID);
 932   if(fOnOff[0]) PostData(3,fOutputTrack);
 933   PostData(4,fCuts);
 934   if(fOnOff[2]) PostData(5,fOutputCounters);
 935   //Post data 6 done in case of centrality on
 936
 937 }
 938
 939 //____________________________________________________________________________
 940 void AliAnalysisTaskSEHFQA::Terminate(Option_t */*option*/){
 941   //terminate analysis
 942
 943   fNEntries = dynamic_cast<TH1F*>(GetOutputData(1));
 944   if(!fNEntries){
 945     printf("ERROR: %s not available\n",GetOutputSlot(1)->GetContainer()->GetName());
 946     return;
 947   }
 948
 949   fOutputPID = dynamic_cast<TList*> (GetOutputData(2));
 950   if (!fOutputPID && fOnOff[1]) {
 951     printf("ERROR: %s not available\n",GetOutputSlot(2)->GetContainer()->GetName());
 952     return;
 953   }
 954
 955   fOutputTrack = dynamic_cast<TList*> (GetOutputData(3));
 956   if (!fOutputTrack && fOnOff[0]) {
 957     printf("ERROR: %s not available\n",GetOutputSlot(3)->GetContainer()->GetName());
 958     return;
 959   }
 960
 961 }
 962