Implement Oleksii's comments

Author: pstahlhu

PWGHF/D2H/Macros/HFInvMassFitter.cxx

@@ -381,7 +381,7 @@ void HFInvMassFitter::doFit()
     countSignal(mRawYieldCounted, mRawYieldCountedErr);
     calculateSignificance(mSignificance, mSignificanceErr);
     // Fit to data ratio
-    mRatioFrame = mass->frame(Title("Fit/Data Ratio"));
+    mRatioFrame = mass->frame(Title(Form("%s", mHistoInvMass->GetTitle())));
     calculateFitToDataRatio();
   }
 }
@@ -669,21 +669,9 @@ void HFInvMassFitter::drawResidual(TVirtualPad* pad)
 void HFInvMassFitter::drawRatio(TVirtualPad* pad)
 {
   pad->cd();
-  mRatioFrame->GetYaxis()->SetTitle("");
-  TPaveText* textInfo = new TPaveText(0.12, 0.65, 0.47, .89, "NDC");
-  textInfo->SetBorderSize(0);
-  textInfo->SetFillStyle(0);
-  textInfo->SetTextColor(kBlue);
-  textInfo->AddText(Form("S = %.0f #pm %.0f ", mRawYield, mRawYieldErr));
-  textInfo->AddText(Form("S_{count} = %.0f #pm %.0f ", mRawYieldCounted, mRawYieldCountedErr));
-  textInfo->AddText(Form("mean = %.3f #pm %.3f", mRooMeanSgn->getVal(), mRooMeanSgn->getError()));
-  if (mTypeOfSgnPdf == DoubleGaus) {
-    textInfo->AddText(Form("sigma = %.3f #pm %.3f", mRooSigmaSgn->getVal(), mRooSigmaSgn->getError()));
-    textInfo->AddText(Form("sigma 2 = %.3f #pm %.3f", mRooSecSigmaSgn->getVal(), mRooSecSigmaSgn->getError()));
-  } else {
-    textInfo->AddText(Form("sigma = %.3f #pm %.3f", mRooSigmaSgn->getVal(), mRooSigmaSgn->getError()));
-  }
-  mRatioFrame->addObject(textInfo);
+  mRatioFrame->GetXaxis()->SetTitleOffset(1.2);
+  mRatioFrame->GetYaxis()->SetTitleOffset(1.5);
+  mRatioFrame->GetYaxis()->SetTitle("Fit / Data");
   double xMin = mRatioFrame->GetXaxis()->GetXmin();
   double xMax = mRatioFrame->GetXaxis()->GetXmax();
   TLine* line = new TLine(xMin, 1.0, xMax, 1.0);
@@ -883,7 +871,7 @@ void HFInvMassFitter::plotRefl(RooAbsPdf* pdf)
 }
 
 // Calculate fit to data ratio
-void HFInvMassFitter::calculateFitToDataRatio()
+void HFInvMassFitter::calculateFitToDataRatio() const
 {
   if (!mInvMassFrame)
     return;
@@ -911,9 +899,9 @@ void HFInvMassFitter::calculateFitToDataRatio()
     ratioHist->SetPointError(i, 0, 0, err, err);
   }
 
+  mRatioFrame->addPlotable(ratioHist, "P");
   mRatioFrame->SetMinimum(0.5);
   mRatioFrame->SetMaximum(1.5);
-  mRatioFrame->addPlotable(ratioHist, "P");
 }
 
 // Fix reflection pdf

PWGHF/D2H/Macros/HFInvMassFitter.h

@@ -229,7 +229,7 @@ class HFInvMassFitter : public TNamed
   void calculateBackground(Double_t& bkg, Double_t& bkgErr) const;
   void calculateSignificance(Double_t& significance, Double_t& significanceErr) const;
   void checkForSignal(Double_t& estimatedSignal);
-  void calculateFitToDataRatio();
+  void calculateFitToDataRatio() const;
   void drawFit(TVirtualPad* c, const std::vector<std::string>& plotLabels, Bool_t writeParInfo = true);
   void drawResidual(TVirtualPad* c);
   void drawRatio(TVirtualPad* c);

PWGHF/D2H/Macros/config_massfitter.json

@@ -1,5 +1,7 @@
 {
     "IsMC": false,
+    "WriteSignalPar": true,
+    "_WriteSignalPar": "write signal parameter values on mass canvas",
     "InFileName": "hMasses.root",
     "_InFileName": "download example file here https://cernbox.cern.ch/s/uoWicuRAVGArDsV",
     "ReflFileName": "",
@@ -35,6 +37,8 @@
         "Dstar",
         "XicToXiPiPi"
     ],
+    "CollisionSystem": "pp #sqrt{#it{s}} = 13.6 TeV",
+    "_CollisionSystems": "Label for mass canvases",
     "EnableRefl": false,
     "FixSigma": false,
     "SigmaFile": "",

PWGHF/D2H/Macros/runMassFitter.C

@@ -651,6 +651,8 @@ int runMassFitter(const TString& configFileName)
   outputFileName.ReplaceAll(".root", ".pdf");
   TString outputFileNameResidual = outputFileName;
   outputFileNameResidual.ReplaceAll(".pdf", "_Residuals.pdf");
+  TString outputFileRatio = outputFileName;
+  outputFileRatio.ReplaceAll(".pdf", "_Ratio.pdf");
   for (int iCanvas = 0; iCanvas < nCanvases; iCanvas++) {
     if (iCanvas == 0 && nCanvases > 1) {
       canvasMass[iCanvas]->SaveAs(Form("%s[", outputFileName.Data()));
@@ -660,13 +662,22 @@ int runMassFitter(const TString& configFileName)
       canvasMass[iCanvas]->SaveAs(Form("%s]", outputFileName.Data()));
     }
     if (!isMc) {
+      // residuals
       if (iCanvas == 0 && nCanvases > 1) {
         canvasResiduals[iCanvas]->SaveAs(Form("%s[", outputFileNameResidual.Data()));
       }
       canvasResiduals[iCanvas]->SaveAs(outputFileNameResidual.Data());
       if (iCanvas == nCanvases - 1 && nCanvases > 1) {
         canvasResiduals[iCanvas]->SaveAs(Form("%s]", outputFileNameResidual.Data()));
       }
+      // ratio
+      if (iCanvas == 0 && nCanvases > 1) {
+        canvasRatio[iCanvas]->SaveAs(Form("%s[", outputFileRatio.Data()));
+      }
+      canvasRatio[iCanvas]->SaveAs(outputFileRatio.Data());
+      if (iCanvas == nCanvases - 1 && nCanvases > 1) {
+        canvasRatio[iCanvas]->SaveAs(Form("%s]", outputFileRatio.Data()));
+      }
     }
   }
   return 0;