ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.41
Committed: Fri Mar 20 18:47:46 2009 UTC (16 years, 1 month ago) by bendavid
Content type: text/plain
Branch: MAIN
Changes since 1.40: +15 -1 lines
Log Message: 
Added tau fillers

File Contents

# User Rev Content
1 bendavid 1.41 // $Id: FillMitTree.cc,v 1.40 2009/03/19 22:20:46 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/FillerConversionElectrons.h"
17     #include "MitProd/TreeFiller/interface/FillerConversions.h"
18     #include "MitProd/TreeFiller/interface/FillerDecayParts.h"
19 loizides 1.6 #include "MitProd/TreeFiller/interface/FillerElectrons.h"
20 sixie 1.22 #include "MitProd/TreeFiller/interface/FillerGenJets.h"
21 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
22     #include "MitProd/TreeFiller/interface/FillerMCEventInfo.h"
23     #include "MitProd/TreeFiller/interface/FillerMCParticles.h"
24     #include "MitProd/TreeFiller/interface/FillerMet.h"
25     #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
26     #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     }
149    
150 loizides 1.9 //--------------------------------------------------------------------------------------------------
151 loizides 1.3 bool FillMitTree::configure(const edm::ParameterSet &cfg)
152 paus 1.1 {
153 loizides 1.36 // Configure our fillers according to given parameter ("fillers").
154 loizides 1.3
155 bendavid 1.33 std::vector<std::string> pars;
156 loizides 1.36 if (cfg.exists("fillers"))
157     pars=cfg.getUntrackedParameter<vector<string> >("fillers");
158 bendavid 1.33 else
159     cfg.getParameterSetNames(pars, false);
160 loizides 1.32
161     // loop over psets
162     for (unsigned int i = 0; i<pars.size(); ++i) {
163    
164     const string name(pars.at(i));
165    
166 loizides 1.37 string ftype("Filler" + name);
167     if (cfg.existsAs<ParameterSet>(name,0)) {
168     ParameterSet next(cfg.getUntrackedParameter<ParameterSet>(name));
169     if (!next.exists("fillerType")) {
170     edm::LogError("FillMitTree") << "Can not determine fillerType for pset named "
171     << name << std::endl;
172     throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
173     << "Can not determine fillerType for pset named "
174     << name << std::endl;
175     }
176     ftype = next.getUntrackedParameter<string>("fillerType");
177 loizides 1.32 }
178 loizides 1.2
179 loizides 1.37 edm::LogInfo("FillMitTree") << "Attempting to configure '" << ftype
180     << "' for '" << name << "'" << std::endl;
181 loizides 1.35
182 loizides 1.32 if (ftype.compare("FillerMetaInfos")==0) {
183     FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg, name.c_str(), defactive_);
184     addActiveFiller(fillerMetaInfos);
185     continue;
186     }
187 loizides 1.37
188 loizides 1.32 if (ftype.compare("FillerMCParticles")==0) {
189     FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg, name.c_str(), defactive_);
190     addActiveFiller(fillerMCParticles);
191     continue;
192     }
193 loizides 1.37
194     if (ftype.compare("FillerMCEventInfo")==0) {
195     FillerMCEventInfo *fillerMCEventInfo = new FillerMCEventInfo(cfg, name.c_str(), defactive_);
196     addActiveFiller(fillerMCEventInfo);
197     continue;
198     }
199    
200 loizides 1.32 if (ftype.compare("FillerBeamSpot")==0) {
201     FillerBeamSpot *fillerBeamSpot = new FillerBeamSpot(cfg, name.c_str(), defactive_);
202     addActiveFiller(fillerBeamSpot);
203     continue;
204     }
205 loizides 1.40
206 loizides 1.32 if (ftype.compare("FillerVertexes")==0) {
207     FillerVertexes *fillerVertexes = new FillerVertexes(cfg, name.c_str(), defactive_);
208     addActiveFiller(fillerVertexes);
209     continue;
210     }
211    
212     if (ftype.compare("FillerCaloTowers")==0) {
213     FillerCaloTowers *fillerCaloTowers = new FillerCaloTowers(cfg, name.c_str(), defactive_);
214     addActiveFiller(fillerCaloTowers);
215     continue;
216     }
217    
218     if (ftype.compare("FillerGenJets")==0) {
219     FillerGenJets *fillerGenJets = new FillerGenJets(cfg, name.c_str(), defactive_);
220     addActiveFiller(fillerGenJets);
221     continue;
222     }
223    
224     if (ftype.compare("FillerCaloJets")==0) {
225     FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg, name.c_str(), defactive_);
226     addActiveFiller(fillerCaloJets);
227     continue;
228     }
229 bendavid 1.34
230     if (ftype.compare("FillerMet")==0) {
231     FillerMet *fillerMet = new FillerMet(cfg, name.c_str(), defactive_);
232     addActiveFiller(fillerMet);
233     continue;
234     }
235    
236 loizides 1.32 if (ftype.compare("FillerCaloMet")==0) {
237     FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg, name.c_str(), defactive_);
238     addActiveFiller(fillerCaloMet);
239     continue;
240 bendavid 1.34 }
241    
242     if (ftype.compare("FillerPFMet")==0) {
243     FillerPFMet *fillerPFMet = new FillerPFMet(cfg, name.c_str(), defactive_);
244     addActiveFiller(fillerPFMet);
245     continue;
246 loizides 1.32 }
247    
248     if (ftype.compare("FillerBasicClusters")==0) {
249     FillerBasicClusters *fillerBasicClusters =
250     new FillerBasicClusters(cfg, name.c_str(), defactive_);
251     addActiveFiller(fillerBasicClusters);
252     continue;
253     }
254    
255     if (ftype.compare("FillerSuperClusters")==0) {
256     FillerSuperClusters *fillerSuperClusters =
257     new FillerSuperClusters(cfg, name.c_str(), defactive_);
258     addActiveFiller(fillerSuperClusters);
259     continue;
260     }
261    
262     if (ftype.compare("FillerTracks")==0) {
263     FillerTracks *fillerTracks = new FillerTracks(cfg, name.c_str(), defactive_);
264     addActiveFiller(fillerTracks);
265     continue;
266     }
267    
268     if (ftype.compare("FillerMuons")==0) {
269     FillerMuons *fillerMuons = new FillerMuons(cfg, name.c_str(), defactive_);
270     addActiveFiller(fillerMuons);
271     continue;
272     }
273    
274     if (ftype.compare("FillerElectrons")==0) {
275     FillerElectrons *fillerElectrons = new FillerElectrons(cfg, name.c_str(), defactive_);
276     addActiveFiller(fillerElectrons);
277     continue;
278     }
279    
280     if (ftype.compare("FillerConversionElectrons")==0) {
281     FillerConversionElectrons *fillerConversionElectrons =
282     new FillerConversionElectrons(cfg, name.c_str(), defactive_);
283     addActiveFiller(fillerConversionElectrons);
284     continue;
285     }
286    
287     if (ftype.compare("FillerConversions")==0) {
288     FillerConversions *fillerConversions = new FillerConversions(cfg, name.c_str(), defactive_);
289     addActiveFiller(fillerConversions);
290     continue;
291     }
292    
293     if (ftype.compare("FillerPhotons")==0) {
294     FillerPhotons *fillerPhotons = new FillerPhotons(cfg, name.c_str(), defactive_);
295     addActiveFiller(fillerPhotons);
296     continue;
297     }
298    
299     if (ftype.compare("FillerStableParts")==0) {
300     FillerStableParts *fillerStableParts = new FillerStableParts(cfg, name.c_str(), defactive_);
301     addActiveFiller(fillerStableParts);
302     continue;
303     }
304    
305     if (ftype.compare("FillerDecayParts")==0) {
306     FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg, name.c_str(), defactive_);
307     addActiveFiller(fillerDecayParts);
308     continue;
309     }
310 bendavid 1.33
311     if (ftype.compare("FillerPFCandidates")==0) {
312 loizides 1.35 FillerPFCandidates *fillerPFCands = new FillerPFCandidates(cfg, name.c_str(), defactive_);
313     addActiveFiller(fillerPFCands);
314 bendavid 1.33 continue;
315     }
316 loizides 1.32
317 bendavid 1.33 if (ftype.compare("FillerPFJets")==0) {
318     FillerPFJets *fillerPFJets = new FillerPFJets(cfg, name.c_str(), defactive_);
319     addActiveFiller(fillerPFJets);
320     continue;
321     }
322 bendavid 1.41
323     if (ftype.compare("FillerCaloTaus")==0) {
324     FillerCaloTaus *fillerCaloTaus = new FillerCaloTaus(cfg, name.c_str(), defactive_);
325     addActiveFiller(fillerCaloTaus);
326     continue;
327     }
328    
329     if (ftype.compare("FillerPFTaus")==0) {
330     FillerPFTaus *fillerPFTaus = new FillerPFTaus(cfg, name.c_str(), defactive_);
331     addActiveFiller(fillerPFTaus);
332     continue;
333     }
334 bendavid 1.33
335 loizides 1.32 edm::LogError("FillMitTree")
336     << "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
337     throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
338     << "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
339     }
340 paus 1.13
341 loizides 1.3 return 1;
342 paus 1.1 }
343    
344 loizides 1.9 //--------------------------------------------------------------------------------------------------
345 paus 1.1 void FillMitTree::endJob()
346     {
347 loizides 1.2 // Delete fillers.
348    
349     for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
350     delete *iF;
351     }
352    
353 paus 1.1 edm::LogInfo("FillMitTree::endJob") << "Ending Job" << endl;
354     }