TrackAssociatorByPosition.h

Go to the documentation of this file.

#ifndef TrackAssociatorByPosition_h
#define TrackAssociatorByPosition_h

#include "SimTracker/TrackAssociation/interface/TrackAssociatorBase.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "SimDataFormats/Vertex/interface/SimVertexContainer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "TrackingTools/GeomPropagators/interface/Propagator.h"
#include "Geometry/CommonDetUnit/interface/GlobalTrackingGeometry.h"

#include <TrackingTools/TrajectoryState/interface/TrajectoryStateOnSurface.h>

#include "SimGeneral/TrackingAnalysis/interface/SimHitTPAssociationProducer.h"

#include<map>

//Note that the Association Map is filled with -ch2 and not chi2 because it is ordered using std::greater:
//the track with the lowest association chi2 will be the first in the output map.

class TrackAssociatorByPosition : public TrackAssociatorBase {

 public:

   TrackAssociatorByPosition(const edm::ParameterSet& iConfig,
                 const TrackingGeometry * geo, 
                 const Propagator * prop){
     theGeometry = geo;
     thePropagator = prop;
     theMinIfNoMatch = iConfig.getParameter<bool>("MinIfNoMatch");
     theQminCut = iConfig.getParameter<double>("QminCut");
     theQCut = iConfig.getParameter<double>("QCut");
     thePositionMinimumDistance = iConfig.getParameter<double>("positionMinimumDistance");
     std::string  meth= iConfig.getParameter<std::string>("method");
     if (meth=="chi2"){ theMethod =0; }
     else if (meth=="dist"){theMethod =1;}
     else if (meth=="momdr"){theMethod = 2;}
     else if (meth=="posdr"){theMethod = 3;}
     else{
       edm::LogError("TrackAssociatorByPosition")<<meth<<" mothed not recognized. Use dr or chi2.";     }

     theConsiderAllSimHits = iConfig.getParameter<bool>("ConsiderAllSimHits");
     _simHitTpMapTag = iConfig.getParameter<edm::InputTag>("simHitTpMapTag");
   };


  ~TrackAssociatorByPosition(){
  };


  virtual
  reco::RecoToSimCollection associateRecoToSim(const edm::RefToBaseVector<reco::Track>&,
                           const edm::RefVector<TrackingParticleCollection>&,
                           const edm::Event * event = 0,
                                               const edm::EventSetup * setup = 0 ) const override;

  virtual
  reco::SimToRecoCollection associateSimToReco(const edm::RefToBaseVector<reco::Track>&,
                           const edm::RefVector<TrackingParticleCollection>&,
                           const edm::Event * event = 0,
                                               const edm::EventSetup * setup = 0 ) const override;

  double quality(const TrajectoryStateOnSurface&, const TrajectoryStateOnSurface &)const;

 private:

  const TrackingGeometry * theGeometry;
  const Propagator * thePropagator;
  unsigned int theMethod;
  double theQminCut;
  double theQCut;
  bool theMinIfNoMatch;
  double thePositionMinimumDistance;
  bool theConsiderAllSimHits;
  
  FreeTrajectoryState getState(const reco::Track &) const;
  TrajectoryStateOnSurface getState(const TrackingParticleRef&, const SimHitTPAssociationProducer::SimHitTPAssociationList& simHitsTPAssoc)const;
  edm::InputTag _simHitTpMapTag;
};

#endif