ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.48
Committed: Fri Sep 25 08:43:19 2009 UTC (15 years, 7 months ago) by loizides
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_011a
Changes since 1.47: +17 -2 lines
Log Message: 
New fillers

File Contents

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