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/TrackReco/interface/TrajectoryStopReasons.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"

#include <utility>

class TrackCandidate {
public:
  typedef edm::OwnVector<TrackingRecHit> RecHitContainer;
  typedef RecHitContainer::const_iterator const_iterator;
  typedef std::pair<const_iterator, const_iterator> range;

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

  range recHits() const { return std::make_pair(rh_.begin(), rh_.end()); }

  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