File indexing completed on 2023-03-17 11:23:27
0001 #ifndef CrossingPtBasedLinearizationPointFinder_H
0002 #define CrossingPtBasedLinearizationPointFinder_H
0003
0004 #include "RecoVertex/VertexTools/interface/LinearizationPointFinder.h"
0005 #include "TrackingTools/TransientTrack/interface/TransientTrack.h"
0006 #include "RecoVertex/VertexTools/interface/ModeFinder3d.h"
0007 #include "RecoVertex/VertexTools/interface/RecTracksDistanceMatrix.h"
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017
0018
0019
0020
0021
0022
0023
0024
0025
0026 class CrossingPtBasedLinearizationPointFinder : public LinearizationPointFinder {
0027 public:
0028
0029
0030
0031 CrossingPtBasedLinearizationPointFinder(const ModeFinder3d &algo, const signed int n_pairs = 5);
0032
0033
0034
0035
0036
0037
0038
0039 CrossingPtBasedLinearizationPointFinder(const RecTracksDistanceMatrix *m,
0040 const ModeFinder3d &algo,
0041 const signed int n_pairs = -1);
0042
0043 CrossingPtBasedLinearizationPointFinder(const CrossingPtBasedLinearizationPointFinder &);
0044
0045 ~CrossingPtBasedLinearizationPointFinder() override;
0046
0047
0048
0049 GlobalPoint getLinearizationPoint(const std::vector<reco::TransientTrack> &) const override;
0050 GlobalPoint getLinearizationPoint(const std::vector<FreeTrajectoryState> &) const override;
0051
0052 CrossingPtBasedLinearizationPointFinder *clone() const override {
0053 return new CrossingPtBasedLinearizationPointFinder(*this);
0054 };
0055
0056 protected:
0057 const bool useMatrix;
0058 signed int theNPairs;
0059 const RecTracksDistanceMatrix *theMatrix;
0060
0061 private:
0062
0063
0064 GlobalPoint find(const std::vector<std::pair<GlobalPoint, float> > &) const;
0065
0066 private:
0067 ModeFinder3d *theAlgo;
0068
0069
0070
0071 struct CompareTwoTracks {
0072 int operator()(const reco::TransientTrack &a, const reco::TransientTrack &b) {
0073 return a.initialFreeState().momentum().mag() > b.initialFreeState().momentum().mag();
0074
0075 };
0076 };
0077 std::vector<reco::TransientTrack> getBestTracks(const std::vector<reco::TransientTrack> &) const;
0078 GlobalPoint useFullMatrix(const std::vector<reco::TransientTrack> &) const;
0079 GlobalPoint useAllTracks(const std::vector<reco::TransientTrack> &) const;
0080 };
0081
0082 #endif