ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.52
Committed: Tue Dec 8 11:34:28 2009 UTC (15 years, 4 months ago) by loizides
Content type: text/plain
Branch: MAIN
Changes since 1.51: +9 -2 lines
Log Message: 
Added evt sel data filler

File Contents

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