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.27 by bendavid, Mon Nov 3 18:11:10 2008 UTC vs.
Revision 1.62 by bendavid, Mon Sep 12 15:21:38 2011 UTC

# Line 2 | Line 2
2  
3   #include "MitProd/TreeFiller/interface/FillMitTree.h"
4   #include "FWCore/MessageLogger/interface/MessageLogger.h"
5 + #include "FWCore/ParameterSet/interface/ParameterSet.h"
6   #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/FillerVertexes.h"
11 #include "MitProd/TreeFiller/interface/FillerTracks.h"
9   #include "MitProd/TreeFiller/interface/FillerBasicClusters.h"
10 < #include "MitProd/TreeFiller/interface/FillerSuperClusters.h"
14 < #include "MitProd/TreeFiller/interface/FillerCaloTowers.h"
15 < #include "MitProd/TreeFiller/interface/FillerMuons.h"
16 < #include "MitProd/TreeFiller/interface/FillerElectrons.h"
17 < #include "MitProd/TreeFiller/interface/FillerGenJets.h"
10 > #include "MitProd/TreeFiller/interface/FillerBeamSpot.h"
11   #include "MitProd/TreeFiller/interface/FillerCaloJets.h"
12   #include "MitProd/TreeFiller/interface/FillerCaloMet.h"
13 + #include "MitProd/TreeFiller/interface/FillerCaloTaus.h"
14 + #include "MitProd/TreeFiller/interface/FillerCaloTowers.h"
15   #include "MitProd/TreeFiller/interface/FillerConversions.h"
16 < #include "MitProd/TreeFiller/interface/FillerConversionElectrons.h"
22 < #include "MitProd/TreeFiller/interface/FillerPhotons.h"
23 < #include "MitProd/TreeFiller/interface/FillerMCParticles.h"
16 > #include "MitProd/TreeFiller/interface/FillerConversionsDecay.h"
17   #include "MitProd/TreeFiller/interface/FillerDecayParts.h"
18 + #include "MitProd/TreeFiller/interface/FillerElectrons.h"
19 + #include "MitProd/TreeFiller/interface/FillerEvtSelData.h"
20 + #include "MitProd/TreeFiller/interface/FillerGenJets.h"
21 + #include "MitProd/TreeFiller/interface/FillerGenMet.h"
22 + #include "MitProd/TreeFiller/interface/FillerJPTJets.h"
23 + #include "MitProd/TreeFiller/interface/FillerMCEventInfo.h"
24 + #include "MitProd/TreeFiller/interface/FillerMCParticles.h"
25 + #include "MitProd/TreeFiller/interface/FillerMCVertexes.h"
26 + #include "MitProd/TreeFiller/interface/FillerMet.h"
27 + #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
28 + #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
29 + #include "MitProd/TreeFiller/interface/FillerMuons.h"
30 + #include "MitProd/TreeFiller/interface/FillerPFCandidates.h"
31 + #include "MitProd/TreeFiller/interface/FillerPFJets.h"
32 + #include "MitProd/TreeFiller/interface/FillerPFMet.h"
33 + #include "MitProd/TreeFiller/interface/FillerPFTaus.h"
34 + #include "MitProd/TreeFiller/interface/FillerNSVFit.h"
35 + #include "MitProd/TreeFiller/interface/FillerPhotons.h"
36 + #include "MitProd/TreeFiller/interface/FillerPileupInfo.h"
37 + #include "MitProd/TreeFiller/interface/FillerPileupEnergyDensity.h"
38 + #include "MitProd/TreeFiller/interface/FillerPixelHits.h"
39   #include "MitProd/TreeFiller/interface/FillerStableParts.h"
40 < #include "MitProd/TreeFiller/interface/FillerPATMuons.h"
41 < #include "MitProd/TreeFiller/interface/FillerPATElectrons.h"
40 > #include "MitProd/TreeFiller/interface/FillerStripHits.h"
41 > #include "MitProd/TreeFiller/interface/FillerSuperClusters.h"
42 > #include "MitProd/TreeFiller/interface/FillerTracks.h"
43 > #include "MitProd/TreeFiller/interface/FillerTrackJets.h"
44 > #include "MitProd/TreeFiller/interface/FillerVertexes.h"
45 > #include "MitAna/DataTree/interface/Names.h"
46 > #include "MitAna/DataTree/interface/BranchTable.h"
47 > #include "MitCommon/OptIO/interface/OptInt.h"
48  
49   using namespace std;
50   using namespace edm;
# Line 34 | Line 54 | mithep::ObjectService *mithep::FillMitTr
54  
55   //--------------------------------------------------------------------------------------------------
56   FillMitTree::FillMitTree(const edm::ParameterSet &cfg) :
57 <  defactive_(cfg.getUntrackedParameter<bool>("defactive",1))
57 >  defactive_(cfg.getUntrackedParameter<bool>("defactive",1)),
58 >  brtable_(0),
59 >  acfnumber_(-1),
60 >  tws_(new TreeWriter(Names::gkEvtTreeName,0))
61   {
62    // Constructor.
63  
# Line 47 | Line 70 | FillMitTree::FillMitTree(const edm::Para
70   //--------------------------------------------------------------------------------------------------
71   FillMitTree::~FillMitTree()
72   {
73 <  // Destructor: nothing to be done here.
73 >  // Destructor.
74 >
75 >  delete brtable_;
76 >  delete tws_;
77   }
78  
79   //--------------------------------------------------------------------------------------------------
# Line 55 | Line 81 | bool FillMitTree::addActiveFiller(BaseFi
81   {
82    // Check if filler is active and add it to list of fillers. Otherwise delete it.
83  
84 +  if (!bf)
85 +    return 0;
86 +
87    if (bf->Active()) {
88      fillers_.push_back(bf);
89      return 1;
# Line 66 | Line 95 | bool FillMitTree::addActiveFiller(BaseFi
95   }
96  
97   //--------------------------------------------------------------------------------------------------
98 + void FillMitTree::beginRun(edm::Run const &run, edm::EventSetup const &setup)
99 + {
100 +  // Access and copy event content.
101 +
102 +  // first step: Loop over the data fillers of the various components
103 +  for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
104 +    (*iF)->FillRunBlock(run,setup);
105 +  }
106 +  
107 + }
108 +
109 + //--------------------------------------------------------------------------------------------------
110   void FillMitTree::analyze(const edm::Event      &event,
111                            const edm::EventSetup &setup)
112   {
113    // Access and copy event content.
114 +  
115 +  //tree writer begin event actions
116 +  tws_->BeginEvent(kTRUE);
117  
118 <  // First step: Loop over the data fillers of the various components
118 >  // first step: Loop over the data fillers of the various components
119    for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
120 +    //printf("%s FillDataBlock\n",(*iF)->Name().c_str());
121      (*iF)->FillDataBlock(event,setup);
122    }
123  
124 <  // Second step: Loop over the link resolution of the various components
124 >  // second step: Loop over the link resolution of the various components
125    for (std::vector<BaseFiller*>::const_iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
126 +    //printf("%s ResolveLinks\n",(*iF)->Name().c_str());
127      (*iF)->ResolveLinks(event,setup);
128    }
129 +
130 +  if (brtable_) { // only the first FillMitTree object has to deal with the branch table
131 +    if (acfnumber_==-1) {
132 +      brtable_->Rehash(brtable_->GetSize());
133 +      if (0)
134 +        brtable_->Print();
135 +    }
136 +    if (acfnumber_ != tws_->GetFileNumber()) {
137 +      tws_->StoreObject(brtable_);
138 +      acfnumber_ = tws_->GetFileNumber();
139 +    }
140 +  }
141 +
142 +  //tree writer end of event actions
143 +  tws_->EndEvent(kTRUE);
144   }
145  
146   //--------------------------------------------------------------------------------------------------
147 < void FillMitTree::beginJob(const edm::EventSetup &event)
147 > void FillMitTree::beginJob()
148   {
149    // Access the tree and book branches.
150  
151 <  Service<TreeService> ts;
91 <  TreeWriter *tws = ts->get();
92 <  if (! tws) {
93 <    throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
94 <      << "Could not get pointer to tree. "
95 <      << "Do you have the TreeServie define in your config?" << "\n";
96 <    return;
97 <  }
98 <
99 <  if (os_==0) {
151 >  if (os_==0) { // only the first FillMitTree object has to deal with this
152      Service<ObjectService> os;
153      if (!os.isAvailable()) {
154        throw edm::Exception(edm::errors::Configuration, "FillMitTree::beginJob()\n")
# Line 105 | Line 157 | void FillMitTree::beginJob(const edm::Ev
157        return;
158      }
159      os_ = &(*os);
160 +    brtable_ = new BranchTable;
161 +    brtable_->SetName(Names::gkBranchTable);
162 +    brtable_->SetOwner();
163 +    os->add(brtable_, brtable_->GetName());
164    }
165  
166 <  // Loop over the various components and book the branches
166 >  // loop over the various components and book the branches
167    for (std::vector<BaseFiller*>::iterator iF = fillers_.begin(); iF != fillers_.end(); ++iF) {
168      edm::LogInfo("FillMitTree::beginJob") << "Booking for " << (*iF)->Name() << endl;
169 <    (*iF)->BookDataBlock(*tws);
169 >    (*iF)->BookDataBlock(*tws_);
170    }
171 +
172 +  // call branch ref for the event tree
173 +  if (brtable_ && tws_->GetTree())
174 +    tws_->GetTree()->BranchRef();
175 +  
176   }
177  
178   //--------------------------------------------------------------------------------------------------
179   bool FillMitTree::configure(const edm::ParameterSet &cfg)
180   {
120  // Configure our fillers.
121
122  FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg,defactive_);
123  addActiveFiller(fillerMetaInfos);
124
125  FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg,"MCParticles",defactive_);
126  addActiveFiller(fillerMCParticles);
127
128  FillerVertexes *fillerPrimaryVertexes =
129    new FillerVertexes(cfg,"PrimaryVertexes", defactive_);
130  addActiveFiller(fillerPrimaryVertexes);
131  
132  //primary vertexes with beamspot constraint
133  FillerVertexes *fillerPrimaryVertexesBS =
134    new FillerVertexes(cfg,"PrimaryVertexesBS", defactive_);
135  addActiveFiller(fillerPrimaryVertexesBS);  
136  
137  FillerCaloTowers *fillerCaloTowers =
138    new FillerCaloTowers(cfg, "CaloTowers", defactive_);
139  addActiveFiller(fillerCaloTowers);
140
141  FillerGenJets *fillerIC5GenJets = new FillerGenJets(cfg,"IC5GenJets",defactive_);
142  addActiveFiller(fillerIC5GenJets);
143
144  FillerGenJets *fillerSC5GenJets = new FillerGenJets(cfg,"SC5GenJets",defactive_);
145  addActiveFiller(fillerSC5GenJets);
146  
147  FillerGenJets *fillerSC7GenJets = new FillerGenJets(cfg,"SC7GenJets",defactive_);
148  addActiveFiller(fillerSC7GenJets);
149
150  FillerGenJets *fillerKT4GenJets = new FillerGenJets(cfg,"KT4GenJets",defactive_);
151  addActiveFiller(fillerKT4GenJets);
152  
153  FillerGenJets *fillerKT6GenJets = new FillerGenJets(cfg,"KT6GenJets",defactive_);
154  addActiveFiller(fillerKT6GenJets);
155
156  FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg,"CaloJets",defactive_);
157  addActiveFiller(fillerCaloJets);
158
159  FillerCaloJets *fillerItrCone5Jets = new FillerCaloJets(cfg,"ItrCone5Jets",defactive_);
160  addActiveFiller(fillerItrCone5Jets);
181  
182 <  FillerCaloJets *fillerSisCone5Jets = new FillerCaloJets(cfg,"SisCone5Jets",defactive_);
183 <  addActiveFiller(fillerSisCone5Jets);
184 <
185 <  FillerCaloJets *fillerSisCone7Jets = new FillerCaloJets(cfg,"SisCone7Jets",defactive_);
186 <  addActiveFiller(fillerSisCone7Jets);
187 <
188 <  FillerCaloJets *fillerKt4Jets = new FillerCaloJets(cfg,"Kt4Jets",defactive_);
189 <  addActiveFiller(fillerKt4Jets);    
190 <  
191 <  FillerCaloJets *fillerKt6Jets = new FillerCaloJets(cfg,"Kt6Jets",defactive_);
192 <  addActiveFiller(fillerKt6Jets);
182 >  // Configure TreeWriter
183 >  const std::string twsConfigName("TreeWriter");
184 >  ParameterSet twsConfig;
185 >  if (cfg.existsAs<ParameterSet>(twsConfigName,0))
186 >    twsConfig = cfg.getUntrackedParameter<ParameterSet>(twsConfigName);
187 >  
188 >  configureTreeWriter(twsConfig);
189 >  
190 >  // Configure our fillers according to given parameter ("fillers").
191 >
192 >  std::vector<std::string> pars;
193 >  if (cfg.exists("fillers"))
194 >    pars=cfg.getUntrackedParameter<vector<string> >("fillers");
195 >  else
196 >    cfg.getParameterSetNames(pars, false);
197 >
198 >  // loop over psets
199 >  for (unsigned int i = 0; i<pars.size(); ++i) {
200 >
201 >    const string name(pars.at(i));
202 >
203 >    string ftype("Filler" + name);
204 >    if (cfg.existsAs<ParameterSet>(name,0)) {
205 >      ParameterSet next(cfg.getUntrackedParameter<ParameterSet>(name));
206 >      if (!next.exists("fillerType")) {
207 >        edm::LogError("FillMitTree") << "Cannot determine fillerType for pset named "
208 >                                     << name << std::endl;
209 >        throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
210 >          << "Cannot determine fillerType for pset named "
211 >          << name << std::endl;
212 >      }
213 >      ftype = next.getUntrackedParameter<string>("fillerType");
214 >    }
215  
216 <  FillerCaloJets *fillerIC5JetPlusTrack = new FillerCaloJets(cfg,"IC5JetPlusTrack",defactive_);
217 <  addActiveFiller(fillerIC5JetPlusTrack);
216 >    edm::LogInfo("FillMitTree") << "Attempting to configure '" << ftype
217 >                                << "' for '" << name << "'" << std::endl;
218  
219 <  FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg,"CaloMet",defactive_);
220 <  addActiveFiller(fillerCaloMet);
219 >    if (ftype.compare("FillerMetaInfos")==0) {
220 >      FillerMetaInfos *fillerMetaInfos = new FillerMetaInfos(cfg, name.c_str(), defactive_);
221 >      addActiveFiller(fillerMetaInfos);
222 >      continue;
223 >    }
224  
225 <  FillerCaloMet *fillerItrCone5Met = new FillerCaloMet(cfg,"ItrCone5Met",defactive_);
226 <  addActiveFiller(fillerItrCone5Met);
227 <  
228 <  FillerCaloMet *fillerSisCone5Met = new FillerCaloMet(cfg,"SisCone5Met",defactive_);
229 <  addActiveFiller(fillerSisCone5Met);
185 <  
186 <  FillerCaloMet *fillerSisCone7Met = new FillerCaloMet(cfg,"SisCone7Met",defactive_);
187 <  addActiveFiller(fillerSisCone7Met);
225 >    if (ftype.compare("FillerMCParticles")==0) {
226 >      FillerMCParticles *fillerMCParticles = new FillerMCParticles(cfg, name.c_str(), defactive_);
227 >      addActiveFiller(fillerMCParticles);
228 >      continue;
229 >    }
230  
231 <  FillerCaloMet *fillerKt4Met = new FillerCaloMet(cfg,"Kt4Met",defactive_);
232 <  addActiveFiller(fillerKt4Met);
233 <  
234 <  FillerCaloMet *fillerKt6Met = new FillerCaloMet(cfg,"Kt6Met",defactive_);
235 <  addActiveFiller(fillerKt6Met);  
231 >    if (ftype.compare("FillerMCEventInfo")==0) {
232 >      FillerMCEventInfo *fillerMCEventInfo = new FillerMCEventInfo(cfg, name.c_str(), defactive_);
233 >      addActiveFiller(fillerMCEventInfo);
234 >      continue;
235 >    }
236  
237 <  FillerBasicClusters *fillerBarrelBasicClusters =
238 <    new FillerBasicClusters(cfg, "BarrelBasicClusters", defactive_);
239 <  addActiveFiller(fillerBarrelBasicClusters);
240 <
241 <  FillerSuperClusters *fillerBarrelSuperClusters =  
242 <    new FillerSuperClusters(cfg,"BarrelSuperClusters", defactive_);
243 <  addActiveFiller(fillerBarrelSuperClusters);
244 <
245 <  FillerBasicClusters *fillerEndcapBasicClusters =
246 <    new FillerBasicClusters(cfg,"EndcapBasicClusters", defactive_);
247 <  addActiveFiller(fillerEndcapBasicClusters);
206 <  
207 <  FillerSuperClusters *fillerEndcapSuperClusters =  
208 <    new FillerSuperClusters(cfg,"EndcapSuperClusters", defactive_);
209 <  addActiveFiller(fillerEndcapSuperClusters);
210 <  
211 <  FillerTracks *fillerGeneralTracks = new FillerTracks(cfg,"GeneralTracks",defactive_);
212 <  addActiveFiller(fillerGeneralTracks);
237 >    if (ftype.compare("FillerMCVertexes")==0) {
238 >      FillerMCVertexes *fillerMCVertexes = new FillerMCVertexes(cfg, name.c_str(), defactive_);
239 >      addActiveFiller(fillerMCVertexes);
240 >      continue;
241 >    }  
242 >
243 >    if (ftype.compare("FillerEvtSelData")==0) {
244 >      FillerEvtSelData *fillerEvtSelData = new FillerEvtSelData(cfg, name.c_str(), defactive_);
245 >      addActiveFiller(fillerEvtSelData);
246 >      continue;
247 >    }
248  
249 <  FillerTracks *fillerStandaloneMuonTracks =
250 <    new FillerTracks(cfg,"StandaloneMuonTracks",defactive_);
251 <  addActiveFiller(fillerStandaloneMuonTracks);
252 <
253 <  FillerTracks *fillerStandaloneMuonTracksVtx =
219 <    new FillerTracks(cfg,"StandaloneMuonTracksWVtxConstraint",defactive_);
220 <  addActiveFiller(fillerStandaloneMuonTracksVtx);
249 >    if (ftype.compare("FillerBeamSpot")==0) {
250 >      FillerBeamSpot *fillerBeamSpot = new FillerBeamSpot(cfg, name.c_str(), defactive_);
251 >      addActiveFiller(fillerBeamSpot);
252 >      continue;
253 >    }
254  
255 <  FillerTracks *fillerGlobalMuonTracks = new FillerTracks(cfg,"GlobalMuonTracks",defactive_);
256 <  addActiveFiller(fillerGlobalMuonTracks);
255 >    if (ftype.compare("FillerVertexes")==0) {
256 >      FillerVertexes *fillerVertexes = new FillerVertexes(cfg, name.c_str(), defactive_);
257 >      addActiveFiller(fillerVertexes);
258 >      continue;
259 >    }  
260 >
261 >    if (ftype.compare("FillerCaloTowers")==0) {
262 >      FillerCaloTowers *fillerCaloTowers = new FillerCaloTowers(cfg, name.c_str(), defactive_);
263 >      addActiveFiller(fillerCaloTowers);
264 >      continue;
265 >    }  
266 >
267 >    if (ftype.compare("FillerGenJets")==0) {
268 >      FillerGenJets *fillerGenJets = new FillerGenJets(cfg, name.c_str(), defactive_);
269 >      addActiveFiller(fillerGenJets);
270 >      continue;
271 >    }  
272 >
273 >    if (ftype.compare("FillerCaloJets")==0) {
274 >      FillerCaloJets *fillerCaloJets = new FillerCaloJets(cfg, name.c_str(), defactive_);
275 >      addActiveFiller(fillerCaloJets);
276 >      continue;
277 >    }  
278      
279 <  FillerTracks *fillerConversionInOutTracks =
280 <    new FillerTracks(cfg,"ConversionInOutTracks",defactive_);
281 <  addActiveFiller(fillerConversionInOutTracks);
282 <
283 <  FillerTracks *fillerConversionOutInTracks =
284 <    new FillerTracks(cfg,"ConversionOutInTracks",defactive_);
285 <  addActiveFiller(fillerConversionOutInTracks);
286 <
287 <  FillerTracks *fillerGsfTracks = new FillerTracks(cfg,"GsfTracks",defactive_);
288 <  addActiveFiller(fillerGsfTracks);
289 <
290 <  FillerMuons *fillerMuons = new FillerMuons(cfg,defactive_);
291 <  addActiveFiller(fillerMuons);
292 <  
293 <  FillerElectrons *fillerElectrons = new FillerElectrons(cfg,defactive_);
294 <  addActiveFiller(fillerElectrons);
295 <
296 <  FillerConversionElectrons *fillerConversionElectrons =
297 <    new FillerConversionElectrons(cfg,defactive_);
298 <  addActiveFiller(fillerConversionElectrons);
279 >    if (ftype.compare("FillerMet")==0) {
280 >      FillerMet *fillerMet = new FillerMet(cfg, name.c_str(), defactive_);
281 >      addActiveFiller(fillerMet);
282 >      continue;
283 >    }  
284 >    
285 >    if (ftype.compare("FillerGenMet")==0) {
286 >      FillerGenMet *fillerGenMet = new FillerGenMet(cfg, name.c_str(), defactive_);
287 >      addActiveFiller(fillerGenMet);
288 >      continue;
289 >    }  
290 >
291 >    if (ftype.compare("FillerCaloMet")==0) {
292 >      FillerCaloMet *fillerCaloMet = new FillerCaloMet(cfg, name.c_str(), defactive_);
293 >      addActiveFiller(fillerCaloMet);
294 >      continue;
295 >    }
296 >    
297 >    if (ftype.compare("FillerPFMet")==0) {
298 >      FillerPFMet *fillerPFMet = new FillerPFMet(cfg, name.c_str(), defactive_);
299 >      addActiveFiller(fillerPFMet);
300 >      continue;
301 >    }  
302 >
303 >    if (ftype.compare("FillerBasicClusters")==0) {
304 >      FillerBasicClusters *fillerBasicClusters =
305 >        new FillerBasicClusters(cfg, name.c_str(), defactive_);
306 >      addActiveFiller(fillerBasicClusters);
307 >      continue;
308 >    }  
309 >
310 >    if (ftype.compare("FillerSuperClusters")==0) {
311 >      FillerSuperClusters *fillerSuperClusters =  
312 >        new FillerSuperClusters(cfg, name.c_str(), defactive_);
313 >      addActiveFiller(fillerSuperClusters);
314 >      continue;
315 >    }  
316 >
317 >    if (ftype.compare("FillerPixelHits")==0) {
318 >      FillerPixelHits *fillerPixelHits =  
319 >        new FillerPixelHits(cfg, name.c_str(), defactive_);
320 >      addActiveFiller(fillerPixelHits);
321 >      continue;
322 >    }  
323 >
324 >    if (ftype.compare("FillerPileupInfo")==0) {
325 >      FillerPileupInfo *fillerPileupInfo =  
326 >        new FillerPileupInfo(cfg, name.c_str(), defactive_);
327 >      addActiveFiller(fillerPileupInfo);
328 >      continue;
329 >    }  
330 >
331 >    if (ftype.compare("FillerPileupEnergyDensity")==0) {
332 >      FillerPileupEnergyDensity *fillerPileupEnergyDensity =  
333 >        new FillerPileupEnergyDensity(cfg, name.c_str(), defactive_);
334 >      addActiveFiller(fillerPileupEnergyDensity);
335 >      continue;
336 >    }  
337 >
338 >    if (ftype.compare("FillerStripHits")==0) {
339 >      FillerStripHits *fillerStripHits =  
340 >        new FillerStripHits(cfg, name.c_str(), defactive_);
341 >      addActiveFiller(fillerStripHits);
342 >      continue;
343 >    }  
344 >
345 >    if (ftype.compare("FillerTracks")==0) {
346 >      FillerTracks *fillerTracks = new FillerTracks(cfg, name.c_str(), defactive_);
347 >      addActiveFiller(fillerTracks);
348 >      continue;
349 >    }  
350 >
351 >    if (ftype.compare("FillerMuons")==0) {
352 >      FillerMuons *fillerMuons = new FillerMuons(cfg, name.c_str(), defactive_);
353 >      addActiveFiller(fillerMuons);
354 >      continue;
355 >    }  
356 >
357 >    if (ftype.compare("FillerElectrons")==0) {
358 >      FillerElectrons *fillerElectrons = new FillerElectrons(cfg, name.c_str(), defactive_);
359 >      addActiveFiller(fillerElectrons);
360 >      continue;
361 >    }  
362 >
363 >    if (ftype.compare("FillerConversions")==0) {
364 >      FillerConversions *fillerConversions = new FillerConversions(cfg, name.c_str(), defactive_);
365 >      addActiveFiller(fillerConversions);
366 >      continue;
367 >    }
368 >    
369 >    if (ftype.compare("FillerConversionsDecay")==0) {
370 >      FillerConversionsDecay *fillerConversionsDecay = new FillerConversionsDecay(cfg, name.c_str(), defactive_);
371 >      addActiveFiller(fillerConversionsDecay);
372 >      continue;
373 >    }    
374 >
375 >    if (ftype.compare("FillerPhotons")==0) {
376 >      FillerPhotons *fillerPhotons = new FillerPhotons(cfg, name.c_str(), defactive_);
377 >      addActiveFiller(fillerPhotons);
378 >      continue;
379 >    }  
380 >
381 >    if (ftype.compare("FillerStableParts")==0) {
382 >      FillerStableParts *fillerStableParts = new FillerStableParts(cfg, name.c_str(), defactive_);
383 >      addActiveFiller(fillerStableParts);
384 >      continue;
385 >    }  
386 >
387 >    if (ftype.compare("FillerDecayParts")==0) {
388 >      FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg, name.c_str(), defactive_);
389 >      addActiveFiller(fillerDecayParts);
390 >      continue;
391 >    }  
392 >    
393 >    if (ftype.compare("FillerPFCandidates")==0) {
394 >      FillerPFCandidates *fillerPFCands = new FillerPFCandidates(cfg, name.c_str(), defactive_);
395 >      addActiveFiller(fillerPFCands);
396 >      continue;
397 >    }  
398 >
399 >    if (ftype.compare("FillerPFJets")==0) {
400 >      FillerPFJets *fillerPFJets = new FillerPFJets(cfg, name.c_str(), defactive_);
401 >      addActiveFiller(fillerPFJets);
402 >      continue;
403 >    }  
404 >
405 >    if (ftype.compare("FillerJPTJets")==0) {
406 >      FillerJPTJets *fillerJPTJets = new FillerJPTJets(cfg, name.c_str(), defactive_);
407 >      addActiveFiller(fillerJPTJets);
408 >      continue;
409 >    }  
410 >
411 >    if (ftype.compare("FillerCaloTaus")==0) {
412 >      FillerCaloTaus *fillerCaloTaus = new FillerCaloTaus(cfg, name.c_str(), defactive_);
413 >      addActiveFiller(fillerCaloTaus);
414 >      continue;
415 >    }
416 >    
417 >    if (ftype.compare("FillerPFTaus")==0) {
418 >      FillerPFTaus *fillerPFTaus = new FillerPFTaus(cfg, name.c_str(), defactive_);
419 >      addActiveFiller(fillerPFTaus);
420 >      continue;
421 >    }  
422 >
423 >    if (ftype.compare("FillerNSVFit")==0) {
424 >      FillerNSVFit *fillerNSVFit = new FillerNSVFit(cfg, name.c_str(), defactive_);
425 >      addActiveFiller(fillerNSVFit);
426 >      continue;
427 >    }  
428 >
429 >    if (ftype.compare("FillerTrackJets")==0) {
430 >      FillerTrackJets *fillerTrackJets = new FillerTrackJets(cfg, name.c_str(), defactive_);
431 >      addActiveFiller(fillerTrackJets);
432 >      continue;
433 >    }
434 >    
435 >    edm::LogError("FillMitTree")
436 >      << "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
437 >    throw edm::Exception(edm::errors::Configuration, "FillMitTree::configure\n")
438 >      << "Unknown fillerType " << ftype << " for pset named " << name << std::endl;
439 >  }
440  
441 <  FillerConversions *fillerConversions = new FillerConversions(cfg,defactive_);
442 <  addActiveFiller(fillerConversions);
441 >  return 1;
442 > }
443  
444 <  FillerPhotons *fillerPhotons = new FillerPhotons(cfg,defactive_);
445 <  addActiveFiller(fillerPhotons);
444 > //--------------------------------------------------------------------------------------------------
445 > bool FillMitTree::configureTreeWriter(const edm::ParameterSet &cfg)
446 > {
447 >  // Configure tree writer with options from config file.
448  
449 <  FillerStableParts *fillerStableParts = new FillerStableParts(cfg,"StableParts",defactive_);
450 <  addActiveFiller(fillerStableParts);
449 >  tws_->SetPrefix(cfg.getUntrackedParameter<string>("fileName","mit-test"));
450 >  tws_->SetBaseURL(cfg.getUntrackedParameter<string>("pathName","."));
451 >  tws_->SetMaxSize((Long64_t)cfg.getUntrackedParameter<unsigned>("maxSize",1024)*1024*1024);
452 >  tws_->SetCompressLevel(cfg.getUntrackedParameter<unsigned>("compLevel",9));
453 >  tws_->SetDefaultSL(cfg.getUntrackedParameter<unsigned>("splitLevel",99));
454 >  tws_->SetDefaultBrSize(cfg.getUntrackedParameter<unsigned>("brSize",16*1024));
455 >  
456 >  if (OptInt::IsActivated()) {
457 >    OptInt::SetZipMode(cfg.getUntrackedParameter<unsigned>("zipMode",99));
458 >    OptInt::SetGzipFraction(cfg.getUntrackedParameter<double>("gZipThres",1.0));
459 >    OptInt::SetBzipFraction(cfg.getUntrackedParameter<double>("bZipThres",-1.0));
460 >    OptInt::SetLzoFraction(cfg.getUntrackedParameter<double>("lzoThres",-1.0));
461 >    OptInt::SetLzmaFraction(cfg.getUntrackedParameter<double>("lzmaThres",0.95));
462 >    OptInt::SetVerbose(cfg.getUntrackedParameter<unsigned>("optIOVerbose",0));
463 >
464 >  } else {
465 >
466 >    if (cfg.exists("zipMode")   || cfg.exists("bZipThres") ||
467 >        cfg.exists("gZipThres") || cfg.exists("lzoThres")  ||
468 >        cfg.exists("lzmaThres")) {
469 >      edm::LogError("FillMitTree") <<
470 >        "OptIO interface not properly pre-loaded, ignoring given settings." << std::endl;
471 >    }
472 >  }
473    
255  FillerDecayParts *fillerDecayParts = new FillerDecayParts(cfg,"DecayParts",defactive_);
256  addActiveFiller(fillerDecayParts);
257
474    return 1;
475   }
476  
# Line 267 | Line 483 | void FillMitTree::endJob()
483      delete *iF;
484    }
485  
486 +  tws_->Clear();
487 +  
488    edm::LogInfo("FillMitTree::endJob") << "Ending Job" << endl;
489   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines