logos-blockchain-specs/mixnet/v2/sim/main.py

import argparse

import dacite
import matplotlib.pyplot as plt
import pandas as pd
import seaborn
import yaml

from config import Config
from simulation import Simulation

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Run simulation', formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument("--config", type=str, required=True, help="Configuration file path")
    args = parser.parse_args()

    with open(args.config, "r") as f:
        config = yaml.safe_load(f)
    config = dacite.from_dict(data_class=Config, data=config)

    sim = Simulation(config)
    sim.run()

    df = pd.DataFrame(sim.p2p.message_sizes, columns=["message_size"])
    print(df.describe())
    plt.figure(figsize=(10, 6))
    seaborn.boxplot(y=df["message_size"])
    plt.title("Message size distribution")
    plt.ylabel("Message Size (bytes)")
    plt.show()

    print("Simulation complete!")