ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.50
Committed: Wed Nov 4 17:03:32 2009 UTC (15 years, 5 months ago) by loizides
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_012b, Mit_012a, Mit_012
Changes since 1.49: +4 -11 lines
Log Message: 
Docu plus cosmetics.

File Contents

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