AliAnalysisTaskSEDs.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKDS_H
#define ALIANALYSISTASKDS_H

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

/* $Id$ */


#include <TROOT.h>
#include <TSystem.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TH3F.h>
#include <THnSparse.h>

#include "AliAnalysisTaskSE.h"
#include "AliRDHFCutsDstoKKpi.h"
#include "AliLog.h"

class AliNormalizationCounter;


class AliAnalysisTaskSEDs : public AliAnalysisTaskSE
{
 public:
    
  enum ECentrality {kCentOff,kCentV0M,kCentV0A,kCentCL1,kCentZNA,kCentInvalid};

  AliAnalysisTaskSEDs();
  AliAnalysisTaskSEDs(const char *name, AliRDHFCutsDstoKKpi* analysiscuts, Int_t fillNtuple=0);
  virtual ~AliAnalysisTaskSEDs();
  void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
  void SetWriteOnlySignalInNtuple(Bool_t opt=kTRUE){
    if(fReadMC) fWriteOnlySignal=opt;
    else AliError("fReadMC has to be kTRUE");
  }
  void SetFillNtuple(Int_t fill=0){fFillNtuple=fill;}
  void SetFillNSparse(Bool_t fill=kTRUE){fFillSparse=fill;}
  void SetFillNSparseDplus(Bool_t fill=kTRUE){fFillSparseDplus=fill;if(fill)fFillSparse=fill;}
  void SetFillNSparseImpPar(Bool_t fill=kTRUE){fFillImpParSparse=fill;}
  void SetFillNSparseAcceptanceLevel(Bool_t fill=kTRUE){fFillAcceptanceLevel=fill;}
  void SetMassRange(Double_t rang=0.4){fMassRange=rang;}
  void SetDoCutVarHistos(Bool_t opt=kTRUE) {fDoCutVarHistos=opt;}
  void SetUseSelectionBit(Bool_t opt=kFALSE){ fUseSelectionBit=opt;}
  void SetAODMismatchProtection(Int_t opt=1) {fAODProtection=opt;}
  void SetUseRotBkg(Bool_t flag=kFALSE) {fDoRotBkg=flag;}
  void SetUseBkgFromPhiSB(Bool_t flag=kFALSE) {fDoBkgPhiSB=flag;}
  void SetPhiMassRange4RotBkg(Double_t range) {fMaxDeltaPhiMass4Rot=range;}
  void SetUseCutV0multVsTPCout(Bool_t flag) {fDoCutV0multTPCout=flag;}
  void SetFillTracklets(Bool_t flag) {fUseTrkl=flag;}
  void SetFillCentralityAxis(Int_t flag=0);    
  Bool_t CheckDaugAcc(TClonesArray* arrayMC,Int_t nProng, Int_t *labDau);
  Bool_t GetUseWeight() const {return fUseWeight;}
  void FillMCGenAccHistos(TClonesArray *arrayMC, AliAODMCHeader *mcHeader, Double_t nTracklets);
  void GenerateRotBkg(AliAODRecoDecayHF3Prong *d, Int_t dec, Int_t iPtBin);
  void CreateCutVarsAndEffSparses();
  void CreateImpactParameterSparses();
  void CreateIPSparse();
  Float_t GetTrueImpactParameterDstoPhiPi(const AliAODMCHeader *mcHeader, TClonesArray* arrayMC, const AliAODMCParticle *partDs) const;
  
  void SetPtWeightsFromFONLL5anddataoverLHC16i2a();
  void SetPtWeightsFromFONLL5overLHC16i2abc();
  void SetPtWeightsFromFONLL5andBAMPSoverLHC16i2abc();
  void SetPtWeightsFromFONLL5andTAMUoverLHC16i2abc();
  void SetPtWeightsFromFONLL13overLHC17c3a12();
  void SetPtWeightsFromFONLL5overLHC18a4a2();

  void SetInvMassBinSize(Double_t binsiz=0.002){fMassBinSize=binsiz;}
  void SetPtBins(Int_t n, Float_t* lim);
  void SetAnalysisCuts(AliRDHFCutsDstoKKpi* cuts){fAnalysisCuts=cuts;}
  void SetSystem(Int_t system){fSystem = system;}

  void SetMultSelectionObjectName(TString str){fMultSelectionObjectName=str;}

  Double_t GetPtWeightFromHistogram(Double_t pt);
    
  virtual void UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);
    
 private:
  Int_t GetHistoIndex(Int_t iPtBin) const { return iPtBin*4;}
  Int_t GetSignalHistoIndex(Int_t iPtBin) const { return iPtBin*4+1;}
  Int_t GetBackgroundHistoIndex(Int_t iPtBin) const { return iPtBin*4+2;}
  Int_t GetReflSignalHistoIndex(Int_t iPtBin) const { return iPtBin*4+3;}

  enum {kMaxPtBins=24,knVarForSparse=15,knVarForSparseAcc=3,knVarForSparseIP=6,kVarForImpPar=3};
    
  AliAnalysisTaskSEDs(const AliAnalysisTaskSEDs &source);
  AliAnalysisTaskSEDs& operator=(const AliAnalysisTaskSEDs& source);
    
  TList*  fOutput;                    
  TH1F*   fHistNEvents;               
  TH1F*   fHistoPtWeight;             //-> user-defined histogram to calculate the Pt weights
  TH1F*   fChanHist[4];               
  TH1F*   fMassHist[4*kMaxPtBins];    
  TH1F*   fMassHistPhi[4*kMaxPtBins];     
  TH1F*   fMassHistK0st[4*kMaxPtBins];    
  TH1F*   fMassHistKK[kMaxPtBins];        
  TH1F*   fMassHistKpi[kMaxPtBins];       
  TH1F*   fMassRotBkgHistPhi[kMaxPtBins]; 
  TH1F*   fMassLSBkgHistPhi[kMaxPtBins];  
  TH1F*   fMassRSBkgHistPhi[kMaxPtBins];  
  TH1F*   fCosPHist[4*kMaxPtBins];        
  TH1F*   fCosPxyHist[4*kMaxPtBins];      
  TH1F*   fDLenHist[4*kMaxPtBins];        
  TH1F*   fDLenxyHist[4*kMaxPtBins];      
  TH1F*   fNDLenxyHist[4*kMaxPtBins];     
  TH1F*   fSumd02Hist[4*kMaxPtBins];      
  TH1F*   fSigVertHist[4*kMaxPtBins];     
  TH1F*   fPtMaxHist[4*kMaxPtBins];       
  TH1F*   fPtCandHist[4*kMaxPtBins];      
  TH1F*   fDCAHist[4*kMaxPtBins];         
  TH1F*   fNormIPHist[4*kMaxPtBins];      
  TH1F*   fCosPiDsHist[4*kMaxPtBins];     
  TH1F*   fCosPiKPhiHist[4*kMaxPtBins];   
  TH1F*   fPtProng0Hist[4*kMaxPtBins];    
  TH1F*   fPtProng1Hist[4*kMaxPtBins];    
  TH1F*   fPtProng2Hist[4*kMaxPtBins];    
  TH2F*   fDalitz[4*kMaxPtBins];          
  TH2F*   fDalitzPhi[4*kMaxPtBins];       
  TH2F*   fDalitzK0st[4*kMaxPtBins];      
  TH2F *fPtVsMass;       
  TH2F *fPtVsMassPhi;    
  TH2F *fPtVsMassK0st;   
  TH2F *fYVsPt;          
  TH2F *fYVsPtSig;       
  TH2F *fHistAllV0multNTPCout;  
  TH2F *fHistSelV0multNTPCout;  
  TH1F *fHistCentrality[3];     
  TH2F *fHistCentralityMult[3]; 
  TH3F*   fCosPHist3D;       
  TH3F*   fCosPxyHist3D;     
  TH3F*   fDLenHist3D;       
  TH3F*   fDLenxyHist3D;     
  TH3F*   fNDLenxyHist3D;    
  TH3F*   fSigVertHist3D;    
  TH3F*   fDCAHist3D;        
  TH3F*   fNormIPHist3D;     
  TH3F*   fCosPiDsHist3D;    
  TH3F*   fCosPiKPhiHist3D;  
  TH3F*   fPtProng0Hist3D;   
  TH3F*   fPtProng1Hist3D;   
  TH3F*   fPtProng2Hist3D;   
  TNtuple *fNtupleDs; 
  Int_t fFillNtuple;                 
  Int_t   fUseCentrAxis; 

  Int_t   fSystem;                    
  Bool_t  fReadMC;                    
  Bool_t  fWriteOnlySignal;           
  Bool_t  fDoCutVarHistos;            
  Bool_t  fUseSelectionBit;           
  Bool_t  fFillSparse;                
  Bool_t  fFillSparseDplus;           
  Bool_t  fFillImpParSparse;          
  Bool_t  fFillAcceptanceLevel;       
  Bool_t fDoRotBkg;                   
  Bool_t fDoBkgPhiSB;                 
  Bool_t fDoCutV0multTPCout;          
  Bool_t fUseWeight;                  
  Bool_t fUseTrkl;                    
  Int_t fAODProtection;               
  UChar_t fNPtBins;                   
  TList *fListCuts; //list of cuts
  Float_t fPtLimits[kMaxPtBins+1];    
  Double_t fMassRange;                
  Double_t fMassBinSize;              
  Double_t fminMass;
  Double_t fmaxMass;
  Double_t fMaxDeltaPhiMass4Rot;     
    
    
  AliNormalizationCounter *fCounter;
  AliRDHFCutsDstoKKpi *fAnalysisCuts; 
    
  THnSparseF *fnSparse;       
  THnSparseF *fnSparseIP;       
  THnSparseF *fnSparseMC[4];  
  THnSparseF *fnSparseMCDplus[4];  
  THnSparseF *fImpParSparse;       
  THnSparseF *fImpParSparseMC[4];     
   
  TString fMultSelectionObjectName; 
  TString fCentEstName; 
    
  ClassDef(AliAnalysisTaskSEDs,27);    
};

#endif