NDExtrapComponent.cxx

Go to the documentation of this file.

#include "CAFAna/Extrap/NDExtrapComponent.h"
#include "CAFAna/Extrap/NDExtrapBeamComponent.h"

#include <utility>

#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/Core/Loaders.h"
#include "CAFAna/Core/SpectrumLoaderBase.h"
#include "CAFAna/Cuts/Cuts.h"

#include "TDirectory.h"
#include "TH1D.h"
#include "TObject.h"
#include "TObjString.h"
#include <cassert>

namespace ana
{
  //---------------------------------------------------------------------------
  NDExtrapComponent::NDExtrapComponent(SpectrumLoaderBase& loader,
                                       const HistAxis& axis,
                                       const Cut& selection,                                
                                       const SystShifts& shiftMC,
                                       const Var& weight)
    :  fKaonComponent(new NDExtrapBeamComponent(loader, axis, selection, 0, shiftMC, weight)),
       fPionComponent(new NDExtrapBeamComponent(loader, axis, selection, 1, shiftMC, weight)),
       fMuonComponent(new NDExtrapBeamComponent(loader, axis, selection, 2, shiftMC, weight))
      {
      }

  //---------------------------------------------------------------------------
  NDOscillatableSpectrum NDExtrapComponent::ExtrapComponent(double KaonScale, double PionScale, double MuonScale)
  {

    NDOscillatableSpectrum totalSpec   = fKaonComponent->Component();
    totalSpec.Scale(KaonScale);
    NDOscillatableSpectrum frompion    = fPionComponent->Component();
    frompion.Scale(PionScale);
    totalSpec += frompion;
    NDOscillatableSpectrum frommuon    = fMuonComponent->Component();
    frommuon.Scale(MuonScale);
    totalSpec +=frommuon;

    return totalSpec;
  }

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

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

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

    fKaonComponent->SaveTo(dir, "ndextrapbeamcompkaon");
    fPionComponent->SaveTo(dir, "ndextrapbeamcomppion");
    fMuonComponent->SaveTo(dir, "ndextrapbeamcompmuon");

    dir->Write();
    delete dir;
  
    tmp->cd();
  }

 //---------------------------------------------------------------------------

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

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

    ret->fKaonComponent = ana::LoadFrom<NDExtrapBeamComponent>(dir, "ndextrapbeamcompkaon");

    ret->fPionComponent = ana::LoadFrom<NDExtrapBeamComponent>(dir, "ndextrapbeamcomppion");

    ret->fMuonComponent = ana::LoadFrom<NDExtrapBeamComponent>(dir, "ndextrapbeamcompmuon");
    
    delete dir;

    return ret;
  }

} /////  end of namespace ana //