Updates for TRD HFE analysis

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskElecHadronCorrel.h

diff --git a/PWGHF/hfe/AliAnalysisTaskElecHadronCorrel.h b/PWGHF/hfe/AliAnalysisTaskElecHadronCorrel.h
index 51b42b9770847e91e46370e868594f504b705bf0..df63b55116a5c0f8a5557e24c5e29f88cfa93497 100644 (file)
--- a/PWGHF/hfe/AliAnalysisTaskElecHadronCorrel.h
+++ b/PWGHF/hfe/AliAnalysisTaskElecHadronCorrel.h
@@ -31,7 +31,7 @@ class AliHFEcuts;
 class AliHFEpid;
 class AliHFEpidQAmanager;
 class AliCFManager;
-//class AliEventPoolManager;
+class AliEventPoolManager;
 
 #include "AliLog.h"
 #include "AliAnalysisTaskSE.h"
@@ -63,6 +63,8 @@ class AliAnalysisTaskElecHadronCorrel : public AliAnalysisTaskSE {
     void ElectronHadCorrelEtaBinsNoPartner(Int_t itrack,Int_t jtrack, AliVTrack *track, TH2F *DphiPtEta1, TH2F *DphiPtEta11,TH2F *DphiPtEta12,TH2F *DphiPtEta13,TH2F *DphiPtEta14,TH2F *DphiPtEta2, TH2F *DphiPtEta21,TH2F *DphiPtEta22,TH2F *DphiPtEta23,TH2F *DphiPtEta24);  
     // void ElectronHadCorrelEtaBinsNoPartner(Int_t itrack,Int_t jtrack, AliVTrack *track, TH3F *DphiPtEta1, TH3F *DphiPtEta11,TH3F *DphiPtEta12,TH3F *DphiPtEta13,TH3F *DphiPtEta14); 
     void HadronInfo(Int_t itrack);
+
+    void SetEventTriggerSelectionMB(Bool_t TriggerMB) {fTriggerMB = TriggerMB;};
     void SetTriggerSelection(Bool_t TriggerCentral) {fTriggerCentral = TriggerCentral;};
     void    SetCentralityParameters(Double_t CentralityMin, Double_t CentralityMax, const char* CentralityMethod); //select centrality
     void    CheckCentrality(AliVEvent *event,Bool_t &centralitypass); //to use only events with the correct centrality....
@@ -76,8 +78,10 @@ class AliAnalysisTaskElecHadronCorrel : public AliAnalysisTaskSE {
     void SetShowerShapeCutsM20(Double_t M20CutMin, Double_t M20CutMax){fM20CutMin=M20CutMin; fM20CutMax=M20CutMax;};
     void SetShowerShapeCutsDisp(Double_t DispCutMin, Double_t DispCutMax){fDispCutMin=DispCutMin; fDispCutMax=DispCutMax;};
     void SetEovPCuts(Double_t EovPMin, Double_t EovPMax){fEovPMin=EovPMin;fEovPMax=EovPMax;};
- //   void MixedEvent(AliVTrack *track, TH2F *DphiPt, TH2F *DphiPt1, TH2F *DphiPt2, TH2F *DphiPt3, TH2F *DphiPt4);
- //   TObjArray* CloneAndReduceTrackList();
+    void SetTPCNClsHad(Int_t TPCNClsHad) {fTPCNClsHad = TPCNClsHad;};
+    void SetAssoElecITSrefit(Bool_t AssoEleITSref){fAssoEleITSref = AssoEleITSref;};
+    void MixedEvent(AliVTrack *track, TH2F *DphiPt, TH2F *DphiPt1, TH2F *DphiPt2, TH2F *DphiPt3, TH2F *DphiPt4);
+    TObjArray* CloneAndReduceTrackList();
 
   private:
 
@@ -122,8 +126,11 @@ class AliAnalysisTaskElecHadronCorrel : public AliAnalysisTaskSE {
     Double_t   fEovPMin;//
     Double_t   fEovPMax;//
     Bool_t     fTriggerCentral;//
+    Bool_t     fTriggerMB;//
+    Int_t      fTPCNClsHad;//
+    Bool_t     fAssoEleITSref;//
 
- //   AliEventPoolManager*     fPoolMgr;         //! event pool manager
+    AliEventPoolManager*     fPoolMgr;         //! event pool manager
 
     TH1F                       *fNoEvents;             //no of events
     //    TH1F                 *fTrkpt;                //track pt
@@ -251,7 +258,7 @@ class AliAnalysisTaskElecHadronCorrel : public AliAnalysisTaskSE {
 
     //   TH1F                  *fTrackPtBefTrkCuts;    //Track pt before track cuts    
     //   TH1F                  *fTrackPtAftTrkCuts;    //Track pt after track cuts
-    //   TH2F                  *fTPCnsigma;            //TPC n sigma vs p      
+    TH2F                       *fTPCnsigma;            //TPC n sigma vs p      
     //   TH1F                  *fNCellv1;              //No of cells in cluster, all EMCAL cluster
     //   TH1F                  *fClsEv1;               //Cluster energy, all EMCAL cluster
     //   TH1F                  *fNClusv1;              //No of clusters in event, all EMCAL cluster
@@ -281,7 +288,7 @@ class AliAnalysisTaskElecHadronCorrel : public AliAnalysisTaskSE {
           TH1F        *fTrakPhiSPDAnd;//
           TH1F        *fTrackHFEcutsITS;//
      */
-/*    TH1F        *fNoMixedEvents;//
+    TH1F        *fNoMixedEvents;//
     TH2F        *fMixStat; //no of events in pool vs multplicity
     TH2F        *fMixStat1; //no of events in pool vs zvtx 
     TH2F        *fMixedIncElecDphi; //Mixed event - inclusive elec DPhi
@@ -309,7 +316,7 @@ class AliAnalysisTaskElecHadronCorrel : public AliAnalysisTaskSE {
     TH2F        *fMixedDphiLSMassLow2;//
     TH2F        *fMixedDphiLSMassLow3;//
     TH2F        *fMixedDphiLSMassLow4;//
-*/
+
     TH1F        *fHadronPt;//
     TH1F       *fCentralityPass; // ! QA histogram of events that pass centrality cut
     TH1F       *fCentralityNoPass; //! QA histogram of events that do not pass centrality cut
@@ -365,7 +372,7 @@ class AliAnalysisTaskElecHadronCorrel : public AliAnalysisTaskSE {
 
     ClassDef(AliAnalysisTaskElecHadronCorrel, 2); //!example of analysis
 };
-/*
+
    class AliehDPhiBasicParticle : public AliVParticle
    {
    public:
@@ -413,7 +420,6 @@ Short_t fCharge;   // charge
 
 ClassDef( AliehDPhiBasicParticle, 1); // class which contains only quantities requires for this analysis to reduce memory consumption for event mixing
 };
- */
 #endif