AliAnalysisTaskDmesonMCPerform.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKDMESONMCPERFORM_H
#define ALIANALYSISTASKDMESONMCPERFORM_H

/* Copyright(c) 1998-2018, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//*************************************************************************
//*************************************************************************

#include "AliAnalysisTaskSE.h"
#include "AliRDHFCutsD0toKpi.h"
#include "AliRDHFCutsDplustoKpipi.h"
#include "TH3F.h"
#include "TH1F.h"

class AliAnalysisTaskDmesonMCPerform : public AliAnalysisTaskSE
{
 public:
  AliAnalysisTaskDmesonMCPerform();
  virtual ~AliAnalysisTaskDmesonMCPerform();
  
  void SetAODMismatchProtection(Int_t opt=1) {fAODProtection=opt;}
  void SetDplusAnalysisCuts(AliRDHFCutsDplustoKpipi* cts){
    fRDHFCutsDplus=cts;
  }
  void SetUseCentrality(Int_t flag){
    fRDHFCuts->SetUseCentrality(flag);
  }
  void SetPtBinning(Int_t nbins, Double_t minpt, Double_t maxpt){
    fNPtBins=nbins; fMinPt=minpt; fMaxPt=maxpt;
  }
  void SetFillExtraHistos(Bool_t opt){
    fEnableExtraHistos=opt;
  }
  virtual void UserCreateOutputObjects();
  virtual void Init(){};
  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);

  AliRDHFCuts* GetCutObject(){return fRDHFCuts;}

 private:

  void FillGenLevelHistos(AliAODEvent* aod, TClonesArray *arrayMC, AliAODMCHeader *mcHeader);
  void FillCandLevelHistos(Int_t idCase, AliAODEvent* aod, TClonesArray *arrayDcand, TClonesArray *arrayMC);

  Bool_t CheckAcceptance(TClonesArray* arrayMC,Int_t nProng, Int_t *labDau);
  AliAODRecoDecayHF* GetRecoDecay(AliAODEvent* aod, Int_t nProng, Int_t *labDau);
  void MapTrackLabels(AliAODEvent* aod);

  AliAnalysisTaskDmesonMCPerform(const AliAnalysisTaskDmesonMCPerform &source);
  AliAnalysisTaskDmesonMCPerform& operator=(const AliAnalysisTaskDmesonMCPerform &source);
  enum {kDecays=6, kMaxLabel=1000000};

  TList* fOutput;                   
  TH1F* fHistNEvents;               
  TH1F* fHistNGenD;                 
  TH2F* fHistNCand;                 
  TH3F* fHistPtYMultGen[2*kDecays];         
  TH3F* fHistPtYMultGenDauInAcc[2*kDecays]; 
  TH3F* fHistPtYMultReco[2*kDecays];        
  TH3F* fHistPtYMultRecoFilt[2*kDecays];    
  TH3F* fHistPtYMultRecoSel[2*kDecays];     

  TH2F* fHistXvtxResVsPt[2*kDecays];    
  TH2F* fHistYvtxResVsPt[2*kDecays];    
  TH2F* fHistZvtxResVsPt[2*kDecays];    
  TH2F* fHistXvtxResRotVsPt[2*kDecays]; 
  TH2F* fHistYvtxResRotVsPt[2*kDecays]; 
  TH2F* fHistXvtxResVsPhi[2*kDecays];   
  TH2F* fHistYvtxResVsPhi[2*kDecays];   
  TH2F* fHistZvtxResVsPhi[2*kDecays];   
  TH3F* fHistXvtxResVsDecLenVsPt[2*kDecays];   
  TH3F* fHistYvtxResVsDecLenVsPt[2*kDecays];   
  TH3F* fHistZvtxResVsDecLenVsPt[2*kDecays];   
  TH2F* fHistInvMassVsPt[2*kDecays];    
  TH2F* fHistDecLenVsPt[2*kDecays];     
  TH2F* fHistNormDLxyVsPt[2*kDecays];     
  TH2F* fHistCosPointVsPt[2*kDecays];   

  Int_t fNPtBins;               
  Double_t fMinPt;              
  Double_t fMaxPt;              
  Bool_t   fEnableExtraHistos;  

  Int_t fAODProtection;         

  AliRDHFCutsD0toKpi *fRDHFCuts;             
  AliRDHFCutsDplustoKpipi *fRDHFCutsDplus;   
  TString fPartName[kDecays];                
  Int_t fMapTrLabel[kMaxLabel];              

  ClassDef(AliAnalysisTaskDmesonMCPerform,4);
};
#endif