HepMcWriterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 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 : virtual public IIOHepMcTool, public AthAlgTool
{ 
 
  // 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