DataFormats/JetReco/interface/TrackExtrapolation.h - master - githubmirror/cmssw

DataFormats/JetReco/interface/TrackExtrapolation.h

55 строк2 KB

Matti Kortelainen

Move stored types in io_v1 namespace, reset class versions to 3

22 май 2026, 18:18

22 май 2026, 18:18909f324

#ifndef DataFormats_TrackReco_interface_TrackExtrapolation_h
#define DataFormats_TrackReco_interface_TrackExtrapolation_h
/* \class reco::TrackExtrapolation TrackExtrapolation.h DataFormats/TrackReco/interface/TrackExtrapolation.h
*
* This class represents the track state at several radii (specified by user in producer).
* It stores a TrackRef to the original track, as well as vectors of the positions and  momenta
* of the track at the various radii. 
*
* \author Salvatore Rappoccio, JHU
*
*
*/

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Math/interface/Point3D.h"

#include <vector>

namespace reco {
  namespace io_v1 {
    class TrackExtrapolation {
      // Next two typedefs use double in ROOT 6 rather than Double32_t due to a bug in ROOT 5,
      // which otherwise would make ROOT5 files unreadable in ROOT6.  This does not increase
      // the size on disk, because due to the bug, double was actually stored on disk in ROOT 5.
      typedef ROOT::Math::PositionVector3D<ROOT::Math::Cartesian3D<double> > Point;
      typedef ROOT::Math::DisplacementVector3D<ROOT::Math::Cartesian3D<double> > Vector;

    public:
      TrackExtrapolation() {}
      TrackExtrapolation(reco::TrackRef const& track, std::vector<Point> const& pos, std::vector<Vector> const& mom)
          : track_(track) {
        pos_.resize(pos.size());
        copy(pos.begin(), pos.end(), pos_.begin());
        mom_.resize(mom.size());
        copy(mom.begin(), mom.end(), mom_.begin());
      }

      ~TrackExtrapolation() {}

      reco::TrackRef const& track() const { return track_; }
      std::vector<Point> const& positions() const { return pos_; }
      std::vector<Vector> const& momenta() const { return mom_; }

    protected:
      reco::TrackRef track_;
      std::vector<Point> pos_;
      std::vector<Vector> mom_;
    };
  }  // namespace io_v1
  using TrackExtrapolation = io_v1::TrackExtrapolation;
}  // namespace reco

#endif