ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.24
Committed: Tue Sep 30 13:03:42 2008 UTC (16 years, 7 months ago) by bendavid
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_004
Changes since 1.23: +11 -1 lines
Log Message:
added vertex filler

File Contents

# User Rev Content
1 bendavid 1.24 // $Id: FillMitTree.cc,v 1.23 2008/09/29 17:07:56 sixie 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 loizides 1.6 #include "MitProd/TreeFiller/interface/FillerTracks.h"
12     #include "MitProd/TreeFiller/interface/FillerGsfTracks.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.24 FillerVertexes *fillerPrimaryVertexes =
130     new FillerVertexes(cfg,"PrimaryVertexes", defactive_);
131     addActiveFiller(fillerPrimaryVertexes);
132    
133     //primary vertexes with beamspot constraint
134     FillerVertexes *fillerPrimaryVertexesBS =
135     new FillerVertexes(cfg,"PrimaryVertexesBS", defactive_);
136     addActiveFiller(fillerPrimaryVertexesBS);
137    
138 sixie 1.20 FillerCaloTowers *fillerCaloTowers =
139     new FillerCaloTowers(cfg, "CaloTowers", defactive_);
140     addActiveFiller(fillerCaloTowers);
141    
142 sixie 1.22 FillerGenJets *fillerIC5GenJets = new FillerGenJets(cfg,"IC5GenJets",defactive_);
143     addActiveFiller(fillerIC5GenJets);
144    
145     FillerGenJets *fillerSC5GenJets = new FillerGenJets(cfg,"SC5GenJets",defactive_);
146     addActiveFiller(fillerSC5GenJets);
147 sixie 1.23
148     FillerGenJets *fillerSC7GenJets = new FillerGenJets(cfg,"SC7GenJets",defactive_);
149     addActiveFiller(fillerSC7GenJets);
150    
151     FillerGenJets *fillerKT4GenJets = new FillerGenJets(cfg,"KT4GenJets",defactive_);
152     addActiveFiller(fillerKT4GenJets);
153    
154     FillerGenJets *fillerKT6GenJets = new FillerGenJets(cfg,"KT6GenJets",defactive_);
155     addActiveFiller(fillerKT6GenJets);
156 sixie 1.22
157 loizides 1.11 FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg,"CaloJets",defactive_);
158 loizides 1.17 addActiveFiller(fillerCaloJets);
159 loizides 1.11
160 sixie 1.23 FillerCaloJets *fillerItrCone5Jets = new FillerCaloJets(cfg,"ItrCone5Jets",defactive_);
161     addActiveFiller(fillerItrCone5Jets);
162    
163 sixie 1.19 FillerCaloJets *fillerSisCone5Jets = new FillerCaloJets(cfg,"SisCone5Jets",defactive_);
164     addActiveFiller(fillerSisCone5Jets);
165    
166 sixie 1.23 FillerCaloJets *fillerSisCone7Jets = new FillerCaloJets(cfg,"SisCone7Jets",defactive_);
167     addActiveFiller(fillerSisCone7Jets);
168    
169     FillerCaloJets *fillerKt4Jets = new FillerCaloJets(cfg,"Kt4Jets",defactive_);
170     addActiveFiller(fillerKt4Jets);
171    
172     FillerCaloJets *fillerKt6Jets = new FillerCaloJets(cfg,"Kt6Jets",defactive_);
173     addActiveFiller(fillerKt6Jets);
174    
175 loizides 1.11 FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg,"CaloMet",defactive_);
176 loizides 1.17 addActiveFiller(fillerCaloMet);
177 loizides 1.11
178 sixie 1.23 FillerCaloMet *fillerItrCone5Met = new FillerCaloMet(cfg,"ItrCone5Met",defactive_);
179     addActiveFiller(fillerItrCone5Met);
180    
181     FillerCaloMet *fillerSisCone5Met = new FillerCaloMet(cfg,"SisCone5Met",defactive_);
182     addActiveFiller(fillerSisCone5Met);
183    
184     FillerCaloMet *fillerSisCone7Met = new FillerCaloMet(cfg,"SisCone7Met",defactive_);
185     addActiveFiller(fillerSisCone7Met);
186    
187     FillerCaloMet *fillerKt4Met = new FillerCaloMet(cfg,"Kt4Met",defactive_);
188     addActiveFiller(fillerKt4Met);
189    
190     FillerCaloMet *fillerKt6Met = new FillerCaloMet(cfg,"Kt6Met",defactive_);
191     addActiveFiller(fillerKt6Met);
192    
193 loizides 1.17 FillerTracks *fillerGeneralTracks = new FillerTracks(cfg,"GeneralTracks",defactive_);
194     addActiveFiller(fillerGeneralTracks);
195 loizides 1.11
196 loizides 1.9 FillerTracks *fillerStandaloneMuonTracks =
197     new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
198 loizides 1.17 addActiveFiller(fillerStandaloneMuonTracks);
199 loizides 1.6
200     FillerTracks *fillerStandaloneMuonTracksVtx =
201 loizides 1.7 new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
202 loizides 1.17 addActiveFiller(fillerStandaloneMuonTracksVtx);
203 loizides 1.11
204 loizides 1.7 FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
205 loizides 1.17 addActiveFiller(fillerGlobalMuonTracks);
206 loizides 1.6
207     FillerTracks *fillerConversionInOutTracks =
208 loizides 1.17 new FillerTracks(cfg,"ConversionInOutTracks",defactive_);
209     addActiveFiller(fillerConversionInOutTracks);
210 loizides 1.11
211 loizides 1.6 FillerTracks *fillerConversionOutInTracks =
212 loizides 1.17 new FillerTracks(cfg,"ConversionOutInTracks",defactive_);
213     addActiveFiller(fillerConversionOutInTracks);
214 loizides 1.11
215 loizides 1.17 FillerGsfTracks *fillerGsfTracks = new FillerGsfTracks(cfg,"GsfTracks",defactive_);
216     addActiveFiller(fillerGsfTracks);
217 loizides 1.11
218 sixie 1.18 FillerBasicClusters *fillerBarrelBasicClusters =
219     new FillerBasicClusters(cfg, "BarrelBasicClusters", defactive_);
220     addActiveFiller(fillerBarrelBasicClusters);
221    
222     FillerSuperClusters *fillerBarrelSuperClusters =
223     new FillerSuperClusters(cfg,"BarrelSuperClusters", defactive_);
224     addActiveFiller(fillerBarrelSuperClusters);
225    
226     FillerBasicClusters *fillerEndcapBasicClusters =
227     new FillerBasicClusters(cfg,"EndcapBasicClusters", defactive_);
228     addActiveFiller(fillerEndcapBasicClusters);
229    
230     FillerSuperClusters *fillerEndcapSuperClusters =
231     new FillerSuperClusters(cfg,"EndcapSuperClusters", defactive_);
232     addActiveFiller(fillerEndcapSuperClusters);
233    
234 loizides 1.17 FillerMuons *fillerMuons = new FillerMuons(cfg,defactive_);
235     addActiveFiller(fillerMuons);
236 sixie 1.18
237 loizides 1.17 FillerElectrons *fillerElectrons = new FillerElectrons(cfg,defactive_);
238     addActiveFiller(fillerElectrons);
239 loizides 1.11
240 loizides 1.9 FillerConversionElectrons *fillerConversionElectrons =
241 loizides 1.17 new FillerConversionElectrons(cfg,defactive_);
242     addActiveFiller(fillerConversionElectrons);
243 loizides 1.11
244 loizides 1.17 FillerConversions *fillerConversions = new FillerConversions(cfg,defactive_);
245     addActiveFiller(fillerConversions);
246 loizides 1.11
247 loizides 1.17 FillerPhotons *fillerPhotons = new FillerPhotons(cfg,defactive_);
248     addActiveFiller(fillerPhotons);
249 loizides 1.5
250 loizides 1.17 FillerStableParts *fillerStableParts = new FillerStableParts(cfg,"StableParts",defactive_);
251     addActiveFiller(fillerStableParts);
252 bendavid 1.14
253 loizides 1.17 FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg,"DecayParts",defactive_);
254     addActiveFiller(fillerDecayParts);
255 paus 1.13
256 loizides 1.3 return 1;
257 paus 1.1 }
258    
259 loizides 1.9 //--------------------------------------------------------------------------------------------------
260 paus 1.1 void FillMitTree::endJob()
261     {
262 loizides 1.2 // Delete fillers.
263    
264     for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
265     delete *iF;
266     }
267    
268 paus 1.1 edm::LogInfo("FillMitTree::endJob") << "Ending Job" << endl;
269     }