PredictionCombinePeriods.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Prediction/IPrediction.h"

namespace ana
{
  /// \brief Sum MC predictions from different periods scaled according to data POT targets
  ///
  /// Be sure you understand how putting one of these in a
  /// SingleSampleExperiment differs from putting regular predictions for the
  /// different periods into a MultiExperiment. This calculates
  /// chisq(data1+data2|mc1+mc2), whereas the MultiExperiment is
  /// chisq(data1|mc1)+chisq(data2+mc2).
  class PredictionCombinePeriods: public IPrediction
  {
  public:
    /// \param preds List of predictions to combine, along with the POT target
    ///              for that period. Should be like:
    ///              {{pred1, data1.POT()}, {pred2, data2.POT()}}
    PredictionCombinePeriods(const std::vector<std::pair<const IPrediction*, double>>& preds);
    ~PredictionCombinePeriods();

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

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

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

    Spectrum PredictComponentSyst(osc::IOscCalc* 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
    void SaveTo(TDirectory* dir, const std::string& name) const override;

    static std::unique_ptr<PredictionCombinePeriods> LoadFrom(TDirectory* dir, const std::string& name);


    unsigned int NPredictions() const {return fPreds.size();}
    const IPrediction* GetPrediction(int i) const {return fPreds[i].first;}
    double GetPOT(int i) const {return fPreds[i].second;}

  protected:
    std::vector<std::pair<const IPrediction*, double>> fPreds;
  };
}