HepMcWriterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
// HepMcWriterTool.h 
// Header file for class HepMcWriterTool
// Author: S.Binet<binet@cern.ch>
#ifndef MCPARTICLETOOLS_HEPMCWRITERTOOL_H 
#define MCPARTICLETOOLS_HEPMCWRITERTOOL_H 
 
// STL includes
#include <string>
 
// FrameWork includes
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
// McParticleKernel includes
#include "TruthIO/IIOHepMcTool.h"
 
// Forward declaration
#include "AtlasHepMC/GenEvent_fwd.h"
#include "AtlasHepMC/IO_BaseClass.h"
#ifdef HEPMC3
#include "HepMC3/Writer.h"
#include "HepMC3/WriterAsciiHepMC2.h"
#endif 
class HepMcWriterTool : public extends<AthAlgTool, IIOHepMcTool>
{ 

  // Public methods: 
 public: 
 
  // Copy constructor: 

  HepMcWriterTool( const std::string& type, const std::string& name,  const IInterface* parent );

  virtual ~HepMcWriterTool(); 
 
  // Athena algorithm's Hooks
  StatusCode  initialize();
  StatusCode  execute();
  StatusCode  finalize();

  // Non-const methods: 

  StatusCode write( const HepMC::GenEvent* evt );

  // Protected methods: 
 protected: 

  HepMcWriterTool();

  void setupBackend( Gaudi::Details::PropertyBase& ioBackendURL );

  // Protected data: 
 protected: 

  StringProperty m_ioBackendURL;

  StringProperty m_mcEventsName;

#ifdef HEPMC3
  HepMC3::Writer* m_ioBackend;
#else   
  HepMC::IO_BaseClass* m_ioBackend;
#endif
 
}; 
#endif //> MCPARTICLETOOLS_HEPMCWRITERTOOL_H