GsfTrackProducerBase.h

Go to the documentation of this file.

#ifndef GsfTrackProducerBase_h
#define GsfTrackProducerBase_h

#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackFwd.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackExtra.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackExtraFwd.h"
#include "DataFormats/GsfTrackReco/interface/GsfComponent5D.h"
#include "RecoTracker/TrackProducer/interface/TrackProducerBase.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "TrackingTools/GsfTools/interface/MultiTrajectoryStateMode.h"

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

class TrajectoryStateOnSurface;
class Propagator;
class TransverseImpactPointExtrapolator;
class TrajectoryStateClosestToBeamLineBuilder;

class GsfTrackProducerBase : public TrackProducerBase<reco::GsfTrack> {
public:
  explicit GsfTrackProducerBase(bool trajectoryInEvent, bool split)
      : TrackProducerBase<reco::GsfTrack>(trajectoryInEvent), useSplitting(split) {}

  virtual void putInEvt(edm::Event&,
                        const Propagator* prop,
                        const MeasurementTracker* measTk,
                        std::unique_ptr<TrackingRecHitCollection>&,
                        std::unique_ptr<reco::GsfTrackCollection>&,
                        std::unique_ptr<reco::TrackExtraCollection>&,
                        std::unique_ptr<reco::GsfTrackExtraCollection>&,
                        std::unique_ptr<std::vector<Trajectory> >&,
                        AlgoProductCollection&,
                        TransientTrackingRecHitBuilder const*,
                        const reco::BeamSpot&,
                        const TrackerTopology* ttopo);

protected:
  void fillStates(TrajectoryStateOnSurface tsos, std::vector<reco::GsfComponent5D>& states) const;
  void fillMode(reco::GsfTrack& track,
                const TrajectoryStateOnSurface innertsos,
                const Propagator& gsfProp,
                const TransverseImpactPointExtrapolator& tipExtrapolator,
                TrajectoryStateClosestToBeamLineBuilder& tscblBuilder,
                const reco::BeamSpot& bs) const;

private:
  void localParametersFromQpMode(const TrajectoryStateOnSurface tsos,
                                 AlgebraicVector5& parameters,
                                 AlgebraicSymMatrix55& covariance) const;
  bool computeModeAtTM(const TrajectoryMeasurement& tm,
                       reco::GsfTrackExtra::Point& position,
                       reco::GsfTrackExtra::Vector& momentum,
                       Measurement1D& deltaP) const;

private:
  bool useSplitting;
};

#endif