McAodTupleWriterTool.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// McAodTupleWriterTool.h 
// Header file for class McAodTupleWriterTool
// Author: S.Binet<binet@cern.ch>
#ifndef MCPARTICLETOOLS_MCAODTUPLEWRITERTOOL_H 
#define MCPARTICLETOOLS_MCAODTUPLEWRITERTOOL_H 
 
// STL includes
#include <string>
 
#include <array>
 
// FrameWork includes
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
// McParticleKernel includes
#include "McParticleKernel/IIOMcAodTool.h"
 
// Forward declaration
class ITHistSvc;
class TTree;
class TruthParticleContainer;
 
class McAodTupleWriterTool : virtual public IIOMcAodTool,
                         public AthAlgTool
{ 

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

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

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

  // Const methods: 

  // Non-const methods: 

  StatusCode write( const TruthParticleContainer* mcAod );

  // Protected methods: 
 protected: 

  McAodTupleWriterTool();

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

  void bookTuple();

  // Protected data: 
 protected: 

  static const int s_nMax = 2000;

  struct McAodParticles {
    unsigned 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;
    // et isolation stuff
    std::array<double, s_nMax> m_etcone10;
    std::array<double, s_nMax> m_etcone20;
    std::array<double, s_nMax> m_etcone30;
    std::array<double, s_nMax> m_etcone40;
    std::array<double, s_nMax> m_etcone45;
    std::array<double, s_nMax> m_etcone50;
    std::array<double, s_nMax> m_etcone60;
    std::array<double, s_nMax> m_etcone70;
  };

  McAodParticles m_particles{};

  ServiceHandle<ITHistSvc> m_tupleSvc;

  StringProperty m_truthParticlesName;

  StringProperty m_outputFileName;

  StringProperty m_outputStreamName;

  TTree* m_tuple;
}; 


 
#endif //> MCPARTICLETOOLS_MCAODTUPLEWRITERTOOL_H