TrivialBkgdEstimator.cxx

Go to the documentation of this file.

#include "CAFAna/Core/LoadFromFile.h"
#include "CAFAna/XSec/TrivialBkgdEstimator.h"

#include "TObjString.h"
#include "TVectorD.h"

namespace ana
{
  int
  TrivialBkgdEstimator::GetNComponents() const
  {
    return fMCBkgd.size();
  }

  const Spectrum *
  TrivialBkgdEstimator::GetComponent(int n) const
  {
    //    assert(n < (int) fMCBkgd.size() && "Index out of range");
    return fMCBkgd[n];
  }

  Spectrum TrivialBkgdEstimator::Background() const
  {
    assert(fMCBkgd.size() > 0 && "No background given to estimator");
    Spectrum ret = *fMCBkgd[0];
    for (int i = 1; i < int(fMCBkgd.size()); i++)
      ret += *fMCBkgd[i];
    return ret;
  }

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

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

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

    // Need a way to save a vector of Spectrum*
    TVectorD nmcbkgd(1);
    nmcbkgd[0] = fMCBkgd.size();
    nmcbkgd.Write("nmcbkgd");
    for (int i = 0; i < int(fMCBkgd.size()); i++){
      fMCBkgd[i]->SaveTo(dir, "fMCBkgd_"+std::to_string(i));
    }

    dir->Write();
    delete dir;

    tmp->cd();
  }

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

    TVectorD* nmcbkgd = (TVectorD*)dir->Get("nmcbkgd");
    assert(nmcbkgd->GetNoElements()==1 && "N MCBkgds is not 1.");
    std::vector<Spectrum*> mcbkgd;
    dir->cd();
    for (int i = 0; i < (*nmcbkgd)[0]; i++){
      mcbkgd.push_back(ana::LoadFrom<Spectrum>(dir, "fMCBkgd_"+std::to_string(i)).release());
    }

    delete dir;

    return std::unique_ptr<TrivialBkgdEstimator>(
      new TrivialBkgdEstimator(mcbkgd));
  }


  TrivialBkgdEstimator::~TrivialBkgdEstimator() {
    for(int i = 0; i < (int) fMCBkgd.size(); i++) {
      delete fMCBkgd[i];
    }
  }

  TrivialBkgdEstimator::TrivialBkgdEstimator(const TrivialBkgdEstimator& copy) {
    fMCBkgd = copy.fMCBkgd;
  }

  TrivialBkgdEstimator& TrivialBkgdEstimator::operator=
  (const TrivialBkgdEstimator& rhs) {
    if(this == & rhs) return *this;
    else {
      fMCBkgd = rhs.fMCBkgd;
      return *this;
    }
  }
  
}