trklet::ProducerTT Class Reference

Converts KF output into TTTracks. More...

Inheritance diagram for trklet::ProducerTT:

Public Member Functions
	ProducerTT (const ParameterSet &)
	~ProducerTT () override
Public Member Functions inherited from edm::stream::EDProducer< T >
	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
void	endJob ()
void	produce (Event &, const EventSetup &) override

Private Attributes
const DataFormats *	dataFormats_ = nullptr
EDGetTokenT< StreamsStub >	edGetTokenStubs_
EDGetTokenT< StreamsTrack >	edGetTokenTracks_
EDPutTokenT< TTTracks >	edPutToken_
ESGetToken< DataFormats, DataFormatsRcd >	esGetTokenDataFormats_
ESGetToken< Setup, SetupRcd >	esGetTokenSetup_
ParameterSet	iConfig_
const Setup *	setup_ = nullptr

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer< T >
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

Converts KF output into TTTracks.

Author

Thomas Schuh

Date

2021, Aug

Definition at line 31 of file ProducerTT.cc.

Constructor & Destructor Documentation

ProducerTT()

trklet::ProducerTT::ProducerTT ( const ParameterSet &  iConfig )

explicit

Definition at line 59 of file ProducerTT.cc.

References edGetTokenStubs_, edGetTokenTracks_, edPutToken_, esGetTokenDataFormats_, esGetTokenSetup_, edm::ParameterSet::getParameter(), ProducerED_cfi::InputTag, and label.

                                                    : iConfig_(iConfig) {
    const string& label = iConfig.getParameter<string>("LabelKF");
    const string& branchStubs = iConfig.getParameter<string>("BranchAcceptedStubs");
    const string& branchTracks = iConfig.getParameter<string>("BranchAcceptedTracks");
    // book in- and output ED products
    edGetTokenStubs_ = consumes<StreamsStub>(InputTag(label, branchStubs));
    edGetTokenTracks_ = consumes<StreamsTrack>(InputTag(label, branchTracks));
    edPutToken_ = produces<TTTracks>(branchTracks);
    // book ES products
    esGetTokenSetup_ = esConsumes<Setup, SetupRcd, Transition::BeginRun>();
    esGetTokenDataFormats_ = esConsumes<DataFormats, DataFormatsRcd, Transition::BeginRun>();
  }

~ProducerTT()

trklet::ProducerTT::~ProducerTT ( )

inlineoverride

Definition at line 34 of file ProducerTT.cc.

{}

Member Function Documentation

beginRun()

void trklet::ProducerTT::beginRun ( const Run &  iRun, const EventSetup &  iSetup  )

overrideprivate

Definition at line 72 of file ProducerTT.cc.

References tt::Setup::checkHistory(), tt::Setup::configurationSupported(), dataFormats_, esGetTokenDataFormats_, esGetTokenSetup_, edm::EventSetup::getData(), edm::ParameterSet::getParameter(), iConfig_, 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_);
  }

endJob()

void trklet::ProducerTT::endJob ( void  )

inlineprivate

Definition at line 39 of file ProducerTT.cc.

{}

produce()

void trklet::ProducerTT::produce ( Event &  iEvent, const EventSetup &  iSetup  )

overrideprivate

Definition at line 84 of file ProducerTT.cc.

References tt::Setup::configurationSupported(), dataFormats_, edGetTokenStubs_, edGetTokenTracks_, edPutToken_, amptDefault_cfi::frame, iEvent, createfilelist::int, eostools::move(), beamSpotPI::nTracks, tt::Setup::numLayers(), hltrates_dqm_sourceclient-live_cfg::offset, edm::Handle< T >::product(), setup_, cms::cuda::stream, and HLT_2024v10_cff::track.

                                                                  {
    // empty KFout product
    TTTracks ttTracks;
    // read in KF Product and produce KFout product
    if (setup_->configurationSupported()) {
      Handle<StreamsStub> handleStubs;
      iEvent.getByToken<StreamsStub>(edGetTokenStubs_, handleStubs);
      const StreamsStub& streamsStubs = *handleStubs.product();
      Handle<StreamsTrack> handleTracks;
      iEvent.getByToken<StreamsTrack>(edGetTokenTracks_, handleTracks);
      const StreamsTrack& streamsTracks = *handleTracks.product();
      // count number of kf tracks
      int nTracks(0);
      for (const StreamTrack& stream : streamsTracks)
        nTracks += accumulate(stream.begin(), stream.end(), 0, [](int sum, const FrameTrack& frame) {
          return sum + (frame.first.isNonnull() ? 1 : 0);
        });
      ttTracks.reserve(nTracks);
      // convert kf track frames per channel and stub frames per channel and layer to TTTracks
      for (int channel = 0; channel < (int)streamsTracks.size(); channel++) {
        const int offset = channel * setup_->numLayers();
        int iTrk(0);
        for (const FrameTrack& frameTrack : streamsTracks[channel]) {
          if (frameTrack.first.isNull())
            continue;
          // convert stub frames to kf stubs
          vector<StubKF> stubs;
          stubs.reserve(setup_->numLayers());
          for (int layer = 0; layer < setup_->numLayers(); layer++) {
            const FrameStub& frameStub = streamsStubs[offset + layer][iTrk];
            if (frameStub.first.isNonnull())
              stubs.emplace_back(frameStub, dataFormats_, layer);
          }
          // convert track frame to kf track
          TrackKF track(frameTrack, dataFormats_);
          // convert kf track and kf stubs to TTTrack
          ttTracks.emplace_back(track.ttTrack(stubs));
          iTrk++;
        }
      }
    }
    // store products
    iEvent.emplace(edPutToken_, std::move(ttTracks));
  }

Member Data Documentation

dataFormats_

const DataFormats* trklet::ProducerTT::dataFormats_ = nullptr

private

Definition at line 56 of file ProducerTT.cc.

Referenced by beginRun(), and produce().

edGetTokenStubs_

EDGetTokenT<StreamsStub> trklet::ProducerTT::edGetTokenStubs_

private

Definition at line 42 of file ProducerTT.cc.

Referenced by produce(), and ProducerTT().

edGetTokenTracks_

EDGetTokenT<StreamsTrack> trklet::ProducerTT::edGetTokenTracks_

private

Definition at line 44 of file ProducerTT.cc.

Referenced by produce(), and ProducerTT().

edPutToken_

EDPutTokenT<TTTracks> trklet::ProducerTT::edPutToken_

private

Definition at line 46 of file ProducerTT.cc.

Referenced by produce(), and ProducerTT().

esGetTokenDataFormats_

ESGetToken<DataFormats, DataFormatsRcd> trklet::ProducerTT::esGetTokenDataFormats_

private

Definition at line 50 of file ProducerTT.cc.

Referenced by beginRun(), and ProducerTT().

esGetTokenSetup_

ESGetToken<Setup, SetupRcd> trklet::ProducerTT::esGetTokenSetup_

private

Definition at line 48 of file ProducerTT.cc.

Referenced by beginRun(), and ProducerTT().

iConfig_

ParameterSet trklet::ProducerTT::iConfig_

private

Definition at line 52 of file ProducerTT.cc.

Referenced by beginRun().

setup_

const Setup* trklet::ProducerTT::setup_ = nullptr

private

Definition at line 54 of file ProducerTT.cc.

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