AliConversionMesonCuts.h

Go to the documentation of this file.

#ifndef ALICONVERSIONMESONCUTS_H
#define ALICONVERSIONMESONCUTS_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 "AliAODMCParticle.h"
#include "AliCaloPhotonCuts.h"

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


class AliConversionMesonCuts : public AliAnalysisCuts {

  public:


    enum cutIds {
      kMesonKind,
      kBackgroundScheme,
      kNumberOfBGEvents,
      kDegreesForRotationMethod,
      kRapidityMesonCut,
      kPtCut,
      kalphaMesonCut,
      kSelectionCut,
      kElecShare,
      kToCloseV0s,
      kuseMCPSmearing,
      kDcaGammaGamma,
      kDcaRPrimVtx,
      kDcaZPrimVtx,
      kMinOpanMesonCut,
      kMaxOpanMesonCut,
      kNCuts
    };

    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];

    Bool_t  InitializeCutsFromCutString(const TString analysisCutSelection);
    void    FillElectonLabelArray(AliAODConversionPhoton* photon, Int_t nV0);

    AliConversionMesonCuts(const char *name="MesonCuts", const char * title="Meson Cuts");
    AliConversionMesonCuts(const AliConversionMesonCuts&);
    AliConversionMesonCuts& operator=(const AliConversionMesonCuts&);

    virtual ~AliConversionMesonCuts();                            //virtual destructor

    virtual Bool_t IsSelected(TObject* /*obj*/){return kTRUE;}
    virtual Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
    Bool_t MesonIsSelectedByMassCut (AliAODConversionMother *meson, Int_t nominalRange);

    TString GetCutNumber();

    // Cut Selection
    Bool_t MesonIsSelected(AliAODConversionMother *pi0,Bool_t IsSignal=kTRUE, Double_t fRapidityShift=0., Int_t leadingCellID1 = 0, Int_t leadingCellID2 = 0);
    Bool_t MesonIsSelectedMC(TParticle *fMCMother,AliMCEvent *mcEvent, Double_t fRapidityShift=0.);
    Bool_t MesonIsSelectedAODMC(AliAODMCParticle *MCMother,TClonesArray *AODMCArray, Double_t fRapidityShift=0.);
    Bool_t MesonIsSelectedMCDalitz(TParticle *fMCMother,AliMCEvent *mcEvent, Int_t &labelelectron, Int_t &labelpositron, Int_t &labelgamma,Double_t fRapidityShift=0.);
    Bool_t MesonIsSelectedAODMCDalitz(AliAODMCParticle *MCMother,TClonesArray *AODMCArray, Int_t &labelelectron, Int_t &labelpositron, Int_t &labelgamma,Double_t fRapidityShift=0.);
    Bool_t MesonIsSelectedMCEtaPiPlPiMiGamma(TParticle *fMCMother,AliMCEvent *mcEvent, Int_t &labelNegPion, Int_t &labelPosPion, Int_t &labelGamma, Double_t fRapidityShift=0);
    Bool_t MesonIsSelectedMCPiPlPiMiPiZero(TParticle *fMCMother,AliMCEvent *mcEvent, Int_t &labelNegPion, Int_t &labelPosPion, Int_t &labelNeutPion, Double_t fRapidityShift=0);
    Bool_t MesonIsSelectedMCPiZeroGamma(TParticle *fMCMother, AliMCEvent *mcEvent, Int_t &labelNeutPion, Int_t &labelGamma, Double_t fRapidityShift=0);
    Bool_t MesonIsSelectedMCChiC(TParticle *fMCMother,AliMCEvent *mcEvent, Int_t &, Int_t &, Int_t &, Double_t fRapidityShift=0. );
    Bool_t MesonIsSelectedPiZeroGammaAngle(AliAODConversionMother *omega, AliAODConversionMother *pi0, AliAODConversionPhoton *gamma,
                                           Bool_t DoPiZeroAngleCut, TF1 *maxfit, Double_t lowerFactor, Double_t upperFactor);
    void   PrintCuts();
    void   PrintCutsWithValues();

    void    SetLightOutput( Bool_t flag ){fDoLightOutput = flag; return;}
    void    SetRunningMode(Int_t mode){fMode = mode; return;}
    void    InitCutHistograms(TString name="",Bool_t additionalHists=kFALSE);
    void    SetFillCutHistograms(TString name=""){if(!fHistograms){InitCutHistograms(name);};}
    TList   *GetCutHistograms(){return fHistograms;}
    void    SmearParticle(AliAODConversionPhoton * photon);
    void    SmearVirtualPhoton(AliAODConversionPhoton* photon);
    TLorentzVector SmearElectron(TLorentzVector particle);

    void    SetDefaultSmearing(Double_t p0, Double_t p1, Double_t p2){fUseMCPSmearing=1.;fPBremSmearing=p0;fPSigSmearing=p1;fPSigSmearingCte=p2;return;}

    //Cut functions
    Bool_t RejectSharedElectronV0s(AliAODConversionPhoton* photon, Int_t nV0, Int_t nV0s);
    Bool_t RejectToCloseV0s(AliAODConversionPhoton* photon, TList *photons, Int_t nV0);

    void SetCaloMesonCutsObject(AliCaloPhotonCuts* cuts){fCaloPhotonCuts = cuts;}

    // Set Individual Cuts
    Bool_t SetMinPtCut(Int_t PtCut);
    Bool_t SetMesonKind(Int_t mesonKind);
    Bool_t SetSelectionWindowCut(Int_t selectionCut);
    Bool_t SetSelectionWindowMergedCut(Int_t selectionCut);
    Bool_t SetAlphaMesonCut(Int_t alphaMesonCut);
    Bool_t SetAlphaMesonMergedCut(Int_t alphaMesonCut);
    Bool_t SetRapidityMesonCut(Int_t RapidityMesonCut);
    Bool_t SetBackgroundScheme(Int_t BackgroundScheme);
    Bool_t SetNDegreesForRotationMethod(Int_t DegreesForRotationMethod);
    Bool_t SetNumberOfBGEvents(Int_t NumberOfBGEvents);
    Bool_t SetMCPSmearing(Int_t useMCPSmearing);
    Bool_t SetSharedElectronCut(Int_t sharedElec);
    Bool_t SetToCloseV0sCut(Int_t toClose);
    Bool_t SetDCAGammaGammaCut(Int_t DCAGammaGamma);
    Bool_t SetDCAZMesonPrimVtxCut(Int_t DCAZMesonPrimVtx);
    Bool_t SetDCARMesonPrimVtxCut(Int_t DCARMesonPrimVtx);
    void   SetOpeningAngleCut(Float_t OpeningAngle){fOpeningAngle = OpeningAngle;}
    Bool_t SetMinOpanMesonCut(Int_t minOpanMesonCut);
    Bool_t SetMaxOpanMesonCut(Int_t maxOpanMesonCut);
    void   SetEnableOpeningAngleCut (Bool_t isOn) {fEnableMinOpeningAngleCut = isOn;}
    void   SetIsMergedClusterCut(Int_t merged)                { fIsMergedClusterCut = merged; return;}
    Int_t  GetIsMergedClusterCut()                            { return fIsMergedClusterCut;}
    Double_t GetRapidityCutValue()                            { return fRapidityCutMeson; }

    Float_t FunctionMinMassCut(Float_t e);
    Float_t FunctionMaxMassCut(Float_t e);

    // Request Flags
    Bool_t   UseRotationMethod(){return fUseRotationMethodInBG;}
    Bool_t   UsePtmaxMethod(){return fUsePtmaxMethodForBG;}
    Bool_t   UseTrackMultiplicity(){return fUseTrackMultiplicityForBG;}
    Int_t    GetNumberOfBGEvents(){return fNumberOfBGEvents;}
    Int_t    NDegreesRotation(){return fNDegreeRotationPMForBG;}
    Bool_t   DoBGCalculation(){return fDoBG;}
    Bool_t   DoBGProbability(){return fDoBGProbability;}
    Bool_t   UseElecSharingCut(){return fDoSharedElecCut;}
    Bool_t   UseToCloseV0sCut(){return fDoToCloseV0sCut;}
    Bool_t   UseMCPSmearing(){return fUseMCPSmearing;}
    Int_t    BackgroundHandlerType(){return fBackgroundHandler;}
    Double_t GetSelectionLow() const {return fSelectionLow;}
    Double_t GetSelectionHigh() const {return fSelectionHigh;}
    Double_t GetAcceptMassFlag() const {return fAcceptMesonMass;}
    Double_t GetMinPt() const {return fMinPt;}
    Bool_t   UseLikeSignMixing() {return fBackgroundUseLikeSign;}
    Bool_t   UseSidebandMixing() {return fBackgroundUseSideband;}
    Bool_t   UseSidebandMixingBothSides() {return fBackgroundUseSidebandBothSides;}
    Double_t GetSidebandMixingLow() const {return fSidebandMixingLow;}
    Double_t GetSidebandMixingHigh() const {return fSidebandMixingHigh;}
    Double_t GetSidebandMixingLeftLow() const {return fSidebandMixingLeftLow;}
    Double_t GetSidebandMixingLeftHigh() const {return fSidebandMixingLeftHigh;}
    Double_t GetSidebandMixingRightLow() const {return fSidebandMixingRightLow;}
    Double_t GetSidebandMixingRightHigh() const {return fSidebandMixingRightHigh;}

  protected:
    TRandom3    fRandom;                        
    AliCaloPhotonCuts* fCaloPhotonCuts;         
    TList*      fHistograms;                    
    TObjString* fCutString;                     
    TString     fCutStringRead;
    // Histograms
    TH2F*       fHistoMesonCuts;                
    TH2F*       fHistoMesonBGCuts;              
    TH1F*       fHistoDCAGGMesonBefore;         
    TH1F*       fHistoDCAZMesonPrimVtxBefore;   
    TH1F*       fHistoDCARMesonPrimVtxBefore;   
    TH1F*       fHistoDCAGGMesonAfter;          
    TH2F*       fHistoDCAZMesonPrimVtxAfter;    
    TH1F*       fHistoDCARMesonPrimVtxAfter;    
    TH1F*       fHistoInvMassBefore;            
    TH1F*       fHistoInvMassAfter;             

    TF1*        fBrem;                          
    TF1*        fFAlphaCut;                     
    TF1*        fFMinOpanCut;                   
    TF1*        fFMaxOpanCut;                   

    Double_t    fMaxR;                          
    Double_t    fMinPt;                         
    Double_t    fSelectionLow;                  
    Double_t    fSelectionHigh;                 
    Double_t    fAlphaMinCutMeson;              
    Double_t    fAlphaCutMeson;                 
    Double_t    fRapidityCutMeson;              
    Double_t    fMinV0Dist;                     
    Double_t    fPBremSmearing;                 
    Double_t    fPSigSmearing;                  
    Double_t    fPSigSmearingCte;               
    Double_t    fDCAGammaGammaCut;              
    Double_t    fDCAZMesonPrimVtxCut;           
    Double_t    fDCARMesonPrimVtxCut;           
    Double_t    fMinOpanCutMeson;               
    Double_t    fMaxOpanCutMeson;               
    Double_t    fSidebandMixingLow;             
    Double_t    fSidebandMixingHigh;            
    Double_t    fSidebandMixingLeftLow;         
    Double_t    fSidebandMixingLeftHigh;        
    Double_t    fSidebandMixingRightLow;        
    Double_t    fSidebandMixingRightHigh;       

    Float_t     fOpeningAngle;                  

    Int_t       fMode;                          
    Int_t       fMesonKind;                     
    Int_t       fIsMergedClusterCut;            
    Int_t       fSelectionWindowCut;            
    Int_t       fNDegreeRotationPMForBG;        
    Int_t       fNumberOfBGEvents;              
    Int_t       fElectronLabelArraySize;        
    Int_t*      fElectronLabelArray;            //[fElectronLabelArraySize] Array with elec/pos v0 label
    Int_t       fBackgroundHandler;             

    Bool_t      fDoLightOutput;                 
    Bool_t      fDoMinPtCut;                    
    Bool_t      fEnableMassCut;                 
    Bool_t      fAcceptMesonMass;               
    Bool_t      fUseRotationMethodInBG;         
    Bool_t      fUsePtmaxMethodForBG;           
    Bool_t      fDoBG;                          
    Bool_t      fDoBGProbability;               
    Bool_t      fUseTrackMultiplicityForBG;     
    Bool_t      fEnableMinOpeningAngleCut;      
    Bool_t      fEnableOneCellDistCut;          
    Bool_t      fDoToCloseV0sCut;               
    Bool_t      fDoSharedElecCut;               
    Bool_t      fUseMCPSmearing;                
    Bool_t      fAlphaPtDepCut;                 
    Bool_t      fDCAGammaGammaCutOn;            
    Bool_t      fDCAZMesonPrimVtxCutOn;         
    Bool_t      fDCARMesonPrimVtxCutOn;         
    Bool_t      fMinOpanPtDepCut;               
    Bool_t      fMaxOpanPtDepCut;               
    Bool_t      fBackgroundUseSideband;         
    Bool_t      fBackgroundUseSidebandBothSides;
    Bool_t      fBackgroundUseLikeSign;         

  private:

    ClassDef(AliConversionMesonCuts,24)
};


#endif