ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.28
Committed: Wed Nov 12 18:21:28 2008 UTC (16 years, 5 months ago) by bendavid
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_006b, Mit_006a, Mit_006
Changes since 1.27: +6 -1 lines
Log Message:
Added BeamSpot filler

File Contents

# User Rev Content
1 bendavid 1.28 // $Id: FillMitTree.cc,v 1.27 2008/11/03 18:11:10 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     #include "FWCore/ServiceRegistry/interface/Service.h"
6 loizides 1.3 #include "MitProd/TreeService/interface/TreeService.h"
7 loizides 1.16 #include "MitProd/ObjectService/interface/ObjectService.h"
8 loizides 1.6 #include "MitProd/TreeFiller/interface/AssociationMaps.h"
9 loizides 1.3 #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
10 bendavid 1.24 #include "MitProd/TreeFiller/interface/FillerVertexes.h"
11 bendavid 1.28 #include "MitProd/TreeFiller/interface/FillerBeamSpot.h"
12 loizides 1.6 #include "MitProd/TreeFiller/interface/FillerTracks.h"
13 sixie 1.18 #include "MitProd/TreeFiller/interface/FillerBasicClusters.h"
14     #include "MitProd/TreeFiller/interface/FillerSuperClusters.h"
15 sixie 1.20 #include "MitProd/TreeFiller/interface/FillerCaloTowers.h"
16 loizides 1.6 #include "MitProd/TreeFiller/interface/FillerMuons.h"
17     #include "MitProd/TreeFiller/interface/FillerElectrons.h"
18 sixie 1.22 #include "MitProd/TreeFiller/interface/FillerGenJets.h"
19 loizides 1.11 #include "MitProd/TreeFiller/interface/FillerCaloJets.h"
20     #include "MitProd/TreeFiller/interface/FillerCaloMet.h"
21     #include "MitProd/TreeFiller/interface/FillerConversions.h"
22 bendavid 1.8 #include "MitProd/TreeFiller/interface/FillerConversionElectrons.h"
23     #include "MitProd/TreeFiller/interface/FillerPhotons.h"
24 bendavid 1.12 #include "MitProd/TreeFiller/interface/FillerMCParticles.h"
25 paus 1.13 #include "MitProd/TreeFiller/interface/FillerDecayParts.h"
26     #include "MitProd/TreeFiller/interface/FillerStableParts.h"
27 sixie 1.18 #include "MitProd/TreeFiller/interface/FillerPATMuons.h"
28     #include "MitProd/TreeFiller/interface/FillerPATElectrons.h"
29 loizides 1.21
30 paus 1.1 using namespace std;
31     using namespace edm;
32     using namespace mithep;
33    
34 loizides 1.16 mithep::ObjectService *mithep::FillMitTree::os_ = 0;
35    
36 loizides 1.9 //--------------------------------------------------------------------------------------------------
37 loizides 1.7 FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
38     defactive_(cfg.getUntrackedParameter<bool>("defactive",1))
39 paus 1.1 {
40 loizides 1.6 // Constructor.
41 loizides 1.15
42 loizides 1.3 if (!configure(cfg)) {
43     throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
44     << "Could not configure fillers." << "\n";
45     }
46 paus 1.1 }
47    
48 loizides 1.9 //--------------------------------------------------------------------------------------------------
49 paus 1.1 FillMitTree::~FillMitTree()
50     {
51 loizides 1.2 // Destructor: nothing to be done here.
52 paus 1.1 }
53    
54 loizides 1.9 //--------------------------------------------------------------------------------------------------
55 loizides 1.17 bool FillMitTree::addActiveFiller(BaseFiller *bf)
56     {
57     // Check if filler is active and add it to list of fillers. Otherwise delete it.
58    
59     if (bf->Active()) {
60     fillers_.push_back(bf);
61     return 1;
62     }
63    
64     delete bf;
65     bf = 0;
66     return 0;
67     }
68    
69     //--------------------------------------------------------------------------------------------------
70 loizides 1.3 void FillMitTree::analyze(const edm::Event &event,
71     const edm::EventSetup &setup)
72     {
73     // Access and copy event content.
74    
75     // First step: Loop over the data fillers of the various components
76     for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
77     (*iF)->FillDataBlock(event,setup);
78     }
79    
80     // Second step: Loop over the link resolution of the various components
81     for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
82     (*iF)->ResolveLinks(event,setup);
83     }
84     }
85    
86 loizides 1.9 //--------------------------------------------------------------------------------------------------
87 loizides 1.3 void FillMitTree::beginJob(const edm::EventSetup &event)
88 paus 1.1 {
89 loizides 1.2 // Access the tree and book branches.
90    
91 paus 1.1 Service<TreeService> ts;
92     TreeWriter *tws = ts->get();
93     if (! tws) {
94     throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
95 loizides 1.17 << "Could not get pointer to tree. "
96     << "Do you have the TreeServie define in your config?" << "\n";
97 paus 1.1 return;
98     }
99 loizides 1.16
100     if (os_==0) {
101     Service<ObjectService> os;
102     if (!os.isAvailable()) {
103     throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
104 loizides 1.17 << "Could not get object service. "
105     << "Do you have the ObjectService defined in your config?" << "\n";
106 loizides 1.16 return;
107     }
108     os_ = &(*os);
109     }
110    
111 paus 1.1 // Loop over the various components and book the branches
112     for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
113 loizides 1.16 edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
114 loizides 1.4 (*iF)->BookDataBlock(*tws);
115 paus 1.1 }
116     }
117    
118 loizides 1.9 //--------------------------------------------------------------------------------------------------
119 loizides 1.3 bool FillMitTree::configure(const edm::ParameterSet &cfg)
120 paus 1.1 {
121 loizides 1.3 // Configure our fillers.
122    
123 loizides 1.17 FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg,defactive_);
124     addActiveFiller(fillerMetaInfos);
125 loizides 1.2
126 bendavid 1.12 FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg,"MCParticles",defactive_);
127 loizides 1.17 addActiveFiller(fillerMCParticles);
128 loizides 1.11
129 bendavid 1.28 FillerBeamSpot *fillerBeamSpot =
130     new FillerBeamSpot(cfg,"BeamSpot", defactive_);
131     addActiveFiller(fillerBeamSpot);
132    
133 bendavid 1.24 FillerVertexes *fillerPrimaryVertexes =
134     new FillerVertexes(cfg,"PrimaryVertexes", defactive_);
135     addActiveFiller(fillerPrimaryVertexes);
136    
137     //primary vertexes with beamspot constraint
138     FillerVertexes *fillerPrimaryVertexesBS =
139     new FillerVertexes(cfg,"PrimaryVertexesBS", defactive_);
140     addActiveFiller(fillerPrimaryVertexesBS);
141    
142 sixie 1.20 FillerCaloTowers *fillerCaloTowers =
143     new FillerCaloTowers(cfg, "CaloTowers", defactive_);
144     addActiveFiller(fillerCaloTowers);
145    
146 sixie 1.22 FillerGenJets *fillerIC5GenJets = new FillerGenJets(cfg,"IC5GenJets",defactive_);
147     addActiveFiller(fillerIC5GenJets);
148    
149     FillerGenJets *fillerSC5GenJets = new FillerGenJets(cfg,"SC5GenJets",defactive_);
150     addActiveFiller(fillerSC5GenJets);
151 sixie 1.23
152     FillerGenJets *fillerSC7GenJets = new FillerGenJets(cfg,"SC7GenJets",defactive_);
153     addActiveFiller(fillerSC7GenJets);
154    
155     FillerGenJets *fillerKT4GenJets = new FillerGenJets(cfg,"KT4GenJets",defactive_);
156     addActiveFiller(fillerKT4GenJets);
157    
158     FillerGenJets *fillerKT6GenJets = new FillerGenJets(cfg,"KT6GenJets",defactive_);
159     addActiveFiller(fillerKT6GenJets);
160 sixie 1.22
161 loizides 1.11 FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg,"CaloJets",defactive_);
162 loizides 1.17 addActiveFiller(fillerCaloJets);
163 loizides 1.11
164 sixie 1.23 FillerCaloJets *fillerItrCone5Jets = new FillerCaloJets(cfg,"ItrCone5Jets",defactive_);
165     addActiveFiller(fillerItrCone5Jets);
166    
167 sixie 1.19 FillerCaloJets *fillerSisCone5Jets = new FillerCaloJets(cfg,"SisCone5Jets",defactive_);
168     addActiveFiller(fillerSisCone5Jets);
169    
170 sixie 1.23 FillerCaloJets *fillerSisCone7Jets = new FillerCaloJets(cfg,"SisCone7Jets",defactive_);
171     addActiveFiller(fillerSisCone7Jets);
172    
173     FillerCaloJets *fillerKt4Jets = new FillerCaloJets(cfg,"Kt4Jets",defactive_);
174     addActiveFiller(fillerKt4Jets);
175    
176     FillerCaloJets *fillerKt6Jets = new FillerCaloJets(cfg,"Kt6Jets",defactive_);
177     addActiveFiller(fillerKt6Jets);
178    
179 sixie 1.25 FillerCaloJets *fillerIC5JetPlusTrack = new FillerCaloJets(cfg,"IC5JetPlusTrack",defactive_);
180     addActiveFiller(fillerIC5JetPlusTrack);
181    
182 loizides 1.11 FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg,"CaloMet",defactive_);
183 loizides 1.17 addActiveFiller(fillerCaloMet);
184 loizides 1.11
185 sixie 1.23 FillerCaloMet *fillerItrCone5Met = new FillerCaloMet(cfg,"ItrCone5Met",defactive_);
186     addActiveFiller(fillerItrCone5Met);
187    
188     FillerCaloMet *fillerSisCone5Met = new FillerCaloMet(cfg,"SisCone5Met",defactive_);
189     addActiveFiller(fillerSisCone5Met);
190    
191     FillerCaloMet *fillerSisCone7Met = new FillerCaloMet(cfg,"SisCone7Met",defactive_);
192     addActiveFiller(fillerSisCone7Met);
193    
194     FillerCaloMet *fillerKt4Met = new FillerCaloMet(cfg,"Kt4Met",defactive_);
195     addActiveFiller(fillerKt4Met);
196    
197     FillerCaloMet *fillerKt6Met = new FillerCaloMet(cfg,"Kt6Met",defactive_);
198     addActiveFiller(fillerKt6Met);
199    
200 bendavid 1.26 FillerBasicClusters *fillerBarrelBasicClusters =
201     new FillerBasicClusters(cfg, "BarrelBasicClusters", defactive_);
202     addActiveFiller(fillerBarrelBasicClusters);
203    
204     FillerSuperClusters *fillerBarrelSuperClusters =
205     new FillerSuperClusters(cfg,"BarrelSuperClusters", defactive_);
206     addActiveFiller(fillerBarrelSuperClusters);
207    
208     FillerBasicClusters *fillerEndcapBasicClusters =
209     new FillerBasicClusters(cfg,"EndcapBasicClusters", defactive_);
210     addActiveFiller(fillerEndcapBasicClusters);
211    
212     FillerSuperClusters *fillerEndcapSuperClusters =
213     new FillerSuperClusters(cfg,"EndcapSuperClusters", defactive_);
214     addActiveFiller(fillerEndcapSuperClusters);
215    
216 loizides 1.17 FillerTracks *fillerGeneralTracks = new FillerTracks(cfg,"GeneralTracks",defactive_);
217     addActiveFiller(fillerGeneralTracks);
218 loizides 1.11
219 loizides 1.9 FillerTracks *fillerStandaloneMuonTracks =
220     new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
221 loizides 1.17 addActiveFiller(fillerStandaloneMuonTracks);
222 loizides 1.6
223     FillerTracks *fillerStandaloneMuonTracksVtx =
224 loizides 1.7 new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
225 loizides 1.17 addActiveFiller(fillerStandaloneMuonTracksVtx);
226 loizides 1.11
227 loizides 1.7 FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
228 loizides 1.17 addActiveFiller(fillerGlobalMuonTracks);
229 loizides 1.6
230     FillerTracks *fillerConversionInOutTracks =
231 loizides 1.17 new FillerTracks(cfg,"ConversionInOutTracks",defactive_);
232     addActiveFiller(fillerConversionInOutTracks);
233 loizides 1.11
234 loizides 1.6 FillerTracks *fillerConversionOutInTracks =
235 loizides 1.17 new FillerTracks(cfg,"ConversionOutInTracks",defactive_);
236     addActiveFiller(fillerConversionOutInTracks);
237 loizides 1.11
238 bendavid 1.27 FillerTracks *fillerGsfTracks = new FillerTracks(cfg,"GsfTracks",defactive_);
239 loizides 1.17 addActiveFiller(fillerGsfTracks);
240 loizides 1.11
241 loizides 1.17 FillerMuons *fillerMuons = new FillerMuons(cfg,defactive_);
242     addActiveFiller(fillerMuons);
243 sixie 1.18
244 loizides 1.17 FillerElectrons *fillerElectrons = new FillerElectrons(cfg,defactive_);
245     addActiveFiller(fillerElectrons);
246 loizides 1.11
247 loizides 1.9 FillerConversionElectrons *fillerConversionElectrons =
248 loizides 1.17 new FillerConversionElectrons(cfg,defactive_);
249     addActiveFiller(fillerConversionElectrons);
250 loizides 1.11
251 loizides 1.17 FillerConversions *fillerConversions = new FillerConversions(cfg,defactive_);
252     addActiveFiller(fillerConversions);
253 loizides 1.11
254 loizides 1.17 FillerPhotons *fillerPhotons = new FillerPhotons(cfg,defactive_);
255     addActiveFiller(fillerPhotons);
256 loizides 1.5
257 loizides 1.17 FillerStableParts *fillerStableParts = new FillerStableParts(cfg,"StableParts",defactive_);
258     addActiveFiller(fillerStableParts);
259 bendavid 1.14
260 loizides 1.17 FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg,"DecayParts",defactive_);
261     addActiveFiller(fillerDecayParts);
262 paus 1.13
263 loizides 1.3 return 1;
264 paus 1.1 }
265    
266 loizides 1.9 //--------------------------------------------------------------------------------------------------
267 paus 1.1 void FillMitTree::endJob()
268     {
269 loizides 1.2 // Delete fillers.
270    
271     for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
272     delete *iF;
273     }
274    
275 paus 1.1 edm::LogInfo("FillMitTree::endJob") << "Ending Job" << endl;
276     }