TrackCandidate.h

Go to the documentation of this file.

#ifndef DATAFORMATS_TRACKCANDIDATE_TRACKCANDIDATE_H
#define DATAFORMATS_TRACKCANDIDATE_TRACKCANDIDATE_H
 
#include "DataFormats/Common/interface/RefToBase.h"
#include "DataFormats/Common/interface/OwnVector.h"
#include "DataFormats/TrackCandidate/interface/TrajectoryStopReasons.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "FWCore/Utilities/interface/Range.h"
 
#include <utility>
 
class TrackCandidate {
public:
  typedef edm::OwnVector<TrackingRecHit> RecHitContainer;
 
  TrackCandidate()
      : rh_(), seed_(), state_(), seedRef_(), nLoops_(0), stopReason_((uint8_t)StopReason::UNINITIALIZED) {}
 
  explicit TrackCandidate(RecHitContainer& rh)
      : rh_(), seed_(), state_(), seedRef_(), nLoops_(0), stopReason_((uint8_t)StopReason::UNINITIALIZED) {
    rh_.swap(rh);
  }
 
  TrackCandidate(RecHitContainer& rh,
                 TrajectorySeed const& s,
                 PTrajectoryStateOnDet const& st,
                 signed char nLoops = 0,
                 uint8_t stopReason = (uint8_t)StopReason::UNINITIALIZED)
      : rh_(), seed_(s), state_(st), seedRef_(), nLoops_(nLoops), stopReason_(stopReason) {
    rh_.swap(rh);
  }
 
  TrackCandidate(RecHitContainer& rh,
                 TrajectorySeed const& s,
                 PTrajectoryStateOnDet const& st,
                 const edm::RefToBase<TrajectorySeed>& seedRef,
                 signed char nLoops = 0,
                 uint8_t stopReason = (uint8_t)StopReason::UNINITIALIZED)
      : rh_(), seed_(s), state_(st), seedRef_(seedRef), nLoops_(nLoops), stopReason_(stopReason) {
    rh_.swap(rh);
  }
 
  PTrajectoryStateOnDet const& trajectoryStateOnDet() const { return state_; }
 
  edm::Range<RecHitContainer::const_iterator> recHits() const { return {rh_.begin(), rh_.end()}; }
  auto nRecHits() const { return rh_.size(); }
 
  TrajectorySeed const& seed() const { return seed_; }
 
  bool isLooper() const { return (nLoops_ > 0); }
  signed char nLoops() const { return nLoops_; }
  uint8_t stopReason() const { return stopReason_; }
 
  void setNLoops(signed char value) { nLoops_ = value; }
  void setStopReason(uint8_t value) { stopReason_ = value; }
 
  edm::RefToBase<TrajectorySeed> seedRef() const { return seedRef_; }
 
  void setSeedRef(edm::RefToBase<TrajectorySeed>& seedRef) { seedRef_ = seedRef; }
 
private:
  RecHitContainer rh_;
  TrajectorySeed seed_;
  PTrajectoryStateOnDet state_;
  edm::RefToBase<TrajectorySeed> seedRef_;
  signed char nLoops_;
  uint8_t stopReason_;
};
#endif