ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
Revision: 1.22
Committed: Tue Sep 16 18:21:33 2008 UTC (16 years, 7 months ago) by sixie
Content type: text/plain
Branch: MAIN
Changes since 1.21: +8 -1 lines
Log Message: 
add IC5GenJets and SC5GenJets collections to be filled

File Contents

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