HepMcTupleWriterTool.h

Go to the documentation of this file.

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