ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.40
Committed: Thu Mar 19 22:20:46 2009 UTC (16 years, 1 month ago) by loizides
Content type: text/plain
Branch: MAIN
Changes since 1.39: +22 -10 lines
Log Message: 
Store BranchTable.

File Contents

# User Rev Content
1 loizides 1.40 // $Id: FillMitTree.cc,v 1.39 2009/03/19 18:18:32 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 sixie 1.20 #include "MitProd/TreeFiller/interface/FillerCaloTowers.h"
15 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerConversionElectrons.h"
16     #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 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
21     #include "MitProd/TreeFiller/interface/FillerMCEventInfo.h"
22     #include "MitProd/TreeFiller/interface/FillerMCParticles.h"
23     #include "MitProd/TreeFiller/interface/FillerMet.h"
24     #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
25     #include "MitProd/TreeFiller/interface/FillerMuons.h"
26     #include "MitProd/TreeFiller/interface/FillerPFCandidates.h"
27 bendavid 1.33 #include "MitProd/TreeFiller/interface/FillerPFJets.h"
28 bendavid 1.34 #include "MitProd/TreeFiller/interface/FillerPFMet.h"
29 bendavid 1.8 #include "MitProd/TreeFiller/interface/FillerPhotons.h"
30 paus 1.13 #include "MitProd/TreeFiller/interface/FillerStableParts.h"
31 loizides 1.37 #include "MitProd/TreeFiller/interface/FillerSuperClusters.h"
32     #include "MitProd/TreeFiller/interface/FillerTracks.h"
33     #include "MitProd/TreeFiller/interface/FillerVertexes.h"
34 loizides 1.35 #include "MitAna/DataTree/interface/Names.h"
35     #include "MitAna/DataTree/interface/BranchTable.h"
36 loizides 1.21
37 paus 1.1 using namespace std;
38     using namespace edm;
39     using namespace mithep;
40    
41 loizides 1.16 mithep::ObjectService *mithep::FillMitTree::os_ = 0;
42    
43 loizides 1.9 //--------------------------------------------------------------------------------------------------
44 loizides 1.7 FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
45 loizides 1.35 defactive_(cfg.getUntrackedParameter<bool>("defactive",1)),
46 loizides 1.40 brtable_(0),
47     acfnumber_(-1)
48 paus 1.1 {
49 loizides 1.6 // Constructor.
50 loizides 1.15
51 loizides 1.3 if (!configure(cfg)) {
52     throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
53     << "Could not configure fillers." << "\n";
54     }
55 paus 1.1 }
56    
57 loizides 1.9 //--------------------------------------------------------------------------------------------------
58 paus 1.1 FillMitTree::~FillMitTree()
59     {
60 loizides 1.35 // Destructor.
61    
62     delete brtable_;
63 paus 1.1 }
64    
65 loizides 1.9 //--------------------------------------------------------------------------------------------------
66 loizides 1.17 bool FillMitTree::addActiveFiller(BaseFiller *bf)
67     {
68     // Check if filler is active and add it to list of fillers. Otherwise delete it.
69    
70 loizides 1.35 if (!bf)
71     return 0;
72    
73 loizides 1.17 if (bf->Active()) {
74     fillers_.push_back(bf);
75     return 1;
76     }
77    
78     delete bf;
79     bf = 0;
80     return 0;
81     }
82    
83     //--------------------------------------------------------------------------------------------------
84 loizides 1.3 void FillMitTree::analyze(const edm::Event &event,
85     const edm::EventSetup &setup)
86     {
87     // Access and copy event content.
88    
89 loizides 1.32 // first step: Loop over the data fillers of the various components
90 loizides 1.3 for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
91     (*iF)->FillDataBlock(event,setup);
92     }
93    
94 loizides 1.32 // second step: Loop over the link resolution of the various components
95 loizides 1.3 for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
96     (*iF)->ResolveLinks(event,setup);
97     }
98 loizides 1.38
99 loizides 1.40 if (brtable_) { // only the first FillMitTree object has to deal with the branch table
100     if (acfnumber_==-1) {
101     brtable_->Rehash(brtable_->GetSize());
102     if (0)
103     brtable_->Print();
104     }
105     if (acfnumber_ != tws_->GetFileNumber()) {
106     tws_->StoreObject(brtable_);
107     acfnumber_ = tws_->GetFileNumber();
108     }
109     }
110 loizides 1.3 }
111    
112 loizides 1.9 //--------------------------------------------------------------------------------------------------
113 loizides 1.3 void FillMitTree::beginJob(const edm::EventSetup &event)
114 paus 1.1 {
115 loizides 1.2 // Access the tree and book branches.
116    
117 paus 1.1 Service<TreeService> ts;
118 loizides 1.40 tws_ = ts->get();
119     if (!tws_) {
120 paus 1.1 throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
121 loizides 1.17 << "Could not get pointer to tree. "
122     << "Do you have the TreeServie define in your config?" << "\n";
123 paus 1.1 return;
124     }
125 loizides 1.16
126 loizides 1.40 if (os_==0) { // only the first FillMitTree object has to deal with this
127 loizides 1.16 Service<ObjectService> os;
128     if (!os.isAvailable()) {
129     throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
130 loizides 1.17 << "Could not get object service. "
131     << "Do you have the ObjectService defined in your config?" << "\n";
132 loizides 1.16 return;
133     }
134     os_ = &(*os);
135 loizides 1.40 brtable_ = new BranchTable;
136     brtable_->SetName(Names::gkBranchTable);
137     brtable_->SetOwner();
138 loizides 1.35 os->add(brtable_, brtable_->GetName());
139 loizides 1.16 }
140    
141 loizides 1.32 // loop over the various components and book the branches
142 paus 1.1 for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
143 loizides 1.16 edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
144 loizides 1.40 (*iF)->BookDataBlock(*tws_);
145 paus 1.1 }
146     }
147    
148 loizides 1.9 //--------------------------------------------------------------------------------------------------
149 loizides 1.3 bool FillMitTree::configure(const edm::ParameterSet &cfg)
150 paus 1.1 {
151 loizides 1.36 // Configure our fillers according to given parameter ("fillers").
152 loizides 1.3
153 bendavid 1.33 std::vector<std::string> pars;
154 loizides 1.36 if (cfg.exists("fillers"))
155     pars=cfg.getUntrackedParameter<vector<string> >("fillers");
156 bendavid 1.33 else
157     cfg.getParameterSetNames(pars, false);
158 loizides 1.32
159     // loop over psets
160     for (unsigned int i = 0; i<pars.size(); ++i) {
161    
162     const string name(pars.at(i));
163    
164 loizides 1.37 string ftype("Filler" + name);
165     if (cfg.existsAs<ParameterSet>(name,0)) {
166     ParameterSet next(cfg.getUntrackedParameter<ParameterSet>(name));
167     if (!next.exists("fillerType")) {
168     edm::LogError("FillMitTree") << "Can not determine fillerType for pset named "
169     << name << std::endl;
170     throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
171     << "Can not determine fillerType for pset named "
172     << name << std::endl;
173     }
174     ftype = next.getUntrackedParameter<string>("fillerType");
175 loizides 1.32 }
176 loizides 1.2
177 loizides 1.37 edm::LogInfo("FillMitTree") << "Attempting to configure '" << ftype
178     << "' for '" << name << "'" << std::endl;
179 loizides 1.35
180 loizides 1.32 if (ftype.compare("FillerMetaInfos")==0) {
181     FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg, name.c_str(), defactive_);
182     addActiveFiller(fillerMetaInfos);
183     continue;
184     }
185 loizides 1.37
186 loizides 1.32 if (ftype.compare("FillerMCParticles")==0) {
187     FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg, name.c_str(), defactive_);
188     addActiveFiller(fillerMCParticles);
189     continue;
190     }
191 loizides 1.37
192     if (ftype.compare("FillerMCEventInfo")==0) {
193     FillerMCEventInfo *fillerMCEventInfo = new FillerMCEventInfo(cfg, name.c_str(), defactive_);
194     addActiveFiller(fillerMCEventInfo);
195     continue;
196     }
197    
198 loizides 1.32 if (ftype.compare("FillerBeamSpot")==0) {
199     FillerBeamSpot *fillerBeamSpot = new FillerBeamSpot(cfg, name.c_str(), defactive_);
200     addActiveFiller(fillerBeamSpot);
201     continue;
202     }
203 loizides 1.40
204 loizides 1.32 if (ftype.compare("FillerVertexes")==0) {
205     FillerVertexes *fillerVertexes = new FillerVertexes(cfg, name.c_str(), defactive_);
206     addActiveFiller(fillerVertexes);
207     continue;
208     }
209    
210     if (ftype.compare("FillerCaloTowers")==0) {
211     FillerCaloTowers *fillerCaloTowers = new FillerCaloTowers(cfg, name.c_str(), defactive_);
212     addActiveFiller(fillerCaloTowers);
213     continue;
214     }
215    
216     if (ftype.compare("FillerGenJets")==0) {
217     FillerGenJets *fillerGenJets = new FillerGenJets(cfg, name.c_str(), defactive_);
218     addActiveFiller(fillerGenJets);
219     continue;
220     }
221    
222     if (ftype.compare("FillerCaloJets")==0) {
223     FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg, name.c_str(), defactive_);
224     addActiveFiller(fillerCaloJets);
225     continue;
226     }
227 bendavid 1.34
228     if (ftype.compare("FillerMet")==0) {
229     FillerMet *fillerMet = new FillerMet(cfg, name.c_str(), defactive_);
230     addActiveFiller(fillerMet);
231     continue;
232     }
233    
234 loizides 1.32 if (ftype.compare("FillerCaloMet")==0) {
235     FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg, name.c_str(), defactive_);
236     addActiveFiller(fillerCaloMet);
237     continue;
238 bendavid 1.34 }
239    
240     if (ftype.compare("FillerPFMet")==0) {
241     FillerPFMet *fillerPFMet = new FillerPFMet(cfg, name.c_str(), defactive_);
242     addActiveFiller(fillerPFMet);
243     continue;
244 loizides 1.32 }
245    
246     if (ftype.compare("FillerBasicClusters")==0) {
247     FillerBasicClusters *fillerBasicClusters =
248     new FillerBasicClusters(cfg, name.c_str(), defactive_);
249     addActiveFiller(fillerBasicClusters);
250     continue;
251     }
252    
253     if (ftype.compare("FillerSuperClusters")==0) {
254     FillerSuperClusters *fillerSuperClusters =
255     new FillerSuperClusters(cfg, name.c_str(), defactive_);
256     addActiveFiller(fillerSuperClusters);
257     continue;
258     }
259    
260     if (ftype.compare("FillerTracks")==0) {
261     FillerTracks *fillerTracks = new FillerTracks(cfg, name.c_str(), defactive_);
262     addActiveFiller(fillerTracks);
263     continue;
264     }
265    
266     if (ftype.compare("FillerMuons")==0) {
267     FillerMuons *fillerMuons = new FillerMuons(cfg, name.c_str(), defactive_);
268     addActiveFiller(fillerMuons);
269     continue;
270     }
271    
272     if (ftype.compare("FillerElectrons")==0) {
273     FillerElectrons *fillerElectrons = new FillerElectrons(cfg, name.c_str(), defactive_);
274     addActiveFiller(fillerElectrons);
275     continue;
276     }
277    
278     if (ftype.compare("FillerConversionElectrons")==0) {
279     FillerConversionElectrons *fillerConversionElectrons =
280     new FillerConversionElectrons(cfg, name.c_str(), defactive_);
281     addActiveFiller(fillerConversionElectrons);
282     continue;
283     }
284    
285     if (ftype.compare("FillerConversions")==0) {
286     FillerConversions *fillerConversions = new FillerConversions(cfg, name.c_str(), defactive_);
287     addActiveFiller(fillerConversions);
288     continue;
289     }
290    
291     if (ftype.compare("FillerPhotons")==0) {
292     FillerPhotons *fillerPhotons = new FillerPhotons(cfg, name.c_str(), defactive_);
293     addActiveFiller(fillerPhotons);
294     continue;
295     }
296    
297     if (ftype.compare("FillerStableParts")==0) {
298     FillerStableParts *fillerStableParts = new FillerStableParts(cfg, name.c_str(), defactive_);
299     addActiveFiller(fillerStableParts);
300     continue;
301     }
302    
303     if (ftype.compare("FillerDecayParts")==0) {
304     FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg, name.c_str(), defactive_);
305     addActiveFiller(fillerDecayParts);
306     continue;
307     }
308 bendavid 1.33
309     if (ftype.compare("FillerPFCandidates")==0) {
310 loizides 1.35 FillerPFCandidates *fillerPFCands = new FillerPFCandidates(cfg, name.c_str(), defactive_);
311     addActiveFiller(fillerPFCands);
312 bendavid 1.33 continue;
313     }
314 loizides 1.32
315 bendavid 1.33 if (ftype.compare("FillerPFJets")==0) {
316     FillerPFJets *fillerPFJets = new FillerPFJets(cfg, name.c_str(), defactive_);
317     addActiveFiller(fillerPFJets);
318     continue;
319     }
320    
321 loizides 1.32 edm::LogError("FillMitTree")
322     << "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
323     throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
324     << "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
325     }
326 paus 1.13
327 loizides 1.3 return 1;
328 paus 1.1 }
329    
330 loizides 1.9 //--------------------------------------------------------------------------------------------------
331 paus 1.1 void FillMitTree::endJob()
332     {
333 loizides 1.2 // Delete fillers.
334    
335     for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
336     delete *iF;
337     }
338    
339 paus 1.1 edm::LogInfo("FillMitTree::endJob") << "Ending Job" << endl;
340     }