PredictionXSecTuning.cxx

Go to the documentation of this file.

#include "CAFAna/Prediction/PredictionXSecTuning.h"

#include "CAFAna/Extrap/IExtrap.h"
#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Core/LoadFromRegistry.h"

#include "CAFAna/Core/Loaders.h"
#include "CAFAna/Extrap/TrivialExtrap.h"

#include "CAFAna/Cuts/TruthCuts.h"

#include "TDirectory.h"
#include "TObjString.h"


namespace ana
{
  REGISTER_LOADFROM("PredictionXSecTuning", IPrediction, PredictionXSecTuning);

  //----------------------------------------------------------------------
  PredictionXSecTuning::PredictionXSecTuning( Loaders& loaders,
                                              const HistAxis& axis,
                                              const Cut& cut,
                                              const SystShifts& shift,
                                              const Var& wei )
    : fSpectrum( loaders.GetLoader( caf::kNEARDET, Loaders::kMC ),
                 axis, cut, shift, wei )
  {
  }

  //----------------------------------------------------------------------
  Spectrum PredictionXSecTuning::PredictComponent( osc::IOscCalc* /*calc*/,
                                                   Flavors::Flavors_t flav,
                                                   Current::Current_t curr,
                                                   Sign::Sign_t sign) const
  {
    // basically the whole prediction is regarded as being one component
    if (flav == ana::Flavors::kNuMuToNuMu && curr == ana::Current::kCC && sign == ana::Sign::kBoth)
    {
      return Predict((osc::IOscCalc*)(nullptr)); // Faster
    }
    else
    {
      auto spec = ana::Spectrum( fSpectrum );
      spec.Clear();
      return std::move( spec );
    }
  }

  //----------------------------------------------------------------------
  void PredictionXSecTuning::SaveTo( TDirectory* dir, const std::string& name ) const
  {
    TDirectory* tmp = gDirectory;

    dir = dir->mkdir(name.c_str()); // switch to subdir
    dir->cd();

    TObjString( "PredictionXSecTuning" ).Write( "type");
    fSpectrum.SaveTo( dir, "spec" );

    dir->Write();
    delete dir;

    tmp->cd();
  }

  //----------------------------------------------------------------------
  std::unique_ptr<PredictionXSecTuning> PredictionXSecTuning::LoadFrom( TDirectory* dir, const std::string& name )
  {
    dir = dir->GetDirectory(name.c_str()); // switch to subdir
    assert(dir);

    PredictionXSecTuning* ret = new PredictionXSecTuning(*ana::LoadFrom<Spectrum>( dir, "spec" ) );

    delete dir;

    // Can't use make_unique because constructor is protected
    return std::unique_ptr<PredictionXSecTuning>( ret );
  }
}