ana::EnsembleSpectrum Class Reference

#include <EnsembleSpectrum.h>

Public Member Functions
	EnsembleSpectrum (SpectrumLoaderBase &loader, const HistAxis &axis, const SpillCut &spillcut, const Cut &cut, const std::vector< SystShifts > &univ_shifts, const Var &cv_wei=kUnweighted)
	EnsembleSpectrum (SpectrumLoaderBase &loader, const HistAxis &axis, const SpillCut &spillcut, const Cut &cut, const std::vector< Var > &univ_weis, const Var &cv_wei=kUnweighted)
Spectrum	Nominal () const
unsigned int	NUniverses () const
Spectrum	Universe (unsigned int i) const
double	POT () const
double	Livetime () const
TGraphAsymmErrors *	ErrorBand (double exposure, EExposureType expotype=kPOT, EBinType bintype=kBinContent) const
	Result can be painted prettily with DrawErrorBand. More...
void	Scale (double c)
EnsembleSpectrum &	operator+= (const EnsembleSpectrum &rhs)
EnsembleSpectrum	operator+ (const EnsembleSpectrum &rhs) const
EnsembleSpectrum &	operator-= (const EnsembleSpectrum &rhs)
EnsembleSpectrum	operator- (const EnsembleSpectrum &rhs) const
EnsembleSpectrum &	operator*= (const EnsembleRatio &rhs)
EnsembleSpectrum	operator* (const EnsembleRatio &rhs) const
EnsembleSpectrum &	operator/= (const EnsembleRatio &rhs)
EnsembleSpectrum	operator/ (const EnsembleRatio &rhs) const
void	SaveTo (TDirectory *dir) const
unsigned int	NDimensions () const
std::vector< std::string >	GetLabels () const
std::vector< Binning >	GetBinnings () const

Static Public Member Functions
static std::unique_ptr < EnsembleSpectrum >	LoadFrom (TDirectory *dir)

Protected Member Functions
	EnsembleSpectrum (const Spectrum &nom)

Protected Attributes
Spectrum	fNom
std::vector< Spectrum >	fUnivs

Detailed Description

Definition at line 16 of file EnsembleSpectrum.h.

Constructor & Destructor Documentation

ana::EnsembleSpectrum::EnsembleSpectrum	(	SpectrumLoaderBase &	loader,
		const HistAxis &	axis,
		const SpillCut &	spillcut,
		const Cut &	cut,
		const std::vector< SystShifts > &	univ_shifts,
		const Var &	cv_wei = kUnweighted
	)

Definition at line 14 of file EnsembleSpectrum.cxx.

    : fNom(loader, axis, spillcut, cut, kNoShift, cv_wei)
  {
    fUnivs.reserve(univ_shifts.size());
    for(const SystShifts& ss: univ_shifts){
      fUnivs.emplace_back(loader, axis, spillcut, cut, ss, cv_wei);
    }
  }

ana::EnsembleSpectrum::EnsembleSpectrum	(	SpectrumLoaderBase &	loader,
		const HistAxis &	axis,
		const SpillCut &	spillcut,
		const Cut &	cut,
		const std::vector< Var > &	univ_weis,
		const Var &	cv_wei = kUnweighted
	)

Definition at line 29 of file EnsembleSpectrum.cxx.

    : fNom(loader, axis, spillcut, cut, kNoShift, cv_wei)
  {
    fUnivs.reserve(univ_weis.size());
    for(const Var& w: univ_weis){
      fUnivs.emplace_back(loader, axis, spillcut, cut, kNoShift, cv_wei * w);
    }
  }

ana::EnsembleSpectrum::EnsembleSpectrum ( const Spectrum &  nom )

inlineprotected

Definition at line 72 of file EnsembleSpectrum.h.

: fNom(nom) {}

Member Function Documentation

TGraphAsymmErrors * ana::EnsembleSpectrum::ErrorBand	(	double	exposure,
		EExposureType	expotype = kPOT,
		EBinType	bintype = kBinContent
	)		const

Result can be painted prettily with DrawErrorBand.

Definition at line 44 of file EnsembleSpectrum.cxx.

  {
    std::unique_ptr<TH1D> hnom(fNom.ToTH1(exposure, expotype, bintype));

    std::vector<std::unique_ptr<TH1D>> hunivs;
    hunivs.reserve(fUnivs.size());
    for(const Spectrum& u: fUnivs) hunivs.emplace_back(u.ToTH1(exposure, expotype, bintype));

    TGraphAsymmErrors* g = new TGraphAsymmErrors;

    for(int binIdx = 0; binIdx < hnom->GetNbinsX()+2; ++binIdx){
      const double xnom = hnom->GetXaxis()->GetBinCenter(binIdx);
      const double ynom = hnom->GetBinContent(binIdx);
      g->SetPoint(binIdx, xnom, ynom);

      const double dx = hnom->GetXaxis()->GetBinWidth(binIdx);

      std::vector<double> ys;
      ys.reserve(hunivs.size());
      for(const std::unique_ptr<TH1D>& hu: hunivs){
        ys.push_back(hu->GetBinContent(binIdx));
      }

      // 1 sigma
      const double y0 = FindQuantile(.5-0.6827/2, ys);
      const double y1 = FindQuantile(.5+0.6827/2, ys);

      // It's theoretically possible for the central value to be outside the
      // error bands - clamp to zero in that case
      g->SetPointError(binIdx, dx/2, dx/2,
                       std::max(ynom-y0, 0.),
                       std::max(y1-ynom, 0.));
    } // end for binIdx

    return g;
  }

std::vector<Binning> ana::EnsembleSpectrum::GetBinnings ( ) const

inline

Definition at line 69 of file EnsembleSpectrum.h.

{return fNom.GetBinnings();}

std::vector<std::string> ana::EnsembleSpectrum::GetLabels ( ) const

inline

Definition at line 68 of file EnsembleSpectrum.h.

{return fNom.GetLabels();}

double ana::EnsembleSpectrum::Livetime ( ) const

inline

Definition at line 43 of file EnsembleSpectrum.h.

{return fNom.Livetime();}

std::unique_ptr< EnsembleSpectrum > ana::EnsembleSpectrum::LoadFrom ( TDirectory *  dir )

static

Definition at line 175 of file EnsembleSpectrum.cxx.

  {
    DontAddDirectory guard;

    TObjString* tag = (TObjString*)dir->Get("type");
    assert(tag);
    assert(tag->GetString() == "EnsembleSpectrum");
    delete tag;

    std::unique_ptr<EnsembleSpectrum> ret(new EnsembleSpectrum(*Spectrum::LoadFrom(dir->GetDirectory("nom"))));

    for(unsigned int i = 0; ; ++i){
      TDirectory* d = dir->GetDirectory(("univ"+std::to_string(i)).c_str());
      if(!d) break;
      ret->fUnivs.push_back(*Spectrum::LoadFrom(d));
    }

    return ret;
  }

unsigned int ana::EnsembleSpectrum::NDimensions ( ) const

inline

Definition at line 67 of file EnsembleSpectrum.h.

{return fNom.NDimensions();}

Spectrum ana::EnsembleSpectrum::Nominal ( ) const

inline

Definition at line 33 of file EnsembleSpectrum.h.

{return fNom;}

unsigned int ana::EnsembleSpectrum::NUniverses ( ) const

inline

Definition at line 34 of file EnsembleSpectrum.h.

{return fUnivs.size();}

EnsembleSpectrum ana::EnsembleSpectrum::operator*

(

const EnsembleRatio &

rhs

)

const

Definition at line 134 of file EnsembleSpectrum.cxx.

  {
    EnsembleSpectrum ret = *this;
    ret *= rhs;
    return ret;
  }

EnsembleSpectrum & ana::EnsembleSpectrum::operator*=

(

const EnsembleRatio &

rhs

)

Definition at line 125 of file EnsembleSpectrum.cxx.

  {
    fNom *= rhs.Nominal();
    assert(rhs.NUniverses() == fUnivs.size());
    for(unsigned int i = 0; i < fUnivs.size(); ++i) fUnivs[i] *= rhs.Universe(i);
    return *this;
  }

EnsembleSpectrum ana::EnsembleSpectrum::operator+

(

const EnsembleSpectrum &

rhs

)

const

Definition at line 100 of file EnsembleSpectrum.cxx.

  {
    EnsembleSpectrum ret = *this;
    ret += rhs;
    return ret;
  }

EnsembleSpectrum & ana::EnsembleSpectrum::operator+=

(

const EnsembleSpectrum &

rhs

)

Definition at line 91 of file EnsembleSpectrum.cxx.

  {
    fNom += rhs.fNom;
    assert(fUnivs.size() == rhs.fUnivs.size());
    for(unsigned int i = 0; i < fUnivs.size(); ++i) fUnivs[i] += rhs.fUnivs[i];
    return *this;
  }

EnsembleSpectrum ana::EnsembleSpectrum::operator-

(

const EnsembleSpectrum &

rhs

)

const

Definition at line 117 of file EnsembleSpectrum.cxx.

  {
    EnsembleSpectrum ret = *this;
    ret -= rhs;
    return ret;
  }

EnsembleSpectrum & ana::EnsembleSpectrum::operator-=

(

const EnsembleSpectrum &

rhs

)

Definition at line 108 of file EnsembleSpectrum.cxx.

  {
    fNom -= rhs.fNom;
    assert(fUnivs.size() == rhs.fUnivs.size());
    for(unsigned int i = 0; i < fUnivs.size(); ++i) fUnivs[i] -= rhs.fUnivs[i];
    return *this;
  }

EnsembleSpectrum ana::EnsembleSpectrum::operator/

(

const EnsembleRatio &

rhs

)

const

Definition at line 151 of file EnsembleSpectrum.cxx.

  {
    EnsembleSpectrum ret = *this;
    ret /= rhs;
    return ret;
  }

EnsembleSpectrum & ana::EnsembleSpectrum::operator/=

(

const EnsembleRatio &

rhs

)

Definition at line 142 of file EnsembleSpectrum.cxx.

  {
    fNom /= rhs.Nominal();
    assert(rhs.NUniverses() == fUnivs.size());
    for(unsigned int i = 0; i < fUnivs.size(); ++i) fUnivs[i] /= rhs.Universe(i);
    return *this;
  }

double ana::EnsembleSpectrum::POT ( ) const

inline

Definition at line 41 of file EnsembleSpectrum.h.

{return fNom.POT();}

void ana::EnsembleSpectrum::SaveTo

(

TDirectory *

dir

)

const

Definition at line 159 of file EnsembleSpectrum.cxx.

  {
    TDirectory* tmp = gDirectory;
    dir->cd();

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

    fNom.SaveTo(dir->mkdir("nom"));
    for(unsigned int i = 0; i < fUnivs.size(); ++i){
      fUnivs[i].SaveTo(dir->mkdir(("univ"+std::to_string(i)).c_str()));
    }

    tmp->cd();
  }

void ana::EnsembleSpectrum::Scale

(

double

c

)

Definition at line 84 of file EnsembleSpectrum.cxx.

  {
    fNom.Scale(c);
    for(Spectrum& s: fUnivs) s.Scale(c);
  }

Spectrum ana::EnsembleSpectrum::Universe ( unsigned int  i ) const

inline

Definition at line 35 of file EnsembleSpectrum.h.

    {
      assert(i < fUnivs.size());
      return fUnivs[i];
    }

Member Data Documentation

Spectrum ana::EnsembleSpectrum::fNom

protected

Definition at line 74 of file EnsembleSpectrum.h.

std::vector<Spectrum> ana::EnsembleSpectrum::fUnivs

protected

Definition at line 75 of file EnsembleSpectrum.h.

---

The documentation for this class was generated from the following files:

• EnsembleSpectrum.h
• EnsembleSpectrum.cxx