|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
#ifndef DataFormatsL1TCorrelator_TkEm_h
#define DataFormatsL1TCorrelator_TkEm_h
// -*- C++ -*-
//
// Package: L1Trigger
// Class : TkEm
//
#include "DataFormats/L1Trigger/interface/L1Candidate.h"
#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/L1Trigger/interface/EGamma.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "DataFormats/L1TParticleFlow/interface/gt_datatypes.h"
#include "FWCore/Utilities/interface/Exception.h"
#include <ap_int.h>
namespace l1t {
class TkEm : public L1Candidate {
public:
TkEm();
TkEm(const LorentzVector& p4, float tkisol = -999., float tkisolPV = -999);
TkEm(const LorentzVector& p4, const edm::Ptr<L1Candidate>& egCaloPtr, float tkisol = -999., float tkisolPV = -999);
enum class HWEncoding { None, CT, GT };
// ---------- const member functions ---------------------
const edm::Ptr<L1Candidate>& egCaloPtr() const { return egCaloPtr_; }
float trkIsol() const { return trkIsol_; } // not constrained to the PV, just track ptSum
float trkIsolPV() const { return trkIsolPV_; } // constrained to the PV by DZ
float pfIsol() const { return pfIsol_; } // not constrained to the PV, just track ptSum
float pfIsolPV() const { return pfIsolPV_; } // constrained to the PV by DZ
float puppiIsol() const { return puppiIsol_; } // not constrained to the PV, just track ptSum
float puppiIsolPV() const { return puppiIsolPV_; } // constrained to the PV by DZ
// ---------- member functions ---------------------------
void setTrkIsol(float TrkIsol) { trkIsol_ = TrkIsol; }
void setTrkIsolPV(float TrkIsolPV) { trkIsolPV_ = TrkIsolPV; }
void setPFIsol(float pfIsol) { pfIsol_ = pfIsol; }
void setPFIsolPV(float pfIsolPV) { pfIsolPV_ = pfIsolPV; }
void setPuppiIsol(float puppiIsol) { puppiIsol_ = puppiIsol; }
void setPuppiIsolPV(float puppiIsolPV) { puppiIsolPV_ = puppiIsolPV; }
void setEgCaloPtr(const edm::Ptr<L1Candidate>& egPtr) { egCaloPtr_ = egPtr; }
template <int N>
void setEgBinaryWord(ap_uint<N> word, HWEncoding encoding) {
egBinaryWord0_ = word;
egBinaryWord1_ = (word >> 32);
egBinaryWord2_ = (word >> 64);
encoding_ = encoding;
}
l1gt::Photon hwObj() const {
if (encoding() != HWEncoding::GT) {
throw cms::Exception("RuntimeError") << "TkEm::hwObj : encoding is not in GT format!" << std::endl;
}
return l1gt::Photon::unpack_ap(egBinaryWord<l1gt::Photon::BITWIDTH>());
}
template <int N>
ap_uint<N> egBinaryWord() const {
return ap_uint<N>(egBinaryWord0_) | (ap_uint<N>(egBinaryWord1_) << 32) | (ap_uint<N>(egBinaryWord2_) << 64);
}
HWEncoding encoding() const { return encoding_; }
private:
edm::Ptr<L1Candidate> egCaloPtr_;
float trkIsol_;
float trkIsolPV_;
float pfIsol_;
float pfIsolPV_;
float puppiIsol_;
float puppiIsolPV_;
uint32_t egBinaryWord0_;
uint32_t egBinaryWord1_;
uint32_t egBinaryWord2_;
HWEncoding encoding_;
};
} // namespace l1t
#endif
|