TrivialExtrap.cxx

Go to the documentation of this file.

#include "CAFAna/Extrap/TrivialExtrap.h"

#include "CAFAna/Cuts/TruthCuts.h"
#include "CAFAna/Core/LoadFromRegistry.h"
#include "CAFAna/Core/Loaders.h"

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

namespace ana
{
  REGISTER_LOADFROM("TrivialExtrap", IExtrap, TrivialExtrap);

  //----------------------------------------------------------------------
  TrivialExtrap::TrivialExtrap(Loaders& loaders,
                               std::string label,
                               const Binning& bins,
                               const Var& var,
                               const Cut& cut,
                               const SystShifts& shift,
                               const Var& wei)
    : TrivialExtrap(loaders, HistAxis(label, bins, var), cut, shift, wei)
  {
  }

  //----------------------------------------------------------------------
  TrivialExtrap::TrivialExtrap(Loaders& loaders,
                               const HistAxis& axis,
                               const Cut& cut,
                               const SystShifts& shift,
                               const Var& wei)
    : TrivialExtrap(loaders.GetLoader(caf::kFARDET, Loaders::kMC, ana::kBeam, Loaders::kNonSwap),
                    loaders.GetLoader(caf::kFARDET, Loaders::kMC, ana::kBeam, Loaders::kFluxSwap),
                    loaders.GetLoader(caf::kFARDET, Loaders::kMC, ana::kBeam, Loaders::kTauSwap),
                    axis, cut, shift, wei)
  {
  }

  //----------------------------------------------------------------------
  TrivialExtrap::TrivialExtrap(SpectrumLoaderBase& loader,
                               SpectrumLoaderBase& loaderSwap,
                               SpectrumLoaderBase& loaderTau,
                               const HistAxis& axis,
                               const Cut& cut,
                               const SystShifts& shift,
                               const Var& wei) :
    fNueApp       (loaderSwap, axis, cut && kIsSig       && !kIsAntiNu, shift, wei),
    fNueAppAnti   (loaderSwap, axis, cut && kIsSig       &&  kIsAntiNu, shift, wei),
    fNumuSurv     (loader,     axis, cut && kIsNumuCC    && !kIsAntiNu, shift, wei),
    fNumuSurvAnti (loader,     axis, cut && kIsNumuCC    &&  kIsAntiNu, shift, wei),
    fNumuApp      (loaderSwap, axis, cut && kIsNumuApp   && !kIsAntiNu, shift, wei),
    fNumuAppAnti  (loaderSwap, axis, cut && kIsNumuApp   &&  kIsAntiNu, shift, wei),
    fNueSurv      (loader,     axis, cut && kIsBeamNue   && !kIsAntiNu, shift, wei),
    fNueSurvAnti  (loader,     axis, cut && kIsBeamNue   &&  kIsAntiNu, shift, wei),
    fTauFromE     (loaderTau,  axis, cut && kIsTauFromE  && !kIsAntiNu, shift, wei),
    fTauFromEAnti (loaderTau,  axis, cut && kIsTauFromE  &&  kIsAntiNu, shift, wei),
    fTauFromMu    (loaderTau,  axis, cut && kIsTauFromMu && !kIsAntiNu, shift, wei),
    fTauFromMuAnti(loaderTau,  axis, cut && kIsTauFromMu &&  kIsAntiNu, shift, wei),
    fNCTot          (loader,   axis, cut && kIsNC,                      shift, wei),
    fNC             (loader,   axis, cut && kIsNC        && !kIsAntiNu, shift, wei),
    fNCAnti         (loader,   axis, cut && kIsNC        && kIsAntiNu,  shift, wei)
  {
    // Swapped files are just as valid a source of NCs as unswapped. This
    // approximately doubles our statistics. SpectrumLoader just adds events
    // and POT for both cases, which is the right thing to do.
    loaderSwap.AddSpectrum(fNCTot, axis.GetVar1D(), cut && kIsNC, shift, wei);
    loaderSwap.AddSpectrum(fNC,    axis.GetVar1D(), cut && kIsNC && !kIsAntiNu, shift, wei);
    loaderSwap.AddSpectrum(fNCAnti,axis.GetVar1D(), cut && kIsNC && kIsAntiNu,  shift, wei);
    loaderTau .AddSpectrum(fNCTot, axis.GetVar1D(), cut && kIsNC, shift, wei);
    loaderTau .AddSpectrum(fNC,    axis.GetVar1D(), cut && kIsNC && !kIsAntiNu, shift, wei);
    loaderTau .AddSpectrum(fNCAnti,axis.GetVar1D(), cut && kIsNC && kIsAntiNu,  shift, wei);
  }

  //----------------------------------------------------------------------
  TrivialExtrap::TrivialExtrap(SpectrumLoaderBase& loader,
                               SpectrumLoaderBase& loaderSwap,
                               SpectrumLoaderBase& loaderTau,
                               std::string label,
                               const Binning& bins,
                               const Var& var,
                               const Cut& cut,
                               const SystShifts& shift,
                               const Var& wei) :
    TrivialExtrap(loader, loaderSwap, loaderTau,
                  HistAxis(label, bins, var),
                  cut, shift, wei)
  {
  }

  //----------------------------------------------------------------------
  TrivialExtrap::TrivialExtrap(SpectrumLoaderBase& loader,
                               SpectrumLoaderBase& loaderSwap,
                               std::string label,
                               const Binning& bins,
                               const Var& var,
                               const Cut& cut,
                               const SystShifts& shift,
                               const Var& wei)
    : TrivialExtrap(loader, loaderSwap, kNullLoader,
                    label, bins, var, cut, shift, wei)
  {
  }

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

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

    TObjString("TrivialExtrap").Write("type");

    fNueApp.SaveTo(dir, "nue_app");
    fNueAppAnti.SaveTo(dir, "nue_app_anti");
    fNCTot.SaveTo(dir, "nc_tot");
    fNC.SaveTo(dir, "nc");
    fNCAnti.SaveTo(dir, "nc_anti");
    fNumuSurv.SaveTo(dir, "numu_surv");
    fNumuSurvAnti.SaveTo(dir, "numu_surv_anti");
    fNumuApp.SaveTo(dir, "numu_app");
    fNumuAppAnti.SaveTo(dir, "numu_app_anti");
    fNueSurv.SaveTo(dir, "nue_surv");
    fNueSurvAnti.SaveTo(dir, "nue_surv_anti");
    fTauFromE.SaveTo(dir, "nutau_from_nue");
    fTauFromEAnti.SaveTo(dir, "nutau_from_nue_anti");
    fTauFromMu.SaveTo(dir, "nutau_from_numu");
    fTauFromMuAnti.SaveTo(dir, "nutau_from_numu_anti");

    dir->Write();
    delete dir;

    tmp->cd();
  }

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

    std::unique_ptr<TrivialExtrap> ret(new TrivialExtrap);

    ret->fNueApp        = *OscillatableSpectrum::LoadFrom(dir, "nue_app");
    ret->fNueAppAnti    = *OscillatableSpectrum::LoadFrom(dir, "nue_app_anti");
    ret->fNumuSurv      = *OscillatableSpectrum::LoadFrom(dir, "numu_surv");
    ret->fNumuSurvAnti  = *OscillatableSpectrum::LoadFrom(dir, "numu_surv_anti");
    ret->fNumuApp       = *OscillatableSpectrum::LoadFrom(dir, "numu_app");
    ret->fNumuAppAnti   = *OscillatableSpectrum::LoadFrom(dir, "numu_app_anti");
    ret->fNueSurv       = *OscillatableSpectrum::LoadFrom(dir, "nue_surv");
    ret->fNueSurvAnti   = *OscillatableSpectrum::LoadFrom(dir, "nue_surv_anti");
    ret->fTauFromE      = *OscillatableSpectrum::LoadFrom(dir, "nutau_from_nue");
    ret->fTauFromEAnti  = *OscillatableSpectrum::LoadFrom(dir, "nutau_from_nue_anti");
    ret->fTauFromMu     = *OscillatableSpectrum::LoadFrom(dir, "nutau_from_numu");
    ret->fTauFromMuAnti = *OscillatableSpectrum::LoadFrom(dir, "nutau_from_numu_anti");

    ret->fNCTot  = *Spectrum::LoadFrom(dir, "nc_tot");
    ret->fNC     = *Spectrum::LoadFrom(dir, "nc");
    ret->fNCAnti = *Spectrum::LoadFrom(dir, "nc_anti");

    delete dir;

    return ret;
  }
}