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.7 by loizides, Tue Jul 1 21:11:47 2008 UTC vs.
Revision 1.40 by loizides, Thu Mar 19 22:20:46 2009 UTC

#	Line 2 | Line 2
2
3		#include "MitProd/TreeFiller/interface/FillMitTree.h"
4		#include "FWCore/MessageLogger/interface/MessageLogger.h"
5	+	#include "FWCore/ParameterSet/interface/ParameterSet.h"
6		#include "FWCore/ServiceRegistry/interface/Service.h"
7		#include "MitProd/TreeService/interface/TreeService.h"
8	+	#include "MitProd/ObjectService/interface/ObjectService.h"
9		#include "MitProd/TreeFiller/interface/AssociationMaps.h"
10	+	#include "MitProd/TreeFiller/interface/FillerBasicClusters.h"
11	+	#include "MitProd/TreeFiller/interface/FillerBeamSpot.h"
12	+	#include "MitProd/TreeFiller/interface/FillerCaloJets.h"
13	+	#include "MitProd/TreeFiller/interface/FillerCaloMet.h"
14	+	#include "MitProd/TreeFiller/interface/FillerCaloTowers.h"
15	+	#include "MitProd/TreeFiller/interface/FillerConversionElectrons.h"
16	+	#include "MitProd/TreeFiller/interface/FillerConversions.h"
17	+	#include "MitProd/TreeFiller/interface/FillerDecayParts.h"
18	+	#include "MitProd/TreeFiller/interface/FillerElectrons.h"
19	+	#include "MitProd/TreeFiller/interface/FillerGenJets.h"
20	+	#include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
21	+	#include "MitProd/TreeFiller/interface/FillerMCEventInfo.h"
22	+	#include "MitProd/TreeFiller/interface/FillerMCParticles.h"
23	+	#include "MitProd/TreeFiller/interface/FillerMet.h"
24		#include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
9	–	#include "MitProd/TreeFiller/interface/FillerSimParticles.h"
10	–	#include "MitProd/TreeFiller/interface/FillerTracks.h"
11	–	#include "MitProd/TreeFiller/interface/FillerGsfTracks.h"
25		#include "MitProd/TreeFiller/interface/FillerMuons.h"
26	<	#include "MitProd/TreeFiller/interface/FillerElectrons.h"
27	<	//#include "MitProd/TreeFiller/interface/FillerConvElectrons.h"
28	<	//#include "MitProd/TreeFiller/interface/FillerPhotons.h"
29	<	#include "MitProd/TreeFiller/interface/FillerGenParts.h"
26	>	#include "MitProd/TreeFiller/interface/FillerPFCandidates.h"
27	>	#include "MitProd/TreeFiller/interface/FillerPFJets.h"
28	>	#include "MitProd/TreeFiller/interface/FillerPFMet.h"
29	>	#include "MitProd/TreeFiller/interface/FillerPhotons.h"
30	>	#include "MitProd/TreeFiller/interface/FillerStableParts.h"
31	>	#include "MitProd/TreeFiller/interface/FillerSuperClusters.h"
32	>	#include "MitProd/TreeFiller/interface/FillerTracks.h"
33	>	#include "MitProd/TreeFiller/interface/FillerVertexes.h"
34	>	#include "MitAna/DataTree/interface/Names.h"
35	>	#include "MitAna/DataTree/interface/BranchTable.h"
36
37		using namespace std;
38		using namespace edm;
39		using namespace mithep;
40
41	<	//-------------------------------------------------------------------------------------------------
41	>	mithep::ObjectService *mithep::FillMitTree::os_ = 0;
42	>
43	>	//--------------------------------------------------------------------------------------------------
44		FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
45	<	defactive_(cfg.getUntrackedParameter<bool>("defactive",1))
45	>	defactive_(cfg.getUntrackedParameter<bool>("defactive",1)),
46	>	brtable_(0),
47	>	acfnumber_(-1)
48		{
49		// Constructor.
50	<
50	>
51		if (!configure(cfg)) {
52		throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
53		<< "Could not configure fillers." << "\n";
54		}
55		}
56
57	<	//-------------------------------------------------------------------------------------------------
57	>	//--------------------------------------------------------------------------------------------------
58		FillMitTree::~FillMitTree()
59		{
60	<	// Destructor: nothing to be done here.
60	>	// Destructor.
61	>
62	>	delete brtable_;
63	>	}
64	>
65	>	//--------------------------------------------------------------------------------------------------
66	>	bool FillMitTree::addActiveFiller(BaseFiller *bf)
67	>	{
68	>	// Check if filler is active and add it to list of fillers. Otherwise delete it.
69	>
70	>	if (!bf)
71	>	return 0;
72	>
73	>	if (bf->Active()) {
74	>	fillers_.push_back(bf);
75	>	return 1;
76	>	}
77	>
78	>	delete bf;
79	>	bf = 0;
80	>	return 0;
81		}
82
83	<	//-------------------------------------------------------------------------------------------------
83	>	//--------------------------------------------------------------------------------------------------
84		void FillMitTree::analyze(const edm::Event      &event,
85		const edm::EventSetup &setup)
86		{
87		// Access and copy event content.
88
89	<	// First step: Loop over the data fillers of the various components
89	>	// first step: Loop over the data fillers of the various components
90		for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
91		(*iF)->FillDataBlock(event,setup);
92		}
93
94	<	// Second step: Loop over the link resolution of the various components
94	>	// second step: Loop over the link resolution of the various components
95		for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
96		(*iF)->ResolveLinks(event,setup);
97		}
98	+
99	+	if (brtable_) { // only the first FillMitTree object has to deal with the branch table
100	+	if (acfnumber_==-1) {
101	+	brtable_->Rehash(brtable_->GetSize());
102	+	if (0)
103	+	brtable_->Print();
104	+	}
105	+	if (acfnumber_ != tws_->GetFileNumber()) {
106	+	tws_->StoreObject(brtable_);
107	+	acfnumber_ = tws_->GetFileNumber();
108	+	}
109	+	}
110		}
111
112	<	//-------------------------------------------------------------------------------------------------
112	>	//--------------------------------------------------------------------------------------------------
113		void FillMitTree::beginJob(const edm::EventSetup &event)
114		{
115		// Access the tree and book branches.
116
117		Service<TreeService> ts;
118	<	TreeWriter *tws = ts->get();
119	<	if (! tws) {
118	>	tws_ = ts->get();
119	>	if (!tws_) {
120		throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
121	<	<< "Could not get pointer to tree." << "\n";
121	>	<< "Could not get pointer to tree. "
122	>	<< "Do you have the TreeServie define in your config?" << "\n";
123		return;
124		}
125	<
126	<	// Loop over the various components and book the branches
125	>
126	>	if (os_==0) { // only the first FillMitTree object has to deal with this
127	>	Service<ObjectService> os;
128	>	if (!os.isAvailable()) {
129	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
130	>	<< "Could not get object service. "
131	>	<< "Do you have the ObjectService defined in your config?" << "\n";
132	>	return;
133	>	}
134	>	os_ = &(*os);
135	>	brtable_ = new BranchTable;
136	>	brtable_->SetName(Names::gkBranchTable);
137	>	brtable_->SetOwner();
138	>	os->add(brtable_, brtable_->GetName());
139	>	}
140	>
141	>	// loop over the various components and book the branches
142		for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
143	<	(*iF)->BookDataBlock(*tws);
143	>	edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
144	>	(*iF)->BookDataBlock(*tws_);
145		}
146		}
147
148	<	//-------------------------------------------------------------------------------------------------
148	>	//--------------------------------------------------------------------------------------------------
149		bool FillMitTree::configure(const edm::ParameterSet &cfg)
150		{
151	<	// Configure our fillers.
151	>	// Configure our fillers according to given parameter ("fillers").
152
153	<	FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg);
154	<	if (fillerMetaInfos->Active())
155	<	fillers_.push_back(fillerMetaInfos);
156	<	else
157	<	delete fillerMetaInfos;
158	<
159	<	FillerGenParts *fillerGenParts = new FillerGenParts(cfg,defactive_);
160	<	if (fillerGenParts->Active())
161	<	fillers_.push_back(fillerGenParts);
162	<	else
163	<	delete fillerGenParts;
164	<
165	<	FillerSimParticles *fillerSimParticles = new FillerSimParticles(cfg,defactive_);
166	<	const SimParticleMap* simParticleMap=0;
167	<	if (fillerSimParticles->Active()) {
168	<	fillers_.push_back(fillerSimParticles);
169	<	simParticleMap = fillerSimParticles->GetSimParticleMap();
170	<	}
171	<	else
172	<	delete fillerSimParticles;
173	<
174	<	FillerTracks *fillerGeneralTracks = new FillerTracks(cfg,"GeneralTracks",defactive_,simParticleMap);
175	<	const TrackMap* generalTrackMap=0;
176	<	if (fillerGeneralTracks->Active()) {
177	<	fillers_.push_back(fillerGeneralTracks);
178	<	generalTrackMap = fillerGeneralTracks->GetTrackMap();
179	<	}
180	<	else
181	<	delete fillerGeneralTracks;
182	<
183	<	FillerTracks *fillerStandaloneMuonTracks = new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
184	<	const TrackMap* standaloneMuonTrackMap=0;
185	<	if (fillerStandaloneMuonTracks->Active()) {
186	<	fillers_.push_back(fillerStandaloneMuonTracks);
187	<	standaloneMuonTrackMap = fillerStandaloneMuonTracks->GetTrackMap();
188	<	}
189	<	else
190	<	delete fillerStandaloneMuonTracks;
191	<
192	<	FillerTracks *fillerStandaloneMuonTracksVtx =
193	<	new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
194	<	const TrackMap* standaloneMuonTrackVtxMap=0;
195	<	if (fillerStandaloneMuonTracksVtx->Active()) {
196	<	fillers_.push_back(fillerStandaloneMuonTracksVtx);
197	<	standaloneMuonTrackVtxMap = fillerStandaloneMuonTracksVtx->GetTrackMap();
198	<	}
199	<	else
200	<	delete fillerStandaloneMuonTracksVtx;
201	<
202	<	FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
203	<	const TrackMap* globalMuonTrackMap=0;
204	<	if (fillerGlobalMuonTracks->Active()) {
205	<	fillers_.push_back(fillerGlobalMuonTracks);
206	<	globalMuonTrackMap = fillerGlobalMuonTracks->GetTrackMap();
207	<	}
208	<	else
209	<	delete fillerGlobalMuonTracks;
210	<
211	<	FillerTracks *fillerConversionInOutTracks =
212	<	new FillerTracks(cfg,"ConversionInOutTracks",defactive_,simParticleMap);
213	<	const TrackMap* conversionInOutTrackMap=0;
214	<	const TrackCol* conversionInOutTracks=0;
215	<	if (fillerConversionInOutTracks->Active()) {
216	<	fillers_.push_back(fillerConversionInOutTracks);
217	<	conversionInOutTrackMap = fillerConversionInOutTracks->GetTrackMap();
218	<	conversionInOutTracks = fillerConversionInOutTracks->GetTrackCol();
219	<	}
220	<	else
221	<	delete fillerConversionInOutTracks;
153	>	std::vector<std::string> pars;
154	>	if (cfg.exists("fillers"))
155	>	pars=cfg.getUntrackedParameter<vector<string> >("fillers");
156	>	else
157	>	cfg.getParameterSetNames(pars, false);
158	>
159	>	// loop over psets
160	>	for (unsigned int i = 0; i<pars.size(); ++i) {
161	>
162	>	const string name(pars.at(i));
163	>
164	>	string ftype("Filler" + name);
165	>	if (cfg.existsAs<ParameterSet>(name,0)) {
166	>	ParameterSet next(cfg.getUntrackedParameter<ParameterSet>(name));
167	>	if (!next.exists("fillerType")) {
168	>	edm::LogError("FillMitTree") << "Can not determine fillerType for pset named "
169	>	<< name << std::endl;
170	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
171	>	<< "Can not determine fillerType for pset named "
172	>	<< name << std::endl;
173	>	}
174	>	ftype = next.getUntrackedParameter<string>("fillerType");
175	>	}
176	>
177	>	edm::LogInfo("FillMitTree") << "Attempting to configure '" << ftype
178	>	<< "' for '" << name << "'" << std::endl;
179	>
180	>	if (ftype.compare("FillerMetaInfos")==0) {
181	>	FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg, name.c_str(), defactive_);
182	>	addActiveFiller(fillerMetaInfos);
183	>	continue;
184	>	}
185	>
186	>	if (ftype.compare("FillerMCParticles")==0) {
187	>	FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg, name.c_str(), defactive_);
188	>	addActiveFiller(fillerMCParticles);
189	>	continue;
190	>	}
191	>
192	>	if (ftype.compare("FillerMCEventInfo")==0) {
193	>	FillerMCEventInfo *fillerMCEventInfo = new FillerMCEventInfo(cfg, name.c_str(), defactive_);
194	>	addActiveFiller(fillerMCEventInfo);
195	>	continue;
196	>	}
197	>
198	>	if (ftype.compare("FillerBeamSpot")==0) {
199	>	FillerBeamSpot *fillerBeamSpot = new FillerBeamSpot(cfg, name.c_str(), defactive_);
200	>	addActiveFiller(fillerBeamSpot);
201	>	continue;
202	>	}
203	>
204	>	if (ftype.compare("FillerVertexes")==0) {
205	>	FillerVertexes *fillerVertexes = new FillerVertexes(cfg, name.c_str(), defactive_);
206	>	addActiveFiller(fillerVertexes);
207	>	continue;
208	>	}
209	>
210	>	if (ftype.compare("FillerCaloTowers")==0) {
211	>	FillerCaloTowers *fillerCaloTowers = new FillerCaloTowers(cfg, name.c_str(), defactive_);
212	>	addActiveFiller(fillerCaloTowers);
213	>	continue;
214	>	}
215	>
216	>	if (ftype.compare("FillerGenJets")==0) {
217	>	FillerGenJets *fillerGenJets = new FillerGenJets(cfg, name.c_str(), defactive_);
218	>	addActiveFiller(fillerGenJets);
219	>	continue;
220	>	}
221	>
222	>	if (ftype.compare("FillerCaloJets")==0) {
223	>	FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg, name.c_str(), defactive_);
224	>	addActiveFiller(fillerCaloJets);
225	>	continue;
226	>	}
227
228	<	FillerTracks *fillerConversionOutInTracks =
229	<	new FillerTracks(cfg,"ConversionOutInTracks",defactive_,simParticleMap);
230	<	const TrackMap* conversionOutInTrackMap=0;
231	<	const TrackCol* conversionOutInTracks=0;
232	<	if (fillerConversionOutInTracks->Active()) {
233	<	fillers_.push_back(fillerConversionOutInTracks);
234	<	conversionOutInTrackMap = fillerConversionOutInTracks->GetTrackMap();
235	<	conversionOutInTracks = fillerConversionOutInTracks->GetTrackCol();
236	<	}
237	<	else
238	<	delete fillerConversionOutInTracks;
228	>	if (ftype.compare("FillerMet")==0) {
229	>	FillerMet *fillerMet = new FillerMet(cfg, name.c_str(), defactive_);
230	>	addActiveFiller(fillerMet);
231	>	continue;
232	>	}
233	>
234	>	if (ftype.compare("FillerCaloMet")==0) {
235	>	FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg, name.c_str(), defactive_);
236	>	addActiveFiller(fillerCaloMet);
237	>	continue;
238	>	}
239
240	<	FillerGsfTracks *fillerGsfTracks = new FillerGsfTracks(cfg,"GsfTracks",defactive_,simParticleMap);
241	<	const GsfTrackMap* gsfTrackMap=0;
242	<	if (fillerGsfTracks->Active()) {
243	<	fillers_.push_back(fillerGsfTracks);
244	<	gsfTrackMap = (GsfTrackMap*)fillerGsfTracks->GetTrackMap();
245	<	}
246	<	else
247	<	delete fillerGsfTracks;
240	>	if (ftype.compare("FillerPFMet")==0) {
241	>	FillerPFMet *fillerPFMet = new FillerPFMet(cfg, name.c_str(), defactive_);
242	>	addActiveFiller(fillerPFMet);
243	>	continue;
244	>	}
245	>
246	>	if (ftype.compare("FillerBasicClusters")==0) {
247	>	FillerBasicClusters *fillerBasicClusters =
248	>	new FillerBasicClusters(cfg, name.c_str(), defactive_);
249	>	addActiveFiller(fillerBasicClusters);
250	>	continue;
251	>	}
252	>
253	>	if (ftype.compare("FillerSuperClusters")==0) {
254	>	FillerSuperClusters *fillerSuperClusters =
255	>	new FillerSuperClusters(cfg, name.c_str(), defactive_);
256	>	addActiveFiller(fillerSuperClusters);
257	>	continue;
258	>	}
259	>
260	>	if (ftype.compare("FillerTracks")==0) {
261	>	FillerTracks *fillerTracks = new FillerTracks(cfg, name.c_str(), defactive_);
262	>	addActiveFiller(fillerTracks);
263	>	continue;
264	>	}
265	>
266	>	if (ftype.compare("FillerMuons")==0) {
267	>	FillerMuons *fillerMuons = new FillerMuons(cfg, name.c_str(), defactive_);
268	>	addActiveFiller(fillerMuons);
269	>	continue;
270	>	}
271	>
272	>	if (ftype.compare("FillerElectrons")==0) {
273	>	FillerElectrons *fillerElectrons = new FillerElectrons(cfg, name.c_str(), defactive_);
274	>	addActiveFiller(fillerElectrons);
275	>	continue;
276	>	}
277	>
278	>	if (ftype.compare("FillerConversionElectrons")==0) {
279	>	FillerConversionElectrons *fillerConversionElectrons =
280	>	new FillerConversionElectrons(cfg, name.c_str(), defactive_);
281	>	addActiveFiller(fillerConversionElectrons);
282	>	continue;
283	>	}
284	>
285	>	if (ftype.compare("FillerConversions")==0) {
286	>	FillerConversions *fillerConversions = new FillerConversions(cfg, name.c_str(), defactive_);
287	>	addActiveFiller(fillerConversions);
288	>	continue;
289	>	}
290	>
291	>	if (ftype.compare("FillerPhotons")==0) {
292	>	FillerPhotons *fillerPhotons = new FillerPhotons(cfg, name.c_str(), defactive_);
293	>	addActiveFiller(fillerPhotons);
294	>	continue;
295	>	}
296	>
297	>	if (ftype.compare("FillerStableParts")==0) {
298	>	FillerStableParts *fillerStableParts = new FillerStableParts(cfg, name.c_str(), defactive_);
299	>	addActiveFiller(fillerStableParts);
300	>	continue;
301	>	}
302	>
303	>	if (ftype.compare("FillerDecayParts")==0) {
304	>	FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg, name.c_str(), defactive_);
305	>	addActiveFiller(fillerDecayParts);
306	>	continue;
307	>	}
308
309	<	FillerMuons *fillerMuons =
310	<	new FillerMuons(cfg,defactive_,globalMuonTrackMap,standaloneMuonTrackMap,
311	<	standaloneMuonTrackVtxMap,generalTrackMap);
312	<	if (fillerMuons->Active())
313	<	fillers_.push_back(fillerMuons);
314	<	else
315	<	delete fillerMuons;
309	>	if (ftype.compare("FillerPFCandidates")==0) {
310	>	FillerPFCandidates *fillerPFCands = new FillerPFCandidates(cfg, name.c_str(), defactive_);
311	>	addActiveFiller(fillerPFCands);
312	>	continue;
313	>	}
314	>
315	>	if (ftype.compare("FillerPFJets")==0) {
316	>	FillerPFJets *fillerPFJets = new FillerPFJets(cfg, name.c_str(), defactive_);
317	>	addActiveFiller(fillerPFJets);
318	>	continue;
319	>	}
320
321	<	FillerElectrons *fillerElectrons = new FillerElectrons(cfg,defactive_,gsfTrackMap,generalTrackMap);
322	<	if (fillerElectrons->Active())
323	<	fillers_.push_back(fillerElectrons);
324	<	else
184	<	delete fillerElectrons;
185	<
186	<	#if 0
187	<	FillerConversionElectrons *fillerConversionElectrons =
188	<	new FillerConversionElectrons(cfg,defactive_,conversionInOutTracks,conversionOutInTracks,
189	<	conversionInOutTrackMap,conversionOutInTrackMap);
190	<
191	<	const ConversionElectronMap* convElectronMap=0;
192	<	if (fillerConversionElectrons->Active()) {
193	<	fillers_.push_back(fillerConversionElectrons);
194	<	convElectronMap = fillerConversionElectrons->GetConversionElectronMap();
321	>	edm::LogError("FillMitTree")
322	>	<< "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
323	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
324	>	<< "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
325		}
196	–	else
197	–	delete fillerConversionElectrons;
198	–
199	–	FillerPhotons *fillerPhotons = new FillerPhotons(cfg,defactive_,convElectronMap);
200	–	if (fillerPhotons->Active())
201	–	fillers_.push_back(fillerPhotons);
202	–	else
203	–	delete fillerPhotons;
204	–	#endif
326
327		return 1;
328		}
329
330	<	//-------------------------------------------------------------------------------------------------
330	>	//--------------------------------------------------------------------------------------------------
331		void FillMitTree::endJob()
332		{
333		// Delete fillers.

Diff Legend

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