#ifndef _TOP_MODULE_H_
#define _TOP_MODULE_H_

#include <atomic>
#include <chrono>
#include <scc/async_queue.h>
#include <scc/async_thread.h>
#include <scc/report.h>
#include <sysc/kernel/sc_initializer_function.h>
#include <sysc/kernel/sc_simcontext.h>
#include <thread>
#include <tlm/scc/quantum_keeper.h>
#include <tlm>
#include <tlm_core/tlm_2/tlm_generic_payload/tlm_gp.h>
#include <tlm_utils/simple_initiator_socket.h>
#include <tlm_utils/simple_target_socket.h>
#include <util/logging.h>

struct initiator : ::sc_core ::sc_module {
    tlm_utils::simple_initiator_socket<initiator, scc::LT> isckt{"isckt"};

    initiator(sc_core::sc_module_name nm)
    : sc_core::sc_module(nm) {
        SC_THREAD(run);
    }

    ~initiator() {}

private:
    void run() {
        wait(sc_core::SC_ZERO_TIME); // guard elaboration phase
        quantum_keeper.reset();
        core_executor.start([this]() { return thread_exec(); });
        wait(core_executor.thread_finish_event());
        sc_core::sc_stop();
    }

    sc_core::sc_time thread_exec() {
        SCCDEBUG(SCMOD) << "starting thread_exec";
        for(auto i = 0u; i < 16; ++i) {
            std::this_thread::sleep_for(std::chrono::milliseconds(1));
            if(i && (i % 3) == 0) {
                tlm::tlm_generic_payload gp;
                sc_core::sc_time t;
                SCCDEBUG(SCMOD) << "initiating b_transport at local time " << quantum_keeper.get_local_absolute_time();
                quantum_keeper.execute_on_sysc([this, &gp, &t]() {
                    SCCDEBUG(SCMOD) << "executing b_transport";
                    this->isckt->b_transport(gp, t);
                });
                if(t.value()) {
                    SCCDEBUG(SCMOD) << "incrementing local time by b_transport delay of " << t;
                    quantum_keeper.inc(t);
                }
            } else {
                quantum_keeper.check_and_sync(1_us);
            }
            SCCDEBUG(SCMOD) << "local time now " << quantum_keeper.get_local_absolute_time();
        }
        SCCDEBUG(SCMOD) << "finished thread_exec at local time " << quantum_keeper.get_local_absolute_time();
        return quantum_keeper.get_local_absolute_time();
    }
    tlm::scc::quantumkeeper_mt quantum_keeper;
    scc::async_thread core_executor;
};

// top_module
struct top_module : ::sc_core ::sc_module {
    initiator core{"core"};

    tlm_utils::simple_target_socket<top_module, scc::LT> tsckt{"tsckt"};

    top_module(sc_core::sc_module_name nm)
    : sc_core::sc_module(nm) {
        core.isckt(tsckt);
        tsckt.register_b_transport(this, &top_module::b_transport);
    }

    ~top_module() {}

    void b_transport(tlm::tlm_generic_payload& gp, sc_core::sc_time& t) {
        SCCDEBUG(SCMOD) << "Received b_transport call at local time " << t;
        t += 5_us;
        gp.set_response_status(tlm::TLM_OK_RESPONSE);
    }
};
#endif // _TOP_MODULE_H_