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/ESHandle.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_;
};

HGCalTriggerGeometryESProducer::HGCalTriggerGeometryESProducer(const edm::ParameterSet& iConfig)
    : geometry_config_(iConfig.getParameterSet("TriggerGeometry")),
      geometry_name_(geometry_config_.getParameter<std::string>("TriggerGeometryName")) {
  setWhatProduced(this);
}

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_));
  geometry->reset();
  edm::ESHandle<CaloGeometry> calo_geometry;
  iRecord.get(calo_geometry);
  // Initialize trigger geometry for V7/V8 HGCAL geometry
  if (calo_geometry.isValid() && calo_geometry->getSubdetectorGeometry(DetId::Forward, HGCEE) &&
      calo_geometry->getSubdetectorGeometry(DetId::Forward, HGCHEF) &&
      calo_geometry->getSubdetectorGeometry(DetId::Hcal, HcalEndcap)) {
    geometry->initialize(calo_geometry);
  }
  // Initialize trigger geometry for V9 HGCAL geometry
  else {
    edm::ESHandle<HGCalGeometry> ee_geometry;
    edm::ESHandle<HGCalGeometry> hsi_geometry;
    edm::ESHandle<HGCalGeometry> hsc_geometry;
    iRecord.getRecord<IdealGeometryRecord>().get("HGCalEESensitive", ee_geometry);
    iRecord.getRecord<IdealGeometryRecord>().get("HGCalHESiliconSensitive", hsi_geometry);
    iRecord.getRecord<IdealGeometryRecord>().get("HGCalHEScintillatorSensitive", hsc_geometry);
    geometry->initialize(ee_geometry, hsi_geometry, hsc_geometry);
  }
  return geometry;
}

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