LSTOutputConverter Class Reference

Inheritance diagram for LSTOutputConverter:

Public Member Functions
	LSTOutputConverter (edm::ParameterSet const &iConfig)
	~LSTOutputConverter () override=default
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

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Member Functions
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) override

Private Attributes
const bool	includeNonpLSTSs_
const bool	includeT5s_
const edm::EDGetTokenT< LSTOutput >	lstOutputToken_
const edm::EDGetTokenT< LSTPhase2OTHitsInput >	lstPhase2OTHitsInputToken_
const edm::EDGetTokenT< TrajectorySeedCollection >	lstPixelSeedToken_
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	mfToken_
const edm::ESGetToken< Propagator, TrackingComponentsRecord >	propagatorAlongToken_
const edm::ESGetToken< Propagator, TrackingComponentsRecord >	propagatorOppositeToken_
std::unique_ptr< SeedCreator >	seedCreator_
const edm::EDPutTokenT< std::vector< SeedStopInfo > >	seedStopInfoPutToken_
const edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord >	tGeomToken_
const edm::EDPutTokenT< TrackCandidateCollection >	trackCandidateNopLSTCPutToken_
const edm::EDPutTokenT< TrackCandidateCollection >	trackCandidatepLSTCPutToken_
const edm::EDPutTokenT< TrackCandidateCollection >	trackCandidatepTCPutToken_
const edm::EDPutTokenT< TrackCandidateCollection >	trackCandidatepTTCPutToken_
const edm::EDPutTokenT< TrackCandidateCollection >	trackCandidatePutToken_
const edm::EDPutTokenT< TrackCandidateCollection >	trackCandidateT5TCPutToken_
const edm::EDPutTokenT< TrajectorySeedCollection >	trajectorySeedpLSPutToken_
const edm::EDPutTokenT< TrajectorySeedCollection >	trajectorySeedPutToken_

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

Definition at line 28 of file LSTOutputConverter.cc.

Constructor & Destructor Documentation

LSTOutputConverter()

LSTOutputConverter::LSTOutputConverter ( edm::ParameterSet const &  iConfig )

explicit

Definition at line 59 of file LSTOutputConverter.cc.

References edm::ParameterSet::getParameter().

    : lstOutputToken_(consumes(iConfig.getParameter<edm::InputTag>("lstOutput"))),
      lstPhase2OTHitsInputToken_{consumes(iConfig.getParameter<edm::InputTag>("phase2OTHits"))},
      lstPixelSeedToken_{consumes(iConfig.getParameter<edm::InputTag>("lstPixelSeeds"))},
      includeT5s_(iConfig.getParameter<bool>("includeT5s")),
      includeNonpLSTSs_(iConfig.getParameter<bool>("includeNonpLSTSs")),
      mfToken_(esConsumes()),
      propagatorAlongToken_{esConsumes(iConfig.getParameter<edm::ESInputTag>("propagatorAlong"))},
      propagatorOppositeToken_{esConsumes(iConfig.getParameter<edm::ESInputTag>("propagatorOpposite"))},
      tGeomToken_(esConsumes()),
      seedCreator_(SeedCreatorFactory::get()->create("SeedFromConsecutiveHitsCreator",
                                                     iConfig.getParameter<edm::ParameterSet>("SeedCreatorPSet"),
                                                     consumesCollector())),
      // FIXME: need to make creation configurable:
      // - A toggle to not produce TSs at all could be useful to save memory;
      //   it won't affect speed though
      // - The minimal set for TCs is t5TCsLST, pTTCsLST and pLSTCsLST.
      //   That would complicate the handling of collections though,
      //   so it is deferred to when we have a clearer picture of what's needed.
      trajectorySeedPutToken_(produces("")),
      trajectorySeedpLSPutToken_(produces("pLSTSsLST")),
      trackCandidatePutToken_(produces("")),
      trackCandidatepTCPutToken_(produces("pTCsLST")),
      trackCandidateT5TCPutToken_(produces("t5TCsLST")),
      trackCandidateNopLSTCPutToken_(produces("nopLSTCsLST")),
      trackCandidatepTTCPutToken_(produces("pTTCsLST")),
      trackCandidatepLSTCPutToken_(produces("pLSTCsLST")),
      seedStopInfoPutToken_(produces()) {}

~LSTOutputConverter()

LSTOutputConverter::~LSTOutputConverter ( )

overridedefault

Member Function Documentation

fillDescriptions()

void LSTOutputConverter::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 88 of file LSTOutputConverter.cc.

References edm::ParameterSetDescription::add(), edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, ProducerED_cfi::InputTag, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                    {
  edm::ParameterSetDescription desc;

  desc.add<edm::InputTag>("lstOutput", edm::InputTag("lstProducer"));
  desc.add<edm::InputTag>("phase2OTHits", edm::InputTag("lstPhase2OTHitsInputProducer"));
  desc.add<edm::InputTag>("lstPixelSeeds", edm::InputTag("lstPixelSeedInputProducer"));
  desc.add<bool>("includeT5s", true);
  desc.add<bool>("includeNonpLSTSs", false);
  desc.add("propagatorAlong", edm::ESInputTag{"", "PropagatorWithMaterial"});
  desc.add("propagatorOpposite", edm::ESInputTag{"", "PropagatorWithMaterialOpposite"});

  edm::ParameterSetDescription psd0;
  psd0.add<std::string>("ComponentName", std::string("SeedFromConsecutiveHitsCreator"));
  psd0.add<std::string>("propagator", std::string("PropagatorWithMaterial"));
  psd0.add<double>("SeedMomentumForBOFF", 5.0);
  psd0.add<double>("OriginTransverseErrorMultiplier", 1.0);
  psd0.add<double>("MinOneOverPtError", 1.0);
  psd0.add<std::string>("magneticField", std::string(""));
  psd0.add<std::string>("TTRHBuilder", std::string("WithTrackAngle"));
  psd0.add<bool>("forceKinematicWithRegionDirection", false);
  desc.add<edm::ParameterSetDescription>("SeedCreatorPSet", psd0);

  descriptions.addWithDefaultLabel(desc);
}

produce()

void LSTOutputConverter::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 113 of file LSTOutputConverter.cc.

References a, funct::abs(), b, edm::EventSetup::getData(), mps_fire::i, iEvent, includeNonpLSTSs_, includeT5s_, GeomDetEnumerators::isBarrel(), GeomDetEnumerators::isInnerTracker(), GeomDetEnumerators::isOuterTracker(), dqmiolumiharvest::j, LogDebug, lstOutputToken_, lstPhase2OTHitsInputToken_, lstPixelSeedToken_, mfToken_, eostools::move(), nHits, trajectoryStateTransform::persistentState(), TrackerGeometry::Ph2PSP, hltEgammaHLTExtra_cfi::pixelSeeds, pLS, propagatorAlongToken_, propagatorOppositeToken_, FastTrackerRecHitMaskProducer_cfi::recHits, TrajectoryStateOnSurface::rescaleError(), fileCollector::seed, seedCreator_, HLT_2024v14_cff::seeds, seedStopInfoPutToken_, contentValuesCheck::ss, T5, tGeomToken_, trackCandidateNopLSTCPutToken_, trackCandidatepLSTCPutToken_, trackCandidatepTCPutToken_, trackCandidatepTTCPutToken_, trackCandidatePutToken_, trackCandidateT5TCPutToken_, PbPb_ZMuSkimMuonDPG_cff::tracker, trajectorySeedpLSPutToken_, trajectorySeedPutToken_, trajectoryStateTransform::transientState(), and mitigatedMETSequence_cff::U.

                                                                              {
  // Setup
  auto const& lstOutput = iEvent.get(lstOutputToken_);
  auto const& phase2OTRecHits = iEvent.get(lstPhase2OTHitsInputToken_);
  auto const& pixelSeeds = iEvent.get(lstPixelSeedToken_);
  auto const& mf = iSetup.getData(mfToken_);
  auto const& propAlo = iSetup.getData(propagatorAlongToken_);
  auto const& propOppo = iSetup.getData(propagatorOppositeToken_);
  auto const& tracker = iSetup.getData(tGeomToken_);

  // Vector definitions
  std::vector<std::vector<unsigned int>> const& lstTC_hitIdx = lstOutput.hitIdx();
  std::vector<unsigned int> const& lstTC_len = lstOutput.len();
  std::vector<int> const& lstTC_seedIdx = lstOutput.seedIdx();
  std::vector<short> const& lstTC_trackCandidateType = lstOutput.trackCandidateType();

  TrajectorySeedCollection outputTS, outputpLSTS;
  outputTS.reserve(lstTC_len.size());
  outputpLSTS.reserve(lstTC_len.size());
  TrackCandidateCollection outputTC, outputpTC, outputT5TC, outputNopLSTC, outputpTTC, outputpLSTC;
  outputTC.reserve(lstTC_len.size());
  outputpTC.reserve(lstTC_len.size());
  outputT5TC.reserve(lstTC_len.size());
  outputNopLSTC.reserve(lstTC_len.size());
  outputpTTC.reserve(lstTC_len.size());
  outputpLSTC.reserve(lstTC_len.size());

  auto const& OTHits = phase2OTRecHits.hits();

  LogDebug("LSTOutputConverter") << "lstTC size " << lstTC_len.size();
  for (unsigned int i = 0; i < lstTC_len.size(); i++) {
    LogDebug("LSTOutputConverter") << " cand " << i << " " << lstTC_len[i] << " " << lstTC_seedIdx[i];
    TrajectorySeed seed;
    if (lstTC_trackCandidateType[i] != LSTOutput::LSTTCType::T5)
      seed = pixelSeeds[lstTC_seedIdx[i]];

    edm::OwnVector<TrackingRecHit> recHits;
    if (lstTC_trackCandidateType[i] != LSTOutput::LSTTCType::T5) {
      for (auto const& hit : seed.recHits())
        recHits.push_back(hit.clone());
    }

    unsigned int const nPixelHits = lstTC_trackCandidateType[i] == LSTOutput::LSTTCType::T5 ? 0 : recHits.size();
    for (unsigned int j = nPixelHits; j < lstTC_hitIdx[i].size(); j++)
      recHits.push_back(OTHits[lstTC_hitIdx[i][j]]->clone());

    recHits.sort([](const auto& a, const auto& b) {
      const auto asub = a.det()->subDetector();
      const auto bsub = b.det()->subDetector();
      if (GeomDetEnumerators::isInnerTracker(asub) && GeomDetEnumerators::isOuterTracker(bsub)) {
        return true;
      } else if (GeomDetEnumerators::isOuterTracker(asub) && GeomDetEnumerators::isInnerTracker(bsub)) {
        return false;
      } else if (asub != bsub) {
        return asub < bsub;
      } else {
        const auto& apos = a.surface();
        const auto& bpos = b.surface();
        if (GeomDetEnumerators::isBarrel(asub)) {
          return apos->rSpan().first < bpos->rSpan().first;
        } else {
          return std::abs(apos->zSpan().first) < std::abs(bpos->zSpan().first);
        }
      }
    });

    TrajectorySeedCollection seeds;
    if (lstTC_trackCandidateType[i] != LSTOutput::LSTTCType::pLS) {
      // Construct a full-length TrajectorySeed always for T5s,
      // only when required by a flag for other pT objects.
      if (includeNonpLSTSs_ || lstTC_trackCandidateType[i] == LSTOutput::LSTTCType::T5) {
        using Hit = SeedingHitSet::ConstRecHitPointer;
        std::vector<Hit> hitsForSeed;
        hitsForSeed.reserve(lstTC_len[i]);
        int nHits = 0;
        for (auto const& hit : recHits) {
          if (lstTC_trackCandidateType[i] == LSTOutput::LSTTCType::T5) {
            auto hType = tracker.getDetectorType(hit.geographicalId());
            if (hType != TrackerGeometry::ModuleType::Ph2PSP && nHits < 2)
              continue;  // the first two should be P
          }
          hitsForSeed.emplace_back(dynamic_cast<Hit>(&hit));
          nHits++;
        }

        seedCreator_->init(GlobalTrackingRegion(), iSetup, nullptr);
        seedCreator_->makeSeed(seeds, hitsForSeed);
        if (seeds.empty()) {
          edm::LogInfo("LSTOutputConverter")
              << "failed to convert a LST object to a seed" << i << " " << lstTC_len[i] << " " << lstTC_seedIdx[i];
          if (lstTC_trackCandidateType[i] == LSTOutput::LSTTCType::T5)
            continue;
        }
        if (lstTC_trackCandidateType[i] == LSTOutput::LSTTCType::T5)
          seed = seeds[0];

        auto trajectorySeed = (seeds.empty() ? seed : seeds[0]);
        outputTS.emplace_back(trajectorySeed);
        auto const& ss = trajectorySeed.startingState();
        LogDebug("LSTOutputConverter") << "Created a seed with " << seed.nHits() << " " << ss.detId() << " " << ss.pt()
                                       << " " << ss.parameters().vector() << " " << ss.error(0);
      }
    } else {
      outputTS.emplace_back(seed);
      outputpLSTS.emplace_back(seed);
    }

    TrajectoryStateOnSurface tsos =
        trajectoryStateTransform::transientState(seed.startingState(), (seed.recHits().end() - 1)->surface(), &mf);
    tsos.rescaleError(100.);
    auto tsosPair = propOppo.propagateWithPath(tsos, *recHits[0].surface());
    if (!tsosPair.first.isValid()) {
      LogDebug("LSTOutputConverter") << "Propagating to startingState opposite to momentum failed, trying along next";
      tsosPair = propAlo.propagateWithPath(tsos, *recHits[0].surface());
    }
    if (tsosPair.first.isValid()) {
      PTrajectoryStateOnDet st =
          trajectoryStateTransform::persistentState(tsosPair.first, recHits[0].det()->geographicalId().rawId());

      if (lstTC_trackCandidateType[i] == LSTOutput::LSTTCType::T5) {
        if (!includeT5s_) {
          continue;
        } else {
          auto tc = TrackCandidate(recHits, seed, st);
          outputTC.emplace_back(tc);
          outputT5TC.emplace_back(tc);
          outputNopLSTC.emplace_back(tc);
        }
      } else {
        auto tc = TrackCandidate(recHits, seed, st);
        outputTC.emplace_back(tc);
        outputpTC.emplace_back(tc);
        if (lstTC_trackCandidateType[i] != LSTOutput::LSTTCType::pLS) {
          outputNopLSTC.emplace_back(tc);
          outputpTTC.emplace_back(tc);
        } else {
          outputpLSTC.emplace_back(tc);
        }
      }
    } else {
      edm::LogInfo("LSTOutputConverter") << "Failed to make a candidate initial state. Seed state is " << tsos
                                         << " TC cand " << i << " " << lstTC_len[i] << " " << lstTC_seedIdx[i]
                                         << " first hit " << recHits.front().globalPosition() << " last hit "
                                         << recHits.back().globalPosition();
    }
  }

  LogDebug("LSTOutputConverter") << "done with conversion: Track candidate output size = " << outputpTC.size()
                                 << " (p* objects) + " << outputT5TC.size() << " (T5 objects)";
  iEvent.emplace(trajectorySeedPutToken_, std::move(outputTS));
  iEvent.emplace(trajectorySeedpLSPutToken_, std::move(outputpLSTS));
  iEvent.emplace(trackCandidatePutToken_, std::move(outputTC));
  iEvent.emplace(trackCandidatepTCPutToken_, std::move(outputpTC));
  iEvent.emplace(trackCandidateT5TCPutToken_, std::move(outputT5TC));
  iEvent.emplace(trackCandidateNopLSTCPutToken_, std::move(outputNopLSTC));
  iEvent.emplace(trackCandidatepTTCPutToken_, std::move(outputpTTC));
  iEvent.emplace(trackCandidatepLSTCPutToken_, std::move(outputpLSTC));
  iEvent.emplace(seedStopInfoPutToken_, 0U);  //dummy stop info
}

Member Data Documentation

includeNonpLSTSs_

const bool LSTOutputConverter::includeNonpLSTSs_

private

Definition at line 42 of file LSTOutputConverter.cc.

Referenced by produce().

includeT5s_

const bool LSTOutputConverter::includeT5s_

private

Definition at line 41 of file LSTOutputConverter.cc.

Referenced by produce().

lstOutputToken_

const edm::EDGetTokenT<LSTOutput> LSTOutputConverter::lstOutputToken_

private

Definition at line 38 of file LSTOutputConverter.cc.

Referenced by produce().

lstPhase2OTHitsInputToken_

const edm::EDGetTokenT<LSTPhase2OTHitsInput> LSTOutputConverter::lstPhase2OTHitsInputToken_

private

Definition at line 39 of file LSTOutputConverter.cc.

Referenced by produce().

lstPixelSeedToken_

const edm::EDGetTokenT<TrajectorySeedCollection> LSTOutputConverter::lstPixelSeedToken_

private

Definition at line 40 of file LSTOutputConverter.cc.

Referenced by produce().

mfToken_

const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> LSTOutputConverter::mfToken_

private

Definition at line 43 of file LSTOutputConverter.cc.

Referenced by produce().

propagatorAlongToken_

const edm::ESGetToken<Propagator, TrackingComponentsRecord> LSTOutputConverter::propagatorAlongToken_

private

Definition at line 44 of file LSTOutputConverter.cc.

Referenced by produce().

propagatorOppositeToken_

const edm::ESGetToken<Propagator, TrackingComponentsRecord> LSTOutputConverter::propagatorOppositeToken_

private

Definition at line 45 of file LSTOutputConverter.cc.

Referenced by produce().

seedCreator_

std::unique_ptr<SeedCreator> LSTOutputConverter::seedCreator_

private

Definition at line 47 of file LSTOutputConverter.cc.

Referenced by produce().

seedStopInfoPutToken_

const edm::EDPutTokenT<std::vector<SeedStopInfo> > LSTOutputConverter::seedStopInfoPutToken_

private

Definition at line 56 of file LSTOutputConverter.cc.

Referenced by produce().

tGeomToken_

const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> LSTOutputConverter::tGeomToken_

private

Definition at line 46 of file LSTOutputConverter.cc.

Referenced by produce().

trackCandidateNopLSTCPutToken_

const edm::EDPutTokenT<TrackCandidateCollection> LSTOutputConverter::trackCandidateNopLSTCPutToken_

private

Definition at line 53 of file LSTOutputConverter.cc.

Referenced by produce().

trackCandidatepLSTCPutToken_

const edm::EDPutTokenT<TrackCandidateCollection> LSTOutputConverter::trackCandidatepLSTCPutToken_

private

Definition at line 55 of file LSTOutputConverter.cc.

Referenced by produce().

trackCandidatepTCPutToken_

const edm::EDPutTokenT<TrackCandidateCollection> LSTOutputConverter::trackCandidatepTCPutToken_

private

Definition at line 51 of file LSTOutputConverter.cc.

Referenced by produce().

trackCandidatepTTCPutToken_

const edm::EDPutTokenT<TrackCandidateCollection> LSTOutputConverter::trackCandidatepTTCPutToken_

private

Definition at line 54 of file LSTOutputConverter.cc.

Referenced by produce().

trackCandidatePutToken_

const edm::EDPutTokenT<TrackCandidateCollection> LSTOutputConverter::trackCandidatePutToken_

private

Definition at line 50 of file LSTOutputConverter.cc.

Referenced by produce().

trackCandidateT5TCPutToken_

const edm::EDPutTokenT<TrackCandidateCollection> LSTOutputConverter::trackCandidateT5TCPutToken_

private

Definition at line 52 of file LSTOutputConverter.cc.

Referenced by produce().

trajectorySeedpLSPutToken_

const edm::EDPutTokenT<TrajectorySeedCollection> LSTOutputConverter::trajectorySeedpLSPutToken_

private

Definition at line 49 of file LSTOutputConverter.cc.

Referenced by produce().

trajectorySeedPutToken_

const edm::EDPutTokenT<TrajectorySeedCollection> LSTOutputConverter::trajectorySeedPutToken_

private

Definition at line 48 of file LSTOutputConverter.cc.

Referenced by produce().