PredictionGenerator.h

Go to the documentation of this file.

#pragma once

#include <memory>

#include "CAFAna/Core/Cut.h"
#include "CAFAna/Core/FwdDeclare.h"
#include "CAFAna/Core/HistAxis.h"
#include "CAFAna/Core/Var.h"
#include "CAFAna/Vars/HistAxes.h"
#include "CAFAna/Core/SystShifts.h"

namespace ana
{
  class Loaders;
  class IPrediction;

  /** \brief Given loaders and an MC shift, Generate() generates an IPrediction

      All other arguments needed to construct the prediction are passed to the
      IPredictionGenerator constructor, and are identical for all generated
      predictions. For standalone use or use with PredictionInterp. **/
  class IPredictionGenerator
  {
    public:
    virtual std::unique_ptr<IPrediction> Generate(
      Loaders& loaders, const SystShifts& shiftMC = kNoShift ) const = 0;
  };

  //---------------------------------------------------------------------------

  /// Generates FD-only predictions (no extrapolation)
  class NoExtrapGenerator: public IPredictionGenerator
  {
    public:
      NoExtrapGenerator(
          const HistAxis axis,
          const Cut cut,
          const Var wei = kUnweighted );
      std::unique_ptr<IPrediction> Generate(
          Loaders& loaders,
          const SystShifts& shiftMC = kNoShift ) const override;
    private:
      const HistAxis fAxis;
      const Cut fCut;
      const Var fWei;
  };

  //.......................................................................................


}