|
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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
// -*- C++ -*-
///bookLayer
// Package: SiPixelMonitorVertexSoAAlpaka
// Class: SiPixelMonitorVertexSoAAlpaka
//
/**\class SiPixelMonitorVertexSoAAlpaka SiPixelMonitorVertexSoAAlpaka.cc
*/
//
// Author: Suvankar Roy Chowdhury
//
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Common/interface/Handle.h"
// DQM Histograming
#include "DQMServices/Core/interface/MonitorElement.h"
#include "DQMServices/Core/interface/DQMEDAnalyzer.h"
#include "DQMServices/Core/interface/DQMStore.h"
#include "DataFormats/VertexSoA/interface/ZVertexHost.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
class SiPixelMonitorVertexSoAAlpaka : public DQMEDAnalyzer {
public:
using IndToEdm = std::vector<uint16_t>;
explicit SiPixelMonitorVertexSoAAlpaka(const edm::ParameterSet&);
~SiPixelMonitorVertexSoAAlpaka() override = default;
void bookHistograms(DQMStore::IBooker& ibooker, edm::Run const& iRun, edm::EventSetup const& iSetup) override;
void analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) override;
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
private:
const edm::EDGetTokenT<ZVertexHost> tokenSoAVertex_;
const edm::EDGetTokenT<reco::BeamSpot> tokenBeamSpot_;
std::string topFolderName_;
MonitorElement* hnVertex;
MonitorElement* hx;
MonitorElement* hy;
MonitorElement* hz;
MonitorElement* hchi2;
MonitorElement* hchi2oNdof;
MonitorElement* hptv2;
MonitorElement* hntrks;
};
//
// constructors
//
SiPixelMonitorVertexSoAAlpaka::SiPixelMonitorVertexSoAAlpaka(const edm::ParameterSet& iConfig)
: tokenSoAVertex_(consumes<ZVertexHost>(iConfig.getParameter<edm::InputTag>("pixelVertexSrc"))),
tokenBeamSpot_(consumes<reco::BeamSpot>(iConfig.getParameter<edm::InputTag>("beamSpotSrc"))),
topFolderName_(iConfig.getParameter<std::string>("topFolderName")) {}
//
// -- Analyze
//
void SiPixelMonitorVertexSoAAlpaka::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
const auto& vsoaHandle = iEvent.getHandle(tokenSoAVertex_);
if (!vsoaHandle.isValid()) {
edm::LogWarning("SiPixelMonitorVertexSoAAlpaka") << "No Vertex SoA found \n returning!" << std::endl;
return;
}
auto const& vsoa = *vsoaHandle;
auto vtx_view = vsoa.view<reco::ZVertexSoA>();
auto trk_view = vsoa.view<reco::ZVertexTracksSoA>();
int nVertices = vtx_view.nvFinal();
auto bsHandle = iEvent.getHandle(tokenBeamSpot_);
float x0 = 0., y0 = 0., z0 = 0., dxdz = 0., dydz = 0.;
if (!bsHandle.isValid()) {
edm::LogWarning("SiPixelMonitorVertexSoAAlpaka") << "No beamspot found. returning vertexes with (0,0,Z) ";
} else {
const reco::BeamSpot& bs = *bsHandle;
x0 = bs.x0();
y0 = bs.y0();
z0 = bs.z0();
dxdz = bs.dxdz();
dydz = bs.dydz();
}
for (int iv = 0; iv < nVertices; iv++) {
auto si = vtx_view[iv].sortInd();
auto z = vtx_view[si].zv();
auto x = x0 + dxdz * z;
auto y = y0 + dydz * z;
z += z0;
hx->Fill(x);
hy->Fill(y);
hz->Fill(z);
auto ndof = trk_view[si].ndof();
hchi2->Fill(vtx_view[si].chi2());
hchi2oNdof->Fill(vtx_view[si].chi2() / ndof);
hptv2->Fill(vtx_view[si].ptv2());
hntrks->Fill(ndof + 1);
}
hnVertex->Fill(nVertices);
}
//
// -- Book Histograms
//
void SiPixelMonitorVertexSoAAlpaka::bookHistograms(DQMStore::IBooker& ibooker,
edm::Run const& iRun,
edm::EventSetup const& iSetup) {
//std::string top_folder = ""//
ibooker.cd();
ibooker.setCurrentFolder(topFolderName_);
hnVertex = ibooker.book1D("nVertex", ";# of Vertices;#entries", 101, -0.5, 100.5);
hx = ibooker.book1D("vx", ";Vertex x;#entries", 10, -5., 5.);
hy = ibooker.book1D("vy", ";Vertex y;#entries", 10, -5., 5.);
hz = ibooker.book1D("vz", ";Vertex z;#entries", 30, -30., 30);
hchi2 = ibooker.book1D("chi2", ";Vertex chi-squared;#entries", 40, 0., 20.);
hchi2oNdof = ibooker.book1D("chi2oNdof", ";Vertex chi-squared/Ndof;#entries", 40, 0., 20.);
hptv2 = ibooker.book1D("ptsq", ";Vertex #sum (p_{T})^{2};#entries", 200, 0., 200.);
hntrks = ibooker.book1D("ntrk", ";#tracks associated;#entries", 100, -0.5, 99.5);
}
void SiPixelMonitorVertexSoAAlpaka::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
// monitorpixelVertexSoA
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("pixelVertexSrc", edm::InputTag("pixelVerticesAlpaka"));
desc.add<edm::InputTag>("beamSpotSrc", edm::InputTag("offlineBeamSpot"));
desc.add<std::string>("topFolderName", "SiPixelHeterogeneous/PixelVertexAlpaka");
descriptions.addWithDefaultLabel(desc);
}
DEFINE_FWK_MODULE(SiPixelMonitorVertexSoAAlpaka);
|