HepMcTupleWriterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
// HepMcTupleWriterTool.h 
// Header file for class HepMcTupleWriterTool
// Author: S.Binet<binet@cern.ch>
#ifndef MCPARTICLETOOLS_HEPMCTUPLEWRITERTOOL_H 
#define MCPARTICLETOOLS_HEPMCTUPLEWRITERTOOL_H 
 
// STL includes
#include <string>
 
 
#include <array>
 
// FrameWork includes
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
// McParticleKernel includes
#include "TruthIO/IIOHepMcTool.h"
 
// Forward declaration
class ITHistSvc;
class TTree;
#include "AtlasHepMC/GenEvent_fwd.h"
 
class HepMcTupleWriterTool : public extends<AthAlgTool, IIOHepMcTool>
{ 

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

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

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

  // Non-const methods: 

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

  // Protected methods: 
 protected: 

  HepMcTupleWriterTool();

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

  void bookTuple();

  // Protected data: 
 protected: 

  static const int s_nMax = 2000;

  struct HepMcParticles {
    int m_nParticles;
    std::array<double, s_nMax> m_px;
    std::array<double, s_nMax> m_py;
    std::array<double, s_nMax> m_pz;
    std::array<double, s_nMax> m_m;
    std::array<double, s_nMax> m_ene;
    std::array<int,    s_nMax> m_pdgId;
    std::array<int,    s_nMax> m_status;
    std::array<int,    s_nMax> m_barcode;
  };

  HepMcParticles m_particles{};

  ServiceHandle<ITHistSvc> m_tupleSvc;

  StringProperty m_mcEventsName;

  StringProperty m_outputFileName;

  StringProperty m_outputStreamName;

  TTree* m_tuple;
}; 
 
#endif //> MCPARTICLETOOLS_HEPMCTUPLEWRITERTOOL_H