TrackExtraBase.h

Go to the documentation of this file.

#ifndef TrackReco_TrackExtraBase_h
#define TrackReco_TrackExtraBase_h

#include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"

namespace reco
{

class TrackExtraBase
{

public:
    TrackExtraBase() : m_firstHit(-1), m_nHits(0) { }

    void setHits(TrackingRecHitRefProd const & prod, unsigned firstH, unsigned int nH) {
        m_hitCollection.pushBackItem(prod.refCore(),true);
        m_firstHit =firstH;  m_nHits=nH;
    }

    void add(const TrackingRecHitRef &ref) {
      m_hitCollection.pushBackItem(ref.refCore(), true);
      if (m_nHits==0) {
        m_firstHit = ref.key();
      }
      assert(m_nHits== ref.key()-m_firstHit);
      ++m_nHits;   
    }


    unsigned int firstRecHit() const {
      return m_firstHit;
    }

    unsigned int recHitsSize() const {
        return m_nHits;
    }


    trackingRecHit_iterator recHitsBegin() const {
        return recHitsProduct().data().begin()+firstRecHit();
    }

    trackingRecHit_iterator recHitsEnd() const {
        return recHitsBegin()+recHitsSize();
    }

    TrackingRecHitRef recHitRef(unsigned int i) const {                                                               
        return TrackingRecHitRef(m_hitCollection,m_firstHit+i);
    }

    TrackingRecHitRef recHit(unsigned int i) const {
        return recHitRef(i);
    }

    TrackingRecHitCollection const & recHitsProduct() const {
      return *edm::getProduct<TrackingRecHitCollection>(m_hitCollection);

    }

private:

    edm::RefCore m_hitCollection;
    unsigned int m_firstHit;
    unsigned int m_nHits;

};

}// namespace reco

#endif