PredictionSystJoint2018.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Prediction/PredictionInterp.h"

namespace ana
{
  enum ENu2018ExtrapType{
    kNoExtrap,
    kProportional,
    kCombo,
    kFake,
    kNuMuNoExtrap,
    kNuMu
  };

  /// Loads shifted spectra from files
  class PredictionSystJoint2018: public PredictionInterp
  {
  public:
    /// Nue constructor
    PredictionSystJoint2018(ENu2018ExtrapType 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);
    
    /// Numu constructor
    PredictionSystJoint2018(ENu2018ExtrapType extrap,
                            osc::IOscCalc* osc,
                            const std::string beam,
                            int WhichQuant,
                            const std::vector<const ISyst*>& systs,
                            const std::string fname,
                            bool minimizeMemory = true);

    virtual ~PredictionSystJoint2018();

    /// 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 << " " << std::endl; }
      return fPreds[syst];
    };

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

  protected:
    PredictionSystJoint2018() {}

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

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

    std::string SigmaToString(int sigma) const;

    /// Interprets fBeam
    bool IsFHC() const;

    std::string fBeam;
  };
}