CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
VertexAssociatorByPositionAndTracks.cc
Go to the documentation of this file.
1 #include "CLHEP/Units/GlobalSystemOfUnits.h"
4 
6 
9  double absZ,
10  double sigmaZ,
11  double maxRecoZ,
12  double absT,
13  double sigmaT,
14  double maxRecoT,
15  double sharedTrackFraction,
16  const reco::RecoToSimCollection *trackRecoToSimAssociation,
17  const reco::SimToRecoCollection *trackSimToRecoAssociation)
18  : productGetter_(productGetter),
19  absZ_(absZ),
20  sigmaZ_(sigmaZ),
21  maxRecoZ_(maxRecoZ),
22  absT_(absT),
23  sigmaT_(sigmaT),
24  maxRecoT_(maxRecoT),
25  sharedTrackFraction_(sharedTrackFraction),
26  trackRecoToSimAssociation_(trackRecoToSimAssociation),
27  trackSimToRecoAssociation_(trackSimToRecoAssociation) {}
28 
31  double absZ,
32  double sigmaZ,
33  double maxRecoZ,
34  double sharedTrackFraction,
35  const reco::RecoToSimCollection *trackRecoToSimAssociation,
36  const reco::SimToRecoCollection *trackSimToRecoAssociation)
37  : productGetter_(productGetter),
38  absZ_(absZ),
39  sigmaZ_(sigmaZ),
40  maxRecoZ_(maxRecoZ),
41  absT_(std::numeric_limits<double>::max()),
42  sigmaT_(std::numeric_limits<double>::max()),
43  maxRecoT_(std::numeric_limits<double>::max()),
44  sharedTrackFraction_(sharedTrackFraction),
45  trackRecoToSimAssociation_(trackRecoToSimAssociation),
46  trackSimToRecoAssociation_(trackSimToRecoAssociation) {}
47 
49 
53 
54  const edm::View<reco::Vertex> &recoVertices = *vCH;
55  const TrackingVertexCollection &simVertices = *tVCH;
56 
57  LogDebug("VertexAssociation") << "VertexAssociatorByPositionAndTracks::"
58  "associateRecoToSim(): associating "
59  << recoVertices.size() << " reco::Vertices to" << simVertices.size()
60  << " TrackingVertices";
61 
62  // filter sim PVs
63  std::vector<size_t> simPVindices;
64  simPVindices.reserve(recoVertices.size());
65  {
66  int current_event = -1;
67  for (size_t iSim = 0; iSim != simVertices.size(); ++iSim) {
68  const TrackingVertex &simVertex = simVertices[iSim];
69 
70  // Associate only to primary vertices of the in-time pileup
71  // events (BX=0, first vertex in each of the events)
72  if (simVertex.eventId().bunchCrossing() != 0)
73  continue;
74  if (simVertex.eventId().event() != current_event) {
75  current_event = simVertex.eventId().event();
76  simPVindices.push_back(iSim);
77  }
78  }
79  }
80 
81  for (size_t iReco = 0; iReco != recoVertices.size(); ++iReco) {
82  const reco::Vertex &recoVertex = recoVertices[iReco];
83 
84  // skip fake vertices
85  if (std::abs(recoVertex.z()) > maxRecoZ_ || recoVertex.isFake() || !recoVertex.isValid() || recoVertex.ndof() < 0.)
86  continue;
87 
88  LogTrace("VertexAssociation") << " reco::Vertex at Z " << recoVertex.z();
89 
90  for (const size_t iSim : simPVindices) {
91  const TrackingVertex &simVertex = simVertices[iSim];
92  LogTrace("VertexAssociation") << " Considering TrackingVertex at Z " << simVertex.position().z();
93 
94  // recoVertex.t() == 0. is a special value
95  // need to change this to std::numeric_limits<double>::max() or something
96  // more clear
97  const bool useTiming = (absT_ != std::numeric_limits<double>::max() && recoVertex.t() != 0.);
98  if (useTiming) {
99  LogTrace("VertexAssociation") << " and T " << recoVertex.t() * CLHEP::second << std::endl;
100  }
101 
102  const double tdiff = std::abs(recoVertex.t() - simVertex.position().t() * CLHEP::second);
103  const double zdiff = std::abs(recoVertex.z() - simVertex.position().z());
104  if (zdiff < absZ_ && zdiff / recoVertex.zError() < sigmaZ_ &&
105  (!useTiming || (tdiff < absT_ && tdiff / recoVertex.tError() < sigmaT_))) {
106  auto sharedTracks = calculateVertexSharedTracks(recoVertex, simVertex, *trackRecoToSimAssociation_);
107  auto fraction = double(sharedTracks) / recoVertex.tracksSize();
108  if (sharedTrackFraction_ < 0 || fraction > sharedTrackFraction_) {
109  LogTrace("VertexAssociation") << " Matched with significance " << zdiff / recoVertex.zError() << " "
110  << tdiff / recoVertex.tError() << " shared tracks " << sharedTracks
111  << " reco Tracks " << recoVertex.tracksSize() << " TrackingParticles "
112  << simVertex.nDaughterTracks();
113 
114  ret.insert(reco::VertexBaseRef(vCH, iReco), std::make_pair(TrackingVertexRef(tVCH, iSim), sharedTracks));
115  }
116  }
117  }
118  }
119 
120  ret.post_insert();
121 
122  LogDebug("VertexAssociation") << "VertexAssociatorByPositionAndTracks::associateRecoToSim(): finished";
123 
124  return ret;
125 }
126 
130 
131  const edm::View<reco::Vertex> &recoVertices = *vCH;
132  const TrackingVertexCollection &simVertices = *tVCH;
133 
134  LogDebug("VertexAssociation") << "VertexAssociatorByPositionAndTracks::"
135  "associateSimToReco(): associating "
136  << simVertices.size() << " TrackingVertices to " << recoVertices.size()
137  << " reco::Vertices";
138 
139  int current_event = -1;
140  for (size_t iSim = 0; iSim != simVertices.size(); ++iSim) {
141  const TrackingVertex &simVertex = simVertices[iSim];
142 
143  // Associate only primary vertices of the in-time pileup
144  // events (BX=0, first vertex in each of the events)
145  if (simVertex.eventId().bunchCrossing() != 0)
146  continue;
147  if (simVertex.eventId().event() != current_event) {
148  current_event = simVertex.eventId().event();
149  } else {
150  continue;
151  }
152 
153  LogTrace("VertexAssociation") << " TrackingVertex at Z " << simVertex.position().z();
154 
155  for (size_t iReco = 0; iReco != recoVertices.size(); ++iReco) {
156  const reco::Vertex &recoVertex = recoVertices[iReco];
157 
158  // skip fake vertices
159  if (std::abs(recoVertex.z()) > maxRecoZ_ || recoVertex.isFake() || !recoVertex.isValid() ||
160  recoVertex.ndof() < 0.)
161  continue;
162 
163  LogTrace("VertexAssociation") << " Considering reco::Vertex at Z " << recoVertex.z();
164  const bool useTiming = (absT_ != std::numeric_limits<double>::max() && recoVertex.t() != 0.);
165  if (useTiming) {
166  LogTrace("VertexAssociation") << " and T " << recoVertex.t() * CLHEP::second << std::endl;
167  }
168 
169  const double tdiff = std::abs(recoVertex.t() - simVertex.position().t() * CLHEP::second);
170  const double zdiff = std::abs(recoVertex.z() - simVertex.position().z());
171  if (zdiff < absZ_ && zdiff / recoVertex.zError() < sigmaZ_ &&
172  (!useTiming || (tdiff < absT_ && tdiff / recoVertex.tError() < sigmaT_))) {
173  auto sharedTracks = calculateVertexSharedTracks(simVertex, recoVertex, *trackSimToRecoAssociation_);
174  auto fraction = double(sharedTracks) / recoVertex.tracksSize();
175  if (sharedTrackFraction_ < 0 || fraction > sharedTrackFraction_) {
176  LogTrace("VertexAssociation") << " Matched with significance " << zdiff / recoVertex.zError() << " "
177  << tdiff / recoVertex.tError() << " shared tracks " << sharedTracks
178  << " reco Tracks " << recoVertex.tracksSize() << " TrackingParticles "
179  << simVertex.nDaughterTracks();
180 
181  ret.insert(TrackingVertexRef(tVCH, iSim), std::make_pair(reco::VertexBaseRef(vCH, iReco), sharedTracks));
182  }
183  }
184  }
185  }
186 
187  ret.post_insert();
188 
189  LogDebug("VertexAssociation") << "VertexAssociatorByPositionAndTracks::associateSimToReco(): finished";
190 
191  return ret;
192 }
tuple ret
prodAgent to be discontinued
int event() const
get the contents of the subdetector field (should be protected?)
double zError() const
error on z
Definition: Vertex.h:141
bool isValid() const
Tells whether the vertex is valid.
Definition: Vertex.h:72
reco::VertexSimToRecoCollection associateSimToReco(const edm::Handle< edm::View< reco::Vertex >> &vCH, const edm::Handle< TrackingVertexCollection > &tVCH) const override
size_type size() const
const reco::RecoToSimCollection * trackRecoToSimAssociation_
EDProductGetter const * productGetter(std::atomic< void const * > const &iCache)
#define LogTrace(id)
U second(std::pair< T, U > const &p)
void post_insert()
post insert action
int bunchCrossing() const
get the detector field from this detid
reco::VertexRecoToSimCollection associateRecoToSim(const edm::Handle< edm::View< reco::Vertex >> &vCH, const edm::Handle< TrackingVertexCollection > &tVCH) const override
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
double z() const
z coordinate
Definition: Vertex.h:133
VertexAssociatorByPositionAndTracks(const edm::EDProductGetter *productGetter, double absZ, double sigmaZ, double maxRecoZ, double absT, double sigmaT, double maxRecoT, double sharedTrackFraction, const reco::RecoToSimCollection *trackRecoToSimAssociation, const reco::SimToRecoCollection *trackSimToRecoAssociation)
size_t tracksSize() const
number of tracks
Definition: Vertex.h:112
double ndof() const
Definition: Vertex.h:123
edm::Ref< TrackingVertexCollection > TrackingVertexRef
bool isFake() const
Definition: Vertex.h:76
unsigned int nDaughterTracks() const
void insert(const key_type &k, const data_type &v)
insert an association
std::vector< TrackingVertex > TrackingVertexCollection
const EncodedEventId & eventId() const
unsigned int calculateVertexSharedTracks(const reco::Vertex &recoV, const TrackingVertex &simV, const reco::RecoToSimCollection &trackRecoToSimAssociation)
const reco::SimToRecoCollection * trackSimToRecoAssociation_
double tError() const
error on t
Definition: Vertex.h:143
double t() const
t coordinate
Definition: Vertex.h:135
#define LogDebug(id)
const LorentzVector & position() const