ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillerMetaInfos.cc
Revision: 1.26
Committed: Tue Mar 10 15:56:01 2009 UTC (16 years, 1 month ago) by loizides
Content type: text/plain
Branch: MAIN
Changes since 1.25: +3 -3 lines
Log Message:
Introduced more dynamic filling interface.

File Contents

# User Rev Content
1 loizides 1.26 // $Id: FillerMetaInfos.cc,v 1.25 2009/03/08 12:16:01 loizides Exp $
2 loizides 1.1
3     #include "MitProd/TreeFiller/interface/FillerMetaInfos.h"
4     #include "FWCore/MessageLogger/interface/MessageLogger.h"
5 loizides 1.5 #include "FWCore/Framework/interface/TriggerNames.h"
6 loizides 1.1 #include "DataFormats/Common/interface/Handle.h"
7 loizides 1.5 #include "DataFormats/Common/interface/TriggerResults.h"
8 loizides 1.14 #include "DataFormats/HLTReco/interface/TriggerEvent.h"
9 loizides 1.1 #include "MitAna/DataTree/interface/Names.h"
10 loizides 1.5 #include "MitAna/DataTree/interface/EventHeader.h"
11     #include "MitAna/DataTree/interface/LAHeader.h"
12 loizides 1.25 #include "MitAna/DataTree/interface/TriggerMask.h"
13 loizides 1.16 #include "MitAna/DataTree/interface/TriggerName.h"
14 loizides 1.5 #include "MitAna/DataTree/interface/RunInfo.h"
15 loizides 1.8 #include <TObjectTable.h>
16 loizides 1.14 #include <TIterator.h>
17 sixie 1.13
18 loizides 1.1 using namespace std;
19     using namespace edm;
20     using namespace mithep;
21    
22 loizides 1.12 bool mithep::FillerMetaInfos::instance_ = 0;
23    
24 loizides 1.8 //--------------------------------------------------------------------------------------------------
25 loizides 1.26 FillerMetaInfos::FillerMetaInfos(const ParameterSet &cfg, const char *name, bool active) :
26     BaseFiller(cfg,name,(instance_==0||active?1:0)),
27 loizides 1.2 evtName_(Conf().getUntrackedParameter<string>("evtName",Names::gkEvtHeaderBrn)),
28     runName_(Conf().getUntrackedParameter<string>("runName",Names::gkRunInfoBrn)),
29 loizides 1.4 lahName_(Conf().getUntrackedParameter<string>("lahName",Names::gkLAHeaderBrn)),
30 loizides 1.16 l1Active_(Conf().getUntrackedParameter<bool>("l1Active",true)),
31     l1TableName_(Conf().getUntrackedParameter<string>("l1TableName",Names::gkL1TableBrn)),
32     l1BitsName_(Conf().getUntrackedParameter<string>("l1BitsName",Names::gkL1BitBrn)),
33     l1ObjsName_(Conf().getUntrackedParameter<string>("l1ObjsName",Names::gkL1ObjBrn)),
34     hltActive_(Conf().getUntrackedParameter<bool>("hltActive",true)),
35     hltProcName_(Conf().getUntrackedParameter<string>("hltProcName","HLT")),
36     hltResName_(Conf().getUntrackedParameter<string>("hltResName","TriggerResults::HLT")),
37     hltEvtName_(Conf().getUntrackedParameter<string>("hltEvtName","hltTriggerSummaryAOD::HLT")),
38     hltTableName_(Conf().getUntrackedParameter<string>("hltTableName",Names::gkHltTableBrn)),
39     hltLabelName_(Conf().getUntrackedParameter<string>("hltLabelName",Names::gkHltLabelBrn)),
40     hltBitsName_(Conf().getUntrackedParameter<string>("hltBitsName",Names::gkHltBitBrn)),
41     hltObjsName_(Conf().getUntrackedParameter<string>("hltObjsName",Names::gkHltObjBrn)),
42 loizides 1.5 tws_(0),
43 loizides 1.2 eventHeader_(new EventHeader()),
44 loizides 1.3 evtLAHeader_(new LAHeader()),
45 loizides 1.2 runInfo_(new RunInfo()),
46     runTree_(0),
47 loizides 1.3 laTree_(0),
48     runEntries_(0),
49 loizides 1.16 l1Entries_(0),
50     l1Table_(0),
51     l1Tree_(0),
52 loizides 1.25 hltBits_(new TriggerMask),
53 loizides 1.22 hltTable_(new vector<string>),
54 loizides 1.16 hltTabMap_(0),
55 loizides 1.22 hltLabels_(new vector<string>),
56 loizides 1.16 hltLabMap_(0),
57     hltObjs_(new TriggerObjectBaseArr),
58     hltRels_(new TriggerObjectRelArr),
59     hltTree_(0),
60     hltEntries_(0),
61 loizides 1.3 fileNum_(0)
62 loizides 1.1 {
63 loizides 1.3 // Constructor.
64 loizides 1.12
65     instance_=1;
66 loizides 1.1 }
67    
68 loizides 1.8 //--------------------------------------------------------------------------------------------------
69 loizides 1.1 FillerMetaInfos::~FillerMetaInfos()
70     {
71 loizides 1.3 // Destructor.
72    
73 loizides 1.2 delete eventHeader_;
74 loizides 1.8 delete evtLAHeader_;
75 loizides 1.2 delete runInfo_;
76 loizides 1.16 delete l1Table_;
77     delete hltTable_;
78     delete hltLabels_;
79     delete hltObjs_;
80     delete hltRels_;
81 loizides 1.3 eventHeader_ = 0;
82 loizides 1.8 evtLAHeader_ = 0;
83 loizides 1.3 runInfo_ = 0;
84 loizides 1.16 l1Table_ = 0;
85     hltTable_ = 0;
86     hltLabels_ = 0;
87     hltObjs_ = 0;
88     hltRels_ = 0;
89 loizides 1.3 runTree_ = 0;
90     laTree_ = 0;
91 loizides 1.16 hltTree_ = 0;
92     l1Tree_ = 0;
93 loizides 1.1 }
94    
95 loizides 1.8 //--------------------------------------------------------------------------------------------------
96 loizides 1.2 void FillerMetaInfos::BookDataBlock(TreeWriter &tws)
97 loizides 1.1 {
98 loizides 1.3 // Create run info tre and book our branches.
99    
100     // add branches to main tree
101 loizides 1.24 tws.AddBranch(evtName_.c_str(),&eventHeader_);
102     tws.AddBranch(hltBitsName_.c_str(),&hltBits_);
103 loizides 1.16 tws.AddBranch(hltObjsName_.c_str(),&hltObjs_);
104     tws.AddBranch(Form("%sRelation",hltObjsName_.c_str()),&hltRels_);
105 loizides 1.20 tws.GetTree()->BranchRef();
106 loizides 1.3
107     // add branches to run info tree
108 loizides 1.24 tws.AddBranchToTree(Names::gkRunTreeName,runName_.c_str(),
109 loizides 1.25 TClass::GetClass(typeid(*runInfo_))->GetName(),&runInfo_);
110 loizides 1.2 tws.SetAutoFill(Names::gkRunTreeName,0);
111     runTree_=tws.GetTree(Names::gkRunTreeName);
112 loizides 1.3
113     // add branches to lookahead tree
114 loizides 1.24 tws.AddBranchToTree(Names::gkLATreeName,Names::gkLAHeaderBrn,
115 loizides 1.25 TClass::GetClass(typeid(*evtLAHeader_))->GetName(),&runInfo_);
116 loizides 1.3 tws.SetAutoFill(Names::gkLATreeName,0);
117     laTree_=tws.GetTree(Names::gkLATreeName);
118    
119 loizides 1.16 // add branches to L1 trigger info tree
120    
121     // add branches to HLT trigger info tree
122 loizides 1.23 tws.AddBranchToTree(Names::gkHltTreeName,hltTableName_.c_str(),
123 loizides 1.25 TClass::GetClass(typeid(*hltTable_))->GetName(),&hltTable_,32000,0);
124 loizides 1.23 tws.AddBranchToTree(Names::gkHltTreeName,hltLabelName_.c_str(),
125 loizides 1.25 TClass::GetClass(typeid(*hltLabels_))->GetName(),&hltLabels_,32000,0);
126 loizides 1.16 tws.SetAutoFill(Names::gkHltTreeName,0);
127     hltTree_=tws.GetTree(Names::gkHltTreeName);
128    
129 loizides 1.3 // store pointer to tree writer
130     tws_ = &tws;
131 loizides 1.1 }
132    
133 loizides 1.8 //--------------------------------------------------------------------------------------------------
134 loizides 1.5 void FillerMetaInfos::FillDataBlock(const edm::Event &event,
135 loizides 1.1 const edm::EventSetup &setup)
136     {
137 loizides 1.3 // Fill our data structures.
138    
139     // clear map if a new file was opened
140     if (tws_->GetFileNumber()!=fileNum_) {
141     runmap_.clear();
142     fileNum_ = tws_->GetFileNumber();
143     runEntries_ = 0;
144 loizides 1.16 l1Entries_ = -1;
145     hltEntries_ = -1;
146 loizides 1.3 }
147    
148     UInt_t runnum = event.id().run();
149    
150     // store look ahead information
151     if (runEntries_>0) {
152     evtLAHeader_->SetRunNum(runnum);
153     laTree_->Fill();
154 loizides 1.21 if(0) { // check for memory leak in usage of ROOT objects
155 loizides 1.14 if(laTree_->GetEntries() % 100==0)
156     gObjectTable->Print();
157     }
158 loizides 1.3 }
159    
160 loizides 1.2 // fill event header
161     eventHeader_->SetEvtNum(event.id().event());
162     eventHeader_->SetLumiSec(event.luminosityBlock());
163     eventHeader_->SetRunNum(runnum);
164    
165 loizides 1.3 // look-up if entry is in map
166 loizides 1.2 map<UInt_t,Int_t>::iterator riter = runmap_.find(runnum);
167     if (riter != runmap_.end()) {
168     Int_t runentry = riter->second;
169     eventHeader_->SetRunEntry(runentry);
170 loizides 1.16 FillHltTrig(event,setup);
171 loizides 1.2 return;
172     }
173 loizides 1.1
174 loizides 1.2 // fill new run info
175     Int_t runentry = runEntries_;
176 loizides 1.16 ++runEntries_;
177 loizides 1.2 eventHeader_->SetRunEntry(runentry);
178     runmap_.insert(pair<UInt_t,Int_t>(runnum,runentry));
179 loizides 1.16 runInfo_->SetRunNum(runnum);
180    
181     Int_t l1entry = l1Entries_;
182     //FillL1Info(event,setup);
183     //FillL1Trigger(event,setup);
184 loizides 1.21 if (l1entry < l1Entries_) {
185     l1Tree_->Fill();
186     runInfo_->SetL1Entry(l1entry);
187     } else {
188     runInfo_->SetL1Entry(l1entry-1);
189     }
190 loizides 1.2
191 loizides 1.16 Int_t hltentry = hltEntries_;
192 loizides 1.14 FillHltInfo(event,setup);
193 loizides 1.16 FillHltTrig(event,setup);
194 loizides 1.21 if (hltentry < hltEntries_) {
195     runInfo_->SetHltEntry(hltentry);
196 loizides 1.16 hltTree_->Fill();
197 loizides 1.21 } else {
198     runInfo_->SetHltEntry(hltentry-1);
199     }
200 loizides 1.14
201 loizides 1.2 runTree_->Fill();
202 loizides 1.1 }
203 loizides 1.5
204 loizides 1.8 //--------------------------------------------------------------------------------------------------
205 loizides 1.5 void FillerMetaInfos::FillHltInfo(const edm::Event &event,
206     const edm::EventSetup &setup)
207     {
208 loizides 1.16 // Fill HLT trigger table if it changed.
209 loizides 1.5
210 loizides 1.14 if (!hltActive_) return;
211    
212 loizides 1.21 // check if we can access the hlt config information
213 loizides 1.16 if (!hltConfig_.init(hltProcName_)) {
214 loizides 1.21 edm::LogError("FillerMetaInfos") << "Can not access hlt config using "
215     << hltProcName_ << std::endl;
216 loizides 1.14 return;
217 loizides 1.16 }
218 loizides 1.14
219 loizides 1.21 // check size of menu... < 256
220 loizides 1.16 if (hltConfig_.size()>hltBits_->Size()) {
221 loizides 1.21 edm::LogError("FillerMetaInfos") << "HLT config contains too many paths "
222 loizides 1.25 << hltConfig_.size() << " > "
223     << hltBits_->Size() << std::endl;
224 loizides 1.16 return;
225     }
226 loizides 1.14
227 loizides 1.22 vector<string> *trigtable = new vector<string>;
228 loizides 1.16 map<string,Short_t> *tabmap = new map<string,Short_t>;
229 loizides 1.22 vector<string> *labels = new vector<string>;
230 loizides 1.16 map<string,Short_t> *labmap = new map<string,Short_t>;
231    
232 loizides 1.21 // loop over hlt paths
233 loizides 1.16 for(UInt_t i=0;i<hltConfig_.size();++i) {
234    
235     tabmap->insert(pair<string,Short_t>(hltConfig_.triggerName(i),i));
236 loizides 1.22 trigtable->push_back(hltConfig_.triggerName(i));
237 loizides 1.16 const vector<string> &mLabels(hltConfig_.moduleLabels(i));
238     for (UInt_t j=0; j<mLabels.size(); ++j) {
239 loizides 1.19
240 loizides 1.16 const string& label(mLabels[j]);
241    
242 loizides 1.19 // remove characters which do not work: negation is included as !ModuleName
243     // (this might require a change in the underlying CMSSW HLT fwk, so watch out).
244     const char *lptr = &label[0];
245     if (label[0] == '!')
246     ++lptr;
247    
248     map<string,Short_t>::iterator riter = labmap->find(lptr);
249 loizides 1.16 if (riter == labmap->end()) {
250 loizides 1.22 labmap->insert(pair<string,Short_t>(lptr,labels->size()));
251     labels->push_back(lptr);
252 loizides 1.16 }
253 loizides 1.19
254     const string type(hltConfig_.moduleType(lptr));
255 loizides 1.16 riter = labmap->find(type);
256     if (riter == labmap->end()) {
257 loizides 1.22 labmap->insert(pair<string,Short_t>(type,labels->size()));
258     labels->push_back(type);
259 loizides 1.14 }
260     }
261     }
262    
263 loizides 1.22 if (hltTable_->size()>=0) {
264 loizides 1.21 // check if existing table contains all necessary paths:
265     // if so keep it, otherwise store the new one
266    
267 loizides 1.22 if ((hltTable_->size()>=trigtable->size()) &&
268     (hltLabels_->size()>=labels->size())) {
269 loizides 1.21
270     bool newEntryFound = false;
271 loizides 1.22 for (UInt_t i=0; i<trigtable->size(); ++i) {
272     map<string,Short_t>::iterator riter = tabmap->find(trigtable->at(i));
273 loizides 1.21 if (riter == tabmap->end()) {
274     newEntryFound = true;
275     break;
276     }
277     }
278     if (!newEntryFound) {
279 loizides 1.22 for (UInt_t i=0; i<labels->size(); ++i) {
280     map<string,Short_t>::iterator riter = labmap->find(labels->at(i));
281 loizides 1.21 if (riter == labmap->end()) {
282     newEntryFound = true;
283     break;
284     }
285     }
286     }
287    
288     if (!newEntryFound) {
289     if (verbose_>1)
290     cout << "FillHltInfo: Kept previous HLT information" << endl;
291    
292     delete trigtable;
293     delete labels;
294     delete labmap;
295     delete tabmap;
296     return;
297     }
298 loizides 1.16 }
299 loizides 1.20 }
300 loizides 1.16
301 loizides 1.21 // new hlt entry
302 loizides 1.16 delete hltTable_;
303     delete hltLabels_;
304     delete hltTabMap_;
305     delete hltLabMap_;
306     hltTable_ = trigtable;
307     hltLabels_ = labels;
308     hltTabMap_ = tabmap;
309     hltLabMap_ = labmap;
310     hltEntries_++;
311     }
312    
313     //--------------------------------------------------------------------------------------------------
314     void FillerMetaInfos::FillHltTrig(const edm::Event &event,
315     const edm::EventSetup &setup)
316     {
317     // Fill HLT trigger objects along triggered paths.
318    
319     if (!hltActive_) return;
320    
321     // reset trigger objects
322     hltObjs_->Reset();
323     hltRels_->Reset();
324 loizides 1.14
325 loizides 1.16 // get HLT trigger information
326 loizides 1.14 Handle<TriggerResults> triggerResultsHLT;
327 loizides 1.16 GetProduct(hltResName_, triggerResultsHLT, event);
328 loizides 1.5
329 loizides 1.16 // get HLT trigger object information
330     Handle<trigger::TriggerEvent> triggerEventHLT;
331     GetProduct(hltEvtName_, triggerEventHLT, event);
332    
333     if (verify_)
334     assert(triggerResultsHLT->size()==hltConfig_.size());
335    
336 loizides 1.25 // new bitmask
337     BitMask256 mask;
338 loizides 1.16
339     //map between EDM and OAK trigger object indices
340     std::map<Int_t,Int_t> objmap;
341    
342     // loop over trigger paths
343     const UInt_t N = hltConfig_.size();
344     for(UInt_t i=0;i<N;++i) {
345    
346     const string &name(hltConfig_.triggerName(i));
347     const UInt_t tind(hltConfig_.triggerIndex(name.c_str()));
348     if (verify_)
349     assert(tind==i);
350    
351     if (verbose_>0)
352     cout << "Trigger: path " << name << " [" << i << "]" << endl;
353    
354     if (!triggerResultsHLT->accept(tind))
355     continue;
356    
357     // get our trigger bit
358     map<string,Short_t>::iterator riter = hltTabMap_->find(name);
359     if (riter == hltTabMap_->end()) {
360     PrintErrorAndExit(Form("Trigger %s not found\n", name.c_str()));
361     }
362    
363     UInt_t mytind = riter->second;
364     if (verify_) {
365     if (mytind!=tind)
366     PrintErrorAndExit(Form("Trigger for %s index does not match: %ud %ud\n",
367     name.c_str(), mytind, tind));
368     }
369    
370     // set trigger bit
371 loizides 1.25 mask.SetBit(mytind);
372 sixie 1.13
373 loizides 1.16 // modules on this trigger path
374     const UInt_t M(hltConfig_.size(tind));
375     const UInt_t mind(triggerResultsHLT->index(tind));
376     assert (mind<M);
377    
378     const vector<string> &mLabels(hltConfig_.moduleLabels(tind));
379    
380     if (verbose_>1)
381     cout << " Last active module - label/type: "
382     << mLabels[mind] << "/" << hltConfig_.moduleType(mLabels[mind])
383     << " [" << mind << " out of 0-" << (M-1) << " on this path]" << endl;
384    
385     // loop over modules on path
386     for (UInt_t j=0; j<=mind; ++j) {
387     const string &mLabel(mLabels[j]);
388    
389     // check whether the module is packed up in TriggerEvent product
390     const UInt_t find(triggerEventHLT->filterIndex(InputTag(mLabel,"",hltProcName_)));
391     if (find>=triggerEventHLT->sizeFilters())
392     continue;
393    
394     const string mType(hltConfig_.moduleType(mLabel));
395     if (verbose_>1)
396     cout << " 'L3' filter in slot " << j << " - label/type "
397     << mLabel << "/" << mType << endl;
398    
399     // find index for module label/type name
400     Short_t modind = -1;
401     map<string,Short_t>::iterator riter = hltLabMap_->find(mLabel);
402     if (riter != hltLabMap_->end()) {
403     modind = riter->second;
404     }
405 loizides 1.17 assert(modind!=-1);
406    
407     Short_t filind = -1;
408 loizides 1.16 riter = hltLabMap_->find(mType);
409     if (riter != hltLabMap_->end()) {
410 loizides 1.17 filind = riter->second;
411 loizides 1.16 }
412 loizides 1.17 assert(filind!=-1);
413 loizides 1.14
414 loizides 1.16 // find trigger objects
415     const trigger::Vids &vids(triggerEventHLT->filterIds(find));
416     const trigger::Keys &keys(triggerEventHLT->filterKeys(find));
417     const trigger::size_type nVids(vids.size());
418     const trigger::size_type nKeys(keys.size());
419     assert(nVids==nKeys);
420    
421     if (verbose_>2)
422     cout << " " << nVids << " accepted 'L3' objects found: " << endl;
423    
424     // loop over trigger objects
425     const trigger::TriggerObjectCollection &toc(triggerEventHLT->getObjects());
426     for (trigger::size_type k=0; k<nVids; ++k) {
427     Int_t tocind = keys[k];
428    
429     // get trigger object
430     const trigger::TriggerObject &tobj(toc[tocind]);
431     if (verbose_>2)
432     cout << " " << k << " " << vids[k] << "/" << keys[k] << ": "
433     << tobj.id() << " " << tobj.pt() << " " << tobj.eta()
434     << " " << tobj.phi() << " " << tobj.mass() << endl;
435    
436     // look-up if entry is in map
437     Int_t objind = -1;
438     map<Int_t,Int_t>::iterator riter = objmap.find(tocind);
439     if (riter == objmap.end()) { // add new trigger object
440     objind = hltObjs_->Entries();
441     objmap.insert(pair<Int_t,Int_t>(tocind,objind));
442     TriggerObjectBase *trigObj = hltObjs_->Allocate();
443 loizides 1.17 new (trigObj) TriggerObjectBase(tobj.id(),tobj.pt(),tobj.eta(),tobj.phi(),tobj.mass());
444 loizides 1.16 } else { // use existing trigger object
445     objind = riter->second;
446     }
447 loizides 1.14
448 loizides 1.16 TriggerObjectRel *trigRel = hltRels_->Allocate();
449 loizides 1.17 new (trigRel) TriggerObjectRel(mytind,vids[k],objind,modind,filind);
450 loizides 1.16 }
451     }
452     }
453 loizides 1.25
454     hltBits_->SetBits(mask);
455 loizides 1.16 hltObjs_->Trim();
456     hltRels_->Trim();
457 loizides 1.5 }