CMS 3D CMS Logo
FastPrimaryVertexWithWeightsProducer.cc
Go to the documentation of this file.
1 // -*- C++ -*-
2 //
3 // Package: FastPrimaryVertexWithWeightsProducer
4 // Class: FastPrimaryVertexWithWeightsProducer
5 //
14 //
15 // Original Author: Silvio DONATO
16 // Created: Wed Dec 18 10:05:40 CET 2013
17 //
18 //
19 
20 // system include files
21 #include <memory>
22 #include <vector>
23 #include <cmath>
24 // user include files
25 #include "FWCore/Framework/interface/Frameworkfwd.h"
26 #include "FWCore/Framework/interface/stream/EDProducer.h"
27 #include "FWCore/Framework/interface/ESHandle.h"
28 #include "FWCore/Framework/interface/Event.h"
29 #include "FWCore/Framework/interface/MakerMacros.h"
30 #include "FWCore/ParameterSet/interface/ParameterSet.h"
31 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
32 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
33 
34 #include "DataFormats/JetReco/interface/CaloJet.h"
35 #include "DataFormats/JetReco/interface/CaloJetCollection.h"
36 #include "DataFormats/Math/interface/deltaPhi.h"
37 #include "DataFormats/VertexReco/interface/VertexFwd.h"
38 #include "DataFormats/VertexReco/interface/Vertex.h"
39 #include "DataFormats/TrackReco/interface/Track.h"
40 #include "DataFormats/SiPixelCluster/interface/SiPixelCluster.h"
41 #include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHit.h"
42 #include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"
43 #include "DataFormats/BeamSpot/interface/BeamSpot.h"
44 
45 #include "RecoLocalTracker/ClusterParameterEstimator/interface/PixelClusterParameterEstimator.h"
46 #include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
47 #include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
48 #include "RecoLocalTracker/Records/interface/TkPixelCPERecord.h"
49 
50 #include "RecoPixelVertexing/PixelVertexFinding/interface/FindPeakFastPV.h"
51 
52 #include "FWCore/Utilities/interface/InputTag.h"
53 
54 using namespace std;
55 
56 class FastPrimaryVertexWithWeightsProducer : public edm::stream::EDProducer<> {
57 public:
58  explicit FastPrimaryVertexWithWeightsProducer(const edm::ParameterSet&);
59  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
60 
61 private:
62  void produce(edm::Event&, const edm::EventSetup&) override;
63 
64  edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> const m_geomToken;
65  edm::ESGetToken<PixelClusterParameterEstimator, TkPixelCPERecord> const m_pixelCPEToken;
66 
67  const double m_maxZ; // Use only pixel clusters with |z| < maxZ
68  const edm::InputTag m_clusters; // PixelClusters InputTag
69  edm::EDGetTokenT<SiPixelClusterCollectionNew> clustersToken;
70  edm::EDGetTokenT<reco::BeamSpot> beamSpotToken;
71  edm::EDGetTokenT<edm::View<reco::Jet> > jetsToken;
72 
73  // PARAMETERS USED IN THE BARREL PIXEL CLUSTERS PROJECTION
74  const int m_njets; // Use only the first njets
75  const double m_maxJetEta; // Use only jets with |eta| < maxJetEta
76  const double m_minJetPt; // Use only jets with Pt > minJetPt
77  const bool m_barrel; // Use clusters from pixel endcap
78  const double m_maxSizeX; // Use only pixel clusters with sizeX <= maxSizeX
79  const double m_maxDeltaPhi; // Use only pixel clusters with DeltaPhi(Jet,Cluster) < maxDeltaPhi
80  const double m_weight_charge_down; // Use only pixel clusters with ClusterCharge > weight_charge_down
81  const double m_weight_charge_up; // Use only pixel clusters with ClusterCharge < weight_charge_up
82  //It is the ratio between pixel cell height and width along z coordinate about 285µm/150µm=1.9
83  const double m_PixelCellHeightOverWidth;
84  // Use only pixel clusters with sizeY > PixelCellHeightOverWidth * |jetZOverRho| + minSizeY_q
85  const double m_minSizeY_q;
86  // Use only pixel clusters with sizeY < PixelCellHeightOverWidth * |jetZOverRho| + maxSizeY_q
87  const double m_maxSizeY_q;
88 
89  // PARAMETERS USED TO WEIGHT THE BARREL PIXEL CLUSTERS
90  // The cluster weight is defined as weight = weight_dPhi * weight_sizeY * weight_rho * weight_sizeX1 * weight_charge
91 
92  const double m_weight_dPhi; // used in weight_dPhi = exp(-|DeltaPhi(JetCluster)|/m_weight_dPhi)
93  const double m_weight_SizeX1; // used in weight_SizeX1 = (ClusterSizeX==2)*1+(ClusterSizeX==1)*m_weight_SizeX1;
94  const double m_weight_rho_up; // used in weight_rho = (m_weight_rho_up - ClusterRho)/m_weight_rho_up
95  const double m_weight_charge_peak; // Give the maximum weight_charge for a cluster with Charge = m_weight_charge_peak
96  // Give the maximum weight_sizeY for a cluster with sizeY = PixelCellHeightOverWidth * |jetZOverRho| + peakSizeY_q
97  const double m_peakSizeY_q;
98 
99  // PARAMETERS USED IN THE ENDCAP PIXEL CLUSTERS PROJECTION
100  const bool m_endCap; // Use clusters from pixel endcap
101  const double m_minJetEta_EC; // Use only jets with |eta| > minJetEta_EC
102  const double m_maxJetEta_EC; // Use only jets with |eta| < maxJetEta_EC
103  const double m_maxDeltaPhi_EC; // Use only pixel clusters with DeltaPhi(Jet,Cluster) < maxDeltaPhi_EC
104 
105  // PARAMETERS USED TO WEIGHT THE ENDCAP PIXEL CLUSTERS
106  const double m_EC_weight; // In EndCap the weight is defined as weight = m_EC_weight*(weight_dPhi)
107  const double m_weight_dPhi_EC; // Used in weight_dPhi = exp(-|DeltaPhi|/m_weight_dPhi_EC )
108 
109  // PARAMETERS USED TO FIND THE FASTPV AS PEAK IN THE Z-PROJECTIONS DISTRIBUTION
110  // First Iteration: look for a cluster with a width = m_zClusterWidth_step1
111  const double m_zClusterWidth_step1; // cluster width in step1
112 
113  // Second Iteration: use only z-projections with weight > weightCut_step2 and look for a cluster with a width = m_zClusterWidth_step2, within of weightCut_step2 of the previous result
114  const double m_zClusterWidth_step2; // cluster width in step2
115  const double m_zClusterSearchArea_step2; // cluster width in step2
116  const double m_weightCut_step2; // minimum z-projections weight required in step2
117 
118  // Third Iteration: use only z-projections with weight > weightCut_step3 and look for a cluster with a width = m_zClusterWidth_step3, within of weightCut_step3 of the previous result
119  const double m_zClusterWidth_step3; // cluster width in step3
120  const double m_zClusterSearchArea_step3; // cluster width in step3
121  const double m_weightCut_step3; // minimum z-projections weight required in step3
122 
123  // use the jetPt weighting
124  const bool m_ptWeighting; // use weight=weight*pt_weigth ?;
125  const double m_ptWeighting_slope; // pt_weigth= pt*ptWeighting_slope +m_ptWeighting_offset;
126  const double m_ptWeighting_offset; //
127 };
128 
129 FastPrimaryVertexWithWeightsProducer::FastPrimaryVertexWithWeightsProducer(const edm::ParameterSet& iConfig)
130  : m_geomToken(esConsumes()),
131  m_pixelCPEToken(esConsumes(edm::ESInputTag("", iConfig.getParameter<std::string>("pixelCPE")))),
132  m_maxZ(iConfig.getParameter<double>("maxZ")),
133  m_njets(iConfig.getParameter<int>("njets")),
134  m_maxJetEta(iConfig.getParameter<double>("maxJetEta")),
135  m_minJetPt(iConfig.getParameter<double>("minJetPt")),
136 
137  m_barrel(iConfig.getParameter<bool>("barrel")),
138  m_maxSizeX(iConfig.getParameter<double>("maxSizeX")),
139  m_maxDeltaPhi(iConfig.getParameter<double>("maxDeltaPhi")),
140  m_weight_charge_down(iConfig.getParameter<double>("weight_charge_down")),
141  m_weight_charge_up(iConfig.getParameter<double>("weight_charge_up")),
142  m_PixelCellHeightOverWidth(iConfig.getParameter<double>("PixelCellHeightOverWidth")),
143  m_minSizeY_q(iConfig.getParameter<double>("minSizeY_q")),
144  m_maxSizeY_q(iConfig.getParameter<double>("maxSizeY_q")),
145 
146  m_weight_dPhi(iConfig.getParameter<double>("weight_dPhi")),
147  m_weight_SizeX1(iConfig.getParameter<double>("weight_SizeX1")),
148  m_weight_rho_up(iConfig.getParameter<double>("weight_rho_up")),
149  m_weight_charge_peak(iConfig.getParameter<double>("weight_charge_peak")),
150  m_peakSizeY_q(iConfig.getParameter<double>("peakSizeY_q")),
151 
152  m_endCap(iConfig.getParameter<bool>("endCap")),
153  m_minJetEta_EC(iConfig.getParameter<double>("minJetEta_EC")),
154  m_maxJetEta_EC(iConfig.getParameter<double>("maxJetEta_EC")),
155  m_maxDeltaPhi_EC(iConfig.getParameter<double>("maxDeltaPhi_EC")),
156  m_EC_weight(iConfig.getParameter<double>("EC_weight")),
157  m_weight_dPhi_EC(iConfig.getParameter<double>("weight_dPhi_EC")),
158 
159  m_zClusterWidth_step1(iConfig.getParameter<double>("zClusterWidth_step1")),
160 
161  m_zClusterWidth_step2(iConfig.getParameter<double>("zClusterWidth_step2")),
162  m_zClusterSearchArea_step2(iConfig.getParameter<double>("zClusterSearchArea_step2")),
163  m_weightCut_step2(iConfig.getParameter<double>("weightCut_step2")),
164 
165  m_zClusterWidth_step3(iConfig.getParameter<double>("zClusterWidth_step3")),
166  m_zClusterSearchArea_step3(iConfig.getParameter<double>("zClusterSearchArea_step3")),
167  m_weightCut_step3(iConfig.getParameter<double>("weightCut_step3")),
168 
169  m_ptWeighting(iConfig.getParameter<bool>("ptWeighting")),
170  m_ptWeighting_slope(iConfig.getParameter<double>("ptWeighting_slope")),
171  m_ptWeighting_offset(iConfig.getParameter<double>("ptWeighting_offset")) {
172  clustersToken = consumes<SiPixelClusterCollectionNew>(iConfig.getParameter<edm::InputTag>("clusters"));
173  beamSpotToken = consumes<reco::BeamSpot>(iConfig.getParameter<edm::InputTag>("beamSpot"));
174  jetsToken = consumes<edm::View<reco::Jet> >(iConfig.getParameter<edm::InputTag>("jets"));
175 
176  produces<reco::VertexCollection>();
177  produces<float>();
178 }
179 
180 void FastPrimaryVertexWithWeightsProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
181  edm::ParameterSetDescription desc;
182  desc.add<edm::InputTag>("clusters", edm::InputTag("hltSiPixelClusters"));
183  desc.add<edm::InputTag>("beamSpot", edm::InputTag("hltOnlineBeamSpot"));
184  desc.add<edm::InputTag>("jets", edm::InputTag("hltCaloJetL1FastJetCorrected"));
185  desc.add<std::string>("pixelCPE", "hltESPPixelCPEGeneric");
186  desc.add<double>("maxZ", 19.0);
187  desc.add<int>("njets", 999);
188  desc.add<double>("maxJetEta", 2.6);
189  desc.add<double>("minJetPt", 40.);
190  desc.add<bool>("barrel", true);
191  desc.add<double>("maxSizeX", 2.1);
192  desc.add<double>("maxDeltaPhi", 0.21);
193  desc.add<double>("PixelCellHeightOverWidth", 1.8);
194  desc.add<double>("weight_charge_down", 11. * 1000.);
195  desc.add<double>("weight_charge_up", 190. * 1000.);
196  desc.add<double>("maxSizeY_q", 2.0);
197  desc.add<double>("minSizeY_q", -0.6);
198  desc.add<double>("weight_dPhi", 0.13888888);
199  desc.add<double>("weight_SizeX1", 0.88);
200  desc.add<double>("weight_rho_up", 22.);
201  desc.add<double>("weight_charge_peak", 22. * 1000.);
202  desc.add<double>("peakSizeY_q", 1.0);
203  desc.add<bool>("endCap", true);
204  desc.add<double>("minJetEta_EC", 1.3);
205  desc.add<double>("maxJetEta_EC", 2.6);
206  desc.add<double>("maxDeltaPhi_EC", 0.14);
207  desc.add<double>("EC_weight", 0.008);
208  desc.add<double>("weight_dPhi_EC", 0.064516129);
209  desc.add<double>("zClusterWidth_step1", 2.0);
210  desc.add<double>("zClusterWidth_step2", 0.65);
211  desc.add<double>("zClusterSearchArea_step2", 3.0);
212  desc.add<double>("weightCut_step2", 0.05);
213  desc.add<double>("zClusterWidth_step3", 0.3);
214  desc.add<double>("zClusterSearchArea_step3", 0.55);
215  desc.add<double>("weightCut_step3", 0.1);
216  desc.add<bool>("ptWeighting", false); // <---- newMethod
217  desc.add<double>("ptWeighting_slope", 1 / 20.);
218  desc.add<double>("ptWeighting_offset", -1);
219  descriptions.add("fastPrimaryVertexWithWeightsProducer", desc);
220 }
221 
222 void FastPrimaryVertexWithWeightsProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
223  using namespace edm;
224  using namespace reco;
225  using namespace std;
226 
227  const float barrel_lenght = 30; //half lenght of the pixel barrel 30 cm
228 
229  //get pixel cluster
230  Handle<SiPixelClusterCollectionNew> cH;
231  iEvent.getByToken(clustersToken, cH);
232  const SiPixelClusterCollectionNew& pixelClusters = *cH.product();
233 
234  //get jets
235  Handle<edm::View<reco::Jet> > jH;
236  iEvent.getByToken(jetsToken, jH);
237  const edm::View<reco::Jet>& jets = *jH.product();
238 
239  vector<const reco::Jet*> selectedJets;
240  int countjet = 0;
241  for (edm::View<reco::Jet>::const_iterator it = jets.begin(); it != jets.end() && countjet < m_njets; it++) {
242  if ( //select jets used in barrel or endcap pixel cluster projections
243  ((it->pt() >= m_minJetPt) && std::abs(it->eta()) <= m_maxJetEta) || //barrel
244  ((it->pt() >= m_minJetPt) && std::abs(it->eta()) <= m_maxJetEta_EC &&
245  std::abs(it->eta()) >= m_minJetEta_EC) //endcap
246  ) {
247  selectedJets.push_back(&(*it));
248  countjet++;
249  }
250  }
251 
252  //get PixelClusterParameterEstimator
253  const PixelClusterParameterEstimator* pp = &iSetup.getData(m_pixelCPEToken);
254 
255  //get beamSpot
256  edm::Handle<BeamSpot> beamSpot;
257  iEvent.getByToken(beamSpotToken, beamSpot);
258 
259  //get TrackerGeometry
260  const TrackerGeometry* trackerGeometry = &iSetup.getData(m_geomToken);
261 
262  // PART I: get z-projections with z-weights
263  std::vector<float> zProjections;
264  std::vector<float> zWeights;
265  int jet_count = 0;
266  for (vector<const reco::Jet*>::iterator jit = selectedJets.begin(); jit != selectedJets.end();
267  jit++) { //loop on selected jets
268  float px = (*jit)->px();
269  float py = (*jit)->py();
270  float pz = (*jit)->pz();
271  float pt = (*jit)->pt();
272  float eta = (*jit)->eta();
273  float jetZOverRho = (*jit)->momentum().Z() / (*jit)->momentum().Rho();
274  float pt_weight = pt * m_ptWeighting_slope + m_ptWeighting_offset;
275  for (SiPixelClusterCollectionNew::const_iterator it = pixelClusters.begin(); it != pixelClusters.end();
276  it++) //Loop on pixel modules with clusters
277  { //loop on pixel modules
278  DetId id = it->detId();
279  const edmNew::DetSet<SiPixelCluster>& detset = (*it);
280  Point3DBase<float, GlobalTag> modulepos = trackerGeometry->idToDet(id)->position();
281  float zmodule = modulepos.z() -
282  ((modulepos.x() - beamSpot->x0()) * px + (modulepos.y() - beamSpot->y0()) * py) / pt * pz / pt;
283  if ((std::abs(deltaPhi((*jit)->momentum().Phi(), modulepos.phi())) < m_maxDeltaPhi * 2) &&
284  (std::abs(zmodule) < (m_maxZ + barrel_lenght))) { //if it is a compatible module
285  for (size_t j = 0; j < detset.size(); j++) { //loop on pixel clusters on this module
286  const SiPixelCluster& aCluster = detset[j];
287  if ( //it is a cluster to project
288  ( // barrel
289  m_barrel && std::abs(modulepos.z()) < barrel_lenght && pt >= m_minJetPt && jet_count < m_njets &&
290  std::abs(eta) <= m_maxJetEta && aCluster.sizeX() <= m_maxSizeX &&
291  aCluster.sizeY() >= m_PixelCellHeightOverWidth * std::abs(jetZOverRho) + m_minSizeY_q &&
292  aCluster.sizeY() <= m_PixelCellHeightOverWidth * std::abs(jetZOverRho) + m_maxSizeY_q) ||
293  ( // EC
294  m_endCap && std::abs(modulepos.z()) > barrel_lenght && pt > m_minJetPt && jet_count < m_njets &&
295  std::abs(eta) <= m_maxJetEta_EC && std::abs(eta) >= m_minJetEta_EC &&
296  aCluster.sizeX() <= m_maxSizeX)) {
297  Point3DBase<float, GlobalTag> v = trackerGeometry->idToDet(id)->surface().toGlobal(
298  pp->localParametersV(aCluster, (*trackerGeometry->idToDetUnit(id)))[0].first);
299  GlobalPoint v_bs(v.x() - beamSpot->x0(), v.y() - beamSpot->y0(), v.z());
300  if ( //it pass DeltaPhi(Jet,Cluster) requirements
301  (m_barrel && std::abs(modulepos.z()) < barrel_lenght &&
302  std::abs(deltaPhi((*jit)->momentum().Phi(), v_bs.phi())) <= m_maxDeltaPhi) || //barrel
303  (m_endCap && std::abs(modulepos.z()) > barrel_lenght &&
304  std::abs(deltaPhi((*jit)->momentum().Phi(), v_bs.phi())) <= m_maxDeltaPhi_EC) //EC
305  ) {
306  //calculate z-projection
307  float z = v.z() - ((v.x() - beamSpot->x0()) * px + (v.y() - beamSpot->y0()) * py) / pt * pz / pt;
308  if (std::abs(z) < m_maxZ) {
309  zProjections.push_back(z); //add z-projection in zProjections
310  float weight = 0;
311  //calculate zWeight
312  if (std::abs(modulepos.z()) < barrel_lenght) { //barrel
313  //calculate weight_sizeY
314  float sizeY = aCluster.sizeY();
315  float sizeY_up = m_PixelCellHeightOverWidth * std::abs(jetZOverRho) + m_maxSizeY_q;
316  float sizeY_peak = m_PixelCellHeightOverWidth * std::abs(jetZOverRho) + m_peakSizeY_q;
317  float sizeY_down = m_PixelCellHeightOverWidth * std::abs(jetZOverRho) + m_minSizeY_q;
318  float weight_sizeY_up = (sizeY_up - sizeY) / (sizeY_up - sizeY_peak);
319  float weight_sizeY_down = (sizeY - sizeY_down) / (sizeY_peak - sizeY_down);
320  weight_sizeY_down = weight_sizeY_down * (weight_sizeY_down > 0) * (weight_sizeY_down < 1);
321  weight_sizeY_up = weight_sizeY_up * (weight_sizeY_up > 0) * (weight_sizeY_up < 1);
322  float weight_sizeY = weight_sizeY_up + weight_sizeY_down;
323 
324  //calculate weight_rho
325  float rho = sqrt(v_bs.x() * v_bs.x() + v_bs.y() * v_bs.y());
326  float weight_rho = ((m_weight_rho_up - rho) / m_weight_rho_up);
327 
328  //calculate weight_dPhi
329  float weight_dPhi = exp(-std::abs(deltaPhi((*jit)->momentum().Phi(), v_bs.phi())) / m_weight_dPhi);
330 
331  //calculate weight_sizeX1
332  float weight_sizeX1 = (aCluster.sizeX() == 2) + (aCluster.sizeX() == 1) * m_weight_SizeX1;
333 
334  //calculate weight_charge
335  float charge = aCluster.charge();
336  float weightCluster_up = (m_weight_charge_up - charge) / (m_weight_charge_up - m_weight_charge_peak);
337  float weightCluster_down =
338  (charge - m_weight_charge_down) / (m_weight_charge_peak - m_weight_charge_down);
339  weightCluster_down = weightCluster_down * (weightCluster_down > 0) * (weightCluster_down < 1);
340  weightCluster_up = weightCluster_up * (weightCluster_up > 0) * (weightCluster_up < 1);
341  float weight_charge = weightCluster_up + weightCluster_down;
342 
343  //calculate the final weight
344  weight = weight_dPhi * weight_sizeY * weight_rho * weight_sizeX1 * weight_charge;
345  } else if (std::abs(modulepos.z()) > barrel_lenght) // EC
346  { // EC
347  //calculate weight_dPhi
348  float weight_dPhi = exp(-std::abs(deltaPhi((*jit)->momentum().Phi(), v_bs.phi())) / m_weight_dPhi_EC);
349  //calculate the final weight
350  weight = m_EC_weight * (weight_dPhi);
351  }
352  if (m_ptWeighting)
353  weight = weight * pt_weight;
354  zWeights.push_back(weight); //add the weight to zWeights
355  }
356  } //if it pass DeltaPhi(Jet,Cluster) requirements
357  }
358  } //loop on pixel clusters on this module
359  } //if it is a compatible module
360  } //loop on pixel modules
361  jet_count++;
362  } //loop on selected jets
363 
364  //order zProjections and zWeights by z
365  std::multimap<float, float> zWithW;
366  size_t i = 0;
367  for (i = 0; i < zProjections.size(); i++)
368  zWithW.insert(std::pair<float, float>(zProjections[i], zWeights[i]));
369  i = 0;
370  for (std::multimap<float, float>::iterator it = zWithW.begin(); it != zWithW.end(); it++, i++) {
371  zProjections[i] = it->first;
372  zWeights[i] = it->second;
373  } //order zProjections and zWeights by z
374 
375  //calculate zWeightsSquared
376  std::vector<float> zWeightsSquared;
377  for (std::vector<float>::iterator it = zWeights.begin(); it != zWeights.end(); it++) {
378  zWeightsSquared.push_back((*it) * (*it));
379  }
380 
381  //do multi-step peak searching
382  float res_step1 = FindPeakFastPV(zProjections, zWeights, 0.0, m_zClusterWidth_step1, 999.0, -1.0);
383  float res_step2 = FindPeakFastPV(
384  zProjections, zWeights, res_step1, m_zClusterWidth_step2, m_zClusterSearchArea_step2, m_weightCut_step2);
385  float res_step3 = FindPeakFastPV(zProjections,
386  zWeightsSquared,
387  res_step2,
388  m_zClusterWidth_step3,
389  m_zClusterSearchArea_step3,
390  m_weightCut_step3 * m_weightCut_step3);
391 
392  float centerWMax = res_step3;
393  //End of PART II
394 
395  //Make the output
396  float res = 0;
397  if (zProjections.size() > 2) {
398  res = centerWMax;
399  Vertex::Error e;
400  e(0, 0) = 0.0015 * 0.0015;
401  e(1, 1) = 0.0015 * 0.0015;
402  e(2, 2) = 1.5 * 1.5;
403  Vertex::Point p(beamSpot->x(res), beamSpot->y(res), res);
404  Vertex thePV(p, e, 1, 1, 0);
405  auto pOut = std::make_unique<reco::VertexCollection>();
406  pOut->push_back(thePV);
407  iEvent.put(std::move(pOut));
408  } else {
409  Vertex::Error e;
410  e(0, 0) = 0.0015 * 0.0015;
411  e(1, 1) = 0.0015 * 0.0015;
412  e(2, 2) = 1.5 * 1.5;
413  Vertex::Point p(beamSpot->x(res), beamSpot->y(res), res);
414  Vertex thePV(p, e, 0, 0, 0);
415  auto pOut = std::make_unique<reco::VertexCollection>();
416  pOut->push_back(thePV);
417  iEvent.put(std::move(pOut));
418  }
419 
420  //Finally, calculate the zClusterQuality as Sum(weights near the fastPV)/sqrt(Sum(total weights)) [a kind of zCluster significance]
421 
422  const float half_width_peak = 1;
423  float nWeightedTot = 0;
424  float nWeightedTotPeak = 0;
425  for (std::vector<float>::iterator it = zProjections.begin(); it != zProjections.end(); it++) {
426  nWeightedTot += zWeights[it - zProjections.begin()];
427  if ((res - half_width_peak) <= (*it) && (*it) <= (res + half_width_peak)) {
428  nWeightedTotPeak += zWeights[it - zProjections.begin()];
429  }
430  }
431 
432  auto zClusterQuality = std::make_unique<float>();
433  *zClusterQuality = -1;
434  if (nWeightedTot != 0) {
435  // where 30 is the beam spot lenght
436  *zClusterQuality = nWeightedTotPeak / sqrt(nWeightedTot / (2 * half_width_peak));
437  iEvent.put(std::move(zClusterQuality));
438  } else
439  iEvent.put(std::move(zClusterQuality));
440 }
441 
442 //define this as a plug-in
443 DEFINE_FWK_MODULE(FastPrimaryVertexWithWeightsProducer);
GeomDet::position
const Surface::PositionType & position() const
The position (origin of the R.F.)
Definition: GeomDet.h:43
ConfigurationDescriptions.h
TrackerGeometry::idToDet
const TrackerGeomDet * idToDet(DetId) const override
Definition: TrackerGeometry.cc:193
FindPeakFastPV.h
PixelClusterParameterEstimator
Definition: PixelClusterParameterEstimator.h:15
CaloJetCollection.h
CaloJet.h
FastPrimaryVertexWithWeightsProducer::m_maxZ
const double m_maxZ
Definition: FastPrimaryVertexWithWeightsProducer.cc:67
electrons_cff.bool
bool
Definition: electrons_cff.py:366
FastPrimaryVertexWithWeightsProducer::m_zClusterWidth_step2
const double m_zClusterWidth_step2
Definition: FastPrimaryVertexWithWeightsProducer.cc:114
mps_fire.i
i
Definition: mps_fire.py:428
pwdgSkimBPark_cfi.beamSpot
beamSpot
Definition: pwdgSkimBPark_cfi.py:5
HLT_FULL_cff.weight_dPhi
weight_dPhi
Definition: HLT_FULL_cff.py:50757
LumiMonitor_cff.pixelClusters
pixelClusters
Definition: LumiMonitor_cff.py:10
TrackerGeometry.h
edm::Handle::product
T const * product() const
Definition: Handle.h:70
FastPrimaryVertexWithWeightsProducer::m_clusters
const edm::InputTag m_clusters
Definition: FastPrimaryVertexWithWeightsProducer.cc:68
ESHandle.h
ESInputTag
FastPrimaryVertexWithWeightsProducer::m_minSizeY_q
const double m_minSizeY_q
Definition: FastPrimaryVertexWithWeightsProducer.cc:85
PV3DBase::x
T x() const
Definition: PV3DBase.h:59
DiDispStaMuonMonitor_cfi.pt
pt
Definition: DiDispStaMuonMonitor_cfi.py:39
deltaPhi.h
edm::EDGetTokenT
Definition: EDGetToken.h:33
multPhiCorr_741_25nsDY_cfi.py
py
Definition: multPhiCorr_741_25nsDY_cfi.py:12
edm
HLT enums.
Definition: AlignableModifier.h:19
reco::Vertex::Error
math::Error< dimension >::type Error
covariance error matrix (3x3)
Definition: Vertex.h:44
PixelClusterParameterEstimator.h
SiPixelCluster.h
HLT_FULL_cff.InputTag
InputTag
Definition: HLT_FULL_cff.py:89301
edm::ParameterSetDescription
Definition: ParameterSetDescription.h:52
edmNew::DetSetVector::const_iterator
boost::transform_iterator< IterHelp, const_IdIter > const_iterator
Definition: DetSetVectorNew.h:197
EDProducer.h
L1TrackObjectNtupleMaker_cfg.pOut
pOut
Definition: L1TrackObjectNtupleMaker_cfg.py:172
TrackingRecHitFwd.h
FastPrimaryVertexWithWeightsProducer::m_barrel
const bool m_barrel
Definition: FastPrimaryVertexWithWeightsProducer.cc:77
singleTopDQM_cfi.jets
jets
Definition: singleTopDQM_cfi.py:42
SiPixelCluster
Pixel cluster – collection of neighboring pixels above threshold.
Definition: SiPixelCluster.h:28
reco
fixed size matrix
Definition: AlignmentAlgorithmBase.h:46
findQualityFiles.v
v
Definition: findQualityFiles.py:179
FastPrimaryVertexWithWeightsProducer::m_zClusterSearchArea_step2
const double m_zClusterSearchArea_step2
Definition: FastPrimaryVertexWithWeightsProducer.cc:115
edm::Handle
Definition: AssociativeIterator.h:50
FastPrimaryVertexWithWeightsProducer::m_endCap
const bool m_endCap
Definition: FastPrimaryVertexWithWeightsProducer.cc:100
FastPrimaryVertexWithWeightsProducer::clustersToken
edm::EDGetTokenT< SiPixelClusterCollectionNew > clustersToken
Definition: FastPrimaryVertexWithWeightsProducer.cc:69
FastPrimaryVertexWithWeightsProducer::m_weightCut_step3
const double m_weightCut_step3
Definition: FastPrimaryVertexWithWeightsProducer.cc:121
edmNew::DetSet::size
size_type size() const
Definition: DetSetNew.h:68
FastPrimaryVertexWithWeightsProducer::m_pixelCPEToken
const edm::ESGetToken< PixelClusterParameterEstimator, TkPixelCPERecord > m_pixelCPEToken
Definition: FastPrimaryVertexWithWeightsProducer.cc:65
TrackerGeometry::idToDetUnit
const TrackerGeomDet * idToDetUnit(DetId) const override
Return the pointer to the GeomDetUnit corresponding to a given DetId.
Definition: TrackerGeometry.cc:183
PV3DBase::z
T z() const
Definition: PV3DBase.h:61
DetId
Definition: DetId.h:17
GeomDet::surface
const Plane & surface() const
The nominal surface of the GeomDet.
Definition: GeomDet.h:37
MakerMacros.h
SiPixelCluster::sizeY
int sizeY() const
Definition: SiPixelCluster.h:140
Track.h
FastPrimaryVertexWithWeightsProducer::m_maxDeltaPhi
const double m_maxDeltaPhi
Definition: FastPrimaryVertexWithWeightsProducer.cc:79
FastPrimaryVertexWithWeightsProducer::m_PixelCellHeightOverWidth
const double m_PixelCellHeightOverWidth
Definition: FastPrimaryVertexWithWeightsProducer.cc:83
DEFINE_FWK_MODULE
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:16
FastPrimaryVertexWithWeightsProducer::m_maxSizeY_q
const double m_maxSizeY_q
Definition: FastPrimaryVertexWithWeightsProducer.cc:87
BeamSpot.h
edm::ConfigurationDescriptions::add
void add(std::string const &label, ParameterSetDescription const &psetDescription)
Definition: ConfigurationDescriptions.cc:57
SiPixelRawToDigiRegional_cfi.deltaPhi
deltaPhi
Definition: SiPixelRawToDigiRegional_cfi.py:9
FastPrimaryVertexWithWeightsProducer::produce
void produce(edm::Event &, const edm::EventSetup &) override
Definition: FastPrimaryVertexWithWeightsProducer.cc:222
PVValHelper::eta
Definition: PVValidationHelpers.h:70
mathSSE::sqrt
T sqrt(T t)
Definition: SSEVec.h:19
fillDescriptions
static void fillDescriptions(edm::ConfigurationDescriptions &descriptions)
Surface::toGlobal
GlobalPoint toGlobal(const Point2DBase< Scalar, LocalTag > lp) const
Definition: Surface.h:79
DDAxes::z
edmNew::DetSet
Definition: DetSetNew.h:22
RecoTauValidation_cfi.sizeY
sizeY
Definition: RecoTauValidation_cfi.py:290
FastPrimaryVertexWithWeightsProducer::m_zClusterSearchArea_step3
const double m_zClusterSearchArea_step3
Definition: FastPrimaryVertexWithWeightsProducer.cc:120
Point3DBase< float, GlobalTag >
ParameterSetDescription.h
SiPixelRecHit.h
edm::ConfigurationDescriptions
Definition: ConfigurationDescriptions.h:28
DDAxes::rho
FastPrimaryVertexWithWeightsProducer::m_weight_dPhi_EC
const double m_weight_dPhi_EC
Definition: FastPrimaryVertexWithWeightsProducer.cc:107
ALCARECOTkAlJpsiMuMu_cff.charge
charge
Definition: ALCARECOTkAlJpsiMuMu_cff.py:47
edm::View
Definition: CaloClusterFwd.h:14
FastPrimaryVertexWithWeightsProducer::m_zClusterWidth_step1
const double m_zClusterWidth_step1
Definition: FastPrimaryVertexWithWeightsProducer.cc:111
TrackerDigiGeometryRecord.h
edm::ParameterSet
Definition: ParameterSet.h:47
AlCaHLTBitMon_ParallelJobs.p
def p
Definition: AlCaHLTBitMon_ParallelJobs.py:153
Event.h
FastPrimaryVertexWithWeightsProducer::m_minJetPt
const double m_minJetPt
Definition: FastPrimaryVertexWithWeightsProducer.cc:76
FastPrimaryVertexWithWeightsProducer::m_maxSizeX
const double m_maxSizeX
Definition: FastPrimaryVertexWithWeightsProducer.cc:78
PV3DBase::y
T y() const
Definition: PV3DBase.h:60
createfilelist.int
int
Definition: createfilelist.py:10
iEvent
int iEvent
Definition: GenABIO.cc:224
FastPrimaryVertexWithWeightsProducer::m_maxJetEta_EC
const double m_maxJetEta_EC
Definition: FastPrimaryVertexWithWeightsProducer.cc:102
edm::stream::EDProducer
Definition: EDProducer.h:36
FastPrimaryVertexWithWeightsProducer::beamSpotToken
edm::EDGetTokenT< reco::BeamSpot > beamSpotToken
Definition: FastPrimaryVertexWithWeightsProducer.cc:70
edm::EventSetup
Definition: EventSetup.h:58
FastPrimaryVertexWithWeightsProducer::m_ptWeighting
const bool m_ptWeighting
Definition: FastPrimaryVertexWithWeightsProducer.cc:124
FastPrimaryVertexWithWeightsProducer::m_weight_rho_up
const double m_weight_rho_up
Definition: FastPrimaryVertexWithWeightsProducer.cc:94
res
Definition: Electron.h:6
FastPrimaryVertexWithWeightsProducer::m_maxDeltaPhi_EC
const double m_maxDeltaPhi_EC
Definition: FastPrimaryVertexWithWeightsProducer.cc:103
AlCaHLTBitMon_QueryRunRegistry.string
string string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:256
edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord >
FastPrimaryVertexWithWeightsProducer::m_weight_charge_down
const double m_weight_charge_down
Definition: FastPrimaryVertexWithWeightsProducer.cc:80
FastPrimaryVertexWithWeightsProducer::m_maxJetEta
const double m_maxJetEta
Definition: FastPrimaryVertexWithWeightsProducer.cc:75
InputTag.h
edm::EventSetup::getData
bool getData(T &iHolder) const
Definition: EventSetup.h:127
FastPrimaryVertexWithWeightsProducer::jetsToken
edm::EDGetTokenT< edm::View< reco::Jet > > jetsToken
Definition: FastPrimaryVertexWithWeightsProducer.cc:71
VertexFwd.h
FastPrimaryVertexWithWeightsProducer::m_ptWeighting_offset
const double m_ptWeighting_offset
Definition: FastPrimaryVertexWithWeightsProducer.cc:126
multPhiCorr_741_25nsDY_cfi.px
px
Definition: multPhiCorr_741_25nsDY_cfi.py:10
edmNew::DetSetVector
Definition: DetSetNew.h:13
submitPVResolutionJobs.desc
string desc
Definition: submitPVResolutionJobs.py:251
FindPeakFastPV
float FindPeakFastPV(const std::vector< float > &zProjections, const std::vector< float > &zWeights, const float oldVertex, const float m_zClusterWidth, const float m_zClusterSearchArea, const float m_weightCut)
Definition: FindPeakFastPV.h:13
reco::Vertex::Point
math::XYZPoint Point
point in the space
Definition: Vertex.h:40
eostools.move
def move(src, dest)
Definition: eostools.py:511
std
Definition: JetResolutionObject.h:76
Vertex.h
FastPrimaryVertexWithWeightsProducer::m_weight_dPhi
const double m_weight_dPhi
Definition: FastPrimaryVertexWithWeightsProducer.cc:92
Frameworkfwd.h
FastPrimaryVertexWithWeightsProducer::FastPrimaryVertexWithWeightsProducer
FastPrimaryVertexWithWeightsProducer(const edm::ParameterSet &)
Definition: FastPrimaryVertexWithWeightsProducer.cc:129
FastPrimaryVertexWithWeightsProducer::m_EC_weight
const double m_EC_weight
Definition: FastPrimaryVertexWithWeightsProducer.cc:106
FastPrimaryVertexWithWeightsProducer::m_minJetEta_EC
const double m_minJetEta_EC
Definition: FastPrimaryVertexWithWeightsProducer.cc:101
SiPixelCluster::charge
int charge() const
Definition: SiPixelCluster.h:142
edm::ParameterSet::getParameter
T getParameter(std::string const &) const
Definition: ParameterSet.h:303
FastPrimaryVertexWithWeightsProducer::m_njets
const int m_njets
Definition: FastPrimaryVertexWithWeightsProducer.cc:74
edm::View::const_iterator
boost::indirect_iterator< typename seq_t::const_iterator > const_iterator
Definition: View.h:86
TkPixelCPERecord.h
FastPrimaryVertexWithWeightsProducer::m_weight_SizeX1
const double m_weight_SizeX1
Definition: FastPrimaryVertexWithWeightsProducer.cc:93
objectSelection_cff.selectedJets
selectedJets
Definition: objectSelection_cff.py:99
FastPrimaryVertexWithWeightsProducer::m_weight_charge_up
const double m_weight_charge_up
Definition: FastPrimaryVertexWithWeightsProducer.cc:81
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
ParameterSet.h
createTree.pp
pp
Definition: createTree.py:17
dqmiolumiharvest.j
j
Definition: dqmiolumiharvest.py:66
JetChargeProducer_cfi.exp
exp
Definition: JetChargeProducer_cfi.py:6
SiPixelCluster::sizeX
int sizeX() const
Definition: SiPixelCluster.h:137
FastPrimaryVertexWithWeightsProducer::m_geomToken
const edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord > m_geomToken
Definition: FastPrimaryVertexWithWeightsProducer.cc:64
DeDxTools::esConsumes
ESGetTokenH3DDVariant esConsumes(std::string const &Reccord, edm::ConsumesCollector &)
Definition: DeDxTools.cc:283
edm::Event
Definition: Event.h:73
FastPrimaryVertexWithWeightsProducer::m_peakSizeY_q
const double m_peakSizeY_q
Definition: FastPrimaryVertexWithWeightsProducer.cc:97
FastPrimaryVertexWithWeightsProducer::m_ptWeighting_slope
const double m_ptWeighting_slope
Definition: FastPrimaryVertexWithWeightsProducer.cc:125
edm::InputTag
Definition: InputTag.h:15
PV3DBase::phi
Geom::Phi< T > phi() const
Definition: PV3DBase.h:66
weight
Definition: weight.py:1
reco::Vertex
Definition: Vertex.h:35
FastPrimaryVertexWithWeightsProducer::m_weightCut_step2
const double m_weightCut_step2
Definition: FastPrimaryVertexWithWeightsProducer.cc:116
FastPrimaryVertexWithWeightsProducer::m_zClusterWidth_step3
const double m_zClusterWidth_step3
Definition: FastPrimaryVertexWithWeightsProducer.cc:119
FastPrimaryVertexWithWeightsProducer
Definition: FastPrimaryVertexWithWeightsProducer.cc:56
MillePedeFileConverter_cfg.e
e
Definition: MillePedeFileConverter_cfg.py:37
TrackerGeometry
Definition: TrackerGeometry.h:14
FastPrimaryVertexWithWeightsProducer::fillDescriptions
static void fillDescriptions(edm::ConfigurationDescriptions &descriptions)
Definition: FastPrimaryVertexWithWeightsProducer.cc:180
FastPrimaryVertexWithWeightsProducer::m_weight_charge_peak
const double m_weight_charge_peak
Definition: FastPrimaryVertexWithWeightsProducer.cc:95
Generated for CMSSW Reference Manual by doxygen 1.8.16