CosmicTrackSelector Class Reference

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	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndRuns () 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

Point

typedef math::XYZPoint CosmicTrackSelector::Point

private

Definition at line 41 of file CosmicTrackSelector.cc.

Constructor & Destructor Documentation

CosmicTrackSelector()

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

explicit

Definition at line 98 of file CosmicTrackSelector.cc.

    : 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.empty()) {
      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");
  }
}

References SiStripOfflineCRack_cfg::alias, looper::cfg, copyExtras_, copyTrajectories_, Exception, keepAllTracks_, qualityToSet_, setQualityBit_, srcTass_, srcTraj_, and AlCaHLTBitMon_QueryRunRegistry::string.

~CosmicTrackSelector()

CosmicTrackSelector::~CosmicTrackSelector ( )

override

Definition at line 151 of file CosmicTrackSelector.cc.

{}

Member Function Documentation

produce()

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

overrideprivate

Definition at line 153 of file CosmicTrackSelector.cc.

                                                                        {
  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_));
  }
}

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(), dqmiodumpmetadata::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_.

select()

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

private

Definition at line 262 of file CosmicTrackSelector.cc.

                                                                                        {
  // 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;
}

References funct::abs(), chi2n, chi2n_par_, d0, 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(), DiDispStaMuonMonitor_cfi::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().

Member Data Documentation

beamspot_

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

private

Definition at line 48 of file CosmicTrackSelector.cc.

Referenced by produce().

chi2n_par_

double CosmicTrackSelector::chi2n_par_

private

Definition at line 64 of file CosmicTrackSelector.cc.

Referenced by select().

copyExtras_

bool CosmicTrackSelector::copyExtras_

private

Definition at line 50 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

copyTrajectories_

bool CosmicTrackSelector::copyTrajectories_

private

Definition at line 52 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

keepAllTracks_

bool CosmicTrackSelector::keepAllTracks_

private

Definition at line 57 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

max_d0_

double CosmicTrackSelector::max_d0_

private

Definition at line 67 of file CosmicTrackSelector.cc.

Referenced by select().

max_eta_

double CosmicTrackSelector::max_eta_

private

Definition at line 71 of file CosmicTrackSelector.cc.

Referenced by select().

max_lostLayers_

uint32_t CosmicTrackSelector::max_lostLayers_

private

Definition at line 79 of file CosmicTrackSelector.cc.

Referenced by select().

max_z0_

double CosmicTrackSelector::max_z0_

private

Definition at line 68 of file CosmicTrackSelector.cc.

Referenced by select().

min_3Dlayers_

uint32_t CosmicTrackSelector::min_3Dlayers_

private

Definition at line 78 of file CosmicTrackSelector.cc.

Referenced by select().

min_layers_

uint32_t CosmicTrackSelector::min_layers_

private

Definition at line 77 of file CosmicTrackSelector.cc.

Referenced by select().

min_nHit_

uint32_t CosmicTrackSelector::min_nHit_

private

Definition at line 73 of file CosmicTrackSelector.cc.

Referenced by select().

min_nPixelHit_

uint32_t CosmicTrackSelector::min_nPixelHit_

private

Definition at line 75 of file CosmicTrackSelector.cc.

Referenced by select().

min_pt_

double CosmicTrackSelector::min_pt_

private

Definition at line 70 of file CosmicTrackSelector.cc.

Referenced by select().

qualityToSet_

TrackBase::TrackQuality CosmicTrackSelector::qualityToSet_

private

Definition at line 60 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

res_par_

std::vector<double> CosmicTrackSelector::res_par_

private

Definition at line 63 of file CosmicTrackSelector.cc.

rHits_

TrackingRecHitRefProd CosmicTrackSelector::rHits_

private

Definition at line 90 of file CosmicTrackSelector.cc.

Referenced by produce().

rTrackExtras_

reco::TrackExtraRefProd CosmicTrackSelector::rTrackExtras_

private

Definition at line 89 of file CosmicTrackSelector.cc.

Referenced by produce().

rTracks_

reco::TrackRefProd CosmicTrackSelector::rTracks_

private

Definition at line 88 of file CosmicTrackSelector.cc.

Referenced by produce().

rTrajectories_

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

private

Definition at line 91 of file CosmicTrackSelector.cc.

Referenced by produce().

selHits_

std::unique_ptr<TrackingRecHitCollection> CosmicTrackSelector::selHits_

private

Definition at line 84 of file CosmicTrackSelector.cc.

Referenced by produce().

selTrackExtras_

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

private

Definition at line 83 of file CosmicTrackSelector.cc.

Referenced by produce().

selTracks_

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

private

Definition at line 82 of file CosmicTrackSelector.cc.

Referenced by produce().

selTrajPtrs_

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

private

Definition at line 86 of file CosmicTrackSelector.cc.

selTrajs_

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

private

Definition at line 85 of file CosmicTrackSelector.cc.

Referenced by produce().

selTTAss_

std::unique_ptr<TrajTrackAssociationCollection> CosmicTrackSelector::selTTAss_

private

Definition at line 87 of file CosmicTrackSelector.cc.

Referenced by produce().

setQualityBit_

bool CosmicTrackSelector::setQualityBit_

private

Definition at line 59 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

src_

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

private

Definition at line 47 of file CosmicTrackSelector.cc.

Referenced by produce().

srcTass_

edm::EDGetTokenT<TrajTrackAssociationCollection> CosmicTrackSelector::srcTass_

private

Definition at line 54 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

srcTraj_

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

private

Definition at line 53 of file CosmicTrackSelector.cc.

Referenced by CosmicTrackSelector(), and produce().

trackRefs_

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

private

Definition at line 92 of file CosmicTrackSelector.cc.

Referenced by produce().