AliAnalysisTaskPIDconfig.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKPIDconfig_H
#define ALIANALYSISTASKPIDconfig_H

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

/* $Id: AliAnalysisTaskPIDconfig.h */
// Author: Naghmeh Mohammadi, 10/07/2014

//==============================================================================
//
//
//
//
//==============================================================================

#include <TVectorDfwd.h>

#include "AliAnalysisTaskSE.h"
#include "AliPID.h"
#include "AliPIDResponse.h"
#include "AliCentrality.h"
#include "TCutG.h"
#include "TF2.h"
#include "THnSparse.h"




class AliESDEvent;
class AliAODEvent;
class AliESDTrack;
class AliAODTrack;
class AliPIDResponse;
class TList;
class AliVEvent;
class TH1F;
class TH2F;
class TH3F;
class TH3F;



class AliAnalysisTaskPIDconfig : public AliAnalysisTaskSE {
    
    
public:
    AliAnalysisTaskPIDconfig();
    AliAnalysisTaskPIDconfig(const char *name);
    virtual ~AliAnalysisTaskPIDconfig();
    
    virtual void UserCreateOutputObjects();
    virtual void UserExec(Option_t * /*option*/);
    
    void SetFilterBit(Double_t b){fFilterBit = b;}
    void SetCentralityPercentileMin(Int_t b){fCentralityPercentileMin = b;}
    void SetCentralityPercentileMax(Int_t b){fCentralityPercentileMax = b;}
    void SetCentralityEstimator(TString b){fCentralityEstimator = b;}
    void SetUseCentrality(Bool_t b=kTRUE){fUseCentrality = b;}
    void SetCentralityTrigger(Int_t b=AliVEvent::kMB){ fTriggerSelection = b;}
    void SetDCAxyCut(Int_t b){fDCAxyCut = b;}
    void SetDCAzCut(Int_t b){fDCAzCut = b;}
    void SetCutTPCmultiplicityOutliersAOD(Bool_t b){fCutTPCmultiplicityOutliersAOD = b;}
    void SetData2011(Bool_t b){fData2011 = b;}
    void CheckCentrality(AliVEvent *event,Double_t centrality, Bool_t &centralitypass); //to use only events with the correct centrality....
    void SetCuts(Bool_t b){fPIDcuts = b;}
    //void MultiplicityOutlierCut(AliVEvent *event,Bool_t &centralitypass,Int_t ntracks);
    //void SetPIDcontoursList(TDirectory* b){fCutContourList = b;}
    
    void SetPIDPurityFunctions(Float_t PurityLevel);
    
protected:
    
    
    
private:
    AliVEvent             *fVevent;             
    AliESDEvent           *fESD;                
    AliAODEvent           *fAOD;                
    AliPIDResponse        *fPIDResponse;        
    Int_t                  fTriggerSelection;   // trigger selection
    Int_t                  fCentralityPercentileMin;    // min centrality
    Int_t                  fCentralityPercentileMax;    // max
    Float_t                fPurityLevel;  //purity level
    Double_t               fFilterBit;                  // filterbit
    Double_t               fDCAxyCut;           // dca cut
    Double_t               fDCAzCut;            // dcz z
    Double_t               fLowPtPIDTPCnsigLow_Pion[6]; //nsig low cut Pion
    Double_t               fLowPtPIDTPCnsigHigh_Pion[6]; //nsig high cut Pion
    Double_t               fLowPtPIDTPCnsigLow_Kaon[6]; //nsig low cut Kaon
    Double_t               fLowPtPIDTPCnsigHigh_Kaon[6]; //nsig high cut Kaon
    Double_t               fvalueAll[4];
    Bool_t                 fData2011;           // 2011 data
    Bool_t                 fUseCentrality;      // use centrality
    Bool_t                 fCutTPCmultiplicityOutliersAOD;      // do outlier cut
    Bool_t                 fPIDcuts;            // pid cuts
    TString                fCentralityEstimator;// cen estimator "V0M","TRK","TKL","ZDC","FMD"
    TFile                 *fPurityFunctionsFile;       
    TDirectory            *fPurityFunctionsList;     
    TList                 *fListQA;             
    TList                 *fListQAtpctof;       
    TList                 *fListQAInfo;         
    TH1F                  *fhistCentralityPassBefore; 
    TH1F                  *fhistCentralityPassAfter; 
    TH1F                  *fNoEvents;           
    TH1F                  *fpVtxZ;              
    TH2F                  *fhistDCABefore;      
    TH2F                  *fhistDCAAfter;       
    TH1F                  *fhistPhiDistBefore;  
    TH1F                  *fhistPhiDistAfter;   
    TH1F                  *fhistEtaDistBefore;  
    TH1F                  *fhistEtaDistAfter;   
    TH2F                  *fTPCvsGlobalMultBeforeOutliers;      
    TH2F                  *fTPCvsGlobalMultAfterOutliers;       
    TH2F                  *fTPCvsGlobalMultAfter;       
    TH2F                  *fHistBetavsPTOFbeforePID;    
    TH2F                  *fHistdEdxvsPTPCbeforePID;    
    TH3F                  *fhistNsigmaP;        
    TH2F                  *fhistTPCnSigmavsP;   
    TH2F                  *fhistTOFnSigmavsP;   
    TH2F                  *fHistBetavsPTOFafterPID;     
    TH2F                  *fHistdEdxvsPTPCafterPID;     
    TH2F                  *fHistBetavsPTOFafterPID_2;     
    TH2F                  *fHistdEdxvsPTPCafterPID_2;     
    TH2F                  *fHistBetavsPTOFafterPIDTPCTOF; 
    TH2F                  *fHistdEdxvsPTPCafterPIDTPCTOF; 
    TH2F                  *fHistBetavsPTOFafterPIDTPConly; 
    TH2F                  *fHistdEdxvsPTPCafterPIDTPConly; 
    TH2F                  *fHistPion_BetavsPTOFafterPIDTPCTOF; 
    TH2F                  *fHistPion_dEdxvsPTPCafterPIDTPCTOF; 
    TH2F                  *fHistKaon_BetavsPTOFafterPIDTPCTOF; 
    TH2F                  *fHistKaon_dEdxvsPTPCafterPIDTPCTOF; 
    TH2F                  *fHistProton_BetavsPTOFafterPIDTPCTOF; 
    TH2F                  *fHistProton_dEdxvsPTPCafterPIDTPCTOF; 
    TH1F                  *fhistPionEtaDistAfter; 
    TH1F                  *fhistKaonEtaDistAfter; 
    TH1F                  *fhistProtonEtaDistAfter; 
    THnSparseD            *fSparseAll;
    TF2                   *fPurityFunction[180];  //purity functions
    
    //qa object initialisation
    void SetupTPCTOFqa();
    void SetupEventInfo();
    //
    
    AliAnalysisTaskPIDconfig(const AliAnalysisTaskPIDconfig &other);
    AliAnalysisTaskPIDconfig& operator=(const AliAnalysisTaskPIDconfig &other);
    
    ClassDef(AliAnalysisTaskPIDconfig,4)  // Task to properly set the PID response functions of all detectors
};

#endif