#ifndef TrackReco_Track_h
#define TrackReco_Track_h
/** \class reco::Track Track.h DataFormats/TrackReco/interface/Track.h
 *
 * This class describes the reconstructed tracks that are stored in the AOD and
 * RECO. It also contains a reference to more detailed information about each
 * track, that is stoed in the TrackExtra object, available only in RECO.
 *
 * Note that most of the functions provided in this Track class rely on the existance of
 * the TrackExtra object, so will not work on AOD.
 *
 * The most useful functions are those provided in the TrackBase class from which this
 * inherits, all of which work on AOD.
 *
 * \author Luca Lista, INFN
 *
 *
 */
#include "DataFormats/TrackReco/interface/TrackBase.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/TrackReco/interface/TrackExtraFwd.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"
#include "FWCore/Utilities/interface/thread_safety_macros.h"

namespace reco {

  class Track : public TrackBase {
  public:
    /// default constructor
    Track() {}

    /// virtual destructor
    ~Track() override;

    /// constructor from fit parameters and error matrix
    Track(double chi2,
          double ndof,
          const Point& referencePoint,
          const Vector& momentum,
          int charge,
          const CovarianceMatrix&,
          TrackAlgorithm = undefAlgorithm,
          TrackQuality quality = undefQuality,
          float t0 = 0,
          float beta = 0,
          float covt0t0 = -1.,
          float covbetabeta = -1.);

    /// return true if the outermost hit is valid
    bool outerOk() const { return extra_.isNonnull() && extra_.isAvailable() && extra_->outerOk(); }

    /// return true if the innermost hit is valid
    bool innerOk() const { return extra_.isNonnull() && extra_.isAvailable() && extra_->innerOk(); }

    /// position of the innermost hit
    const math::XYZPoint& innerPosition() const { return extra_->innerPosition(); }

    /// momentum vector at the innermost hit position
    const math::XYZVector& innerMomentum() const { return extra_->innerMomentum(); }

    /// position of the outermost hit
    const math::XYZPoint& outerPosition() const { return extra_->outerPosition(); }

    /// momentum vector at the outermost hit position
    const math::XYZVector& outerMomentum() const { return extra_->outerMomentum(); }

    /// outermost trajectory state curvilinear errors
    CovarianceMatrix outerStateCovariance() const { return extra_->outerStateCovariance(); }

    /// innermost trajectory state curvilinear errors
    CovarianceMatrix innerStateCovariance() const { return extra_->innerStateCovariance(); }

    /// fill outermost trajectory state curvilinear errors
    CovarianceMatrix& fillOuter CMS_THREAD_SAFE(CovarianceMatrix& v) const { return extra_->fillOuter(v); }

    CovarianceMatrix& fillInner CMS_THREAD_SAFE(CovarianceMatrix& v) const { return extra_->fillInner(v); }

    /// DetId of the detector on which surface the outermost state is located
    unsigned int outerDetId() const { return extra_->outerDetId(); }

    /// DetId of the detector on which surface the innermost state is located
    unsigned int innerDetId() const { return extra_->innerDetId(); }

    /// Access to reconstructed hits on the track.
    auto recHits() const { return extra_->recHits(); }

    /// Iterator to first hit on the track.
    trackingRecHit_iterator recHitsBegin() const { return extra_->recHitsBegin(); }

    /// Iterator to last hit on the track.
    trackingRecHit_iterator recHitsEnd() const { return extra_->recHitsEnd(); }

    /// Get i-th hit on the track.
    TrackingRecHitRef recHit(size_t i) const { return extra_->recHit(i); }

    /// Get number of RecHits. (Warning, this includes invalid hits, which are not physical hits).
    size_t recHitsSize() const { return extra_->recHitsSize(); }

    /// x coordinate of momentum vector at the outermost hit position
    double outerPx() const { return extra_->outerPx(); }

    /// y coordinate of momentum vector at the outermost hit position
    double outerPy() const { return extra_->outerPy(); }

    /// z coordinate of momentum vector at the outermost hit position
    double outerPz() const { return extra_->outerPz(); }

    /// x coordinate of the outermost hit position
    double outerX() const { return extra_->outerX(); }

    /// y coordinate of the outermost hit position
    double outerY() const { return extra_->outerY(); }

    /// z coordinate of the outermost hit position
    double outerZ() const { return extra_->outerZ(); }

    /// magnitude of momentum vector at the outermost hit position
    double outerP() const { return extra_->outerP(); }

    /// transverse momentum at the outermost hit position
    double outerPt() const { return extra_->outerPt(); }

    /// azimuthal angle of the  momentum vector at the outermost hit position
    double outerPhi() const { return extra_->outerPhi(); }

    /// pseudorapidity of the  momentum vector at the outermost hit position
    double outerEta() const { return extra_->outerEta(); }

    /// polar angle of the  momentum vector at the outermost hit position
    double outerTheta() const { return extra_->outerTheta(); }

    /// polar radius of the outermost hit position
    double outerRadius() const { return extra_->outerRadius(); }

    /// set reference to "extra" object
    void setExtra(const TrackExtraRef& ref) { extra_ = ref; }

    /// reference to "extra" object
    const TrackExtraRef& extra() const { return extra_; }

    /// Number of valid hits on track.
    unsigned short found() const { return numberOfValidHits(); }

    /// Number of lost (=invalid) hits on track.
    unsigned short lost() const { return numberOfLostHits(); }

    /// direction of how the hits were sorted in the original seed
    const PropagationDirection& seedDirection() const { return extra_->seedDirection(); }

    /**  return the edm::reference to the trajectory seed in the original
     *   seeds collection. If the collection has been dropped from the
     *   Event, the reference may be invalid. Its validity should be tested,
     *   before the reference is actually used.
     */
    const edm::RefToBase<TrajectorySeed>& seedRef() const { return extra_->seedRef(); }

    /// get the residuals
    const TrackResiduals& residuals() const { return extra_->residuals(); }

    // Check validity of track extra and rechits
    bool recHitsOk() const { return extra_.isNonnull() && extra_.isAvailable() && extra_->recHitsOk(); }

  private:
    /// Reference to additional information stored only on RECO.
    TrackExtraRef extra_;
  };

}  // namespace reco

#endif