Adding D meson efficiency maps and Coverity fixes (Sandro)

[u/mrichter/AliRoot.git] / PWGHF / correlationHF / AliAnalysisTaskDStarCorrelations.h

diff --git a/PWGHF/correlationHF/AliAnalysisTaskDStarCorrelations.h b/PWGHF/correlationHF/AliAnalysisTaskDStarCorrelations.h
index 7cefe51b9aac5f7b39d8f4363300c5ef3e85b021..ade088c547c83dc7569d294c08edd5e6be710095 100644 (file)
--- a/PWGHF/correlationHF/AliAnalysisTaskDStarCorrelations.h
+++ b/PWGHF/correlationHF/AliAnalysisTaskDStarCorrelations.h
@@ -40,7 +40,7 @@
 #include "AliHFCorrelator.h"\r
 #include <THnSparse.h>\r
 #include "AliAnalysisUtils.h"\r
-\r
+#include "AliVertexingHFUtils.h"\r
 class TParticle ;\r
 class TClonesArray ;\r
 class AliAODMCParticle;\r
@@ -73,6 +73,7 @@ class AliAnalysisTaskDStarCorrelations : public AliAnalysisTaskSE
   \r
   void DefineThNSparseForAnalysis();\r
   void DefineHistoForAnalysis();\r
+  void EnlargeDZeroMassWindow();\r
   \r
   \r
   // checker for event mixing\r
@@ -98,8 +99,8 @@ class AliAnalysisTaskDStarCorrelations : public AliAnalysisTaskSE
   void SetDim(){fDim = 4;\r
     fDMesonSigmas = new Float_t[4];}\r
   void SetDeffMapvsPt(TH1D * map){fDeffMapvsPt = map;}\r
-  void SetDeffMapvsPtvsMult(TH2D * map){fDeffMapvsPtvsMult = map;}\r
-  void SetDeffMapvsPtvsMultvsEta(TH3D * map){fDeffMapvsPtvsMultvsEta = map;}\r
+  void SetDeffMapvsPtvsMult(TH2D * map){fDeffMapvsPtvsMult = (TH2D*)map;}\r
+  void SetDeffMapvsPtvsMultvsEta(TH2D * map){fDeffMapvsPtvsEta = map;}\r
   void SetNofPhiBins(Int_t nbins){fPhiBins = nbins;}\r
   \r
   \r
@@ -124,12 +125,15 @@ private:
   Bool_t fReco; // use reconstruction or MC truth\r
   Bool_t fUseEfficiencyCorrection; // boolean variable to use or not the efficiency correction\r
   Bool_t fUseDmesonEfficiencyCorrection; // boolean flag for the use of Dmeson efficiency correction\r
+    Bool_t fUseCentrality;// boolean to switch in between centrality or multiplicity\r
   Int_t fPhiBins;\r
   Int_t fEvents; //! number of event\r
   Int_t fDebugLevel; //! debug level\r
   Int_t fDisplacement; // set 0 for no displacement cut, 1 for absolute d0, 2 for d0/sigma_d0\r
   Int_t fDim;//\r
+    Int_t fNofPtBins;\r
   Float_t *fDMesonSigmas;//[fDim]\r
+ Float_t * fD0Window;  //[fNofPtBins]\r
   \r
   \r
   \r
@@ -142,9 +146,9 @@ private:
   \r
   TH1D * fDeffMapvsPt; // histo for Deff mappin\r
   TH2D * fDeffMapvsPtvsMult; // histo for Deff mappin\r
-  TH3D * fDeffMapvsPtvsMultvsEta; // histo for Deff mappin\r
+  TH2D * fDeffMapvsPtvsEta; // histo for Deff mappin\r
   \r
-  ClassDef(AliAnalysisTaskDStarCorrelations,4); // class for D meson correlations\r
+  ClassDef(AliAnalysisTaskDStarCorrelations,5); // class for D meson correlations\r
   \r
 };\r
 \r