PCaloGeometryBuilder.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "Geometry/EcalAlgo/interface/EcalBarrelGeometry.h"
#include "Geometry/EcalAlgo/interface/EcalEndcapGeometry.h"
#include "Geometry/EcalAlgo/interface/EcalPreshowerGeometry.h"
#include "Geometry/HGCalGeometry/interface/HGCalGeometry.h"
#include "Geometry/HcalTowerAlgo/interface/HcalGeometry.h"
#include "Geometry/HcalTowerAlgo/interface/CaloTowerGeometry.h"
#include "Geometry/ForwardGeometry/interface/ZdcGeometry.h"
#include "Geometry/ForwardGeometry/interface/CastorGeometry.h"

class PCaloGeometryBuilder : public edm::one::EDAnalyzer<edm::one::WatchRuns> {
public:
  PCaloGeometryBuilder(const edm::ParameterSet& pset)
      : m_ecalE(pset.getUntrackedParameter<bool>("EcalE", true)),
        m_ecalP(pset.getUntrackedParameter<bool>("EcalP", true)),
        m_hgcal(pset.getUntrackedParameter<bool>("HGCal", false)) {}

  void beginRun(edm::Run const& iEvent, edm::EventSetup const&) override;
  void analyze(edm::Event const& iEvent, edm::EventSetup const&) override {}
  void endRun(edm::Run const& iEvent, edm::EventSetup const&) override {}

private:
  bool m_ecalE;
  bool m_ecalP;
  bool m_hgcal;
};

void PCaloGeometryBuilder::beginRun(const edm::Run&, edm::EventSetup const& es) {
  const std::string toDB("_toDB");

  std::cout << "Writing out " << EcalBarrelGeometry::producerTag() << std::endl;
  edm::ESHandle<CaloSubdetectorGeometry> pGeb;
  es.get<EcalBarrelGeometry::AlignedRecord>().get(EcalBarrelGeometry::producerTag() + toDB, pGeb);

  if (m_ecalE) {
    std::cout << "Writing out " << EcalEndcapGeometry::producerTag() << std::endl;
    edm::ESHandle<CaloSubdetectorGeometry> pGee;
    es.get<EcalEndcapGeometry::AlignedRecord>().get(EcalEndcapGeometry::producerTag() + toDB, pGee);
  }

  if (m_ecalP) {
    std::cout << "Writing out " << EcalPreshowerGeometry::producerTag() << std::endl;
    edm::ESHandle<CaloSubdetectorGeometry> pGes;
    es.get<EcalPreshowerGeometry::AlignedRecord>().get(EcalPreshowerGeometry::producerTag() + toDB, pGes);
  }

  if (m_hgcal) {
    std::cout << "Writing out " << HGCalGeometry::producerTag() << std::endl;
    edm::ESHandle<CaloSubdetectorGeometry> pGee;
    es.get<HGCalGeometry::AlignedRecord>().get(HGCalGeometry::producerTag() + toDB, pGee);
  }

  std::cout << "Writing out " << HcalGeometry::producerTag() << std::endl;
  edm::ESHandle<CaloSubdetectorGeometry> pGhcal;
  es.get<HcalGeometry::AlignedRecord>().get(HcalGeometry::producerTag() + toDB, pGhcal);

  std::cout << "Writing out " << CaloTowerGeometry::producerTag() << std::endl;
  edm::ESHandle<CaloSubdetectorGeometry> pGct;
  es.get<CaloTowerGeometry::AlignedRecord>().get(CaloTowerGeometry::producerTag() + toDB, pGct);

  std::cout << "Writing out " << ZdcGeometry::producerTag() << std::endl;
  edm::ESHandle<CaloSubdetectorGeometry> pGzdc;
  es.get<ZdcGeometry::AlignedRecord>().get(ZdcGeometry::producerTag() + toDB, pGzdc);

  std::cout << "Writing out " << CastorGeometry::producerTag() << std::endl;
  edm::ESHandle<CaloSubdetectorGeometry> pGcast;
  es.get<CastorGeometry::AlignedRecord>().get(CastorGeometry::producerTag() + toDB, pGcast);
}

DEFINE_FWK_MODULE(PCaloGeometryBuilder);