JetAnalysis/interface/HiInclusiveJetAnalyzer.h

#ifndef MNguyen_HiInclusiveJetAnalyzer_inclusiveJetAnalyzer_
#define MNguyen_HiInclusiveJetAnalyzer_inclusiveJetAnalyzer_

// system include files
#include <memory>
#include <string>
#include <iostream>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "TFile.h"
#include "TTree.h"
#include "TH1.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "FWCore/ServiceRegistry/interface/Service.h"

#include "DataFormats/HeavyIonEvent/interface/CentralityProvider.h"

#include "DataFormats/PatCandidates/interface/Jet.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
//

/**\class HiInclusiveJetAnalyzer 

\author Matt Nguyen
\date   November 2010
*/


class HiInclusiveJetAnalyzer : public edm::EDAnalyzer {
 public:

  explicit HiInclusiveJetAnalyzer(const edm::ParameterSet&);

  ~HiInclusiveJetAnalyzer();
  
  virtual void analyze(const edm::Event&, const edm::EventSetup&);

  virtual void beginRun(const edm::Run & r, const edm::EventSetup & c);

  virtual void beginJob();

  void fillL1Bits(const edm::Event &iEvent);

  void fillHLTBits(const edm::Event &iEvent);

  template <typename TYPE>
    void                          getProduct(const std::string name, edm::Handle<TYPE> &prod,
                                             const edm::Event &event) const;    
  template <typename TYPE>
    bool                          getProductSafe(const std::string name, edm::Handle<TYPE> &prod,
                                                 const edm::Event &event) const;
  

 private:
  
  int getPFJetMuon(const pat::Jet& pfJet, const reco::PFCandidateCollection *pfCandidateColl);

  double getPtRel(const reco::PFCandidate lep, const pat::Jet& jet );

  edm::InputTag   jetTag_, vtxTag_, genjetTag_, eventInfoTag_, L1gtReadout_, pfCandidateLabel_; 


  /// verbose ?
  bool   verbose_;

  bool useCentrality_;
  bool useVtx_;
  bool useJEC_;
  bool usePat_;
  bool isMC_;

  double genPtMin_;
  bool doLifeTimeTagging_;
  bool doLifeTimeTaggingExtras_;

  TTree *t;
  edm::Service<TFileService> fs1;

  CentralityProvider * centrality_;


  std::string                   hltResName_;         //HLT trigger results name
  std::vector<std::string>      hltProcNames_;       //HLT process name(s)
  std::vector<std::string>      hltTrgNames_;        //HLT trigger name(s)

  std::vector<int>              hltTrgBits_;         //HLT trigger bit(s)
  std::vector<bool>             hltTrgDeci_;         //HLT trigger descision(s)
  std::vector<std::string>      hltTrgUsedNames_;    //HLT used trigger name(s)
  std::string                   hltUsedResName_;     //used HLT trigger results name


  static const int MAXJETS = 500;
  static const int MAXTRACKS = 5000;
  static const int MAXHLTBITS = 5000;


  struct JRA{
    
    int nref;
    int run;
    int evt;
    int lumi;
    int bin;
    float vx, vy, vz;
    float b;
    float hf;

    float rawpt[MAXJETS];
    float jtpt[MAXJETS];
    float jteta[MAXJETS];
    float jtphi[MAXJETS];
    float jty[MAXJETS];
    float jtpu[MAXJETS];

    float discr_csvMva[MAXJETS];
    float discr_csvSimple[MAXJETS];
    float discr_muByIp3[MAXJETS];
    float discr_muByPt[MAXJETS];
    float discr_prob[MAXJETS];
    float discr_probb[MAXJETS];
    float discr_tcHighEff[MAXJETS];    
    float discr_tcHighPur[MAXJETS];

    int nsvtx[MAXJETS];    
    int svtxntrk[MAXJETS]; 
    float svtxdl[MAXJETS];   
    float svtxdls[MAXJETS];  
    float svtxm[MAXJETS];    
    float svtxpt[MAXJETS];   

    int nIPtrk[MAXJETS];
    int nselIPtrk[MAXJETS];
    
    int ipJetIndex[MAXTRACKS];
    float ipPt[MAXTRACKS];
    float ipProb0[MAXTRACKS];
    float ipProb1[MAXTRACKS];
    float ip2d[MAXTRACKS];
    float ip2dSig[MAXTRACKS];
    float ip3d[MAXTRACKS];
    float ip3dSig[MAXTRACKS];
    float ipDist2Jet[MAXTRACKS];
    float ipDist2JetSig[MAXTRACKS];
    float ipClosest2Jet[MAXTRACKS];

    float mue[MAXJETS];     
    float mupt[MAXJETS];    
    float mueta[MAXJETS];   
    float muphi[MAXJETS];   
    float mudr[MAXJETS];    
    float muptrel[MAXJETS]; 
    int muchg[MAXJETS];   
    
    float refpt[MAXJETS];
    float refeta[MAXJETS];
    float refphi[MAXJETS];
    float refy[MAXJETS];
    float refdphijt[MAXJETS];
    float refdrjt[MAXJETS];
    float refparton_pt[MAXJETS];
    int refparton_flavor[MAXJETS];
    int refparton_flavorForB[MAXJETS];

    float pthat;
    int beamId1, beamId2;
    int ngen;
    int genmatchindex[MAXJETS];
    float genpt[MAXJETS];
    float geneta[MAXJETS];
    float genphi[MAXJETS];
    float geny[MAXJETS];
    float gendphijt[MAXJETS];
    float gendrjt[MAXJETS];

    // hlt
    int nHLTBit;
    bool hltBit[MAXHLTBITS];

    // l1
    int nL1TBit;
    bool l1TBit[MAXHLTBITS];
    int nL1ABit;
    bool l1ABit[MAXHLTBITS];


  };

  JRA jets_;

};

#endif
Revision:	1.5
Committed:	Sat Apr 28 12:18:43 2012 UTC (13 years ago) by mnguyen
Content type:	text/plain
Branch:	MAIN
Changes since 1.4:	+21 -6 lines
Log Message:	Add extra b-tagging info
#	User	Rev	Content
1	yilmaz	1.1	#ifndef MNguyen_HiInclusiveJetAnalyzer_inclusiveJetAnalyzer_
2			#define MNguyen_HiInclusiveJetAnalyzer_inclusiveJetAnalyzer_
3
4			// system include files
5			#include <memory>
6			#include <string>
7			#include <iostream>
8
9			// user include files
10			#include "FWCore/Framework/interface/Frameworkfwd.h"
11			#include "FWCore/Framework/interface/EDAnalyzer.h"
12			#include "FWCore/ParameterSet/interface/ParameterSet.h"
13			#include "FWCore/Framework/interface/Event.h"
14			#include "FWCore/Framework/interface/MakerMacros.h"
15
16			#include "TFile.h"
17			#include "TTree.h"
18			#include "TH1.h"
19			#include "CommonTools/UtilAlgos/interface/TFileService.h"
20			#include "FWCore/ServiceRegistry/interface/Service.h"
21
22			#include "DataFormats/HeavyIonEvent/interface/CentralityProvider.h"
23
24	mnguyen	1.3	#include "DataFormats/PatCandidates/interface/Jet.h"
25			#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
26	yilmaz	1.1	//
27
28			/**\class HiInclusiveJetAnalyzer
29
30			\author Matt Nguyen
31			\date November 2010
32			*/
33
34
35
36
37			class HiInclusiveJetAnalyzer : public edm::EDAnalyzer {
38			public:
39
40			explicit HiInclusiveJetAnalyzer(const edm::ParameterSet&);
41
42			~HiInclusiveJetAnalyzer();
43
44			virtual void analyze(const edm::Event&, const edm::EventSetup&);
45
46			virtual void beginRun(const edm::Run & r, const edm::EventSetup & c);
47
48			virtual void beginJob();
49
50			void fillL1Bits(const edm::Event &iEvent);
51
52			void fillHLTBits(const edm::Event &iEvent);
53
54			template <typename TYPE>
55			void getProduct(const std::string name, edm::Handle<TYPE> &prod,
56			const edm::Event &event) const;
57			template <typename TYPE>
58			bool getProductSafe(const std::string name, edm::Handle<TYPE> &prod,
59			const edm::Event &event) const;
60
61
62			private:
63
64	mnguyen	1.3	int getPFJetMuon(const pat::Jet& pfJet, const reco::PFCandidateCollection *pfCandidateColl);
65	yilmaz	1.1
66	mnguyen	1.3	double getPtRel(const reco::PFCandidate lep, const pat::Jet& jet );
67	yilmaz	1.1
68	mnguyen	1.3	edm::InputTag jetTag_, vtxTag_, genjetTag_, eventInfoTag_, L1gtReadout_, pfCandidateLabel_;
69	yilmaz	1.1
70
71			/// verbose ?
72			bool verbose_;
73
74			bool useCentrality_;
75			bool useVtx_;
76			bool useJEC_;
77	yilmaz	1.2	bool usePat_;
78	yilmaz	1.1	bool isMC_;
79
80	yilmaz	1.4	double genPtMin_;
81	mnguyen	1.3	bool doLifeTimeTagging_;
82	mnguyen	1.5	bool doLifeTimeTaggingExtras_;
83	yilmaz	1.1
84			TTree *t;
85			edm::Service<TFileService> fs1;
86
87			CentralityProvider * centrality_;
88
89
90
91			std::string hltResName_; //HLT trigger results name
92			std::vector<std::string> hltProcNames_; //HLT process name(s)
93			std::vector<std::string> hltTrgNames_; //HLT trigger name(s)
94
95			std::vector<int> hltTrgBits_; //HLT trigger bit(s)
96			std::vector<bool> hltTrgDeci_; //HLT trigger descision(s)
97			std::vector<std::string> hltTrgUsedNames_; //HLT used trigger name(s)
98			std::string hltUsedResName_; //used HLT trigger results name
99
100
101
102	mnguyen	1.5	static const int MAXJETS = 500;
103			static const int MAXTRACKS = 5000;
104			static const int MAXHLTBITS = 5000;
105	yilmaz	1.1
106
107			struct JRA{
108
109			int nref;
110			int run;
111			int evt;
112			int lumi;
113			int bin;
114			float vx, vy, vz;
115			float b;
116			float hf;
117
118			float rawpt[MAXJETS];
119			float jtpt[MAXJETS];
120			float jteta[MAXJETS];
121			float jtphi[MAXJETS];
122			float jty[MAXJETS];
123	yilmaz	1.2	float jtpu[MAXJETS];
124
125	mnguyen	1.3	float discr_csvMva[MAXJETS];
126			float discr_csvSimple[MAXJETS];
127			float discr_muByIp3[MAXJETS];
128			float discr_muByPt[MAXJETS];
129			float discr_prob[MAXJETS];
130			float discr_probb[MAXJETS];
131			float discr_tcHighEff[MAXJETS];
132			float discr_tcHighPur[MAXJETS];
133
134	mnguyen	1.5	int nsvtx[MAXJETS];
135			int svtxntrk[MAXJETS];
136	mnguyen	1.3	float svtxdl[MAXJETS];
137			float svtxdls[MAXJETS];
138			float svtxm[MAXJETS];
139			float svtxpt[MAXJETS];
140
141	mnguyen	1.5	int nIPtrk[MAXJETS];
142			int nselIPtrk[MAXJETS];
143
144			int ipJetIndex[MAXTRACKS];
145			float ipPt[MAXTRACKS];
146			float ipProb0[MAXTRACKS];
147			float ipProb1[MAXTRACKS];
148			float ip2d[MAXTRACKS];
149			float ip2dSig[MAXTRACKS];
150			float ip3d[MAXTRACKS];
151			float ip3dSig[MAXTRACKS];
152			float ipDist2Jet[MAXTRACKS];
153			float ipDist2JetSig[MAXTRACKS];
154			float ipClosest2Jet[MAXTRACKS];
155	mnguyen	1.3
156			float mue[MAXJETS];
157			float mupt[MAXJETS];
158			float mueta[MAXJETS];
159			float muphi[MAXJETS];
160			float mudr[MAXJETS];
161			float muptrel[MAXJETS];
162			int muchg[MAXJETS];
163
164	yilmaz	1.1	float refpt[MAXJETS];
165			float refeta[MAXJETS];
166			float refphi[MAXJETS];
167			float refy[MAXJETS];
168			float refdphijt[MAXJETS];
169			float refdrjt[MAXJETS];
170			float refparton_pt[MAXJETS];
171	mnguyen	1.3	int refparton_flavor[MAXJETS];
172			int refparton_flavorForB[MAXJETS];
173	yilmaz	1.1
174			float pthat;
175	mnguyen	1.5	int beamId1, beamId2;
176	yilmaz	1.1	int ngen;
177			int genmatchindex[MAXJETS];
178			float genpt[MAXJETS];
179			float geneta[MAXJETS];
180			float genphi[MAXJETS];
181			float geny[MAXJETS];
182			float gendphijt[MAXJETS];
183			float gendrjt[MAXJETS];
184
185			// hlt
186			int nHLTBit;
187			bool hltBit[MAXHLTBITS];
188
189			// l1
190			int nL1TBit;
191			bool l1TBit[MAXHLTBITS];
192			int nL1ABit;
193			bool l1ABit[MAXHLTBITS];
194
195	mnguyen	1.3
196	yilmaz	1.1	};
197
198			JRA jets_;
199
200			};
201
202			#endif