VHbbDataFormats/interface/CutsAndHistos.h

#ifndef CUTANDHISTOS_H
#define CUTANDHISTOS_H
#include "VHbbAnalysis/VHbbDataFormats/interface/VHbbEvent.h"
#include "VHbbAnalysis/VHbbDataFormats/interface/VHbbProxy.h"
#include <string>
#include <sstream>
#include <vector>
#include <TROOT.h>
#include <TFile.h>

class VHbbEvent;

class Histos {
public:
   virtual void book(TFile &f, std::string suffix) = 0;
   virtual void fill(VHbbProxy &, float w) = 0;
};

class Cut {
 public:
   virtual bool pass(VHbbProxy &) = 0;
   virtual std::string name() = 0;
   virtual bool operator()(VHbbProxy &iProxy) {return pass(iProxy); }
};

class CutSet : public Cut {
 public:
 void add(Cut *c) {cuts.push_back(c);}  
 bool pass(VHbbProxy &iProxy) {
  bool result=true;
  for(size_t i=0; i< cuts.size(); i++) 
    if( ! (cuts.at(i)->pass(iProxy)) ) 
      result=false;
  return result;
 } 
 std::string name() {
   std::stringstream s;
   for(size_t i=0; i< cuts.size(); i++) {
     s << "_" << cuts.at(i)->name();
   }
 return s.str();
 }

private:
 std::vector<Cut *> cuts;
 
};

class CutsAndHistos {
public:
  CutsAndHistos() {}
  CutsAndHistos(CutSet & c, std::vector<Histos *> & h):
    cutset(c),
    histos(h) {}
  
  void book(TFile &f) {
    std::string suffix=cutset.name();
    for(size_t i=0; i< histos.size(); i++) 
      histos.at(i)->book(f,suffix);
  }
  
  void process(VHbbProxy & iProxy,float w)
  {
    if(cutset.pass(iProxy))
      for(size_t i=0; i< histos.size(); i++) 
        histos.at(i)->fill(iProxy,w);
  }

  
  CutSet cutset;
  std::vector<Histos *> histos;
    
}; 


#endif
Revision:	1.1
Committed:	Fri Jul 8 07:32:35 2011 UTC (13 years, 9 months ago) by tboccali
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Jul22nd2011, Jul21st2011, Jul20th2011, Jul18th2011, Jul17th2011, Jul8th2011
Log Message:	split in DF and EDM parts
#	User	Rev	Content
1	tboccali	1.1	#ifndef CUTANDHISTOS_H
2			#define CUTANDHISTOS_H
3			#include "VHbbAnalysis/VHbbDataFormats/interface/VHbbEvent.h"
4			#include "VHbbAnalysis/VHbbDataFormats/interface/VHbbProxy.h"
5			#include <string>
6			#include <sstream>
7			#include <vector>
8			#include <TROOT.h>
9			#include <TFile.h>
10
11			class VHbbEvent;
12
13			class Histos {
14			public:
15			virtual void book(TFile &f, std::string suffix) = 0;
16			virtual void fill(VHbbProxy &, float w) = 0;
17			};
18
19			class Cut {
20			public:
21			virtual bool pass(VHbbProxy &) = 0;
22			virtual std::string name() = 0;
23			virtual bool operator()(VHbbProxy &iProxy) {return pass(iProxy); }
24			};
25
26			class CutSet : public Cut {
27			public:
28			void add(Cut *c) {cuts.push_back(c);}
29			bool pass(VHbbProxy &iProxy) {
30			bool result=true;
31			for(size_t i=0; i< cuts.size(); i++)
32			if( ! (cuts.at(i)->pass(iProxy)) )
33			result=false;
34			return result;
35			}
36			std::string name() {
37			std::stringstream s;
38			for(size_t i=0; i< cuts.size(); i++) {
39			s << "_" << cuts.at(i)->name();
40			}
41			return s.str();
42			}
43
44			private:
45			std::vector<Cut *> cuts;
46
47			};
48
49			class CutsAndHistos {
50			public:
51			CutsAndHistos() {}
52			CutsAndHistos(CutSet & c, std::vector<Histos *> & h):
53			cutset(c),
54			histos(h) {}
55
56			void book(TFile &f) {
57			std::string suffix=cutset.name();
58			for(size_t i=0; i< histos.size(); i++)
59			histos.at(i)->book(f,suffix);
60			}
61
62			void process(VHbbProxy & iProxy,float w)
63			{
64			if(cutset.pass(iProxy))
65			for(size_t i=0; i< histos.size(); i++)
66			histos.at(i)->fill(iProxy,w);
67			}
68
69
70			CutSet cutset;
71			std::vector<Histos *> histos;
72
73			};
74
75
76
77
78
79			#endif