|
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
|
#include "CommonTools/RecoAlgos/interface/PrimaryVertexSorting.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
#include "DataFormats/Math/interface/deltaR.h"
#include "TrackingTools/IPTools/interface/IPTools.h"
#include <fastjet/internal/base.hh>
#include "fastjet/PseudoJet.hh"
#include "fastjet/JetDefinition.hh"
#include "fastjet/ClusterSequence.hh"
#include "fastjet/Selector.hh"
#include "fastjet/PseudoJet.hh"
#include "FWCore/Utilities/interface/isFinite.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
using namespace fastjet;
using namespace std;
float PrimaryVertexSorting::score(const reco::Vertex &pv,
const std::vector<const reco::Candidate *> &cands,
bool useMet) const {
typedef math::XYZTLorentzVector LorentzVector;
float sumPt2 = 0;
float sumEt = 0;
LorentzVector met;
std::vector<fastjet::PseudoJet> fjInputs_;
fjInputs_.clear();
size_t countScale0 = 0;
for (size_t i = 0; i < cands.size(); i++) {
const reco::Candidate *c = cands[i];
float scale = 1.;
if (c->bestTrack() != nullptr) {
if (c->pt() != 0) {
scale = (c->pt() - c->bestTrack()->ptError()) / c->pt();
}
if (edm::isNotFinite(scale)) {
edm::LogWarning("PrimaryVertexSorting") << "Scaling is NAN ignoring this candidate/track" << std::endl;
scale = 0;
}
if (scale < 0) {
scale = 0;
countScale0++;
}
}
int absId = abs(c->pdgId());
if (absId == 13 or absId == 11) {
float pt = c->pt() * scale;
sumPt2 += pt * pt;
met += c->p4() * scale;
sumEt += c->pt() * scale;
} else {
if (scale != 0) { // otherwise, what is the point to cluster zeroes
fjInputs_.push_back(fastjet::PseudoJet(c->px() * scale, c->py() * scale, c->pz() * scale, c->p4().E() * scale));
// fjInputs_.back().set_user_index(i);
}
}
}
fastjet::ClusterSequence sequence(fjInputs_, JetDefinition(antikt_algorithm, 0.4));
auto jets = fastjet::sorted_by_pt(sequence.inclusive_jets(0));
for (const auto &pj : jets) {
auto p4 = LorentzVector(pj.px(), pj.py(), pj.pz(), pj.e());
sumPt2 += (p4.pt() * p4.pt()) * 0.8 * 0.8;
met += p4;
sumEt += p4.pt();
}
float metAbove = met.pt() - 2 * sqrt(sumEt);
if (metAbove > 0 and useMet) {
sumPt2 += metAbove * metAbove;
}
if (countScale0 == cands.size())
sumPt2 = countScale0 * 0.01; //leave some epsilon value to sort vertices with unknown pt
return sumPt2;
}
|