AnaTools/src/CutFlow.cc

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

CutFlow::CutFlow (map<string, TH1D *> &hists1D, const edm::Service<TFileService> &fs, const string &prefix)
{
  sw_.Start ();
  string cutFlowName = "cutFlow",
         selectionName = "selection",
         complementarySelectionName = "complementarySelection";

  prefix_ = prefix;
  if (prefix_ != "")
    {
      cutFlowName = prefix_ + "CutFlow";
      selectionName = prefix_ + "Selection";
      complementarySelectionName = prefix_ + "ComplementarySelection";
    }

  TH1::SetDefaultSumw2 ();
  hists1D[cutFlowName] = fs->make<TH1D> (cutFlowName.c_str (), "", 1, 0.0, 1.0);
  hists1D[selectionName] = fs->make<TH1D> (selectionName.c_str (), "", 1, 0.0, 1.0);
  hists1D[complementarySelectionName] = fs->make<TH1D> (complementarySelectionName.c_str (), "", 1, 0.0, 1.0);

  hists1D[cutFlowName]->GetXaxis ()->SetBinLabel (1, "Total");
  hists1D[selectionName]->GetXaxis ()->SetBinLabel (1, "Total");
  hists1D[complementarySelectionName]->GetXaxis ()->SetBinLabel (1, "Total");

  cutFlow_ = hists1D[cutFlowName];
  selection_ = hists1D[selectionName];
  complementarySelection_ = hists1D[complementarySelectionName];
}

CutFlow::~CutFlow ()
{
  sw_.Stop ();
  outputTime ();
  outputCutFlow ();
}

bool &
CutFlow::operator[] (const string &cutName)
{
  if (!cuts_.count (cutName))
    {
      cutNames_.push_back (cutName);
      cuts_[cutName] = false;

      cutFlow_->SetBins (cutNames_.size () + 1, 0.0, cutNames_.size () + 1.0);
      selection_->SetBins (cutNames_.size () + 1, 0.0, cutNames_.size () + 1.0);
      complementarySelection_->SetBins (cutNames_.size () + 1, 0.0, cutNames_.size () + 1.0);

      cutFlow_->GetXaxis ()->SetBinLabel (cutNames_.size () + 1, cutName.c_str ());
      selection_->GetXaxis ()->SetBinLabel (cutNames_.size () + 1, cutName.c_str ());
      complementarySelection_->GetXaxis ()->SetBinLabel (cutNames_.size () + 1, cutName.c_str ());
    }

  return cuts_[cutName];
}

void
CutFlow::fillCutFlow ()
{
  bool fillCumulative = true, fillComplement = true;
  double complement = -1.0;

  cutFlow_->Fill (0.5);
  selection_->Fill (0.5);
  for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
    {
      if (cuts_[*cut])
        {
          double binCenter = selection_->GetBinCenter (cut - cutNames_.begin () + 2);

          selection_->Fill (binCenter);
          if (fillCumulative)
            cutFlow_->Fill (binCenter);
        }
      else
        {
          fillCumulative = false;
          if (complement < 0.0)
            complement = complementarySelection_->GetBinCenter (cut - cutNames_.begin () + 2);
          else
            fillComplement = false;
        }
    }
  if (fillCumulative)
    {
      complementarySelection_->Fill (0.5);
      for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
        {
          double binCenter = complementarySelection_->GetBinCenter (cut - cutNames_.begin () + 2);

          complementarySelection_->Fill (binCenter);
        }
    }
  if (!fillCumulative && fillComplement)
    complementarySelection_->Fill (complement);
}

void
CutFlow::outputTime ()
{
  double cpu, real;
  int days, hours, minutes;

  cpu = sw_.CpuTime ();
  real = sw_.RealTime ();

  days = (int) (cpu / (60.0 * 60.0 * 24.0));
  cpu -= days * (60.0 * 60.0 * 24.0);
  hours = (int) (cpu / (60.0 * 60.0));
  cpu -= hours * (60.0 * 60.0);
  minutes = (int) (cpu / 60.0);
  cpu -= minutes * 60.0;

  cout << endl;
  cout << "=============================================" << endl;
  cout << endl;

  cout << "CPU Time:  ";
  if (days)
    cout << days << " days, ";
  if (days || hours)
    cout << hours << " hours, ";
  if (days || hours || minutes)
    cout << minutes << " minutes, ";
  if (days || hours || minutes || cpu)
    cout << cpu << " seconds" << endl;

  days = (int) (real / (60.0 * 60.0 * 24.0));
  real -= days * (60.0 * 60.0 * 24.0);
  hours = (int) (real / (60.0 * 60.0));
  real -= hours * (60.0 * 60.0);
  minutes = (int) (real / 60.0);
  real -= minutes * 60.0;

  cout << "Real Time: ";
  if (days)
    cout << days << " days, ";
  if (days || hours)
    cout << hours << " hours, ";
  if (days || hours || minutes)
    cout << minutes << " minutes, ";
  if (days || hours || minutes || cpu)
    cout << real << " seconds" << endl;
}

void
CutFlow::outputCutFlow ()
{
  int totalEvents, totalEventsComplement;
  string title = prefix_;

  cout << endl;
  cout << "=============================================" << endl;
  cout << endl;

  if (prefix_ != "")
    {
      title[0] = toupper (title[0]);
      title += " Cuts";
      cout << title << endl;
    }

  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
  cout << setw (25) << left << "Cut Name" << right << setw (25) << "Cut Flow" << setw (25) << "Efficiency" << endl;
  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
  totalEvents = cutFlow_->GetBinContent (1);
  cout << setw (25) << left << "Total:" << right << setw (25) << totalEvents << setw (25) << "100%" << endl;
  for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
    {
      double cutFlow = cutFlow_->GetBinContent (cut - cutNames_.begin () + 2);

      cout << setw (25) << left << (*cut + ":") << right << setw (25) << cutFlow << setw (24) << 100.0 * (cutFlow / (double) totalEvents) << "%" << endl;
    }
  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;

  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
  cout << setw (25) << left << "Cut Name" << right << setw (25) << "Selection" << setw (25) << "Efficiency" << endl;
  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
  cout << setw (25) << left << "Total:" << right << setw (25) << totalEvents << setw (25) << "100%" << endl;
  for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
    {
      double selection  = selection_->GetBinContent (cut - cutNames_.begin () + 2);

      cout << setw (25) << left << (*cut + ":") << right << setw (25) << selection << setw (24) << 100.0 * (selection / (double) totalEvents) << "%" << endl;
    }
  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;

  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
  cout << setw (25) << left << "Cut Name" << right << setw (25) << "Complementary Selection" << setw (25) << "Efficiency" << endl;
  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
  totalEventsComplement = complementarySelection_->GetBinContent (1);
  cout << setw (25) << left << "Total:" << right << setw (25) << totalEventsComplement << setw (24) << 100.0 * (totalEventsComplement / (double) totalEvents) << "%" << endl;
  for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
    {
      double complement = complementarySelection_->GetBinContent (cut - cutNames_.begin () + 2);

      cout << setw (25) << left << (*cut + ":") << right << setw (25) << complement << setw (24) << 100.0 * (complement / (double) totalEvents) << "%" << endl;
    }
  cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
  cout << endl;
}
Revision:	1.1
Committed:	Mon Jul 23 08:20:07 2012 UTC (12 years, 9 months ago) by ahart
Content type:	text/plain
Branch:	MAIN
Log Message:	First commit of CutFlow class.
#	User	Rev	Content
1	ahart	1.1	#include "OSUT3Analysis/AnaTools/interface/CutFlow.h"
2
3			CutFlow::CutFlow (map<string, TH1D *> &hists1D, const edm::Service<TFileService> &fs, const string &prefix)
4			{
5			sw_.Start ();
6			string cutFlowName = "cutFlow",
7			selectionName = "selection",
8			complementarySelectionName = "complementarySelection";
9
10			prefix_ = prefix;
11			if (prefix_ != "")
12			{
13			cutFlowName = prefix_ + "CutFlow";
14			selectionName = prefix_ + "Selection";
15			complementarySelectionName = prefix_ + "ComplementarySelection";
16			}
17
18			TH1::SetDefaultSumw2 ();
19			hists1D[cutFlowName] = fs->make<TH1D> (cutFlowName.c_str (), "", 1, 0.0, 1.0);
20			hists1D[selectionName] = fs->make<TH1D> (selectionName.c_str (), "", 1, 0.0, 1.0);
21			hists1D[complementarySelectionName] = fs->make<TH1D> (complementarySelectionName.c_str (), "", 1, 0.0, 1.0);
22
23			hists1D[cutFlowName]->GetXaxis ()->SetBinLabel (1, "Total");
24			hists1D[selectionName]->GetXaxis ()->SetBinLabel (1, "Total");
25			hists1D[complementarySelectionName]->GetXaxis ()->SetBinLabel (1, "Total");
26
27			cutFlow_ = hists1D[cutFlowName];
28			selection_ = hists1D[selectionName];
29			complementarySelection_ = hists1D[complementarySelectionName];
30			}
31
32			CutFlow::~CutFlow ()
33			{
34			sw_.Stop ();
35			outputTime ();
36			outputCutFlow ();
37			}
38
39			bool &
40			CutFlow::operator[] (const string &cutName)
41			{
42			if (!cuts_.count (cutName))
43			{
44			cutNames_.push_back (cutName);
45			cuts_[cutName] = false;
46
47			cutFlow_->SetBins (cutNames_.size () + 1, 0.0, cutNames_.size () + 1.0);
48			selection_->SetBins (cutNames_.size () + 1, 0.0, cutNames_.size () + 1.0);
49			complementarySelection_->SetBins (cutNames_.size () + 1, 0.0, cutNames_.size () + 1.0);
50
51			cutFlow_->GetXaxis ()->SetBinLabel (cutNames_.size () + 1, cutName.c_str ());
52			selection_->GetXaxis ()->SetBinLabel (cutNames_.size () + 1, cutName.c_str ());
53			complementarySelection_->GetXaxis ()->SetBinLabel (cutNames_.size () + 1, cutName.c_str ());
54			}
55
56			return cuts_[cutName];
57			}
58
59			void
60			CutFlow::fillCutFlow ()
61			{
62			bool fillCumulative = true, fillComplement = true;
63			double complement = -1.0;
64
65			cutFlow_->Fill (0.5);
66			selection_->Fill (0.5);
67			for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
68			{
69			if (cuts_[*cut])
70			{
71			double binCenter = selection_->GetBinCenter (cut - cutNames_.begin () + 2);
72
73			selection_->Fill (binCenter);
74			if (fillCumulative)
75			cutFlow_->Fill (binCenter);
76			}
77			else
78			{
79			fillCumulative = false;
80			if (complement < 0.0)
81			complement = complementarySelection_->GetBinCenter (cut - cutNames_.begin () + 2);
82			else
83			fillComplement = false;
84			}
85			}
86			if (fillCumulative)
87			{
88			complementarySelection_->Fill (0.5);
89			for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
90			{
91			double binCenter = complementarySelection_->GetBinCenter (cut - cutNames_.begin () + 2);
92
93			complementarySelection_->Fill (binCenter);
94			}
95			}
96			if (!fillCumulative && fillComplement)
97			complementarySelection_->Fill (complement);
98			}
99
100			void
101			CutFlow::outputTime ()
102			{
103			double cpu, real;
104			int days, hours, minutes;
105
106			cpu = sw_.CpuTime ();
107			real = sw_.RealTime ();
108
109			days = (int) (cpu / (60.0 * 60.0 * 24.0));
110			cpu -= days * (60.0 * 60.0 * 24.0);
111			hours = (int) (cpu / (60.0 * 60.0));
112			cpu -= hours * (60.0 * 60.0);
113			minutes = (int) (cpu / 60.0);
114			cpu -= minutes * 60.0;
115
116			cout << endl;
117			cout << "=============================================" << endl;
118			cout << endl;
119
120			cout << "CPU Time: ";
121			if (days)
122			cout << days << " days, ";
123			if (days \|\| hours)
124			cout << hours << " hours, ";
125			if (days \|\| hours \|\| minutes)
126			cout << minutes << " minutes, ";
127			if (days \|\| hours \|\| minutes \|\| cpu)
128			cout << cpu << " seconds" << endl;
129
130			days = (int) (real / (60.0 * 60.0 * 24.0));
131			real -= days * (60.0 * 60.0 * 24.0);
132			hours = (int) (real / (60.0 * 60.0));
133			real -= hours * (60.0 * 60.0);
134			minutes = (int) (real / 60.0);
135			real -= minutes * 60.0;
136
137			cout << "Real Time: ";
138			if (days)
139			cout << days << " days, ";
140			if (days \|\| hours)
141			cout << hours << " hours, ";
142			if (days \|\| hours \|\| minutes)
143			cout << minutes << " minutes, ";
144			if (days \|\| hours \|\| minutes \|\| cpu)
145			cout << real << " seconds" << endl;
146			}
147
148			void
149			CutFlow::outputCutFlow ()
150			{
151			int totalEvents, totalEventsComplement;
152			string title = prefix_;
153
154			cout << endl;
155			cout << "=============================================" << endl;
156			cout << endl;
157
158			if (prefix_ != "")
159			{
160			title[0] = toupper (title[0]);
161			title += " Cuts";
162			cout << title << endl;
163			}
164
165			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
166			cout << setw (25) << left << "Cut Name" << right << setw (25) << "Cut Flow" << setw (25) << "Efficiency" << endl;
167			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
168			totalEvents = cutFlow_->GetBinContent (1);
169			cout << setw (25) << left << "Total:" << right << setw (25) << totalEvents << setw (25) << "100%" << endl;
170			for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
171			{
172			double cutFlow = cutFlow_->GetBinContent (cut - cutNames_.begin () + 2);
173
174			cout << setw (25) << left << (cut + ":") << right << setw (25) << cutFlow << setw (24) << 100.0 (cutFlow / (double) totalEvents) << "%" << endl;
175			}
176			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
177
178			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
179			cout << setw (25) << left << "Cut Name" << right << setw (25) << "Selection" << setw (25) << "Efficiency" << endl;
180			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
181			cout << setw (25) << left << "Total:" << right << setw (25) << totalEvents << setw (25) << "100%" << endl;
182			for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
183			{
184			double selection = selection_->GetBinContent (cut - cutNames_.begin () + 2);
185
186			cout << setw (25) << left << (cut + ":") << right << setw (25) << selection << setw (24) << 100.0 (selection / (double) totalEvents) << "%" << endl;
187			}
188			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
189
190			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
191			cout << setw (25) << left << "Cut Name" << right << setw (25) << "Complementary Selection" << setw (25) << "Efficiency" << endl;
192			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
193			totalEventsComplement = complementarySelection_->GetBinContent (1);
194			cout << setw (25) << left << "Total:" << right << setw (25) << totalEventsComplement << setw (24) << 100.0 * (totalEventsComplement / (double) totalEvents) << "%" << endl;
195			for (vector<string>::const_iterator cut = cutNames_.begin (); cut != cutNames_.end (); cut++)
196			{
197			double complement = complementarySelection_->GetBinContent (cut - cutNames_.begin () + 2);
198
199			cout << setw (25) << left << (cut + ":") << right << setw (25) << complement << setw (24) << 100.0 (complement / (double) totalEvents) << "%" << endl;
200			}
201			cout << setw (75) << setfill ('-') << '-' << setfill (' ') << endl;
202			cout << endl;
203			}