ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/VHbbAnalysis/VHbbDataFormats/interface/TriggerReader.h
Revision: 1.3
Committed: Fri Sep 2 19:07:27 2011 UTC (13 years, 8 months ago) by arizzi
Content type: text/plain
Branch: MAIN
Changes since 1.2: +9 -6 lines
Log Message: 
 fix cacheRun to be per path

File Contents

# User Rev Content
1 arizzi 1.1 #ifndef TRIGGERREADER_H
2     #define TRIGGERREADER_H
3     #include "DataFormats/FWLite/interface/Event.h"
4     #include "DataFormats/FWLite/interface/Handle.h"
5     #include "DataFormats/Common/interface/TriggerResults.h"
6     #include "FWCore/Common/interface/TriggerNames.h"
7     #include <string>
8     #include <map>
9     #include <TString.h>
10     #include <TRegexp.h>
11     #include <regex.h>
12     #include <stdio.h>
13    
14     class TriggerReader {
15     public:
16 arizzi 1.3 TriggerReader(bool passAllEvents=false) : passAll(passAllEvents) {}
17 arizzi 1.1
18     void setEvent( fwlite::Event * e) { ev=e;}
19    
20     bool accept(const std::string & triggername)
21     {
22 arizzi 1.2 if(passAll) return true;
23 arizzi 1.1 fwlite::Handle<edm::TriggerResults> hTriggerResults;
24     hTriggerResults.getByLabel(*ev,"TriggerResults","","HLT");
25    
26    
27     regex_t regex;
28     int reti;
29    
30    
31     reti = regcomp(&regex, triggername.c_str(), 0);
32     // reti = regcomp(&regex, "HLT_Ele32_CaloIdVT_CaloIsoT_TrkIdT_TrkIsoT_v.*", 0);
33     if( reti ){ std::cerr << "Could not compile regex" << std::endl;}
34    
35    
36     std::map<std::string,size_t>::iterator nit;
37 arizzi 1.3 if(ev->getRun().run() != cacheRun[triggername] || nameMap.find(triggername) == nameMap.end())
38 arizzi 1.1 {
39 arizzi 1.3 std::cout << "new run" << ev->getRun().run() << std::endl;
40     cacheRun[triggername]=ev->getRun().run();
41 arizzi 1.1 edm::TriggerNames const& triggerNames = ev->triggerNames(*hTriggerResults);
42     std::string oldiname="whatever";
43     nit=nameMap.find(triggername);
44     if(nit!=nameMap.end() && nit->second < triggerNames.size())
45     {
46     oldiname=triggerNames.triggerName(nit->second);
47     }
48     // std::cout << "whatever3" << std::endl;
49     // std::cout << "reg1 " << regexec(&regex, oldiname.c_str(), 0, NULL, 0) << std::endl;
50     if(nit==nameMap.end() || ( regexec(&regex, oldiname.c_str(), 0, NULL, 0) != 0 )) //; ! oldiname.Contains( TRegexp(triggername)) ) // changed, search it
51     {
52     std::cout << "searching " << triggername << std::endl;
53     for (unsigned i = 0; i < triggerNames.size(); ++i) {
54     std::string iname(triggerNames.triggerName(i));
55    
56     if( regexec(&regex, iname.c_str(), 0, NULL, 0) == 0 ) //iname.Contains(TRegexp(triggername)) )
57     {
58     nameMap[triggername]= i;
59     std::cout << "FOUND: " << triggerNames.triggerName(i) << " is bit " << i << std::endl;
60     }
61     }
62    
63     }
64     }
65     nit=nameMap.find(triggername);
66     if(nit==nameMap.end())
67     {
68     std::cout << "ERROR: trigger name not found" << std::endl;
69     edm::TriggerNames const& triggerNames = ev->triggerNames(*hTriggerResults);
70 arizzi 1.3 // for (unsigned i = 0; i < triggerNames.size(); ++i) std::cout << triggerNames.triggerName(i) << " is bit " << i << "looking for: "<< triggername << std::endl;
71     nameMap[triggername]=100000000; // meaning not found in this run
72 arizzi 1.1
73     return false;
74     }
75 arizzi 1.3 if(nit->second==100000000) return false;
76 arizzi 1.1 return hTriggerResults->accept(nit->second);
77     }
78    
79     private:
80     std::map<std::string,size_t> nameMap;
81 arizzi 1.3 std::map<std::string,size_t> cacheRun;
82     // unsigned int cacheRun;
83 arizzi 1.1 fwlite::Event * ev;
84 arizzi 1.2 bool passAll;
85 arizzi 1.1
86     };
87    
88     #endif