ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.34
Committed: Thu Mar 12 16:00:23 2009 UTC (16 years, 1 month ago) by bendavid
Content type: text/plain
Branch: MAIN
Changes since 1.33: +16 -1 lines
Log Message: 
Updated Met fillers to reflect changes to Met classes, added FillerPFMet

File Contents

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