CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes
SiPixelDigisClustersFromSoAT< TrackerTraits > Class Template Reference
Inheritance diagram for SiPixelDigisClustersFromSoAT< TrackerTraits >:
edm::global::EDProducer<> edm::global::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 SiPixelDigisClustersFromSoAT (const edm::ParameterSet &iConfig)
 
 ~SiPixelDigisClustersFromSoAT () override=default
 
- Public Member Functions inherited from edm::global::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
 
EDProduceroperator= (const EDProducer &)=delete
 
bool wantsGlobalLuminosityBlocks () const final
 
bool wantsGlobalRuns () const final
 
bool wantsInputProcessBlocks () const final
 
bool wantsProcessBlocks () const final
 
bool wantsStreamLuminosityBlocks () const final
 
bool wantsStreamRuns () const final
 
- Public Member Functions inherited from edm::global::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
 ~EDProducerBase () override
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
std::vector< edm::ProductResolverIndex > const & putTokenIndexToProductResolverIndex () const
 
std::vector< bool > const & recordProvenanceList () const
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
 
TypeLabelList const & typeLabelList () const
 used by the fwk to register the list of products of this module More...
 
 ~ProducerBase () noexcept(false) override
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ESResolverIndex const * esGetTokenIndices (edm::Transition iTrans) const
 
std::vector< ESResolverIndex > const & esGetTokenIndicesVector (edm::Transition iTrans) const
 
std::vector< ESRecordIndex > const & esGetTokenRecordIndicesVector (edm::Transition iTrans) const
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFrom (BranchType iType) const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::array< std::vector< ModuleDescription const *> *, NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, ProductRegistry const &preg, std::map< std::string, ModuleDescription const *> const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
void selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)
 
ProductResolverIndexAndSkipBit uncheckedIndexFrom (EDGetToken) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
void updateLookup (eventsetup::ESRecordsToProductResolverIndices const &)
 
virtual ~EDConsumerBase () noexcept(false)
 

Static Public Member Functions

static void fillDescriptions (edm::ConfigurationDescriptions &descriptions)
 
- Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 

Private Member Functions

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

Private Attributes

edm::EDPutTokenT< SiPixelClusterCollectionNewclusterPutToken_
 
const SiPixelClusterThresholds clusterThresholds_
 
edm::EDGetTokenT< legacy::SiPixelDigisSoAdigiGetToken_
 
edm::EDPutTokenT< edm::DetSetVector< PixelDigi > > digiPutToken_
 
const bool produceDigis_
 
const bool storeDigis_
 
const edm::ESGetToken< TrackerTopology, TrackerTopologyRcdtopoToken_
 

Additional Inherited Members

- Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase ModuleType
 
- Public Types inherited from edm::ProducerBase
template<typename T >
using BranchAliasSetterT = ProductRegistryHelper::BranchAliasSetterT< T >
 
using ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex > >
 
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Protected Member Functions inherited from edm::ProducerBase
template<Transition Tr = Transition::Event>
auto produces (std::string instanceName) noexcept
 declare what type of product will make and with which optional label More...
 
template<Transition B>
BranchAliasSetter produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
 
template<BranchType B>
BranchAliasSetter produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
 
BranchAliasSetter produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
 
template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType > produces (std::string instanceName)
 
template<class ProductType >
BranchAliasSetterT< ProductType > produces ()
 
template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType > produces (std::string instanceName)
 
template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType > produces ()
 
template<class ProductType >
BranchAliasSetterT< ProductType > produces (std::string instanceName)
 
template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType > produces ()
 
template<Transition Tr = Transition::Event>
auto produces () noexcept
 
ProducesCollector producesCollector ()
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
template<BranchType B = InEvent>
EDConsumerBaseAdaptor< Bconsumes (edm::InputTag tag) noexcept
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto esConsumes ()
 
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto esConsumes (ESInputTag const &tag)
 
template<Transition Tr = Transition::Event>
constexpr auto esConsumes ()
 
template<Transition Tr = Transition::Event>
auto esConsumes (ESInputTag tag)
 
template<Transition Tr = Transition::Event>
ESGetTokenGeneric esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
 Used with EventSetupRecord::doGet. More...
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
void resetItemsToGetFrom (BranchType iType)
 

Detailed Description

template<typename TrackerTraits>
class SiPixelDigisClustersFromSoAT< TrackerTraits >

Definition at line 27 of file SiPixelDigisClustersFromSoA.cc.

Constructor & Destructor Documentation

◆ SiPixelDigisClustersFromSoAT()

template<typename TrackerTraits >
SiPixelDigisClustersFromSoAT< TrackerTraits >::SiPixelDigisClustersFromSoAT ( const edm::ParameterSet iConfig)
explicit

Definition at line 51 of file SiPixelDigisClustersFromSoA.cc.

References SiPixelDigisClustersFromSoAT< TrackerTraits >::digiPutToken_, and SiPixelDigisClustersFromSoAT< TrackerTraits >::produceDigis_.

53  digiGetToken_(consumes<legacy::SiPixelDigisSoA>(iConfig.getParameter<edm::InputTag>("src"))),
54  clusterPutToken_(produces<SiPixelClusterCollectionNew>()),
55  clusterThresholds_(iConfig.getParameter<int>("clusterThreshold_layer1"),
56  iConfig.getParameter<int>("clusterThreshold_otherLayers")),
57  produceDigis_(iConfig.getParameter<bool>("produceDigis")),
58  storeDigis_(iConfig.getParameter<bool>("produceDigis") && iConfig.getParameter<bool>("storeDigis")) {
59  if (produceDigis_)
60  digiPutToken_ = produces<edm::DetSetVector<PixelDigi>>();
61 }
T getParameter(std::string const &) const
Definition: ParameterSet.h:307
const edm::ESGetToken< TrackerTopology, TrackerTopologyRcd > topoToken_
const SiPixelClusterThresholds clusterThresholds_
edm::EDPutTokenT< edm::DetSetVector< PixelDigi > > digiPutToken_
edm::EDGetTokenT< legacy::SiPixelDigisSoA > digiGetToken_
edm::EDPutTokenT< SiPixelClusterCollectionNew > clusterPutToken_

◆ ~SiPixelDigisClustersFromSoAT()

template<typename TrackerTraits >
SiPixelDigisClustersFromSoAT< TrackerTraits >::~SiPixelDigisClustersFromSoAT ( )
overridedefault

Member Function Documentation

◆ fillDescriptions()

template<typename TrackerTraits >
void SiPixelDigisClustersFromSoAT< TrackerTraits >::fillDescriptions ( edm::ConfigurationDescriptions descriptions)
static

Definition at line 64 of file SiPixelDigisClustersFromSoA.cc.

References edm::ConfigurationDescriptions::addWithDefaultLabel(), gpuClustering::clusterThresholdLayerOne, gpuClustering::clusterThresholdOtherLayers, submitPVResolutionJobs::desc, and ProducerED_cfi::InputTag.

64  {
66  desc.add<edm::InputTag>("src", edm::InputTag("siPixelDigisSoA"));
67  desc.add<int>("clusterThreshold_layer1", gpuClustering::clusterThresholdLayerOne);
68  desc.add<int>("clusterThreshold_otherLayers", gpuClustering::clusterThresholdOtherLayers);
69  desc.add<bool>("produceDigis", true);
70  desc.add<bool>("storeDigis", true);
71 
72  descriptions.addWithDefaultLabel(desc);
73 }
void addWithDefaultLabel(ParameterSetDescription const &psetDescription)
constexpr uint16_t clusterThresholdOtherLayers
constexpr uint16_t clusterThresholdLayerOne

◆ produce()

template<typename TrackerTraits >
void SiPixelDigisClustersFromSoAT< TrackerTraits >::produce ( edm::StreamID  ,
edm::Event iEvent,
const edm::EventSetup iSetup 
) const
overrideprivatevirtual

Implements edm::global::EDProducerBase.

Definition at line 76 of file SiPixelDigisClustersFromSoA.cc.

References edmNew::DetSetVector< T >::FastFiller::abort(), PixelClusterizerBase::AccretionCluster::add(), cms::cuda::assert(), edmNew::DetSetVector< T >::FastFiller::back(), edmNew::DetSetVector< T >::FastFiller::begin(), GetRecoTauVFromDQM_MC_cff::cl2, PixelClusterizerBase::AccretionCluster::clear(), EMEnrichingFilter_cfi::clusterThreshold, cuy::col, universalConfigTemplate::collection, ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), gather_cfg::cout, hcalRecHitTable_cff::detId, edmNew::DetSetVector< T >::FastFiller::emplace_back(), edmNew::DetSetVector< T >::FastFiller::empty(), edmNew::DetSetVector< T >::FastFiller::end(), edm::EventSetup::getData(), mps_fire::i, iEvent, nano_mu_digi_cff::layer, LogDebug, SiStripPI::max, pixelClustering::maxNumClustersPerModules, SiPixelCluster::minPixelRow(), eostools::move(), phase1PixelTopology::numberOfModules, slimmedTrackExtras_cff::outputClusters, and DetId::subdetId().

78  {
79  const auto& digis = iEvent.get(digiGetToken_);
80  const uint32_t nDigis = digis.size();
81  const auto& ttopo = iSetup.getData(topoToken_);
82  constexpr auto maxModules = TrackerTraits::numberOfModules;
83 
84  std::unique_ptr<edm::DetSetVector<PixelDigi>> collection;
85  if (produceDigis_)
86  collection = std::make_unique<edm::DetSetVector<PixelDigi>>();
87  if (storeDigis_)
88  collection->reserve(maxModules);
89  auto outputClusters = std::make_unique<SiPixelClusterCollectionNew>();
90  outputClusters->reserve(maxModules, nDigis / 2);
91 
92  edm::DetSet<PixelDigi>* detDigis = nullptr;
93  uint32_t detId = 0;
94  for (uint32_t i = 0; i < nDigis; i++) {
95  // check for uninitialized digis
96  // this is set in RawToDigi_kernel in SiPixelRawToClusterGPUKernel.cu
97  if (digis.rawIdArr(i) == 0)
98  continue;
99  // check for noisy/dead pixels (electrons set to 0)
100  if (digis.adc(i) == 0)
101  continue;
102 
103  detId = digis.rawIdArr(i);
104  if (storeDigis_) {
105  detDigis = &collection->find_or_insert(detId);
106  if ((*detDigis).empty())
107  (*detDigis).data.reserve(64); // avoid the first relocations
108  }
109  break;
110  }
111 
112  int32_t nclus = -1;
114 #ifdef EDM_ML_DEBUG
115  auto totClustersFilled = 0;
116 #endif
117 
118  auto fillClusters = [&](uint32_t detId) {
119  if (nclus < 0)
120  return; // this in reality should never happen
122  auto layer = (DetId(detId).subdetId() == 1) ? ttopo.pxbLayer(detId) : 0;
124  for (int32_t ic = 0; ic < nclus + 1; ++ic) {
125  auto const& acluster = aclusters[ic];
126  // in any case we cannot go out of sync with gpu...
127  if (acluster.charge < clusterThreshold)
128  edm::LogWarning("SiPixelDigisClustersFromSoA") << "cluster below charge Threshold "
129  << "Layer/DetId/clusId " << layer << '/' << detId << '/' << ic
130  << " size/charge " << acluster.isize << '/' << acluster.charge;
131  // sort by row (x)
132  spc.emplace_back(acluster.isize, acluster.adc, acluster.x, acluster.y, acluster.xmin, acluster.ymin, ic);
133  aclusters[ic].clear();
134 #ifdef EDM_ML_DEBUG
135  ++totClustersFilled;
136  const auto& cluster{spc.back()};
137  LogDebug("SiPixelDigisClustersFromSoA")
138  << "putting in this cluster " << ic << " " << cluster.charge() << " " << cluster.pixelADC().size();
139 #endif
140  std::push_heap(spc.begin(), spc.end(), [](SiPixelCluster const& cl1, SiPixelCluster const& cl2) {
141  return cl1.minPixelRow() < cl2.minPixelRow();
142  });
143  }
144  nclus = -1;
145  // sort by row (x)
146  std::sort_heap(spc.begin(), spc.end(), [](SiPixelCluster const& cl1, SiPixelCluster const& cl2) {
147  return cl1.minPixelRow() < cl2.minPixelRow();
148  });
149  if (spc.empty())
150  spc.abort();
151  };
152 
153 #ifdef GPU_DEBUG
154  std::cout << "Dumping all digis. nDigis = " << nDigis << std::endl;
155 #endif
156 
157  for (uint32_t i = 0; i < nDigis; i++) {
158  // check for uninitialized digis
159  if (digis.rawIdArr(i) == 0)
160  continue;
161  // check for noisy/dead pixels (electrons set to 0)
162  if (digis.adc(i) == 0)
163  continue;
164  if (digis.clus(i) > 9000)
165  continue; // not in cluster; TODO add an assert for the size
166 #ifdef EDM_ML_DEBUG
167  assert(digis.rawIdArr(i) > 109999);
168 #endif
169  if (detId != digis.rawIdArr(i)) {
170 #ifdef GPU_DEBUG
171  std::cout << ">> Closed module --" << detId << "; nclus = " << nclus << std::endl;
172 #endif
173  // new module
174  fillClusters(detId);
175 #ifdef EDM_ML_DEBUG
176  assert(nclus == -1);
177 #endif
178  detId = digis.rawIdArr(i);
179  if (storeDigis_) {
180  detDigis = &collection->find_or_insert(detId);
181  if ((*detDigis).empty())
182  (*detDigis).data.reserve(64); // avoid the first relocations
183  else {
184  edm::LogWarning("SiPixelDigisClustersFromSoA")
185  << "Problem det present twice in input! " << (*detDigis).detId();
186  }
187  }
188  }
189  PixelDigi dig(digis.pdigi(i));
190 
191 #ifdef GPU_DEBUG
192  std::cout << i << ";" << digis.rawIdArr(i) << ";" << digis.clus(i) << ";" << digis.pdigi(i) << ";" << digis.adc(i)
193  << ";" << dig.row() << ";" << dig.column() << std::endl;
194 #endif
195 
196  if (storeDigis_)
197  (*detDigis).data.emplace_back(dig);
198  // fill clusters
199 #ifdef EDM_ML_DEBUG
200  assert(digis.clus(i) >= 0);
201  assert(digis.clus(i) < static_cast<int32_t>(TrackerTraits::maxNumClustersPerModules));
202 #endif
203  nclus = std::max(digis.clus(i), nclus);
204  auto row = dig.row();
205  auto col = dig.column();
206  SiPixelCluster::PixelPos pix(row, col);
207  aclusters[digis.clus(i)].add(pix, digis.adc(i));
208  }
209 
210  // fill final clusters
211  if (detId > 0)
212  fillClusters(detId);
213 
214 #ifdef EDM_ML_DEBUG
215  LogDebug("SiPixelDigisClustersFromSoA") << "filled " << totClustersFilled << " clusters";
216 #endif
217 
218  if (produceDigis_)
221 }
T const & getData(const ESGetToken< T, R > &iToken) const noexcept(false)
Definition: EventSetup.h:119
assert(be >=bs)
const edm::ESGetToken< TrackerTopology, TrackerTopologyRcd > topoToken_
constexpr uint16_t numberOfModules
int minPixelRow() const
constexpr int32_t getThresholdForLayerOnCondition(bool isLayer1) const noexcept
int iEvent
Definition: GenABIO.cc:224
const SiPixelClusterThresholds clusterThresholds_
constexpr int subdetId() const
get the contents of the subdetector field (not cast into any detector&#39;s numbering enum) ...
Definition: DetId.h:48
Definition: DetId.h:17
Pixel cluster – collection of neighboring pixels above threshold.
constexpr int32_t maxNumClustersPerModules
edm::EDPutTokenT< edm::DetSetVector< PixelDigi > > digiPutToken_
col
Definition: cuy.py:1009
edm::EDGetTokenT< legacy::SiPixelDigisSoA > digiGetToken_
bool add(SiPixelCluster::PixelPos const &p, uint16_t const iadc)
Log< level::Warning, false > LogWarning
edm::EDPutTokenT< SiPixelClusterCollectionNew > clusterPutToken_
def move(src, dest)
Definition: eostools.py:511
#define LogDebug(id)

Member Data Documentation

◆ clusterPutToken_

template<typename TrackerTraits >
edm::EDPutTokenT<SiPixelClusterCollectionNew> SiPixelDigisClustersFromSoAT< TrackerTraits >::clusterPutToken_
private

Definition at line 42 of file SiPixelDigisClustersFromSoA.cc.

◆ clusterThresholds_

template<typename TrackerTraits >
const SiPixelClusterThresholds SiPixelDigisClustersFromSoAT< TrackerTraits >::clusterThresholds_
private

Definition at line 44 of file SiPixelDigisClustersFromSoA.cc.

◆ digiGetToken_

template<typename TrackerTraits >
edm::EDGetTokenT<legacy::SiPixelDigisSoA> SiPixelDigisClustersFromSoAT< TrackerTraits >::digiGetToken_
private

Definition at line 39 of file SiPixelDigisClustersFromSoA.cc.

◆ digiPutToken_

template<typename TrackerTraits >
edm::EDPutTokenT<edm::DetSetVector<PixelDigi> > SiPixelDigisClustersFromSoAT< TrackerTraits >::digiPutToken_
private

◆ produceDigis_

template<typename TrackerTraits >
const bool SiPixelDigisClustersFromSoAT< TrackerTraits >::produceDigis_
private

◆ storeDigis_

template<typename TrackerTraits >
const bool SiPixelDigisClustersFromSoAT< TrackerTraits >::storeDigis_
private

Definition at line 47 of file SiPixelDigisClustersFromSoA.cc.

◆ topoToken_

template<typename TrackerTraits >
const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> SiPixelDigisClustersFromSoAT< TrackerTraits >::topoToken_
private

Definition at line 37 of file SiPixelDigisClustersFromSoA.cc.