PredictionSyst3Flavor2020.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Prediction/PredictionInterp.h"

namespace ana
{
  enum class ENu2020ExtrapType
  {
    kNoExtrap,
    kProportional,
    kCombo,
    kPropPtExtrap,
    kComboPtExtrap,
    kFake,
    kNuMuNoExtrap,
    kNuMu,
    kNuMuPtExtrap
  };

  /// Loads shifted spectra from files
  class PredictionSyst3Flavor2020: public PredictionInterp
  {
  public:
    /// Nue constructor
    PredictionSyst3Flavor2020(ENu2020ExtrapType extrap,
                              osc::IOscCalc* osc,
                              const std::vector<const ISyst*>& systs,
                              std::string beam = "fhc",
                              bool isFakeData = false,
                              bool mergePeripheral = true,
                              const std::string fname = "",
                              bool minimizeMemory = true,
                              bool NERSC = false);
    
    /// Numu constructor
    PredictionSyst3Flavor2020(ENu2020ExtrapType extrap,
                              osc::IOscCalc* osc,
                              const std::string beam,
                              const int EhadFracQuant,
                              const std::vector<const ISyst*>& systs,
                              bool isFakeData = false,
                              const std::string fname = "",
                              bool minimizeMemory = true,
                              bool NERSC = false);

    virtual ~PredictionSyst3Flavor2020();

    /// Method to expose fPredNom
    ///
    /// \returns fPredNom Unique ptr to nominal prediction
    IPrediction* GetNominalPrediction()
    {
      //std::cout << "GETNOMPRED " << fPredNom.get() << std::endl;
      return fPredNom.get();
    };

    /// Method to expose fPreds
    ///
    /// \returns shifted prediction (ShiftedPreds) from fPreds
    const ShiftedPreds& GetShiftedPrediction(const ISyst* syst)
    {
      //      for (auto& shift:fPreds[syst].preds) { std::cout << "\t Func shift: " << shift.get() << " " << std::endl; }
      return fPreds[syst];
    };

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

  protected:
    PredictionSyst3Flavor2020() {}

    /// Must set fBeam first
    void AddNormSysts(const std::vector<const ISyst*>& systs, ENu2020ExtrapType extrap);

    void AddNormSyst(const ISyst* syst,
                     const std::vector<const ISyst*>& systs,
                     double fracNueApp, double fracOthers);

    std::string SigmaToString(int sigma) const;

    IPrediction* LoadNuePtExtrapPred( TFile* file, ENu2020ExtrapType extrap, std::string syst_name, std::string sigma_str ) const;

    IPrediction* LoadNumuPtExtrapPred( TFile* file, int EhadFracQuant, std::string syst_name, std::string sigma_str ) const;

    /// Interprets fBeam
    bool IsFHC() const;

    std::string fBeam;
  };
}