ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.54
Committed: Thu Feb 11 14:55:27 2010 UTC (15 years, 2 months ago) by yilmaz
Content type: text/plain
Branch: MAIN
Changes since 1.53: +14 -13 lines
Log Message: 
event setup stuff moved to beginRun from beginJob

File Contents

# User Rev Content
1 yilmaz 1.54 // $Id: FillMitTree.cc,v 1.53 2009/12/12 22:33:52 bendavid 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.16 #include "MitProd/ObjectService/interface/ObjectService.h"
8 loizides 1.6 #include "MitProd/TreeFiller/interface/AssociationMaps.h"
9 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerBasicClusters.h"
10 bendavid 1.28 #include "MitProd/TreeFiller/interface/FillerBeamSpot.h"
11 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerCaloJets.h"
12     #include "MitProd/TreeFiller/interface/FillerCaloMet.h"
13 bendavid 1.41 #include "MitProd/TreeFiller/interface/FillerCaloTaus.h"
14 sixie 1.20 #include "MitProd/TreeFiller/interface/FillerCaloTowers.h"
15 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerConversions.h"
16     #include "MitProd/TreeFiller/interface/FillerDecayParts.h"
17 loizides 1.6 #include "MitProd/TreeFiller/interface/FillerElectrons.h"
18 loizides 1.52 #include "MitProd/TreeFiller/interface/FillerEvtSelData.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 loizides 1.48 #include "MitProd/TreeFiller/interface/FillerMCVertexes.h"
24 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerMet.h"
25     #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
26 loizides 1.45 #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
27 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerMuons.h"
28     #include "MitProd/TreeFiller/interface/FillerPFCandidates.h"
29 bendavid 1.33 #include "MitProd/TreeFiller/interface/FillerPFJets.h"
30 bendavid 1.34 #include "MitProd/TreeFiller/interface/FillerPFMet.h"
31 bendavid 1.41 #include "MitProd/TreeFiller/interface/FillerPFTaus.h"
32 bendavid 1.8 #include "MitProd/TreeFiller/interface/FillerPhotons.h"
33 loizides 1.48 #include "MitProd/TreeFiller/interface/FillerPixelHits.h"
34 paus 1.13 #include "MitProd/TreeFiller/interface/FillerStableParts.h"
35 loizides 1.51 #include "MitProd/TreeFiller/interface/FillerStripHits.h"
36 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerSuperClusters.h"
37     #include "MitProd/TreeFiller/interface/FillerTracks.h"
38     #include "MitProd/TreeFiller/interface/FillerVertexes.h"
39 loizides 1.35 #include "MitAna/DataTree/interface/Names.h"
40     #include "MitAna/DataTree/interface/BranchTable.h"
41 bendavid 1.49 #include "MitCommon/OptIO/interface/OptInt.h"
42 loizides 1.21
43 paus 1.1 using namespace std;
44     using namespace edm;
45     using namespace mithep;
46    
47 loizides 1.16 mithep::ObjectService *mithep::FillMitTree::os_ = 0;
48    
49 loizides 1.9 //--------------------------------------------------------------------------------------------------
50 loizides 1.7 FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
51 loizides 1.35 defactive_(cfg.getUntrackedParameter<bool>("defactive",1)),
52 loizides 1.40 brtable_(0),
53 bendavid 1.49 acfnumber_(-1),
54     tws_(new TreeWriter(Names::gkEvtTreeName,0))
55 paus 1.1 {
56 loizides 1.6 // Constructor.
57 loizides 1.15
58 loizides 1.3 if (!configure(cfg)) {
59     throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
60     << "Could not configure fillers." << "\n";
61     }
62 paus 1.1 }
63    
64 loizides 1.9 //--------------------------------------------------------------------------------------------------
65 paus 1.1 FillMitTree::~FillMitTree()
66     {
67 loizides 1.35 // Destructor.
68    
69     delete brtable_;
70 bendavid 1.49 delete tws_;
71 paus 1.1 }
72    
73 loizides 1.9 //--------------------------------------------------------------------------------------------------
74 loizides 1.17 bool FillMitTree::addActiveFiller(BaseFiller *bf)
75     {
76     // Check if filler is active and add it to list of fillers. Otherwise delete it.
77    
78 loizides 1.35 if (!bf)
79     return 0;
80    
81 loizides 1.17 if (bf->Active()) {
82     fillers_.push_back(bf);
83     return 1;
84     }
85    
86     delete bf;
87     bf = 0;
88     return 0;
89     }
90    
91     //--------------------------------------------------------------------------------------------------
92 bendavid 1.53 void FillMitTree::beginRun(edm::Run const &run, edm::EventSetup const &setup)
93     {
94 yilmaz 1.54
95     // loop over the various components and book the branches
96     for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
97     edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
98     (*iF)->BookDataBlock(*tws_, setup);
99     }
100    
101     // call branch ref for the event tree
102     if (brtable_ && tws_->GetTree())
103     tws_->GetTree()->BranchRef();
104    
105 bendavid 1.53 // Access and copy event content.
106    
107     // first step: Loop over the data fillers of the various components
108     for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
109     (*iF)->FillRunBlock(run,setup);
110     }
111 yilmaz 1.54
112 bendavid 1.53 }
113    
114     //--------------------------------------------------------------------------------------------------
115 loizides 1.3 void FillMitTree::analyze(const edm::Event &event,
116     const edm::EventSetup &setup)
117     {
118     // Access and copy event content.
119 bendavid 1.49
120     //tree writer begin event actions
121     tws_->BeginEvent(kTRUE);
122 loizides 1.3
123 loizides 1.32 // first step: Loop over the data fillers of the various components
124 loizides 1.3 for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
125     (*iF)->FillDataBlock(event,setup);
126     }
127    
128 loizides 1.32 // second step: Loop over the link resolution of the various components
129 loizides 1.3 for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
130     (*iF)->ResolveLinks(event,setup);
131     }
132 loizides 1.38
133 loizides 1.40 if (brtable_) { // only the first FillMitTree object has to deal with the branch table
134     if (acfnumber_==-1) {
135     brtable_->Rehash(brtable_->GetSize());
136     if (0)
137     brtable_->Print();
138     }
139     if (acfnumber_ != tws_->GetFileNumber()) {
140     tws_->StoreObject(brtable_);
141     acfnumber_ = tws_->GetFileNumber();
142     }
143     }
144 bendavid 1.49
145     //tree writer end of event actions
146     tws_->EndEvent(kTRUE);
147    
148 loizides 1.3 }
149    
150 loizides 1.9 //--------------------------------------------------------------------------------------------------
151 yilmaz 1.54 void FillMitTree::beginJob()
152 paus 1.1 {
153 loizides 1.2 // Access the tree and book branches.
154    
155 loizides 1.40 if (os_==0) { // only the first FillMitTree object has to deal with this
156 loizides 1.16 Service<ObjectService> os;
157     if (!os.isAvailable()) {
158     throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
159 loizides 1.17 << "Could not get object service. "
160     << "Do you have the ObjectService defined in your config?" << "\n";
161 loizides 1.16 return;
162     }
163     os_ = &(*os);
164 loizides 1.40 brtable_ = new BranchTable;
165     brtable_->SetName(Names::gkBranchTable);
166     brtable_->SetOwner();
167 loizides 1.35 os->add(brtable_, brtable_->GetName());
168 loizides 1.16 }
169    
170 paus 1.1 }
171    
172 loizides 1.9 //--------------------------------------------------------------------------------------------------
173 loizides 1.3 bool FillMitTree::configure(const edm::ParameterSet &cfg)
174 paus 1.1 {
175 bendavid 1.49
176     // Configure TreeWriter
177     const std::string twsConfigName("TreeWriter");
178     ParameterSet twsConfig;
179     if (cfg.existsAs<ParameterSet>(twsConfigName,0))
180     twsConfig = cfg.getUntrackedParameter<ParameterSet>(twsConfigName);
181    
182     configureTreeWriter(twsConfig);
183    
184 loizides 1.36 // Configure our fillers according to given parameter ("fillers").
185 loizides 1.3
186 bendavid 1.33 std::vector<std::string> pars;
187 loizides 1.36 if (cfg.exists("fillers"))
188     pars=cfg.getUntrackedParameter<vector<string> >("fillers");
189 bendavid 1.33 else
190     cfg.getParameterSetNames(pars, false);
191 loizides 1.32
192     // loop over psets
193     for (unsigned int i = 0; i<pars.size(); ++i) {
194    
195     const string name(pars.at(i));
196    
197 loizides 1.37 string ftype("Filler" + name);
198     if (cfg.existsAs<ParameterSet>(name,0)) {
199     ParameterSet next(cfg.getUntrackedParameter<ParameterSet>(name));
200     if (!next.exists("fillerType")) {
201 loizides 1.42 edm::LogError("FillMitTree") << "Cannot determine fillerType for pset named "
202 loizides 1.37 << name << std::endl;
203     throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
204 loizides 1.42 << "Cannot determine fillerType for pset named "
205 loizides 1.37 << name << std::endl;
206     }
207     ftype = next.getUntrackedParameter<string>("fillerType");
208 loizides 1.32 }
209 loizides 1.2
210 loizides 1.37 edm::LogInfo("FillMitTree") << "Attempting to configure '" << ftype
211     << "' for '" << name << "'" << std::endl;
212 loizides 1.35
213 loizides 1.32 if (ftype.compare("FillerMetaInfos")==0) {
214     FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg, name.c_str(), defactive_);
215     addActiveFiller(fillerMetaInfos);
216     continue;
217     }
218 loizides 1.37
219 loizides 1.32 if (ftype.compare("FillerMCParticles")==0) {
220     FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg, name.c_str(), defactive_);
221     addActiveFiller(fillerMCParticles);
222     continue;
223     }
224 loizides 1.37
225     if (ftype.compare("FillerMCEventInfo")==0) {
226     FillerMCEventInfo *fillerMCEventInfo = new FillerMCEventInfo(cfg, name.c_str(), defactive_);
227     addActiveFiller(fillerMCEventInfo);
228     continue;
229     }
230    
231 loizides 1.48 if (ftype.compare("FillerMCVertexes")==0) {
232     FillerMCVertexes *fillerMCVertexes = new FillerMCVertexes(cfg, name.c_str(), defactive_);
233     addActiveFiller(fillerMCVertexes);
234     continue;
235     }
236    
237 loizides 1.52 if (ftype.compare("FillerEvtSelData")==0) {
238     FillerEvtSelData *fillerEvtSelData = new FillerEvtSelData(cfg, name.c_str(), defactive_);
239     addActiveFiller(fillerEvtSelData);
240     continue;
241     }
242    
243 loizides 1.32 if (ftype.compare("FillerBeamSpot")==0) {
244     FillerBeamSpot *fillerBeamSpot = new FillerBeamSpot(cfg, name.c_str(), defactive_);
245     addActiveFiller(fillerBeamSpot);
246     continue;
247     }
248 loizides 1.52
249 loizides 1.32 if (ftype.compare("FillerVertexes")==0) {
250     FillerVertexes *fillerVertexes = new FillerVertexes(cfg, name.c_str(), defactive_);
251     addActiveFiller(fillerVertexes);
252     continue;
253     }
254    
255     if (ftype.compare("FillerCaloTowers")==0) {
256     FillerCaloTowers *fillerCaloTowers = new FillerCaloTowers(cfg, name.c_str(), defactive_);
257     addActiveFiller(fillerCaloTowers);
258     continue;
259     }
260    
261     if (ftype.compare("FillerGenJets")==0) {
262     FillerGenJets *fillerGenJets = new FillerGenJets(cfg, name.c_str(), defactive_);
263     addActiveFiller(fillerGenJets);
264     continue;
265     }
266    
267     if (ftype.compare("FillerCaloJets")==0) {
268     FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg, name.c_str(), defactive_);
269     addActiveFiller(fillerCaloJets);
270     continue;
271     }
272 bendavid 1.34
273     if (ftype.compare("FillerMet")==0) {
274     FillerMet *fillerMet = new FillerMet(cfg, name.c_str(), defactive_);
275     addActiveFiller(fillerMet);
276     continue;
277     }
278 bendavid 1.44
279     if (ftype.compare("FillerGenMet")==0) {
280     FillerGenMet *fillerGenMet = new FillerGenMet(cfg, name.c_str(), defactive_);
281     addActiveFiller(fillerGenMet);
282     continue;
283     }
284 bendavid 1.34
285 loizides 1.32 if (ftype.compare("FillerCaloMet")==0) {
286     FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg, name.c_str(), defactive_);
287     addActiveFiller(fillerCaloMet);
288     continue;
289 bendavid 1.34 }
290    
291     if (ftype.compare("FillerPFMet")==0) {
292     FillerPFMet *fillerPFMet = new FillerPFMet(cfg, name.c_str(), defactive_);
293     addActiveFiller(fillerPFMet);
294     continue;
295 loizides 1.32 }
296    
297     if (ftype.compare("FillerBasicClusters")==0) {
298     FillerBasicClusters *fillerBasicClusters =
299     new FillerBasicClusters(cfg, name.c_str(), defactive_);
300     addActiveFiller(fillerBasicClusters);
301     continue;
302     }
303    
304     if (ftype.compare("FillerSuperClusters")==0) {
305     FillerSuperClusters *fillerSuperClusters =
306     new FillerSuperClusters(cfg, name.c_str(), defactive_);
307     addActiveFiller(fillerSuperClusters);
308     continue;
309     }
310    
311 loizides 1.48 if (ftype.compare("FillerPixelHits")==0) {
312     FillerPixelHits *fillerPixelHits =
313     new FillerPixelHits(cfg, name.c_str(), defactive_);
314     addActiveFiller(fillerPixelHits);
315     continue;
316     }
317    
318 loizides 1.51 if (ftype.compare("FillerStripHits")==0) {
319     FillerStripHits *fillerStripHits =
320     new FillerStripHits(cfg, name.c_str(), defactive_);
321     addActiveFiller(fillerStripHits);
322     continue;
323     }
324    
325 loizides 1.32 if (ftype.compare("FillerTracks")==0) {
326     FillerTracks *fillerTracks = new FillerTracks(cfg, name.c_str(), defactive_);
327     addActiveFiller(fillerTracks);
328     continue;
329     }
330    
331     if (ftype.compare("FillerMuons")==0) {
332     FillerMuons *fillerMuons = new FillerMuons(cfg, name.c_str(), defactive_);
333     addActiveFiller(fillerMuons);
334     continue;
335     }
336    
337     if (ftype.compare("FillerElectrons")==0) {
338     FillerElectrons *fillerElectrons = new FillerElectrons(cfg, name.c_str(), defactive_);
339     addActiveFiller(fillerElectrons);
340     continue;
341     }
342    
343     if (ftype.compare("FillerConversions")==0) {
344     FillerConversions *fillerConversions = new FillerConversions(cfg, name.c_str(), defactive_);
345     addActiveFiller(fillerConversions);
346     continue;
347     }
348    
349     if (ftype.compare("FillerPhotons")==0) {
350     FillerPhotons *fillerPhotons = new FillerPhotons(cfg, name.c_str(), defactive_);
351     addActiveFiller(fillerPhotons);
352     continue;
353     }
354    
355     if (ftype.compare("FillerStableParts")==0) {
356     FillerStableParts *fillerStableParts = new FillerStableParts(cfg, name.c_str(), defactive_);
357     addActiveFiller(fillerStableParts);
358     continue;
359     }
360    
361     if (ftype.compare("FillerDecayParts")==0) {
362     FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg, name.c_str(), defactive_);
363     addActiveFiller(fillerDecayParts);
364     continue;
365     }
366 bendavid 1.33
367     if (ftype.compare("FillerPFCandidates")==0) {
368 loizides 1.35 FillerPFCandidates *fillerPFCands = new FillerPFCandidates(cfg, name.c_str(), defactive_);
369     addActiveFiller(fillerPFCands);
370 bendavid 1.33 continue;
371     }
372 loizides 1.32
373 bendavid 1.33 if (ftype.compare("FillerPFJets")==0) {
374     FillerPFJets *fillerPFJets = new FillerPFJets(cfg, name.c_str(), defactive_);
375     addActiveFiller(fillerPFJets);
376     continue;
377     }
378 bendavid 1.41
379     if (ftype.compare("FillerCaloTaus")==0) {
380     FillerCaloTaus *fillerCaloTaus = new FillerCaloTaus(cfg, name.c_str(), defactive_);
381     addActiveFiller(fillerCaloTaus);
382     continue;
383     }
384    
385     if (ftype.compare("FillerPFTaus")==0) {
386     FillerPFTaus *fillerPFTaus = new FillerPFTaus(cfg, name.c_str(), defactive_);
387     addActiveFiller(fillerPFTaus);
388     continue;
389     }
390 bendavid 1.33
391 loizides 1.32 edm::LogError("FillMitTree")
392     << "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
393     throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
394     << "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
395     }
396 paus 1.13
397 loizides 1.3 return 1;
398 paus 1.1 }
399    
400 loizides 1.9 //--------------------------------------------------------------------------------------------------
401 bendavid 1.49 bool FillMitTree::configureTreeWriter(const edm::ParameterSet &cfg)
402     {
403 loizides 1.50 // Configure tree writer with options from config file.
404 bendavid 1.49
405     tws_->SetPrefix(cfg.getUntrackedParameter<string>("fileName","mit-test"));
406     tws_->SetBaseURL(cfg.getUntrackedParameter<string>("pathName","."));
407     tws_->SetMaxSize((Long64_t)cfg.getUntrackedParameter<unsigned>("maxSize",1024)*1024*1024);
408     tws_->SetCompressLevel(cfg.getUntrackedParameter<unsigned>("compLevel",9));
409     tws_->SetDefaultSL(cfg.getUntrackedParameter<unsigned>("splitLevel",99));
410     tws_->SetDefaultBrSize(cfg.getUntrackedParameter<unsigned>("brSize",16*1024));
411    
412     if (OptInt::IsActivated()) {
413     OptInt::SetZipMode(cfg.getUntrackedParameter<unsigned>("zipMode",99));
414     OptInt::SetGzipFraction(cfg.getUntrackedParameter<double>("gZipThres",1.0));
415     OptInt::SetBzipFraction(cfg.getUntrackedParameter<double>("bZipThres",-1.0));
416     OptInt::SetLzoFraction(cfg.getUntrackedParameter<double>("lzoThres",-1.0));
417     OptInt::SetLzmaFraction(cfg.getUntrackedParameter<double>("lzmaThres",0.95));
418     OptInt::SetVerbose(cfg.getUntrackedParameter<unsigned>("optIOVerbose",0));
419    
420     } else {
421    
422     if (cfg.exists("zipMode") || cfg.exists("bZipThres") ||
423     cfg.exists("gZipThres") || cfg.exists("lzoThres") ||
424     cfg.exists("lzmaThres")) {
425 loizides 1.50 edm::LogError("FillMitTree") <<
426     "OptIO interface not properly pre-loaded, ignoring given settings." << std::endl;
427 bendavid 1.49 }
428     }
429    
430     return 1;
431     }
432    
433     //--------------------------------------------------------------------------------------------------
434 paus 1.1 void FillMitTree::endJob()
435     {
436 loizides 1.2 // Delete fillers.
437    
438     for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
439     delete *iF;
440     }
441    
442 bendavid 1.49 tws_->Clear();
443    
444 paus 1.1 edm::LogInfo("FillMitTree::endJob") << "Ending Job" << endl;
445     }