HGCalVFEProcessorSums.cc

Go to the documentation of this file.

#include "L1Trigger/L1THGCal/interface/veryfrontend/HGCalVFEProcessorSums.h"
#include <limits>

#include "DataFormats/Candidate/interface/LeafCandidate.h"
#include "DataFormats/L1THGCal/interface/HGCalTriggerCell.h"

DEFINE_EDM_PLUGIN(HGCalVFEProcessorBaseFactory, HGCalVFEProcessorSums, "HGCalVFEProcessorSums");

HGCalVFEProcessorSums::HGCalVFEProcessorSums(const edm::ParameterSet& conf) : HGCalVFEProcessorBase(conf) {
  vfeLinearizationImpl_ = std::make_unique<HGCalVFELinearizationImpl>(conf);
  vfeSummationImpl_ = std::make_unique<HGCalVFESummationImpl>(conf);
  vfeCompressionImpl_ = std::make_unique<HGCalVFECompressionImpl>(conf);
  calibration_ = std::make_unique<HGCalTriggerCellCalibration>(conf);
}

void HGCalVFEProcessorSums::run(const HGCalDigiCollection& digiColl,
                                l1t::HGCalTriggerCellBxCollection& triggerCellColl,
                                const edm::EventSetup& es) {
  vfeSummationImpl_->eventSetup(es);
  calibration_->eventSetup(es);

  std::vector<HGCalDataFrame> dataframes;
  std::vector<std::pair<DetId, uint32_t>> linearized_dataframes;
  std::unordered_map<uint32_t, uint32_t> payload;
  std::unordered_map<uint32_t, std::array<uint32_t, 2>> compressed_payload;

  // convert ee and fh hit collections into the same object
  for (const auto& digiData : digiColl) {
    if (DetId(digiData.id()).det() == DetId::Hcal && HcalDetId(digiData.id()).subdetId() != HcalEndcap)
      continue;
    uint32_t module = geometry_->getModuleFromCell(digiData.id());
    if (geometry_->disconnectedModule(module))
      continue;
    dataframes.emplace_back(digiData.id());
    for (int i = 0; i < digiData.size(); i++) {
      dataframes.back().setSample(i, digiData.sample(i));
    }
  }

  vfeLinearizationImpl_->linearize(dataframes, linearized_dataframes);
  vfeSummationImpl_->triggerCellSums(*geometry_, linearized_dataframes, payload);
  vfeCompressionImpl_->compress(payload, compressed_payload);

  // Transform map to trigger cell vector vector<HGCalTriggerCell>
  for (const auto& id_value : payload) {
    if (id_value.second > 0) {
      l1t::HGCalTriggerCell triggerCell(
          reco::LeafCandidate::LorentzVector(), compressed_payload[id_value.first][1], 0, 0, 0, id_value.first);
      triggerCell.setCompressedCharge(compressed_payload[id_value.first][0]);
      triggerCell.setUncompressedCharge(id_value.second);
      GlobalPoint point = geometry_->getTriggerCellPosition(id_value.first);

      // 'value' is hardware, so p4 is meaningless, except for eta and phi
      math::PtEtaPhiMLorentzVector p4((double)id_value.second / cosh(point.eta()), point.eta(), point.phi(), 0.);
      triggerCell.setP4(p4);
      triggerCell.setPosition(point);

      // calibration part ---------------------------
      if (triggerCell.hwPt() > 0) {
        l1t::HGCalTriggerCell calibratedtriggercell(triggerCell);
        calibration_->calibrateInGeV(calibratedtriggercell);
        triggerCellColl.push_back(0, calibratedtriggercell);
      }
    }
  }
}