MultiExperiment.h

Go to the documentation of this file.

#pragma once

#include "sbnana/CAFAna/Experiment/IExperiment.h"

#include <memory>
#include <vector>

namespace ana
{
  /// Combine multiple component experiments
  class MultiExperiment: public IExperiment
  {
  public:
    MultiExperiment(std::vector<const IExperiment*> expts = {}) : fExpts(expts)
    {
      fSystCorrelations.resize(expts.size());
    }

    void Add(const IExperiment* expt){fExpts.push_back(expt);}

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

    /// For the subexperiment \a idx, set up a mapping between systematics
    ///
    /// Each element in the vector is a pair from a "primary" systematic to a
    /// "secondary". When this MultiExperiment is called with a primary
    /// systematic shifted, the sub-experiment will be called with the
    /// secondary systematic set to the same value (and the primary unset).
    ///
    /// You can pass NULL for a secondary to indicate that the systematic
    /// simply has no effect on the experiment in question and should be
    /// filtered out.
    void SetSystCorrelations(int idx,
                             const std::vector<std::pair<const ISyst*,
                                                         const ISyst*>>& corrs);

    virtual void SaveTo(TDirectory* dir) const override;
    static std::unique_ptr<MultiExperiment> LoadFrom(TDirectory* dir);

  protected:
    std::vector<const IExperiment*> fExpts;

    std::vector<std::vector<std::pair<const ISyst*, const ISyst*>>> fSystCorrelations;
  };
}