mirror of
				https://github.com/YosysHQ/yosys
				synced 2025-11-03 21:09:12 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			130 lines
		
	
	
	
		
			3.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			130 lines
		
	
	
	
		
			3.9 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/*
 | 
						|
 *  yosys -- Yosys Open SYnthesis Suite
 | 
						|
 *
 | 
						|
 *  Copyright (C) 2012  Claire Xenia Wolf <claire@yosyshq.com>
 | 
						|
 *
 | 
						|
 *  Permission to use, copy, modify, and/or distribute this software for any
 | 
						|
 *  purpose with or without fee is hereby granted, provided that the above
 | 
						|
 *  copyright notice and this permission notice appear in all copies.
 | 
						|
 *
 | 
						|
 *  THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 | 
						|
 *  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 | 
						|
 *  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 | 
						|
 *  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 | 
						|
 *  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 | 
						|
 *  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 | 
						|
 *  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 | 
						|
 *
 | 
						|
 */
 | 
						|
 | 
						|
#include "kernel/yosys.h"
 | 
						|
#include "kernel/sigtools.h"
 | 
						|
 | 
						|
USING_YOSYS_NAMESPACE
 | 
						|
PRIVATE_NAMESPACE_BEGIN
 | 
						|
 | 
						|
struct InsbufPass : public Pass {
 | 
						|
	InsbufPass() : Pass("insbuf", "insert buffer cells for connected wires") { }
 | 
						|
	void help() override
 | 
						|
	{
 | 
						|
		log("\n");
 | 
						|
		log("    insbuf [options] [selection]\n");
 | 
						|
		log("\n");
 | 
						|
		log("Insert buffer cells into the design for directly connected wires.\n");
 | 
						|
		log("\n");
 | 
						|
		log("    -buf <celltype> <in-portname> <out-portname>\n");
 | 
						|
		log("        Use the given cell type instead of $_BUF_. (Notice that the next\n");
 | 
						|
		log("        call to \"clean\" will remove all $_BUF_ in the design.)\n");
 | 
						|
		log("\n");
 | 
						|
		log("    -chain\n");
 | 
						|
		log("        Chain buffer cells\n");
 | 
						|
		log("\n");
 | 
						|
	}
 | 
						|
	void execute(std::vector<std::string> args, RTLIL::Design *design) override
 | 
						|
	{
 | 
						|
		log_header(design, "Executing INSBUF pass (insert buffer cells for connected wires).\n");
 | 
						|
 | 
						|
		IdString celltype = ID($_BUF_), in_portname = ID::A, out_portname = ID::Y;
 | 
						|
		bool chain_mode = false;
 | 
						|
 | 
						|
		size_t argidx;
 | 
						|
		for (argidx = 1; argidx < args.size(); argidx++)
 | 
						|
		{
 | 
						|
			std::string arg = args[argidx];
 | 
						|
			if (arg == "-buf" && argidx+3 < args.size()) {
 | 
						|
				celltype = RTLIL::escape_id(args[++argidx]);
 | 
						|
				in_portname = RTLIL::escape_id(args[++argidx]);
 | 
						|
				out_portname = RTLIL::escape_id(args[++argidx]);
 | 
						|
				continue;
 | 
						|
			}
 | 
						|
			if (arg == "-chain") {
 | 
						|
				chain_mode = true;
 | 
						|
				continue;
 | 
						|
			}
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		extra_args(args, argidx, design);
 | 
						|
 | 
						|
		for (auto module : design->selected_modules())
 | 
						|
		{
 | 
						|
			std::vector<RTLIL::SigSig> new_connections;
 | 
						|
			pool<Cell*> bufcells;
 | 
						|
			SigMap sigmap;
 | 
						|
 | 
						|
			for (auto &conn : module->connections())
 | 
						|
			{
 | 
						|
				RTLIL::SigSig new_conn;
 | 
						|
 | 
						|
				for (int i = 0; i < GetSize(conn.first); i++)
 | 
						|
				{
 | 
						|
					SigBit lhs = conn.first[i];
 | 
						|
					SigBit rhs = conn.second[i];
 | 
						|
 | 
						|
					if (!lhs.wire || !design->selected(module, lhs.wire)) {
 | 
						|
						new_conn.first.append(lhs);
 | 
						|
						new_conn.second.append(rhs);
 | 
						|
						log("Skip %s: %s -> %s\n", log_id(module), log_signal(rhs), log_signal(lhs));
 | 
						|
						continue;
 | 
						|
					}
 | 
						|
 | 
						|
					if (chain_mode && rhs.wire) {
 | 
						|
						rhs = sigmap(rhs);
 | 
						|
						SigBit outbit = sigmap(lhs);
 | 
						|
						sigmap.add(lhs, rhs);
 | 
						|
						sigmap.add(outbit);
 | 
						|
					}
 | 
						|
 | 
						|
					Cell *cell = module->addCell(NEW_ID, celltype);
 | 
						|
					cell->setPort(in_portname, rhs);
 | 
						|
					cell->setPort(out_portname, lhs);
 | 
						|
 | 
						|
					log("Add %s/%s: %s -> %s\n", log_id(module), log_id(cell), log_signal(rhs), log_signal(lhs));
 | 
						|
					bufcells.insert(cell);
 | 
						|
				}
 | 
						|
 | 
						|
				if (GetSize(new_conn.first))
 | 
						|
					new_connections.push_back(new_conn);
 | 
						|
			}
 | 
						|
 | 
						|
			if (chain_mode) {
 | 
						|
				for (auto &cell : module->selected_cells()) {
 | 
						|
					if (bufcells.count(cell))
 | 
						|
						continue;
 | 
						|
					for (auto &port : cell->connections())
 | 
						|
						if (cell->input(port.first)) {
 | 
						|
							auto s = sigmap(port.second);
 | 
						|
							if (s == port.second)
 | 
						|
								continue;
 | 
						|
							log("Rewrite %s/%s/%s: %s -> %s\n", log_id(module), log_id(cell),
 | 
						|
									log_id(port.first), log_signal(port.second), log_signal(s));
 | 
						|
							cell->setPort(port.first, s);
 | 
						|
						}
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			module->new_connections(new_connections);
 | 
						|
		}
 | 
						|
	}
 | 
						|
} InsbufPass;
 | 
						|
 | 
						|
PRIVATE_NAMESPACE_END
 |