CrossingPtBasedLinearizationPointFinder.h

Go to the documentation of this file.

#ifndef CrossingPtBasedLinearizationPointFinder_H
#define CrossingPtBasedLinearizationPointFinder_H

#include "RecoVertex/VertexTools/interface/LinearizationPointFinder.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "RecoVertex/VertexTools/interface/ModeFinder3d.h"
#include "RecoVertex/VertexTools/interface/RecTracksDistanceMatrix.h"

class CrossingPtBasedLinearizationPointFinder : public LinearizationPointFinder {
public:
  CrossingPtBasedLinearizationPointFinder(const ModeFinder3d &algo, const signed int n_pairs = 5);

  CrossingPtBasedLinearizationPointFinder(const RecTracksDistanceMatrix *m,
                                          const ModeFinder3d &algo,
                                          const signed int n_pairs = -1);

  CrossingPtBasedLinearizationPointFinder(const CrossingPtBasedLinearizationPointFinder &);

  ~CrossingPtBasedLinearizationPointFinder() override;

  GlobalPoint getLinearizationPoint(const std::vector<reco::TransientTrack> &) const override;
  GlobalPoint getLinearizationPoint(const std::vector<FreeTrajectoryState> &) const override;

  CrossingPtBasedLinearizationPointFinder *clone() const override {
    return new CrossingPtBasedLinearizationPointFinder(*this);
  };

protected:
  const bool useMatrix;
  signed int theNPairs;
  const RecTracksDistanceMatrix *theMatrix;

private:
  GlobalPoint find(const std::vector<std::pair<GlobalPoint, float> > &) const;

private:
  ModeFinder3d *theAlgo;

  struct CompareTwoTracks {
    int operator()(const reco::TransientTrack &a, const reco::TransientTrack &b) {
      return a.initialFreeState().momentum().mag() > b.initialFreeState().momentum().mag();
      //       return a.p() > b.p();
    };
  };
  std::vector<reco::TransientTrack> getBestTracks(const std::vector<reco::TransientTrack> &) const;
  GlobalPoint useFullMatrix(const std::vector<reco::TransientTrack> &) const;
  GlobalPoint useAllTracks(const std::vector<reco::TransientTrack> &) const;
};

#endif