ConversionTrackPairFinder.h

Go to the documentation of this file.

#ifndef ConversionTrackPairFinder_H
#define ConversionTrackPairFinder_H

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackFwd.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackExtra.h"

//
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"
#include "DataFormats/Common/interface/ValueMap.h"

//
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "TrackingTools/TransientTrack/interface/TrackTransientTrack.h"
#include "DataFormats/EgammaTrackReco/interface/TrackCaloClusterAssociation.h"


struct CompareTwoTracks {  
  bool operator() ( const reco::TransientTrack& a, const reco::TransientTrack & b ) const{
    return a.impactPointState().globalMomentum().perp() > b.impactPointState().globalMomentum().perp();}
};

struct CompareTwoTracksVectors {
   bool operator() ( const std::vector<reco::TransientTrack> & a, const std::vector<reco::TransientTrack> & b ) const{
           return ( a[0].impactPointState().globalMomentum().perp() >
                     b[0].impactPointState().globalMomentum().perp() ) ;}
   };


class ConversionTrackPairFinder {

public:

  ConversionTrackPairFinder();


  ~ConversionTrackPairFinder();



  std::map<std::vector<reco::TransientTrack>,  reco::CaloClusterPtr, CompareTwoTracksVectors>  run(std::vector<reco::TransientTrack> outIn,  
                          const edm::Handle<reco::TrackCollection>&  outInTrkHandle,
                          const edm::Handle<reco::TrackCaloClusterPtrAssociation>&  outInTrackSCAssH, 
                          std::vector<reco::TransientTrack> inOut,  
                          const edm::Handle<reco::TrackCollection>& inOutTrkHandle,
                          const edm::Handle<reco::TrackCaloClusterPtrAssociation>& inOutTrackSCAssH  );




 private:

class ByNumOfHits {
 public:
  bool operator()(reco::TransientTrack const & a, reco::TransientTrack  const & b) {
    if (a.numberOfValidHits()  == b.numberOfValidHits()  ) {
      return a.normalizedChi2() < b.normalizedChi2();
    } else {
      return a.numberOfValidHits()  > b.numberOfValidHits() ;
    }
  }
};



};

#endif // ConversionTrackPairFinder_H