Morgan/interface/VertexAnalyzer.h

#ifndef VertexAnalyzer_h
#define VertexAnalyzer_h

// system include files
#include <iostream>

// user include files
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/Common/interface/RefToBase.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"

#include "../interface/TRootEvent.h"
#include "../interface/TRootBeamSpot.h"
#include "../interface/TRootVertex.h"

#include "TClonesArray.h"


class VertexAnalyzer
{

   public:
      VertexAnalyzer(const edm::ParameterSet& producersNames);
      VertexAnalyzer(const edm::ParameterSet& producersNames, int verbosity);
      ~VertexAnalyzer();
      void setVerbosity(int verbosity) {verbosity_ = verbosity; };
      bool getBeamSpot(const edm::Event& iEvent, TRootBeamSpot* rootBeamSpot);
      bool getVertices(const edm::Event& iEvent, TClonesArray* rootVertices);
      void selectPrimary(TRootEvent* rootEvent, TClonesArray* rootVertices);

   private:
      int verbosity_;
      edm::InputTag primaryVertexProducer_;
      edm::InputTag beamSpotProducer_;
      bool allowMissingCollection_;

};

#endif
Revision:	1.6
Committed:	Wed Jun 10 11:17:06 2009 UTC (15 years, 10 months ago) by lethuill
Content type:	text/plain
Branch:	MAIN
CVS Tags:	all_3_3_2_01, all_3_2_5_02, all_3_2_5_01, all_2_2_9_03, all_2_2_9_02, all_2_2_9_01, HEAD
Branch point for:	CMSSW_2_2_X_br
Changes since 1.5:	+16 -14 lines
Log Message:	Better protection against missing collection / Cleaning data format selection / Last iteration for migration to PAT of Photons
#	User	Rev	Content
1	lethuill	1.1	#ifndef VertexAnalyzer_h
2			#define VertexAnalyzer_h
3
4			// system include files
5			#include <iostream>
6
7			// user include files
8			#include "FWCore/Framework/interface/Event.h"
9			#include "FWCore/ParameterSet/interface/ParameterSet.h"
10			#include "FWCore/Framework/interface/ESHandle.h"
11
12	lethuill	1.3	#include "DataFormats/TrackReco/interface/Track.h"
13			#include "DataFormats/TrackReco/interface/TrackFwd.h"
14	lethuill	1.1	#include "DataFormats/VertexReco/interface/Vertex.h"
15			#include "DataFormats/VertexReco/interface/VertexFwd.h"
16	lethuill	1.3	#include "DataFormats/Common/interface/RefToBase.h"
17	lethuill	1.5	#include "DataFormats/BeamSpot/interface/BeamSpot.h"
18	lethuill	1.1
19	lethuill	1.2	#include "../interface/TRootEvent.h"
20	lethuill	1.5	#include "../interface/TRootBeamSpot.h"
21	lethuill	1.3	#include "../interface/TRootVertex.h"
22	lethuill	1.1
23			#include "TClonesArray.h"
24
25
26	lethuill	1.6	class VertexAnalyzer
27			{
28	lethuill	1.1
29	lethuill	1.6	public:
30			VertexAnalyzer(const edm::ParameterSet& producersNames);
31			VertexAnalyzer(const edm::ParameterSet& producersNames, int verbosity);
32			~VertexAnalyzer();
33			void setVerbosity(int verbosity) {verbosity_ = verbosity; };
34			bool getBeamSpot(const edm::Event& iEvent, TRootBeamSpot* rootBeamSpot);
35			bool getVertices(const edm::Event& iEvent, TClonesArray* rootVertices);
36			void selectPrimary(TRootEvent* rootEvent, TClonesArray* rootVertices);
37
38			private:
39			int verbosity_;
40			edm::InputTag primaryVertexProducer_;
41			edm::InputTag beamSpotProducer_;
42			bool allowMissingCollection_;
43	lethuill	1.1
44			};
45
46			#endif