From c8ed08691fab56f86ea2de3675eb6a1a85fe065e Mon Sep 17 00:00:00 2001
From: Honza <jan.klhufek@gmail.com>
Date: Tue, 16 Nov 2021 00:02:52 +0100
Subject: [PATCH] Updated functionality of the extend_bus method.
---
ariths_gen/wire_components/buses.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/ariths_gen/wire_components/buses.py b/ariths_gen/wire_components/buses.py
index 02e68fc..ed6d040 100644
--- a/ariths_gen/wire_components/buses.py
+++ b/ariths_gen/wire_components/buses.py
@@ -48,18 +48,23 @@ class Bus():
"""
return self.out_bus
- def bus_extend(self, N: int, prefix: str = "bus"):
+ def bus_extend(self, N: int, prefix: str = "bus", last_wire_extend: bool = True):
"""Provides bus extension to contain more wires.
Args:
N (int): Number of wires in the bus. Defaults to 1.
prefix (str, optional): Prefix name of the bus. Defaults to "bus".
+ last_wire_extend (bool, optional): Specifies whether the last wire of the bus should be extended (connected) to all the extending wires. Defaults to True.
"""
# Checks if any extension is neccesarry and if so, proceeds to wire extend the bus
if self.N < N:
# Adding wires into current bus's wires list (wire names are concatenated from bus prefix and their index position inside the bus in square brackets)
self.bus += [Wire(name=prefix+f"[{i}]", prefix=prefix, index=i, parent_bus=self) for i in range(self.N, N)]
+ if last_wire_extend is True:
+ for w_index in range(self.N, N):
+ self.connect(bus_wire_index=w_index, inner_component_out_wire=self.get_wire(self.N - 1))
self.N = N
+
def get_wire(self, wire_index: int = 0):
"""Retrieves a wire from the bus by a given index.