PredictionXSecTuning.h

Go to the documentation of this file.

#pragma once

#include "CAFAna/Prediction/IPrediction.h"
#include "CAFAna/Prediction/PredictionGenerator.h"

namespace ana
{
  class Loaders;

  /// Prediction that wraps a simple Spectrum
  class PredictionXSecTuning: public IPrediction
  {
  public:
    PredictionXSecTuning( Loaders& loaders,
                          const HistAxis& axis,
                          const Cut& cut,
                          const SystShifts& shift = kNoShift,
                          const Var& wei = kUnweighted);

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

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

    virtual Spectrum Predict( osc::IOscCalc*  /*calc*/ ) const override
    {
      return fSpectrum;
    }

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

  protected:
    PredictionXSecTuning( const Spectrum& s )
      : fSpectrum( s )
    {}

    Spectrum fSpectrum;
  };

  class XSecTuningPredGenerator: public IPredictionGenerator
  {
  public:
    XSecTuningPredGenerator( const HistAxis axis,
                             const Cut cut,
                             const Var wei = kUnweighted )
      : fAxis( axis ), fCut( cut ), fWei( wei )
    {
    }

    virtual std::unique_ptr<IPrediction>
    Generate( Loaders& loaders, const SystShifts& shiftMC = kNoShift ) const override
    {
      return std::unique_ptr<IPrediction>( new PredictionXSecTuning( 
                                           loaders, fAxis, fCut, shiftMC, fWei ) );
    }
  protected:
    const HistAxis fAxis;
    const Cut fCut;
    const Var fWei;
  };

}