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;

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

private:
  static 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);
  bool checkExtrapolator() const;
  bool checkGeometry() const;

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

#endif