TestbenchAxisModuleGenerator.cpp

Go to the documentation of this file.

/*
 *
 *                   _/_/_/    _/_/   _/    _/ _/_/_/    _/_/
 *                  _/   _/ _/    _/ _/_/  _/ _/   _/ _/    _/
 *                 _/_/_/  _/_/_/_/ _/  _/_/ _/   _/ _/_/_/_/
 *                _/      _/    _/ _/    _/ _/   _/ _/    _/
 *               _/      _/    _/ _/    _/ _/_/_/  _/    _/
 *
 *             ***********************************************
 *                              PandA Project
 *                     URL: http://panda.dei.polimi.it
 *                       Politecnico di Milano - DEIB
 *                        System Architectures Group
 *             ***********************************************
 *              Copyright (C) 2023-2024 Politecnico di Milano
 *
 *   This file is part of the PandA framework.
 *
 *   The PandA framework is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
#include "TestbenchAxisModuleGenerator.hpp"

#include "behavioral_helper.hpp"
#include "function_behavior.hpp"
#include "hls_manager.hpp"
#include "language_writer.hpp"
#include "structural_manager.hpp"
#include "tree_helper.hpp"
#include "tree_manager.hpp"

TestbenchAxisModuleGenerator::TestbenchAxisModuleGenerator(const HLS_managerRef& _HLSMgr) : Registrar(_HLSMgr)
{
}

void TestbenchAxisModuleGenerator::InternalExec(std::ostream& out, structural_objectRef mod_cir,
                                                unsigned int function_id, vertex /* op_v */,
                                                const HDLWriter_Language language,
                                                const std::vector<ModuleGenerator::parameter>& /* _p */,
                                                const std::vector<ModuleGenerator::parameter>& /* _ports_in */,
                                                const std::vector<ModuleGenerator::parameter>& /* _ports_out */,
                                                const std::vector<ModuleGenerator::parameter>& /* _ports_inout */)
{
   if(language != HDLWriter_Language::VERILOG)
   {
      THROW_UNREACHABLE("Unsupported output language");
      return;
   }

   const auto arg_name = mod_cir->get_id().substr(sizeof("if_axis_") - 1U, std::string::npos);

   const auto top_bh = HLSMgr->CGetFunctionBehavior(function_id)->CGetBehavioralHelper();
   const auto top_fname = top_bh->GetMangledFunctionName();
   const auto top_fnode = HLSMgr->get_tree_manager()->CGetTreeReindex(function_id);
   const auto return_type = tree_helper::GetFunctionReturnType(top_fnode);
   const auto& iface_attrs = HLSMgr->module_arch->GetArchitecture(top_fname)->ifaces.at(arg_name);
   const auto if_dir = port_o::to_port_direction(iface_attrs.at(FunctionArchitecture::iface_direction));
   const auto if_alignment = iface_attrs.at(FunctionArchitecture::iface_alignment);
   const auto if_ndir = if_dir == port_o::IN ? port_o::OUT : port_o::IN;
   const auto port_prefix = (if_dir == port_o::IN ? "s_axis_" : "m_axis_") + arg_name;
   std::string np_library = mod_cir->get_id() + " index";
   std::string ip_components;
   const auto add_port_parametric = [&](const std::string& suffix, port_o::port_direction dir, unsigned port_size) {
      const auto port_name = port_prefix + suffix;
      structural_manager::add_port(port_name, dir, mod_cir,
                                   structural_type_descriptorRef(new structural_type_descriptor("bool", port_size)));
      np_library += " " + port_name;
   };
   add_port_parametric("_TDATA", if_ndir, 1U);
   add_port_parametric("_TVALID", if_ndir, 0U);
   add_port_parametric("_TREADY", if_dir, 0U);
   out << "localparam BITSIZE_data=BITSIZE_" << port_prefix << "_TDATA;\n";

   if(if_dir == port_o::IN)
   {
      ip_components = "TestbenchFifoRead";
      out << "assign tb_done_port = 1'b1;\n\n"
          << "TestbenchFifoRead #(.index(index),\n"
          << "  .CHECK_ACK(1),\n"
          << "  .BITSIZE_dout(BITSIZE_data)) fifo_read(.clock(clock),\n"
          << "  .setup_port(setup_port),\n"
          << "  .done_port(done_port),\n"
          << "  .empty_n(" << port_prefix << "_TVALID),\n"
          << "  .read(" << port_prefix << "_TREADY),\n"
          << "  .dout(" << port_prefix << "_TDATA));\n";
   }
   else if(if_dir == port_o::OUT)
   {
      ip_components = "TestbenchFifoWrite";
      out << "wire _full_n;\n"
          << "assign " << port_prefix << "_TREADY = _full_n;\n"
          << "assign tb_done_port = ~_full_n;\n\n"
          << "TestbenchFifoWrite #(.index(index),\n"
          << "  .BITSIZE_din(BITSIZE_data)) fifo_write(.clock(clock),\n"
          << "  .setup_port(setup_port),\n"
          << "  .done_port(done_port),\n"
          << "  .full_n(_full_n),\n"
          << "  .write(" << port_prefix << "_TVALID),\n"
          << "  .din(" << port_prefix << "_TDATA));\n";
   }
   else
   {
      THROW_UNREACHABLE("Unknown AXIS interface port direction: " + port_o::GetString(if_dir));
   }

   structural_manager::add_NP_functionality(mod_cir, NP_functionality::IP_COMPONENT, ip_components);
   structural_manager::add_NP_functionality(mod_cir, NP_functionality::LIBRARY, np_library);
}