SingleSampleExperiment.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Experiment/IExperiment.h"
#include "CAFAna/Prediction/IPrediction.h"
#include "CAFAna/Core/Spectrum.h"
#include "CAFAna/Core/ISyst.h"

namespace ana
{
  /// Dummy syst to communicate with \ref SingleSampleExperiment
  class CosmicBkgScaleSyst: public ISyst
  {
  public:
    CosmicBkgScaleSyst() : ISyst("cosmicScale", "Cosmic background scale") {}
    void Shift(double, caf::SRProxy*, double&) const {}
  };

  extern const CosmicBkgScaleSyst kCosmicBkgScaleSyst;

  /// Compare a single data spectrum to the MC + cosmics expectation
  class SingleSampleExperiment: public IExperiment
  {
  public:
    /// \param pred   Source of oscillated MC beam predictions
    /// \param data   Data spectrum to compare to
    /// \param cosmic Cosmic ray background component
    /// \param cosmicScaleError fractional uncertainty on cosmic normalization
    SingleSampleExperiment(const IPrediction* pred,
                           const Spectrum& data,
                           const Spectrum& cosmic,
                           double cosmicScaleError = 0,
                           bool PoissonError = false);

    /// In MC studies you might not want to bother with cosmics
    SingleSampleExperiment(const IPrediction* pred,
                           const Spectrum& data)
      : fMC(pred), fData(data)
    {
    }

    virtual ~SingleSampleExperiment();

    virtual double ChiSq(osc::IOscCalcAdjustable* osc,
                         const SystShifts& syst = SystShifts::Nominal()) const override;

    const IPrediction * GetPrediction() const { return fMC; }

    virtual stan::math::var LogLikelihood(osc::IOscCalcAdjustableStan* osc,
                                          const SystShifts& syst = kNoShift) const override;

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

    // Didn't make provisions for copying fCosmic or fMC
    SingleSampleExperiment(const SingleSampleExperiment&) = delete;
    SingleSampleExperiment& operator=(const SingleSampleExperiment&) = delete;

    // need to explicitly declare move constructor since copy constructor is deleted
    SingleSampleExperiment(SingleSampleExperiment&& s)
      : fMC(s.fMC), fData(std::move(s.fData)), fCosmic(std::move(s.fCosmic)), fCosmicScaleError(s.fCosmicScaleError)
    {
      s.fMC = nullptr;
    };

  protected:
    template <typename T>
    Eigen::Array<T, Eigen::Dynamic, 1> PredHistIncCosmics(osc::_IOscCalc<T>* calc, const SystShifts& syst) const;

    const IPrediction* fMC;
    Spectrum fData;
    std::optional<Spectrum> fCosmic;
    double fCosmicScaleError;
    bool fPoissonError;
    std::vector < std::pair <double, double> > fBinErrors;
  };
}