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

Diff Legend

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