|
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#include "DataFormats/BTauReco/interface/TauMassTagInfo.h"
#include "DataFormats/EgammaReco/interface/BasicCluster.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/Common/interface/RefVector.h"
using namespace edm;
using namespace reco;
using namespace std;
float reco::TauMassTagInfo::discriminator(
double matching_cone, double leading_trk_pt, double signal_cone, double cluster_track_cone, double m_cut) const {
float discriminator = 0.0;
double invariantMass = getInvariantMass(matching_cone, leading_trk_pt, signal_cone, cluster_track_cone);
if (invariantMass >= 0.0 && invariantMass < m_cut)
discriminator = 1.0;
return discriminator;
}
//
// -- Set IsolatedTauTag
//
void reco::TauMassTagInfo::setIsolatedTauTag(const IsolatedTauTagInfoRef isolationRef) { isolatedTau = isolationRef; }
//
// -- Get IsolatedTauTag
//
const IsolatedTauTagInfoRef& reco::TauMassTagInfo::getIsolatedTauTag() const { return isolatedTau; }
//
// -- Set Cluster Collection
//
void reco::TauMassTagInfo::storeClusterTrackCollection(reco::BasicClusterRef clusterRef, float dr) {
clusterMap.insert(clusterRef, dr);
}
//
// -- Calculate 4 momentum vector from tracks only
//
bool reco::TauMassTagInfo::calculateTrkP4(double matching_cone,
double leading_trk_pt,
double signal_cone,
math::XYZTLorentzVector& p4) const {
const TrackRef leadTk = isolatedTau->leadingSignalTrack(matching_cone, leading_trk_pt);
if (!leadTk) {
std::cout << " TauMassTagInfo:: No Leading Track !! " << std::endl;
return false;
}
math::XYZVector momentum = (*leadTk).momentum();
const RefVector<TrackCollection> signalTracks = isolatedTau->tracksInCone(momentum, signal_cone, 1.0);
// if (signalTracks.size() == 0 || signalTracks.size()%2 == 0) return false;
if (signalTracks.empty())
return false;
double px_inv = 0.0;
double py_inv = 0.0;
double pz_inv = 0.0;
double e_inv = 0.0;
for (RefVector<TrackCollection>::const_iterator itrack = signalTracks.begin(); itrack != signalTracks.end();
itrack++) {
double p = (*itrack)->p();
double energy = sqrt(p * p + 0.139 * 0.139); // use correct value!
px_inv += (*itrack)->px();
py_inv += (*itrack)->py();
pz_inv += (*itrack)->pz();
e_inv += energy;
}
p4.SetPx(px_inv);
p4.SetPy(py_inv);
p4.SetPz(pz_inv);
p4.SetE(e_inv);
return true;
}
//
// -- Get Invariant Mass
//
double reco::TauMassTagInfo::getInvariantMassTrk(double matching_cone,
double leading_trk_pt,
double signal_cone) const {
math::XYZTLorentzVector totalP4;
if (!calculateTrkP4(matching_cone, leading_trk_pt, signal_cone, totalP4))
return -1.0;
return totalP4.M();
}
//
// -- Get Invariant Mass
//
double reco::TauMassTagInfo::getInvariantMass(double matching_cone,
double leading_trk_pt,
double signal_cone,
double track_cone) const {
math::XYZTLorentzVector totalP4;
if (!calculateTrkP4(matching_cone, leading_trk_pt, signal_cone, totalP4))
return -1.0;
// Add Clusters away from tracks
for (ClusterTrackAssociationCollection::const_iterator mapIter = clusterMap.begin(); mapIter != clusterMap.end();
mapIter++) {
const reco::BasicClusterRef& iclus = mapIter->key;
float dr = mapIter->val;
if (dr > track_cone) {
math::XYZVector clus3Vec(iclus->x(), iclus->y(), iclus->z());
double e = iclus->energy();
double theta = clus3Vec.theta();
double phi = clus3Vec.phi();
double px = e * sin(theta) * cos(phi);
double py = e * sin(theta) * sin(phi);
double pz = e * cos(theta);
math::XYZTLorentzVector p4(px, py, pz, e);
totalP4 += p4;
}
}
return totalP4.M();
}
|