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