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