Inheritance diagram for CosmicTrackSelector:

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

virtual	~CosmicTrackSelector ()

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

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

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::auto_ptr < TrackingRecHitCollection >	selHits_

std::auto_ptr < reco::TrackExtraCollection >	selTrackExtras_

std::auto_ptr < reco::TrackCollection >	selTracks_

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

std::auto_ptr< std::vector < Trajectory > >	selTrajs_

std::auto_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

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

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 copyExtras_, copyTrajectories_, edm::hlt::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 ( )

virtual

Definition at line 152 of file CosmicTrackSelector.cc.

152 {

153 }

Member Function Documentation

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

overrideprivatevirtual

Implements edm::stream::EDProducerBase.

Definition at line 155 of file CosmicTrackSelector.cc.

References reco::TrackExtraBase::add(), beamspot_, edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, copyExtras_, copyTrajectories_, cond::rpcobimon::current, archive::emptyString, edm::Event::getByToken(), edm::Event::getRefBeforePut(), i, reco::Track::innerDetId(), reco::Track::innerMomentum(), reco::Track::innerOk(), reco::Track::innerPosition(), reco::Track::innerStateCovariance(), keepAllTracks_, edm::Ref< C, T, F >::key(), match(), n, convertSQLiteXML::ok, reco::Track::outerDetId(), reco::Track::outerMomentum(), reco::Track::outerOk(), reco::Track::outerPosition(), reco::Track::outerStateCovariance(), edm::Event::put(), qualityToSet_, reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), dt_dqm_sourceclient_common_cff::reco, reco::Track::residuals(), rHits_, rTrackExtras_, rTracks_, rTrajectories_, reco::Track::seedDirection(), reco::Track::seedRef(), select(), selHits_, selTrackExtras_, selTracks_, selTrajs_, selTTAss_, setQualityBit_, reco::TrackExtra::setResiduals(), 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_ = auto_ptr<TrackCollection>(new TrackCollection());
   rTracks_ = evt.getRefBeforePut<TrackCollection>();      
   if (copyExtras_) {
     selTrackExtras_ = auto_ptr<TrackExtraCollection>(new TrackExtraCollection());
     selHits_ = auto_ptr<TrackingRecHitCollection>(new 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
       for( trackingRecHit_iterator hit = trk.recHitsBegin(); hit != trk.recHitsEnd(); ++ hit ) {
     selHits_->push_back( (*hit)->clone() );
     tx.add( TrackingRecHitRef( rHits_, selHits_->size() - 1) );
       }
     }
     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_ = auto_ptr< vector<Trajectory> >(new vector<Trajectory>()); 
     rTrajectories_ = evt.getRefBeforePut< vector<Trajectory> >();
     selTTAss_ = auto_ptr< TrajTrackAssociationCollection >(new TrajTrackAssociationCollection());
     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(selTracks_);
   if (copyExtras_ ) {
     evt.put(selTrackExtras_); 
     evt.put(selHits_);
   }
   if ( copyTrajectories_ ) {
     evt.put(selTrajs_);
     evt.put(selTTAss_);
   }
 }

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

private

Definition at line 253 of file CosmicTrackSelector.cc.

References funct::abs(), chi2n_par_, reco::TrackBase::dxy(), reco::TrackBase::dz(), eta(), reco::TrackBase::eta(), reco::TrackBase::hitPattern(), max_d0_, max_eta_, max_lostLayers_, max_z0_, min_3Dlayers_, min_layers_, min_nHit_, min_nPixelHit_, min_pt_, reco::TrackBase::normalizedChi2(), reco::HitPattern::numberOfValidStripLayersWithMonoAndStereo(), PixelSubdetector::PixelBarrel, PixelSubdetector::PixelEndcap, reco::HitPattern::pixelLayersWithMeasurement(), reco::BeamSpot::position(), EnergyCorrector::pt, reco::TrackBase::pt(), 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::auto_ptr< TrackingRecHitCollection> CosmicTrackSelector::selHits_

private

Definition at line 84 of file CosmicTrackSelector.cc.

Referenced by produce().

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

private

Definition at line 83 of file CosmicTrackSelector.cc.

Referenced by produce().

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

private

Definition at line 82 of file CosmicTrackSelector.cc.

Referenced by produce().

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

private

Definition at line 86 of file CosmicTrackSelector.cc.

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

private

Definition at line 85 of file CosmicTrackSelector.cc.

Referenced by produce().

std::auto_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