AnaTools/src/SFWeight.cc

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

MuonSFWeight::MuonSFWeight (const string &sfFile, const string &dataOverMC)
{

  TFile *fin = TFile::Open (sfFile.c_str ());
  TGraphAsymmErrors *data = (TGraphAsymmErrors *) fin->Get (dataOverMC.c_str ());

  double bins[data->GetN () + 1];
  for (int i = 0; i < data->GetN (); i++)
    {
      double x, y, exl, exh;

      data->GetPoint (i, x, y);
      exl = data->GetErrorXlow (i);
      exh = data->GetErrorXhigh (i);
      if (i == 0)
        bins[i] = x - exl;
      bins[i + 1] = x + exh;
    }

  muonSFWeight_ = new TH1D ("muonSFWeight", "", data->GetN (), bins);
  muonSFWeight_->SetDirectory (0);
  for (int i = 0; i < data->GetN (); i++)
    {
      double dataX, dataY;

      data->GetPoint (i, dataX, dataY);
      muonSFWeight_->SetBinContent (i + 1, dataY);
    }

  fin->Close ();
  delete data;
}

MuonSFWeight::MuonSFWeight (const string &sfFile, const string &dataEff, const string &mcEff)
{
  TFile *fin = TFile::Open (sfFile.c_str ());
  TGraphAsymmErrors *mc = (TGraphAsymmErrors *) fin->Get (mcEff.c_str ());
  TGraphAsymmErrors *data = (TGraphAsymmErrors *) fin->Get (dataEff.c_str ());

  double bins[mc->GetN () + 1];
  for (int i = 0; i < mc->GetN (); i++)
    {
      double x, y, exl, exh;

      mc->GetPoint (i, x, y);
      exl = mc->GetErrorXlow (i);
      exh = mc->GetErrorXhigh (i);
      if (i == 0)
        bins[i] = x - exl;
      bins[i + 1] = x + exh;
    }

  muonSFWeight_ = new TH1D ("muonSFWeight", "", mc->GetN (), bins);
  muonSFWeight_->SetDirectory (0);
  for (int i = 0; i < mc->GetN (); i++)
    {
      double mcX, mcY, dataX, dataY;

      mc->GetPoint (i, mcX, mcY);
      data->GetPoint (i, dataX, dataY);
      muonSFWeight_->SetBinContent (i + 1, dataY / mcY);
    }

  fin->Close ();
  delete mc;
  delete data;
}

MuonSFWeight::~MuonSFWeight ()
{
  delete muonSFWeight_;
}

ElectronSFWeight::ElectronSFWeight (const string &cmsswRelease, const string &id) :
  cmsswRelease_ (cmsswRelease),
  id_ (id)
{
}

double
ElectronSFWeight::at (const double &eta, const double &pt)
{
  //cout << eta << " " << pt << endl;
  //cout << "doing it\n";
  if (cmsswRelease_ == "53X")
    {
      if (id_ == "tight")
        {
          if (fabs (eta) < 0.8)
            {
              if (pt < 15.0 && pt >= 10.0)
                return 0.818;
              if (pt < 20.0 && pt >= 15.0)
                return 0.928;
              if (pt < 30.0 && pt >= 20.0)
                return 0.973;
              if (pt < 40.0 && pt >= 30.0)
                return 0.979;
              if (pt < 50.0 && pt >= 40.0)
                return 0.984;
              if (pt < 200.0 && pt >= 50.0)
                return 0.983;
            }
          if (fabs (eta) < 1.442 && fabs (eta) >= 0.8)
            {
              if (pt < 15.0 && pt >= 10.0)
                return 0.840;
              if (pt < 20.0 && pt >= 15.0)
                return 0.914;
              if (pt < 30.0 && pt >= 20.0)
                return 0.948;
              if (pt < 40.0 && pt >= 30.0)
                return 0.961;
              if (pt < 50.0 && pt >= 40.0)
                return 0.972;
              if (pt < 200.0 && pt >= 50.0)
                return 0.977;
            }
          if (fabs (eta) < 1.556 && fabs (eta) >= 1.442)
            {
              if (pt < 15.0 && pt >= 10.0)
                return 1.008;
              if (pt < 20.0 && pt >= 15.0)
                return 0.877;
              if (pt < 30.0 && pt >= 20.0)
                return 0.983;
              if (pt < 40.0 && pt >= 30.0)
                return 0.983;
              if (pt < 50.0 && pt >= 40.0)
                return 0.957;
              if (pt < 200.0 && pt >= 50.0)
                return 0.978;
            }
          if (fabs (eta) < 2.0 && fabs (eta) >= 1.556)
            {
              if (pt < 15.0 && pt >= 10.0)
                return 0.906;
              if (pt < 20.0 && pt >= 15.0)
                return 0.907;
              if (pt < 30.0 && pt >= 20.0)
                return 0.957;
              if (pt < 40.0 && pt >= 30.0)
                return 0.962;
              if (pt < 50.0 && pt >= 40.0)
                return 0.985;
              if (pt < 200.0 && pt >= 50.0)
                return 0.986;
            }
          if (fabs (eta) < 2.5 && fabs (eta) >= 2.0)
            {
              if (pt < 15.0 && pt >= 10.0)
                return 0.991;
              if (pt < 20.0 && pt >= 15.0)
                return 0.939;
              if (pt < 30.0 && pt >= 20.0)
                return 1.001;
              if (pt < 40.0 && pt >= 30.0)
                return 1.002;
              if (pt < 50.0 && pt >= 40.0)
                return 0.999;
              if (pt < 200.0 && pt >= 50.0)
                return 0.995;
            }
        }
      if (id_ == "mvaNonTrig")
        {
          cout << "mvaNonTrig\n";
          if (fabs (eta) < 0.8)
            {
              if (pt < 20.0)
                return 1.0;
              if (pt < 30.0 && pt >= 20.0)
                return 0.994;
              if (pt < 40.0 && pt >= 30.0)
                return 0.974;
              if (pt < 50.0 && pt >= 40.0)
                return 0.988;
              if (pt < 200.0 && pt >= 50.0)
                return 0.999;
            }
          if (fabs (eta) < 1.478 && fabs (eta) >= 0.8)
            {
              if (pt < 20.0)
                return 1.0;
              if (pt < 30.0 && pt >= 20.0)
                return 0.971;
              if (pt < 40.0 && pt >= 30.0)
                return 0.967;
              if (pt < 50.0 && pt >= 40.0)
                return 0.984;
              if (pt < 200.0 && pt >= 50.0)
                return 0.998;
            }
          if (fabs (eta) <= 2.5 && fabs (eta) >= 1.478)
            {
              if (pt < 20.0)
                return 1.0;
              if (pt < 30.0 && pt >= 20.0)
                return 0.921;
              if (pt < 40.0 && pt >= 30.0)
                return 0.932;
              if (pt < 50.0 && pt >= 40.0)
                return 0.968;
              if (pt < 200.0 && pt >= 50.0)
                return 0.983;
            }
        }
    }

  return 0.0;
}

ElectronSFWeight::~ElectronSFWeight ()
{
}
Revision:	1.3
Committed:	Wed May 1 19:50:37 2013 UTC (12 years ago) by ahart
Content type:	text/plain
Branch:	MAIN
CVS Tags:	V01-01-00, V01-00-01, V01-00-00
Changes since 1.2:	+2 -2 lines
Log Message:	Commented out the cout statements in the ElectronSFWeight::at.
#	User	Rev	Content
1	ahart	1.1	#include "OSUT3Analysis/AnaTools/interface/SFWeight.h"
2
3			MuonSFWeight::MuonSFWeight (const string &sfFile, const string &dataOverMC)
4			{
5	lantonel	1.2
6	ahart	1.1	TFile *fin = TFile::Open (sfFile.c_str ());
7			TGraphAsymmErrors data = (TGraphAsymmErrors ) fin->Get (dataOverMC.c_str ());
8
9			double bins[data->GetN () + 1];
10			for (int i = 0; i < data->GetN (); i++)
11			{
12			double x, y, exl, exh;
13
14			data->GetPoint (i, x, y);
15			exl = data->GetErrorXlow (i);
16			exh = data->GetErrorXhigh (i);
17			if (i == 0)
18			bins[i] = x - exl;
19			bins[i + 1] = x + exh;
20			}
21
22			muonSFWeight_ = new TH1D ("muonSFWeight", "", data->GetN (), bins);
23			muonSFWeight_->SetDirectory (0);
24			for (int i = 0; i < data->GetN (); i++)
25			{
26			double dataX, dataY;
27
28			data->GetPoint (i, dataX, dataY);
29			muonSFWeight_->SetBinContent (i + 1, dataY);
30			}
31
32			fin->Close ();
33			delete data;
34			}
35
36			MuonSFWeight::MuonSFWeight (const string &sfFile, const string &dataEff, const string &mcEff)
37			{
38			TFile *fin = TFile::Open (sfFile.c_str ());
39			TGraphAsymmErrors mc = (TGraphAsymmErrors ) fin->Get (mcEff.c_str ());
40			TGraphAsymmErrors data = (TGraphAsymmErrors ) fin->Get (dataEff.c_str ());
41
42			double bins[mc->GetN () + 1];
43			for (int i = 0; i < mc->GetN (); i++)
44			{
45			double x, y, exl, exh;
46
47			mc->GetPoint (i, x, y);
48			exl = mc->GetErrorXlow (i);
49			exh = mc->GetErrorXhigh (i);
50			if (i == 0)
51			bins[i] = x - exl;
52			bins[i + 1] = x + exh;
53			}
54
55			muonSFWeight_ = new TH1D ("muonSFWeight", "", mc->GetN (), bins);
56			muonSFWeight_->SetDirectory (0);
57			for (int i = 0; i < mc->GetN (); i++)
58			{
59			double mcX, mcY, dataX, dataY;
60
61			mc->GetPoint (i, mcX, mcY);
62			data->GetPoint (i, dataX, dataY);
63			muonSFWeight_->SetBinContent (i + 1, dataY / mcY);
64			}
65
66			fin->Close ();
67			delete mc;
68			delete data;
69			}
70
71			MuonSFWeight::~MuonSFWeight ()
72			{
73			delete muonSFWeight_;
74			}
75
76			ElectronSFWeight::ElectronSFWeight (const string &cmsswRelease, const string &id) :
77			cmsswRelease_ (cmsswRelease),
78			id_ (id)
79			{
80			}
81
82			double
83			ElectronSFWeight::at (const double &eta, const double &pt)
84			{
85	ahart	1.3	//cout << eta << " " << pt << endl;
86			//cout << "doing it\n";
87	ahart	1.1	if (cmsswRelease_ == "53X")
88			{
89			if (id_ == "tight")
90			{
91			if (fabs (eta) < 0.8)
92			{
93			if (pt < 15.0 && pt >= 10.0)
94			return 0.818;
95			if (pt < 20.0 && pt >= 15.0)
96			return 0.928;
97			if (pt < 30.0 && pt >= 20.0)
98			return 0.973;
99			if (pt < 40.0 && pt >= 30.0)
100			return 0.979;
101			if (pt < 50.0 && pt >= 40.0)
102			return 0.984;
103			if (pt < 200.0 && pt >= 50.0)
104			return 0.983;
105			}
106			if (fabs (eta) < 1.442 && fabs (eta) >= 0.8)
107			{
108			if (pt < 15.0 && pt >= 10.0)
109			return 0.840;
110			if (pt < 20.0 && pt >= 15.0)
111			return 0.914;
112			if (pt < 30.0 && pt >= 20.0)
113			return 0.948;
114			if (pt < 40.0 && pt >= 30.0)
115			return 0.961;
116			if (pt < 50.0 && pt >= 40.0)
117			return 0.972;
118			if (pt < 200.0 && pt >= 50.0)
119			return 0.977;
120			}
121			if (fabs (eta) < 1.556 && fabs (eta) >= 1.442)
122			{
123			if (pt < 15.0 && pt >= 10.0)
124			return 1.008;
125			if (pt < 20.0 && pt >= 15.0)
126			return 0.877;
127			if (pt < 30.0 && pt >= 20.0)
128			return 0.983;
129			if (pt < 40.0 && pt >= 30.0)
130			return 0.983;
131			if (pt < 50.0 && pt >= 40.0)
132			return 0.957;
133			if (pt < 200.0 && pt >= 50.0)
134			return 0.978;
135			}
136			if (fabs (eta) < 2.0 && fabs (eta) >= 1.556)
137			{
138			if (pt < 15.0 && pt >= 10.0)
139			return 0.906;
140			if (pt < 20.0 && pt >= 15.0)
141			return 0.907;
142			if (pt < 30.0 && pt >= 20.0)
143			return 0.957;
144			if (pt < 40.0 && pt >= 30.0)
145			return 0.962;
146			if (pt < 50.0 && pt >= 40.0)
147			return 0.985;
148			if (pt < 200.0 && pt >= 50.0)
149			return 0.986;
150			}
151			if (fabs (eta) < 2.5 && fabs (eta) >= 2.0)
152			{
153			if (pt < 15.0 && pt >= 10.0)
154			return 0.991;
155			if (pt < 20.0 && pt >= 15.0)
156			return 0.939;
157			if (pt < 30.0 && pt >= 20.0)
158			return 1.001;
159			if (pt < 40.0 && pt >= 30.0)
160			return 1.002;
161			if (pt < 50.0 && pt >= 40.0)
162			return 0.999;
163			if (pt < 200.0 && pt >= 50.0)
164			return 0.995;
165			}
166			}
167	lantonel	1.2	if (id_ == "mvaNonTrig")
168			{
169			cout << "mvaNonTrig\n";
170			if (fabs (eta) < 0.8)
171			{
172			if (pt < 20.0)
173			return 1.0;
174			if (pt < 30.0 && pt >= 20.0)
175			return 0.994;
176			if (pt < 40.0 && pt >= 30.0)
177			return 0.974;
178			if (pt < 50.0 && pt >= 40.0)
179			return 0.988;
180			if (pt < 200.0 && pt >= 50.0)
181			return 0.999;
182			}
183			if (fabs (eta) < 1.478 && fabs (eta) >= 0.8)
184			{
185			if (pt < 20.0)
186			return 1.0;
187			if (pt < 30.0 && pt >= 20.0)
188			return 0.971;
189			if (pt < 40.0 && pt >= 30.0)
190			return 0.967;
191			if (pt < 50.0 && pt >= 40.0)
192			return 0.984;
193			if (pt < 200.0 && pt >= 50.0)
194			return 0.998;
195			}
196			if (fabs (eta) <= 2.5 && fabs (eta) >= 1.478)
197			{
198			if (pt < 20.0)
199			return 1.0;
200			if (pt < 30.0 && pt >= 20.0)
201			return 0.921;
202			if (pt < 40.0 && pt >= 30.0)
203			return 0.932;
204			if (pt < 50.0 && pt >= 40.0)
205			return 0.968;
206			if (pt < 200.0 && pt >= 50.0)
207			return 0.983;
208			}
209			}
210	ahart	1.1	}
211
212			return 0.0;
213			}
214
215			ElectronSFWeight::~ElectronSFWeight ()
216			{
217			}