CaloRingerReaderJetUtils.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#ifndef CALORINGERTOOLS_CALORINGERREADERJetUTILS_H
#define CALORINGERTOOLS_CALORINGERREADERJetUTILS_H
 
// STL includes:
#include <string>
#include <memory>
#include <vector>
 
// Athena framework include:
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/IDataHandleHolder.h"
#include "StoreGate/WriteDecorHandle.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "AthenaBaseComps/AthCheckMacros.h"
#include "AthenaBaseComps/AthMessaging.h"
#include "AthContainers/DataVector.h"
 
// xAOD include:
#include "xAODCaloRings/tools/getCaloRingsDecorator.h"
 
// Interface includes:
#include "CaloRingerTools/ICaloRingsBuilder.h"
 
// Local includes:
#include "RingerKnownParticles.h"
 
// Forward declarations:
class CaloCellContainer;
class IMessageSvc;
 
namespace Ringer {

class BuildCaloRingsJetFctorBase : public AthMessaging {
 
  protected:
    BuildCaloRingsJetFctorBase(
        ToolHandle<ICaloRingsBuilder> &builder,
        IMessageSvc* msgSvc,
        const std::string& name):
          AthMessaging(msgSvc, name),
          m_builder(builder),
          m_part_counter(0),
          m_part_size(0),
          m_crContH(nullptr),
          m_rsContH(nullptr)
      {;}
 

    ToolHandle<ICaloRingsBuilder> &m_builder;
    size_t m_part_counter;
    size_t m_part_size;


    StatusCode prepareJetToLoopFor( std::size_t nParticles );

    void incrementCounter() { ++m_part_counter; }

    void checkJetRelease();

    virtual ~BuildCaloRingsJetFctorBase(){;}
 
  private:
    SG::WriteHandle<xAOD::CaloRingsContainer>* m_crContH;
    SG::WriteHandle<xAOD::RingSetContainer>* m_rsContH;
 
};

template< typename container_t >
class BuildCaloRingsJetFctor : public BuildCaloRingsJetFctorBase
{
  public:
    typedef typename container_t::base_value_type particle_t;
    typedef typename SG::WriteDecorHandle< container_t, xAOD::CaloRingsLinks >
      decor_t;
 
  private:
    typedef BuildCaloRingsJetFctorBase base_t;
    // Import base template properties:
    using base_t::m_builder;
    using base_t::m_part_counter;
    using base_t::m_part_size;
 
    using base_t::checkJetRelease;

    static constexpr const char *m_particle_name = Ringer::GetParticleProp<particle_t>::name;

    void displayLoopingMessage() const;

    SG::WriteDecorHandleKey< container_t > m_decorKey;

    decor_t* m_decor{nullptr};
 
  public:
    BuildCaloRingsJetFctor(
        const std::string &decoContName,
        ToolHandle<ICaloRingsBuilder> &builder,
        IMessageSvc* msgSvc,
        IDataHandleHolder* owningAlg)
    : BuildCaloRingsJetFctorBase(
        builder,
        msgSvc,
        owningAlg->name()),
      m_decorKey( decoContName + "." + xAOD::caloRingsLinksDecorKey() )
         { m_decorKey.setOwner(owningAlg);}


    void operator() (const particle_t *part);

    StatusCode initialize();

    StatusCode prepareJetToLoopFor( std::size_t nParticles );

    void checkJetRelease();
};
 
} // namespace Ringer
 
#endif // CALORINGERTOOLS_CALORINGERREADERUTILS_H
 
#ifndef INCLUDE_HEADER_ONLY // Use to avoid circular includes:
#include "CaloRingerReaderJetUtils.icc"
#endif // INCLUDE_HEADER_ONLY