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_;
  };

}

#endif