File indexing completed on 2025-06-20 01:53:36
0001 #include "L1Trigger/Phase2L1ParticleFlow/interface/regionizer/tdr_regionizer_ref.h"
0002 #include "L1Trigger/Phase2L1ParticleFlow/interface/regionizer/tdr_regionizer_elements_ref.icc"
0003
0004 #include <iostream>
0005
0006 #ifdef CMSSW_GIT_HASH
0007 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0008 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0009
0010 l1ct::TDRRegionizerEmulator::TDRRegionizerEmulator(const edm::ParameterSet& iConfig)
0011 : TDRRegionizerEmulator(iConfig.getParameter<uint32_t>("nTrack"),
0012 iConfig.getParameter<uint32_t>("nCalo"),
0013 iConfig.getParameter<uint32_t>("nEmCalo"),
0014 iConfig.getParameter<uint32_t>("nMu"),
0015 iConfig.getParameter<uint32_t>("nClocks"),
0016 iConfig.getParameter<std::vector<int32_t>>("bigRegionEdges"),
0017 iConfig.getParameter<bool>("doSort")) {
0018 debug_ = iConfig.getUntrackedParameter<bool>("debug");
0019 }
0020
0021 edm::ParameterSetDescription l1ct::TDRRegionizerEmulator::getParameterSetDescription() {
0022 edm::ParameterSetDescription description;
0023 description.add<uint32_t>("nClocks", 162);
0024 description.add<uint32_t>("nTrack", 22);
0025 description.add<uint32_t>("nCalo", 15);
0026 description.add<uint32_t>("nEmCalo", 12);
0027 description.add<uint32_t>("nMu", 2);
0028 description.add<bool>("doSort", false);
0029 description.add<std::vector<int32_t>>("bigRegionEdges", {-560, -80, 400, -560});
0030 description.addUntracked<bool>("debug", false);
0031 return description;
0032 }
0033 #endif
0034
0035 l1ct::TDRRegionizerEmulator::TDRRegionizerEmulator(uint32_t ntk,
0036 uint32_t ncalo,
0037 uint32_t nem,
0038 uint32_t nmu,
0039 uint32_t nclocks,
0040 std::vector<int32_t> bigRegionEdges,
0041 bool dosort)
0042 : RegionizerEmulator(),
0043 ntk_(ntk),
0044 ncalo_(ncalo),
0045 nem_(nem),
0046 nmu_(nmu),
0047 nclocks_(nclocks),
0048 bigRegionEdges_(bigRegionEdges),
0049 dosort_(dosort),
0050 netaInBR_(6),
0051 nphiInBR_(3),
0052 init_(false) {
0053 nBigRegions_ = bigRegionEdges_.size() - 1;
0054 }
0055
0056 l1ct::TDRRegionizerEmulator::~TDRRegionizerEmulator() {}
0057
0058 void l1ct::TDRRegionizerEmulator::initSectorsAndRegions(const RegionizerDecodedInputs& in,
0059 const std::vector<PFInputRegion>& out) {
0060 if (debug_) {
0061 dbgCout() << "doing init, out_size = " << out.size() << std::endl;
0062 }
0063 assert(!init_);
0064 init_ = true;
0065
0066 for (unsigned int i = 0; i < nBigRegions_; i++) {
0067 tkRegionizers_.emplace_back(
0068 netaInBR_, nphiInBR_, ntk_, bigRegionEdges_[i], bigRegionEdges_[i + 1], nclocks_, 1, false);
0069 hadCaloRegionizers_.emplace_back(
0070 netaInBR_, nphiInBR_, ncalo_, bigRegionEdges_[i], bigRegionEdges_[i + 1], nclocks_ / 3, 1, false);
0071 emCaloRegionizers_.emplace_back(
0072 netaInBR_, nphiInBR_, nem_, bigRegionEdges_[i], bigRegionEdges_[i + 1], nclocks_ / 3, 1, false);
0073 muRegionizers_.emplace_back(
0074 netaInBR_, nphiInBR_, nmu_, bigRegionEdges_[i], bigRegionEdges_[i + 1], nclocks_, 1, false);
0075 }
0076
0077 if (debug_) {
0078 dbgCout() << "in.track.size() = " << in.track.size() << std::endl;
0079 dbgCout() << "in.hadcalo.size() = " << in.hadcalo.size() << std::endl;
0080 dbgCout() << "in.emcalo.size() = " << in.emcalo.size() << std::endl;
0081 }
0082
0083 if (ntk_) {
0084 for (unsigned int i = 0; i < nBigRegions_; i++) {
0085 tkRegionizers_[i].initSectors(in.track);
0086 tkRegionizers_[i].initRegions(out);
0087 }
0088 }
0089 if (ncalo_) {
0090 for (unsigned int i = 0; i < nBigRegions_; i++) {
0091 hadCaloRegionizers_[i].initSectors(in.hadcalo);
0092 hadCaloRegionizers_[i].initRegions(out);
0093 }
0094 }
0095 if (nem_) {
0096 for (unsigned int i = 0; i < nBigRegions_; i++) {
0097 emCaloRegionizers_[i].initSectors(in.emcalo);
0098 emCaloRegionizers_[i].initRegions(out);
0099 }
0100 }
0101 if (nmu_) {
0102 for (unsigned int i = 0; i < nBigRegions_; i++) {
0103 muRegionizers_[i].initSectors(in.muon);
0104 muRegionizers_[i].initRegions(out);
0105 }
0106 }
0107 }
0108
0109 void l1ct::TDRRegionizerEmulator::run(const RegionizerDecodedInputs& in, std::vector<PFInputRegion>& out) {
0110 if (debug_) {
0111 dbgCout() << "TDRRegionizerEmulator::run called, out.size = " << out.size() << std::endl;
0112 }
0113
0114 if (!init_) {
0115 initSectorsAndRegions(in, out);
0116 }
0117
0118 for (unsigned int ie = 0; ie < nBigRegions_; ie++) {
0119
0120 tkRegionizers_[ie].reset();
0121 tkRegionizers_[ie].fillBuffers(in.track);
0122 tkRegionizers_[ie].run();
0123
0124 emCaloRegionizers_[ie].reset();
0125 emCaloRegionizers_[ie].fillBuffers(in.emcalo);
0126 emCaloRegionizers_[ie].run();
0127
0128 hadCaloRegionizers_[ie].reset();
0129 hadCaloRegionizers_[ie].fillBuffers(in.hadcalo);
0130 hadCaloRegionizers_[ie].run();
0131
0132 muRegionizers_[ie].reset();
0133 muRegionizers_[ie].fillBuffers(in.muon);
0134 muRegionizers_[ie].run();
0135 }
0136
0137 for (unsigned int ie = 0; ie < nBigRegions_; ie++) {
0138 auto regionTrackMap = tkRegionizers_[ie].fillRegions(dosort_);
0139 for (auto& pr : regionTrackMap) {
0140 out[pr.first].track = pr.second;
0141 }
0142 auto regionEmCaloMap = emCaloRegionizers_[ie].fillRegions(dosort_);
0143 for (auto& pr : regionEmCaloMap) {
0144 out[pr.first].emcalo = pr.second;
0145 }
0146 auto regionHadCaloMap = hadCaloRegionizers_[ie].fillRegions(dosort_);
0147 for (auto& pr : regionHadCaloMap) {
0148 out[pr.first].hadcalo = pr.second;
0149 }
0150 auto regionMuMap = muRegionizers_[ie].fillRegions(dosort_);
0151 for (auto& pr : regionMuMap) {
0152 out[pr.first].muon = pr.second;
0153 }
0154 }
0155 }
