DumpFWRecoGeometry.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "Fireworks/Geometry/interface/FWRecoGeometry.h"
#include "Fireworks/Geometry/interface/FWRecoGeometryRecord.h"

#include "TFile.h"
#include "TTree.h"
#include "TError.h"
#include "TSystem.h"

class DumpFWRecoGeometry : public edm::EDAnalyzer
{
public:
  explicit DumpFWRecoGeometry( const edm::ParameterSet& config );
  ~DumpFWRecoGeometry( void ) override {}

private:
  void analyze( const edm::Event& event, const edm::EventSetup& eventSetup ) override;
  void beginJob( void ) override;
  void endJob( void ) override;

  int m_level;
  std::string m_tag;
  std::string m_outputFileName;
};

DumpFWRecoGeometry::DumpFWRecoGeometry( const edm::ParameterSet& config )
  : m_level( config.getUntrackedParameter<int>( "level", 1 )),
    m_tag( config.getUntrackedParameter<std::string>( "tagInfo", "unknown" )),
    
    m_outputFileName( config.getUntrackedParameter<std::string>( "outputFileName", "cmsRecoGeo.root" ))
{}

void
DumpFWRecoGeometry::analyze( const edm::Event& event, const edm::EventSetup& eventSetup )
{
  using namespace edm;

  ESTransientHandle<FWRecoGeometry> geoh;
  eventSetup.get<FWRecoGeometryRecord>().get( geoh );
  TFile file( m_outputFileName.c_str(), "RECREATE" );

  TTree *tree = new TTree("idToGeo", "raw detector id association with geometry ANT");

  UInt_t v_id;
  Float_t v_vertex[24];
  Float_t v_params[9];
  Float_t v_shape[5];
  Float_t v_translation[3];
  Float_t v_matrix[9];

  tree->SetBranchStyle( 0 );
  tree->Branch( "id", &v_id, "id/i" );
  tree->Branch( "points", &v_vertex, "points[24]/F" );
  tree->Branch( "topology", &v_params, "topology[9]/F" );
  tree->Branch( "shape", &v_shape, "shape[5]/F" );
  tree->Branch( "translation", &v_translation, "translation[3]/F" );
  tree->Branch( "matrix", &v_matrix, "matrix[9]/F" );

  for( FWRecoGeom::InfoMapItr it = geoh.product()->idToName.begin(),
                 end = geoh.product()->idToName.end();
       it != end; ++it )
  {
    v_id = it->id;
    for( unsigned int i = 0; i < 24; ++i )
      v_vertex[i] = it->points[i];
    for( unsigned int i = 0; i < 9; ++i )
      v_params[i] = it->topology[i];
    for( unsigned int i = 0; i < 5; ++i )
      v_shape[i] = it->shape[i];
    for( unsigned int i = 0; i < 3; ++i )
      v_translation[i] = it->translation[i];
    for( unsigned int i = 0; i < 9; ++i )
      v_matrix[i] = it->matrix[i];
    tree->Fill();
  }
  file.WriteTObject( tree );


  file.WriteTObject(new TNamed("CMSSW_VERSION", gSystem->Getenv( "CMSSW_VERSION" )));
  file.WriteTObject(new TNamed("tag", m_tag.c_str()));
  file.WriteTObject(&geoh.product()->extraDet, "ExtraDetectors");
  file.WriteTObject(new TNamed("PRODUCER_VERSION", "1")); // version 2 changes pixel parameters


  file.WriteTObject(new TNamed("TrackerTopology",  geoh.product()->trackerTopologyXML));

  file.Close();
}

void 
DumpFWRecoGeometry::beginJob( void )
{}

void 
DumpFWRecoGeometry::endJob( void )
{}

DEFINE_FWK_MODULE( DumpFWRecoGeometry );