CMS 3D CMS Logo
List of all members | Public Member Functions | Private Member Functions | Private Attributes
TrackVertexArbitration< VTX > Class Template Reference

#include <TrackVertexArbitration.h>

Public Member Functions

 TrackVertexArbitration (const edm::ParameterSet &params)
 
std::vector< VTX > trackVertexArbitrator (edm::Handle< reco::BeamSpot > &beamSpot, const reco::Vertex &pv, std::vector< reco::TransientTrack > &selectedTracks, std::vector< VTX > &secondaryVertices)
 

Private Member Functions

bool trackFilterArbitrator (const reco::TransientTrack &track) const
 

Private Attributes

edm::InputTag beamSpotCollection
 
double distCut
 
double dLenFraction
 
double dRCut
 
double fitterRatio
 
double fitterSigmacut
 
double fitterTini
 
double maxTimeSignificance
 
edm::InputTag primaryVertexCollection
 
edm::InputTag secondaryVertexCollection
 
double sigCut
 
double sign
 
edm::InputTag trackCollection
 
int trackMinLayers
 
int trackMinPixels
 
double trackMinPt
 

Detailed Description

template<class VTX>
class TrackVertexArbitration< VTX >

Definition at line 53 of file TrackVertexArbitration.h.

Constructor & Destructor Documentation

◆ TrackVertexArbitration()

template<class VTX >
TrackVertexArbitration< VTX >::TrackVertexArbitration ( const edm::ParameterSet &  params)

Definition at line 85 of file TrackVertexArbitration.h.

References TrackVertexArbitration< VTX >::dRCut, and TrackVertexArbitration< VTX >::sign.

86  : primaryVertexCollection(params.getParameter<edm::InputTag>("primaryVertices")),
87  secondaryVertexCollection(params.getParameter<edm::InputTag>("secondaryVertices")),
88  trackCollection(params.getParameter<edm::InputTag>("tracks")),
89  beamSpotCollection(params.getParameter<edm::InputTag>("beamSpot")),
90  dRCut(params.getParameter<double>("dRCut")),
91  distCut(params.getParameter<double>("distCut")),
92  sigCut(params.getParameter<double>("sigCut")),
93  dLenFraction(params.getParameter<double>("dLenFraction")),
94  fitterSigmacut(params.getParameter<double>("fitterSigmacut")),
95  fitterTini(params.getParameter<double>("fitterTini")),
96  fitterRatio(params.getParameter<double>("fitterRatio")),
97  trackMinLayers(params.getParameter<int32_t>("trackMinLayers")),
98  trackMinPt(params.getParameter<double>("trackMinPt")),
99  trackMinPixels(params.getParameter<int32_t>("trackMinPixels")),
100  maxTimeSignificance(params.getParameter<double>("maxTimeSignificance")) {
101  sign = 1.0;
102  if (dRCut < 0) {
103  sign = -1.0;
104  }
105  dRCut *= dRCut;
106 }
TrackVertexArbitration::secondaryVertexCollection
edm::InputTag secondaryVertexCollection
Definition: TrackVertexArbitration.h:66

Member Function Documentation

◆ trackFilterArbitrator()

template<class VTX >
bool TrackVertexArbitration< VTX >::trackFilterArbitrator ( const reco::TransientTrack &  track) const
private

Definition at line 108 of file TrackVertexArbitration.h.

References HLT_2024v12_cff::track, HLT_2024v12_cff::trackMinLayers, HLT_2024v12_cff::trackMinPixels, and HLT_2024v12_cff::trackMinPt.

108  {
109  if (!track.isValid())
110  return false;
111  if (track.track().hitPattern().trackerLayersWithMeasurement() < trackMinLayers)
112  return false;
113  if (track.track().pt() < trackMinPt)
114  return false;
115  if (track.track().hitPattern().numberOfValidPixelHits() < trackMinPixels)
116  return false;
117 
118  return true;
119 }
HLT_2024v12_cff.track
track
Definition: HLT_2024v12_cff.py:9556

◆ trackVertexArbitrator()

template<class VTX >
std::vector< VTX > TrackVertexArbitration< VTX >::trackVertexArbitrator ( edm::Handle< reco::BeamSpot > &  beamSpot,
const reco::Vertex &  pv,
std::vector< reco::TransientTrack > &  selectedTracks,
std::vector< VTX > &  secondaryVertices 
)

Definition at line 140 of file TrackVertexArbitration.h.

References funct::abs(), IPTools::absoluteImpactParameter3D(), pwdgSkimBPark_cfi::beamSpot, RecoVertex::convertError(), RecoVertex::convertPos(), gather_cfg::cout, svhelper::cov33(), PbPb_ZMuSkimMuonDPG_cff::deltaR, HLTMuonOfflineAnalyzer_cfi::deltaR2, VertexDistance3D::distance(), MaterialEffects_cfi::distCut, HLT_2024v12_cff::dLenFraction, HGC3DClusterGenMatchSelector_cfi::dR, HLT_2024v12_cff::dRCut, HLT_2024v12_cff::fitterRatio, HLT_2024v12_cff::fitterSigmacut, HLT_2024v12_cff::fitterTini, edm::isFinite(), TransientVertex::isValid(), HLT_2024v12_cff::maxTimeSignificance, funct::pow(), HLT_2024v12_cff::secondaryVertices, TrackCollections2monitor_cff::selectedTracks, HLT_2024v12_cff::sigCut, Validation_hcalonly_cfi::sign, Measurement1D::significance(), mathSSE::sqrt(), pfDeepBoostedJetPreprocessParams_cfi::sv, trackWeight(), svhelper::updateVertexTime(), Measurement1D::value(), AdaptiveVertexFitter::vertex(), and w().

143  {
144  using namespace reco;
145 
146  //std::cout << "PV: " << pv.position() << std::endl;
147  VertexDistance3D dist;
148  AdaptiveVertexFitter theAdaptiveFitter(GeometricAnnealing(fitterSigmacut, fitterTini, fitterRatio),
149  DefaultLinearizationPointFinder(),
150  KalmanVertexUpdator<5>(),
151  KalmanVertexTrackCompatibilityEstimator<5>(),
152  KalmanVertexSmoother());
153 
154  std::vector<VTX> recoVertices;
155  VertexDistance3D vdist;
156  GlobalPoint ppv(pv.position().x(), pv.position().y(), pv.position().z());
157 
158  std::unordered_map<unsigned int, Measurement1D> cachedIP;
159  for (typename std::vector<VTX>::const_iterator sv = secondaryVertices.begin(); sv != secondaryVertices.end(); ++sv) {
160  const double svTime(sv->t()), svTimeCov(svhelper::cov33(*sv));
161  const bool svHasTime = svTimeCov > 0.;
162 
163  GlobalPoint ssv(sv->position().x(), sv->position().y(), sv->position().z());
164  GlobalVector flightDir = ssv - ppv;
165  // std::cout << "Vertex : " << sv-secondaryVertices->begin() << " " << sv->position() << std::endl;
166  Measurement1D dlen =
167  vdist.distance(pv, VertexState(RecoVertex::convertPos(sv->position()), RecoVertex::convertError(sv->error())));
168  std::vector<reco::TransientTrack> selTracks;
169  for (unsigned int itrack = 0; itrack < selectedTracks.size(); itrack++) {
170  TransientTrack &tt = (selectedTracks)[itrack];
171  if (!trackFilterArbitrator(tt))
172  continue;
173  tt.setBeamSpot(*beamSpot);
174  float w = trackWeight(*sv, tt);
175  Measurement1D ipv;
176  if (cachedIP.count(itrack)) {
177  ipv = cachedIP[itrack];
178  } else {
179  std::pair<bool, Measurement1D> ipvp = IPTools::absoluteImpactParameter3D(tt, pv);
180  cachedIP[itrack] = ipvp.second;
181  ipv = ipvp.second;
182  }
183 
184  AnalyticalImpactPointExtrapolator extrapolator(tt.field());
185  TrajectoryStateOnSurface tsos =
186  extrapolator.extrapolate(tt.impactPointState(), RecoVertex::convertPos(sv->position()));
187  if (!tsos.isValid())
188  continue;
189  GlobalPoint refPoint = tsos.globalPosition();
190  GlobalError refPointErr = tsos.cartesianError().position();
191  Measurement1D isv =
192  dist.distance(VertexState(RecoVertex::convertPos(sv->position()), RecoVertex::convertError(sv->error())),
193  VertexState(refPoint, refPointErr));
194 
195  float dR = Geom::deltaR2(((sign > 0) ? flightDir : -flightDir),
196  tt.track()); //.eta(), flightDir.phi(), tt.track().eta(), tt.track().phi());
197 
198  double timeSig = 0.;
199  if (svHasTime && edm::isFinite(tt.timeExt())) {
200  double tError = std::sqrt(std::pow(tt.dtErrorExt(), 2) + svTimeCov);
201  timeSig = std::abs(tt.timeExt() - svTime) / tError;
202  }
203 
204  if (w > 0 || (isv.significance() < sigCut && isv.value() < distCut && isv.value() < dlen.value() * dLenFraction &&
205  timeSig < maxTimeSignificance)) {
206  if ((isv.value() < ipv.value()) && isv.value() < distCut && isv.value() < dlen.value() * dLenFraction &&
207  dR < dRCut && timeSig < maxTimeSignificance) {
208 #ifdef VTXDEBUG
209  if (w > 0.5)
210  std::cout << " = ";
211  else
212  std::cout << " + ";
213 #endif
214  selTracks.push_back(tt);
215  } else {
216 #ifdef VTXDEBUG
217  if (w > 0.5 && isv.value() > ipv.value())
218  std::cout << " - ";
219  else
220  std::cout << " ";
221 #endif
222  //add also the tracks used in previous fitting that are still closer to Sv than Pv
223  if (w > 0.5 && isv.value() <= ipv.value() && dR < dRCut && timeSig < maxTimeSignificance) {
224  selTracks.push_back(tt);
225 #ifdef VTXDEBUG
226  std::cout << " = ";
227 #endif
228  }
229  if (w > 0.5 && isv.value() <= ipv.value() && dR >= dRCut) {
230 #ifdef VTXDEBUG
231  std::cout << " - ";
232 #endif
233  }
234  }
235 #ifdef VTXDEBUG
236 
237  std::cout << "t : " << itrack << " ref " << ref.key() << " w: " << w << " svip: " << isv.significance() << " "
238  << isv.value() << " pvip: " << ipv.significance() << " " << ipv.value() << " res "
239  << tt.track().residualX(0) << "," << tt.track().residualY(0)
240 // << " tpvip: " << itpv.second.significance() << " " << itpv.second.value() << " dr: " << dR << std::endl;
241 #endif
242 
243  } else {
244 #ifdef VTXDEBUG
245 
246  std::cout << " . t : " << itrack << " ref " << ref.key() << " w: " << w
247  << " svip: " << isv.second.significance() << " " << isv.second.value()
248  << " pvip: " << ipv.significance() << " " << ipv.value() << " dr: " << dR << std::endl;
249 #endif
250  }
251  }
252 
253  if (selTracks.size() >= 2) {
254  TransientVertex singleFitVertex;
255  singleFitVertex = theAdaptiveFitter.vertex(selTracks, ssv);
256 
257  if (singleFitVertex.isValid()) {
258  if (pv.covariance(3, 3) > 0.)
259  svhelper::updateVertexTime(singleFitVertex);
260  recoVertices.push_back(VTX(singleFitVertex));
261 
262 #ifdef VTXDEBUG
263  const VTX &extVertex = recoVertices.back();
264  GlobalVector vtxDir = GlobalVector(extVertex.p4().X(), extVertex.p4().Y(), extVertex.p4().Z());
265 
266  std::cout << " pointing : " << Geom::deltaR(extVertex.position() - pv.position(), vtxDir) << std::endl;
267 #endif
268  }
269  }
270  }
271  return recoVertices;
272 }
RecoVertex::convertPos
reco::Vertex::Point convertPos(const GlobalPoint &p)
Definition: ConvertToFromReco.h:7
TrackCollections2monitor_cff.selectedTracks
selectedTracks
Definition: TrackCollections2monitor_cff.py:34
w
T w() const
Definition: extBasic3DVector.h:225
RecoVertex::convertError
reco::Vertex::Error convertError(const GlobalError &ge)
Definition: ConvertError.h:8
IPTools::absoluteImpactParameter3D
std::pair< bool, Measurement1D > absoluteImpactParameter3D(const reco::TransientTrack &transientTrack, const reco::Vertex &vertex)
Definition: IPTools.cc:38
PbPb_ZMuSkimMuonDPG_cff.deltaR
deltaR
Definition: PbPb_ZMuSkimMuonDPG_cff.py:63
pwdgSkimBPark_cfi.beamSpot
beamSpot
Definition: pwdgSkimBPark_cfi.py:5
pfDeepBoostedJetPreprocessParams_cfi.sv
sv
Definition: pfDeepBoostedJetPreprocessParams_cfi.py:352
edm::isFinite
constexpr bool isFinite(T x)
HLTMuonOfflineAnalyzer_cfi.deltaR2
deltaR2
Definition: HLTMuonOfflineAnalyzer_cfi.py:53
tt
Definition: TTTypes.h:54
TransientVertex::isValid
bool isValid() const
Definition: TransientVertex.h:195
mathSSE::sqrt
T sqrt(T t)
Definition: SSEVec.h:19
funct::abs
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
VertexDistance3D::distance
Measurement1D distance(const GlobalPoint &vtx1Position, const GlobalError &vtx1PositionError, const GlobalPoint &vtx2Position, const GlobalError &vtx2PositionError) const override
Definition: VertexDistance3D.cc:17
HLT_2024v12_cff.secondaryVertices
secondaryVertices
Definition: HLT_2024v12_cff.py:38160
HGC3DClusterGenMatchSelector_cfi.dR
dR
Definition: HGC3DClusterGenMatchSelector_cfi.py:7
Measurement1D::value
double value() const
Definition: Measurement1D.h:25
Measurement1D::significance
double significance() const
Definition: Measurement1D.h:29
svhelper::updateVertexTime
void updateVertexTime(TransientVertex &vtx)
Definition: SVTimeHelpers.h:11
reco
fixed size matrix
Definition: AlignmentAlgorithmBase.h:46
TrackVertexArbitration::trackFilterArbitrator
bool trackFilterArbitrator(const reco::TransientTrack &track) const
Definition: TrackVertexArbitration.h:108
svhelper::cov33
double cov33(const reco::Vertex &sv)
Definition: TrackVertexArbitration.h:48
gather_cfg.cout
cout
Definition: gather_cfg.py:144
funct::pow
Power< A, B >::type pow(const A &a, const B &b)
Definition: Power.h:29
trackWeight
float trackWeight(const reco::Vertex &sv, const reco::TransientTrack &track)
Definition: TrackVertexArbitration.h:121
GlobalVector
Global3DVector GlobalVector
Definition: GlobalVector.h:10

Member Data Documentation

◆ beamSpotCollection

template<class VTX >
edm::InputTag TrackVertexArbitration< VTX >::beamSpotCollection
private

Definition at line 68 of file TrackVertexArbitration.h.

◆ distCut

template<class VTX >
double TrackVertexArbitration< VTX >::distCut
private

Definition at line 70 of file TrackVertexArbitration.h.

◆ dLenFraction

template<class VTX >
double TrackVertexArbitration< VTX >::dLenFraction
private

Definition at line 72 of file TrackVertexArbitration.h.

◆ dRCut

template<class VTX >
double TrackVertexArbitration< VTX >::dRCut
private

Definition at line 69 of file TrackVertexArbitration.h.

Referenced by TrackVertexArbitration< VTX >::TrackVertexArbitration().

◆ fitterRatio

template<class VTX >
double TrackVertexArbitration< VTX >::fitterRatio
private

Definition at line 75 of file TrackVertexArbitration.h.

◆ fitterSigmacut

template<class VTX >
double TrackVertexArbitration< VTX >::fitterSigmacut
private

Definition at line 73 of file TrackVertexArbitration.h.

◆ fitterTini

template<class VTX >
double TrackVertexArbitration< VTX >::fitterTini
private

Definition at line 74 of file TrackVertexArbitration.h.

◆ maxTimeSignificance

template<class VTX >
double TrackVertexArbitration< VTX >::maxTimeSignificance
private

Definition at line 79 of file TrackVertexArbitration.h.

◆ primaryVertexCollection

template<class VTX >
edm::InputTag TrackVertexArbitration< VTX >::primaryVertexCollection
private

Definition at line 65 of file TrackVertexArbitration.h.

◆ secondaryVertexCollection

template<class VTX >
edm::InputTag TrackVertexArbitration< VTX >::secondaryVertexCollection
private

Definition at line 66 of file TrackVertexArbitration.h.

◆ sigCut

template<class VTX >
double TrackVertexArbitration< VTX >::sigCut
private

Definition at line 71 of file TrackVertexArbitration.h.

◆ sign

template<class VTX >
double TrackVertexArbitration< VTX >::sign
private

Definition at line 80 of file TrackVertexArbitration.h.

Referenced by TrackVertexArbitration< VTX >::TrackVertexArbitration().

◆ trackCollection

template<class VTX >
edm::InputTag TrackVertexArbitration< VTX >::trackCollection
private

Definition at line 67 of file TrackVertexArbitration.h.

◆ trackMinLayers

template<class VTX >
int TrackVertexArbitration< VTX >::trackMinLayers
private

Definition at line 76 of file TrackVertexArbitration.h.

◆ trackMinPixels

template<class VTX >
int TrackVertexArbitration< VTX >::trackMinPixels
private

Definition at line 78 of file TrackVertexArbitration.h.

◆ trackMinPt

template<class VTX >
double TrackVertexArbitration< VTX >::trackMinPt
private

Definition at line 77 of file TrackVertexArbitration.h.

Generated for CMSSW Reference Manual by doxygen 1.8.14