EscapedEnergyRegistry.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef CALOG4SIM_ESCAPEDENERGYREGISTRY_H
#define CALOG4SIM_ESCAPEDENERGYREGISTRY_H
 
#include "CaloG4Sim/VEscapedEnergyProcessing.h"
#include "CxxUtils/checker_macros.h"
#include "globals.hh"
 
#include <map>
#include <memory>
 
#include <thread>
#include "G4Types.hh"
#ifdef G4MULTITHREADED
#  include "tbb/concurrent_unordered_map.h"
#endif
namespace CaloG4
{
 
  class EscapedEnergyRegistry
  {
  public:
 
    static EscapedEnergyRegistry* GetInstance();
 
    ~EscapedEnergyRegistry() = default;
 
    // Add a VEscapedEnergyProcessing object to the registry.  The
    // name include "Adopt" because we assume responsibility for
    // deleting the pointer.
    void AddAndAdoptProcessing( const G4String& name, std::unique_ptr<VEscapedEnergyProcessing> process );
 
    // Given a volume name,
    VEscapedEnergyProcessing* GetProcessing( const G4String& volumeName ) const;
 
  protected:
 
    // Constructor protected according to singleton pattern.
    EscapedEnergyRegistry() = default;
 
  private:
 
    typedef std::map< const G4String, std::unique_ptr<VEscapedEnergyProcessing> > m_processingMap_t;
    typedef m_processingMap_t::iterator                           m_processingMap_ptr_t;
    typedef m_processingMap_t::const_iterator                     m_processingMap_const_ptr_t;
    m_processingMap_t                                             m_processingMap;
#ifdef G4MULTITHREADED
     // Thread-to-EscapeEnergyRegistry concurrent map type
    using EERThreadMap_t = tbb::concurrent_unordered_map< std::thread::id, EscapedEnergyRegistry*, std::hash<std::thread::id> >;
    // Concurrent map of EERs, one for each thread
    static EERThreadMap_t m_EERThreadMap ATLAS_THREAD_SAFE;
    // @brief Search inside m_EERThreadMap the element with the current threadID 
    // and return it or return a null pointer if the element is not found
    static EscapedEnergyRegistry* getEER();
    // @brief Insert the current EER in m_EERThreadMap and 
    // associate it with the current threadID
    static EscapedEnergyRegistry* setEER();
#endif
  };
 
} // namespace CaloG4
 
#endif // CaloG4_EscapedEnergyRegistry_H