AliConversionPhotonCuts.h

Go to the documentation of this file.

#ifndef ALICONVERSIONPHOTONCUTS_H
#define ALICONVERSIONPHOTONCUTS_H

#include "AliAODpidUtil.h"
#include "AliConversionPhotonBase.h"
#include "AliAODConversionMother.h"
#include "AliAODTrack.h"
#include "AliESDtrack.h"
#include "AliVTrack.h"
#include "AliAODTrack.h"
#include "AliMCEvent.h"
#include "AliAnalysisCuts.h"
#include "TH1F.h"
#include "TF1.h"
#include "TProfile.h"
#include "AliAnalysisUtils.h"
#include "AliAnalysisManager.h"


class AliESDEvent;
class AliAODEvent;
class AliConversionPhotonBase;
class AliPIDResponse;
class AliKFVertex;
class TH1F;
class TH2F;
class TF1;
class TProfile;
class AliAnalysisCuts;
class iostream;
class TList;
class AliAnalysisManager;
class AliAODMCParticle;

class AliConversionPhotonCuts : public AliAnalysisCuts {
      
  public: 
    enum cutIds {
      kv0FinderType,                
      ketaCut,                                     
      kRCut,
      kEtaForPhiSector,
      kMinPhiSector,
      kMaxPhiSector,
      ksinglePtCut,                 
      kclsTPCCut,                   
      kededxSigmaCut,               
      kpidedxSigmaCut,              
      kpiMomdedxSigmaCut,        
      kpiMaxMomdedxSigmaCut,        
      kLowPRejectionSigmaCut,       
      kTOFelectronPID, 
      kITSelectronPID,
      kTRDelectronPID,
      kQtMaxCut,                    
      kchi2GammaCut,                
      kPsiPair, 
      kdoPhotonAsymmetryCut,
      kCosPAngle,
      kElecShare,
      kToCloseV0s,
      kDcaRPrimVtx,
      kDcaZPrimVtx,
      kInPlaneOutOfPlane,
      kNCuts
    };

    enum photonCuts {
        kPhotonIn=0,
        kOnFly,
        kNoTracks,
        kNoV0,
        kTrackCuts,
        kdEdxCuts,
        kConvPointFail,
        kPhotonCuts,
        kEventPlane,
        kPhotonOut
    };


    Bool_t SetCutIds(TString cutString);
    Int_t fCuts[kNCuts];
    Bool_t SetCut(cutIds cutID, Int_t cut);
    Bool_t UpdateCutString();

    static const char * fgkCutNames[kNCuts];

    Double_t GetCosineOfPointingAngle(const AliConversionPhotonBase * photon, AliVEvent * event) const; 
    Bool_t InitializeCutsFromCutString(const TString analysisCutSelection);
    void FillElectonLabelArray(AliAODConversionPhoton* photon, Int_t nV0);
    void SetPreSelectionCutFlag(Bool_t preSelFlag){fPreSelCut = preSelFlag;}   

    AliConversionPhotonCuts(const char *name="V0Cuts", const char * title="V0 Cuts");
    AliConversionPhotonCuts(const AliConversionPhotonCuts&);
    AliConversionPhotonCuts& operator=(const AliConversionPhotonCuts&);

    virtual ~AliConversionPhotonCuts();                            //virtual destructor

    static AliConversionPhotonCuts * GetStandardCuts2010PbPb();
    static AliConversionPhotonCuts * GetStandardCuts2010pp();

    Bool_t InitPIDResponse();
    void SetPIDResponse(AliPIDResponse * pidResponse) {fPIDResponse = pidResponse;}
    AliPIDResponse * GetPIDResponse() { return fPIDResponse;}

    
    virtual Bool_t IsSelected(TObject* /*obj*/){return kTRUE;}
    virtual Bool_t IsSelected(TList* /*list*/) {return kTRUE;}

    TString GetCutNumber();
    
    Float_t GetKappaTPC(AliConversionPhotonBase *gamma, AliVEvent *event);
    
    // Cut Selection
    Bool_t PhotonIsSelected(AliConversionPhotonBase * photon, AliVEvent  * event);
    Bool_t PhotonIsSelectedMC(TParticle *particle,AliMCEvent *mcEvent,Bool_t checkForConvertedGamma=kTRUE);
    Bool_t PhotonIsSelectedAODMC(AliAODMCParticle *particle,TClonesArray *aodmcArray,Bool_t checkForConvertedGamma=kTRUE);
    //Bool_t ElectronIsSelectedMC(TParticle *particle,AliMCEvent *mcEvent);
    Bool_t TracksAreSelected(AliVTrack * negTrack, AliVTrack * posTrack);
    //Bool_t MesonIsSelected(AliAODConversionMother *pi0,Bool_t IsSignal=kTRUE);
    //Bool_t MesonIsSelectedMC(TParticle *fMCMother,AliMCEvent *mcEvent, Bool_t bMCDaughtersInAcceptance=kFALSE);
      
    void PrintCuts();
    void PrintCutsWithValues();

    void SetLightOutput( Bool_t flag ){fDoLightOutput = flag; return;}
    void InitCutHistograms(TString name="",Bool_t preCut = kTRUE);
    void SetFillCutHistograms(TString name="",Bool_t preCut = kTRUE){if(!fHistograms){InitCutHistograms(name,preCut);};}
    TList *GetCutHistograms(){return fHistograms;}
    void FillPhotonCutIndex(Int_t photoncut){if(fHistoCutIndex)fHistoCutIndex->Fill(photoncut);}    
    void FillV0EtaBeforedEdxCuts(Float_t v0Eta){if(fHistoEtaDistV0s)fHistoEtaDistV0s->Fill(v0Eta);}
    void FillV0EtaAfterdEdxCuts(Float_t v0Eta){if(fHistoEtaDistV0sAfterdEdxCuts)fHistoEtaDistV0sAfterdEdxCuts->Fill(v0Eta);}

    void SetV0ReaderName(TString name){fV0ReaderName = name; return;}
    void SetProcessAODCheck(Bool_t flag){fProcessAODCheck = flag; return;}

    AliVTrack * GetTrack(AliVEvent * event, Int_t label);
    AliESDtrack *GetESDTrack(AliESDEvent * event, Int_t label);
    
    Bool_t SpecificTrackCuts(AliAODTrack * negTrack, AliAODTrack * posTrack,Int_t &cutIndex);
    Bool_t SpecificTrackCuts(AliESDtrack * negTrack, AliESDtrack * posTrack,Int_t &cutIndex);
    Bool_t AcceptanceCuts(AliConversionPhotonBase *photon);
    Bool_t AcceptanceCut(TParticle *particle, TParticle * ePos,TParticle* eNeg);
    Bool_t PhiSectorCut(AliConversionPhotonBase * photon);
    Bool_t dEdxCuts(AliVTrack * track);
    Bool_t KappaCuts(AliConversionPhotonBase * photon,AliVEvent *event);
    Bool_t ArmenterosQtCut(AliConversionPhotonBase *photon);
    Bool_t AsymmetryCut(AliConversionPhotonBase *photon,AliVEvent *event);
    Bool_t PIDProbabilityCut(AliConversionPhotonBase *photon, AliVEvent * event);
    Bool_t SelectV0Finder(Bool_t onfly){
      if(onfly == fUseOnFlyV0Finder) return kTRUE;
      else return kFALSE;
    }
    Bool_t PhotonCuts(AliConversionPhotonBase *photon,AliVEvent *event);
    Bool_t CorrectedTPCClusterCut(AliConversionPhotonBase *photon, AliVEvent * event);
    Bool_t PsiPairCut(const AliConversionPhotonBase * photon) const;
    Bool_t CosinePAngleCut(const AliConversionPhotonBase * photon, AliVEvent * event) const;
    Bool_t RejectSharedElectronV0s(AliAODConversionPhoton* photon, Int_t nV0, Int_t nV0s);
    Bool_t RejectToCloseV0s(AliAODConversionPhoton* photon, TList *photons, Int_t nV0);

    UChar_t DeterminePhotonQualityAOD(AliAODConversionPhoton*, AliVEvent*);
    Bool_t InPlaneOutOfPlaneCut(Double_t photonPhi, Double_t eventPlaneAngle = -100, Bool_t fill = kTRUE);
    Int_t GetInPlaneOutOfPlaneCut(){return fInPlaneOutOfPlane;}

    // Set Individual Cuts
    Bool_t SetRCut(Int_t RCut);
    Bool_t SetV0Finder(Int_t v0FinderType);
    Bool_t SetChi2GammaCut(Int_t chi2GammaCut);
    Bool_t SetTPCdEdxCutPionLine(Int_t pidedxSigmaCut);
    Bool_t SetTPCdEdxCutElectronLine(Int_t ededxSigmaCut);
    Bool_t SetSinglePtCut(Int_t singlePtCut);
    Bool_t SetTPCClusterCut(Int_t clsTPCCut);
    Bool_t SetEtaCut(Int_t etaCut);
    Bool_t SetEtaForPhiCut(Int_t etaPhiCut);
    Bool_t SetMinPhiSectorCut(Int_t minPhiCut);
    Bool_t SetMaxPhiSectorCut(Int_t maxPhiCut);
    Bool_t SetMinMomPiondEdxCut(Int_t piMinMomdedxSigmaCut);
    Bool_t SetMaxMomPiondEdxCut(Int_t piMaxMomdedxSigmaCut);
    Bool_t SetLowPRejectionCuts(Int_t LowPRejectionSigmaCut);
    Bool_t SetQtMaxCut(Int_t QtMaxCut);
    Bool_t SetTOFElectronPIDCut(Int_t TOFelectronPID);
    Bool_t SetTRDElectronCut(Int_t TRDElectronCut);
    Bool_t SetPhotonAsymmetryCut(Int_t doPhotonAsymmetryCut);
    Bool_t SetCosPAngleCut(Int_t cosCut);
    Bool_t SetPsiPairCut(Int_t psiCut);
    Bool_t SetSharedElectronCut(Int_t sharedElec);
    Bool_t SetToCloseV0sCut(Int_t toClose);
    Bool_t SetDCARPhotonPrimVtxCut(Int_t DCARPhotonPrimVtx);
    Bool_t SetDCAZPhotonPrimVtxCut(Int_t DCAZPhotonPrimVtx);
    Bool_t SetInPlaneOutOfPlane(Int_t inOutPlane);
    Bool_t SetKappaTPCCut(Int_t kappaCut);
    void SetIsHeavyIon(Int_t isHeavyIon){fIsHeavyIon=isHeavyIon;}
    Int_t GetFirstTPCRow(Double_t radius);
    
    Bool_t SetITSElectronPIDCut(Int_t ITSelectronPID);
    Bool_t SetTRDElectronPIDCut(Int_t TRDelectronPID);
    
    // Request Flags
    Bool_t UseElecSharingCut(){return fDoSharedElecCut;}
    Bool_t UseToCloseV0sCut(){return fDoToCloseV0sCut;}
    Double_t GetEtaCut(){return fEtaCut;}
    void SetDodEdxSigmaCut(Bool_t k=kTRUE)  {fDodEdxSigmaCut=k;}
    void SetSwitchToKappaInsteadOfNSigdEdxTPC(Bool_t k=kTRUE) {fSwitchToKappa=k;}
    
    Bool_t GetMaterialBudgetWeightsInitialized() {return fMaterialBudgetWeightsInitialized;}
    Bool_t InitializeMaterialBudgetWeights(Int_t flag, TString filename);
    Float_t GetMaterialBudgetCorrectingWeightForTrueGamma(AliAODConversionPhoton* gamma);

    Int_t GetV0FinderSameSign(){return fUseOnFlyV0FinderSameSign;}
      
  protected:
    TList*            fHistograms;                          
    AliPIDResponse*   fPIDResponse;                         

    Bool_t            fDoLightOutput;                       
    TString           fV0ReaderName;               

    //cuts
    Double_t          fMaxR;                                
    Double_t          fMinR;                                
    Double_t          fEtaCut;                              
    Double_t          fEtaCutMin;                           
    Float_t           fEtaForPhiCutMin;                     
    Float_t           fEtaForPhiCutMax;                     
    Float_t           fMinPhiCut;                           
    Float_t           fMaxPhiCut;                           
    Bool_t            fDoShrinkTPCAcceptance;               
    Double_t          fPtCut;                               
    Double_t          fSinglePtCut;                         
    Double_t          fSinglePtCut2;                        
    Bool_t            fDoAsymPtCut;                         
    Double_t          fMaxZ;                                
    Double_t          fMinClsTPC;                           
    Double_t          fMinClsTPCToF;                        
    Double_t          fLineCutZRSlope;                      
    Double_t          fLineCutZValue;                       
    Double_t          fLineCutZRSlopeMin;                   
    Double_t          fLineCutZValueMin;                    
    Double_t          fChi2CutConversion;                   
    Double_t          fPIDProbabilityCutNegativeParticle;   
    Double_t          fPIDProbabilityCutPositiveParticle;   
    Bool_t            fDodEdxSigmaCut;                      
    Bool_t            fDoTOFsigmaCut;                       
    Double_t          fPIDTRDEfficiency;                    
    Bool_t            fDoTRDPID;                            
    Double_t          fPIDnSigmaAboveElectronLine;          
    Double_t          fPIDnSigmaBelowElectronLine;          
    Double_t          fTofPIDnSigmaAboveElectronLine;       
    Double_t          fTofPIDnSigmaBelowElectronLine;       
    Double_t          fPIDnSigmaAbovePionLine;              
    Double_t          fPIDnSigmaAbovePionLineHighPt;        
    Double_t          fPIDMinPnSigmaAbovePionLine;          
    Double_t          fPIDMaxPnSigmaAbovePionLine;          
    Double_t          fDoKaonRejectionLowP;                 
    Double_t          fDoProtonRejectionLowP;               
    Double_t          fDoPionRejectionLowP;                 
    Double_t          fPIDnSigmaAtLowPAroundKaonLine;       
    Double_t          fPIDnSigmaAtLowPAroundProtonLine;     
    Double_t          fPIDnSigmaAtLowPAroundPionLine;       
    Double_t          fPIDMinPKaonRejectionLowP;            
    Double_t          fPIDMinPProtonRejectionLowP;          
    Double_t          fPIDMinPPionRejectionLowP;            
    Bool_t            fDoQtGammaSelection;                  
    Bool_t            fDo2DQt;                              
    Double_t          fQtMax;                               
    Double_t          fNSigmaMass;                          
    Bool_t            fUseEtaMinCut;                        
    Bool_t            fUseOnFlyV0Finder;                    
    Int_t             fUseOnFlyV0FinderSameSign;            
    Bool_t            fDoPhotonAsymmetryCut;                
    Bool_t            fDoPhotonPDependentAsymCut;           
    TF1 *             fFAsymmetryCut;                       
    Double_t          fMinPPhotonAsymmetryCut;              
    Double_t          fMinPhotonAsymmetry;                  
    Bool_t            fUseCorrectedTPCClsInfo;              
    Bool_t            fUseTOFpid;                           
    Float_t           fOpeningAngle;                        
    Float_t           fPsiPairCut;                          
    Bool_t            fDo2DPsiPairChi2;                     
    Bool_t            fIncludeRejectedPsiPair;              
    Float_t           fCosPAngleCut;                        
    Bool_t            fDoToCloseV0sCut;                     
    Double_t          fminV0Dist;                           
    Bool_t            fDoSharedElecCut;                     
    Bool_t            fDoPhotonQualitySelectionCut;         
    Int_t             fPhotonQualityCut;                    
    TRandom3          fRandom;                              
    Int_t             fElectronArraySize;                   
    Int_t*            fElectronLabelArray;                  //[fElectronArraySize]
    Double_t          fDCAZPrimVtxCut;                      
    Double_t          fDCARPrimVtxCut;                      
    Int_t             fInPlaneOutOfPlane;                   
    Float_t           fConversionPointXArray;               
    Float_t           fConversionPointYArray;               
    Float_t           fConversionPointZArray;               
    TObjString*       fCutString;                           
    TString           fCutStringRead;                       
    Int_t             fIsHeavyIon;                          
    Bool_t            fUseITSpid;                           
    Double_t          fITSPIDnSigmaAboveElectronLine;       
    Double_t          fITSPIDnSigmaBelowElectronLine;       
    Double_t          fMaxPtPIDITS;                         
    Double_t          fTRDPIDAboveCut;                      
    Double_t          fTRDPIDBelowCut;                      
    Bool_t            fDoDoubleCountingCut;                 
    Double_t          fMinRDC;                              
    Double_t          fDeltaR;                              
    Double_t          fOpenAngle;                           
    Bool_t            fSwitchToKappa;                       
    Float_t           fKappaMinCut;                         
    Float_t           fKappaMaxCut;                         
    
    // Histograms
    TH1F*             fHistoEtaDistV0s;                     
    TH1F*             fHistoEtaDistV0sAfterdEdxCuts;        
    TH2F*             fHistodEdxCuts;                       
    TH2F*             fHistoTPCdEdxbefore;                  
    TH2F*             fHistoTPCdEdxafter;                   
    TH2F*             fHistoTPCdEdxSigbefore;               
    TH2F*             fHistoTPCdEdxSigafter;                
    TH2F*             fHistoKappaafter;                     
    TH2F*             fHistoTOFbefore;                      
    TH2F*             fHistoTOFSigbefore;                   
    TH2F*             fHistoTOFSigafter;                    
    TH2F*             fHistoITSSigbefore;                   
    TH2F*             fHistoITSSigafter;                    
    TH2F*             fHistoPsiPairDeltaPhiafter;           
    TH1F*             fHistoTrackCuts;                      
    TH2F*             fHistoPhotonCuts;                     
    TH1F*             fHistoInvMassbefore;                  
    TH2F*             fHistoArmenterosbefore;               
    TH1F*             fHistoInvMassafter;                   
    TH2F*             fHistoArmenterosafter;                
    TH2F*             fHistoAsymmetryafter;                 
    TH2F*             fHistoAcceptanceCuts;                 
    TH1F*             fHistoCutIndex;                       
    TH1F*             fHistoEventPlanePhi;                  
    Bool_t            fPreSelCut;                           
    Bool_t            fProcessAODCheck;                     
    Bool_t            fMaterialBudgetWeightsInitialized;    
    TProfile*         fProfileContainingMaterialBudgetWeights;      

  private:
    ClassDef(AliConversionPhotonCuts,15)
};

#endif