Geometry/HGCalMapping/plugins/HGCalMappingTriggerESProducer.cc - master - githubmirror/cmssw

Geometry/HGCalMapping/plugins/HGCalMappingTriggerESProducer.cc
134 строки6 KB
Christopher Jones
Move to EventSetupRecordInfiniteIntervalFinder
16 июн 2026, 21:21
16 июн 2026, 21:21419a1ec
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/SourceFactory.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESProducer.h"
#include "FWCore/Framework/interface/ESProducts.h"
#include "FWCore/Framework/interface/EventSetupRecordInfiniteIntervalFinder.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/do_nothing_deleter.h"
#include "CondFormats/DataRecord/interface/HGCalElectronicsMappingRcd.h"

#include "CondFormats/HGCalObjects/interface/HGCalMappingModuleIndexerTrigger.h"
#include "CondFormats/HGCalObjects/interface/HGCalMappingCellIndexerTrigger.h"
#include "CondFormats/HGCalObjects/interface/HGCalMappingParameterHost.h"

#include "DataFormats/HGCalDigi/interface/HGCalElectronicsId.h"
#include "DataFormats/ForwardDetId/interface/HGCSiliconDetId.h"
#include "DataFormats/ForwardDetId/interface/HGCScintillatorDetId.h"
#include "Geometry/HGCalMapping/interface/HGCalMappingTools.h"
#include <regex>  // regular expression

/**
   @short plugin parses the module/cell locator files to produce the indexer records
 */
class HGCalMappingTriggerESProducer : public edm::ESProducer, public edm::EventSetupRecordInfiniteIntervalFinder {
public:
  explicit HGCalMappingTriggerESProducer(const edm::ParameterSet& iConfig) {
    //parse the files and hold the list of entities in memory
    modulesMap_.buildFrom(iConfig.getParameter<edm::FileInPath>("modules").fullPath());
    sicellsMap_.buildFrom(iConfig.getParameter<edm::FileInPath>("si").fullPath());
    sipmCellsMap_.buildFrom(iConfig.getParameter<edm::FileInPath>("sipm").fullPath());

    setWhatProduced(this, &HGCalMappingTriggerESProducer::produceCellMapIndexer);
    setWhatProduced(this, &HGCalMappingTriggerESProducer::produceModuleMapIndexer);

    findingRecord<HGCalElectronicsMappingRcd>();

    prepareCellMapperIndexer();
    prepareModuleMapperIndexer();
  }

  std::shared_ptr<HGCalMappingModuleIndexerTrigger> produceModuleMapIndexer(const HGCalElectronicsMappingRcd&) {
    return std::shared_ptr<HGCalMappingModuleIndexerTrigger>(&modIndexer_, edm::do_nothing_deleter());
  }

  std::shared_ptr<HGCalMappingCellIndexerTrigger> produceCellMapIndexer(const HGCalElectronicsMappingRcd&) {
    return std::shared_ptr<HGCalMappingCellIndexerTrigger>(&cellIndexer_, edm::do_nothing_deleter());
  }

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;
    desc.add<edm::FileInPath>("modules")->setComment("module locator file");
    desc.add<edm::FileInPath>("si")->setComment("file containing the mapping of the readout cells in Si modules");
    desc.add<edm::FileInPath>("sipm")->setComment(
        "file containing the mapping of the readout cells in SiPM-on-tile modules");
    descriptions.addWithDefaultLabel(desc);
  }

private:
  void prepareCellMapperIndexer();
  void prepareModuleMapperIndexer();

  hgcal::mappingtools::HGCalEntityList modulesMap_, sicellsMap_, sipmCellsMap_;
  HGCalMappingCellIndexerTrigger cellIndexer_;
  HGCalMappingModuleIndexerTrigger modIndexer_;
};

//
void HGCalMappingTriggerESProducer::prepareCellMapperIndexer() {
  for (size_t i = 0; i < 2; i++) {
    const auto& pmap = i == 0 ? sicellsMap_ : sipmCellsMap_;
    const auto& entities = pmap.getEntries();
    for (const auto& row : entities) {
      auto typecode = pmap.getAttr("Typecode", row);
      int ROC = pmap.getIntAttr("ROC", row);
      int trLink = pmap.getIntAttr("TrLink", row);
      int trCell = pmap.getIntAttr("TrCell", row);
      cellIndexer_.processNewCell(typecode, ROC, trLink, trCell);
    }
  }

  // all {hex,tile}board types are loaded finalize the mapping indexer
  cellIndexer_.update();
}

//
void HGCalMappingTriggerESProducer::prepareModuleMapperIndexer() {
  //default values to assign in case module type has not yet been mapped
  //a high density module (max possible) will be assigned so that the mapping doesn't block
  auto defaultTypeCodeIdx = cellIndexer_.getEnumFromTypecode("MH-F");
  auto typecodeidx = defaultTypeCodeIdx;
  auto defaultNTrLinks = cellIndexer_.getNTrLinkExpectedFor(defaultTypeCodeIdx);
  auto nTrLinks = defaultNTrLinks;
  auto defaultTypeNTCs = cellIndexer_.getNWordsExpectedFor(defaultTypeCodeIdx);
  auto nwords = defaultTypeNTCs;

  const auto& entities = modulesMap_.getEntries();
  for (const auto& row : entities) {
    auto typecode = modulesMap_.getAttr("typecode", row);  // module type code
    std::string wtypecode;                                 // wafer type code

    // match module type code to regular expression pattern (MM-TTTT-LL-NNNN)
    // see https://edms.cern.ch/ui/#!master/navigator/document?D:101059405:101148061:subDocs
    const std::regex typecode_regex_si("(([MX])([LH])-([FTBLR5])).*");  // MM-T*

    // match typecode to regular expression SiPM (TX-LYYSZ-NNNN)
    // https://indico.cern.ch/event/1558202/contributions/6567912/attachments/3092451/5477467/hgcalweek_DPG_2025.pdf : slide 8
    const std::regex typecode_regex_sipm("(T[HL]-L[0-9]+S[123]).*");  // SiPM typecode format TM-
    std::smatch typecode_match_si, typecode_match_sipm;               // match object for string objects

    bool matched_si = std::regex_match(typecode, typecode_match_si, typecode_regex_si);
    bool matched_sipm = std::regex_match(typecode, typecode_match_sipm, typecode_regex_sipm);
    if (matched_si) {
      wtypecode = typecode_match_si[1].str();  // wafer type following MM-T pattern, e.g. "MH-F"
    } else if (matched_sipm) {
      wtypecode = typecode_match_sipm[1].str();
    } else {
      edm::LogWarning("HGCalMappingIndexESSource")
          << "Could not match module type code to expected pattern: " << typecode;
    }

    typecodeidx = cellIndexer_.getEnumFromTypecode(wtypecode);
    nTrLinks = cellIndexer_.getNTrLinkExpectedFor(wtypecode);
    nwords = cellIndexer_.getNWordsExpectedFor(wtypecode);

    int fedid = modulesMap_.getIntAttr("trig_fedid", row);
    int econtidx = modulesMap_.getIntAttr("econtidx", row);
    modIndexer_.processNewModule(fedid, econtidx, typecodeidx, nTrLinks, nwords, typecode);
  }

  modIndexer_.finalize();
}

DEFINE_FWK_EVENTSETUP_SOURCE(HGCalMappingTriggerESProducer);