ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/OSUT3Analysis/AnaTools/plugins/OSUAnalysis.h
(Generate patch)

Comparing UserCode/OSUT3Analysis/AnaTools/plugins/OSUAnalysis.h (file contents):
Revision 1.30 by ahart, Sun May 5 23:40:58 2013 UTC vs.
Revision 1.42 by lantonel, Wed Jun 19 22:42:22 2013 UTC

# Line 15 | Line 15
15   #include "TH1D.h"
16   #include "TH2D.h"
17   #include "TLorentzVector.h"
18 + #include "TTree.h"
19  
20 < #include "FWCore/Framework/interface/EDAnalyzer.h"
20 > #include "FWCore/Framework/interface/EDProducer.h"
21   #include "FWCore/Framework/interface/Event.h"
22   #include "FWCore/Framework/interface/EventSetup.h"
23   #include "FWCore/ParameterSet/interface/ParameterSet.h"
# Line 48 | Line 49
49   #include "OSUT3Analysis/AnaTools/interface/CutFlow.h"
50   #include "OSUT3Analysis/AnaTools/interface/PUWeight.h"
51   #include "OSUT3Analysis/AnaTools/interface/SFWeight.h"
52 <
52 > #include "OSUT3Analysis/AnaTools/interface/BtagSFWeight.h"
53   #include "OSUT3Analysis/AnaTools/interface/StopCTauWeight.h"
54  
55   using namespace std;
56  
57 < class OSUAnalysis : public edm::EDAnalyzer
57 > class OSUAnalysis : public edm::EDProducer
58    {
59      public:
60        OSUAnalysis (const edm::ParameterSet &);
61        ~OSUAnalysis ();
62  
63 <      void analyze (const edm::Event &, const edm::EventSetup &);
63 >      void produce (edm::Event &, const edm::EventSetup &);
64        bool evaluateComparison (double, string, double);
65 <      bool evaluateTriggers   (vector<string>,const BNtriggerCollection*);
65 >      bool evaluateComparison (string, string, string);
66 >      bool evaluateTriggers   (vector<string>, vector<string>, const BNtriggerCollection*);
67        double applyFunction(string, double);
68  
69 <      double valueLookup (const BNjet* object, string variable, string function = "");
70 <      double valueLookup (const BNmuon* object, string variable, string function = "");
71 <      double valueLookup (const BNmuon* object1, const BNmuon* object2, string variable, string function = "");
72 <      double valueLookup (const BNelectron* object, string variable, string function = "");
73 <      double valueLookup (const BNelectron* object1, const BNelectron* object2, string variable, string function = "");
74 <      double valueLookup (const BNelectron* object1, const BNmuon* object2, string variable, string function = "");
75 <      double valueLookup (const BNelectron* object1, const BNjet* object2, string variable, string function = "");
76 <      double valueLookup (const BNelectron* object1, const BNtrack* object2, string variable, string function = "");
77 <      double valueLookup (const BNmuon* object1, const BNjet* object2, string variable, string function = "");
78 <      double valueLookup (const BNmuon* object1, const BNtrack* object2, string variable, string function = "");
79 <      double valueLookup (const BNmuon* object1, const BNtau* object2, string variable, string function = "");
80 <      double valueLookup (const BNtau* object1, const BNtau* object2, string variable, string function = "");
81 <      double valueLookup (const BNtau* object1, const BNtrack* object2, string variable, string function);
82 <      double valueLookup (const BNjet* object1, const BNjet* object2, string variable, string function = "");
83 <      double valueLookup (const BNevent* object, string variable, string function = "");
84 <      double valueLookup (const BNtau* object, string variable, string function = "");
85 <      double valueLookup (const BNmet* object, string variable, string function = "");
86 <      double valueLookup (const BNtrack* object, string variable, string function = "");
87 <      double valueLookup (const BNtrack* object1, const BNevent* object2, string variable, string function = "");
88 <      double valueLookup (const BNgenjet* object, string variable, string function = "");
89 <      double valueLookup (const BNmcparticle* object, string variable, string function = "");
90 <      double valueLookup (const BNprimaryvertex* object, string variable, string function = "");
91 <      double valueLookup (const BNbxlumi* object, string variable, string function = "");
92 <      double valueLookup (const BNphoton* object, string variable, string function = "");
93 <      double valueLookup (const BNsupercluster* object, string variable, string function = "");
94 <      double valueLookup (const BNstop* object, string variable, string function = "");
69 >      double valueLookup (const BNjet* object, string variable, string function, string &stringValue);
70 >      double valueLookup (const BNmuon* object, string variable, string function, string &stringValue);
71 >      double valueLookup (const BNmuon* object1, const BNmuon* object2, string variable, string function, string &stringValue);
72 >      double valueLookup (const BNelectron* object, string variable, string function, string &stringValue);
73 >      double valueLookup (const BNelectron* object1, const BNelectron* object2, string variable, string function, string &stringValue);
74 >      double valueLookup (const BNelectron* object1, const BNmuon* object2, string variable, string function, string &stringValue);
75 >      double valueLookup (const BNelectron* object1, const BNjet* object2, string variable, string function, string &stringValue);
76 >      double valueLookup (const BNelectron* object1, const BNtrack* object2, string variable, string function, string &stringValue);
77 >      double valueLookup (const BNmuon* object1, const BNjet* object2, string variable, string function, string &stringValue);
78 >      double valueLookup (const BNmuon* object1, const BNtrack* object2, string variable, string function, string &stringValue);
79 >      double valueLookup (const BNmuon* object1, const BNtau* object2, string variable, string function, string &stringValue);
80 >      double valueLookup (const BNtau* object1, const BNtau* object2, string variable, string function, string &stringValue);
81 >      double valueLookup (const BNtau* object1, const BNtrack* object2, string variable, string function, string &stringValue);
82 >      double valueLookup (const BNjet* object1, const BNjet* object2, string variable, string function, string &stringValue);
83 >      double valueLookup (const BNevent* object, string variable, string function, string &stringValue);
84 >      double valueLookup (const BNtau* object, string variable, string function, string &stringValue);
85 >      double valueLookup (const BNmet* object, string variable, string function, string &stringValue);
86 >      double valueLookup (const BNtrack* object, string variable, string function, string &stringValue);
87 >      double valueLookup (const BNtrack* object1, const BNevent* object2, string variable, string function, string &stringValue);
88 >      double valueLookup (const BNgenjet* object, string variable, string function, string &stringValue);
89 >      double valueLookup (const BNmcparticle* object, string variable, string function, string &stringValue);
90 >      double valueLookup (const BNprimaryvertex* object, string variable, string function, string &stringValue);
91 >      double valueLookup (const BNbxlumi* object, string variable, string function, string &stringValue);
92 >      double valueLookup (const BNphoton* object, string variable, string function, string &stringValue);
93 >      double valueLookup (const BNsupercluster* object, string variable, string function, string &stringValue);
94 >      double valueLookup (const BNtrigobj* object, string variable, string function, string &stringValue);
95 >      double valueLookup (const BNelectron* object1, const BNtrigobj* object2, string variable, string function, string &stringValue);
96 >      double valueLookup (const BNmuon* object1, const BNtrigobj* object2, string variable, string function, string &stringValue);
97 >      double valueLookup (const BNstop* object, string variable, string function, string &stringValue);
98  
99        int getTrkIsIso (const BNtrack* track1, const BNtrackCollection* trackColl);
100        double getTrkPtTrue (const BNtrack* track1, const BNmcparticleCollection* genPartColl);
# Line 103 | Line 108 | class OSUAnalysis : public edm::EDAnalyz
108        //BNtrigobj
109  
110        vector<string> splitString (string);
111 <      void getTwoObjs(string tempInputCollection, string& obj1, string& obj2);
112 <      string getObjToGet(string obj);  
111 >      void getTwoObjs(string tempInputCollection, string& obj1, string& obj2);
112 >      string getObjToGet(string obj);
113  
114      private:
115  
# Line 112 | Line 117 | class OSUAnalysis : public edm::EDAnalyz
117        //string       holds input collection type
118        //outer vector corresponds to each cut
119        //inner vector corresponds to each object in input collection
120 <      //bool         tells if object passes or fails cuts
121 <      typedef map<string, vector < vector<bool> > > flagMap;
120 >      //pair(bool) first bool counts towards the event passing, second bool determines whether to plot the object        
121 >      typedef map<string, vector<vector<pair<bool,bool> > > > flagMap;
122 >
123 >      typedef vector<pair<bool,bool> > flagPair;
124  
125        //counterMap:
126        //string holds input collection type
# Line 137 | Line 144 | class OSUAnalysis : public edm::EDAnalyz
144        edm::InputTag photons_;
145        edm::InputTag superclusters_;
146        edm::InputTag triggers_;
147 <      std::string puFile_;
148 <      std::string deadEcalFile_;
149 <      std::string muonSFFile_;
150 <      std::string dataPU_;
151 <      std::string electronSFID_;
152 <      std::string muonSF_;
153 <      std::string dataset_;
154 <      std::string datasetType_;
147 >      edm::InputTag trigobjs_;
148 >      string puFile_;
149 >      string deadEcalFile_;
150 >      string muonSFFile_;
151 >      string dataPU_;
152 >      string electronSFID_;
153 >      string muonSF_;
154 >      string dataset_;
155 >      string datasetType_;
156        vector<edm::ParameterSet> channels_;
157        vector<edm::ParameterSet> histogramSets_;
158 +      bool useEDMFormat_;
159 +      vector<edm::ParameterSet> treeBranchSets_;  
160        bool plotAllObjectsInPassingEvents_;
161        bool doPileupReweighting_;
162        bool applyLeptonSF_;
163 <      bool printEventInfo_;  
164 <      bool useTrackCaloRhoCorr_;  // to use the calo-based rho correction for the by-hand calculation of the track isolation energy
163 >      bool applyBtagSF_;
164 >      int  minBtag_;
165 >      int  maxBtag_;
166 >      bool printEventInfo_;
167 >      bool printAllTriggers_;
168 >      bool useTrackCaloRhoCorr_;  // to use the calo-based rho correction for the by-hand calculation of the track isolation energy
169        vector<double> stopCTau_;
170 <
170 >      bool GetPlotsAfterEachCut_;
171 >      
172        struct DeadEcal {
173          double etaEcal;
174          double phiEcal;
175        };
176  
177 <      std::vector<DeadEcal> DeadEcalVec;
177 >      vector<DeadEcal> DeadEcalVec;
178  
179  
180        //Collections
181        edm::Handle<BNtriggerCollection> triggers;
182 +      edm::Handle<BNtrigobjCollection> trigobjs;
183        edm::Handle<BNjetCollection> jets;
184        edm::Handle<BNmuonCollection> muons;
185        edm::Handle<BNelectronCollection> electrons;
# Line 179 | Line 195 | class OSUAnalysis : public edm::EDAnalyz
195        edm::Handle<BNsuperclusterCollection> superclusters;
196        edm::Handle<BNstopCollection> stops;
197  
198 <      edm::Handle<double> rhokt6CaloJetsHandle_;  
198 >      edm::Handle<double> rhokt6CaloJetsHandle_;
199  
200        flagMap cumulativeFlags;
201 +      flagMap vetoFlags;
202  
203 <      vector<map<string, TH1D*> > oneDHists_;
204 <      vector<map<string, TH2D*> > twoDHists_;
203 >      vector<vector<map<string, TH1D*>>> oneDHists_;
204 >      vector<vector<map<string, TH2D*>>> twoDHists_;
205 >      vector<TTree*> BNTrees_;  // one tree per channel  
206 >      map<string, vector<float>> BNTreeBranchVals_;  // data structure to hold the values of the branches to be stored in the BNTrees_; the string is the name of a variable  
207  
208        edm::Service<TFileService> fs_;
209  
# Line 201 | Line 220 | class OSUAnalysis : public edm::EDAnalyz
220          vector<string> inputVariables;
221        } histogram;
222  
223 +      typedef struct {
224 +        string name;
225 +        string inputCollection;
226 +        string inputVariable;
227 +      } BranchSpecs;
228 +
229 +
230        struct cut {
231          string inputCollection;
232          vector<string> functions;
233          vector<string> variables;
234          vector<string> comparativeOperators;// >, <, =, etc.
235          vector<double> cutValues;
236 +        vector<string> cutStringValues;
237          vector<string> logicalOperators;//and, or
238 <        int numSubcuts;
238 >        int    numSubcuts;
239          int    numberRequired;
240          string eventComparativeOperator;
241          string name;
242 +        bool   isVeto;
243        };
244  
245        struct channel {
246          string name;
247          vector<string> triggers;
248 +        vector<string> triggersToVeto;
249          vector<cut> cuts;
250        };
251  
# Line 225 | Line 254 | class OSUAnalysis : public edm::EDAnalyz
254        vector<string>  objectsToPlot;
255        vector<channel> channels;
256        vector<histogram> histograms;
257 +      vector<BranchSpecs> treeBranches_;
258  
259        PUWeight *puWeight_;
260        MuonSFWeight *muonSFWeight_;
261 +      double muonScaleFactor_;
262        ElectronSFWeight *electronSFWeight_;
263 +      double electronScaleFactor_;
264 +      BtagSFWeight *bTagSFWeight_;
265 +      double bTagScaleFactor_;    
266  
267        StopCTauWeight *stopCTauWeight_;
268        double stopCTauScaleFactor_;
269  
270        template <class InputCollection> void setObjectFlags(cut &, uint, flagMap &, flagMap &, InputCollection, string);
271 <      template <class InputCollection1, class InputCollection2> void setObjectFlags(cut &, uint, flagMap &, flagMap &, InputCollection1, InputCollection2, vector<bool>, vector<bool>, string);
271 >      template <class InputCollection1, class InputCollection2> void setObjectFlags(cut &, uint, flagMap &, flagMap &, InputCollection1, InputCollection2, flagPair, flagPair, string);
272 >
273 >      template <class InputCollection> void assignTreeBranch(BranchSpecs brSpecs, InputCollection inputCollection, flagPair flags);  
274 >      template <class InputCollection> void fill1DHistogram(TH1*, histogram, InputCollection, flagPair, double);
275 >      template <class InputCollection1, class InputCollection2> void fill1DHistogram(TH1*, histogram, InputCollection1, InputCollection2, flagPair, flagPair, flagPair, double);
276 >      template <class InputCollection> void fill2DHistogram(TH2*, histogram, InputCollection, flagPair, double);
277 >      template <class InputCollection1, class InputCollection2> void fill2DHistogram(TH2*, histogram, InputCollection1, InputCollection2, flagPair, flagPair, flagPair, double);
278 >      bool getPreviousCumulativeFlags(uint currentCutIndex, flagMap &individualFlags, string obj1Type, uint object1, string flagType);
279 >
280  
239 –      template <class InputCollection> void fill1DHistogram(TH1*, histogram, InputCollection, vector<bool>, double);
240 –      template <class InputCollection1, class InputCollection2> void fill1DHistogram(TH1*, histogram, InputCollection1, InputCollection2, vector<bool>, vector<bool>, vector<bool>, double);
241 –      template <class InputCollection> void fill2DHistogram(TH2*, histogram, InputCollection, vector<bool>, double);
242 –      template <class InputCollection1, class InputCollection2> void fill2DHistogram(TH2*, histogram, InputCollection1, InputCollection2, vector<bool>, vector<bool>, vector<bool>, double);
243 –      bool getPreviousCumulativeFlags(uint currentCutIndex, flagMap &individualFlags, string obj1Type, uint object1);  
281  
282        template <class InputObject> int getGenMatchedParticleIndex(InputObject);
283        int getPdgIdBinValue(int);

Diff Legend

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