#include <RooDataSet.h>
#include <RooExponential.h>
#include <RooPlot.h>
#include <RooRealVar.h>
#include <TCanvas.h>

using namespace RooFit;

void rf212_plottingInRanges_blinding()
{
  // Make a fit model
  RooRealVar x("x", "The observable", 1, 30);
  RooRealVar tau("tau", "The exponent", -0.1337, -10., -0.1);
  RooExponential expo("expo", "A falling exponential function", x, tau);

  // Define the sidebands (e.g. background regions)
  x.setRange("full", 1, 30);
  x.setRange("left", 1, 10);
  x.setRange("right", 20, 30);

  // Generate toy data, and cut out the blinded region.
  std::unique_ptr<RooDataSet> data{expo.generate(x, 1000)};
  std::unique_ptr<RooAbsData> blindedData{data->reduce(CutRange("left,right"))};

  // Kick tau a bit, and run an unbinned fit where the blinded data are missing.
  // ----------------------------------------------------------------------------------------------------------
  // The fit should be done only in the unblinded regions, otherwise it would
  // try to make the model adapt to the empty bins in the blinded region.
  tau.setVal(-2.);
  expo.fitTo(*blindedData, Range("left,right"), PrintLevel(-1));

  // Clear the "fitrange" attribute of the PDF. Otherwise, the fitrange would
  // be automatically taken as the NormRange() for plotting. We want to avoid
  // this, because the point of this tutorial is to show what can go wrong when
  // the NormRange() is not specified.
  expo.removeStringAttribute("fitrange");


  // Here we will plot the results
  TCanvas *canvas=new TCanvas("canvas","canvas",800,600);
  canvas->Divide(2,1);


  // Wrong:
  // ----------------------------------------------------------------------------------------------------------
  // Plotting each slice on its own normalises the PDF over its plotting range. For the full curve, that means
  // that the blinded region where data is missing is included in the normalisation calculation. The PDF therefore
  // comes out too low, and doesn't match up with the slices in the side bands, which are normalised to "their" data.

  std::cout << "Now plotting with unique normalisation for each slice." << std::endl;
  canvas->cd(1);
  RooPlot* plotFrame = x.frame(RooFit::Title("Wrong: Each slice normalised over its plotting range"));

  // Plot only the blinded data, and then plot the PDF over the full range as well as both sidebands
  blindedData->plotOn(plotFrame);
  expo.plotOn(plotFrame, LineColor(kRed),   Range("full"));
  expo.plotOn(plotFrame, LineColor(kBlue),  Range("left"));
  expo.plotOn(plotFrame, LineColor(kGreen), Range("right"));

  plotFrame->Draw();

  // Right:
  // ----------------------------------------------------------------------------------------------------------
  // Make the same plot, but normalise each piece with respect to the regions "left" AND "right". This requires setting
  // a "NormRange", which tells RooFit over which range the PDF has to be integrated to normalise.
  // This means that the normalisation of the blue and green curves is slightly different from the left plot,
  // because they get a common scale factor.

  std::cout << "\n\nNow plotting with correct norm ranges:" << std::endl;
  canvas->cd(2);
  RooPlot* plotFrameWithNormRange = x.frame(RooFit::Title("Right: All slices have common normalisation"));

  // Plot only the blinded data, and then plot the PDF over the full range as well as both sidebands
  blindedData->plotOn(plotFrameWithNormRange);
  expo.plotOn(plotFrameWithNormRange, LineColor(kBlue),  Range("left"),  RooFit::NormRange("left,right"));
  expo.plotOn(plotFrameWithNormRange, LineColor(kGreen), Range("right"), RooFit::NormRange("left,right"));
  expo.plotOn(plotFrameWithNormRange, LineColor(kRed),   Range("full"),  RooFit::NormRange("left,right"), LineStyle(10));

  plotFrameWithNormRange->Draw();

  canvas->Draw();

}