PredictionGenerator3Flavor.cxx

Go to the documentation of this file.

#include "3FlavorAna/Prediction/PredictionGenerator3Flavor.h"

#include "CAFAna/Extrap/ModularExtrap.h"
#include "3FlavorAna/Extrap/NueBkgdOnlyExtrap.h"
#include "3FlavorAna/Extrap/NueExtrap.h"
#include "3FlavorAna/Extrap/NueRHCExtrap.h"
#include "3FlavorAna/Extrap/NueSignalOnlyExtrap.h"
#include "3FlavorAna/Extrap/NumuExtrap.h"
#include "CAFAna/Decomp/ProportionalDecomp.h"
#include "CAFAna/Prediction/PredictionExtrap.h"
#include "CAFAna/Prediction/PredictionGenerator.h"
#include "3FlavorAna/Decomp/BENDecomp.h"
#include "3FlavorAna/Decomp/MichelDecomp.h"
#include "3FlavorAna/Decomp/NumuDecomp.h"

namespace ana
{
  //--------------------------------------------------------------------------
  //rhc
  NuePropExtrapRHCGenerator::NuePropExtrapRHCGenerator(
      const HistAxis axis,
      const HistAxis axisNDNumu,
      const Cut cutFD,
      const Cut cutNDNue,
      const Cut cutNDNumu,
      const SystShifts shiftDataND,
      const Var wei
  ) : fAxis(axis), fAxisNDNumu(axisNDNumu), fCutFD(cutFD), fCutNDNue(cutNDNue),
      fCutNDNumu(cutNDNumu), fShiftDataND(shiftDataND), fWei(wei)
  {}

  std::unique_ptr<ana::IPrediction> NuePropExtrapRHCGenerator::Generate(
      Loaders &loaders,
      const SystShifts &shiftMC
  ) const
  {
    NumuDecomp *numuDecomp = new NumuDecomp(
        loaders, fAxisNDNumu, fCutNDNumu, shiftMC, fShiftDataND, fWei);
    ProportionalDecomp *nueDecomp = new ProportionalDecomp(
        loaders, fAxis, fCutNDNue, shiftMC, fShiftDataND, fWei);
    auto extrap = std::make_unique<NueRHCExtrap>(
        loaders, *nueDecomp, *numuDecomp, fAxis, fAxisNDNumu,
        fCutFD, fCutNDNue, fCutNDNumu, shiftMC, fWei);
    return std::make_unique<PredictionExtrap>(std::move(extrap));
  }

  NueComboExtrapGenerator::NueComboExtrapGenerator(
      const HistAxis axis,
      const HistAxis axisNDNumu,
      const Cut cutFD,
      const Cut cutNDNue,
      const Cut cutNDNumu,
      const SystShifts shiftDataND,
      const Var wei,
      const EBENAnaStyle benStyle
  ) : fAxis(axis), fAxisNDNumu(axisNDNumu), fCutFD(cutFD), fCutNDNue(cutNDNue),
      fCutNDNumu(cutNDNumu), fShiftDataND(shiftDataND), fWei(wei), fBenStyle(benStyle)
  {}

  std::unique_ptr<ana::IPrediction> NueComboExtrapGenerator::Generate(
      Loaders &loaders,
      const SystShifts &shiftMC
  ) const
  {
    NumuDecomp *numuDecomp = new NumuDecomp(
        loaders, fAxisNDNumu, fCutNDNumu, shiftMC, fShiftDataND, fWei);
    BENDecomp *benDecomp = new BENDecomp(BENDecomp::kCalcKaonScale,
                                         loaders, fAxis, fCutNDNue, shiftMC, fShiftDataND, fWei);
    if (fBenStyle == kSAKaScale)
    {
      delete benDecomp;
      benDecomp = new BENDecomp(
          loaders, fAxis, fCutNDNue, shiftMC, fShiftDataND, fWei);
    }
    MichelDecomp *nueDecomp = new MichelDecomp(
        loaders, fAxis, fCutNDNue, benDecomp, shiftMC, fShiftDataND, fWei);
    auto extrap = std::make_unique<NueExtrap>(
        loaders, *nueDecomp, *numuDecomp, fAxis, fAxisNDNumu,
        fCutFD, fCutNDNue, fCutNDNumu, shiftMC, fWei);
    return std::make_unique<PredictionExtrap>(std::move(extrap));
  }

  NueSignalExtrapGenerator::NueSignalExtrapGenerator(
      const HistAxis axis,
      const HistAxis axisNDNumu,
      const Cut cutFD,
      const Cut cutNDNumu,
      const SystShifts shiftDataND,
      const Var wei
  ) : fAxis(axis), fAxisNDNumu(axisNDNumu), fCutFD(cutFD),
      fCutNDNumu(cutNDNumu), fShiftDataND(shiftDataND), fWei(wei)
  {}

  std::unique_ptr<ana::IPrediction> NueSignalExtrapGenerator::Generate(
      Loaders &loaders,
      const SystShifts &shiftMC
  ) const
  {
    NumuDecomp *numuDecomp = new NumuDecomp(
        loaders, fAxisNDNumu, fCutNDNumu, shiftMC, fShiftDataND, fWei);
    auto extrap = std::make_unique<NueSignalOnlyExtrap>(
        loaders, *numuDecomp, fAxis, fAxisNDNumu, fCutFD, fCutNDNumu, shiftMC, fWei);
    return std::make_unique<PredictionExtrap>(std::move(extrap));
  }

  NueBkgdComboExtrapGenerator::NueBkgdComboExtrapGenerator(
      const bool isRHC,
      const HistAxis axis,
      const Cut cutFD,
      const Cut cutNDNue,
      const SystShifts shiftDataND,
      const Var wei,
      const EBENAnaStyle benStyle
  ) : fIsRHC(isRHC), fAxis(axis), fCutFD(cutFD), fCutNDNue(cutNDNue),
      fShiftDataND(shiftDataND), fWei(wei), fBenStyle(benStyle)
  {}

  std::unique_ptr<ana::IPrediction> NueBkgdComboExtrapGenerator::Generate(
      Loaders &loaders,
      const SystShifts &shiftMC
  ) const
  {
    BENDecomp *benDecomp = new BENDecomp(BENDecomp::kCalcKaonScale,
                                         loaders, fAxis, fCutNDNue, shiftMC, fShiftDataND, fWei);
    if (fBenStyle == kSAKaScale)
    {
      delete benDecomp;
      benDecomp = new BENDecomp(
          loaders, fAxis, fCutNDNue, shiftMC, fShiftDataND, fWei);
    }
    MichelDecomp *nueDecomp = new MichelDecomp(
        loaders, fAxis, fCutNDNue, benDecomp, shiftMC, fShiftDataND, fWei);
    auto extrap = std::make_unique<NueBkgdOnlyExtrap>(
        fIsRHC, loaders, *nueDecomp, fAxis, fCutFD, fCutNDNue, shiftMC, fWei);
    return std::make_unique<PredictionExtrap>(std::move(extrap));
  }

  NueBkgdPropExtrapGenerator::NueBkgdPropExtrapGenerator(
      const bool isRHC,
      const HistAxis axis,
      const Cut cutFD,
      const Cut cutNDNue,
      const SystShifts shiftDataND,
      const Var wei
  ) : fIsRHC(isRHC), fAxis(axis), fCutFD(cutFD), fCutNDNue(cutNDNue),
      fShiftDataND(shiftDataND), fWei(wei)
  {}

  std::unique_ptr<ana::IPrediction> NueBkgdPropExtrapGenerator::Generate(
      Loaders &loaders,
      const SystShifts &shiftMC
  ) const
  {
    ProportionalDecomp *nueDecomp = new ProportionalDecomp(
        loaders, fAxis, fCutNDNue, shiftMC, fShiftDataND, fWei);
    auto extrap = std::make_unique<NueBkgdOnlyExtrap>(
        fIsRHC, loaders, *nueDecomp, fAxis, fCutFD, fCutNDNue, shiftMC, fWei);
    return std::make_unique<PredictionExtrap>(std::move(extrap));
  }

  NumuExtrapGenerator::NumuExtrapGenerator(
      const HistAxis axis,
      const Cut cutFD,
      const Cut cutND,
      const SystShifts shiftDataND,
      const Var wei
  ) : fAxis(axis), fCutFD(cutFD), fCutND(cutND),
      fShiftDataND(shiftDataND), fWei(wei)
  {}

  std::unique_ptr<ana::IPrediction> NumuExtrapGenerator::Generate(
      Loaders &loaders,
      const SystShifts &shiftMC
  ) const
  {
    NumuDecomp *decomp = new NumuDecomp(
        loaders, fAxis, fCutND, shiftMC, fShiftDataND, fWei);
    auto extrap = std::make_unique<NumuExtrap>(
        loaders, *decomp, fAxis, fCutFD, fCutND, shiftMC, fWei);
    return std::make_unique<PredictionExtrap>(std::move(extrap));
  }

  NuePropExtrapGenerator::NuePropExtrapGenerator(
      const HistAxis axis,
      const HistAxis axisNDNumu,
      const Cut cutFD,
      const Cut cutNDNue,
      const Cut cutNDNumu,
      const SystShifts shiftDataND,
      const Var wei
  ) : fAxis(axis), fAxisNDNumu(axisNDNumu), fCutFD(cutFD), fCutNDNue(cutNDNue),
      fCutNDNumu(cutNDNumu), fShiftDataND(shiftDataND), fWei(wei)
  {}

  std::unique_ptr<ana::IPrediction> NuePropExtrapGenerator::Generate(
      Loaders &loaders,
      const SystShifts &shiftMC
  ) const
  {
    NumuDecomp *numuDecomp = new NumuDecomp(
        loaders, fAxisNDNumu, fCutNDNumu, shiftMC, fShiftDataND, fWei);
    ProportionalDecomp *nueDecomp = new ProportionalDecomp(
        loaders, fAxis, fCutNDNue, shiftMC, fShiftDataND, fWei);
    auto extrap = std::make_unique<NueExtrap>(
        loaders, *nueDecomp, *numuDecomp, fAxis, fAxisNDNumu,
        fCutFD, fCutNDNue, fCutNDNumu, shiftMC, fWei);
    return std::make_unique<PredictionExtrap>(std::move(extrap));
  }
}