trklet::ChannelAssignment Class Reference

Class to assign tracklet tracks and stubs to output channel based on their Pt or seed type as well as DTC stubs to input channel. More...

#include <ChannelAssignment.h>

Public Member Functions
	ChannelAssignment ()
	ChannelAssignment (const edm::ParameterSet &iConfig, const tt::Setup *setup)
const std::vector< int > &	channelEncoding () const
int	channelId (const TTTrackRef &ttTrackRef) const
int	channelId (int seedType, int layerId) const
int	depthMemory () const
bool	layerId (int seedType, const TTStubRef &ttStubRef, int &layerId) const
int	layerId (int seedType, int channel) const
int	maxNumProjectionLayers () const
int	minIdenticalStubs () const
int	nodeDR (const TTTrackRef &ttTrackRef) const
int	numChannelsStub () const
int	numChannelsTrack () const
int	numComparisonModules () const
int	numNodesDR () const
int	numProjectionLayers (int seedType) const
int	numSeedingLayers () const
int	numSeedTypes () const
int	offsetStub (int channelTrack) const
const std::vector< int > &	seedingLayers (int seedType) const
tt::SensorModule::Type	type (const TTStubRef &ttStubRef) const
int	widthLayerId () const
int	widthPSTilt () const
int	widthSeedStubId () const
int	widthStubId () const
	~ChannelAssignment ()

Private Attributes
std::vector< int >	channelEncoding_
int	depthMemory_
int	maxNumProjectionLayers_
int	minIdenticalStubs_
int	numChannelsStub_
int	numChannelsTrack_
int	numComparisonModules_
int	numNodesDR_
int	numSeedingLayers_
int	numSeedTypes_
std::vector< int >	offsetsStubs_
edm::ParameterSet	pSetDR_
edm::ParameterSet	pSetDRin_
std::vector< double >	ptBoundaries_
std::vector< std::string >	seedTypeNames_
std::vector< std::vector< int > >	seedTypesProjectionLayers_
std::vector< std::vector< int > >	seedTypesSeedLayers_
const tt::Setup *	setup_
int	widthLayerId_
int	widthPSTilt_
int	widthSeedStubId_
int	widthStubId_

Detailed Description

Class to assign tracklet tracks and stubs to output channel based on their Pt or seed type as well as DTC stubs to input channel.

Author

Thomas Schuh

Date

2020, Nov; updated 2021 Oct

Definition at line 20 of file ChannelAssignment.h.

Constructor & Destructor Documentation

ChannelAssignment() [1/2]

trklet::ChannelAssignment::ChannelAssignment ( )

inline

Definition at line 22 of file ChannelAssignment.h.

{}

ChannelAssignment() [2/2]

trklet::ChannelAssignment::ChannelAssignment	(	const edm::ParameterSet &	iConfig,
		const tt::Setup *	setup
	)

Definition at line 17 of file ChannelAssignment.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), cppFunctionSkipper::exception, edm::ParameterSet::getParameter(), createfilelist::int, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, maxNumProjectionLayers_, GetRecoTauVFromDQM_MC_cff::next, tt::Setup::numBarrelLayer(), numChannelsStub_, numSeedingLayers_, numSeedTypes_, tt::Setup::offsetLayerDisks(), tt::Setup::offsetLayerId(), offsetsStubs_, trklet::Settings::projdisks(), trklet::Settings::projlayers(), alignCSCRings::s, fileCollector::seed, trklet::Settings::seedlayers(), seedTypeNames_, seedTypesProjectionLayers_, seedTypesSeedLayers_, DBoxMetadataHelper::set_difference(), DBoxMetadataHelper::set_intersection(), and setup_.

      : setup_(setup),
        pSetDRin_(iConfig.getParameter<ParameterSet>("DRin")),
        widthLayerId_(pSetDRin_.getParameter<int>("WidthLayerId")),
        widthStubId_(pSetDRin_.getParameter<int>("WidthStubId")),
        widthSeedStubId_(pSetDRin_.getParameter<int>("WidthSeedStubId")),
        widthPSTilt_(pSetDRin_.getParameter<int>("WidthPSTilt")),
        depthMemory_(pSetDRin_.getParameter<int>("DepthMemory")),
        ptBoundaries_(pSetDRin_.getParameter<vector<double>>("PtBoundaries")),
        pSetDR_(iConfig.getParameter<ParameterSet>("DR")),
        numComparisonModules_(pSetDR_.getParameter<int>("NumComparisonModules")),
        minIdenticalStubs_(pSetDR_.getParameter<int>("MinIdenticalStubs")),
        numNodesDR_(2 * (ptBoundaries_.size() + 1)),
        seedTypeNames_(iConfig.getParameter<vector<string>>("SeedTypes")),
        numSeedTypes_(seedTypeNames_.size()),
        numChannelsTrack_(numSeedTypes_),
        channelEncoding_(iConfig.getParameter<vector<int>>("IRChannelsIn")) {
    const ParameterSet& pSetSeedTypesSeedLayers = iConfig.getParameter<ParameterSet>("SeedTypesSeedLayers");
    const ParameterSet& pSetSeedTypesProjectionLayers = iConfig.getParameter<ParameterSet>("SeedTypesProjectionLayers");
    seedTypesSeedLayers_.reserve(numSeedTypes_);
    seedTypesProjectionLayers_.reserve(numSeedTypes_);
    for (const string& s : seedTypeNames_) {
      seedTypesSeedLayers_.emplace_back(pSetSeedTypesSeedLayers.getParameter<vector<int>>(s));
      seedTypesProjectionLayers_.emplace_back(pSetSeedTypesProjectionLayers.getParameter<vector<int>>(s));
    }
    // consistency check
    const int offsetBarrel = setup_->offsetLayerId();
    const int numBarrelLayer = setup_->numBarrelLayer();
    const int offsetDisk = setup_->offsetLayerDisks() + offsetBarrel;
    static constexpr int invalidSeedLayer = -1;
    static constexpr int invalidLayerDisk = 0;
    const Settings settings;
    const auto& seedlayers = settings.seedlayers();
    const auto& projlayers = settings.projlayers();
    const auto& projdisks = settings.projdisks();
    // convert Seetings layer ids into ChannelAssignment layer ids
    vector<set<int>> allSeedingLayer(seedlayers.size());
    vector<set<int>> allProjectionLayer(seedlayers.size());
    for (int iSeed = 0; iSeed < (int)seedlayers.size(); iSeed++)
      for (const auto& layer : seedlayers[iSeed])
        if (layer != invalidSeedLayer)
          allSeedingLayer[iSeed].insert(layer < numBarrelLayer ? layer + offsetBarrel
                                                               : layer + offsetDisk - numBarrelLayer);
    for (int iSeed = 0; iSeed < (int)projlayers.size(); iSeed++)
      for (const auto& layer : projlayers[iSeed])
        if (layer != invalidLayerDisk)
          allProjectionLayer[iSeed].insert(layer);
    for (int iSeed = 0; iSeed < (int)projdisks.size(); iSeed++)
      for (const auto& disk : projdisks[iSeed])
        if (disk != invalidLayerDisk)
          allProjectionLayer[iSeed].insert(disk - offsetBarrel + offsetDisk);
    // check if ChannelAssignment seedTypesSeedLayers_ and seedTypesProjectionLayers_ are subsets of Settings pendants
    for (int iSubSeed = 0; iSubSeed < numSeedTypes_; iSubSeed++) {
      const vector<int>& seedLayers = seedTypesSeedLayers_[iSubSeed];
      bool subset(false);
      for (int iAllSeed = 0; iAllSeed < (int)seedlayers.size(); iAllSeed++) {
        // compare seed layer
        const set<int>& asl = allSeedingLayer[iAllSeed];
        set<int> sl(seedLayers.begin(), seedLayers.end());
        set<int> intersect;
        set_intersection(sl.begin(), sl.end(), asl.begin(), asl.end(), inserter(intersect, intersect.begin()));
        if (intersect == sl) {
          subset = true;
          // compare projection layer
          const vector<int>& projectionLayers = seedTypesProjectionLayers_[iSubSeed];
          const set<int>& apl = allProjectionLayer[iAllSeed];
          set<int> pl(projectionLayers.begin(), projectionLayers.end());
          set<int> intersect;
          set_intersection(pl.begin(), pl.end(), apl.begin(), apl.end(), inserter(intersect, intersect.begin()));
          if (intersect == pl)
            break;
          set<int> difference;
          set_difference(
              pl.begin(), pl.end(), intersect.begin(), intersect.end(), inserter(difference, difference.begin()));
          cms::Exception exception("LogicError.");
          exception << "ProjectionLayers ( ";
          for (int layer : difference)
            exception << layer << " ";
          exception << ") are not supported with seed type ( ";
          for (int layer : seedLayers)
            exception << layer << " ";
          exception << ")";
          exception.addContext("trklet::ChannelAssignment::ChannelAssignment");
          throw exception;
        }
      }
      if (subset)
        continue;
      cms::Exception exception("LogicError.");
      exception << "SeedLayers ( ";
      for (int layer : seedLayers)
        exception << layer << " ";
      exception << ") are not supported";
      exception.addContext("trklet::ChannelAssignment::ChannelAssignment");
      throw exception;
    }
    auto bigger = [](const vector<int>& lhs, const vector<int>& rhs) { return lhs.size() < rhs.size(); };
    numSeedingLayers_ = max_element(seedTypesSeedLayers_.begin(), seedTypesSeedLayers_.end(), bigger)->size();
    maxNumProjectionLayers_ =
        max_element(seedTypesProjectionLayers_.begin(), seedTypesProjectionLayers_.end(), bigger)->size();
    auto acc = [](int sum, vector<int> ints) { return sum += (int)ints.size(); };
    offsetsStubs_.reserve(numSeedTypes_);
    numChannelsStub_ = accumulate(
        seedTypesProjectionLayers_.begin(), seedTypesProjectionLayers_.end(), numSeedingLayers_ * numSeedTypes_, acc);
    for (int seed = 0; seed < numSeedTypes_; seed++) {
      const auto it = next(seedTypesProjectionLayers_.begin(), seed);
      offsetsStubs_.emplace_back(accumulate(seedTypesProjectionLayers_.begin(), it, numSeedingLayers_ * seed, acc));
    }
  }

~ChannelAssignment()

trklet::ChannelAssignment::~ChannelAssignment ( )

inline

Definition at line 24 of file ChannelAssignment.h.

{}

Member Function Documentation

channelEncoding()

const std::vector<int>& trklet::ChannelAssignment::channelEncoding ( ) const

inline

Definition at line 56 of file ChannelAssignment.h.

References channelEncoding_.

{ return channelEncoding_; }

channelId() [1/2]

int trklet::ChannelAssignment::channelId

(

const TTTrackRef &

ttTrackRef

)

const

Definition at line 128 of file ChannelAssignment.cc.

References cppFunctionSkipper::exception, numSeedTypes_, alignCSCRings::s, and seedTypeNames_.

Referenced by trklet::ProducerTBout::produce(), and L1FPGATrackProducer::produce().

                                                                     {
    const int seedType = ttTrackRef->trackSeedType();
    if (seedType >= numSeedTypes_) {
      cms::Exception exception("logic_error");
      exception << "TTTracks form seed type" << seedType << " not in supported list: (";
      for (const auto& s : seedTypeNames_)
        exception << s << " ";
      exception << ").";
      exception.addContext("trklet:ChannelAssignment:channelId");
      throw exception;
    }
    return ttTrackRef->phiSector() * numSeedTypes_ + seedType;
  }

channelId() [2/2]

int trklet::ChannelAssignment::channelId	(	int	seedType,
		int	layerId
	)		const

Definition at line 177 of file ChannelAssignment.cc.

References HLT_2024v14_cff::distance, spr::find(), layerId(), HLT_2024v14_cff::seeds, seedTypesProjectionLayers_, and seedTypesSeedLayers_.

                                                                  {
    const vector<int>& projections = seedTypesProjectionLayers_.at(seedType);
    const auto itp = find(projections.begin(), projections.end(), layerId);
    if (itp != projections.end())
      return distance(projections.begin(), itp);
    const vector<int>& seeds = seedTypesSeedLayers_.at(seedType);
    const auto its = find(seeds.begin(), seeds.end(), layerId);
    if (its != seeds.end())
      return (int)projections.size() + distance(seeds.begin(), its);
    return -1;
  }

depthMemory()

int trklet::ChannelAssignment::depthMemory ( ) const

inline

Definition at line 40 of file ChannelAssignment.h.

References depthMemory_.

Referenced by trklet::DRin::produce().

{ return depthMemory_; }

layerId() [1/2]

bool trklet::ChannelAssignment::layerId	(	int	seedType,
		const TTStubRef &	ttStubRef,
		int &	layerId
	)		const

Definition at line 143 of file ChannelAssignment.cc.

References HLT_2024v14_cff::distance, cppFunctionSkipper::exception, spr::find(), tt::Setup::layerId(), mergeVDriftHistosByStation::name, numSeedTypes_, seedingLayers(), seedTypeNames_, seedTypesProjectionLayers_, seedTypesSeedLayers_, and setup_.

Referenced by channelId(), and trklet::DRin::consume().

                                                                                              {
    layerId = -1;
    if (seedType < 0 || seedType >= numSeedTypes_) {
      cms::Exception exception("logic_error");
      exception.addContext("trklet::ChannelAssignment::layerId");
      exception << "TTTracks with with seed type " << seedType << " not supported.";
      throw exception;
    }
    const int layer = setup_->layerId(ttStubRef);
    const vector<int>& seedingLayers = seedTypesSeedLayers_[seedType];
    if (find(seedingLayers.begin(), seedingLayers.end(), layer) != seedingLayers.end())
      return false;
    const vector<int>& projectingLayers = seedTypesProjectionLayers_[seedType];
    const auto pos = find(projectingLayers.begin(), projectingLayers.end(), layer);
    if (pos == projectingLayers.end()) {
      const string& name = seedTypeNames_[seedType];
      cms::Exception exception("logic_error");
      exception.addContext("trklet::ChannelAssignment::layerId");
      exception << "TTStub from layer " << layer << " (barrel: 1-6; discs: 11-15) from seed type " << name
                << " not supported.";
      throw exception;
    }
    layerId = distance(projectingLayers.begin(), pos);
    return true;
  }

layerId() [2/2]

int trklet::ChannelAssignment::layerId	(	int	seedType,
		int	channel
	)		const

Definition at line 206 of file ChannelAssignment.cc.

References numProjectionLayers(), seedTypesProjectionLayers_, and seedTypesSeedLayers_.

                                                                {
    if (channel < numProjectionLayers(seedType))
      return seedTypesProjectionLayers_.at(seedType).at(channel);
    return seedTypesSeedLayers_.at(seedType).at(channel - numProjectionLayers(seedType));
  }

maxNumProjectionLayers()

int trklet::ChannelAssignment::maxNumProjectionLayers ( ) const

inline

Definition at line 54 of file ChannelAssignment.h.

References maxNumProjectionLayers_.

Referenced by L1FPGATrackProducer::produce().

{ return maxNumProjectionLayers_; }

minIdenticalStubs()

int trklet::ChannelAssignment::minIdenticalStubs ( ) const

inline

Definition at line 44 of file ChannelAssignment.h.

References minIdenticalStubs_.

Referenced by trklet::DR::equalEnough().

{ return minIdenticalStubs_; }

nodeDR()

int trklet::ChannelAssignment::nodeDR

(

const TTTrackRef &

ttTrackRef

)

const

Definition at line 190 of file ChannelAssignment.cc.

References b, newFWLiteAna::bin, numNodesDR_, DiDispStaMuonMonitor_cfi::pt, and ptBoundaries_.

Referenced by trklet::DRin::produce().

                                                                  {
    const double pt = ttTrackRef->momentum().perp();
    int bin(0);
    for (double b : ptBoundaries_) {
      if (pt < b)
        break;
      bin++;
    }
    if (ttTrackRef->rInv() >= 0.)
      bin += numNodesDR_ / 2;
    else
      bin = numNodesDR_ / 2 - 1 - bin;
    return bin;
  }

numChannelsStub()

int trklet::ChannelAssignment::numChannelsStub ( ) const

inline

Definition at line 30 of file ChannelAssignment.h.

References numChannelsStub_.

Referenced by trklet::ProducerTBout::produce(), and L1FPGATrackProducer::produce().

{ return numChannelsStub_; }

numChannelsTrack()

int trklet::ChannelAssignment::numChannelsTrack ( ) const

inline

Definition at line 28 of file ChannelAssignment.h.

References numChannelsTrack_.

Referenced by trklet::DRin::consume(), trklet::DRin::produce(), trklet::ProducerTBout::produce(), and L1FPGATrackProducer::produce().

{ return numChannelsTrack_; }

numComparisonModules()

int trklet::ChannelAssignment::numComparisonModules ( ) const

inline

Definition at line 42 of file ChannelAssignment.h.

References numComparisonModules_.

Referenced by trklet::DR::produce().

{ return numComparisonModules_; }

numNodesDR()

int trklet::ChannelAssignment::numNodesDR ( ) const

inline

Definition at line 46 of file ChannelAssignment.h.

References numNodesDR_.

Referenced by trklet::DR::consume(), trklet::KFin::consume(), trklet::DR::produce(), trklet::KFin::produce(), trklet::DRin::produce(), trklet::ProducerDR::produce(), and trklet::ProducerDRin::produce().

{ return numNodesDR_; }

numProjectionLayers()

int trklet::ChannelAssignment::numProjectionLayers ( int  seedType ) const

inline

Definition at line 52 of file ChannelAssignment.h.

References seedTypesProjectionLayers_.

Referenced by trklet::DRin::consume(), layerId(), and L1FPGATrackProducer::produce().

{ return (int)seedTypesProjectionLayers_.at(seedType).size(); }

numSeedingLayers()

int trklet::ChannelAssignment::numSeedingLayers ( ) const

inline

Definition at line 68 of file ChannelAssignment.h.

References numSeedingLayers_.

Referenced by trklet::DRin::consume(), and L1FPGATrackProducer::produce().

{ return numSeedingLayers_; }

numSeedTypes()

int trklet::ChannelAssignment::numSeedTypes ( ) const

inline

Definition at line 48 of file ChannelAssignment.h.

References numSeedTypes_.

{ return numSeedTypes_; }

offsetStub()

int trklet::ChannelAssignment::offsetStub

(

int

channelTrack

)

const

Definition at line 170 of file ChannelAssignment.cc.

References numChannelsStub_, numChannelsTrack_, offsetsStubs_, and nano_mu_digi_cff::region.

Referenced by trklet::DRin::consume(), and L1FPGATrackProducer::produce().

                                                          {
    const int region = channelTrack / numChannelsTrack_;
    const int channel = channelTrack % numChannelsTrack_;
    return region * numChannelsStub_ + offsetsStubs_[channel];
  }

seedingLayers()

const std::vector<int>& trklet::ChannelAssignment::seedingLayers ( int  seedType ) const

inline

Definition at line 60 of file ChannelAssignment.h.

References seedTypesSeedLayers_.

Referenced by layerId().

{ return seedTypesSeedLayers_.at(seedType); }

type()

tt::SensorModule::Type trklet::ChannelAssignment::type ( const TTStubRef &  ttStubRef ) const

inline

Definition at line 62 of file ChannelAssignment.h.

References setup_, and tt::Setup::type().

{ return setup_->type(ttStubRef); }

widthLayerId()

int trklet::ChannelAssignment::widthLayerId ( ) const

inline

Definition at line 32 of file ChannelAssignment.h.

References widthLayerId_.

Referenced by trklet::DR::consume(), trklet::KFin::consume(), and trklet::DRin::produce().

{ return widthLayerId_; }

widthPSTilt()

int trklet::ChannelAssignment::widthPSTilt ( ) const

inline

Definition at line 38 of file ChannelAssignment.h.

References widthPSTilt_.

Referenced by trklet::DR::consume(), and trklet::KFin::consume().

{ return widthPSTilt_; }

widthSeedStubId()

int trklet::ChannelAssignment::widthSeedStubId ( ) const

inline

Definition at line 36 of file ChannelAssignment.h.

References widthSeedStubId_.

Referenced by trklet::DR::consume(), trklet::DRin::consume(), and trklet::DRin::produce().

{ return widthSeedStubId_; }

widthStubId()

int trklet::ChannelAssignment::widthStubId ( ) const

inline

Definition at line 34 of file ChannelAssignment.h.

References widthStubId_.

{ return widthStubId_; }

Member Data Documentation

channelEncoding_

std::vector<int> trklet::ChannelAssignment::channelEncoding_

private

Definition at line 112 of file ChannelAssignment.h.

Referenced by channelEncoding().

depthMemory_

int trklet::ChannelAssignment::depthMemory_

private

Definition at line 86 of file ChannelAssignment.h.

Referenced by depthMemory().

maxNumProjectionLayers_

int trklet::ChannelAssignment::maxNumProjectionLayers_

private

Definition at line 110 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), and maxNumProjectionLayers().

minIdenticalStubs_

int trklet::ChannelAssignment::minIdenticalStubs_

private

Definition at line 94 of file ChannelAssignment.h.

Referenced by minIdenticalStubs().

numChannelsStub_

int trklet::ChannelAssignment::numChannelsStub_

private

Definition at line 104 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), numChannelsStub(), and offsetStub().

numChannelsTrack_

int trklet::ChannelAssignment::numChannelsTrack_

private

Definition at line 102 of file ChannelAssignment.h.

Referenced by numChannelsTrack(), and offsetStub().

numComparisonModules_

int trklet::ChannelAssignment::numComparisonModules_

private

Definition at line 92 of file ChannelAssignment.h.

Referenced by numComparisonModules().

numNodesDR_

int trklet::ChannelAssignment::numNodesDR_

private

Definition at line 96 of file ChannelAssignment.h.

Referenced by nodeDR(), and numNodesDR().

numSeedingLayers_

int trklet::ChannelAssignment::numSeedingLayers_

private

Definition at line 116 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), and numSeedingLayers().

numSeedTypes_

int trklet::ChannelAssignment::numSeedTypes_

private

Definition at line 100 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), channelId(), layerId(), and numSeedTypes().

offsetsStubs_

std::vector<int> trklet::ChannelAssignment::offsetsStubs_

private

Definition at line 114 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), and offsetStub().

pSetDR_

edm::ParameterSet trklet::ChannelAssignment::pSetDR_

private

Definition at line 90 of file ChannelAssignment.h.

pSetDRin_

edm::ParameterSet trklet::ChannelAssignment::pSetDRin_

private

Definition at line 76 of file ChannelAssignment.h.

ptBoundaries_

std::vector<double> trklet::ChannelAssignment::ptBoundaries_

private

Definition at line 88 of file ChannelAssignment.h.

Referenced by nodeDR().

seedTypeNames_

std::vector<std::string> trklet::ChannelAssignment::seedTypeNames_

private

Definition at line 98 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), channelId(), and layerId().

seedTypesProjectionLayers_

std::vector<std::vector<int> > trklet::ChannelAssignment::seedTypesProjectionLayers_

private

Definition at line 108 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), channelId(), layerId(), and numProjectionLayers().

seedTypesSeedLayers_

std::vector<std::vector<int> > trklet::ChannelAssignment::seedTypesSeedLayers_

private

Definition at line 106 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), channelId(), layerId(), and seedingLayers().

setup_

const tt::Setup* trklet::ChannelAssignment::setup_

private

Definition at line 74 of file ChannelAssignment.h.

Referenced by ChannelAssignment(), layerId(), upgradeWorkflowComponents.UpgradeWorkflow::setup(), and type().

widthLayerId_

int trklet::ChannelAssignment::widthLayerId_

private

Definition at line 78 of file ChannelAssignment.h.

Referenced by widthLayerId().

widthPSTilt_

int trklet::ChannelAssignment::widthPSTilt_

private

Definition at line 84 of file ChannelAssignment.h.

Referenced by widthPSTilt().

widthSeedStubId_

int trklet::ChannelAssignment::widthSeedStubId_

private

Definition at line 82 of file ChannelAssignment.h.

Referenced by widthSeedStubId().

widthStubId_

int trklet::ChannelAssignment::widthStubId_

private

Definition at line 80 of file ChannelAssignment.h.

Referenced by widthStubId().