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.4 by loizides, Fri Jun 20 17:52:57 2008 UTC vs.
Revision 1.19 by sixie, Sat Aug 30 19:09:09 2008 UTC

# Line 4 | Line 4
4   #include "FWCore/MessageLogger/interface/MessageLogger.h"
5   #include "FWCore/ServiceRegistry/interface/Service.h"
6   #include "MitProd/TreeService/interface/TreeService.h"
7 + #include "MitProd/ObjectService/interface/ObjectService.h"
8 + #include "MitProd/TreeFiller/interface/AssociationMaps.h"
9   #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
10 < #include "MitProd/TreeFiller/interface/FillerGlobalMuons.h"
10 > #include "MitProd/TreeFiller/interface/FillerTracks.h"
11 > #include "MitProd/TreeFiller/interface/FillerGsfTracks.h"
12 > #include "MitProd/TreeFiller/interface/FillerBasicClusters.h"
13 > #include "MitProd/TreeFiller/interface/FillerSuperClusters.h"
14 > #include "MitProd/TreeFiller/interface/FillerMuons.h"
15 > #include "MitProd/TreeFiller/interface/FillerElectrons.h"
16 > #include "MitProd/TreeFiller/interface/FillerCaloJets.h"
17 > #include "MitProd/TreeFiller/interface/FillerCaloMet.h"
18 > #include "MitProd/TreeFiller/interface/FillerConversions.h"
19 > #include "MitProd/TreeFiller/interface/FillerConversionElectrons.h"
20 > #include "MitProd/TreeFiller/interface/FillerPhotons.h"
21 > #include "MitProd/TreeFiller/interface/FillerMCParticles.h"
22 > #include "MitProd/TreeFiller/interface/FillerDecayParts.h"
23 > #include "MitProd/TreeFiller/interface/FillerStableParts.h"
24  
25 + #include "MitProd/TreeFiller/interface/FillerPATMuons.h"
26 + #include "MitProd/TreeFiller/interface/FillerPATElectrons.h"
27   using namespace std;
28   using namespace edm;
29   using namespace mithep;
30  
31 < //-------------------------------------------------------------------------------------------------
32 < FillMitTree::FillMitTree(const edm::ParameterSet &cfg)
31 > mithep::ObjectService *mithep::FillMitTree::os_ = 0;
32 >
33 > //--------------------------------------------------------------------------------------------------
34 > FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
35 >  defactive_(cfg.getUntrackedParameter<bool>("defactive",1))
36   {
37 <  // Constructor: initialize fillers
37 >  // Constructor.
38  
39    if (!configure(cfg)) {
40      throw edm::Exception(edm::errors::Configuration, "FillMitTree::FillMitTree()\n")
# Line 22 | Line 42 | FillMitTree::FillMitTree(const edm::Para
42    }
43   }
44  
45 < //-------------------------------------------------------------------------------------------------
45 > //--------------------------------------------------------------------------------------------------
46   FillMitTree::~FillMitTree()
47   {
48    // Destructor: nothing to be done here.
49   }
50  
51 < //-------------------------------------------------------------------------------------------------
51 > //--------------------------------------------------------------------------------------------------
52 > bool FillMitTree::addActiveFiller(BaseFiller *bf)
53 > {
54 >  // Check if filler is active and add it to list of fillers. Otherwise delete it.
55 >
56 >  if (bf->Active()) {
57 >    fillers_.push_back(bf);
58 >    return 1;
59 >  }
60 >
61 >  delete bf;
62 >  bf = 0;
63 >  return 0;
64 > }
65 >
66 > //--------------------------------------------------------------------------------------------------
67   void FillMitTree::analyze(const edm::Event      &event,
68                            const edm::EventSetup &setup)
69   {
# Line 45 | Line 80 | void FillMitTree::analyze(const edm::Eve
80    }
81   }
82  
83 < //-------------------------------------------------------------------------------------------------
83 > //--------------------------------------------------------------------------------------------------
84   void FillMitTree::beginJob(const edm::EventSetup &event)
85   {
86    // Access the tree and book branches.
# Line 54 | Line 89 | void FillMitTree::beginJob(const edm::Ev
89    TreeWriter *tws = ts->get();
90    if (! tws) {
91      throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
92 <      << "Could not get pointer to tree." << "\n";
92 >      << "Could not get pointer to tree. "
93 >      << "Do you have the TreeServie define in your config?" << "\n";
94      return;
95    }
96  
97 +  if (os_==0) {
98 +    Service<ObjectService> os;
99 +    if (!os.isAvailable()) {
100 +      throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
101 +        << "Could not get object service. "
102 +        << "Do you have the ObjectService defined in your config?" << "\n";
103 +      return;
104 +    }
105 +    os_ = &(*os);
106 +  }
107 +
108    // Loop over the various components and book the branches
109    for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
110 +    edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
111      (*iF)->BookDataBlock(*tws);
112    }
113   }
114  
115 < //-------------------------------------------------------------------------------------------------
115 > //--------------------------------------------------------------------------------------------------
116   bool FillMitTree::configure(const edm::ParameterSet &cfg)
117   {
118    // Configure our fillers.
119  
120 +  FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg,defactive_);
121 +  addActiveFiller(fillerMetaInfos);
122 +
123 +  FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg,"MCParticles",defactive_);
124 +  addActiveFiller(fillerMCParticles);
125 +
126 +  FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg,"CaloJets",defactive_);
127 +  addActiveFiller(fillerCaloJets);
128  
129 <  FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg);
130 <  if (fillerMetaInfos->Active())
131 <    fillers_.push_back(fillerMetaInfos);
132 <  else
133 <    delete fillerMetaInfos;
134 <
135 <  FillerGlobalMuons *fillerGlobalMuons = new FillerGlobalMuons(cfg);
136 <  if (fillerGlobalMuons->Active())
137 <    fillers_.push_back(fillerGlobalMuons);
138 <  else
139 <    delete fillerGlobalMuons;
129 >  FillerCaloJets *fillerSisCone5Jets = new FillerCaloJets(cfg,"SisCone5Jets",defactive_);
130 >  addActiveFiller(fillerSisCone5Jets);
131 >
132 >  FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg,"CaloMet",defactive_);
133 >  addActiveFiller(fillerCaloMet);
134 >
135 >  FillerTracks *fillerGeneralTracks = new FillerTracks(cfg,"GeneralTracks",defactive_);
136 >  addActiveFiller(fillerGeneralTracks);
137 >
138 >  FillerTracks *fillerStandaloneMuonTracks =
139 >    new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
140 >  addActiveFiller(fillerStandaloneMuonTracks);
141 >
142 >  FillerTracks *fillerStandaloneMuonTracksVtx =
143 >    new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
144 >  addActiveFiller(fillerStandaloneMuonTracksVtx);
145 >
146 >  FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
147 >  addActiveFiller(fillerGlobalMuonTracks);
148 >    
149 >  FillerTracks *fillerConversionInOutTracks =
150 >    new FillerTracks(cfg,"ConversionInOutTracks",defactive_);
151 >  addActiveFiller(fillerConversionInOutTracks);
152 >
153 >  FillerTracks *fillerConversionOutInTracks =
154 >    new FillerTracks(cfg,"ConversionOutInTracks",defactive_);
155 >  addActiveFiller(fillerConversionOutInTracks);
156 >
157 >  FillerGsfTracks *fillerGsfTracks = new FillerGsfTracks(cfg,"GsfTracks",defactive_);
158 >  addActiveFiller(fillerGsfTracks);
159 >
160 >  FillerBasicClusters *fillerBarrelBasicClusters =
161 >    new FillerBasicClusters(cfg, "BarrelBasicClusters", defactive_);
162 >  addActiveFiller(fillerBarrelBasicClusters);
163 >
164 >  FillerSuperClusters *fillerBarrelSuperClusters =  
165 >    new FillerSuperClusters(cfg,"BarrelSuperClusters", defactive_);
166 >  addActiveFiller(fillerBarrelSuperClusters);
167 >
168 >  FillerBasicClusters *fillerEndcapBasicClusters =
169 >    new FillerBasicClusters(cfg,"EndcapBasicClusters", defactive_);
170 >  addActiveFiller(fillerEndcapBasicClusters);
171 >  
172 >  FillerSuperClusters *fillerEndcapSuperClusters =  
173 >    new FillerSuperClusters(cfg,"EndcapSuperClusters", defactive_);
174 >  addActiveFiller(fillerEndcapSuperClusters);
175 >  
176 >  FillerMuons *fillerMuons = new FillerMuons(cfg,defactive_);
177 >  addActiveFiller(fillerMuons);
178 >  
179 >  FillerElectrons *fillerElectrons = new FillerElectrons(cfg,defactive_);
180 >  addActiveFiller(fillerElectrons);
181 >
182 >  FillerConversionElectrons *fillerConversionElectrons =
183 >    new FillerConversionElectrons(cfg,defactive_);
184 >  addActiveFiller(fillerConversionElectrons);
185 >
186 >  FillerConversions *fillerConversions = new FillerConversions(cfg,defactive_);
187 >  addActiveFiller(fillerConversions);
188 >
189 >  FillerPhotons *fillerPhotons = new FillerPhotons(cfg,defactive_);
190 >  addActiveFiller(fillerPhotons);
191 >
192 >  FillerStableParts *fillerStableParts = new FillerStableParts(cfg,"StableParts",defactive_);
193 >  addActiveFiller(fillerStableParts);
194 >  
195 >  FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg,"DecayParts",defactive_);
196 >  addActiveFiller(fillerDecayParts);
197  
198    return 1;
199   }
200  
201 < //-------------------------------------------------------------------------------------------------
201 > //--------------------------------------------------------------------------------------------------
202   void FillMitTree::endJob()
203   {
204    // Delete fillers.

Diff Legend

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