XMLIdealMagneticFieldGeometryESProducer.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducer.h"
 
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
 
#include "FWCore/Concurrency/interface/SharedResourceNames.h"
 
#include "DetectorDescription/Core/interface/DDCompactView.h"
#include "DetectorDescription/Core/interface/DDRoot.h"
#include "DetectorDescription/Parser/interface/DDLParser.h"
#include "CondFormats/Common/interface/FileBlob.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
 
#include "DetectorDescription/Core/interface/DDMaterial.h"
#include "DetectorDescription/Core/interface/DDSolid.h"
#include "DetectorDescription/Core/interface/DDSpecifics.h"
#include "DetectorDescription/Core/interface/DDRotationMatrix.h"
 
#include "DetectorDescription/Core/src/Material.h"
#include "DetectorDescription/Core/src/Solid.h"
#include "DetectorDescription/Core/src/LogicalPart.h"
#include "DetectorDescription/Core/src/Specific.h"
 
#include <memory>
 
class XMLIdealMagneticFieldGeometryESProducer : public edm::ESProducer {
public:
  XMLIdealMagneticFieldGeometryESProducer(const edm::ParameterSet&);
  ~XMLIdealMagneticFieldGeometryESProducer() override;
 
  typedef std::unique_ptr<DDCompactView> ReturnType;
 
  ReturnType produce(const IdealMagneticFieldRecord&);
 
private:
  std::string rootDDName_;  // this must be the form namespace:name
  std::string label_;
 
  DDI::Store<DDName, DDI::Material*> matStore_;
  DDI::Store<DDName, DDI::Solid*> solidStore_;
  DDI::Store<DDName, DDI::LogicalPart*> lpStore_;
  DDI::Store<DDName, DDI::Specific*> specStore_;
  DDI::Store<DDName, DDRotationMatrix*> rotStore_;
};
 
XMLIdealMagneticFieldGeometryESProducer::XMLIdealMagneticFieldGeometryESProducer(const edm::ParameterSet& iConfig)
    : rootDDName_(iConfig.getParameter<std::string>("rootDDName")), label_(iConfig.getParameter<std::string>("label")) {
  usesResources({{edm::ESSharedResourceNames::kDDGeometry}});
  setWhatProduced(this);
}
 
XMLIdealMagneticFieldGeometryESProducer::~XMLIdealMagneticFieldGeometryESProducer(void) {}
 
XMLIdealMagneticFieldGeometryESProducer::ReturnType XMLIdealMagneticFieldGeometryESProducer::produce(
    const IdealMagneticFieldRecord& iRecord) {
  using namespace edm::es;
 
  edm::ESTransientHandle<FileBlob> gdd;
  iRecord.getRecord<MFGeometryFileRcd>().get(label_, gdd);
 
  DDName ddName(rootDDName_);
  DDLogicalPart rootNode(ddName);
  DDRootDef::instance().set(rootNode);
  ReturnType returnValue(new DDCompactView(rootNode));
  DDLParser parser(*returnValue);
  parser.getDDLSAX2FileHandler()->setUserNS(true);
  parser.clearFiles();
 
  std::unique_ptr<std::vector<unsigned char> > tb = (*gdd).getUncompressedBlob();
 
  parser.parse(*tb, tb->size());
 
  returnValue->lockdown();
 
  return returnValue;
}
 
DEFINE_FWK_EVENTSETUP_MODULE(XMLIdealMagneticFieldGeometryESProducer);