helper::TrackCollectionStoreManager Struct Reference

Class to manage copying of RecHits and Clusters from Tracks. More...

#include <TrackSelector.h>

Public Types
typedef reco::TrackCollection	collection

Public Member Functions
template<typename I >
void	cloneAndStore (const I &begin, const I &end, edm::Event &evt)
	Method to clone tracks, track extras and their hits and clusters. typename I = this is an interator over a track collection, **I needs to dereference into a Track. More...
bool	cloneClusters ()
edm::OrphanHandle< reco::TrackCollection >	put (edm::Event &evt)
	Put tracks, track extras and hits+clusters into the event. More...
void	setCloneClusters (bool w)
size_t	size () const
	Get the size. More...
	TrackCollectionStoreManager (const edm::Handle< reco::TrackCollection > &)

Private Member Functions
void	processTrack (const reco::Track &trk)
	Process a single track. More...

Private Attributes
bool	cloneClusters_
	Clone clusters, or not? Default: true. More...
ClusterStorer	clusterStorer_
size_t	hidx_
	index to tracking rec hits More...
size_t	idx_
	index to track extra coll More...
TrackingRecHitRefProd	rHits_
reco::TrackExtraRefProd	rTrackExtras_
reco::TrackRefProd	rTracks_
std::unique_ptr< TrackingRecHitCollection >	selHits_
std::unique_ptr< edmNew::DetSetVector< Phase2TrackerCluster1D > >	selPhase2OTClusters_
std::unique_ptr< edmNew::DetSetVector< SiPixelCluster > >	selPixelClusters_
std::unique_ptr< edmNew::DetSetVector< SiStripCluster > >	selStripClusters_
std::unique_ptr< reco::TrackExtraCollection >	selTrackExtras_
std::unique_ptr< reco::TrackCollection >	selTracks_

Detailed Description

Class to manage copying of RecHits and Clusters from Tracks.

Definition at line 35 of file TrackSelector.h.

Member Typedef Documentation

collection

typedef reco::TrackCollection helper::TrackCollectionStoreManager::collection

Definition at line 37 of file TrackSelector.h.

Constructor & Destructor Documentation

TrackCollectionStoreManager()

helper::TrackCollectionStoreManager::TrackCollectionStoreManager

(

const edm::Handle< reco::TrackCollection > &

)

Definition at line 7 of file TrackSelector.cc.

      : selTracks_(new reco::TrackCollection),
        selTrackExtras_(new reco::TrackExtraCollection),
        selHits_(new TrackingRecHitCollection),
        selStripClusters_(new edmNew::DetSetVector<SiStripCluster>),
        selPixelClusters_(new edmNew::DetSetVector<SiPixelCluster>),
        selPhase2OTClusters_(new edmNew::DetSetVector<Phase2TrackerCluster1D>),
        rTracks_(),
        rTrackExtras_(),
        rHits_(),
        clusterStorer_(),
        idx_(0),
        hidx_(0),
        cloneClusters_(true) {}

Member Function Documentation

cloneAndStore()

template<typename I >

void helper::TrackCollectionStoreManager::cloneAndStore	(	const I &	begin,
		const I &	end,
		edm::Event &	evt
	)

Method to clone tracks, track extras and their hits and clusters. typename I = this is an interator over a track collection, **I needs to dereference into a Track.

< index to track extra coll

< index to tracking rec hits

Definition at line 101 of file TrackSelector.h.

References helper::ClusterStorer::clear(), clusterStorer_, hidx_, Exhume::I, mps_fire::i, idx_, helper::ClusterStorer::processAllClusters(), processTrack(), rHits_, rTrackExtras_, rTracks_, selPhase2OTClusters_, selPixelClusters_, selStripClusters_, and trk.

                                                                                             {
    using namespace reco;

    rTracks_ = evt.template getRefBeforePut<TrackCollection>();
    rTrackExtras_ = evt.template getRefBeforePut<TrackExtraCollection>();
    rHits_ = evt.template getRefBeforePut<TrackingRecHitCollection>();
    //--- New: save clusters too
    edm::RefProd<edmNew::DetSetVector<SiPixelCluster> > rPixelClusters =
        evt.template getRefBeforePut<edmNew::DetSetVector<SiPixelCluster> >();
    edm::RefProd<edmNew::DetSetVector<SiStripCluster> > rStripClusters =
        evt.template getRefBeforePut<edmNew::DetSetVector<SiStripCluster> >();
    edm::RefProd<edmNew::DetSetVector<Phase2TrackerCluster1D> > rPhase2OTClusters =
        evt.template getRefBeforePut<edmNew::DetSetVector<Phase2TrackerCluster1D> >();

    //--- Indices into collections handled with RefProd
    idx_ = 0;   
    hidx_ = 0;  
    clusterStorer_.clear();

    //--- Loop over tracks
    for (I i = begin; i != end; ++i) {
      //--- Whatever type the iterator i is, deref to reco::Track &
      const reco::Track& trk = **i;
      //--- Clone this track, and store references aside
      processTrack(trk);
    }
    //--- Clone the clusters and fixup refs
    clusterStorer_.processAllClusters(*selPixelClusters_,
                                      rPixelClusters,
                                      *selStripClusters_,
                                      rStripClusters,
                                      *selPhase2OTClusters_,
                                      rPhase2OTClusters);
  }

cloneClusters()

bool helper::TrackCollectionStoreManager::cloneClusters ( )

inline

Use these to turn off/on the cloning of clusters. The default is to clone them. To not clone (and save space in a quick local job, do: setCloneClusters(false);

Definition at line 47 of file TrackSelector.h.

References cloneClusters_.

Referenced by processTrack().

{ return cloneClusters_; }

processTrack()

void helper::TrackCollectionStoreManager::processTrack ( const reco::Track &  trk )

private

Process a single track.

Definition at line 25 of file TrackSelector.cc.

References helper::ClusterStorer::addCluster(), cloneClusters(), clusterStorer_, hidx_, idx_, TrackingRecHit::isValid(), rHits_, rTrackExtras_, selHits_, selTrackExtras_, selTracks_, reco::TrackExtraBase::setHits(), DetId::Tracker, and trk.

Referenced by cloneAndStore().

                                                                 {
    selTracks_->push_back(Track(trk));
    selTracks_->back().setExtra(TrackExtraRef(rTrackExtras_, idx_++));
    selTrackExtras_->push_back(TrackExtra(trk.outerPosition(),
                                          trk.outerMomentum(),
                                          trk.outerOk(),
                                          trk.innerPosition(),
                                          trk.innerMomentum(),
                                          trk.innerOk(),
                                          trk.outerStateCovariance(),
                                          trk.outerDetId(),
                                          trk.innerStateCovariance(),
                                          trk.innerDetId(),
                                          trk.seedDirection()));
    TrackExtra& tx = selTrackExtras_->back();
    auto const firstHitIndex = hidx_;
    unsigned int nHitsAdded = 0;
    for (trackingRecHit_iterator hit = trk.recHitsBegin(); hit != trk.recHitsEnd(); ++hit, ++hidx_) {
      selHits_->push_back((*hit)->clone());
      TrackingRecHit* newHit = &(selHits_->back());
      ++nHitsAdded;

      //--- Skip the rest for this hit if we don't want to clone the cluster.
      //--- The copy constructer in the rec hit will copy the link properly.
      if (cloneClusters() && newHit->isValid() && ((*hit)->geographicalId().det() == DetId::Tracker)) {
        clusterStorer_.addCluster(*selHits_, hidx_);
      }
    }  // end of for loop over tracking rec hits on this track
    tx.setHits(rHits_, firstHitIndex, nHitsAdded);

  }  // end of track, and function

put()

edm::OrphanHandle< reco::TrackCollection > helper::TrackCollectionStoreManager::put

(

edm::Event &

evt

)

Put tracks, track extras and hits+clusters into the event.

Definition at line 60 of file TrackSelector.cc.

References h, eostools::move(), edm::Event::put(), selHits_, selPhase2OTClusters_, selPixelClusters_, selStripClusters_, selTrackExtras_, and selTracks_.

                                                                                     {
    edm::OrphanHandle<reco::TrackCollection> h = evt.put(std::move(selTracks_));
    evt.put(std::move(selTrackExtras_));
    evt.put(std::move(selHits_));
    evt.put(std::move(selStripClusters_));
    evt.put(std::move(selPixelClusters_));
    evt.put(std::move(selPhase2OTClusters_));
    return h;
  }

setCloneClusters()

void helper::TrackCollectionStoreManager::setCloneClusters ( bool  w )

inline

Definition at line 48 of file TrackSelector.h.

References cloneClusters_, and w().

{ cloneClusters_ = w; }

size()

size_t helper::TrackCollectionStoreManager::size ( void  ) const

inline

Get the size.

Definition at line 58 of file TrackSelector.h.

References selTracks_.

Referenced by ntupleDataFormat._Collection::__iter__(), and ntupleDataFormat._Collection::__len__().

{ return selTracks_->size(); }

Member Data Documentation

cloneClusters_

bool helper::TrackCollectionStoreManager::cloneClusters_

private

Clone clusters, or not? Default: true.

Definition at line 90 of file TrackSelector.h.

Referenced by cloneClusters(), and setCloneClusters().

clusterStorer_

ClusterStorer helper::TrackCollectionStoreManager::clusterStorer_

private

Helper to treat copies of selected clusters and make the hits refer to the output cluster collections:

Definition at line 83 of file TrackSelector.h.

Referenced by cloneAndStore(), and processTrack().

hidx_

size_t helper::TrackCollectionStoreManager::hidx_

private

index to tracking rec hits

Definition at line 87 of file TrackSelector.h.

Referenced by cloneAndStore(), and processTrack().

idx_

size_t helper::TrackCollectionStoreManager::idx_

private

index to track extra coll

Definition at line 86 of file TrackSelector.h.

Referenced by cloneAndStore(), and processTrack().

rHits_

TrackingRecHitRefProd helper::TrackCollectionStoreManager::rHits_

private

Definition at line 80 of file TrackSelector.h.

Referenced by cloneAndStore(), and processTrack().

rTrackExtras_

reco::TrackExtraRefProd helper::TrackCollectionStoreManager::rTrackExtras_

private

Definition at line 79 of file TrackSelector.h.

Referenced by cloneAndStore(), and processTrack().

rTracks_

reco::TrackRefProd helper::TrackCollectionStoreManager::rTracks_

private

Definition at line 78 of file TrackSelector.h.

Referenced by cloneAndStore().

selHits_

std::unique_ptr<TrackingRecHitCollection> helper::TrackCollectionStoreManager::selHits_

private

Definition at line 72 of file TrackSelector.h.

Referenced by processTrack(), and put().

selPhase2OTClusters_

std::unique_ptr<edmNew::DetSetVector<Phase2TrackerCluster1D> > helper::TrackCollectionStoreManager::selPhase2OTClusters_

private

Definition at line 75 of file TrackSelector.h.

Referenced by cloneAndStore(), and put().

selPixelClusters_

std::unique_ptr<edmNew::DetSetVector<SiPixelCluster> > helper::TrackCollectionStoreManager::selPixelClusters_

private

Definition at line 74 of file TrackSelector.h.

Referenced by cloneAndStore(), and put().

selStripClusters_

std::unique_ptr<edmNew::DetSetVector<SiStripCluster> > helper::TrackCollectionStoreManager::selStripClusters_

private

Definition at line 73 of file TrackSelector.h.

Referenced by cloneAndStore(), and put().

selTrackExtras_

std::unique_ptr<reco::TrackExtraCollection> helper::TrackCollectionStoreManager::selTrackExtras_

private

Definition at line 71 of file TrackSelector.h.

Referenced by processTrack(), and put().

selTracks_

std::unique_ptr<reco::TrackCollection> helper::TrackCollectionStoreManager::selTracks_

private

Definition at line 70 of file TrackSelector.h.

Referenced by processTrack(), put(), and size().