runonsysc.h

/*
 * Copyright (c) 2022-2023 Qualcomm Innovation Center, Inc. All Rights Reserved.
 * Author: GreenSocs 2022
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#ifndef RUNONSYSTEMC_H
#define RUNONSYSTEMC_H
 
#include <thread>
#include <mutex>
#include <condition_variable>
#include <queue>
#include <future>
#include <chrono>
 
#include <async_event.h>
#include <uutils.h>
 
namespace gs {
class runonsysc : public sc_core::sc_module
{
protected:
    class AsyncJob
    {
    public:
        using Ptr = std::shared_ptr<AsyncJob>;
 
    private:
        std::packaged_task<void()> m_task;
 
        bool m_cancelled = false;
 
        void run_job() { m_task(); }
 
    public:
        AsyncJob(std::function<void()>&& job): m_task(job) {}
 
        AsyncJob(std::function<void()>& job): m_task(job) {}
 
        AsyncJob() = delete;
        AsyncJob(const AsyncJob&) = delete;
 
        void operator()() { run_job(); }
 
        void cancel()
        {
            m_cancelled = true;
            m_task.reset();
        }
 
        void wait()
        {
            auto future = m_task.get_future();
 
            while (!m_cancelled && future.wait_for(std::chrono::milliseconds(500)) == std::future_status::timeout) {
            }
 
            if (!m_cancelled) {
                future.get();
            }
        }
 
        bool is_cancelled() const { return m_cancelled; }
    };
 
    std::thread::id m_thread_id;
 
    /* Async job queue */
    std::queue<AsyncJob::Ptr> m_async_jobs;
    AsyncJob::Ptr m_running_job;
    std::mutex m_async_jobs_mutex;
 
    async_event m_jobs_handler_event;
    std::atomic<bool> running = true;
 
    // Process inside a thread incase the job calls wait
    void jobs_handler()
    {
        std::unique_lock<std::mutex> lock(m_async_jobs_mutex);
        running = true;
        for (; running;) {
            while (running && !m_async_jobs.empty()) {
                m_running_job = m_async_jobs.front();
                m_async_jobs.pop();
 
                lock.unlock();
                sc_core::sc_unsuspendable(); // a wait in the job will cause systemc time to advance
                (*m_running_job)();
                sc_core::sc_suspendable();
                lock.lock();
 
                m_running_job.reset();
            }
            lock.unlock();
            wait(m_jobs_handler_event);
            lock.lock();
        }
        SC_REPORT_WARNING("RunOnSysc", "Stopped");
        sc_core::sc_stop();
    }
 
    void cancel_pendings_locked()
    {
        while (!m_async_jobs.empty()) {
            m_async_jobs.front()->cancel();
            m_async_jobs.pop();
        }
    }
 
public:
    runonsysc(const sc_core::sc_module_name& n = sc_core::sc_module_name("run-on-sysc"))
        : sc_module(n)
        , m_thread_id(std::this_thread::get_id())
        , m_jobs_handler_event(false) // starve if no more jobs provided
    {
        SC_THREAD(jobs_handler);
    }
 
    void cancel_pendings()
    {
        std::lock_guard<std::mutex> lock(m_async_jobs_mutex);
 
        cancel_pendings_locked();
    }
 
    void cancel_all()
    {
        std::lock_guard<std::mutex> lock(m_async_jobs_mutex);
 
        cancel_pendings_locked();
 
        if (m_running_job) {
            m_running_job->cancel();
            m_running_job.reset();
        }
    }
    void stop()
    {
        running = false;
        m_jobs_handler_event.async_notify();
    }
    void end_of_simulation()
    {
        running = false;
        cancel_all();
    }
 
    void fork_on_systemc(std::function<void()> job_entry) { run_on_sysc(job_entry, false); }
 
    bool run_on_sysc(std::function<void()> job_entry, bool wait = true)
    {
        if (!running) return false;
        if (is_on_sysc()) {
            job_entry();
            return true;
        } else {
            AsyncJob::Ptr job(new AsyncJob(job_entry));
 
            {
                std::lock_guard<std::mutex> lock(m_async_jobs_mutex);
                if (running)
                    m_async_jobs.push(job);
                else
                    return false;
            }
 
            m_jobs_handler_event.async_notify();
 
            if (wait) {
                /* Wait for job completion */
                try {
                    job->wait();
                } catch (std::runtime_error const& e) {
                    /* Report unknown runtime errors, without causing a futher excetion */
                    auto old = sc_core::sc_report_handler::set_actions(sc_core::SC_ERROR,
                                                                       sc_core::SC_LOG | sc_core::SC_DISPLAY);
                    SC_REPORT_ERROR(
                        "RunOnSysc",
                        ("Run on systemc received a runtime error from job: " + std::string(e.what())).c_str());
                    sc_core::sc_report_handler::set_actions(sc_core::SC_ERROR, old);
                    stop();
                    return false;
                } catch (const std::exception& exc) {
                    if (sc_core::sc_report_handler::get_count(sc_core::SC_ERROR) == 0) {
                        /* Report exceptions that were not caused by SC_ERRORS (which have already been reported)*/
                        auto old = sc_core::sc_report_handler::set_actions(sc_core::SC_ERROR,
                                                                           sc_core::SC_LOG | sc_core::SC_DISPLAY);
                        SC_REPORT_ERROR(
                            "RunOnSysc",
                            ("Run on systemc received an exception from job: " + std::string(exc.what())).c_str());
                        sc_core::sc_report_handler::set_actions(sc_core::SC_ERROR, old);
                    }
                    stop();
                    return false;
                } catch (...) {
                    auto old = sc_core::sc_report_handler::set_actions(sc_core::SC_ERROR,
                                                                       sc_core::SC_LOG | sc_core::SC_DISPLAY);
                    SC_REPORT_ERROR("RunOnSysc", "Run on systemc received an unknown exception from job");
                    sc_core::sc_report_handler::set_actions(sc_core::SC_ERROR, old);
                    stop();
                    return false;
                }
 
                return !job->is_cancelled();
            }
 
            return true;
        }
    }
 
    bool is_on_sysc() const { return std::this_thread::get_id() == m_thread_id; }
};
} // namespace gs
 
#endif // RUNONSYSTEMC_H