DDTestDumpFile.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DetectorDescription/DDCMS/interface/DetectorDescriptionRcd.h"
#include "DetectorDescription/DDCMS/interface/DDDetector.h"
#include "DD4hep/Detector.h"
#include "DD4hep/DD4hepRootPersistency.h"

#include "TGeoManager.h"
#include "TFile.h"
#include "TSystem.h"

#include <iostream>
#include <string>

using namespace std;
using namespace cms;
using namespace dd4hep;

class DDTestDumpFile : public edm::one::EDAnalyzer<> {
public:
  explicit DDTestDumpFile( const edm::ParameterSet& );

  void beginJob() override {}
  void analyze( edm::Event const& iEvent, edm::EventSetup const& ) override;
  void endJob() override {}

private:
  std::string m_tag;
  std::string m_outputFileName;
};

DDTestDumpFile::DDTestDumpFile(const edm::ParameterSet& iConfig)
{
  m_tag =  iConfig.getUntrackedParameter<std::string>( "tag", "unknown" );
  m_outputFileName = iConfig.getUntrackedParameter<std::string>( "outputFileName", "cmsDD4HepGeom.root" );
}

void
DDTestDumpFile::analyze(const edm::Event&, const edm::EventSetup& iEventSetup)
{
  std::cout << "DDTestDumpFile::analyze:\n";
  edm::ESTransientHandle<DDDetector> description;
  iEventSetup.get<DetectorDescriptionRcd>().get(description);

  TGeoManager& geom = description->description().manager();
  
  int level = 1 + geom.GetTopVolume()->CountNodes( 100, 3 );
  
  std::cout << "In the DDTestDumpFile::analyze method...obtained main geometry, level="
        << level << std::endl;

  TFile file( m_outputFileName.c_str(), "RECREATE" );
  file.WriteTObject( &geom );
  file.WriteTObject( new TNamed( "CMSSW_VERSION", gSystem->Getenv( "CMSSW_VERSION" )));
  file.WriteTObject( new TNamed( "tag", m_tag.c_str()));
  file.Close();
}

DEFINE_FWK_MODULE( DDTestDumpFile );