AnaTools/interface/CutFlow.h

#ifndef CUT_FLOW

#define CUT_FLOW

#include <iostream>
#include <iomanip>
#include <map>
#include <string>
#include <vector>

#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

#include "TH1D.h"
#include "TStopwatch.h"

using namespace std;

class CutFlow
  {
    public:
      CutFlow () {};
      CutFlow (map<string, TH1D *> &, const edm::Service<TFileService> &, const string &prefix = "");
      ~CutFlow ();
      bool &operator[] (const string &);
      bool &at (const string &cutName) { return (*this)[cutName]; };

      void fillCutFlow ();
      void outputTime ();
      void outputCutFlow ();

    private:
      TH1D *cutFlow_;
      TH1D *selection_;
      TH1D *complementarySelection_;

      TStopwatch sw_;
      string prefix_;
      vector<string> cutNames_;
      map<string, bool> cuts_;
  };

#endif
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
CVS Tags:	mytag_0_0_0, V00-00-03, V00-00-02
Log Message:	First commit of CutFlow class.
#	Content
1	#ifndef CUT_FLOW
2
3	#define CUT_FLOW
4
5	#include <iostream>
6	#include <iomanip>
7	#include <map>
8	#include <string>
9	#include <vector>
10
11	#include "FWCore/ServiceRegistry/interface/Service.h"
12	#include "CommonTools/UtilAlgos/interface/TFileService.h"
13
14	#include "TH1D.h"
15	#include "TStopwatch.h"
16
17	using namespace std;
18
19	class CutFlow
20	{
21	public:
22	CutFlow () {};
23	CutFlow (map<string, TH1D *> &, const edm::Service<TFileService> &, const string &prefix = "");
24	~CutFlow ();
25	bool &operator[] (const string &);
26	bool &at (const string &cutName) { return (*this)[cutName]; };
27
28	void fillCutFlow ();
29	void outputTime ();
30	void outputCutFlow ();
31
32	private:
33	TH1D *cutFlow_;
34	TH1D *selection_;
35	TH1D *complementarySelection_;
36
37	TStopwatch sw_;
38	string prefix_;
39	vector<string> cutNames_;
40	map<string, bool> cuts_;
41	};
42
43	#endif