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.23 by sixie, Mon Sep 29 17:07:56 2008 UTC vs.
Revision 1.37 by loizides, Thu Mar 19 16:15:35 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"
9	–	#include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
10	–	#include "MitProd/TreeFiller/interface/FillerTracks.h"
11	–	#include "MitProd/TreeFiller/interface/FillerGsfTracks.h"
10		#include "MitProd/TreeFiller/interface/FillerBasicClusters.h"
11	<	#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/FillerGenJets.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/FillerConversions.h"
14	>	#include "MitProd/TreeFiller/interface/FillerCaloTowers.h"
15		#include "MitProd/TreeFiller/interface/FillerConversionElectrons.h"
16	<	#include "MitProd/TreeFiller/interface/FillerPhotons.h"
23	<	#include "MitProd/TreeFiller/interface/FillerMCParticles.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"
25	+	#include "MitProd/TreeFiller/interface/FillerMuons.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/FillerPATMuons.h"
32	<	#include "MitProd/TreeFiller/interface/FillerPATElectrons.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;
#	Line 34 | Line 42 | mithep::ObjectService *mithep::FillMitTr
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_(new BranchTable)
47		{
48		// Constructor.
49
#	Line 42 | Line 51 | FillMitTree::FillMitTree(const edm::Para
51		throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
52		<< "Could not configure fillers." << "\n";
53		}
54	+
55	+	brtable_->SetName(Names::gkBranchTable);
56	+	brtable_->SetOwner();
57		}
58
59		//--------------------------------------------------------------------------------------------------
60		FillMitTree::~FillMitTree()
61		{
62	<	// Destructor: nothing to be done here.
62	>	// Destructor.
63	>
64	>	delete brtable_;
65		}
66
67		//--------------------------------------------------------------------------------------------------
#	Line 55 | Line 69 | bool FillMitTree::addActiveFiller(BaseFi
69		{
70		// Check if filler is active and add it to list of fillers. Otherwise delete it.
71
72	+	if (!bf)
73	+	return 0;
74	+
75		if (bf->Active()) {
76		fillers_.push_back(bf);
77		return 1;
#	Line 71 | Line 88 | void FillMitTree::analyze(const edm::Eve
88		{
89		// Access and copy event content.
90
91	<	// First step: Loop over the data fillers of the various components
91	>	// first step: Loop over the data fillers of the various components
92		for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
93		(*iF)->FillDataBlock(event,setup);
94		}
95
96	<	// Second step: Loop over the link resolution of the various components
96	>	// second step: Loop over the link resolution of the various components
97		for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
98		(*iF)->ResolveLinks(event,setup);
99		}
#	Line 105 | Line 122 | void FillMitTree::beginJob(const edm::Ev
122		return;
123		}
124		os_ = &(*os);
125	+	os->add(brtable_, brtable_->GetName());
126		}
127
128	<	// Loop over the various components and book the branches
128	>	// loop over the various components and book the branches
129		for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
130		edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
131		(*iF)->BookDataBlock(*tws);
#	Line 117 | Line 135 | void FillMitTree::beginJob(const edm::Ev
135		//--------------------------------------------------------------------------------------------------
136		bool FillMitTree::configure(const edm::ParameterSet &cfg)
137		{
138	<	// Configure our fillers.
138	>	// Configure our fillers according to given parameter ("fillers").
139	>
140	>	std::vector<std::string> pars;
141	>	if (cfg.exists("fillers"))
142	>	pars=cfg.getUntrackedParameter<vector<string> >("fillers");
143	>	else
144	>	cfg.getParameterSetNames(pars, false);
145	>
146	>	// loop over psets
147	>	for (unsigned int i = 0; i<pars.size(); ++i) {
148	>
149	>	const string name(pars.at(i));
150	>
151	>	string ftype("Filler" + name);
152	>	if (cfg.existsAs<ParameterSet>(name,0)) {
153	>	ParameterSet next(cfg.getUntrackedParameter<ParameterSet>(name));
154	>	if (!next.exists("fillerType")) {
155	>	edm::LogError("FillMitTree") << "Can not determine fillerType for pset named "
156	>	<< name << std::endl;
157	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
158	>	<< "Can not determine fillerType for pset named "
159	>	<< name << std::endl;
160	>	}
161	>	ftype = next.getUntrackedParameter<string>("fillerType");
162	>	}
163	>
164	>	edm::LogInfo("FillMitTree") << "Attempting to configure '" << ftype
165	>	<< "' for '" << name << "'" << std::endl;
166
167	<	FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg,defactive_);
168	<	addActiveFiller(fillerMetaInfos);
167	>	if (ftype.compare("FillerMetaInfos")==0) {
168	>	FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg, name.c_str(), defactive_);
169	>	addActiveFiller(fillerMetaInfos);
170	>	continue;
171	>	}
172
173	<	FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg,"MCParticles",defactive_);
174	<	addActiveFiller(fillerMCParticles);
173	>	if (ftype.compare("FillerMCParticles")==0) {
174	>	FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg, name.c_str(), defactive_);
175	>	addActiveFiller(fillerMCParticles);
176	>	continue;
177	>	}
178
179	<	FillerCaloTowers *fillerCaloTowers =
180	<	new FillerCaloTowers(cfg, "CaloTowers", defactive_);
181	<	addActiveFiller(fillerCaloTowers);
182	<
183	<	FillerGenJets *fillerIC5GenJets = new FillerGenJets(cfg,"IC5GenJets",defactive_);
133	<	addActiveFiller(fillerIC5GenJets);
134	<
135	<	FillerGenJets *fillerSC5GenJets = new FillerGenJets(cfg,"SC5GenJets",defactive_);
136	<	addActiveFiller(fillerSC5GenJets);
137	<
138	<	FillerGenJets *fillerSC7GenJets = new FillerGenJets(cfg,"SC7GenJets",defactive_);
139	<	addActiveFiller(fillerSC7GenJets);
140	<
141	<	FillerGenJets *fillerKT4GenJets = new FillerGenJets(cfg,"KT4GenJets",defactive_);
142	<	addActiveFiller(fillerKT4GenJets);
143	<
144	<	FillerGenJets *fillerKT6GenJets = new FillerGenJets(cfg,"KT6GenJets",defactive_);
145	<	addActiveFiller(fillerKT6GenJets);
146	<
147	<	FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg,"CaloJets",defactive_);
148	<	addActiveFiller(fillerCaloJets);
149	<
150	<	FillerCaloJets *fillerItrCone5Jets = new FillerCaloJets(cfg,"ItrCone5Jets",defactive_);
151	<	addActiveFiller(fillerItrCone5Jets);
152	<
153	<	FillerCaloJets *fillerSisCone5Jets = new FillerCaloJets(cfg,"SisCone5Jets",defactive_);
154	<	addActiveFiller(fillerSisCone5Jets);
155	<
156	<	FillerCaloJets *fillerSisCone7Jets = new FillerCaloJets(cfg,"SisCone7Jets",defactive_);
157	<	addActiveFiller(fillerSisCone7Jets);
158	<
159	<	FillerCaloJets *fillerKt4Jets = new FillerCaloJets(cfg,"Kt4Jets",defactive_);
160	<	addActiveFiller(fillerKt4Jets);
161	<
162	<	FillerCaloJets *fillerKt6Jets = new FillerCaloJets(cfg,"Kt6Jets",defactive_);
163	<	addActiveFiller(fillerKt6Jets);
164	<
165	<	FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg,"CaloMet",defactive_);
166	<	addActiveFiller(fillerCaloMet);
167	<
168	<	FillerCaloMet *fillerItrCone5Met = new FillerCaloMet(cfg,"ItrCone5Met",defactive_);
169	<	addActiveFiller(fillerItrCone5Met);
170	<
171	<	FillerCaloMet *fillerSisCone5Met = new FillerCaloMet(cfg,"SisCone5Met",defactive_);
172	<	addActiveFiller(fillerSisCone5Met);
173	<
174	<	FillerCaloMet *fillerSisCone7Met = new FillerCaloMet(cfg,"SisCone7Met",defactive_);
175	<	addActiveFiller(fillerSisCone7Met);
176	<
177	<	FillerCaloMet *fillerKt4Met = new FillerCaloMet(cfg,"Kt4Met",defactive_);
178	<	addActiveFiller(fillerKt4Met);
179	<
180	<	FillerCaloMet *fillerKt6Met = new FillerCaloMet(cfg,"Kt6Met",defactive_);
181	<	addActiveFiller(fillerKt6Met);
182	<
183	<	FillerTracks *fillerGeneralTracks = new FillerTracks(cfg,"GeneralTracks",defactive_);
184	<	addActiveFiller(fillerGeneralTracks);
185	<
186	<	FillerTracks *fillerStandaloneMuonTracks =
187	<	new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
188	<	addActiveFiller(fillerStandaloneMuonTracks);
189	<
190	<	FillerTracks *fillerStandaloneMuonTracksVtx =
191	<	new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
192	<	addActiveFiller(fillerStandaloneMuonTracksVtx);
179	>	if (ftype.compare("FillerMCEventInfo")==0) {
180	>	FillerMCEventInfo *fillerMCEventInfo = new FillerMCEventInfo(cfg, name.c_str(), defactive_);
181	>	addActiveFiller(fillerMCEventInfo);
182	>	continue;
183	>	}
184
185	<	FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
186	<	addActiveFiller(fillerGlobalMuonTracks);
185	>	if (ftype.compare("FillerBeamSpot")==0) {
186	>	FillerBeamSpot *fillerBeamSpot = new FillerBeamSpot(cfg, name.c_str(), defactive_);
187	>	addActiveFiller(fillerBeamSpot);
188	>	continue;
189	>	}
190	>	if (ftype.compare("FillerVertexes")==0) {
191	>	FillerVertexes *fillerVertexes = new FillerVertexes(cfg, name.c_str(), defactive_);
192	>	addActiveFiller(fillerVertexes);
193	>	continue;
194	>	}
195	>
196	>	if (ftype.compare("FillerCaloTowers")==0) {
197	>	FillerCaloTowers *fillerCaloTowers = new FillerCaloTowers(cfg, name.c_str(), defactive_);
198	>	addActiveFiller(fillerCaloTowers);
199	>	continue;
200	>	}
201	>
202	>	if (ftype.compare("FillerGenJets")==0) {
203	>	FillerGenJets *fillerGenJets = new FillerGenJets(cfg, name.c_str(), defactive_);
204	>	addActiveFiller(fillerGenJets);
205	>	continue;
206	>	}
207	>
208	>	if (ftype.compare("FillerCaloJets")==0) {
209	>	FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg, name.c_str(), defactive_);
210	>	addActiveFiller(fillerCaloJets);
211	>	continue;
212	>	}
213
214	<	FillerTracks *fillerConversionInOutTracks =
215	<	new FillerTracks(cfg,"ConversionInOutTracks",defactive_);
216	<	addActiveFiller(fillerConversionInOutTracks);
217	<
218	<	FillerTracks *fillerConversionOutInTracks =
219	<	new FillerTracks(cfg,"ConversionOutInTracks",defactive_);
220	<	addActiveFiller(fillerConversionOutInTracks);
221	<
222	<	FillerGsfTracks *fillerGsfTracks = new FillerGsfTracks(cfg,"GsfTracks",defactive_);
223	<	addActiveFiller(fillerGsfTracks);
224	<
225	<	FillerBasicClusters *fillerBarrelBasicClusters =
226	<	new FillerBasicClusters(cfg, "BarrelBasicClusters", defactive_);
227	<	addActiveFiller(fillerBarrelBasicClusters);
228	<
229	<	FillerSuperClusters *fillerBarrelSuperClusters =
230	<	new FillerSuperClusters(cfg,"BarrelSuperClusters", defactive_);
231	<	addActiveFiller(fillerBarrelSuperClusters);
232	<
233	<	FillerBasicClusters *fillerEndcapBasicClusters =
234	<	new FillerBasicClusters(cfg,"EndcapBasicClusters", defactive_);
235	<	addActiveFiller(fillerEndcapBasicClusters);
236	<
237	<	FillerSuperClusters *fillerEndcapSuperClusters =
238	<	new FillerSuperClusters(cfg,"EndcapSuperClusters", defactive_);
239	<	addActiveFiller(fillerEndcapSuperClusters);
240	<
241	<	FillerMuons *fillerMuons = new FillerMuons(cfg,defactive_);
242	<	addActiveFiller(fillerMuons);
243	<
244	<	FillerElectrons *fillerElectrons = new FillerElectrons(cfg,defactive_);
245	<	addActiveFiller(fillerElectrons);
246	<
247	<	FillerConversionElectrons *fillerConversionElectrons =
248	<	new FillerConversionElectrons(cfg,defactive_);
249	<	addActiveFiller(fillerConversionElectrons);
250	<
251	<	FillerConversions *fillerConversions = new FillerConversions(cfg,defactive_);
252	<	addActiveFiller(fillerConversions);
253	<
254	<	FillerPhotons *fillerPhotons = new FillerPhotons(cfg,defactive_);
255	<	addActiveFiller(fillerPhotons);
256	<
257	<	FillerStableParts *fillerStableParts = new FillerStableParts(cfg,"StableParts",defactive_);
258	<	addActiveFiller(fillerStableParts);
259	<
260	<	FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg,"DecayParts",defactive_);
261	<	addActiveFiller(fillerDecayParts);
214	>	if (ftype.compare("FillerMet")==0) {
215	>	FillerMet *fillerMet = new FillerMet(cfg, name.c_str(), defactive_);
216	>	addActiveFiller(fillerMet);
217	>	continue;
218	>	}
219	>
220	>	if (ftype.compare("FillerCaloMet")==0) {
221	>	FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg, name.c_str(), defactive_);
222	>	addActiveFiller(fillerCaloMet);
223	>	continue;
224	>	}
225	>
226	>	if (ftype.compare("FillerPFMet")==0) {
227	>	FillerPFMet *fillerPFMet = new FillerPFMet(cfg, name.c_str(), defactive_);
228	>	addActiveFiller(fillerPFMet);
229	>	continue;
230	>	}
231	>
232	>	if (ftype.compare("FillerBasicClusters")==0) {
233	>	FillerBasicClusters *fillerBasicClusters =
234	>	new FillerBasicClusters(cfg, name.c_str(), defactive_);
235	>	addActiveFiller(fillerBasicClusters);
236	>	continue;
237	>	}
238	>
239	>	if (ftype.compare("FillerSuperClusters")==0) {
240	>	FillerSuperClusters *fillerSuperClusters =
241	>	new FillerSuperClusters(cfg, name.c_str(), defactive_);
242	>	addActiveFiller(fillerSuperClusters);
243	>	continue;
244	>	}
245	>
246	>	if (ftype.compare("FillerTracks")==0) {
247	>	FillerTracks *fillerTracks = new FillerTracks(cfg, name.c_str(), defactive_);
248	>	addActiveFiller(fillerTracks);
249	>	continue;
250	>	}
251	>
252	>	if (ftype.compare("FillerMuons")==0) {
253	>	FillerMuons *fillerMuons = new FillerMuons(cfg, name.c_str(), defactive_);
254	>	addActiveFiller(fillerMuons);
255	>	continue;
256	>	}
257	>
258	>	if (ftype.compare("FillerElectrons")==0) {
259	>	FillerElectrons *fillerElectrons = new FillerElectrons(cfg, name.c_str(), defactive_);
260	>	addActiveFiller(fillerElectrons);
261	>	continue;
262	>	}
263	>
264	>	if (ftype.compare("FillerConversionElectrons")==0) {
265	>	FillerConversionElectrons *fillerConversionElectrons =
266	>	new FillerConversionElectrons(cfg, name.c_str(), defactive_);
267	>	addActiveFiller(fillerConversionElectrons);
268	>	continue;
269	>	}
270	>
271	>	if (ftype.compare("FillerConversions")==0) {
272	>	FillerConversions *fillerConversions = new FillerConversions(cfg, name.c_str(), defactive_);
273	>	addActiveFiller(fillerConversions);
274	>	continue;
275	>	}
276	>
277	>	if (ftype.compare("FillerPhotons")==0) {
278	>	FillerPhotons *fillerPhotons = new FillerPhotons(cfg, name.c_str(), defactive_);
279	>	addActiveFiller(fillerPhotons);
280	>	continue;
281	>	}
282	>
283	>	if (ftype.compare("FillerStableParts")==0) {
284	>	FillerStableParts *fillerStableParts = new FillerStableParts(cfg, name.c_str(), defactive_);
285	>	addActiveFiller(fillerStableParts);
286	>	continue;
287	>	}
288	>
289	>	if (ftype.compare("FillerDecayParts")==0) {
290	>	FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg, name.c_str(), defactive_);
291	>	addActiveFiller(fillerDecayParts);
292	>	continue;
293	>	}
294	>
295	>	if (ftype.compare("FillerPFCandidates")==0) {
296	>	FillerPFCandidates *fillerPFCands = new FillerPFCandidates(cfg, name.c_str(), defactive_);
297	>	addActiveFiller(fillerPFCands);
298	>	continue;
299	>	}
300	>
301	>	if (ftype.compare("FillerPFJets")==0) {
302	>	FillerPFJets *fillerPFJets = new FillerPFJets(cfg, name.c_str(), defactive_);
303	>	addActiveFiller(fillerPFJets);
304	>	continue;
305	>	}
306	>
307	>	edm::LogError("FillMitTree")
308	>	<< "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
309	>	throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
310	>	<< "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
311	>	}
312
313		return 1;
314		}

Diff Legend

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