ViewVC Help

View File | Revision Log | Show Annotations | Root Listing

root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc

Comparing UserCode/MitProd/TreeFiller/src/FillMitTree.cc (file contents):
Revision 1.2 by loizides, Wed Jun 18 19:17:21 2008 UTC vs.
Revision 1.18 by sixie, Fri Aug 8 11:21:08 2008 UTC

#	Line 1 | Line 1
1		// $Id$
2
3	+	#include "MitProd/TreeFiller/interface/FillMitTree.h"
4		#include "FWCore/MessageLogger/interface/MessageLogger.h"
4	–	#include "FWCore/Framework/interface/ESHandle.h"
5	–	#include "DataFormats/Common/interface/Handle.h"
5		#include "FWCore/ServiceRegistry/interface/Service.h"
6	+	#include "MitProd/TreeService/interface/TreeService.h"
7	+	#include "MitProd/ObjectService/interface/ObjectService.h"
8	+	#include "MitProd/TreeFiller/interface/AssociationMaps.h"
9	+	#include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
10	+	#include "MitProd/TreeFiller/interface/FillerTracks.h"
11	+	#include "MitProd/TreeFiller/interface/FillerGsfTracks.h"
12	+	#include "MitProd/TreeFiller/interface/FillerBasicClusters.h"
13	+	#include "MitProd/TreeFiller/interface/FillerSuperClusters.h"
14	+	#include "MitProd/TreeFiller/interface/FillerMuons.h"
15	+	#include "MitProd/TreeFiller/interface/FillerElectrons.h"
16	+	#include "MitProd/TreeFiller/interface/FillerCaloJets.h"
17	+	#include "MitProd/TreeFiller/interface/FillerCaloMet.h"
18	+	#include "MitProd/TreeFiller/interface/FillerConversions.h"
19	+	#include "MitProd/TreeFiller/interface/FillerConversionElectrons.h"
20	+	#include "MitProd/TreeFiller/interface/FillerPhotons.h"
21	+	#include "MitProd/TreeFiller/interface/FillerMCParticles.h"
22	+	#include "MitProd/TreeFiller/interface/FillerDecayParts.h"
23	+	#include "MitProd/TreeFiller/interface/FillerStableParts.h"
24
25	<	#include "MitProd/TreeFiller/interface/FillMitTree.h"
26	<	#include "MitProd/TreeFiller/interface/FillerGlobalMuons.h"
10	<
25	>	#include "MitProd/TreeFiller/interface/FillerPATMuons.h"
26	>	#include "MitProd/TreeFiller/interface/FillerPATElectrons.h"
27		using namespace std;
28		using namespace edm;
29		using namespace mithep;
30
31	<	//-------------------------------------------------------------------------------------------------
32	<	FillMitTree::FillMitTree(const edm::ParameterSet &cfg)
31	>	mithep::ObjectService *mithep::FillMitTree::os_ = 0;
32	>
33	>	//--------------------------------------------------------------------------------------------------
34	>	FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
35	>	defactive_(cfg.getUntrackedParameter<bool>("defactive",1))
36		{
37	<	// Constructor: initialize fillers
37	>	// Constructor.
38
39	<	FillerGlobalMuons *fillerGlobalMuons = new FillerGlobalMuons(cfg);
40	<	if (fillerGlobalMuons->Active())
41	<	fillers_.push_back(fillerGlobalMuons);
42	<	else
24	<	delete fillerGlobalMuons;
39	>	if (!configure(cfg)) {
40	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
41	>	<< "Could not configure fillers." << "\n";
42	>	}
43		}
44
45	<	//-------------------------------------------------------------------------------------------------
45	>	//--------------------------------------------------------------------------------------------------
46		FillMitTree::~FillMitTree()
47		{
48		// Destructor: nothing to be done here.
49		}
50
51	<	//-------------------------------------------------------------------------------------------------
52	<	void FillMitTree::beginJob(edm::EventSetup const &event)
51	>	//--------------------------------------------------------------------------------------------------
52	>	bool FillMitTree::addActiveFiller(BaseFiller *bf)
53		{
54	<	// Access the tree and book branches.
54	>	// Check if filler is active and add it to list of fillers. Otherwise delete it.
55
56	<	Service<TreeService> ts;
57	<	TreeWriter *tws = ts->get();
58	<	if (! tws) {
41	<	throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
42	<	<< "Could not get pointer to tree." << "\n";
43	<	return;
56	>	if (bf->Active()) {
57	>	fillers_.push_back(bf);
58	>	return 1;
59		}
60
61	<	// Loop over the various components and book the branches
62	<	for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
63	<	(*iF)->BookDataBlock(tws);
49	<	}
50	<	return;
61	>	delete bf;
62	>	bf = 0;
63	>	return 0;
64		}
65
66	<	//-------------------------------------------------------------------------------------------------
66	>	//--------------------------------------------------------------------------------------------------
67		void FillMitTree::analyze(const edm::Event      &event,
68		const edm::EventSetup &setup)
69		{
#	Line 67 | Line 80 | void FillMitTree::analyze(const edm::Eve
80		}
81		}
82
83	<	//-------------------------------------------------------------------------------------------------
83	>	//--------------------------------------------------------------------------------------------------
84	>	void FillMitTree::beginJob(const edm::EventSetup &event)
85	>	{
86	>	// Access the tree and book branches.
87	>
88	>	Service<TreeService> ts;
89	>	TreeWriter *tws = ts->get();
90	>	if (! tws) {
91	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
92	>	<< "Could not get pointer to tree. "
93	>	<< "Do you have the TreeServie define in your config?" << "\n";
94	>	return;
95	>	}
96	>
97	>	if (os_==0) {
98	>	Service<ObjectService> os;
99	>	if (!os.isAvailable()) {
100	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
101	>	<< "Could not get object service. "
102	>	<< "Do you have the ObjectService defined in your config?" << "\n";
103	>	return;
104	>	}
105	>	os_ = &(*os);
106	>	}
107	>
108	>	// Loop over the various components and book the branches
109	>	for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
110	>	edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
111	>	(*iF)->BookDataBlock(*tws);
112	>	}
113	>	}
114	>
115	>	//--------------------------------------------------------------------------------------------------
116	>	bool FillMitTree::configure(const edm::ParameterSet &cfg)
117	>	{
118	>	// Configure our fillers.
119	>
120	>	FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg,defactive_);
121	>	addActiveFiller(fillerMetaInfos);
122	>
123	>	FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg,"MCParticles",defactive_);
124	>	addActiveFiller(fillerMCParticles);
125	>
126	>	FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg,"CaloJets",defactive_);
127	>	addActiveFiller(fillerCaloJets);
128	>
129	>	FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg,"CaloMet",defactive_);
130	>	addActiveFiller(fillerCaloMet);
131	>
132	>	FillerTracks *fillerGeneralTracks = new FillerTracks(cfg,"GeneralTracks",defactive_);
133	>	addActiveFiller(fillerGeneralTracks);
134	>
135	>	FillerTracks *fillerStandaloneMuonTracks =
136	>	new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
137	>	addActiveFiller(fillerStandaloneMuonTracks);
138	>
139	>	FillerTracks *fillerStandaloneMuonTracksVtx =
140	>	new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
141	>	addActiveFiller(fillerStandaloneMuonTracksVtx);
142	>
143	>	FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
144	>	addActiveFiller(fillerGlobalMuonTracks);
145	>
146	>	FillerTracks *fillerConversionInOutTracks =
147	>	new FillerTracks(cfg,"ConversionInOutTracks",defactive_);
148	>	addActiveFiller(fillerConversionInOutTracks);
149	>
150	>	FillerTracks *fillerConversionOutInTracks =
151	>	new FillerTracks(cfg,"ConversionOutInTracks",defactive_);
152	>	addActiveFiller(fillerConversionOutInTracks);
153	>
154	>	FillerGsfTracks *fillerGsfTracks = new FillerGsfTracks(cfg,"GsfTracks",defactive_);
155	>	addActiveFiller(fillerGsfTracks);
156	>
157	>	FillerBasicClusters *fillerBarrelBasicClusters =
158	>	new FillerBasicClusters(cfg, "BarrelBasicClusters", defactive_);
159	>	addActiveFiller(fillerBarrelBasicClusters);
160	>
161	>	FillerSuperClusters *fillerBarrelSuperClusters =
162	>	new FillerSuperClusters(cfg,"BarrelSuperClusters", defactive_);
163	>	addActiveFiller(fillerBarrelSuperClusters);
164	>
165	>	FillerBasicClusters *fillerEndcapBasicClusters =
166	>	new FillerBasicClusters(cfg,"EndcapBasicClusters", defactive_);
167	>	addActiveFiller(fillerEndcapBasicClusters);
168	>
169	>	FillerSuperClusters *fillerEndcapSuperClusters =
170	>	new FillerSuperClusters(cfg,"EndcapSuperClusters", defactive_);
171	>	addActiveFiller(fillerEndcapSuperClusters);
172	>
173	>	FillerMuons *fillerMuons = new FillerMuons(cfg,defactive_);
174	>	addActiveFiller(fillerMuons);
175	>
176	>	FillerElectrons *fillerElectrons = new FillerElectrons(cfg,defactive_);
177	>	addActiveFiller(fillerElectrons);
178	>
179	>	FillerConversionElectrons *fillerConversionElectrons =
180	>	new FillerConversionElectrons(cfg,defactive_);
181	>	addActiveFiller(fillerConversionElectrons);
182	>
183	>	FillerConversions *fillerConversions = new FillerConversions(cfg,defactive_);
184	>	addActiveFiller(fillerConversions);
185	>
186	>	FillerPhotons *fillerPhotons = new FillerPhotons(cfg,defactive_);
187	>	addActiveFiller(fillerPhotons);
188	>
189	>	FillerStableParts *fillerStableParts = new FillerStableParts(cfg,"StableParts",defactive_);
190	>	addActiveFiller(fillerStableParts);
191	>
192	>	FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg,"DecayParts",defactive_);
193	>	addActiveFiller(fillerDecayParts);
194	>
195	>	return 1;
196	>	}
197	>
198	>	//--------------------------------------------------------------------------------------------------
199		void FillMitTree::endJob()
200		{
201		// Delete fillers.

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines