PredictionNoOsc.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Prediction/IPrediction.h"
#include "CAFAna/Prediction/PredictionGenerator.h"

namespace ana
{
  class Loaders;

  /// Prediction that wraps a simple Spectrum
  class PredictionNoOsc: public IPrediction
  {
  public:
    PredictionNoOsc(SpectrumLoaderBase& loader,
                    const HistAxis& axis,
                    const Cut& cut,
                    const SystShifts& shift = kNoShift,
                    const Var& wei = kUnweighted);

    PredictionNoOsc(SpectrumLoaderBase& loader,
                    const std::string& label,
                    const Binning& bins,
                    const Var& var,
                    const Cut& cut,
                    const SystShifts& shift = kNoShift,
                    const Var& wei = kUnweighted);

    static std::unique_ptr<PredictionNoOsc> LoadFrom(TDirectory* dir, const std::string& name);
    virtual void SaveTo(TDirectory* dir, const std::string& name) const override;

    // un-hide inherited method stubs so we don't get warnings from the compiler
    using IPrediction::Predict;
    using IPrediction::PredictComponent;
    using IPrediction::PredictSyst;


    virtual Spectrum Predict(osc::IOscCalc* /*calc*/) const override
    {
      return fSpectrum;
    }

    virtual Spectrum PredictComponent(osc::IOscCalc* calc,
                                      Flavors::Flavors_t flav,
                                      Current::Current_t curr,
                                      Sign::Sign_t sign) const override;

  protected:
    PredictionNoOsc(const Spectrum& s,
                    const Spectrum& sNC,
                    const Spectrum& sNumu, const Spectrum& sNumubar,
                    const Spectrum& sNue, const Spectrum& sNuebar)
      : fSpectrum(s),
        fSpectrumNC(sNC),
        fSpectrumNumu(sNumu), fSpectrumNumubar(sNumubar),
        fSpectrumNue(sNue), fSpectrumNuebar(sNuebar)
    {
    }

    Spectrum fSpectrum;

    Spectrum fSpectrumNC;
    Spectrum fSpectrumNumu;
    Spectrum fSpectrumNumubar;
    Spectrum fSpectrumNue;
    Spectrum fSpectrumNuebar;
  };


  class NoOscPredictionGenerator: public IPredictionGenerator
  {
  public:
    NoOscPredictionGenerator(SpectrumLoaderBase& loader,
                             HistAxis axis,
                             Cut cut,
                             Var wei)
      : fLoader(loader), fAxis(axis), fCut(cut), fWei(wei)
    {
    }

    virtual std::unique_ptr<IPrediction>
    Generate(Loaders& loaders, const SystShifts& shiftMC = kNoShift) const override
    {
      return std::unique_ptr<IPrediction>(new PredictionNoOsc(fLoader,
                                                              fAxis,
                                                              fCut,
                                                              shiftMC,
                                                              fWei));
    }
  protected:
    SpectrumLoaderBase& fLoader;
    HistAxis fAxis;
    Cut fCut;
    Var fWei;
  };

}