ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.46
Committed: Tue Aug 11 17:28:57 2009 UTC (15 years, 8 months ago) by loizides
Content type: text/plain
Branch: MAIN
Changes since 1.45: +4 -1 lines
Log Message: 
Put BranchRef call in FillMitTree.

File Contents

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