TrackSelector.cc

Go to the documentation of this file.

#include "CommonTools/RecoAlgos/interface/TrackSelector.h"

using namespace reco;

namespace helper 
{

  TrackCollectionStoreManager::
  TrackCollectionStoreManager(const edm::Handle<reco::TrackCollection> & ) 
    :
    selTracks_( new reco::TrackCollection ),
    selTrackExtras_( new reco::TrackExtraCollection ),
    selHits_( new TrackingRecHitCollection ),
    selStripClusters_( new edmNew::DetSetVector<SiStripCluster> ),
    selPixelClusters_( new edmNew::DetSetVector<SiPixelCluster> ),
    rTracks_(), rTrackExtras_(), rHits_(),
    clusterStorer_(),
    idx_(0), hidx_(0),
    cloneClusters_ (true)
  {
  }
  
  
  //------------------------------------------------------------------
  //------------------------------------------------------------------
  void 
  TrackCollectionStoreManager::
  processTrack( const Track & trk ) 
  {
    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


  //------------------------------------------------------------------
  //------------------------------------------------------------------
  edm::OrphanHandle<reco::TrackCollection> 
  TrackCollectionStoreManager::
  put( edm::Event & evt ) {
    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_));
    return h; 
  }
  
} // end of namespace helper