PredictionAddRock.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Prediction/IPrediction.h"

#include "CAFAna/Core/FwdDeclare.h"
#include "CAFAna/Core/ISyst.h"

namespace ana
{
  class PredictionNoExtrap;

  class PredictionAddRock : public IPrediction
  {
  public:
    /// Takes ownership of \a predCore and \a predNoExtrap
    PredictionAddRock(IPrediction        *predFid,
                      PredictionNoExtrap *predRock);
    virtual ~PredictionAddRock();

    virtual Spectrum Predict(osc::IOscCalc* calc) const override;
    virtual Spectrum Predict(osc::IOscCalcStan* calc) const override;
    virtual Spectrum PredictSyst(osc::IOscCalc* calc,
                                 const SystShifts& syst) const override;
    virtual Spectrum PredictSyst(osc::IOscCalcStan* calc,
                                 const SystShifts& syst) const override;

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

    OscillatableSpectrum ComponentCC(int from, int to) const override;
    //nc
    Spectrum ComponentNCTotal() const override;
    Spectrum ComponentNC()      const override;
    Spectrum ComponentNCAnti()  const override;
    //end nc

    const IPrediction*        GetFiducialPred() const { return fPredFid; }
    const PredictionNoExtrap* GetRockPred()     const { return fPredRock; }

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

    PredictionAddRock() = delete;

  protected:
    template <typename T>
    Spectrum _PredictComponentSyst(osc::_IOscCalc<T>* calc,
                                   const SystShifts& syst,
                                   Flavors::Flavors_t flav,
                                   Current::Current_t curr,
                                   Sign::Sign_t sign) const;


    template <typename T>
    Spectrum RockComponent(osc::_IOscCalc<T>* calc,
                           Flavors::Flavors_t flav,
                           Current::Current_t curr,
                           Sign::Sign_t sign) const;

    IPrediction*        fPredFid;
    PredictionNoExtrap* fPredRock;

  };

}