ME0GeometryESModule.cc

Go to the documentation of this file.

/*
//\class ME0GeometryESModule
 
 Description: ME0 GeometryESModule from DD & DD4hep
              DD4hep part added to the original old file (DD version) made by M. Maggi (INFN Bari)
//
// Author:  Sergio Lo Meo (sergio.lo.meo@cern.ch) following what Ianna Osborne made for DTs (DD4HEP migration)
//          Created:  29 Apr 2020 
*/
 
#include "Geometry/GEMGeometryBuilder/src/ME0GeometryBuilder.h"
#include "Geometry/GEMGeometryBuilder/src/ME0GeometryBuilderFromCondDB.h"
 
#include "Geometry/Records/interface/IdealGeometryRecord.h"
#include "Geometry/MuonNumbering/interface/MuonGeometryConstants.h"
#include "DetectorDescription/Core/interface/DDCompactView.h"
 
#include "Geometry/MuonNumbering/interface/DD4hep_MuonNumbering.h"
#include <DetectorDescription/DDCMS/interface/DDCompactView.h>
 
#include "Geometry/Records/interface/ME0RecoGeometryRcd.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"
#include "Geometry/GEMGeometry/interface/ME0Geometry.h"
#include "CondFormats/GeometryObjects/interface/RecoIdealGeometry.h"
 
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include <memory>
 
using namespace edm;
 
class ME0GeometryESModule : public edm::ESProducer {
public:
  ME0GeometryESModule(const edm::ParameterSet& p);
  ~ME0GeometryESModule() override;
 
  std::unique_ptr<ME0Geometry> produce(const MuonGeometryRecord& record);
 
private:
  edm::ESGetToken<DDCompactView, IdealGeometryRecord> cpvToken_;
  edm::ESGetToken<MuonGeometryConstants, IdealGeometryRecord> mdcToken_;
  edm::ESGetToken<cms::DDCompactView, IdealGeometryRecord> dd4hepcpvToken_;
  edm::ESGetToken<cms::MuonNumbering, MuonNumberingRecord> dd4hepmdcToken_;
  edm::ESGetToken<RecoIdealGeometry, ME0RecoGeometryRcd> rigme0Token_;
  // use the DDD or DD4hep as Geometry source
  bool useDDD_;
  bool useDD4hep_;
};
 
ME0GeometryESModule::ME0GeometryESModule(const edm::ParameterSet& p) {
  useDDD_ = p.getParameter<bool>("useDDD");
  useDD4hep_ = p.getUntrackedParameter<bool>("useDD4hep", false);
  auto cc = setWhatProduced(this);
  if (useDDD_) {
    cc.setConsumes(cpvToken_).setConsumes(mdcToken_);
  } else if (useDD4hep_) {
    cc.setConsumes(dd4hepcpvToken_).setConsumes(dd4hepmdcToken_);
  } else {
    cc.setConsumes(rigme0Token_);
  }
}
 
ME0GeometryESModule::~ME0GeometryESModule() {}
 
std::unique_ptr<ME0Geometry> ME0GeometryESModule::produce(const MuonGeometryRecord& record) {
  LogTrace("ME0GeometryESModule") << "ME0GeometryESModule::produce with useDDD = " << useDDD_;
  if (useDDD_) {
    LogTrace("ME0GeometryESModule") << "ME0GeometryESModule::produce :: ME0GeometryBuilder builder";
    auto cpv = record.getTransientHandle(cpvToken_);
    const auto& mdc = record.get(mdcToken_);
    ME0GeometryBuilder builder;
    return std::unique_ptr<ME0Geometry>(builder.build(cpv.product(), mdc));
  } else if (useDD4hep_) {
    LogTrace("ME0GeometryESModule") << "ME0GeometryESModule::produce :: ME0GeometryBuilder builder DD4hep";
    auto cpv = record.getTransientHandle(dd4hepcpvToken_);
    const auto& mdc = record.get(dd4hepmdcToken_);
    ME0GeometryBuilder builder;
    return std::unique_ptr<ME0Geometry>(builder.build(cpv.product(), mdc));
  } else {
    LogTrace("ME0GeometryESModule") << "ME0GeometryESModule::produce :: ME0GeometryBuilderFromCondDB builder";
    const auto& rigme0 = record.get(rigme0Token_);
    ME0GeometryBuilderFromCondDB builder;
    return std::unique_ptr<ME0Geometry>(builder.build(rigme0));
  }
}
 
DEFINE_FWK_EVENTSETUP_MODULE(ME0GeometryESModule);