HGCalTriggerGeometryESProducer.cc

Go to the documentation of this file.

#include <memory>

#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducer.h"
#include "FWCore/Framework/interface/ESProducts.h"

#include "Geometry/Records/interface/CaloGeometryRecord.h"

#include "L1Trigger/L1THGCal/interface/HGCalTriggerGeometryBase.h"

class HGCalTriggerGeometryESProducer : public edm::ESProducer {
public:
  HGCalTriggerGeometryESProducer(const edm::ParameterSet&);
  ~HGCalTriggerGeometryESProducer() override;

  typedef std::unique_ptr<HGCalTriggerGeometryBase> ReturnType;

  ReturnType produce(const CaloGeometryRecord&);

private:
  edm::ParameterSet geometry_config_;
  std::string geometry_name_;
  edm::ESGetToken<CaloGeometry, CaloGeometryRecord> calo_geometry_token_;
  edm::ESGetToken<HGCalGeometry, IdealGeometryRecord> ee_geometry_token_;
  edm::ESGetToken<HGCalGeometry, IdealGeometryRecord> hsi_geometry_token_;
  edm::ESGetToken<HGCalGeometry, IdealGeometryRecord> hsc_geometry_token_;
  bool isV9Geometry_;
};

HGCalTriggerGeometryESProducer::HGCalTriggerGeometryESProducer(const edm::ParameterSet& iConfig)
    : geometry_config_(iConfig.getParameterSet("TriggerGeometry")),
      geometry_name_(geometry_config_.getParameter<std::string>("TriggerGeometryName")),
      isV9Geometry_(iConfig.getParameter<bool>("isV9Geometry")) {
  auto cc = setWhatProduced(this);
  if (isV9Geometry_) {
    cc.setConsumes(ee_geometry_token_, edm::ESInputTag{"", "HGCalEESensitive"})
        .setConsumes(hsi_geometry_token_, edm::ESInputTag{"", "HGCalHESiliconSensitive"})
        .setConsumes(hsc_geometry_token_, edm::ESInputTag{"", "HGCalHEScintillatorSensitive"});
  } else {
    cc.setConsumes(calo_geometry_token_);
  }
}

HGCalTriggerGeometryESProducer::~HGCalTriggerGeometryESProducer() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
}

HGCalTriggerGeometryESProducer::ReturnType HGCalTriggerGeometryESProducer::produce(const CaloGeometryRecord& iRecord) {
  //using namespace edm::es;
  ReturnType geometry(HGCalTriggerGeometryFactory::get()->create(geometry_name_, geometry_config_));
  if (isV9Geometry_) {
    // Initialize trigger geometry for V9 HGCAL geometry
    geometry->initialize(
        &iRecord.get(ee_geometry_token_), &iRecord.get(hsi_geometry_token_), &iRecord.get(hsc_geometry_token_));
  } else {
    // Initialize trigger geometry for V7/V8 HGCAL geometry
    const auto& calo_geometry = iRecord.get(calo_geometry_token_);
    if (not(calo_geometry.getSubdetectorGeometry(DetId::Forward, HGCEE) &&
            calo_geometry.getSubdetectorGeometry(DetId::Forward, HGCHEF) &&
            calo_geometry.getSubdetectorGeometry(DetId::Hcal, HcalEndcap))) {
      throw cms::Exception("LogicError")
          << "Configuration asked for non-V9 geometry, but the CaloGeometry does not look like one";
    }
    geometry->initialize(&calo_geometry);
  }
  return geometry;
}

//define this as a plug-in
DEFINE_FWK_EVENTSETUP_MODULE(HGCalTriggerGeometryESProducer);