MultiTrajectoryStateTransform.h

Go to the documentation of this file.

#ifndef MultiTrajectoryStateTransform_H
#define MultiTrajectoryStateTransform_H

#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackExtra.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"

class TrajectoryStateOnSurface;
class TrackingGeometry;
class Surface;
class MagneticField;
class TransverseImpactPointExtrapolator;

class MultiTrajectoryStateTransform {
private:
  typedef reco::GsfTrackExtra::LocalParameterVector ParameterVector;
  typedef reco::GsfTrackExtra::LocalCovarianceMatrix CovarianceMatrix;
  enum { dimension = reco::GsfTrackExtra::dimension };

public:

  MultiTrajectoryStateTransform () :
  geometry_(nullptr), field_(nullptr), extrapolator_(nullptr) {}
  MultiTrajectoryStateTransform (const TrackingGeometry* geom, const MagneticField* field) :
  geometry_(geom), field_(field), extrapolator_(nullptr) {}
  ~MultiTrajectoryStateTransform ();

  TrajectoryStateOnSurface innerStateOnSurface( const reco::GsfTrack& tk) const;
  TrajectoryStateOnSurface outerStateOnSurface( const reco::GsfTrack& tk) const;

  bool innerMomentumFromMode (const reco::GsfTrack& tk, GlobalVector& momentum) const;
  bool outerMomentumFromMode (const reco::GsfTrack& tk,GlobalVector& momentum) const;

  TrajectoryStateOnSurface extrapolatedState (const TrajectoryStateOnSurface tsos,
                          const GlobalPoint& point) const;

  TrajectoryStateOnSurface innerStateOnSurface( const reco::GsfTrack& tk, 
                        const TrackingGeometry& geom,
                        const MagneticField* field) const;
  TrajectoryStateOnSurface outerStateOnSurface( const reco::GsfTrack& tk, 
                        const TrackingGeometry& geom,
                        const MagneticField* field) const;

private:
  TrajectoryStateOnSurface stateOnSurface (const std::vector<double>& weights,
                       const std::vector<ParameterVector>& parameters,
                       const std::vector<CovarianceMatrix>& covariances,
                       const double& pzSign,
                       const Surface& surface,
                       const MagneticField* field) const;
  bool checkExtrapolator () const;
  bool checkGeometry () const;

private:
  const TrackingGeometry* geometry_;
  const MagneticField* field_;
  mutable TransverseImpactPointExtrapolator* extrapolator_;
};

#endif