GsfTrack.h

Go to the documentation of this file.

#ifndef GsfTrackReco_GsfTrack_h
#define GsfTrackReco_GsfTrack_h

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/GsfTrackReco/interface/GsfTrackExtraFwd.h"
#include "FWCore/Utilities/interface/thread_safety_macros.h"

namespace reco {

  class GsfTrack : public Track {
  public:
    enum { dimensionMode = 3 };
    enum { covarianceSizeMode = dimensionMode * (dimensionMode + 1) / 2 };
    typedef math::Vector<dimensionMode>::type ParameterVectorMode;
    typedef math::Error<dimensionMode>::type CovarianceMatrixMode;
    GsfTrack();
    GsfTrack(double chi2, double ndof, const Point&, const Vector&, int charge, const CovarianceMatrix&);
    void setGsfExtra(const GsfTrackExtraRef& ref) { gsfExtra_ = ref; }
    const GsfTrackExtraRef& gsfExtra() const { return gsfExtra_; }

    void setMode(int chargeMode, const Vector& momentumMode, const CovarianceMatrixMode& covarianceMode);

    int chargeMode() const { return chargeMode_; }
    double qoverpMode() const { return chargeMode() / pMode(); }
    double thetaMode() const { return momentumMode_.theta(); }
    double lambdaMode() const { return M_PI / 2 - momentumMode_.theta(); }
    double pMode() const { return momentumMode_.R(); }
    double ptMode() const { return sqrt(momentumMode_.Perp2()); }
    double pxMode() const { return momentumMode_.x(); }
    double pyMode() const { return momentumMode_.y(); }
    double pzMode() const { return momentumMode_.z(); }
    double phiMode() const { return momentumMode_.Phi(); }
    double etaMode() const { return momentumMode_.Eta(); }

    const Vector& momentumMode() const { return momentumMode_; }

    ParameterVectorMode parametersMode() const { return ParameterVectorMode(qoverpMode(), lambdaMode(), phiMode()); }
    CovarianceMatrixMode covarianceMode() const {
      CovarianceMatrixMode m;
      fill(m);
      return m;
    }

    double parameterMode(int i) const { return parametersMode()[i]; }
    double covarianceMode(int i, int j) const { return covarianceMode_[covIndex(i, j)]; }
    double errorMode(int i) const { return sqrt(covarianceMode_[covIndex(i, i)]); }

    double qoverpModeError() const { return errorMode(i_qoverp); }
    double ptModeError() const {
      return (chargeMode() != 0)
                 ? sqrt(ptMode() * ptMode() * pMode() * pMode() / chargeMode() / chargeMode() *
                            covarianceMode(i_qoverp, i_qoverp) +
                        2 * ptMode() * pMode() / chargeMode() * pzMode() * covarianceMode(i_qoverp, i_lambda) +
                        pzMode() * pzMode() * covarianceMode(i_lambda, i_lambda))
                 : 1.e6;
    }
    double thetaModeError() const { return errorMode(i_lambda); }
    double lambdaModeError() const { return errorMode(i_lambda); }
    double etaModeError() const { return errorMode(i_lambda) * pMode() / ptMode(); }
    double phiModeError() const { return errorMode(i_phi); }

  private:
    CovarianceMatrixMode& fill CMS_THREAD_SAFE(CovarianceMatrixMode& v) const;

  private:
    GsfTrackExtraRef gsfExtra_;
    char chargeMode_;
    Vector momentumMode_;
    float covarianceMode_[covarianceSizeMode];
  };

}  // namespace reco

#endif