UnfoldSVD.cxx

Go to the documentation of this file.

#include "CAFAna/Unfold/UnfoldSVD.h"

#include "CAFAna/Core/Var.h"

#include "TSVDUnfold.h"

#include "TH2.h"

namespace ana
{
  //----------------------------------------------------------------------
  UnfoldSVD::UnfoldSVD(const ReweightableSpectrum& rt,
                       int reg)
    : fRT(rt), fReg(reg)
  {
  }

  //----------------------------------------------------------------------
  UnfoldSVD::~UnfoldSVD()
  {
  }

  //----------------------------------------------------------------------
  Spectrum UnfoldSVD::Truth(const Spectrum& reco)
  {
    std::unique_ptr<TH1D> hReco(reco.ToTH1(reco.POT()));
    std::unique_ptr<TH1D> hMCReco(fRT.UnWeighted().ToTH1(reco.POT()));
    std::unique_ptr<TH1D> hMCTrue(fRT.WeightingVariable().ToTH1(reco.POT()));
    std::unique_ptr<TH2D> hRT(fRT.ToTH2(reco.POT()));

    TSVDUnfold uf(hReco.get(), hMCReco.get(), hMCTrue.get(), hRT.get());

    std::unique_ptr<TH1D> h_unf(uf.Unfold(fReg));

    Eigen::ArrayXd ret(h_unf->GetNbinsX()+2);
    for(int i = 0; i < h_unf->GetNbinsX()+2; ++i) ret[i] = h_unf->GetBinContent(i);
    // Enforce matching normalization
    ret *= reco.Integral(reco.POT())/ret.sum();

    return Spectrum(ret,
                    HistAxis(reco.GetLabels(), reco.GetBinnings()),
                    reco.POT(), reco.Livetime());
  }
}