DumpSimGeometry.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    DumpSimGeometry
// Class:      DumpSimGeometry
//
//
// Original Author:  Chris D Jones
//         Created:  Wed Sep 26 08:27:23 EDT 2007
//
//
 
// system include files
#include <memory>
#include <iostream>
 
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
 
#include "Fireworks/Geometry/interface/DisplayGeomRecord.h"
 
#include "TGeoManager.h"
#include "TGeoMatrix.h"
 
#include "TFile.h"
#include "TError.h"
#include "TSystem.h"
 
//
// class declaration
//
 
class DumpSimGeometry : public edm::EDAnalyzer {
public:
  explicit DumpSimGeometry(const edm::ParameterSet&);
  ~DumpSimGeometry() override;
 
private:
  void analyze(const edm::Event&, const edm::EventSetup&) override;
 
  std::string m_tag;
  std::string m_outputFileName;
};
 
//
// constructors and destructor
//
DumpSimGeometry::DumpSimGeometry(const edm::ParameterSet& ps) {
  m_tag = ps.getUntrackedParameter<std::string>("tag", "unknown");
  m_outputFileName = ps.getUntrackedParameter<std::string>("outputFileName", "cmsSimGeom.root");
}
 
DumpSimGeometry::~DumpSimGeometry() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
}
 
// ------------ method called to for each event  ------------
void DumpSimGeometry::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
  std::cout << "In the DumpSimGeometry::analyze method..." << std::endl;
  using namespace edm;
 
  ESTransientHandle<TGeoManager> geoh;
  iSetup.get<DisplayGeomRecord>().get(geoh);
  const TGeoManager* geom = geoh.product();  // const_cast<TGeoManager*>(geoh.product());
 
  int level = 1 + geom->GetTopVolume()->CountNodes(100, 3);
 
  std::cout << "In the DumpSimGeometry::analyze method...obtained main geometry, level=" << level << std::endl;
 
  // TFile f(TString::Format("cmsSimGeom-%d.root", level), "RECREATE");
  TFile f(m_outputFileName.c_str(), "RECREATE");
  f.WriteTObject(geom);
  f.WriteTObject(new TNamed("CMSSW_VERSION", gSystem->Getenv("CMSSW_VERSION")));
  f.WriteTObject(new TNamed("tag", m_tag.c_str()));
  f.Close();
}
 
//define this as a plug-in
DEFINE_FWK_MODULE(DumpSimGeometry);