o First Version of TRDnSigma implementation (Xianguo) o still requires some catching...

[u/mrichter/AliRoot.git] / PWGLF / totEt / AliAnalysisEtTrackMatchCorrections.h

diff --git a/PWGLF/totEt/AliAnalysisEtTrackMatchCorrections.h b/PWGLF/totEt/AliAnalysisEtTrackMatchCorrections.h
index 95cd46237ee912dd226826a76a79f32968e23e79..9ee9921992131f35dfff7d8ebd5de23451f8ed25 100644 (file)
--- a/PWGLF/totEt/AliAnalysisEtTrackMatchCorrections.h
+++ b/PWGLF/totEt/AliAnalysisEtTrackMatchCorrections.h
@@ -44,8 +44,8 @@ public:
         return *fSecondaryContr;
     }
 
-    TH2F TrackMatchingEfficiency() const {
-      return *fRecoEff;
+    TH2F *TrackMatchingEfficiency() const {
+      return fRecoEff;
     }
     
     Double_t ChargedContr(int mult) const {
@@ -64,7 +64,7 @@ public:
         return fSecondaryContr->Eval(mult)*fMeanSecondary;
     }
 
-    Double_t TrackMatchingEfficiency(Float_t pT, Int_t mult) const;
+    Double_t TrackMatchingEfficiency(Float_t pT, Int_t cent) const;
 
 
 // Setters
@@ -89,6 +89,15 @@ public:
         *fRecoEff = recoEff;
     }
 
+    void SetNeutronCorrection(Int_t cb, Double_t val){fNeutronCorrection[cb] = val;}
+    void SetHadronCorrection(Int_t cb, Double_t val){fHadronCorrection[cb] = val;}
+    void SetKaonCorrection(Int_t cb, Double_t val){fKaonCorrection[cb] = val;}
+    void SetSecondaryCorrection(Int_t cb, Double_t val){fSecondaryCorrection[cb] = val;}
+
+    Double_t GetNeutronCorrection(Int_t cb){if(cb>=0 && cb <20){return fNeutronCorrection[cb];}else{return 0.0;}}
+    Double_t GetHadronCorrection(Int_t cb){if(cb>=0 && cb <20){return fHadronCorrection[cb];}else{return 0.0;}}
+    Double_t GetKaonCorrection(Int_t cb){if(cb>=0 && cb <20){return fKaonCorrection[cb];}else{return 0.0;}}
+    Double_t GetSecondaryCorrection(Int_t cb){if(cb>=0 && cb <20){return fSecondaryCorrection[cb];}else{return 0.0;}}
 
 private:
 
@@ -112,6 +121,11 @@ private:
     // Mean deposited energy from secondaries 
     Double_t fMeanSecondary;
 
+    Double_t fNeutronCorrection[20];//NeutronCorrection
+    Double_t fHadronCorrection[20];//NeutronCorrection
+    Double_t fKaonCorrection[20];//NeutronCorrection
+    Double_t fSecondaryCorrection[20];//NeutronCorrection
+
     // Prohibited
 //! Equality operator
     bool operator==(const AliAnalysisEtTrackMatchCorrections &other) const;