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
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

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<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::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.

References SiStripOfflineCRack_cfg::alias, looper::cfg, copyExtras_, copyTrajectories_, Exception, keepAllTracks_, pixelTrack::qualityByName(), 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.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");
  }
}

~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.

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_.

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

select()

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

private

Definition at line 262 of file CosmicTrackSelector.cc.

References funct::abs(), chi2n, chi2n_par_, d0, reco::TrackBase::dxy(), PVValHelper::dz, reco::TrackBase::dz(), PVValHelper::eta, reco::TrackBase::eta(), reco::TrackBase::hitPattern(), l1ctLayer2EG_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().

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

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().