timeVortexBinnedMap.h

// Copyright 2009-2023 NTESS. Under the terms
// of Contract DE-NA0003525 with NTESS, the U.S.
// Government retains certain rights in this software.
//
// Copyright (c) 2009-2023, NTESS
// All rights reserved.
//
// This file is part of the SST software package. For license
// information, see the LICENSE file in the top level directory of the
// distribution.
 
#ifndef SST_CORE_IMPL_TIMEVORTEX_TIMEVORTEXBINNEDMAP_H
#define SST_CORE_IMPL_TIMEVORTEX_TIMEVORTEXBINNEDMAP_H
 
#include <atomic>
#include <queue>
#include <sst/core/timeVortex.h>
#include <vector>
 
namespace SST {
 
class Output;
 
namespace IMPL {
 
 
template <typename T>
class Pool
{
 
    std::vector<T*> pool;
 
public:
    ~Pool()
    {
        for ( auto x : pool ) {
            delete x;
        }
    }
 
    T* remove()
    {
        if ( pool.empty() ) { pool.push_back(new T()); }
        auto ret = pool.back();
        pool.pop_back();
        return ret;
    }
 
    void insert(T* item) { pool.push_back(item); }
};
 
/**
 * Primary Event Queue
 */
template <bool TS>
class TimeVortexBinnedMapBase : public TimeVortex
{
 
private:
    //  Class to hold a vector and a delivery time
    class TimeUnit
    {
    private:
        SimTime_t              sort_time;
        std::vector<Activity*> activities;
        bool                   sorted;
 
        CACHE_ALIGNED(SST::Core::ThreadSafe::Spinlock, tu_lock);
 
    public:
        // Create with initial event
        TimeUnit() : sorted(false) {}
 
        ~TimeUnit()
        {
            for ( auto x : activities ) {
                delete x;
            }
        }
 
        inline SimTime_t getSortTime() { return sort_time; }
        inline void      setSortTime(SimTime_t time) { sort_time = time; }
 
 
        // Inserts can happen by multiple threads
        void insert(Activity* act)
        {
            if ( TS ) tu_lock.lock();
            activities.push_back(act);
            sorted = false;
            if ( TS ) tu_lock.unlock();
        }
 
        // pop only happens by one thread
        Activity* pop()
        {
            if ( 0 == activities.size() ) return nullptr;
            if ( !sorted ) sort();
            auto ret = activities.back();
            activities.pop_back();
            return ret;
        }
 
        // front only happens by one thread
        Activity* front()
        {
            if ( 0 == activities.size() ) return nullptr;
            if ( !sorted ) sort();
            return activities.back();
        }
 
        void sort();
 
        inline bool operator<(const TimeUnit& rhs) { return this->sort_time < rhs.sort_time; }
 
        /** To use with STL priority queues, that order in reverse. */
        class pq_less
        {
        public:
            /** Compare pointers */
            inline bool operator()(const TimeUnit* lhs, const TimeUnit* rhs) { return lhs->sort_time > rhs->sort_time; }
 
            /** Compare references */
            inline bool operator()(const TimeUnit& lhs, const TimeUnit& rhs) { return lhs.sort_time > rhs.sort_time; }
        };
    };
 
 
public:
    TimeVortexBinnedMapBase(Params& params);
    ~TimeVortexBinnedMapBase();
 
    bool      empty() override;
    int       size() override;
    void      insert(Activity* activity) override;
    Activity* pop() override;
    Activity* front() override;
 
 
    /** Print the state of the TimeVortex */
    void print(Output& out) const override;
 
    uint64_t getCurrentDepth() const override { return current_depth; }
    uint64_t getMaxDepth() const override { return max_depth; }
 
private:
    // Should only ever be accessed by the "active" thread.  Not safe
    // for concurrent access.
    TimeUnit* current_time_unit;
 
    typedef std::map<SimTime_t, TimeUnit*> mapType_t;
 
    // Accessed by multiple threads, must be locked when accessing
    mapType_t                                                            map;
    typename std::conditional<TS, std::atomic<uint64_t>, uint64_t>::type insertOrder;
 
    typename std::conditional<TS, std::atomic<uint64_t>, uint64_t>::type current_depth;
 
    // Should only ever be accessed by the "active" thread, or in a
    // mutex.  There are no internal mutexes.
    Pool<TimeUnit> pool;
 
    CACHE_ALIGNED(SST::Core::ThreadSafe::Spinlock, slock);
};
 
} // namespace IMPL
} // namespace SST
 
#endif // SST_CORE_IMPL_TIMEVORTEX_TIMEVORTEXBINNEDMAP_H