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