KalmanAlignmentTrackRefitter.h

Go to the documentation of this file.

#ifndef Alignment_KalmanAlignmentAlgorithm_KalmanAlignmentTrackRefitter_h
#define Alignment_KalmanAlignmentAlgorithm_KalmanAlignmentTrackRefitter_h

#include "RecoTracker/TrackProducer/interface/TrackProducerBase.h"
#include "RecoTracker/TrackProducer/interface/TrackProducerAlgorithm.h"

#include "Alignment/CommonAlignment/interface/AlignableNavigator.h"

#include "Alignment/KalmanAlignmentAlgorithm/interface/KalmanAlignmentTracklet.h"
#include "Alignment/KalmanAlignmentAlgorithm/interface/KalmanAlignmentSetup.h"

#include "TrackingTools/TransientTrack/interface/TransientTrack.h"

#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHitBuilder.h"

class TrajectoryFitter;

namespace reco { class BeamSpot; }



class KalmanAlignmentTrackRefitter : public TrackProducerBase<reco::Track>
//class KalmanAlignmentTrackRefitter : public TrackProducerBase
{

public:

  typedef std::vector< KalmanAlignmentSetup* > AlignmentSetupCollection;
  typedef KalmanAlignmentSetup::SortingDirection SortingDirection;

  typedef edm::OwnVector< TrackingRecHit > RecHitContainer;

  typedef KalmanAlignmentTracklet::TrajTrackPairCollection TrajTrackPairCollection;
  typedef KalmanAlignmentTracklet::TrackletPtr TrackletPtr;
  typedef std::vector< TrackletPtr > TrackletCollection;

  typedef AlignmentAlgorithmBase::ConstTrajTrackPair ConstTrajTrackPair;
  typedef AlignmentAlgorithmBase::ConstTrajTrackPairCollection ConstTrajTrackPairCollection;

  KalmanAlignmentTrackRefitter( const edm::ParameterSet& config, AlignableNavigator* navigator );

  ~KalmanAlignmentTrackRefitter( void );

  TrackletCollection refitTracks( const edm::EventSetup& eventSetup,
                  const AlignmentSetupCollection& algoSetups,
                  const ConstTrajTrackPairCollection& tracks,
                  const reco::BeamSpot* beamSpot );

  virtual void produce( edm::Event&, const edm::EventSetup& ) {}

private:

  TrajTrackPairCollection refitSingleTracklet( const TrackingGeometry* geometry,
                           const MagneticField* magneticField,
                           const TrajectoryFitter* fitter,
                           const Propagator* propagator,
                           const TransientTrackingRecHitBuilder* recHitBuilder,
                           const reco::TransientTrack& originalTrack,
                           RecHitContainer& recHits,
                           const reco::BeamSpot* beamSpot,
                           const SortingDirection& sortingDir,
                           bool useExternalEstimate,
                           bool reuseMomentumEstimate,
                           const std::string identifier = std::string("RefitSingle_") );

  void sortRecHits( RecHitContainer& hits,
            const TransientTrackingRecHitBuilder* builder,
            const SortingDirection& sortingDir ) const;


  bool rejectTrack( const reco::Track* track ) const;


  void debugTrackData( const std::string identifier,
               const Trajectory* traj,
               const reco::Track* track,
               const reco::BeamSpot* bs );

  TrackProducerAlgorithm<reco::Track> theRefitterAlgo;
  //TrackProducerAlgorithm theRefitterAlgo;
  AlignableNavigator* theNavigator;
  bool theDebugFlag;
};


#endif