ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillMitTree.cc
(Generate patch)

Comparing UserCode/MitProd/TreeFiller/src/FillMitTree.cc (file contents):
Revision 1.1 by paus, Wed Jun 18 13:23:22 2008 UTC vs.
Revision 1.9 by loizides, Thu Jul 3 07:56:14 2008 UTC

# Line 1 | Line 1
1   // $Id$
2  
3 + #include "MitProd/TreeFiller/interface/FillMitTree.h"
4   #include "FWCore/MessageLogger/interface/MessageLogger.h"
4 – #include "FWCore/Framework/interface/ESHandle.h"
5 – #include "DataFormats/Common/interface/Handle.h"
5   #include "FWCore/ServiceRegistry/interface/Service.h"
6 <
7 < #include "MitProd/TreeFiller/interface/FillerGlobalMuons.h"
8 < #include "MitProd/TreeFiller/interface/FillMitTree.h"
6 > #include "MitProd/TreeService/interface/TreeService.h"
7 > #include "MitProd/TreeFiller/interface/AssociationMaps.h"
8 > #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
9 > #include "MitProd/TreeFiller/interface/FillerSimParticles.h"
10 > #include "MitProd/TreeFiller/interface/FillerTracks.h"
11 > #include "MitProd/TreeFiller/interface/FillerGsfTracks.h"
12 > #include "MitProd/TreeFiller/interface/FillerMuons.h"
13 > #include "MitProd/TreeFiller/interface/FillerElectrons.h"
14 > #include "MitProd/TreeFiller/interface/FillerConversionElectrons.h"
15 > #include "MitProd/TreeFiller/interface/FillerConversions.h"
16 > #include "MitProd/TreeFiller/interface/FillerPhotons.h"
17 > #include "MitProd/TreeFiller/interface/FillerGenParts.h"
18  
19   using namespace std;
20   using namespace edm;
21   using namespace mithep;
22  
23 < //-------------------------------------------------------------------------------------------------
23 > //--------------------------------------------------------------------------------------------------
24   FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
25 <  fillerGlobalMuons_(0)
25 >  defactive_(cfg.getUntrackedParameter<bool>("defactive",1))
26   {
27 <  // initialize fillers
28 <  fillerGlobalMuons_ = new FillerGlobalMuons(cfg);
29 <  if (fillerGlobalMuons_->Active())
30 <    fillers_.push_back(fillerGlobalMuons_);
27 >  // Constructor.
28 >    
29 >  if (!configure(cfg)) {
30 >    throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
31 >      << "Could not configure fillers." << "\n";
32 >  }
33   }
34  
35 < //-------------------------------------------------------------------------------------------------
35 > //--------------------------------------------------------------------------------------------------
36   FillMitTree::~FillMitTree()
37   {
38 <  // We own the fillers so we have to delete them
39 <  delete fillerGlobalMuons_;
38 >  // Destructor: nothing to be done here.
39 > }
40 >
41 > //--------------------------------------------------------------------------------------------------
42 > void FillMitTree::analyze(const edm::Event      &event,
43 >                          const edm::EventSetup &setup)
44 > {
45 >  // Access and copy event content.
46 >
47 >  // First step: Loop over the data fillers of the various components
48 >  for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
49 >    (*iF)->FillDataBlock(event,setup);
50 >  }
51 >
52 >  // Second step: Loop over the link resolution of the various components
53 >  for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
54 >    (*iF)->ResolveLinks(event,setup);
55 >  }
56   }
57  
58 < //-------------------------------------------------------------------------------------------------
59 < void FillMitTree::beginJob(edm::EventSetup const &event)
58 > //--------------------------------------------------------------------------------------------------
59 > void FillMitTree::beginJob(const edm::EventSetup &event)
60   {
61 +  // Access the tree and book branches.
62 +
63    Service<TreeService> ts;
64    TreeWriter *tws = ts->get();
65    if (! tws) {
66      throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
67 <      << "Could not get pointer to Tree with name " << tws->GetName() << "\n";
67 >      << "Could not get pointer to tree." << "\n";
68      return;
69    }
70 <
70 >  
71    // Loop over the various components and book the branches
72    for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
73 <    (*iF)->BookDataBlock(tws);
73 >    (*iF)->BookDataBlock(*tws);
74    }
47 –  return;
75   }
76  
77 < //-------------------------------------------------------------------------------------------------
78 < void FillMitTree::analyze(const edm::Event      &event,
52 <                          const edm::EventSetup &setup)
77 > //--------------------------------------------------------------------------------------------------
78 > bool FillMitTree::configure(const edm::ParameterSet &cfg)
79   {
80 <  // First step: Loop over the data fillers of the various components
81 <  for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
82 <    (*iF)->FillDataBlock(event,setup);
80 >  // Configure our fillers.
81 >
82 >  FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg);
83 >  if (fillerMetaInfos->Active())
84 >    fillers_.push_back(fillerMetaInfos);
85 >  else
86 >    delete fillerMetaInfos;
87 >
88 >  FillerGenParts *fillerGenParts = new FillerGenParts(cfg,defactive_);
89 >  if (fillerGenParts->Active())
90 >    fillers_.push_back(fillerGenParts);
91 >  else
92 >    delete fillerGenParts;  
93 >    
94 >  FillerSimParticles *fillerSimParticles = new FillerSimParticles(cfg,defactive_);
95 >  const SimParticleMap* simParticleMap=0;
96 >  if (fillerSimParticles->Active()) {
97 >    fillers_.push_back(fillerSimParticles);
98 >    simParticleMap = fillerSimParticles->GetSimParticleMap();
99 >  }
100 >  else
101 >    delete fillerSimParticles;  
102 >  
103 >  FillerTracks *fillerGeneralTracks =
104 >    new FillerTracks(cfg,"GeneralTracks",defactive_,simParticleMap);
105 >  const TrackMap* generalTrackMap=0;
106 >  if (fillerGeneralTracks->Active()) {
107 >    fillers_.push_back(fillerGeneralTracks);
108 >    generalTrackMap = fillerGeneralTracks->GetTrackMap();
109    }
110 +  else
111 +    delete fillerGeneralTracks;
112 +    
113 +  FillerTracks *fillerStandaloneMuonTracks =
114 +    new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
115 +  const TrackMap* standaloneMuonTrackMap=0;
116 +  if (fillerStandaloneMuonTracks->Active()) {
117 +    fillers_.push_back(fillerStandaloneMuonTracks);
118 +    standaloneMuonTrackMap = fillerStandaloneMuonTracks->GetTrackMap();
119 +  }
120 +  else
121 +    delete fillerStandaloneMuonTracks;
122  
123 <  // Second step: Loop over the link resolution of the various components
124 <  for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
125 <    (*iF)->ResolveLinks(event,setup);
123 >  FillerTracks *fillerStandaloneMuonTracksVtx =
124 >    new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
125 >  const TrackMap* standaloneMuonTrackVtxMap=0;
126 >  if (fillerStandaloneMuonTracksVtx->Active()) {
127 >    fillers_.push_back(fillerStandaloneMuonTracksVtx);
128 >    standaloneMuonTrackVtxMap = fillerStandaloneMuonTracksVtx->GetTrackMap();
129 >  }
130 >  else
131 >    delete fillerStandaloneMuonTracksVtx;
132 >    
133 >  FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
134 >  const TrackMap* globalMuonTrackMap=0;
135 >  if (fillerGlobalMuonTracks->Active()) {
136 >    fillers_.push_back(fillerGlobalMuonTracks);
137 >    globalMuonTrackMap = fillerGlobalMuonTracks->GetTrackMap();
138    }
139 +  else
140 +    delete fillerGlobalMuonTracks;  
141 +    
142 +  FillerTracks *fillerConversionInOutTracks =
143 +    new FillerTracks(cfg,"ConversionInOutTracks",defactive_,simParticleMap);
144 +  const TrackMap* conversionInOutTrackMap=0;
145 +  const TrackCol* conversionInOutTracks=0;
146 +  if (fillerConversionInOutTracks->Active()) {
147 +    fillers_.push_back(fillerConversionInOutTracks);
148 +    conversionInOutTrackMap = fillerConversionInOutTracks->GetTrackMap();
149 +    conversionInOutTracks = fillerConversionInOutTracks->GetTrackCol();
150 +  }
151 +  else
152 +    delete fillerConversionInOutTracks;
153 +    
154 +  FillerTracks *fillerConversionOutInTracks =
155 +    new FillerTracks(cfg,"ConversionOutInTracks",defactive_,simParticleMap);
156 +  const TrackMap* conversionOutInTrackMap=0;
157 +  const TrackCol* conversionOutInTracks=0;
158 +  if (fillerConversionOutInTracks->Active()) {
159 +    fillers_.push_back(fillerConversionOutInTracks);
160 +    conversionOutInTrackMap = fillerConversionOutInTracks->GetTrackMap();
161 +    conversionOutInTracks = fillerConversionOutInTracks->GetTrackCol();
162 +  }
163 +  else
164 +    delete fillerConversionOutInTracks;
165 +    
166 +  FillerGsfTracks *fillerGsfTracks =
167 +    new FillerGsfTracks(cfg,"GsfTracks",defactive_,simParticleMap);
168 +  const GsfTrackMap* gsfTrackMap=0;
169 +  if (fillerGsfTracks->Active()) {
170 +    fillers_.push_back(fillerGsfTracks);
171 +    gsfTrackMap = (GsfTrackMap*)fillerGsfTracks->GetTrackMap();
172 +  }
173 +  else
174 +    delete fillerGsfTracks;
175 +    
176 +  FillerMuons *fillerMuons =
177 +    new FillerMuons(cfg,defactive_,globalMuonTrackMap,standaloneMuonTrackMap,
178 +                    standaloneMuonTrackVtxMap,generalTrackMap);
179 +  if (fillerMuons->Active())
180 +    fillers_.push_back(fillerMuons);
181 +  else
182 +    delete fillerMuons;
183 +    
184 +  FillerElectrons *fillerElectrons =
185 +    new FillerElectrons(cfg,defactive_,gsfTrackMap,generalTrackMap);
186 +  if (fillerElectrons->Active())
187 +    fillers_.push_back(fillerElectrons);
188 +  else
189 +    delete fillerElectrons;
190 +  
191 +  FillerConversionElectrons *fillerConversionElectrons =
192 +    new FillerConversionElectrons(cfg, defactive_, conversionInOutTracks, conversionOutInTracks,
193 +                                  conversionInOutTrackMap, conversionOutInTrackMap);
194 +  const ConversionElectronMap* convElectronMap=0;
195 +  if (fillerConversionElectrons->Active()) {
196 +    fillers_.push_back(fillerConversionElectrons);
197 +    convElectronMap = fillerConversionElectrons->GetConversionElectronMap();
198 +  }
199 +  else
200 +    delete fillerConversionElectrons;
201 +    
202 +  FillerConversions *fillerConversions = new FillerConversions(cfg, defactive_, convElectronMap);
203 +  const ConversionMap* conversionMap=0;
204 +  if (fillerConversions->Active()) {
205 +    fillers_.push_back(fillerConversions);
206 +    conversionMap = fillerConversions->GetConversionMap();
207 +  }
208 +  else
209 +    delete fillerConversions;
210 +    
211 +  FillerPhotons *fillerPhotons = new FillerPhotons(cfg, defactive_, conversionMap);
212 +  if (fillerPhotons->Active())
213 +    fillers_.push_back(fillerPhotons);
214 +  else
215 +    delete fillerPhotons;
216 +
217 +
218 +  return 1;
219   }
220  
221 < //-------------------------------------------------------------------------------------------------
221 > //--------------------------------------------------------------------------------------------------
222   void FillMitTree::endJob()
223   {
224 +  // Delete fillers.
225 +
226 +  for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
227 +    delete *iF;
228 +  }
229 +
230    edm::LogInfo("FillMitTree::endJob") << "Ending Job" << endl;
231   }

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines