ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.33
Committed: Wed Mar 11 18:16:22 2009 UTC (16 years, 1 month ago) by bendavid
Content type: text/plain
Branch: MAIN
Changes since 1.32: +18 -2 lines
Log Message: 
Changed to more specific function to get parameter names, added PF classes

File Contents

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