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 a24c8eca2d6a7c9fff0dd48d1c24501d8e72de32..df63b55116a5c0f8a5557e24c5e29f88cfa93497 100644 (file)
--- a/PWGHF/hfe/AliAnalysisTaskElecHadronCorrel.h
+++ b/PWGHF/hfe/AliAnalysisTaskElecHadronCorrel.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,6 +78,8 @@ 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 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();
 
@@ -122,6 +126,9 @@ 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
 
@@ -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
@@ -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