CondTools/Geometry/plugins/XMLGeometryBuilder.cc - master - githubmirror/cmssw

CondTools/Geometry/plugins/XMLGeometryBuilder.cc

55 строк2 KB

francescobrivio

Migrate Geometry to new `PoolDBOutputService` methods

22 ноя 2021, 01:51

22 ноя 2021, 01:510c74cae

#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "CondFormats/Common/interface/FileBlob.h"
#include "Geometry/Records/interface/GeometryFileRcd.h"

#include <string>
#include <vector>
#include <fstream>

class XMLGeometryBuilder : public edm::one::EDAnalyzer<edm::one::WatchRuns> {
public:
  XMLGeometryBuilder(const edm::ParameterSet&);

  void beginJob() override;
  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:
  std::string m_fname;
  bool m_zip;
  std::string m_record;
};

XMLGeometryBuilder::XMLGeometryBuilder(const edm::ParameterSet& iConfig) {
  m_fname = iConfig.getUntrackedParameter<std::string>("XMLFileName", "test.xml");
  m_zip = iConfig.getUntrackedParameter<bool>("ZIP", true);
  m_record = iConfig.getUntrackedParameter<std::string>("record", "GeometryFileRcd");
}

void XMLGeometryBuilder::beginJob() {
  edm::LogInfo("XMLGeometryBuilder") << "XMLGeometryBuilder::beginJob";
  edm::Service<cond::service::PoolDBOutputService> mydbservice;
  if (!mydbservice.isAvailable()) {
    edm::LogError("XMLGeometryBuilder") << "PoolDBOutputService unavailable";
    return;
  }

  FileBlob pgf(m_fname, m_zip);

  if (mydbservice->isNewTagRequest(m_record)) {
    mydbservice->createOneIOV(pgf, mydbservice->beginOfTime(), m_record);
  } else {
    edm::LogError("XMLGeometryBuilder") << "GeometryFileRcd Tag already exist";
  }
}

DEFINE_FWK_MODULE(XMLGeometryBuilder);