GsfTrackExtra.h

Go to the documentation of this file.

#ifndef GsfTrackReco_GsfTrackExtra_h
#define GsfTrackReco_GsfTrackExtra_h

#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Math/interface/Point3D.h"
#include "DataFormats/Math/interface/Vector.h"
// #include "DataFormats/Math/interface/Error.h"
#include "DataFormats/GsfTrackReco/interface/GsfComponent5D.h"
#include "DataFormats/GsfTrackReco/interface/GsfTangent.h"

#include <iostream>

namespace reco {
  class GsfTrackExtra {
  public:
    enum { dimension = 5 };
    typedef math::Vector<dimension>::type LocalParameterVector;
    typedef math::Error<dimension>::type LocalCovarianceMatrix;
    typedef math::XYZPoint Point;
    typedef math::XYZVector Vector;

    GsfTrackExtra() {}
    GsfTrackExtra(const std::vector<GsfComponent5D>& outerStates,
                  const double& outerLocalPzSign,
                  const std::vector<GsfComponent5D>& innerStates,
                  const double& innerLocalPzSign,
                  const std::vector<GsfTangent>& tangents);
    double outerStateLocalPzSign() const { return positiveOuterStatePz_ ? 1. : -1.; }
    std::vector<double> outerStateWeights() const { return weights(outerStates_); }
    std::vector<LocalParameterVector> outerStateLocalParameters() const { return parameters(outerStates_); }
    std::vector<LocalCovarianceMatrix> outerStateCovariances() const { return covariances(outerStates_); }
    double innerStateLocalPzSign() const { return positiveInnerStatePz_ ? 1. : -1.; }
    std::vector<double> innerStateWeights() const { return weights(innerStates_); }
    std::vector<LocalParameterVector> innerStateLocalParameters() const { return parameters(innerStates_); }
    std::vector<LocalCovarianceMatrix> innerStateCovariances() const { return covariances(innerStates_); }
    inline unsigned int tangentsSize() const { return tangents_.size(); }
    const std::vector<GsfTangent>& tangents() const { return tangents_; }
    const Point& tangentPosition(unsigned int index) const { return tangents_[index].position(); }
    const Vector& tangentMomentum(unsigned int index) const { return tangents_[index].momentum(); }
    Measurement1D tangentDeltaP(unsigned int index) const { return tangents_[index].deltaP(); }

  private:
    std::vector<double> weights(const std::vector<GsfComponent5D>& states) const;
    std::vector<LocalParameterVector> parameters(const std::vector<GsfComponent5D>& states) const;
    std::vector<LocalCovarianceMatrix> covariances(const std::vector<GsfComponent5D>& states) const;

  private:
    std::vector<GsfComponent5D> outerStates_;
    bool positiveOuterStatePz_;
    std::vector<GsfComponent5D> innerStates_;
    bool positiveInnerStatePz_;
    std::vector<GsfTangent> tangents_;
  };

}  // namespace reco

#endif