axi_initiator.h

/*
 * Copyright 2021 Arteris IP
 *
 * 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.
 */
 
#pragma once
 
#include <axi/axi_tlm.h>
#include <axi/fsm/protocol_fsm.h>
#include <tlm/scc/pe/intor_if.h>
#include <scc/ordered_semaphore.h>
#include <scc/peq.h>
#include <systemc>
#include <tlm_utils/peq_with_get.h>
#include <tuple>
#include <unordered_map>
 
namespace axi {
namespace pe {
 
class axi_initiator_b :
        public sc_core::sc_module,
        public axi::ace_bw_transport_if<axi::axi_protocol_types>,
        public tlm::scc::pe::intor_fw_b
{
public:
 
    using payload_type = axi::axi_protocol_types::tlm_payload_type;
    using phase_type = axi::axi_protocol_types::tlm_phase_type;
 
    sc_core::sc_in<bool> clk_i{"clk_i"};
 
    sc_core::sc_export<tlm::scc::pe::intor_fw_b> fw_i{"fw_i"};
 
    sc_core::sc_port<tlm::scc::pe::intor_bw_b, 1, sc_core::SC_ZERO_OR_MORE_BOUND> bw_o{"bw_o"};
 
    void b_snoop(payload_type& trans, sc_core::sc_time& t) override;
 
    tlm::tlm_sync_enum nb_transport_bw(payload_type& trans, phase_type& phase, sc_core::sc_time& t) override;
 
    void invalidate_direct_mem_ptr(sc_dt::uint64 start_range, sc_dt::uint64 end_range) override;
 
    size_t get_transferwith_in_bytes() const { return transfer_width_in_bytes; }
    void transport(payload_type& trans, bool blocking) override;
    void snoop_resp(payload_type& trans, bool sync = false) override;
 
    axi_initiator_b(sc_core::sc_module_name nm, sc_core::sc_port_b<axi::axi_fw_transport_if<axi_protocol_types>>& port,
                    size_t transfer_width, flavor_e flavor);
 
    virtual ~axi_initiator_b();
 
    axi_initiator_b() = delete;
 
    axi_initiator_b(axi_initiator_b const&) = delete;
 
    axi_initiator_b(axi_initiator_b&&) = delete;
 
    axi_initiator_b& operator=(axi_initiator_b const&) = delete;
 
    axi_initiator_b& operator=(axi_initiator_b&&) = delete;
 
    // AXI4 does not allow write data interleaving, and ncore3 only supports AXI4.
    // However, AXI3 allows data interleaving and there may be support for AXI3 in Symphony, so keep it configurable in
    // the testbench.
    sc_core::sc_attribute<bool> data_interleaving{"data_interleaving", false};
    sc_core::sc_attribute<unsigned> artv{"artv", 1};
    sc_core::sc_attribute<unsigned> awtv{"awtv", 1};
    sc_core::sc_attribute<unsigned> wbv{"wbv", 1};
    sc_core::sc_attribute<unsigned> rbr{"rbr", 0};
    sc_core::sc_attribute<unsigned> br{"br", 0};
    sc_core::sc_attribute<unsigned> rla{"rla", 1};
    sc_core::sc_attribute<unsigned> ba{"ba", 1};
    sc_core::sc_attribute<bool> enable_id_serializing{"enable_id_serializing", false};
    sc_core::sc_attribute<unsigned> outstanding_snoops{"outstanding_snoops", 8};
 
    void add_protocol_cb(axi::fsm::protocol_time_point_e e, std::function<void(payload_type&, bool)> cb) {
        assert(e < axi::fsm::CB_CNT);
        protocol_cb[e] = cb;
    }
 
 
protected:
    unsigned calculate_beats(payload_type& p) {
        sc_assert(p.get_data_length() > 0);
        return p.get_data_length() < transfer_width_in_bytes ? 1 : p.get_data_length() / transfer_width_in_bytes;
    }
 
    void snoop_thread();
 
    const size_t transfer_width_in_bytes;
 
    const flavor_e flavor;
 
    sc_core::sc_port_b<axi::axi_fw_transport_if<axi_protocol_types>>& socket_fw;
 
    struct tx_state {
        payload_type* active_tx{nullptr};
        scc::peq<std::tuple<payload_type*, tlm::tlm_phase>> peq;
    };
    std::unordered_map<void*, tx_state*> tx_state_by_tx;
    std::unordered_map<unsigned, scc::ordered_semaphore*> id_mtx;
 
    tlm_utils::peq_with_get<payload_type> snp_peq{"snp_peq"};
 
    std::unordered_map<void*, tx_state*> snp_state_by_id;
 
    scc::ordered_semaphore rd_chnl{1};
 
    scc::ordered_semaphore wr_chnl{1};
 
    scc::ordered_semaphore sresp_chnl{1};
 
    sc_core::sc_event any_tx_finished;
 
    sc_core::sc_time clk_period{10, sc_core::SC_NS};
 
private:
    sc_core::sc_clock* clk_if{nullptr};
    void end_of_elaboration() override;
    void clk_counter() { m_clock_counter++; }
    unsigned get_clk_cnt() { return m_clock_counter; }
 
    tlm::tlm_phase send(payload_type& trans, axi::pe::axi_initiator_b::tx_state* txs, tlm::tlm_phase phase);
 
    unsigned m_clock_counter{0};
    unsigned m_prev_clk_cnt{0};
    unsigned snoops_in_flight{0};
 
    std::array<std::function<void(payload_type&, bool)>, axi::fsm::CB_CNT> protocol_cb;
};
 
template <unsigned int BUSWIDTH = 32, typename TYPES = axi::axi_protocol_types, int N = 1,
          sc_core::sc_port_policy POL = sc_core::SC_ONE_OR_MORE_BOUND>
class axi_initiator : public axi_initiator_b {
public:
    using base = axi_initiator_b;
 
    using payload_type = base::payload_type;
    using phase_type = base::phase_type;
    axi_initiator(const sc_core::sc_module_name& nm, axi::axi_initiator_socket<BUSWIDTH, TYPES, N, POL>& socket_)
    : axi_initiator_b(nm, socket_.get_base_port(), BUSWIDTH, flavor_e::AXI)
    , socket(socket_) {
        socket(*this);
    }
 
    axi_initiator() = delete;
 
    axi_initiator(axi_initiator const&) = delete;
 
    axi_initiator(axi_initiator&&) = delete;
 
    axi_initiator& operator=(axi_initiator const&) = delete;
 
    axi_initiator& operator=(axi_initiator&&) = delete;
 
private:
    axi::axi_initiator_socket<BUSWIDTH, TYPES, N, POL>& socket;
};
 
template <unsigned int BUSWIDTH = 32, typename TYPES = axi::axi_protocol_types, int N = 1,
          sc_core::sc_port_policy POL = sc_core::SC_ONE_OR_MORE_BOUND>
class ace_lite_initiator : public axi_initiator_b {
public:
    using base = axi_initiator_b;
 
    using payload_type = base::payload_type;
    using phase_type = base::phase_type;
    ace_lite_initiator(const sc_core::sc_module_name& nm, axi::axi_initiator_socket<BUSWIDTH, TYPES, N, POL>& socket_)
    : axi_initiator_b(nm, socket_.get_base_port(), BUSWIDTH, flavor_e::ACEL)
    , socket(socket_) {
        socket(*this);
    }
 
    ace_lite_initiator() = delete;
 
    ace_lite_initiator(ace_lite_initiator const&) = delete;
 
    ace_lite_initiator(ace_lite_initiator&&) = delete;
 
    ace_lite_initiator& operator=(ace_lite_initiator const&) = delete;
 
    ace_lite_initiator& operator=(ace_lite_initiator&&) = delete;
 
private:
    axi::axi_initiator_socket<BUSWIDTH, TYPES, N, POL>& socket;
};
 
template <unsigned int BUSWIDTH = 32, typename TYPES = axi::axi_protocol_types, int N = 1,
          sc_core::sc_port_policy POL = sc_core::SC_ONE_OR_MORE_BOUND>
class ace_initiator : public axi_initiator_b {
public:
    using base = axi_initiator_b;
 
    using payload_type = base::payload_type;
    using phase_type = base::phase_type;
    ace_initiator(const sc_core::sc_module_name& nm, axi::ace_initiator_socket<BUSWIDTH, TYPES, N, POL>& socket_)
    : axi_initiator_b(nm, socket_.get_base_port(), BUSWIDTH, flavor_e::ACE)
    , socket(socket_) {
        socket(*this);
    }
 
    ace_initiator() = delete;
 
    ace_initiator(ace_initiator const&) = delete;
 
    ace_initiator(ace_initiator&&) = delete;
 
    ace_initiator& operator=(ace_initiator const&) = delete;
 
    ace_initiator& operator=(ace_initiator&&) = delete;
 
private:
    axi::ace_initiator_socket<BUSWIDTH, TYPES, N, POL>& socket;
};
 
} /* namespace pe */
} /* namespace axi */