Replacing Mult and Zvtx axes with pool index in THnSparse histos (Sandro)

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

diff --git a/PWGHF/hfe/AliAnalysisTaskFlowTPCEMCalEP.h b/PWGHF/hfe/AliAnalysisTaskFlowTPCEMCalEP.h
index 07937d99ba8f539d345e3dc02334bdf164ffbcf1..b5af7c180a7f2a368a9f2bbd7f32b7efe745b568 100644 (file)
--- a/PWGHF/hfe/AliAnalysisTaskFlowTPCEMCalEP.h
+++ b/PWGHF/hfe/AliAnalysisTaskFlowTPCEMCalEP.h
@@ -58,6 +58,7 @@ class AliAnalysisTaskFlowTPCEMCalEP : public AliAnalysisTaskSE {
   Double_t GetDeltaPhi(Double_t phiA,Double_t phiB)    const;
   Double_t GetPi0weight(Double_t mcPi0pT) const;
   Double_t GetEtaweight(Double_t mcEtapT) const;
+  Double_t GetDweight(Int_t whichD, Double_t mcDpT) const;
   
  private:
   
@@ -114,6 +115,16 @@ class AliAnalysisTaskFlowTPCEMCalEP : public AliAnalysisTaskSE {
   TH2F                 *fGammaWeight;          //! gamma weight
   TH2F                 *fPi0Weight;            //! pi0 weight
   TH2F                 *fEtaWeight;            //! eta weight
+  TH2F                 *fD0Weight;             //! D0 weight
+  TH2F                 *fDplusWeight;          //! D+ weight
+  TH2F                 *fDminusWeight;         //! D- weight
+  
+  TH1F                 *fDe[6];
+  TH1F                 *fD0e[6];
+  TH1F                 *fDpluse[6];
+  TH1F                 *fDminuse[6];
+  
+  TH2F                 *fD0_e;
   
   TH1F                 *fTot_pi0e;             //! inclusive electron
   TH1F                 *fPhot_pi0e;            //! ULS pair