TrackProducerAlgorithm.h

Go to the documentation of this file.

#ifndef TrackProducerAlgorithm_h
#define TrackProducerAlgorithm_h

#include "AlgoProductTraits.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/TrackCandidate/interface/TrackCandidateCollection.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHit.h"
#include "TrackingTools/PatternTools/interface/TrackConstraintAssociation.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "RecoTracker/TransientTrackingRecHit/interface/TkTransientTrackingRecHitBuilder.h"
#include "TrackingTools/TrackFitters/interface/TrajectoryFitter.h"


class MagneticField;
class TrackingGeometry;
class Propagator;
class Trajectory;
class TrajectoryStateOnSurface;

struct FitterCloner {
   std::unique_ptr<TrajectoryFitter> fitter;
   TkClonerImpl hitCloner;

  FitterCloner(const TrajectoryFitter * theFitter,const TransientTrackingRecHitBuilder* builder):
    fitter(theFitter->clone()),
    hitCloner(static_cast<TkTransientTrackingRecHitBuilder const *>(builder)->cloner()){
    fitter->setHitCloner(&hitCloner);
  }
};


template <class T>
class TrackProducerAlgorithm : public AlgoProductTraits<T> {
public:
  using Base = AlgoProductTraits<T>;
  using TrackCollection = typename Base::TrackCollection;
  using AlgoProductCollection = typename Base::AlgoProductCollection;

  using SeedRef= edm::RefToBase<TrajectorySeed>;
  using VtxConstraintAssociationCollection = edm::AssociationMap<edm::OneToOne<std::vector<T>,std::vector<VertexConstraint> > > ;
  
public:

  TrackProducerAlgorithm(const edm::ParameterSet& conf) : 
    algo_(reco::TrackBase::algoByName(conf.getParameter<std::string>("AlgorithmName"))),
    originalAlgo_(reco::TrackBase::undefAlgorithm),
    stopReason_(0),
    reMatchSplitHits_(false),
    usePropagatorForPCA_(false)
      {
        geometricInnerState_ = (conf.exists("GeometricInnerState") ?
      conf.getParameter<bool>( "GeometricInnerState" ) : true);
    if (conf.exists("reMatchSplitHits"))
      reMatchSplitHits_=conf.getParameter<bool>("reMatchSplitHits");
        if (conf.exists("usePropagatorForPCA"))
          usePropagatorForPCA_ = conf.getParameter<bool>("usePropagatorForPCA");
      }

  ~TrackProducerAlgorithm() {}
  
  void runWithCandidate(const TrackingGeometry *, 
            const MagneticField *, 
            const TrackCandidateCollection&,
            const TrajectoryFitter *,
            const Propagator *,
            const TransientTrackingRecHitBuilder*,
            const reco::BeamSpot&,
            AlgoProductCollection &);

  void runWithTrack(const TrackingGeometry *, 
            const MagneticField *, 
            const TrackCollection&,
            const TrajectoryFitter *,
            const Propagator *,
            const TransientTrackingRecHitBuilder*,
            const reco::BeamSpot&,
            AlgoProductCollection &);

  void runWithMomentum(const TrackingGeometry *, 
               const MagneticField *, 
               const TrackMomConstraintAssociationCollection&,
               const TrajectoryFitter *,
               const Propagator *,
               const TransientTrackingRecHitBuilder*,
               const reco::BeamSpot&,
               AlgoProductCollection &);

  void runWithVertex(const TrackingGeometry *, 
             const MagneticField *, 
             const VtxConstraintAssociationCollection&,
             const TrajectoryFitter *,
             const Propagator *,
             const TransientTrackingRecHitBuilder*,
             const reco::BeamSpot&,
             AlgoProductCollection &);

  void runWithTrackParameters(const TrackingGeometry *, 
                  const MagneticField *, 
                  const TrackParamConstraintAssociationCollection&,
                  const TrajectoryFitter *,
                  const Propagator *,
                  const TransientTrackingRecHitBuilder*,
                  const reco::BeamSpot&,
                  AlgoProductCollection &);

  bool buildTrack(const TrajectoryFitter *,
          const Propagator *,
          AlgoProductCollection& ,
          TransientTrackingRecHit::RecHitContainer&,
          TrajectoryStateOnSurface& ,
          const TrajectorySeed&,          
          float,
          const reco::BeamSpot&,
          SeedRef seedRef = SeedRef(),
          int qualityMask=0,
          signed char nLoops=0);


 private:
  reco::TrackBase::TrackAlgorithm algo_;
  reco::TrackBase::TrackAlgorithm originalAlgo_;
  reco::TrackBase::AlgoMask algoMask_;
  uint8_t stopReason_;

  bool reMatchSplitHits_;
  bool geometricInnerState_;
  bool usePropagatorForPCA_;

  TrajectoryStateOnSurface getInitialState(const T * theT,
                       TransientTrackingRecHit::RecHitContainer& hits,
                       const TrackingGeometry * theG,
                       const MagneticField * theMF);

};

#include "RecoTracker/TrackProducer/interface/TrackProducerAlgorithm.icc"

template <> bool
TrackProducerAlgorithm<reco::Track>::buildTrack(const TrajectoryFitter *,
                        const Propagator *,
                        AlgoProductCollection& ,
                        TransientTrackingRecHit::RecHitContainer&,
                        TrajectoryStateOnSurface& ,
                        const TrajectorySeed&,
                        float,
                        const reco::BeamSpot&,
                        SeedRef seedRef,
                        int qualityMask,
                        signed char nLoops);


template <> bool
TrackProducerAlgorithm<reco::GsfTrack>::buildTrack(const TrajectoryFitter *,
                           const Propagator *,
                           AlgoProductCollection& ,
                           TransientTrackingRecHit::RecHitContainer&,
                           TrajectoryStateOnSurface& ,
                           const TrajectorySeed&,
                           float,
                           const reco::BeamSpot&,
                           SeedRef seedRef,
                           int qualityMask,
                           signed char nLoops);

#endif