PredictionGeneratorNuX.cxx

Go to the documentation of this file.

#include "NuXAna/Prediction/PredictionGeneratorNuX.h"

#include "CAFAna/Extrap/FDExtrap.h"
#include "CAFAna/Extrap/NDExtrap.h"
#include "NuXAna/Prediction/FDPredictionSterile.h"
#include "NuXAna/Prediction/NDPredictionSterile.h"
#include "NuXAna/Prediction/PredictionSterile.h"
#include "NuXAna/Extrap/ExtrapSterile.h"
#include "CAFAna/Decomp/ProportionalDecomp.h"
#include "CAFAna/Prediction/PredictionGenerator.h"

namespace ana
{

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

  std::unique_ptr<IPrediction> SterileGenerator::Generate(
    Loaders& loaders,
    const SystShifts& shiftMC
  ) const {
    ProportionalDecomp* decompNumu = new ProportionalDecomp(
      loaders, fAxisNDNumu, fCutNDNumu, shiftMC, fShiftDataND, fWei );
    ProportionalDecomp* decompNC = new ProportionalDecomp(
      loaders, fAxis, fCutNDNC, shiftMC, fShiftDataND, fWei );
    auto extrap = std::make_unique<ModularExtrapSterile>(ModularExtrapSterile::NCDisappearance(
      loaders, *decompNC, *decompNumu, fAxis, fAxisNDNumu,
      fCutFD, fCutNDNC, fCutNDNumu, shiftMC, fWei ));
    return std::make_unique<PredictionSterile>( std::move(extrap) );
  }

  std::unique_ptr<IPrediction> SterileGenerator::Generate(
    Loaders& loaders_nd,
    Loaders& loaders_fd,
    const SystShifts& shiftMC
  ) const {
    ProportionalDecomp* decompNumu = new ProportionalDecomp(
      loaders_nd, fAxisNDNumu, fCutNDNumu, shiftMC, fShiftDataND, fWei );
    ProportionalDecomp* decompNC = new ProportionalDecomp(
      loaders_nd, fAxis, fCutNDNC, shiftMC, fShiftDataND, fWei );
    auto extrap = std::make_unique<ModularExtrapSterile>(ModularExtrapSterile::NCDisappearance(
      loaders_nd.GetLoader(caf::kNEARDET, Loaders::kMC, kBeam, Loaders::kNonSwap),
      loaders_fd.GetLoader(caf::kFARDET,  Loaders::kMC, kBeam, Loaders::kFluxSwap),
      loaders_fd.GetLoader(caf::kFARDET,  Loaders::kMC, kBeam, Loaders::kNonSwap),
      loaders_fd.GetLoader(caf::kFARDET,  Loaders::kMC, kBeam, Loaders::kTauSwap),
      *decompNC, *decompNumu, fAxis, fAxisNDNumu,
      fCutFD, fCutNDNC, fCutNDNumu, shiftMC, fWei));
    return std::make_unique<PredictionSterile>( std::move(extrap) );
  }

  //..........................................................................
  NDPredictionGenerator::NDPredictionGenerator(
      const HistAxis axis,
      const Cut cutND,
      const SystShifts shiftND,
      const Var wei
      //const std::string anaLabel
  ) : fAxis(axis), fCutND(cutND),
      fShiftND(shiftND), fWei(wei) {}

  std::unique_ptr<IPrediction> NDPredictionGenerator::Generate(
      Loaders& loaders,
      const SystShifts& shiftMC  ) const
  {
    NDExtrap* extrap = new NDExtrap(NDExtrap::NDExtrap_c(
        loaders, fAxis, fCutND, shiftMC, fWei));
    return std::make_unique<NDPredictionSterile>(extrap);
  }

  //............................................................................
  FDPredictionGenerator::FDPredictionGenerator(
      const HistAxis axis,
      const Cut cutFD,
      const SystShifts shiftFD,
      const Var wei
  ) : fAxis(axis), fCutFD(cutFD),
      fShiftFD(shiftFD), fWei(wei) {}

  std::unique_ptr<IPrediction> FDPredictionGenerator::Generate(
      Loaders& loaders,
      const SystShifts& shiftMC  ) const
  {
    FDExtrap* extrap = new FDExtrap(FDExtrap::FDExtrap_c(
        loaders, fAxis, fCutFD, shiftMC, fWei));
    return std::make_unique<FDPredictionSterile>(extrap);
  }

}