MultivariateAnalysis/interface/LJetsTopoVars.h

/***

Adopted from the D0 framework,
used to use event.get("caf_util::MET", met), etc, so needs something like METSelector to run upstream

***/

#ifndef LJETSTOPOVARS
#define LJETSTOPOVARS

#include <string>
#include <vector>
#include "FWCore/Framework/interface/Event.h"
#include "LJMet/MultivariateAnalysis/interface/TMBLorentzVector.h"
//#include "cafe/Event.hpp"
#include "TVectorD.h"


class LJetsTopoVars
{
public:
  LJetsTopoVars(std::string jets, std::string met, std::string lepton, bool isMuon)
    :m_jetBranch(jets),
     m_metBranch(met),
     m_leptonBranch(lepton),
     m_isMuon(isMuon),


    //_electron(NULL),
    //_muon(    NULL),
                        
     _ht(          std::vector<double>(22, 0.) ),
     _htOK(        false),
     _evtTopo(     std::vector<double>( 3, 0.) ),
     _evtTopoOK(   false),
     _kt(          std::vector<double>( 3, 0.) ),
     _ktOK(        false),
     _mt(          std::vector<double>( 2, 0.) ),
     _mtOK(        false)
  {};

    // returns # of removed jets due to insufficient isolation from lepton
    //void setEvent(const edm::Event& event);
    //int setEvent(const edm::Event& event, double min_dr_jet_lepton=-0.01);


    //###################################################################################################
    //##
    //## Initiate LJetsTopoVars using one lepton, one MET and 4 leading jets momentum. This initiation is 
    //## used when one of the leptons is converted to a neutrino.
    //##
    //###################################################################################################

  int setEventMetFixed(TLorentzVector&,TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //4 jets
  int setEventMetFixed(TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //3 jets
  int setEventMetFixed(TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //2 jets
  int setEventMetFixed(TLorentzVector&, TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //1 jet
  int setEventMetFixed(TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //0 jet
  int setEvent(const edm::Event& event, double, double min_dr_jet_lepton=-0.01); //This is for corrected MET

    
  double aplanarity() const;
  double centrality() const;
  double sphericity() const;
  double ht() const;
  double htpluslepton() const;
  double methtpluslepton() const;
  double h() const;
  double ktMinPrime() const;
  double dphiLepMet() const;
  double minDijetMass() const;
  double maxJetEta() const;
  double Et3() const;
  double minDijetDeltaR() const;
  double LeptonJet_DeltaR();           //btw lepton and leading 2 jet, minimum
  double Jet1Jet2_DeltaR();     
  double Jet1Jet2_DeltaPhi();   
  double Jet1Jet2_M();     
  double Jet1Jet2_Pt();   
  double Jet1Jet2W_M();     
  double Jet1Jet2W_Pt();   
                
  double Hz();        //scalar sum of longitudinal energies of first four jets, the muon, and the neutrino
  double HT2();       //scalar sum of transverse energies of the 2nd, 3rd, and 4th jets
  double HT2prime();  //HT2/Hz
  double W_MT();      //W transverse mass
  double W_M();       //W mass
  double W_Pt();       //W 
  double DphiJMET();  //Delta phi btw leading jet and MET

  //muon only (throws exception if not)
  double Muon_DeltaR();           //btw muon and jet, minimum
  double Muon_etHaloScaled();     //TMBMuon::etHalo()/pT
  double Muon_etTrkConeScaled();  //TMBMuon::etTrkCone()/pT

  //electron only (throws exception if not)
  double Electron_iso();    //TMBEMCluster::iso()
  double Electron_lhood();  //TMBEMCluster::Lhood8(); is this right?
                
  // Ht
  double getHt()        {if(!_htOK) calcHt(); return _ht[ 0];} 
  double getHtp()       {if(!_htOK) calcHt(); return _ht[ 1];} 
  double getHtpp()      {if(!_htOK) calcHt(); return _ht[ 2];}
  double getHt2()       {if(!_htOK) calcHt(); return _ht[ 3];}
  double getHt2p()      {if(!_htOK) calcHt(); return _ht[ 4];}
  double getHt2pp()     {if(!_htOK) calcHt(); return _ht[ 5];} 
  double getHt3()       {if(!_htOK) calcHt(); return _ht[ 6];}
  double getHt3p()      {if(!_htOK) calcHt(); return _ht[ 7];} 
  double getHt3pp()     {if(!_htOK) calcHt(); return _ht[ 8];} 
  double getCen()       {if(!_htOK) calcHt(); return _ht[ 9];} 
  double getNJW()       {if(!_htOK) calcHt(); return _ht[10];}
  double getJetEtaMax() {if(!_htOK) calcHt(); return _ht[11];}
  double getMdijetMin() {if(!_htOK) calcHt(); return _ht[12];}
  double getMtjets()    {if(!_htOK) calcHt(); return _ht[13];}
  double getSqrtsT()    {if(!_htOK) calcHt(); return _ht[14];}
  double getMtAurelio() {if(!_htOK) calcHt(); return _ht[15];}
  double getPzOverHT()  {if(!_htOK) calcHt(); return _ht[16];}
  double getMevent()    {if(!_htOK) calcHt(); return _ht[17];}
  double getM123inv()   {if(!_htOK) calcHt(); return _ht[18];}
  double getEta2Sum()   {if(!_htOK) calcHt(); return _ht[19];}
  double getMwRec()     {if(!_htOK) calcHt(); return _ht[20];}
  double getH()         {if(!_htOK) calcHt(); return _ht[21];}

  // event topo
  double getSph()   {if(!_evtTopoOK) calcEvtTopo(); return _evtTopo[0];}
  double getApl()   {if(!_evtTopoOK) calcEvtTopo(); return _evtTopo[1];}
  double getAplMu() {if(!_evtTopoOK) calcEvtTopo(); return _evtTopo[2];}

  // Kt
  double getKtminp()        {if(!_ktOK) calcKt(); return _kt[0];}
  double getKtminpReduced() {if(!_ktOK) calcKt(); return _kt[1];}
  double getDrMinJetJet()   {if(!_ktOK) calcKt(); return _kt[2];}

  // mT
  double getDphiMuMet() {if(!_mtOK) calcMt(); return _mt[0];}
  double getMt()        {if(!_mtOK) calcMt(); return _mt[1];}

  // Momentum tensor eigenvalues
  TVectorD getEigen() {if(!_evtTopoOK) calcEvtTopo(); return eigenval;}

  // Neutrino
  TMBLorentzVector GetNeutrino() {return _neutrino;}

private:
  std::string m_jetBranch, m_metBranch, m_leptonBranch;
  std::vector<TMBLorentzVector> m_jets;
  TMBLorentzVector m_met;
  TMBLorentzVector m_lepton;

  TVectorD eigenval;

  bool m_isMuon;

  /***** removed temporarily            
  const TMBEMCluster* _electron;
  const TMBMuon*      _muon;
  *****/
                
  //std::vector<TheJetClass*>   _jetRAW;
  //std::vector<TLorentzVector> _jet;
  //TLorentzVector              _lepton;
  //TLorentzVector              _neutrino;
  TMBLorentzVector _neutrino;
                
  void calcHt();
  std::vector<double> _ht;
  bool _htOK;

  void calcEvtTopo();
  std::vector<double> _evtTopo;
  bool _evtTopoOK;

  void calcKt();
  std::vector<double> _kt;
  bool _ktOK;

  void calcMt();
  std::vector<double> _mt;
  bool _mtOK;
};

#endif //LJETSTOPOVARS
Revision:	1.4
Committed:	Tue Mar 30 01:32:31 2010 UTC (15 years, 1 month ago) by msegala
Content type:	text/plain
Branch:	MAIN
CVS Tags:	V00-03-01, ZMorph_BASE_20100408, gak040610_morphing, HEAD
Branch point for:	ZMorph-V00-03-01
Changes since 1.3:	+7 -2 lines
Log Message:	msegala32910
#	User	Rev	Content
1	kukartse	1.1	/***
2
3			Adopted from the D0 framework,
4			used to use event.get("caf_util::MET", met), etc, so needs something like METSelector to run upstream
5
6			***/
7
8			#ifndef LJETSTOPOVARS
9			#define LJETSTOPOVARS
10
11			#include <string>
12			#include <vector>
13			#include "FWCore/Framework/interface/Event.h"
14			#include "LJMet/MultivariateAnalysis/interface/TMBLorentzVector.h"
15			//#include "cafe/Event.hpp"
16			#include "TVectorD.h"
17
18
19			class LJetsTopoVars
20			{
21			public:
22			LJetsTopoVars(std::string jets, std::string met, std::string lepton, bool isMuon)
23			:m_jetBranch(jets),
24			m_metBranch(met),
25			m_leptonBranch(lepton),
26			m_isMuon(isMuon),
27
28
29			//_electron(NULL),
30			//_muon( NULL),
31
32			_ht( std::vector<double>(22, 0.) ),
33			_htOK( false),
34			_evtTopo( std::vector<double>( 3, 0.) ),
35			_evtTopoOK( false),
36			_kt( std::vector<double>( 3, 0.) ),
37			_ktOK( false),
38			_mt( std::vector<double>( 2, 0.) ),
39			_mtOK( false)
40			{};
41
42	kukartse	1.2	// returns # of removed jets due to insufficient isolation from lepton
43			//void setEvent(const edm::Event& event);
44	msegala	1.3	//int setEvent(const edm::Event& event, double min_dr_jet_lepton=-0.01);
45
46
47			//###################################################################################################
48			//##
49			//## Initiate LJetsTopoVars using one lepton, one MET and 4 leading jets momentum. This initiation is
50			//## used when one of the leptons is converted to a neutrino.
51			//##
52			//###################################################################################################
53
54	msegala	1.4	int setEventMetFixed(TLorentzVector&,TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //4 jets
55			int setEventMetFixed(TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //3 jets
56			int setEventMetFixed(TLorentzVector&, TLorentzVector&, TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //2 jets
57			int setEventMetFixed(TLorentzVector&, TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //1 jet
58			int setEventMetFixed(TLorentzVector&, TLorentzVector&,double min_dr_jet_lepton=-0.01); //0 jet
59			int setEvent(const edm::Event& event, double, double min_dr_jet_lepton=-0.01); //This is for corrected MET
60	msegala	1.3
61	msegala	1.4
62	msegala	1.3
63
64	kukartse	1.1
65			double aplanarity() const;
66			double centrality() const;
67			double sphericity() const;
68			double ht() const;
69			double htpluslepton() const;
70			double methtpluslepton() const;
71			double h() const;
72			double ktMinPrime() const;
73			double dphiLepMet() const;
74			double minDijetMass() const;
75			double maxJetEta() const;
76			double Et3() const;
77			double minDijetDeltaR() const;
78			double LeptonJet_DeltaR(); //btw lepton and leading 2 jet, minimum
79			double Jet1Jet2_DeltaR();
80			double Jet1Jet2_DeltaPhi();
81			double Jet1Jet2_M();
82			double Jet1Jet2_Pt();
83			double Jet1Jet2W_M();
84			double Jet1Jet2W_Pt();
85
86			double Hz(); //scalar sum of longitudinal energies of first four jets, the muon, and the neutrino
87			double HT2(); //scalar sum of transverse energies of the 2nd, 3rd, and 4th jets
88			double HT2prime(); //HT2/Hz
89			double W_MT(); //W transverse mass
90			double W_M(); //W mass
91			double W_Pt(); //W
92			double DphiJMET(); //Delta phi btw leading jet and MET
93
94			//muon only (throws exception if not)
95			double Muon_DeltaR(); //btw muon and jet, minimum
96			double Muon_etHaloScaled(); //TMBMuon::etHalo()/pT
97			double Muon_etTrkConeScaled(); //TMBMuon::etTrkCone()/pT
98
99			//electron only (throws exception if not)
100			double Electron_iso(); //TMBEMCluster::iso()
101			double Electron_lhood(); //TMBEMCluster::Lhood8(); is this right?
102
103			// Ht
104			double getHt() {if(!_htOK) calcHt(); return _ht[ 0];}
105			double getHtp() {if(!_htOK) calcHt(); return _ht[ 1];}
106			double getHtpp() {if(!_htOK) calcHt(); return _ht[ 2];}
107			double getHt2() {if(!_htOK) calcHt(); return _ht[ 3];}
108			double getHt2p() {if(!_htOK) calcHt(); return _ht[ 4];}
109			double getHt2pp() {if(!_htOK) calcHt(); return _ht[ 5];}
110			double getHt3() {if(!_htOK) calcHt(); return _ht[ 6];}
111			double getHt3p() {if(!_htOK) calcHt(); return _ht[ 7];}
112			double getHt3pp() {if(!_htOK) calcHt(); return _ht[ 8];}
113			double getCen() {if(!_htOK) calcHt(); return _ht[ 9];}
114			double getNJW() {if(!_htOK) calcHt(); return _ht[10];}
115			double getJetEtaMax() {if(!_htOK) calcHt(); return _ht[11];}
116			double getMdijetMin() {if(!_htOK) calcHt(); return _ht[12];}
117			double getMtjets() {if(!_htOK) calcHt(); return _ht[13];}
118			double getSqrtsT() {if(!_htOK) calcHt(); return _ht[14];}
119			double getMtAurelio() {if(!_htOK) calcHt(); return _ht[15];}
120			double getPzOverHT() {if(!_htOK) calcHt(); return _ht[16];}
121			double getMevent() {if(!_htOK) calcHt(); return _ht[17];}
122			double getM123inv() {if(!_htOK) calcHt(); return _ht[18];}
123			double getEta2Sum() {if(!_htOK) calcHt(); return _ht[19];}
124			double getMwRec() {if(!_htOK) calcHt(); return _ht[20];}
125			double getH() {if(!_htOK) calcHt(); return _ht[21];}
126
127			// event topo
128			double getSph() {if(!_evtTopoOK) calcEvtTopo(); return _evtTopo[0];}
129			double getApl() {if(!_evtTopoOK) calcEvtTopo(); return _evtTopo[1];}
130			double getAplMu() {if(!_evtTopoOK) calcEvtTopo(); return _evtTopo[2];}
131
132			// Kt
133			double getKtminp() {if(!_ktOK) calcKt(); return _kt[0];}
134			double getKtminpReduced() {if(!_ktOK) calcKt(); return _kt[1];}
135			double getDrMinJetJet() {if(!_ktOK) calcKt(); return _kt[2];}
136
137			// mT
138			double getDphiMuMet() {if(!_mtOK) calcMt(); return _mt[0];}
139			double getMt() {if(!_mtOK) calcMt(); return _mt[1];}
140
141			// Momentum tensor eigenvalues
142			TVectorD getEigen() {if(!_evtTopoOK) calcEvtTopo(); return eigenval;}
143
144			// Neutrino
145			TMBLorentzVector GetNeutrino() {return _neutrino;}
146
147			private:
148			std::string m_jetBranch, m_metBranch, m_leptonBranch;
149			std::vector<TMBLorentzVector> m_jets;
150			TMBLorentzVector m_met;
151			TMBLorentzVector m_lepton;
152
153			TVectorD eigenval;
154
155			bool m_isMuon;
156
157			/***** removed temporarily
158			const TMBEMCluster* _electron;
159			const TMBMuon* _muon;
160			*****/
161
162			//std::vector<TheJetClass*> _jetRAW;
163			//std::vector<TLorentzVector> _jet;
164			//TLorentzVector _lepton;
165			//TLorentzVector _neutrino;
166			TMBLorentzVector _neutrino;
167
168			void calcHt();
169			std::vector<double> _ht;
170			bool _htOK;
171
172			void calcEvtTopo();
173			std::vector<double> _evtTopo;
174			bool _evtTopoOK;
175
176			void calcKt();
177			std::vector<double> _kt;
178			bool _ktOK;
179
180			void calcMt();
181			std::vector<double> _mt;
182			bool _mtOK;
183			};
184
185			#endif //LJETSTOPOVARS