peq.h

/*******************************************************************************
 * Copyright 2018 MINRES Technologies GmbH
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *******************************************************************************/
 
#ifndef _SCC_PEQ_H_
#define _SCC_PEQ_H_
 
#include <boost/optional.hpp>
#include <deque>
#include <map>
#include <systemc>
#include <type_traits>
#include <vector>

namespace scc {
template <class TYPE> struct peq : public sc_core::sc_object {
 
    // static_assert(std::is_copy_constructible<TYPE>::value, "TYPE needs to be copy-constructible");
 
    using pair_type = std::pair<const sc_core::sc_time, TYPE>;
    using map_type = std::map<const sc_core::sc_time, std::deque<TYPE>*>;
    peq()
    : sc_core::sc_object(sc_core::sc_gen_unique_name("peq")) {}

    explicit peq(const char* name)
    : sc_core::sc_object(name) {}

    ~peq();
    void notify(TYPE const& entry, const sc_core::sc_time& t) {
        insert_entry(entry, t + sc_core::sc_time_stamp());
        m_event.notify(m_scheduled_events.begin()->first - sc_core::sc_time_stamp());
    }

    void notify(TYPE&& entry, const sc_core::sc_time& t) {
        insert_entry(std::move(entry), t + sc_core::sc_time_stamp());
        m_event.notify(m_scheduled_events.begin()->first - sc_core::sc_time_stamp());
    }
    void notify(TYPE&& entry) {
        insert_entry(std::move(entry), sc_core::sc_time_stamp());
        m_event.notify(); // immediate notification
    }
    void notify(TYPE const& entry) {
        insert_entry(entry, sc_core::sc_time_stamp());
        m_event.notify(); // immediate notification
    }

    boost::optional<TYPE> get_next() {
        if(m_scheduled_events.empty())
            return boost::none;
        sc_core::sc_time now = sc_core::sc_time_stamp();
        if(!m_scheduled_events.size() || m_scheduled_events.begin()->first > now) {
            if(m_scheduled_events.size())
                m_event.notify(m_scheduled_events.begin()->first - now);
            return boost::none;
        } else
            return get_entry();
    }

    TYPE get() {
        while(!has_next()) {
            sc_core::wait(event());
        }
        return get_entry();
    }

    sc_core::sc_event& event() { return m_event; }
    void cancel_all() {
        m_scheduled_events.clear();
        m_event.cancel();
    }

    bool has_next() { return !(m_scheduled_events.empty() || m_scheduled_events.begin()->first > sc_core::sc_time_stamp()); }
 
    void clear() {
        while(!m_scheduled_events.empty()) {
            auto queue = m_scheduled_events.begin()->second;
            queue->clear();
            free_pool.push_back(queue);
            m_scheduled_events.erase(m_scheduled_events.begin());
        }
    }
 
    size_t size() const { return m_scheduled_events.size(); }
 
private:
    map_type m_scheduled_events;
    std::deque<std::deque<TYPE>*> free_pool;
    sc_core::sc_event m_event;
 
    void insert_entry(const TYPE& entry, sc_core::sc_time abs_time) {
        auto it = m_scheduled_events.find(abs_time);
        if(it == m_scheduled_events.end()) {
            if(free_pool.size()) {
                auto r = m_scheduled_events.insert(std::make_pair(abs_time, free_pool.front()));
                free_pool.pop_front();
                r.first->second->push_back(entry);
            } else {
                auto r = m_scheduled_events.insert(std::make_pair(abs_time, new std::deque<TYPE>()));
                r.first->second->push_back(entry);
            }
        } else
            it->second->push_back(entry);
    }
 
    void insert_entry(TYPE&& entry, sc_core::sc_time abs_time) {
        auto it = m_scheduled_events.find(abs_time);
        if(it == m_scheduled_events.end()) {
            if(free_pool.size()) {
                auto r = m_scheduled_events.insert(std::make_pair(abs_time, free_pool.front()));
                free_pool.pop_front();
                r.first->second->push_back(std::move(entry));
            } else {
                auto r = m_scheduled_events.insert(std::make_pair(abs_time, new std::deque<TYPE>()));
                r.first->second->push_back(std::move(entry));
            }
        } else
            it->second->push_back(std::move(entry));
    }
 
    TYPE get_entry() {
        auto entry = m_scheduled_events.begin()->second;
        auto ret = std::move(entry->front());
        entry->pop_front();
        if(!entry->size()) {
            free_pool.push_back(entry);
            m_scheduled_events.erase(m_scheduled_events.begin());
        }
        if(m_scheduled_events.size())
            m_event.notify(m_scheduled_events.begin()->first - sc_core::sc_time_stamp());
        return ret;
    }
};
 
template <class TYPE> inline peq<TYPE>::~peq() {
    while(m_scheduled_events.size()) {
        free_pool.push_back(m_scheduled_events.begin()->second);
        m_scheduled_events.erase(m_scheduled_events.begin());
    }
    for(auto* p : free_pool)
        delete p;
}
 
} // namespace scc // end of scc-sysc
#endif    /* _SCC_PEQ_H_ */