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.13 by paus, Mon Jul 28 23:13:44 2008 UTC vs.
Revision 1.22 by sixie, Tue Sep 16 18:21:33 2008 UTC

#	Line 4 | Line 4
4		#include "FWCore/MessageLogger/interface/MessageLogger.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/FillerCaloTowers.h"
15		#include "MitProd/TreeFiller/interface/FillerMuons.h"
16		#include "MitProd/TreeFiller/interface/FillerElectrons.h"
17	<	#include "MitProd/TreeFiller/interface/FillerElectrons.h"
17	>	#include "MitProd/TreeFiller/interface/FillerGenJets.h"
18		#include "MitProd/TreeFiller/interface/FillerCaloJets.h"
19		#include "MitProd/TreeFiller/interface/FillerCaloMet.h"
20		#include "MitProd/TreeFiller/interface/FillerConversions.h"
#	Line 19 | Line 23
23		#include "MitProd/TreeFiller/interface/FillerMCParticles.h"
24		#include "MitProd/TreeFiller/interface/FillerDecayParts.h"
25		#include "MitProd/TreeFiller/interface/FillerStableParts.h"
26	+	#include "MitProd/TreeFiller/interface/FillerPATMuons.h"
27	+	#include "MitProd/TreeFiller/interface/FillerPATElectrons.h"
28
29		using namespace std;
30		using namespace edm;
31		using namespace mithep;
32
33	+	mithep::ObjectService *mithep::FillMitTree::os_ = 0;
34	+
35		//--------------------------------------------------------------------------------------------------
36		FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
37		defactive_(cfg.getUntrackedParameter<bool>("defactive",1))
38		{
39		// Constructor.
40	<
40	>
41		if (!configure(cfg)) {
42		throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
43		<< "Could not configure fillers." << "\n";
#	Line 43 | Line 51 | FillMitTree::~FillMitTree()
51		}
52
53		//--------------------------------------------------------------------------------------------------
54	+	bool FillMitTree::addActiveFiller(BaseFiller *bf)
55	+	{
56	+	// Check if filler is active and add it to list of fillers. Otherwise delete it.
57	+
58	+	if (bf->Active()) {
59	+	fillers_.push_back(bf);
60	+	return 1;
61	+	}
62	+
63	+	delete bf;
64	+	bf = 0;
65	+	return 0;
66	+	}
67	+
68	+	//--------------------------------------------------------------------------------------------------
69		void FillMitTree::analyze(const edm::Event      &event,
70		const edm::EventSetup &setup)
71		{
#	Line 68 | Line 91 | void FillMitTree::beginJob(const edm::Ev
91		TreeWriter *tws = ts->get();
92		if (! tws) {
93		throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
94	<	<< "Could not get pointer to tree." << "\n";
94	>	<< "Could not get pointer to tree. "
95	>	<< "Do you have the TreeServie define in your config?" << "\n";
96		return;
97		}
98	<
98	>
99	>	if (os_==0) {
100	>	Service<ObjectService> os;
101	>	if (!os.isAvailable()) {
102	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
103	>	<< "Could not get object service. "
104	>	<< "Do you have the ObjectService defined in your config?" << "\n";
105	>	return;
106	>	}
107	>	os_ = &(*os);
108	>	}
109	>
110		// Loop over the various components and book the branches
111		for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
112	+	edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
113		(*iF)->BookDataBlock(*tws);
114		}
115		}
#	Line 83 | Line 119 | bool FillMitTree::configure(const edm::P
119		{
120		// Configure our fillers.
121
122	<	FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg);
123	<	if (fillerMetaInfos->Active())
88	<	fillers_.push_back(fillerMetaInfos);
89	<	else {
90	<	delete fillerMetaInfos;
91	<	fillerMetaInfos = 0;
92	<	}
122	>	FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg,defactive_);
123	>	addActiveFiller(fillerMetaInfos);
124
125		FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg,"MCParticles",defactive_);
126	<	const GenParticleMap* genParticleMap = 0;
127	<	const SimParticleMap *simParticleMap = 0;
128	<	if (fillerMCParticles->Active()) {
129	<	fillers_.push_back(fillerMCParticles);
130	<	genParticleMap = fillerMCParticles->GetGenParticleMap();
131	<	simParticleMap = fillerMCParticles->GetSimParticleMap();
132	<	}
133	<	else {
134	<	delete fillerMCParticles;
135	<	fillerMCParticles = 0;
136	<	}
126	>	addActiveFiller(fillerMCParticles);
127	>
128	>	FillerCaloTowers *fillerCaloTowers =
129	>	new FillerCaloTowers(cfg, "CaloTowers", defactive_);
130	>	addActiveFiller(fillerCaloTowers);
131	>
132	>	FillerGenJets *fillerIC5GenJets = new FillerGenJets(cfg,"IC5GenJets",defactive_);
133	>	addActiveFiller(fillerIC5GenJets);
134	>
135	>	FillerGenJets *fillerSC5GenJets = new FillerGenJets(cfg,"SC5GenJets",defactive_);
136	>	addActiveFiller(fillerSC5GenJets);
137
138		FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg,"CaloJets",defactive_);
139	<	if (fillerCaloJets->Active()) {
140	<	fillers_.push_back(fillerCaloJets);
141	<	}
142	<	else {
112	<	delete fillerCaloJets;
113	<	fillerCaloJets = 0;
114	<	}
139	>	addActiveFiller(fillerCaloJets);
140	>
141	>	FillerCaloJets *fillerSisCone5Jets = new FillerCaloJets(cfg,"SisCone5Jets",defactive_);
142	>	addActiveFiller(fillerSisCone5Jets);
143
144		FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg,"CaloMet",defactive_);
145	<	if (fillerCaloMet->Active()) {
118	<	fillers_.push_back(fillerCaloMet);
119	<	}
120	<	else {
121	<	delete fillerCaloMet;
122	<	fillerCaloMet = 0;
123	<	}
145	>	addActiveFiller(fillerCaloMet);
146
147	<	FillerTracks *fillerGeneralTracks =
148	<	new FillerTracks(cfg,"GeneralTracks",defactive_,simParticleMap);
127	<	const TrackMap *generalTrackMap=0;
128	<	if (fillerGeneralTracks->Active()) {
129	<	fillers_.push_back(fillerGeneralTracks);
130	<	generalTrackMap = fillerGeneralTracks->GetTrackMap();
131	<	}
132	<	else {
133	<	delete fillerGeneralTracks;
134	<	fillerGeneralTracks = 0;
135	<	}
147	>	FillerTracks *fillerGeneralTracks = new FillerTracks(cfg,"GeneralTracks",defactive_);
148	>	addActiveFiller(fillerGeneralTracks);
149
150		FillerTracks *fillerStandaloneMuonTracks =
151		new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
152	<	const TrackMap *standaloneMuonTrackMap=0;
140	<	if (fillerStandaloneMuonTracks->Active()) {
141	<	fillers_.push_back(fillerStandaloneMuonTracks);
142	<	standaloneMuonTrackMap = fillerStandaloneMuonTracks->GetTrackMap();
143	<	}
144	<	else {
145	<	delete fillerStandaloneMuonTracks;
146	<	fillerStandaloneMuonTracks = 0;
147	<	}
152	>	addActiveFiller(fillerStandaloneMuonTracks);
153
154		FillerTracks *fillerStandaloneMuonTracksVtx =
155		new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
156	<	const TrackMap *standaloneMuonTrackVtxMap=0;
152	<	if (fillerStandaloneMuonTracksVtx->Active()) {
153	<	fillers_.push_back(fillerStandaloneMuonTracksVtx);
154	<	standaloneMuonTrackVtxMap = fillerStandaloneMuonTracksVtx->GetTrackMap();
155	<	}
156	<	else {
157	<	delete fillerStandaloneMuonTracksVtx;
158	<	fillerStandaloneMuonTracksVtx = 0;
159	<	}
156	>	addActiveFiller(fillerStandaloneMuonTracksVtx);
157
158		FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
159	<	const TrackMap *globalMuonTrackMap=0;
163	<	if (fillerGlobalMuonTracks->Active()) {
164	<	fillers_.push_back(fillerGlobalMuonTracks);
165	<	globalMuonTrackMap = fillerGlobalMuonTracks->GetTrackMap();
166	<	}
167	<	else {
168	<	delete fillerGlobalMuonTracks;
169	<	fillerGlobalMuonTracks = 0;
170	<	}
159	>	addActiveFiller(fillerGlobalMuonTracks);
160
161		FillerTracks *fillerConversionInOutTracks =
162	<	new FillerTracks(cfg,"ConversionInOutTracks",defactive_,simParticleMap);
163	<	const TrackMap *conversionInOutTrackMap=0;
175	<	const TrackCol *conversionInOutTracks=0;
176	<	if (fillerConversionInOutTracks->Active()) {
177	<	fillers_.push_back(fillerConversionInOutTracks);
178	<	conversionInOutTrackMap = fillerConversionInOutTracks->GetTrackMap();
179	<	conversionInOutTracks = fillerConversionInOutTracks->GetTrackCol();
180	<	}
181	<	else {
182	<	delete fillerConversionInOutTracks;
183	<	fillerConversionInOutTracks = 0;
184	<	}
162	>	new FillerTracks(cfg,"ConversionInOutTracks",defactive_);
163	>	addActiveFiller(fillerConversionInOutTracks);
164
165		FillerTracks *fillerConversionOutInTracks =
166	<	new FillerTracks(cfg,"ConversionOutInTracks",defactive_,simParticleMap);
167	<	const TrackMap *conversionOutInTrackMap=0;
189	<	const TrackCol *conversionOutInTracks=0;
190	<	if (fillerConversionOutInTracks->Active()) {
191	<	fillers_.push_back(fillerConversionOutInTracks);
192	<	conversionOutInTrackMap = fillerConversionOutInTracks->GetTrackMap();
193	<	conversionOutInTracks = fillerConversionOutInTracks->GetTrackCol();
194	<	}
195	<	else {
196	<	delete fillerConversionOutInTracks;
197	<	fillerConversionOutInTracks = 0;
198	<	}
166	>	new FillerTracks(cfg,"ConversionOutInTracks",defactive_);
167	>	addActiveFiller(fillerConversionOutInTracks);
168
169	<	FillerGsfTracks *fillerGsfTracks =
170	<	new FillerGsfTracks(cfg,"GsfTracks",defactive_,simParticleMap);
202	<	const GsfTrackMap *gsfTrackMap=0;
203	<	if (fillerGsfTracks->Active()) {
204	<	fillers_.push_back(fillerGsfTracks);
205	<	gsfTrackMap = (GsfTrackMap*)fillerGsfTracks->GetTrackMap();
206	<	}
207	<	else {
208	<	delete fillerGsfTracks;
209	<	fillerGsfTracks = 0;
210	<	}
169	>	FillerGsfTracks *fillerGsfTracks = new FillerGsfTracks(cfg,"GsfTracks",defactive_);
170	>	addActiveFiller(fillerGsfTracks);
171
172	<	FillerMuons *fillerMuons =
173	<	new FillerMuons(cfg,defactive_,globalMuonTrackMap,standaloneMuonTrackMap,
174	<	standaloneMuonTrackVtxMap,generalTrackMap);
175	<	if (fillerMuons->Active())
176	<	fillers_.push_back(fillerMuons);
177	<	else {
178	<	delete fillerMuons;
179	<	fillerMuons = 0;
180	<	}
181	<
182	<	FillerElectrons *fillerElectrons =
183	<	new FillerElectrons(cfg,defactive_,gsfTrackMap,generalTrackMap);
184	<	if (fillerElectrons->Active())
185	<	fillers_.push_back(fillerElectrons);
186	<	else {
187	<	delete fillerElectrons;
188	<	fillerElectrons = 0;
189	<	}
172	>	FillerBasicClusters *fillerBarrelBasicClusters =
173	>	new FillerBasicClusters(cfg, "BarrelBasicClusters", defactive_);
174	>	addActiveFiller(fillerBarrelBasicClusters);
175	>
176	>	FillerSuperClusters *fillerBarrelSuperClusters =
177	>	new FillerSuperClusters(cfg,"BarrelSuperClusters", defactive_);
178	>	addActiveFiller(fillerBarrelSuperClusters);
179	>
180	>	FillerBasicClusters *fillerEndcapBasicClusters =
181	>	new FillerBasicClusters(cfg,"EndcapBasicClusters", defactive_);
182	>	addActiveFiller(fillerEndcapBasicClusters);
183	>
184	>	FillerSuperClusters *fillerEndcapSuperClusters =
185	>	new FillerSuperClusters(cfg,"EndcapSuperClusters", defactive_);
186	>	addActiveFiller(fillerEndcapSuperClusters);
187	>
188	>	FillerMuons *fillerMuons = new FillerMuons(cfg,defactive_);
189	>	addActiveFiller(fillerMuons);
190	>
191	>	FillerElectrons *fillerElectrons = new FillerElectrons(cfg,defactive_);
192	>	addActiveFiller(fillerElectrons);
193
194		FillerConversionElectrons *fillerConversionElectrons =
195	<	new FillerConversionElectrons(cfg, defactive_, conversionInOutTracks, conversionOutInTracks,
196	<	conversionInOutTrackMap, conversionOutInTrackMap);
234	<	const ConversionElectronMap *convElectronMap=0;
235	<	if (fillerConversionElectrons->Active()) {
236	<	fillers_.push_back(fillerConversionElectrons);
237	<	convElectronMap = fillerConversionElectrons->GetConversionElectronMap();
238	<	}
239	<	else {
240	<	delete fillerConversionElectrons;
241	<	fillerConversionElectrons = 0;
242	<	}
243	<
244	<	FillerConversions *fillerConversions = new FillerConversions(cfg, defactive_, convElectronMap);
245	<	const ConversionMap *conversionMap=0;
246	<	if (fillerConversions->Active()) {
247	<	fillers_.push_back(fillerConversions);
248	<	conversionMap = fillerConversions->GetConversionMap();
249	<	}
250	<	else {
251	<	delete fillerConversions;
252	<	fillerConversions = 0;
253	<	}
195	>	new FillerConversionElectrons(cfg,defactive_);
196	>	addActiveFiller(fillerConversionElectrons);
197
198	<	FillerPhotons *fillerPhotons = new FillerPhotons(cfg, defactive_, conversionMap);
199	<	if (fillerPhotons->Active())
257	<	fillers_.push_back(fillerPhotons);
258	<	else {
259	<	delete fillerPhotons;
260	<	fillerPhotons = 0;
261	<	}
198	>	FillerConversions *fillerConversions = new FillerConversions(cfg,defactive_);
199	>	addActiveFiller(fillerConversions);
200
201	+	FillerPhotons *fillerPhotons = new FillerPhotons(cfg,defactive_);
202	+	addActiveFiller(fillerPhotons);
203
204		FillerStableParts *fillerStableParts = new FillerStableParts(cfg,"StableParts",defactive_);
205	<	if (fillerStableParts->Active()) {
206	<	fillers_.push_back(fillerStableParts);
267	<	}
268	<	else {
269	<	delete fillerStableParts;
270	<	fillerStableParts = 0;
271	<	}
205	>	addActiveFiller(fillerStableParts);
206	>
207		FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg,"DecayParts",defactive_);
208	<	if (fillerDecayParts->Active()) {
274	<	fillers_.push_back(fillerDecayParts);
275	<	}
276	<	else {
277	<	delete fillerDecayParts;
278	<	fillerDecayParts = 0;
279	<	}
208	>	addActiveFiller(fillerDecayParts);
209
210		return 1;
211		}

Diff Legend

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