|
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
|
// -*- C++ -*-
//
// Package: HLTrigger/JetMET
// Class: HLTScoutingPrimaryVertexProducer
//
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/Scouting/interface/Run3ScoutingVertex.h"
#include "DataFormats/Math/interface/libminifloat.h"
#include <memory>
#include <utility>
class HLTScoutingPrimaryVertexProducer : public edm::global::EDProducer<> {
public:
explicit HLTScoutingPrimaryVertexProducer(const edm::ParameterSet&);
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
private:
void produce(edm::StreamID sid, edm::Event& iEvent, edm::EventSetup const& setup) const final;
edm::EDGetTokenT<reco::VertexCollection> const vertexCollToken_;
int const mantissaPrecision_;
};
HLTScoutingPrimaryVertexProducer::HLTScoutingPrimaryVertexProducer(const edm::ParameterSet& iConfig)
: vertexCollToken_(consumes(iConfig.getParameter<edm::InputTag>("vertexCollection"))),
mantissaPrecision_(iConfig.getParameter<int>("mantissaPrecision")) {
produces<Run3ScoutingVertexCollection>("primaryVtx");
}
void HLTScoutingPrimaryVertexProducer::produce(edm::StreamID sid,
edm::Event& iEvent,
edm::EventSetup const& setup) const {
auto outVertices = std::make_unique<Run3ScoutingVertexCollection>();
auto const vertexCollHandle = iEvent.getHandle(vertexCollToken_);
if (vertexCollHandle.isValid()) {
outVertices->reserve(vertexCollHandle->size());
for (auto const& vtx : *vertexCollHandle) {
outVertices->emplace_back(
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.x(), mantissaPrecision_),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.y(), mantissaPrecision_),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.z(), mantissaPrecision_),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.zError(), mantissaPrecision_),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.xError(), mantissaPrecision_),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.yError(), mantissaPrecision_),
vtx.tracksSize(),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.chi2(), mantissaPrecision_),
vtx.ndof(),
vtx.isValid(),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.covariance(0, 1), mantissaPrecision_),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.covariance(0, 2), mantissaPrecision_),
MiniFloatConverter::reduceMantissaToNbitsRounding(vtx.covariance(1, 2), mantissaPrecision_));
}
}
iEvent.put(std::move(outVertices), "primaryVtx");
}
void HLTScoutingPrimaryVertexProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("vertexCollection", edm::InputTag("hltPixelVertices"))
->setComment("InputTag of input collection of primary vertices");
desc.add<int>("mantissaPrecision", 10)->setComment("default float16, change to 23 for float32");
descriptions.addWithDefaultLabel(desc);
}
DEFINE_FWK_MODULE(HLTScoutingPrimaryVertexProducer);
|