Add Ds treatment in macros for pt spectra (Gian Michele)

diff --git a/PWGHF/vertexingHF/macros/CombineFeedDownMCSubtractionMethodsUncertainties.C b/PWGHF/vertexingHF/macros/CombineFeedDownMCSubtractionMethodsUncertainties.C
index 76dcf812a8e04987c056fb2ae9c507c4c4e87f78..9f1c30c0a33c098967bbbd7e59946dfdbbd47387 100644 (file)
--- a/PWGHF/vertexingHF/macros/CombineFeedDownMCSubtractionMethodsUncertainties.C
+++ b/PWGHF/vertexingHF/macros/CombineFeedDownMCSubtractionMethodsUncertainties.C
@@ -1,3 +1,4 @@
+#if !defined(__CINT__) || defined(__MAKECINT__)
 #include "TFile.h"
 #include "TH1.h"
 #include "TH1D.h"
@@ -9,10 +10,12 @@
 #include "TMath.h"
 #include "TROOT.h"
 #include "TStyle.h"
-
 #include "AliHFSystErr.h"
-
 #include <Riostream.h>
+#endif
+
+/* $Id$ */ 
+
 
 //_________________________________________________________________________________________
 //
@@ -74,12 +77,16 @@ void CombineFeedDownMCSubtractionMethodsUncertainties(const char *fcfilename="HF
   TGraphAsymmErrors * thD0KpifromBprediction = (TGraphAsymmErrors*)thfile->Get("D0Kpiprediction");
   TGraphAsymmErrors * thDpluskpipiprediction = (TGraphAsymmErrors*)thfile->Get("Dpluskpipiprediction");
   TGraphAsymmErrors * thDstarD0piprediction = (TGraphAsymmErrors*)thfile->Get("DstarD0piprediction");
+  TGraphAsymmErrors * thDsKKpiprediction = (TGraphAsymmErrors*)thfile->Get("DsKkpiprediction");
+
   thD0KpifromBprediction->SetLineColor(4);
   thD0KpifromBprediction->SetFillColor(kAzure+9);
   thDpluskpipiprediction->SetLineColor(4);
   thDpluskpipiprediction->SetFillColor(kAzure+9);
   thDstarD0piprediction->SetLineColor(4);
   thDstarD0piprediction->SetFillColor(kAzure+9);
+  thDsKKpiprediction->SetLineColor(4);
+  thDsKKpiprediction->SetFillColor(kAzure+9);
 
   //
   // Get the spectra bins & limits
@@ -349,6 +356,13 @@ void CombineFeedDownMCSubtractionMethodsUncertainties(const char *fcfilename="HF
     thDstarD0piprediction->Draw("3CA");
     thDstarD0piprediction->Draw("CX");
   }
+  else if ( decay==4 ) {
+    thDsKKpiprediction->SetLineColor(kGreen+2);
+    thDsKKpiprediction->SetLineWidth(3);
+    thDsKKpiprediction->SetFillColor(kGreen-6);
+    thDsKKpiprediction->Draw("3CA");
+    thDsKKpiprediction->Draw("CX");
+  }
   //
   gSigmaCorr->SetLineColor(kRed);
   gSigmaCorr->SetLineWidth(1);
@@ -367,6 +381,7 @@ void CombineFeedDownMCSubtractionMethodsUncertainties(const char *fcfilename="HF
   if ( decay==1 ) leg->AddEntry(thD0KpifromBprediction,"FONLL ","fl");
   else if ( decay==2 ) leg->AddEntry(thDpluskpipiprediction,"FONLL ","fl");
   else if ( decay==3 ) leg->AddEntry(thDstarD0piprediction,"FONLL ","fl");
+  else if ( decay==4 ) leg->AddEntry(thDsKKpiprediction,"FONLL ","fl");
   leg->AddEntry(histoSigmaCorr,"data stat. unc.","pl");
   leg->AddEntry(gSigmaCorr,"data syst. unc.","f");
   leg->Draw();

diff --git a/PWGHF/vertexingHF/macros/HFPtSpectrum.C b/PWGHF/vertexingHF/macros/HFPtSpectrum.C
index 2128db55343804bc1417952b229a38fe2d15f87c..903bbc60d4fa5ac09a920d9df21cec4454a71b60 100644 (file)
--- a/PWGHF/vertexingHF/macros/HFPtSpectrum.C
+++ b/PWGHF/vertexingHF/macros/HFPtSpectrum.C
@@ -1,10 +1,4 @@
-//
-// Macro to use the AliHFPtSpectrum class
-//  to compute the feed-down corrections for heavy-flavor
-//
-//  Z.Conesa, September 2010 (zconesa@in2p3.fr)
-//
-
+#if !defined(__CINT__) || defined(__MAKECINT__)
 #include <Riostream.h>
 #include "TH1D.h"
 #include "TH1.h"
@@ -17,8 +11,19 @@
 #include "TStyle.h"
 #include "TLegend.h"
 #include "AliHFSystErr.h"
-
 #include "AliHFPtSpectrum.h"
+#endif
+
+/* $Id$ */ 
+
+//
+// Macro to use the AliHFPtSpectrum class
+//  to compute the feed-down corrections for heavy-flavor
+//
+//  Z.Conesa, September 2010 (zconesa@in2p3.fr)
+//
+
+
 
 //
 // Macro execution parameters: 
@@ -50,11 +55,12 @@ void HFPtSpectrum ( const char *mcfilename="FeedDownCorrectionMC.root",
   Bool_t asym = true;
 
   // Set the meson and decay
-  // (only D0 -> K pi, D+--> K pi pi & D* --> D0 pi implemented here)
+  // (only D0 -> K pi, D+--> K pi pi & D* --> D0 pi & D+s -->KKpi implemented here)
   Bool_t isD0Kpi = true;
   Bool_t isDplusKpipi = false;
   Bool_t isDstarD0pi = false;
-  if (isD0Kpi && isDplusKpipi && isDstarD0pi) {
+  Bool_t isDsKKpi = false;
+  if (isD0Kpi && isDplusKpipi && isDstarD0pi && isDsKKpi) {
     cout << "Sorry, can not deal with more than one correction at the same time"<<endl;
     return;
   }
@@ -148,6 +154,15 @@ void HFPtSpectrum ( const char *mcfilename="FeedDownCorrectionMC.root",
     hFeedDownMCptMin = (TH1D*)mcfile->Get("hDstarD0pifromBpred_min_corr");
     //    gPrediction = (TGraphAsymmErrors*)mcfile->Get("DstarD0piprediction");
   }
+  else if (isDsKKpi){
+    decay = 4;
+    hDirectMCpt = (TH1D*)mcfile->Get("hDsKkpipred_central");
+    hFeedDownMCpt = (TH1D*)mcfile->Get("hDsKkpifromBpred_central");
+    hDirectMCptMax = (TH1D*)mcfile->Get("hDsKkpipred_max");
+    hDirectMCptMin = (TH1D*)mcfile->Get("hDsKkpipred_min");
+    hFeedDownMCptMax = (TH1D*)mcfile->Get("hDsKkpifromBpred_max");
+    hFeedDownMCptMin = (TH1D*)mcfile->Get("hDsKkpifromBpred_min");
+  }
   //
   hDirectMCpt->SetNameTitle("hDirectMCpt","direct MC spectra");
   hFeedDownMCpt->SetNameTitle("hFeedDownMCpt","feed-down MC spectra");