logos-blockchain/logos-blockchain-specs
1
0
Fork 0
mirror of https://github.com/logos-blockchain/logos-blockchain-specs.git synced 2026-01-09 00:23:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

plots for bandwidths per conn

Browse Source
This commit is contained in:
Youngjoon Lee 2024-07-08 16:02:42 +09:00
parent a95b61c451
commit 27ad7f2b77
No known key found for this signature in database
GPG Key ID: B4253AFBA618BF4D
2 changed files with 52 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mixnet/sim/simulation.py
View File

@ -1,4 +1,3 @@
import matplotlib.pyplot as plt
import usim
import mixnet.framework.usim as usimfw
@ -22,8 +21,6 @@ class Simulation:
conn_stats, all_node_states = await self._run()
conn_stats.bandwidths()
all_node_states.analyze()
# Show all plots that have been created so far
plt.show()
async def _run(self) -> tuple[ConnectionStats, AllNodeStates]:
async with usim.until(usim.time + self.config.simulation.duration_sec) as scope:

74
mixnet/sim/stats.py
View File

@ -28,11 +28,42 @@ class ConnectionStats:
self.conns_per_node[node][1].append(outbound_conn)
def bandwidths(self):
plt.figure(figsize=(12, 6))
self._bandwidths_per_conn()
self._bandwidths_per_node()
plt.subplot(2, 1, 1)
def _bandwidths_per_conn(self):
_, axs = plt.subplots(nrows=2, ncols=1, figsize=(12, 6))
for i, (_, (inbound_conns, _)) in enumerate(self.conns_per_node.items()):
for _, (inbound_conns, outbound_conns) in self.conns_per_node.items():
for conn in inbound_conns:
inbound_bandwidths = conn.input_bandwidths().map(lambda x: x / 1024)
axs[0].plot(inbound_bandwidths.index, inbound_bandwidths)
for conn in outbound_conns:
outbound_bandwidths = conn.output_bandwidths().map(lambda x: x / 1024)
axs[1].plot(outbound_bandwidths.index, outbound_bandwidths)
axs[0].set_title("Inbound Bandwidths per Connection")
axs[0].set_xlabel("Time (s)")
axs[0].set_ylabel("Bandwidth (KB/s)")
axs[0].set_ylim(bottom=0)
axs[0].grid(True)
axs[1].set_title("Outbound Bandwidths per Connection")
axs[1].set_xlabel("Time (s)")
axs[1].set_ylabel("Bandwidth (KB/s)")
axs[1].set_ylim(bottom=0)
axs[1].grid(True)
plt.tight_layout()
plt.show()
def _bandwidths_per_node(self):
_, axs = plt.subplots(nrows=2, ncols=1, figsize=(12, 6))
for i, (_, (inbound_conns, outbound_conns)) in enumerate(
self.conns_per_node.items()
):
inbound_bandwidths = (
pandas.concat(
[conn.input_bandwidths() for conn in inbound_conns], axis=1
@ -40,18 +71,6 @@ class ConnectionStats:
.sum(axis=1)
.map(lambda x: x / 1024)
)
plt.plot(inbound_bandwidths.index, inbound_bandwidths, label=f"Node-{i}")
plt.xlabel("Time (s)")
plt.ylabel("Bandwidth (KB/s)")
plt.title("Inbound Bandwidths per Node")
plt.legend()
plt.ylim(bottom=0)
plt.grid(True)
plt.subplot(2, 1, 2)
for i, (_, (_, outbound_conns)) in enumerate(self.conns_per_node.items()):
outbound_bandwidths = (
pandas.concat(
[conn.output_bandwidths() for conn in outbound_conns], axis=1
@ -59,13 +78,24 @@ class ConnectionStats:
.sum(axis=1)
.map(lambda x: x / 1024)
)
plt.plot(outbound_bandwidths.index, outbound_bandwidths, label=f"Node-{i}")
axs[0].plot(inbound_bandwidths.index, inbound_bandwidths, label=f"Node-{i}")
axs[1].plot(
outbound_bandwidths.index, outbound_bandwidths, label=f"Node-{i}"
)
plt.xlabel("Time (s)")
plt.ylabel("Bandwidth (KB/s)")
plt.title("Outbound Bandwidths per Node")
plt.legend()
plt.ylim(bottom=0)
plt.grid(True)
axs[0].set_title("Inbound Bandwidths per Node")
axs[0].set_xlabel("Time (s)")
axs[0].set_ylabel("Bandwidth (KB/s)")
axs[0].legend()
axs[0].set_ylim(bottom=0)
axs[0].grid(True)
axs[1].set_title("Outbound Bandwidths per Node")
axs[1].set_xlabel("Time (s)")
axs[1].set_ylabel("Bandwidth (KB/s)")
axs[1].legend()
axs[1].set_ylim(bottom=0)
axs[1].grid(True)
plt.tight_layout()
plt.show()