AnaTools/src/PUWeight.cc

#include "OSUT3Analysis/AnaTools/interface/PUWeight.h"

PUWeight::PUWeight (const string &puFile, const string &dataPU, const string &mcPU)
{
  TFile *fin = TFile::Open (puFile.c_str ());
  if (!fin || fin->IsZombie()) {
    cout << "ERROR [PUWeight]: Could not find file: " << puFile 
         << "; will cause a seg fault." << endl;
    exit(1);
  }  

  TH1D *mc;
  fin->GetObject(mcPU.c_str(), mc);  
  fin->GetObject(dataPU.c_str(), puWeight_);
  if (!mc) {
    cout << "ERROR [PUWeight]: Could not find histogram: " << mcPU
         << "; will cause a seg fault." << endl;
    exit(1);
  }
  if (!puWeight_) {
    cout << "ERROR [PUWeight]: Could not find histogram: " << dataPU
         << "; will cause a seg fault." << endl;
    exit(1);
  }

  mc->SetDirectory (0);
  puWeight_->SetDirectory (0);
  mc->Scale (puWeight_->Integral () / mc->Integral ());
  TH1D *trimmedMC = new TH1D ("bla", "bla", puWeight_->GetNbinsX(), 0, puWeight_->GetNbinsX());
  for (int bin = 1; bin <= puWeight_->GetNbinsX(); bin++)
    trimmedMC->SetBinContent (bin, mc->GetBinContent (bin));
  puWeight_->Divide (trimmedMC);
  fin->Close ();
  delete mc;
}
  
PUWeight::~PUWeight ()
{
  delete puWeight_;
}
Revision:	1.4
Committed:	Wed Mar 20 16:13:08 2013 UTC (12 years, 1 month ago) by wulsin
Content type:	text/plain
Branch:	MAIN
CVS Tags:	V02-03-02, V02-03-01, V02-03-00, V02-02-00, V02-01-01, V02-01-00, V01-01-00, V01-00-01, V01-00-00, V00-01-00, HEAD
Changes since 1.3:	+21 -3 lines
Log Message:	Added error protection, in case the file cannot be opened or the histograms cannot be found.
#	Content
1	#include "OSUT3Analysis/AnaTools/interface/PUWeight.h"
2
3	PUWeight::PUWeight (const string &puFile, const string &dataPU, const string &mcPU)
4	{
5	TFile *fin = TFile::Open (puFile.c_str ());
6	if (!fin \|\| fin->IsZombie()) {
7	cout << "ERROR [PUWeight]: Could not find file: " << puFile
8	<< "; will cause a seg fault." << endl;
9	exit(1);
10	}
11
12	TH1D *mc;
13	fin->GetObject(mcPU.c_str(), mc);
14	fin->GetObject(dataPU.c_str(), puWeight_);
15	if (!mc) {
16	cout << "ERROR [PUWeight]: Could not find histogram: " << mcPU
17	<< "; will cause a seg fault." << endl;
18	exit(1);
19	}
20	if (!puWeight_) {
21	cout << "ERROR [PUWeight]: Could not find histogram: " << dataPU
22	<< "; will cause a seg fault." << endl;
23	exit(1);
24	}
25
26	mc->SetDirectory (0);
27	puWeight_->SetDirectory (0);
28	mc->Scale (puWeight_->Integral () / mc->Integral ());
29	TH1D *trimmedMC = new TH1D ("bla", "bla", puWeight_->GetNbinsX(), 0, puWeight_->GetNbinsX());
30	for (int bin = 1; bin <= puWeight_->GetNbinsX(); bin++)
31	trimmedMC->SetBinContent (bin, mc->GetBinContent (bin));
32	puWeight_->Divide (trimmedMC);
33	fin->Close ();
34	delete mc;
35	}
36
37	PUWeight::~PUWeight ()
38	{
39	delete puWeight_;
40	}