Inheritance diagram for CosmicTrackSelector:

Public Member Functions
	CosmicTrackSelector (const edm::ParameterSet &cfg)

	~CosmicTrackSelector () override

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Private Types
typedef math::XYZPoint	Point

Private Member Functions
void	produce (edm::Event &evt, const edm::EventSetup &es) override

bool	select (const reco::BeamSpot &vertexBeamSpot, const reco::Track &tk)

Private Attributes
edm::EDGetTokenT< reco::BeamSpot >	beamspot_

double	chi2n_par_

bool	copyExtras_

bool	copyTrajectories_

bool	keepAllTracks_

double	max_d0_

double	max_eta_

uint32_t	max_lostLayers_

double	max_z0_

uint32_t	min_3Dlayers_

uint32_t	min_layers_

uint32_t	min_nHit_

uint32_t	min_nPixelHit_

double	min_pt_

TrackBase::TrackQuality	qualityToSet_

std::vector< double >	res_par_

TrackingRecHitRefProd	rHits_

reco::TrackExtraRefProd	rTrackExtras_

reco::TrackRefProd	rTracks_

edm::RefProd< std::vector< Trajectory > >	rTrajectories_

std::unique_ptr< TrackingRecHitCollection >	selHits_

std::unique_ptr< reco::TrackExtraCollection >	selTrackExtras_

std::unique_ptr< reco::TrackCollection >	selTracks_

std::unique_ptr< std::vector< const Trajectory * > >	selTrajPtrs_

std::unique_ptr< std::vector< Trajectory > >	selTrajs_

std::unique_ptr< TrajTrackAssociationCollection >	selTTAss_

bool	setQualityBit_

edm::EDGetTokenT< reco::TrackCollection >	src_

edm::EDGetTokenT< TrajTrackAssociationCollection >	srcTass_

edm::EDGetTokenT< std::vector< Trajectory > >	srcTraj_

std::vector< reco::TrackRef >	trackRefs_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

selects a subset of a track collection, copying extra information on demand

Author: Paolo Azzurri, Giovanni Petrucciani

Definition at line 32 of file CosmicTrackSelector.cc.

Member Typedef Documentation

typedef math::XYZPoint CosmicTrackSelector::Point

private

Definition at line 41 of file CosmicTrackSelector.cc.

Constructor & Destructor Documentation

CosmicTrackSelector::CosmicTrackSelector ( const edm::ParameterSet & cfg )

explicit

Definition at line 99 of file CosmicTrackSelector.cc.

References met_cff::alias, copyExtras_, copyTrajectories_, Exception, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), keepAllTracks_, qualityToSet_, setQualityBit_, srcTass_, srcTraj_, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                       :
   src_( consumes<reco::TrackCollection>( cfg.getParameter<edm::InputTag>( "src" ) ) ),
   beamspot_( consumes<reco::BeamSpot>( cfg.getParameter<edm::InputTag>( "beamspot" ) ) ),
   copyExtras_(cfg.getUntrackedParameter<bool>("copyExtras", false)),
   copyTrajectories_(cfg.getUntrackedParameter<bool>("copyTrajectories", false)),
   keepAllTracks_( cfg.exists("keepAllTracks") ?
           cfg.getParameter<bool>("keepAllTracks") :
           false ),  // as this is what you expect from a well behaved selector
   setQualityBit_( false ),
   qualityToSet_( TrackBase::undefQuality ),
   chi2n_par_( cfg.getParameter<double>("chi2n_par") ),
   // Impact parameter absolute cuts.
   max_d0_(cfg.getParameter<double>("max_d0")),
   max_z0_(cfg.getParameter<double>("max_z0")),
   // Track parameter cuts.
   min_pt_(cfg.getParameter<double>("min_pt")),
   max_eta_(cfg.getParameter<double>("max_eta")),
   // Cut on number of valid hits
   min_nHit_(cfg.getParameter<uint32_t>("min_nHit")),
   // Cut on number of valid hits
   min_nPixelHit_(cfg.getParameter<uint32_t>("min_nPixelHit")),
   // Cuts on numbers of layers with hits/3D hits/lost hits.
   min_layers_(cfg.getParameter<uint32_t>("minNumberLayers") ),
   min_3Dlayers_(cfg.getParameter<uint32_t>("minNumber3DLayers") ),
   max_lostLayers_(cfg.getParameter<uint32_t>("maxNumberLostLayers") )
 {
   if (cfg.exists("qualityBit")) {
     std::string qualityStr = cfg.getParameter<std::string>("qualityBit");
     if (qualityStr != "") {
       setQualityBit_ = true;
       qualityToSet_  = TrackBase::qualityByName(cfg.getParameter<std::string>("qualityBit"));
     }
   }
   if (keepAllTracks_ && !setQualityBit_) throw cms::Exception("Configuration") << 
     "If you set 'keepAllTracks' to true, you must specify which qualityBit to set.\n";
   if (setQualityBit_ && (qualityToSet_ == TrackBase::undefQuality)) throw cms::Exception("Configuration") <<
     "You can't set the quality bit " << cfg.getParameter<std::string>("qualityBit") << " as it is 'undefQuality' or unknown.\n";
   
   std::string alias( cfg.getParameter<std::string>( "@module_label" ) );
   produces<reco::TrackCollection>().setBranchAlias( alias + "Tracks");
   if (copyExtras_) {
     produces<reco::TrackExtraCollection>().setBranchAlias( alias + "TrackExtras");
     produces<TrackingRecHitCollection>().setBranchAlias( alias + "RecHits");
   }
   if (copyTrajectories_) {
     srcTraj_ = consumes<std::vector<Trajectory> >(cfg.getParameter<edm::InputTag>( "src" ));
     srcTass_ = consumes<TrajTrackAssociationCollection>(cfg.getParameter<edm::InputTag>( "src" ));
     produces< std::vector<Trajectory> >().setBranchAlias( alias + "Trajectories");
     produces< TrajTrackAssociationCollection >().setBranchAlias( alias + "TrajectoryTrackAssociations");
   }
   
 }

CosmicTrackSelector::~CosmicTrackSelector ( )

override

Definition at line 152 of file CosmicTrackSelector.cc.

152 {

153 }

Member Function Documentation

void CosmicTrackSelector::produce	(	edm::Event &	evt,
		const edm::EventSetup &	es
	)

overrideprivate

Definition at line 155 of file CosmicTrackSelector.cc.

References beamspot_, edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, copyExtras_, copyTrajectories_, edm::friendlyname::emptyString(), edm::AssociationMap< Tag >::end(), reco::Track::extra(), edm::AssociationMap< Tag >::find(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), mps_fire::i, reco::Track::innerDetId(), reco::Track::innerMomentum(), reco::Track::innerOk(), reco::Track::innerPosition(), reco::Track::innerStateCovariance(), keepAllTracks_, edm::Ref< C, T, F >::key(), match(), eostools::move(), gen::n, convertSQLiteXML::ok, reco::Track::outerDetId(), reco::Track::outerMomentum(), reco::Track::outerOk(), reco::Track::outerPosition(), reco::Track::outerStateCovariance(), edm::Event::productGetter(), edm::Event::put(), qualityToSet_, reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), reco::Track::residuals(), rHits_, rTrackExtras_, rTracks_, rTrajectories_, reco::Track::seedDirection(), reco::Track::seedRef(), select(), selHits_, selTrackExtras_, selTracks_, selTrajs_, selTTAss_, reco::TrackExtraBase::setHits(), setQualityBit_, reco::TrackExtra::setResiduals(), reco::TrackExtraBase::setTrajParams(), src_, srcTass_, srcTraj_, AlCaHLTBitMon_QueryRunRegistry::string, and trackRefs_.

 {
   using namespace std; 
   using namespace edm;
   using namespace reco;
   
   Handle<TrackCollection> hSrcTrack;
   Handle< vector<Trajectory> > hTraj;
   Handle< vector<Trajectory> > hTrajP;
   Handle< TrajTrackAssociationCollection > hTTAss;
   
   // looking for the beam spot
   edm::Handle<reco::BeamSpot> hBsp;
   evt.getByToken(beamspot_, hBsp);
   reco::BeamSpot vertexBeamSpot;
   vertexBeamSpot = *hBsp;
   
   // Get tracks 
   evt.getByToken( src_, hSrcTrack );
   
   selTracks_ = std::make_unique<TrackCollection>();
   rTracks_ = evt.getRefBeforePut<TrackCollection>();      
   if (copyExtras_) {
     selTrackExtras_ = std::make_unique<TrackExtraCollection>();
     selHits_ = std::make_unique<TrackingRecHitCollection>();
     rHits_ = evt.getRefBeforePut<TrackingRecHitCollection>();
     rTrackExtras_ = evt.getRefBeforePut<TrackExtraCollection>();
   }
   
   if (copyTrajectories_) trackRefs_.resize(hSrcTrack->size());
   
   // Loop over tracks
   size_t current = 0;
   for (TrackCollection::const_iterator it = hSrcTrack->begin(), ed = hSrcTrack->end(); it != ed; ++it, ++current) {
     const Track & trk = * it;
     // Check if this track passes cuts
     bool ok = select(vertexBeamSpot, trk);
     if (!ok) {
       if (copyTrajectories_) trackRefs_[current] = reco::TrackRef();
       if (!keepAllTracks_) continue;
     }
     selTracks_->push_back( Track( trk ) ); // clone and store
     if (ok && setQualityBit_) selTracks_->back().setQuality(qualityToSet_);
     if (copyExtras_) {
       // TrackExtras
       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(), trk.seedRef() ) );
       selTracks_->back().setExtra( TrackExtraRef( rTrackExtras_, selTrackExtras_->size() - 1) );
       TrackExtra & tx = selTrackExtras_->back();
       tx.setResiduals(trk.residuals());
       // TrackingRecHits
       auto const firstHitIndex = selHits_->size();
       for( trackingRecHit_iterator hit = trk.recHitsBegin(); hit != trk.recHitsEnd(); ++ hit ) {
     selHits_->push_back( (*hit)->clone() );
       }
       tx.setHits( rHits_, firstHitIndex, selHits_->size() - firstHitIndex );
       tx.setTrajParams(trk.extra()->trajParams(),trk.extra()->chi2sX5());
     }
     if (copyTrajectories_) {
       trackRefs_[current] = TrackRef(rTracks_, selTracks_->size() - 1);
     }
     }
   if ( copyTrajectories_ ) {
     Handle< vector<Trajectory> > hTraj;
     Handle< TrajTrackAssociationCollection > hTTAss;
     evt.getByToken(srcTass_, hTTAss);
     evt.getByToken(srcTraj_, hTraj);
     selTrajs_ = std::make_unique<std::vector<Trajectory>>(); 
     rTrajectories_ = evt.getRefBeforePut< vector<Trajectory> >();
     selTTAss_ = std::make_unique<TrajTrackAssociationCollection>(&evt.productGetter());
     for (size_t i = 0, n = hTraj->size(); i < n; ++i) {
       Ref< vector<Trajectory> > trajRef(hTraj, i);
       TrajTrackAssociationCollection::const_iterator match = hTTAss->find(trajRef);
       if (match != hTTAss->end()) {
     const Ref<TrackCollection> &trkRef = match->val; 
     short oldKey = static_cast<short>(trkRef.key());
     if (trackRefs_[oldKey].isNonnull()) {
       selTrajs_->push_back( Trajectory(*trajRef) );
       selTTAss_->insert ( Ref< vector<Trajectory> >(rTrajectories_, selTrajs_->size() - 1), trackRefs_[oldKey] );
     }
       }
     }
   }
   
   static const std::string emptyString;
   evt.put(std::move(selTracks_));
   if (copyExtras_ ) {
     evt.put(std::move(selTrackExtras_)); 
     evt.put(std::move(selHits_));
   }
   if ( copyTrajectories_ ) {
     evt.put(std::move(selTrajs_));
     evt.put(std::move(selTTAss_));
   }
 }

bool CosmicTrackSelector::select	(	const reco::BeamSpot &	vertexBeamSpot,
		const reco::Track &	tk
	)

private

Definition at line 255 of file CosmicTrackSelector.cc.

References funct::abs(), chi2n, chi2n_par_, allConversions_cfi::d0, DEFINE_FWK_MODULE, reco::TrackBase::dxy(), PVValHelper::dz, reco::TrackBase::dz(), PVValHelper::eta, reco::TrackBase::eta(), reco::TrackBase::hitPattern(), triggerObjects_cff::id, createfilelist::int, max_d0_, max_eta_, max_lostLayers_, max_z0_, min_3Dlayers_, min_layers_, min_nHit_, min_nPixelHit_, min_pt_, nlayers, reco::TrackBase::normalizedChi2(), reco::HitPattern::numberOfValidStripLayersWithMonoAndStereo(), PixelSubdetector::PixelBarrel, PixelSubdetector::PixelEndcap, reco::HitPattern::pixelLayersWithMeasurement(), reco::BeamSpot::position(), EnergyCorrector::pt, reco::TrackBase::pt(), rpcPointValidation_cfi::recHit, reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), reco::HitPattern::TRACK_HITS, reco::HitPattern::trackerLayersWithMeasurement(), and reco::HitPattern::trackerLayersWithoutMeasurement().

Referenced by produce().

                                                                                         {
   // Decide if the given track passes selection cuts.
   
   using namespace std; 
   
   // Cuts on numbers of layers with hits/3D hits/lost hits.
   uint32_t nlayers     = tk.hitPattern().trackerLayersWithMeasurement();
   uint32_t nlayers3D   = tk.hitPattern().pixelLayersWithMeasurement() +
       tk.hitPattern().numberOfValidStripLayersWithMonoAndStereo();
   uint32_t nlayersLost = tk.hitPattern().trackerLayersWithoutMeasurement(reco::HitPattern::TRACK_HITS);
   
   // Get the number of valid hits and PixelHits
   uint32_t nHit = 0;
   uint32_t nPixelHit = 0;
   for ( trackingRecHit_iterator recHit = tk.recHitsBegin(); recHit != tk.recHitsEnd(); ++recHit ) {
     if ( !((*recHit)->isValid()) ) continue; 
     ++nHit;
     DetId id((*recHit)->geographicalId());
     if ( (unsigned int)id.subdetId() == PixelSubdetector::PixelBarrel 
      || (unsigned int)id.subdetId() == PixelSubdetector::PixelEndcap )
       ++nPixelHit;
   }  
   
   // Cut on the number of valid hits
   if (nHit < min_nHit_) return false;
   // Cut on the number of valid Pixel hits
   if (nPixelHit < min_nPixelHit_) return false;
   if (nlayers < min_layers_) return false;
   if (nlayers3D < min_3Dlayers_) return false;
   if (nlayersLost > max_lostLayers_) return false;
   
   // Get track parameters
   double pt = tk.pt(),eta = tk.eta(), chi2n =  tk.normalizedChi2();
   double d0 = -tk.dxy(vertexBeamSpot.position()), dz = tk.dz();
   
   // Absolute cuts on all tracks impact parameters with respect to beam-spot.
   if (abs(d0) > max_d0_) return false;
   if (abs(dz) > max_z0_) return false;
   
   // optimized cuts adapted to the track eta, pt and  chiquare/ndof 
   if (abs(eta) > max_eta_) return false;
   if (pt < min_pt_) return false;
   if (chi2n > chi2n_par_*nlayers) return false;
 
   
   else    
     return true;
   
 }

Member Data Documentation

edm::EDGetTokenT<reco::BeamSpot> CosmicTrackSelector::beamspot_

private

Definition at line 48 of file CosmicTrackSelector.cc.

Referenced by produce().

double CosmicTrackSelector::chi2n_par_

private

Definition at line 64 of file CosmicTrackSelector.cc.

Referenced by select().

bool CosmicTrackSelector::copyExtras_

private

Definition at line 50 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

bool CosmicTrackSelector::copyTrajectories_

private

Definition at line 52 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

bool CosmicTrackSelector::keepAllTracks_

private

Definition at line 57 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

double CosmicTrackSelector::max_d0_

private

Definition at line 67 of file CosmicTrackSelector.cc.

Referenced by select().

double CosmicTrackSelector::max_eta_

private

Definition at line 71 of file CosmicTrackSelector.cc.

Referenced by select().

uint32_t CosmicTrackSelector::max_lostLayers_

private

Definition at line 79 of file CosmicTrackSelector.cc.

Referenced by select().

double CosmicTrackSelector::max_z0_

private

Definition at line 68 of file CosmicTrackSelector.cc.

Referenced by select().

uint32_t CosmicTrackSelector::min_3Dlayers_

private

Definition at line 78 of file CosmicTrackSelector.cc.

Referenced by select().

uint32_t CosmicTrackSelector::min_layers_

private

Definition at line 77 of file CosmicTrackSelector.cc.

Referenced by select().

uint32_t CosmicTrackSelector::min_nHit_

private

Definition at line 73 of file CosmicTrackSelector.cc.

Referenced by select().

uint32_t CosmicTrackSelector::min_nPixelHit_

private

Definition at line 75 of file CosmicTrackSelector.cc.

Referenced by select().

double CosmicTrackSelector::min_pt_

private

Definition at line 70 of file CosmicTrackSelector.cc.

Referenced by select().

TrackBase::TrackQuality CosmicTrackSelector::qualityToSet_

private

Definition at line 60 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

std::vector<double> CosmicTrackSelector::res_par_

private

Definition at line 63 of file CosmicTrackSelector.cc.

TrackingRecHitRefProd CosmicTrackSelector::rHits_

private

Definition at line 90 of file CosmicTrackSelector.cc.

Referenced by produce().

reco::TrackExtraRefProd CosmicTrackSelector::rTrackExtras_

private

Definition at line 89 of file CosmicTrackSelector.cc.

Referenced by produce().

reco::TrackRefProd CosmicTrackSelector::rTracks_

private

Definition at line 88 of file CosmicTrackSelector.cc.

Referenced by produce().

edm::RefProd< std::vector<Trajectory> > CosmicTrackSelector::rTrajectories_

private

Definition at line 91 of file CosmicTrackSelector.cc.

Referenced by produce().

std::unique_ptr<TrackingRecHitCollection> CosmicTrackSelector::selHits_

private

Definition at line 84 of file CosmicTrackSelector.cc.

Referenced by produce().

std::unique_ptr<reco::TrackExtraCollection> CosmicTrackSelector::selTrackExtras_

private

Definition at line 83 of file CosmicTrackSelector.cc.

Referenced by produce().

std::unique_ptr<reco::TrackCollection> CosmicTrackSelector::selTracks_

private

Definition at line 82 of file CosmicTrackSelector.cc.

Referenced by produce().

std::unique_ptr<std::vector<const Trajectory *> > CosmicTrackSelector::selTrajPtrs_

private

Definition at line 86 of file CosmicTrackSelector.cc.

std::unique_ptr<std::vector<Trajectory> > CosmicTrackSelector::selTrajs_

private

Definition at line 85 of file CosmicTrackSelector.cc.

Referenced by produce().

std::unique_ptr<TrajTrackAssociationCollection > CosmicTrackSelector::selTTAss_

private

Definition at line 87 of file CosmicTrackSelector.cc.

Referenced by produce().

bool CosmicTrackSelector::setQualityBit_

private

Definition at line 59 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

edm::EDGetTokenT<reco::TrackCollection> CosmicTrackSelector::src_

private

Definition at line 47 of file CosmicTrackSelector.cc.

Referenced by produce().

edm::EDGetTokenT<TrajTrackAssociationCollection > CosmicTrackSelector::srcTass_

private

Definition at line 54 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

edm::EDGetTokenT<std::vector<Trajectory> > CosmicTrackSelector::srcTraj_

private

Definition at line 53 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

std::vector<reco::TrackRef> CosmicTrackSelector::trackRefs_

private

Definition at line 92 of file CosmicTrackSelector.cc.

Referenced by produce().

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Member Typedef Documentation

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation