AliAnalysisTaskSEDvsMultiplicity.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKSEDVSMULTIPLICITY_H
#define ALIANALYSISTASKSEDVSMULTIPLICITY_H

/* Copyright(c) 1998-2008, 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 <TArrayD.h>
#include <TFile.h>
#include <TRandom.h>
#include <TProfile.h>
#include "AliRDHFCutsDplustoKpipi.h"
#include "AliAnalysisTaskSE.h"
#include "AliAnalysisVertexingHF.h"
#include "AliNormalizationCounter.h"
#include "AliAODMCHeader.h"
#include "AliAODMCParticle.h"
#include "AliVertexingHFUtils.h"
#include "AliVEvent.h"


class AliAnalysisTaskSEDvsMultiplicity : public AliAnalysisTaskSE
{
 public:

  AliAnalysisTaskSEDvsMultiplicity();
  AliAnalysisTaskSEDvsMultiplicity(const char *name, Int_t pdgMeson, AliRDHFCuts* cuts, Bool_t switchPPb);
  virtual ~AliAnalysisTaskSEDvsMultiplicity();


  void SetMassLimits(Double_t lowlimit, Double_t uplimit);
  void SetMassLimits(Int_t pdg, Double_t range);
  Double_t GetUpperMassLimit() const {return fUpmasslimit;}
  Double_t GetLowerMassLimit() const {return fLowmasslimit;}
  void SetNMassBins(Int_t nbins){fNMassBins=nbins;}
  Int_t GetNMassBins() const{return fNMassBins;}
  Bool_t GetSubtractTrackletsFromDaughters() const {return fSubtractTrackletsFromDau;}

  void SetImpactParameterBinning(Int_t nbins, Double_t dmin, Double_t dmax){
    fNImpParBins=nbins;
    fLowerImpPar=dmin;
    fHigherImpPar=dmax;
  }

  void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
  void SetMCOption(Int_t option=0){ fMCOption = option; }
  void SetIsPPbData(Bool_t flag=kTRUE){ 
    fisPPbData=flag;
  }
  void SetUseBit(Bool_t use=kTRUE){fUseBit=use;}
  void SetDoImpactParameterHistos(Bool_t doImp=kTRUE){fDoImpPar=doImp;}
  void SetKeepEstimatorCorrelationPlots(Bool_t use=kTRUE){fKeepCorrPlots=use;}

  void SetMultiplVsZProfileLHC10b(TProfile* hprof){
    if(fMultEstimatorAvg[0]) delete fMultEstimatorAvg[0];
    fMultEstimatorAvg[0]=new TProfile(*hprof);
  }
  void SetMultiplVsZProfileLHC10c(TProfile* hprof){
    if(fMultEstimatorAvg[1]) delete fMultEstimatorAvg[1];
    fMultEstimatorAvg[1]=new TProfile(*hprof);
  }
  void SetMultiplVsZProfileLHC10d(TProfile* hprof){
    if(fMultEstimatorAvg[2]) delete fMultEstimatorAvg[2];
    fMultEstimatorAvg[2]=new TProfile(*hprof);
  }
  void SetMultiplVsZProfileLHC10e(TProfile* hprof){
    if(fMultEstimatorAvg[3]) delete fMultEstimatorAvg[3];
    fMultEstimatorAvg[3]=new TProfile(*hprof);
  }
  
  void SetMultiplVsZProfileLHC13b(TProfile* hprof){
    if(fMultEstimatorAvg[0]) delete fMultEstimatorAvg[0];
    fMultEstimatorAvg[0]=new TProfile(*hprof);
    fYearNumber = 13;
  }
  void SetMultiplVsZProfileLHC13c(TProfile* hprof){
    if(fMultEstimatorAvg[1]) delete fMultEstimatorAvg[1];
    fMultEstimatorAvg[1]=new TProfile(*hprof);
    fYearNumber = 13;
  }
    
  void SetMultiplVsZProfileLHC16qt1stBunch(TProfile* hprof){
    if(fMultEstimatorAvg[0]) delete fMultEstimatorAvg[0];
    fMultEstimatorAvg[0]=new TProfile(*hprof);
    fYearNumber = 16;
  }
  void SetMultiplVsZProfileLHC16qt2ndBunch(TProfile* hprof){
    if(fMultEstimatorAvg[1]) delete fMultEstimatorAvg[1];
    fMultEstimatorAvg[1]=new TProfile(*hprof);
    fYearNumber = 16;
  }
  void SetMultiplVsZProfileLHC16qt3rdBunch(TProfile* hprof){
    if(fMultEstimatorAvg[2]) delete fMultEstimatorAvg[2];
    fMultEstimatorAvg[2]=new TProfile(*hprof);
    fYearNumber = 16;
  }
  void SetMultiplVsZProfileLHC16qt4thBunch(TProfile* hprof){
    if(fMultEstimatorAvg[3]) delete fMultEstimatorAvg[3];
    fMultEstimatorAvg[3]=new TProfile(*hprof);
    fYearNumber = 16;
  }
    
  void SetReferenceMultiplcity(Double_t rmu){fRefMult=rmu;}

  void UseMCNchWeight(Int_t flag) { fUseNchWeight = flag; }
  void SetHistoNchWeight(TH1F *h){
    if(fHistoMCNch) delete fHistoMCNch;
    fHistoMCNch = new TH1F(*h);
  }
  void SetMeasuredNchHisto(TH1F* h){
    if(fHistoMeasNch) delete fHistoMeasNch;
    fHistoMeasNch = new TH1F(*h);
  }

  void SetSubtractTrackletsFromDaughters(Bool_t opt){fSubtractTrackletsFromDau=opt;}
  Int_t CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;

  void SetUseVZEROParameterizedVertexCorr(Int_t flag) { fDoVZER0ParamVertexCorr=flag; }
  Int_t GetUseVZEROParameterizedVertexCorr() { return fDoVZER0ParamVertexCorr; }

  enum { kNtrk10=0, kNtrk10to16=1, kVZERO=2, kNtrk03=3, kNtrk05=4, kVZEROA=5, kVZEROEq=6, kVZEROAEq=7 };
  void SetMultiplicityEstimator(Int_t value){ fMultiplicityEstimator=value; }
  Int_t GetMultiplicityEstimator(){ return fMultiplicityEstimator; }
  enum { kEta10=0, kEta10to16=1, kEtaVZERO=2, kEta03=3, kEta05=5, kEtaVZEROA=5 };
  void SetMCPrimariesEstimator(Int_t value){ fMCPrimariesEstimator=value; }
  Int_t GetMCPrimariesEstimator(){ return fMCPrimariesEstimator; }

  virtual void UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);
    
 private:

  AliAnalysisTaskSEDvsMultiplicity(const AliAnalysisTaskSEDvsMultiplicity &source);
  AliAnalysisTaskSEDvsMultiplicity& operator=(const AliAnalysisTaskSEDvsMultiplicity& source); 

  TProfile* GetEstimatorHistogram(const AliVEvent *event);
  void CreateImpactParameterHistos();
  void CreateMeasuredNchHisto();
  void FillMCMassHistos(TClonesArray *arrayMC, Int_t labD, Int_t countMult,Double_t nchWeight);

  TList  *fOutput; 
  TList  *fListCuts; 
  TList  *fOutputCounters; 
  TList  *fListProfiles; 

  TH1F *fHistNEvents;     

  TH2F* fHistNtrEta16vsNtrEta1EvSel; 
  TH2F* fHistNtrEta05vsNtrEta1EvSel; 
  TH2F* fHistNtrEta03vsNtrEta1EvSel; 
  TH2F* fHistNtrEtaV0AvsNtrEta1EvSel; 
  TH2F* fHistNtrEtaV0MvsNtrEta1EvSel; 
  TH2F* fHistNtrEtaV0AvsV0AEqEvSel;   
  TH2F* fHistNtrEtaV0MvsV0MEqEvSel;   
  TH2F* fHistNtrCorrEta1vsNtrRawEta1EvSel; 
  TH2F* fHistMultCorrvsMultRawEvSel;       
  TH2F* fHistNtrEta16vsNtrEta1EvWithCand; 
  TH2F* fHistNtrEta05vsNtrEta1EvWithCand; 
  TH2F* fHistNtrEta03vsNtrEta1EvWithCand; 
  TH2F* fHistNtrEtaV0AvsNtrEta1EvWithCand; 
  TH2F* fHistNtrEtaV0MvsNtrEta1EvWithCand; 
  TH2F* fHistNtrEtaV0AvsV0AEqEvWithCand;     
  TH2F* fHistNtrEtaV0MvsV0MEqEvWithCand;     
  TH2F* fHistNtrCorrEta1vsNtrRawEta1EvWithCand; 
  TH2F* fHistMultCorrvsMultRawEvWithCand;       
  TH2F* fHistNtrEta16vsNtrEta1EvWithD; 
  TH2F* fHistNtrEta05vsNtrEta1EvWithD; 
  TH2F* fHistNtrEta03vsNtrEta1EvWithD; 
  TH2F* fHistNtrEtaV0AvsNtrEta1EvWithD; 
  TH2F* fHistNtrEtaV0MvsNtrEta1EvWithD; 
  TH2F* fHistNtrEtaV0AvsV0AEqEvWithD;   
  TH2F* fHistNtrEtaV0MvsV0MEqEvWithD;   
  TH2F* fHistNtrCorrEta1vsNtrRawEta1EvWithD; 
  TH2F* fHistMultCorrvsMultRawEvWithD;       

  TH2F* fHistNtrVsZvtx; 
  TH2F* fHistNtrCorrVsZvtx; 

  TH2F* fHistNtrVsNchMC; 
  TH2F* fHistNtrCorrVsNchMC; 
  TH2F* fHistNtrVsNchMCPrimary; 
  TH2F* fHistNtrCorrVsNchMCPrimary; 
  TH2F* fHistNtrVsNchMCPhysicalPrimary; 
  TH2F* fHistNtrCorrVsNchMCPhysicalPrimary; 
  TH1F* fHistGenPrimaryParticlesInelGt0; 
  TH3F* fHistNchMCVsNchMCPrimaryVsNchMCPhysicalPrimary; 
  
  TH1F* fHistNtrUnCorrPSSel; 
  TH1F* fHistNtrUnCorrPSTrigSel; 
  TH1F* fHistNtrUnCorrPSTrigPileUpSel; 
  TH1F* fHistNtrUnCorrPSTrigPileUpVtxSel; 
  TH1F* fHistNtrUnCorrPSTrigPileUpVtxContSel; 
  TH1F* fHistNtrUnCorrPSTrigPileUpVtxRangeSel; 
  TH1F* fHistNtrUnCorrPSTrigPileUpVtxRangeCentrSel; 
  TH1F* fHistNtrUnCorrEvSel; 
  TH1F* fHistNtrUnCorrEvWithCand; 
  TH1F* fHistNtrUnCorrEvWithD;
  TH1F* fHistNtrCorrPSSel; 
  TH1F* fHistNtrCorrEvSel; 
  TH1F* fHistNtrCorrEvWithCand; 
  TH1F* fHistNtrCorrEvWithD;


  TH3F *fPtVsMassVsMult;  
  TH3F *fPtVsMassVsMultNoPid;  
  TH3F *fPtVsMassVsMultUncorr;  
  TH3F *fPtVsMassVsMultPart;  
  TH3F *fPtVsMassVsMultAntiPart;  
  TH3F *fPtVsMassVsMultMC;  

  THnSparseF *fHistMassPtImpPar[5];

  Double_t fUpmasslimit;  
  Double_t fLowmasslimit; 
  Int_t   fNMassBins;    

  AliRDHFCuts *fRDCutsAnalysis; 
  AliNormalizationCounter *fCounterC;           
  AliNormalizationCounter *fCounterU;           
  AliNormalizationCounter *fCounterCandidates;  


  Bool_t fDoImpPar;  
  Int_t  fNImpParBins;   
  Double_t fLowerImpPar;  
  Double_t fHigherImpPar; 

  Bool_t fReadMC;    
  Int_t  fMCOption;  
  Bool_t fisPPbData; 
  Bool_t fUseBit;    
  Bool_t fSubtractTrackletsFromDau; 
  Bool_t fKeepCorrPlots; 
  Int_t fAODProtection;  

  Int_t fUseNchWeight; 
  TH1F* fHistoMCNch;    
  TH1F* fHistoMeasNch;  
  
  TProfile* fMultEstimatorAvg[4]; 
  Double_t fRefMult;   
  Int_t fPdgMeson;   

  Int_t fMultiplicityEstimator; 
  Int_t fMCPrimariesEstimator;  

  Int_t fDoVZER0ParamVertexCorr; 
  
  Int_t fYearNumber; 

  ClassDef(AliAnalysisTaskSEDvsMultiplicity,18); 
};

#endif