target_mixin.h

/*******************************************************************************
 * Copyright 2016, 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 _TLM_NW_TARGET_MIXIN_H_
#define _TLM_NW_TARGET_MIXIN_H_
 
#ifndef SC_INCLUDE_DYNAMIC_PROCESSES // needed for sc_spawn
#define SC_INCLUDE_DYNAMIC_PROCESSES
#endif
 
#include "scc/utilities.h"
#include "tlm_network_sockets.h"
#include <functional>
#include <sstream>
#include <tlm>
#include <tlm_utils/peq_with_get.h>

namespace tlm {
namespace nw {
template <typename BASE_TYPE, bool REQUESTOR = true, typename TYPES = typename BASE_TYPE::protocol_types>
class target_mixin : public BASE_TYPE {
    //    friend class fw_process;
    //    friend class bw_process;
 
public:
    using transaction_type = typename TYPES::tlm_payload_type;
    using phase_type = typename TYPES::tlm_phase_type;
    using sync_enum_type = tlm::tlm_sync_enum;
    using fw_interface_type = tlm::nw::tlm_network_fw_transport_if<TYPES>;
    using bw_interface_type = tlm::nw::tlm_network_bw_transport_if<TYPES>;
 
public:
    target_mixin()
    : target_mixin(sc_core::sc_gen_unique_name("target_mixin_socket")) {}

    explicit target_mixin(const sc_core::sc_module_name& n)
    : BASE_TYPE(n)
    , m_fw_process(this)
    , m_bw_process(this)
    , m_current_transaction(nullptr) {
        bind(m_fw_process);
    }

    using BASE_TYPE::bind;
    tlm::nw::tlm_network_bw_transport_if<TYPES>* operator->() { return &m_bw_process; }
    void register_nb_transport_fw(std::function<sync_enum_type(transaction_type&, phase_type&, sc_core::sc_time&)> cb) {
        assert(!sc_core::sc_get_curr_simcontext()->elaboration_done());
        m_fw_process.set_nb_transport_ptr(cb);
    }

    void register_b_transport(std::function<void(transaction_type&, sc_core::sc_time&)> cb) {
        assert(!sc_core::sc_get_curr_simcontext()->elaboration_done());
        m_fw_process.set_b_transport_ptr(cb);
    }

    void register_transport_dbg(std::function<unsigned int(transaction_type&)> cb) {
        assert(!sc_core::sc_get_curr_simcontext()->elaboration_done());
        m_fw_process.set_transport_dbg_ptr(cb);
    }

    void register_get_direct_mem_ptr(std::function<bool(transaction_type&, tlm::tlm_dmi&)> cb) {
        assert(!sc_core::sc_get_curr_simcontext()->elaboration_done());
        m_fw_process.set_get_direct_mem_ptr(cb);
    }
 
private:
    // make call on bw path.
    sync_enum_type bw_nb_transport(transaction_type& trans, phase_type& phase, sc_core::sc_time& t) {
        return BASE_TYPE::operator->()->nb_transport_bw(trans, phase, t);
    }
 
    // Helper class to handle bw path calls
    // Needed to detect transaction end when called from b_transport.
    class bw_process : public tlm::nw::tlm_network_bw_transport_if<TYPES> {
    public:
        bw_process(target_mixin* p_own)
        : m_owner(p_own) {}
 
        sync_enum_type nb_transport_bw(transaction_type& trans, phase_type& phase, sc_core::sc_time& t) {
            typename std::map<transaction_type*, sc_core::sc_event*>::iterator it;
 
            it = m_owner->m_pending_trans.find(&trans);
            if(it == m_owner->m_pending_trans.end()) {
                // Not a blocking call, forward.
                return m_owner->bw_nb_transport(trans, phase, t);
 
            } else {
                if(phase == tlm::nw::CONFIRM || phase == tlm::nw::RESPONSE) {
                    m_owner->m_end_request.notify(sc_core::SC_ZERO_TIME);
                    return tlm::TLM_ACCEPTED;
                } else {
                    assert(false);
                    exit(1);
                }
            }
        }
 
    private:
        target_mixin* m_owner;
    };
 
    class fw_process : public tlm::nw::tlm_network_fw_transport_if<TYPES>, public tlm::nw::tlm_base_mm_interface {
    public:
        using NBTransportPtr = std::function<sync_enum_type(transaction_type&, phase_type&, sc_core::sc_time&)>;
        using BTransportPtr = std::function<void(transaction_type&, sc_core::sc_time&)>;
        using TransportDbgPtr = std::function<unsigned int(transaction_type&)>;
        using GetDirectMemPtr = std::function<bool(transaction_type&, tlm::tlm_dmi&)>;
 
        fw_process(target_mixin* p_own)
        : m_name(p_own->name())
        , m_owner(p_own)
        , m_nb_transport_ptr(0)
        , m_b_transport_ptr(0)
        , m_transport_dbg_ptr(0)
        , m_get_direct_mem_ptr(0)
        , m_peq(sc_core::sc_gen_unique_name("m_peq")) {
            sc_core::sc_spawn_options opts;
            opts.set_sensitivity(&m_peq.get_event());
            sc_core::sc_spawn(sc_bind(&fw_process::b2nb_thread, this), sc_core::sc_gen_unique_name("b2nb_thread"), &opts);
        }
 
        void set_nb_transport_ptr(NBTransportPtr p) {
            if(m_nb_transport_ptr) {
                std::stringstream s;
                s << m_name << ": non-blocking callback allready registered";
                SC_REPORT_WARNING("/OSCI_TLM-2/simple_socket", s.str().c_str());
            } else {
                m_nb_transport_ptr = p;
            }
        }
 
        void set_b_transport_ptr(BTransportPtr p) {
            if(m_b_transport_ptr) {
                std::stringstream s;
                s << m_name << ": blocking callback allready registered";
                SC_REPORT_WARNING("/OSCI_TLM-2/simple_socket", s.str().c_str());
            } else {
                m_b_transport_ptr = p;
            }
        }
 
        void set_transport_dbg_ptr(TransportDbgPtr p) {
            if(m_transport_dbg_ptr) {
                std::stringstream s;
                s << m_name << ": debug callback allready registered";
                SC_REPORT_WARNING("/OSCI_TLM-2/simple_socket", s.str().c_str());
            } else {
                m_transport_dbg_ptr = p;
            }
        }
 
        void set_get_direct_mem_ptr(GetDirectMemPtr p) {
            if(m_get_direct_mem_ptr) {
                std::stringstream s;
                s << m_name << ": get DMI pointer callback allready registered";
                SC_REPORT_WARNING("/OSCI_TLM-2/simple_socket", s.str().c_str());
            } else {
                m_get_direct_mem_ptr = p;
            }
        }
        // Interface implementation
        sync_enum_type nb_transport_fw(transaction_type& trans, phase_type& phase, sc_core::sc_time& t) {
            if(m_nb_transport_ptr) {
                // forward call
                return m_nb_transport_ptr(trans, phase, t);
            } else if(m_b_transport_ptr) {
                if(phase == tlm::nw::REQUEST || phase == tlm::nw::INDICATION) {
                    // prepare thread to do blocking call
                    process_handle_class* ph = m_process_handle.get_handle(&trans, phase);
 
                    if(!ph) { // create new dynamic process
                        ph = new process_handle_class(&trans, phase);
                        m_process_handle.put_handle(ph);
 
                        sc_core::sc_spawn_options opts;
                        opts.dont_initialize();
                        opts.set_sensitivity(&ph->m_e);
                        sc_core::sc_spawn(sc_bind(&fw_process::nb2b_thread, this, ph), sc_core::sc_gen_unique_name("nb2b_thread"), &opts);
                    }
                    ph->m_e.notify(t);
                    return tlm::TLM_ACCEPTED;
                } else {
                    assert(false);
                    exit(1);
                }
            } else {
                std::stringstream s;
                s << m_name << ": no non-blocking transport callback registered";
                SC_REPORT_ERROR("/OSCI_TLM-2/simple_socket", s.str().c_str());
            }
            return tlm::TLM_ACCEPTED; 
        }
 
        void b_transport(transaction_type& trans, sc_core::sc_time& t) {
            if(m_b_transport_ptr) {
                // forward call
                m_b_transport_ptr(trans, t);
                return;
 
            } else if(m_nb_transport_ptr) {
                m_peq.notify(trans, t);
                t = sc_core::SC_ZERO_TIME;
 
                mm_end_event_ext mm_ext;
                const bool mm_added = !trans.has_mm();
 
                if(mm_added) {
                    trans.set_mm(this);
                    trans.set_auto_extension(&mm_ext);
                    trans.acquire();
                }
 
                // wait until transaction is finished
                sc_core::sc_event end_event;
                m_owner->m_pending_trans[&trans] = &end_event;
                sc_core::wait(end_event);
 
                if(mm_added) {
                    // release will not delete the transaction, it will notify mm_ext.done
                    trans.release();
                    if(trans.get_ref_count()) {
                        sc_core::wait(mm_ext.done);
                    }
                    trans.set_mm(0);
                }
 
            } else {
                std::stringstream s;
                s << m_name << ": no blocking transport callback registered";
                SC_REPORT_ERROR("/OSCI_TLM-2/simple_socket", s.str().c_str());
            }
        }
 
        unsigned int transport_dbg(transaction_type& trans) {
            if(m_transport_dbg_ptr) {
                // forward call
                return m_transport_dbg_ptr(trans);
            } else {
                // No debug support
                return 0;
            }
        }
 
        bool get_direct_mem_ptr(transaction_type& trans, tlm::tlm_dmi& dmi_data) {
            if(m_get_direct_mem_ptr) {
                // forward call
                return m_get_direct_mem_ptr(trans, dmi_data);
 
            } else {
                // No DMI support
                dmi_data.allow_read_write();
                dmi_data.set_start_address(0x0);
                dmi_data.set_end_address((sc_dt::uint64)-1);
                return false;
            }
        }
 
    private:
        // dynamic process handler for nb2b conversion
 
        class process_handle_class {
        public:
            explicit process_handle_class(transaction_type* trans, phase_type phase)
            : m_trans(trans)
            , m_phase(phase)
            , m_suspend(false) {}
 
            transaction_type* m_trans;
            phase_type m_phase;
            sc_core::sc_event m_e;
            bool m_suspend;
        };
 
        class process_handle_list {
        public:
            process_handle_list() = default;
 
            ~process_handle_list() {
                for(typename std::vector<process_handle_class*>::iterator it = v.begin(), end = v.end(); it != end; ++it)
                    delete *it;
            }
 
            process_handle_class* get_handle(transaction_type* trans, phase_type phase) {
                typename std::vector<process_handle_class*>::iterator it;
 
                for(it = v.begin(); it != v.end(); it++) {
                    if((*it)->m_suspend) {      // found suspended dynamic process, re-use it
                        (*it)->m_trans = trans; // replace to new one
                        (*it)->m_phase = phase;
                        (*it)->m_suspend = false;
                        return *it;
                    }
                }
                return NULL; // no suspended process
            }
 
            void put_handle(process_handle_class* ph) { v.push_back(ph); }
 
        private:
            std::vector<process_handle_class*> v;
        };
 
        process_handle_list m_process_handle;
 
        void nb2b_thread(process_handle_class* h) {
            while(true) {
                transaction_type* trans = h->m_trans;
                sc_core::sc_time t = sc_core::SC_ZERO_TIME;
                // forward call
                m_b_transport_ptr(*trans, t);
                sc_core::wait(t);
                t = sc_core::SC_ZERO_TIME;
                phase_type phase = tlm::nw::RESPONSE;
                if(h->m_phase == tlm::nw::REQUEST)
                    phase = tlm::nw::CONFIRM;
                sync_enum_type sync = m_owner->bw_nb_transport(*trans, phase, t);
                // suspend until next transaction
                h->m_suspend = true;
                sc_core::wait();
            }
        }
 
        void b2nb_thread() {
            while(true) {
                sc_core::wait(m_peq.get_event());
                transaction_type* trans;
                while((trans = m_peq.get_next_transaction()) != 0) {
                    assert(m_nb_transport_ptr);
                    phase_type phase = tlm::nw::REQUEST;
                    if(REQUESTOR)
                        phase = tlm::nw::INDICATION;
                    sc_core::sc_time t = sc_core::SC_ZERO_TIME;
 
                    switch(m_nb_transport_ptr(*trans, phase, t)) {
                    case tlm::TLM_COMPLETED: {
                        assert(phase == tlm::nw::CONFIRM || phase == tlm::nw::RESPONSE);
                        // notify transaction is finished
                        typename std::map<transaction_type*, sc_core::sc_event*>::iterator it = m_owner->m_pending_trans.find(trans);
                        assert(it != m_owner->m_pending_trans.end());
                        it->second->notify(t);
                        m_owner->m_pending_trans.erase(it);
                        break;
                    }
                    case tlm::TLM_ACCEPTED:
                    case tlm::TLM_UPDATED:
                        if(phase == tlm::nw::REQUEST || phase == tlm::nw::INDICATION) {
                            m_owner->m_current_transaction = trans;
                            sc_core::wait(m_owner->m_end_request);
                            m_owner->m_current_transaction = 0;
                        } else if(phase == tlm::nw::CONFIRM || phase == tlm::nw::RESPONSE) {
                            sc_core::wait(t);
                        } else
                            assert(false && "Illegal phase received");
                        break;
                    default:
                        assert(false && "Illegate sync enum received");
                        exit(1);
                    };
                }
            }
        }
 
        void free(tlm::nw::tlm_network_payload_base* trans) {
            mm_end_event_ext* ext = trans->template get_extension<mm_end_event_ext>();
            assert(ext);
            // notif event first before freeing extensions (reset)
            ext->done.notify();
            trans->reset();
        }
 
    private:
        class mm_end_event_ext : public tlm::tlm_extension<mm_end_event_ext> {
        public:
            tlm::tlm_extension_base* clone() const { return NULL; }
            void free() {}
            void copy_from(tlm::tlm_extension_base const&) {}
            sc_core::sc_event done;
        };
 
        const std::string m_name;
        target_mixin* m_owner;
        NBTransportPtr m_nb_transport_ptr;
        BTransportPtr m_b_transport_ptr;
        TransportDbgPtr m_transport_dbg_ptr;
        GetDirectMemPtr m_get_direct_mem_ptr;
        tlm_utils::peq_with_get<transaction_type> m_peq;
    };
 
private:
    fw_process m_fw_process;
    bw_process m_bw_process;
    std::map<transaction_type*, sc_core::sc_event*> m_pending_trans;
    sc_core::sc_event m_end_request;
    transaction_type* m_current_transaction;
};
} // namespace nw
} // namespace tlm
 
#endif //_TLM_NW_TARGET_MIXIN_H_