Geant4GDMLWriteAction.cpp

Go to the documentation of this file.

//==========================================================================
//  AIDA Detector description implementation 
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================
#ifndef DD4HEP_DDG4_GEANT4GDMLWRITEACTION_H
#define DD4HEP_DDG4_GEANT4GDMLWRITEACTION_H
 
// Framework include files
#include <DDG4/Geant4Action.h>
 
namespace dd4hep {
 
  namespace sim {
 
 
    class Geant4GDMLWriteAction : public Geant4Action {
    public:
      std::string m_output;
      int         m_overWrite;
      int         m_exportRegions;
      int         m_exportEnergyCuts;
      int         m_exportSensitiveDetectors;
 
    public:
      Geant4GDMLWriteAction(Geant4Context* context, const std::string& nam);
      virtual ~Geant4GDMLWriteAction();
      virtual void installCommandMessenger()  override;
      virtual void writeGDML();
    };
 
  }    // End namespace sim
}      // End namespace dd4hep
 
#endif /* DD4HEP_DDG4_GEANT4GDMLWRITEACTION_H */
 
//====================================================================
//  AIDA Detector description implementation 
//--------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//====================================================================
 
// Framework include files
#include <DD4hep/InstanceCount.h>
#include <DD4hep/Printout.h>
#include <DD4hep/Primitives.h>
#include <DDG4/Geant4DataDump.h>
#include <DDG4/Geant4UIMessenger.h>
 
// Geant 4 includes
#ifndef GEANT4_NO_GDML
#include <G4GDMLParser.hh>
#endif
#include <G4Version.hh>
 
// C/C++ include files
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <memory>
 
using namespace dd4hep::sim;
 
Geant4GDMLWriteAction::Geant4GDMLWriteAction(Geant4Context* ctxt, const std::string& nam)
  : Geant4Action(ctxt, nam)
{
  m_needsControl = true;
  declareProperty("Output",                   m_output = "");
  declareProperty("OverWrite",                m_overWrite = 1);
  declareProperty("exportRegions",            m_exportRegions = 1);
  declareProperty("exportEnergyCuts",         m_exportEnergyCuts = 1);
  declareProperty("exportSensitiveDetectors", m_exportSensitiveDetectors = 1);
  InstanceCount::increment(this);
}
 
Geant4GDMLWriteAction::~Geant4GDMLWriteAction() {
  InstanceCount::decrement(this);
}
 
void Geant4GDMLWriteAction::installCommandMessenger()   {
  Callback cb = Callback(this).make(&Geant4GDMLWriteAction::writeGDML);
  m_control->addCall("write", "Write geometry to GDML file",cb);
}
 
void Geant4GDMLWriteAction::writeGDML()   {
  std::string fname = m_output;
  struct stat buff;
  if ( fname.empty() )   {
    error("+++ No GDML file name given. Please set the output file (property Output)");
    return;
  }
  if ( (0 == ::stat(fname.c_str(), &buff)) && !m_overWrite )  {
    error("+++ GDML file %s elready exists. Please set another output file (property Output)",
          fname.c_str());
    return;
  }
  else if ( 0 == ::stat(fname.c_str(), &buff) )  {
    warning("+++ GDML file %s already exists. Overwriting existing file.", fname.c_str());
    ::unlink(fname.c_str());
  }
#ifdef GEANT4_NO_GDML
  warning("+++ writeGDML: GDML not found in the present Geant4 build! Output: %s not written", fname.c_str());
#else
  std::unique_ptr<G4GDMLParser> parser(new G4GDMLParser());
  parser->SetRegionExport(m_exportRegions != 0);
  parser->SetEnergyCutsExport(m_exportEnergyCuts != 0);
#if G4VERSION_NUMBER>=1030
  parser->SetSDExport(m_exportSensitiveDetectors != 0);
#endif
  info("+++ Writing GDML file: %s", fname.c_str());
  parser->Write(fname, context()->world());
#endif
}
 
#include <DDG4/Factories.h>
DECLARE_GEANT4ACTION(Geant4GDMLWriteAction)