2024-05-14 14:36:05 +09:00

16 lines
407 B
Python

import simpy
from node import Node
from p2p import P2p
class Simulation:
def __init__(self, num_nodes: int, node_params: Node.Parameters):
self.env = simpy.Environment()
self.p2p = P2p(self.env)
self.nodes = [Node(i, self.env, self.p2p, node_params) for i in range(num_nodes)]
self.p2p.add_node(self.nodes)
def run(self, until):
self.env.run(until=until)