Skip to content

Instantly share code, notes, and snippets.

@musinsky
Last active December 20, 2015 18:09
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save musinsky/6173976 to your computer and use it in GitHub Desktop.
Save musinsky/6173976 to your computer and use it in GitHub Desktop.
Simplified basic functions of ROOT TH1 class
void TH1D::AddBinContent(Int_t bin) {
// AbstractMethod, overridden by TH1D (TH1.h)
++fArray[bin];
}
void TH1D::AddBinContent(Int_t bin, Double_t w) {
// AbstractMethod, overridden by TH1D (TH1.h)
fArray[bin] += Double_t (w);
}
Int_t TH1::Fill(Double_t x)
{
Int_t bin = fXaxis.FindBin(x);
fEntries++;
AddBinContent(bin);
if (fSumw2.fN) ++fSumw2.fArray[bin];
++fTsumw; // stats[0]
++fTsumw2; // stats[1]
fTsumwx += x; // stats[2]
fTsumwx2 += x*x; // stats[3]
return bin;
}
Int_t TH1::Fill(Double_t x, Double_t w)
{
Int_t bin = fXaxis.FindBin(x);
fEntries++;
AddBinContent(bin, w);
if (fSumw2.fN) fSumw2.fArray[bin] += w*w;
fTsumw += w; // stats[0]
fTsumw2 += w*w; // stats[1]
fTsumwx += w*x; // stats[2]
fTsumwx2 += w*x*x; // stats[3]
return bin;
}
Double_t TH1D::GetBinContent(Int_t bin) const
{
// AbstractMethod, overridden by TH1D
return Double_t (fArray[bin]);
}
Double_t TH1::GetBinError(Int_t bin) const
{
if (fSumw2.fN) return TMath::Sqrt(fSumw2.fArray[bin]);
return TMath::Sqrt(TMath::Abs(GetBinContent(bin)));
}
Double_t TH1::GetEffectiveEntries() const
{
Stat_t s[kNstat];
this->GetStats(s); // s[0] sum of weights, s[1] sum of squares of weights
return (s[1] ? s[0]*s[0]/s[1] : TMath::Abs(s[0]));
}
void TH1::GetStats(Double_t *stats) const
{
if ((fTsumw == 0 && fEntries > 0) || fXaxis.TestBit(TAxis::kAxisRange)) {
for (bin = 0; bin < 4; bin++) stats[bin] = 0;
// in original code include underflow/overflow if fgStatOverflows
for (binx = firstBinX; binx <= lastBinX; binx++) {
x = fXaxis.GetBinCenter(binx);
w = GetBinContent(binx);
err = TMath::Abs(GetBinError(binx));
stats[0] += w;
stats[1] += err*err;
stats[2] += w*x;
stats[3] += w*x*x;
}
// if (stats[0] < 0) do something ?!
} else {
stats[0] = fTsumw;
stats[1] = fTsumw2;
stats[2] = fTsumwx;
stats[3] = fTsumwx2;
}
}
void TH1::PutStats(Double_t *stats)
{
fTsumw = stats[0];
fTsumw2 = stats[1];
fTsumwx = stats[2];
fTsumwx2 = stats[3];
}
void TH1::ResetStats()
{
Double_t stats[kNstat] = {0}; // kNstat = 13
fTsumw = 0;
fEntries = 1; // to force re-calculation of the statistics in GetStats
GetStats(stats);
PutStats(stats);
fEntries = TMath::Abs(fTsumw);
// use effective entries for weighted histograms: (sum_w) ^2 / sum_w2
if (fSumw2.fN > 0 && fTsumw > 0 && stats[1] > 0 )
fEntries = stats[0]*stats[0]/stats[1];
}
void TH1D::SetBinContent(Int_t bin, Double_t content)
{
// AbstractMethod, overridden by TH1D
fEntries++;
fTsumw = 0; // stats[0]
fArray[bin] = content;
}
void TH1::SetBinError(Int_t bin, Double_t error)
{
if (!fSumw2.fN) Sumw2();
fSumw2.fArray[bin] = error*error;
}
void TH1::Sumw2(Bool_t flag) // default flag = KTRUE
{
if (!flag) {
if (fSumw2.fN > 0 ) fSumw2.Set(0);
return;
}
if (fSumw2.fN == fNcells) return;
fSumw2.Set(fNcells);
if (fEntries > 0)
for (Int_t bin = 0; bin < fNcells; bin++) {
fSumw2.fArray[bin] = TMath::Abs(GetBinContent(bin));
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment