trackerTFP::ProducerKFin Class Reference

transforms TTTracks into KF input More...

Inheritance diagram for trackerTFP::ProducerKFin:

Public Member Functions
	ProducerKFin (const ParameterSet &)
	~ProducerKFin () 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 Member Functions
void	beginRun (const Run &, const EventSetup &) override
virtual void	endJob ()
void	produce (Event &, const EventSetup &) override

Private Attributes
const DataFormats *	dataFormats_ = nullptr
EDGetTokenT< StreamsStub >	edGetTokenStubs_
EDGetTokenT< vector< TTTrack< Ref_Phase2TrackerDigi_ > > >	edGetTokenTTTracks_
EDPutTokenT< StreamsStub >	edPutTokenAcceptedStubs_
EDPutTokenT< StreamsTrack >	edPutTokenAcceptedTracks_
EDPutTokenT< StreamsStub >	edPutTokenLostStubs_
EDPutTokenT< StreamsTrack >	edPutTokenLostTracks_
bool	enableTruncation_
ESGetToken< DataFormats, DataFormatsRcd >	esGetTokenDataFormats_
ESGetToken< LayerEncoding, LayerEncodingRcd >	esGetTokenLayerEncoding_
ESGetToken< Setup, SetupRcd >	esGetTokenSetup_
ParameterSet	iConfig_
const LayerEncoding *	layerEncoding_ = nullptr
const Setup *	setup_ = nullptr

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

transforms TTTracks into KF input

Author

Thomas Schuh

Date

2020, July

Definition at line 37 of file ProducerKFin.cc.

Constructor & Destructor Documentation

ProducerKFin()

trackerTFP::ProducerKFin::ProducerKFin ( const ParameterSet &  iConfig )

explicit

Definition at line 75 of file ProducerKFin.cc.

References edGetTokenStubs_, edGetTokenTTTracks_, edPutTokenAcceptedStubs_, edPutTokenAcceptedTracks_, edPutTokenLostStubs_, edPutTokenLostTracks_, enableTruncation_, esGetTokenDataFormats_, esGetTokenLayerEncoding_, esGetTokenSetup_, edm::ParameterSet::getParameter(), and ProducerED_cfi::InputTag.

                                                        : iConfig_(iConfig) {
    const string& labelTTTracks = iConfig.getParameter<string>("LabelZHTout");
    const string& labelStubs = iConfig.getParameter<string>("LabelZHT");
    const string& branchAcceptedStubs = iConfig.getParameter<string>("BranchAcceptedStubs");
    const string& branchAcceptedTracks = iConfig.getParameter<string>("BranchAcceptedTracks");
    const string& branchLostStubs = iConfig.getParameter<string>("BranchLostStubs");
    const string& branchLostTracks = iConfig.getParameter<string>("BranchLostTracks");
    // book in- and output ED products
    edGetTokenTTTracks_ =
        consumes<vector<TTTrack<Ref_Phase2TrackerDigi_>>>(InputTag(labelTTTracks, branchAcceptedTracks));
    edGetTokenStubs_ = consumes<StreamsStub>(InputTag(labelStubs, branchAcceptedStubs));
    edPutTokenAcceptedStubs_ = produces<StreamsStub>(branchAcceptedStubs);
    edPutTokenAcceptedTracks_ = produces<StreamsTrack>(branchAcceptedTracks);
    edPutTokenLostStubs_ = produces<StreamsStub>(branchLostStubs);
    edPutTokenLostTracks_ = produces<StreamsTrack>(branchLostTracks);
    // book ES products
    esGetTokenSetup_ = esConsumes<Setup, SetupRcd, Transition::BeginRun>();
    esGetTokenDataFormats_ = esConsumes<DataFormats, DataFormatsRcd, Transition::BeginRun>();
    esGetTokenLayerEncoding_ = esConsumes<LayerEncoding, LayerEncodingRcd, Transition::BeginRun>();
    //
    enableTruncation_ = iConfig.getParameter<bool>("EnableTruncation");
  }

~ProducerKFin()

trackerTFP::ProducerKFin::~ProducerKFin ( )

inlineoverride

Definition at line 40 of file ProducerKFin.cc.

{}

Member Function Documentation

beginRun()

void trackerTFP::ProducerKFin::beginRun ( const Run &  iRun, const EventSetup &  iSetup  )

overrideprivate

Definition at line 98 of file ProducerKFin.cc.

References tt::Setup::checkHistory(), tt::Setup::configurationSupported(), dataFormats_, esGetTokenDataFormats_, esGetTokenLayerEncoding_, esGetTokenSetup_, edm::EventSetup::getData(), edm::ParameterSet::getParameter(), iConfig_, layerEncoding_, edm::Run::processHistory(), and setup_.

                                                                       {
    // helper class to store configurations
    setup_ = &iSetup.getData(esGetTokenSetup_);
    if (!setup_->configurationSupported())
      return;
    // check process history if desired
    if (iConfig_.getParameter<bool>("CheckHistory"))
      setup_->checkHistory(iRun.processHistory());
    // helper class to extract structured data from tt::Frames
    dataFormats_ = &iSetup.getData(esGetTokenDataFormats_);
    // helper class to encode layer
    layerEncoding_ = &iSetup.getData(esGetTokenLayerEncoding_);
  }

endJob()

virtual void trackerTFP::ProducerKFin::endJob ( void  )

inlineprivatevirtual

Definition at line 45 of file ProducerKFin.cc.

{}

produce()

void trackerTFP::ProducerKFin::produce ( Event &  iEvent, const EventSetup &  iSetup  )

overrideprivate

Definition at line 112 of file ProducerKFin.cc.

References tt::Setup::configurationSupported(), trackerTFP::cot, trackerTFP::StubZHT::cot(), dataFormats_, trackerTFP::DataFormat::digi(), trackerTFP::dPhi, tt::Setup::dPhi(), trackerTFP::dZ, tt::Setup::dZ(), edGetTokenStubs_, edGetTokenTTTracks_, edPutTokenAcceptedStubs_, edPutTokenAcceptedTracks_, edPutTokenLostStubs_, edPutTokenLostTracks_, enableTruncation_, trackerTFP::DataFormat::floating(), trackerTFP::DataFormats::format(), amptDefault_cfi::frame, mps_fire::i, iEvent, createfilelist::int, trackerTFP::DataFormat::integer(), trackerTFP::inv2R, trackerTFP::StubZHT::inv2R(), trackerTFP::kf, trackerTFP::kfin, MainPageGenerator::l, layerEncoding_, tt::Setup::layerId(), trackerTFP::LayerEncoding::layerIdKF(), trackerTFP::LayerEncoding::maybePattern(), SiStripPI::min, eostools::move(), GetRecoTauVFromDQM_MC_cff::next, trackerTFP::DataFormats::numChannel(), tt::Setup::numFrames(), tt::Setup::numLayers(), tt::Setup::numRegions(), tt::Setup::numSectorsPhi(), trackerTFP::DataFormats::numStreamsStubs(), trackerTFP::DataFormats::numStreamsTracks(), edm::Handle< T >::product(), nano_mu_digi_cff::region, tt::Setup::sectorCot(), trackerTFP::sectorPhi, setup_, cms::cuda::stream, HLT_2024v12_cff::streams, trackerTFP::DataFormat::toUnsigned(), HLT_2024v12_cff::track, DiMuonV_cfg::tracks, trackerTFP::zht, tt::Setup::zhtMaxStubsPerLayer(), and trackerTFP::zT.

                                                                    {
    const DataFormat& dfcot = dataFormats_->format(Variable::cot, Process::kfin);
    const DataFormat& dfzT = dataFormats_->format(Variable::zT, Process::kfin);
    const DataFormat& dfinv2R = dataFormats_->format(Variable::inv2R, Process::kfin);
    const DataFormat& dfdPhi = dataFormats_->format(Variable::dPhi, Process::kfin);
    const DataFormat& dfdZ = dataFormats_->format(Variable::dZ, Process::kfin);
    // empty KFin products
    StreamsStub streamAcceptedStubs(dataFormats_->numStreamsStubs(Process::kf));
    StreamsTrack streamAcceptedTracks(dataFormats_->numStreamsTracks(Process::kf));
    StreamsStub streamLostStubs(dataFormats_->numStreamsStubs(Process::kf));
    StreamsTrack streamLostTracks(dataFormats_->numStreamsTracks(Process::kf));
    // read in SFout Product and produce KFin product
    if (setup_->configurationSupported()) {
      Handle<StreamsStub> handleStubs;
      iEvent.getByToken<StreamsStub>(edGetTokenStubs_, handleStubs);
      const StreamsStub& streams = *handleStubs.product();
      Handle<vector<TTTrack<Ref_Phase2TrackerDigi_>>> handleTTTracks;
      iEvent.getByToken<vector<TTTrack<Ref_Phase2TrackerDigi_>>>(edGetTokenTTTracks_, handleTTTracks);
      const vector<TTTrack<Ref_Phase2TrackerDigi_>>& ttTracks = *handleTTTracks.product();
      for (int region = 0; region < setup_->numRegions(); region++) {
        // Unpack input SF data into vector
        int nStubsZHR(0);
        for (int channel = 0; channel < dataFormats_->numChannel(Process::zht); channel++) {
          const int index = region * dataFormats_->numChannel(Process::zht) + channel;
          const StreamStub& stream = streams[index];
          nStubsZHR += accumulate(stream.begin(), stream.end(), 0, [](int sum, const FrameStub& frame) {
            return sum + (frame.first.isNonnull() ? 1 : 0);
          });
        }
        vector<StubZHT> stubsZHT;
        stubsZHT.reserve(nStubsZHR);
        for (int channel = 0; channel < dataFormats_->numChannel(Process::zht); channel++) {
          const int index = region * dataFormats_->numChannel(Process::zht) + channel;
          for (const FrameStub& frame : streams[index])
            if (frame.first.isNonnull())
              stubsZHT.emplace_back(frame, dataFormats_);
        }
        vector<deque<FrameStub>> dequesStubs(dataFormats_->numChannel(Process::kf) * setup_->numLayers());
        vector<deque<FrameTrack>> dequesTracks(dataFormats_->numChannel(Process::kf));
        int i(0);
        for (const TTTrack<Ref_Phase2TrackerDigi_>& ttTrack : ttTracks) {
          if ((int)ttTrack.phiSector() / setup_->numSectorsPhi() != region) {
            i++;
            continue;
          }
          const int sectorPhi = ttTrack.phiSector() % setup_->numSectorsPhi();
          deque<FrameTrack>& tracks = dequesTracks[sectorPhi];
          const int binEta = ttTrack.etaSector();
          const int binZT = dfzT.toUnsigned(dfzT.integer(ttTrack.z0()));
          const int binCot = dfcot.toUnsigned(dfcot.integer(ttTrack.tanL()));
          StubZHT* stubZHT = nullptr;
          vector<int> layerCounts(setup_->numLayers(), 0);
          for (const TTStubRef& ttStubRef : ttTrack.getStubRefs()) {
            const int layerId = setup_->layerId(ttStubRef);
            const int layerIdKF = layerEncoding_->layerIdKF(binEta, binZT, binCot, layerId);
            if (layerIdKF == -1)
              continue;
            if (layerCounts[layerIdKF] == setup_->zhtMaxStubsPerLayer())
              continue;
            layerCounts[layerIdKF]++;
            deque<FrameStub>& stubs = dequesStubs[sectorPhi * setup_->numLayers() + layerIdKF];
            auto identical = [ttStubRef, ttTrack](const StubZHT& stub) {
              return (int)ttTrack.trackSeedType() == stub.trackId() && ttStubRef == stub.ttStubRef();
            };
            stubZHT = &*find_if(stubsZHT.begin(), stubsZHT.end(), identical);
            const double inv2R = dfinv2R.floating(stubZHT->inv2R());
            const double cot = dfcot.floating(stubZHT->cot()) + setup_->sectorCot(binEta);
            const double dPhi = dfdPhi.digi(setup_->dPhi(ttStubRef, inv2R));
            const double dZ = dfdZ.digi(setup_->dZ(ttStubRef, cot));
            stubs.emplace_back(StubKFin(*stubZHT, dPhi, dZ, layerIdKF).frame());
          }
          const int size = *max_element(layerCounts.begin(), layerCounts.end());
          int layerIdKF(0);
          for (int layerCount : layerCounts) {
            deque<FrameStub>& stubs = dequesStubs[sectorPhi * setup_->numLayers() + layerIdKF++];
            const int nGaps = size - layerCount;
            stubs.insert(stubs.end(), nGaps, FrameStub());
          }
          const TTBV& maybePattern = layerEncoding_->maybePattern(binEta, binZT, binCot);
          const TrackKFin track(*stubZHT, TTTrackRef(handleTTTracks, i++), maybePattern);
          tracks.emplace_back(track.frame());
          const int nGaps = size - 1;
          tracks.insert(tracks.end(), nGaps, FrameTrack());
        }
        // transform deques to vectors & emulate truncation
        for (int channel = 0; channel < dataFormats_->numChannel(Process::kf); channel++) {
          const int index = region * dataFormats_->numChannel(Process::kf) + channel;
          deque<FrameTrack>& tracks = dequesTracks[channel];
          auto limitTracks = next(tracks.begin(), min(setup_->numFrames(), (int)tracks.size()));
          if (!enableTruncation_)
            limitTracks = tracks.end();
          streamAcceptedTracks[index] = StreamTrack(tracks.begin(), limitTracks);
          streamLostTracks[index] = StreamTrack(limitTracks, tracks.end());
          for (int l = 0; l < setup_->numLayers(); l++) {
            deque<FrameStub>& stubs = dequesStubs[channel * setup_->numLayers() + l];
            auto limitStubs = next(stubs.begin(), min(setup_->numFrames(), (int)stubs.size()));
            if (!enableTruncation_)
              limitStubs = stubs.end();
            streamAcceptedStubs[index * setup_->numLayers() + l] = StreamStub(stubs.begin(), limitStubs);
            streamLostStubs[index * setup_->numLayers() + l] = StreamStub(limitStubs, stubs.end());
          }
        }
      }
    }
    // store products
    iEvent.emplace(edPutTokenAcceptedStubs_, std::move(streamAcceptedStubs));
    iEvent.emplace(edPutTokenAcceptedTracks_, std::move(streamAcceptedTracks));
    iEvent.emplace(edPutTokenLostStubs_, std::move(streamLostStubs));
    iEvent.emplace(edPutTokenLostTracks_, std::move(streamLostTracks));
  }

Member Data Documentation

dataFormats_

const DataFormats* trackerTFP::ProducerKFin::dataFormats_ = nullptr

private

Definition at line 68 of file ProducerKFin.cc.

Referenced by beginRun(), and produce().

edGetTokenStubs_

EDGetTokenT<StreamsStub> trackerTFP::ProducerKFin::edGetTokenStubs_

private

Definition at line 50 of file ProducerKFin.cc.

Referenced by produce(), and ProducerKFin().

edGetTokenTTTracks_

EDGetTokenT<vector<TTTrack<Ref_Phase2TrackerDigi_> > > trackerTFP::ProducerKFin::edGetTokenTTTracks_

private

Definition at line 48 of file ProducerKFin.cc.

Referenced by produce(), and ProducerKFin().

edPutTokenAcceptedStubs_

EDPutTokenT<StreamsStub> trackerTFP::ProducerKFin::edPutTokenAcceptedStubs_

private

Definition at line 52 of file ProducerKFin.cc.

Referenced by produce(), and ProducerKFin().

edPutTokenAcceptedTracks_

EDPutTokenT<StreamsTrack> trackerTFP::ProducerKFin::edPutTokenAcceptedTracks_

private

Definition at line 55 of file ProducerKFin.cc.

Referenced by produce(), and ProducerKFin().

edPutTokenLostStubs_

EDPutTokenT<StreamsStub> trackerTFP::ProducerKFin::edPutTokenLostStubs_

private

Definition at line 53 of file ProducerKFin.cc.

Referenced by produce(), and ProducerKFin().

edPutTokenLostTracks_

EDPutTokenT<StreamsTrack> trackerTFP::ProducerKFin::edPutTokenLostTracks_

private

Definition at line 56 of file ProducerKFin.cc.

Referenced by produce(), and ProducerKFin().

enableTruncation_

bool trackerTFP::ProducerKFin::enableTruncation_

private

Definition at line 72 of file ProducerKFin.cc.

Referenced by produce(), and ProducerKFin().

esGetTokenDataFormats_

ESGetToken<DataFormats, DataFormatsRcd> trackerTFP::ProducerKFin::esGetTokenDataFormats_

private

Definition at line 60 of file ProducerKFin.cc.

Referenced by beginRun(), and ProducerKFin().

esGetTokenLayerEncoding_

ESGetToken<LayerEncoding, LayerEncodingRcd> trackerTFP::ProducerKFin::esGetTokenLayerEncoding_

private

Definition at line 62 of file ProducerKFin.cc.

Referenced by beginRun(), and ProducerKFin().

esGetTokenSetup_

ESGetToken<Setup, SetupRcd> trackerTFP::ProducerKFin::esGetTokenSetup_

private

Definition at line 58 of file ProducerKFin.cc.

Referenced by beginRun(), and ProducerKFin().

iConfig_

ParameterSet trackerTFP::ProducerKFin::iConfig_

private

Definition at line 64 of file ProducerKFin.cc.

Referenced by beginRun().

layerEncoding_

const LayerEncoding* trackerTFP::ProducerKFin::layerEncoding_ = nullptr

private

Definition at line 70 of file ProducerKFin.cc.

Referenced by beginRun(), and produce().

setup_

const Setup* trackerTFP::ProducerKFin::setup_ = nullptr

private

Definition at line 66 of file ProducerKFin.cc.

Referenced by beginRun(), produce(), and upgradeWorkflowComponents.UpgradeWorkflow::setup().