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