TransformMap.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef GEOMODELUTILITIES_TRANSFORMMAP_H
#define GEOMODELUTILITIES_TRANSFORMMAP_H
 
#include <CxxUtils/ConcurrentToValMap.h>
#include <CxxUtils/SimpleUpdater.h>
 
#include <memory>
#include <unordered_map>
#include <shared_mutex>

template <typename T, typename X> class TransformMap {
public:
    TransformMap() =default;
    /*** @brief: Copy constructor from an existing map*/
    TransformMap(const TransformMap& other) = default;
    TransformMap& operator=(const TransformMap& other) = default;
 
    ~TransformMap() = default;
    bool setTransform(const T *obj, const X &xf) const;
    bool setTransform(const T* obj, const X& xf);
    bool setTransform(const T *obj, std::shared_ptr<const X> xf) const;
    bool setTransform(const T* obj, std::shared_ptr<const X> xf);    
    
    const X *getTransform(const T *obj) const;

    bool append(const TransformMap &other);
    void clear();
    void lock();
    std::vector<const T*> getStoredKeys() const;
 
private:
    using ConCurrentMap_t = CxxUtils::ConcurrentToValMap<const T*, std::shared_ptr<const X>, CxxUtils::SimpleUpdater>;
    using ConCurrentMap_ptr = std::unique_ptr<ConCurrentMap_t>;
    using CanonicalMap_t = std::unordered_map<const T*, std::shared_ptr<const X>>;
    using CanonicalMap_ptr = std::unique_ptr<CanonicalMap_t>;
    
 
    mutable CanonicalMap_t m_container ATLAS_THREAD_SAFE{};
    mutable bool m_locked ATLAS_THREAD_SAFE{false};
    mutable std::shared_mutex m_mutex ATLAS_THREAD_SAFE{};
    mutable ConCurrentMap_ptr m_mutableCont ATLAS_THREAD_SAFE{
                                std::make_unique<ConCurrentMap_t>(typename ConCurrentMap_t::Updater_t())};
 
 
};
#include <GeoModelUtilities/TransformMap.icc>
#endif