mirror of
https://github.com/logos-blockchain/logos-blockchain-simulations.git
synced 2026-01-10 00:53:11 +00:00
69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
import os
|
|
import typer
|
|
import json
|
|
import pandas as pd
|
|
import numpy as np
|
|
import logging as log
|
|
from pathlib import Path
|
|
import matplotlib.pyplot as plt
|
|
|
|
def read_json(fname):
|
|
with open(fname) as f:
|
|
cdata = json.load(f)
|
|
return cdata
|
|
|
|
|
|
def read_csv(fname):
|
|
df = pd.read_csv(fname, header=0, comment='#', skipinitialspace = True )
|
|
return df
|
|
|
|
|
|
def write_csv(df, fname):
|
|
df.to_csv(fname)
|
|
|
|
|
|
app = typer.Typer()
|
|
|
|
@app.command()
|
|
def views(ctx: typer.Context,
|
|
data_file: Path = typer.Option("simout.csv",
|
|
exists=True, file_okay=True, readable=True,
|
|
help="Set the simulation data file"),
|
|
oprefix: str = typer.Option("output",
|
|
help="Set the output prefix for the plots"),
|
|
debug: bool = typer.Option(True,
|
|
help="Set debug")
|
|
):
|
|
log.basicConfig(level=log.INFO)
|
|
tag = os.path.splitext(os.path.basename(data_file))[0]
|
|
|
|
df = read_csv(data_file)
|
|
steps_df = data=df.drop_duplicates('current_view').step_id.diff().values[1:]
|
|
views_df = df['current_view'].unique()[1:]
|
|
|
|
fig, axes = plt.subplots(1, 1, layout='constrained', sharey=False)
|
|
fig.set_figwidth(12)
|
|
fig.set_figheight(10)
|
|
|
|
fig.suptitle(f'View installation times :: {tag}')
|
|
axes.set_ylabel("Number of Epochs")
|
|
axes.set_xlabel("Views")
|
|
axes.set_xticks([x + 1 for x in range(max(views_df.astype(int)))])
|
|
axes.set_yticks([x + 1 for x in range(max(steps_df.astype(int)))])
|
|
|
|
axes.plot(views_df, steps_df, linestyle='--', marker='o')
|
|
plt.show()
|
|
plt.savefig(f'{oprefix}-view-installion-times.pdf', format="pdf", bbox_inches="tight")
|
|
|
|
|
|
@app.command()
|
|
def other_commands():
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|
|
|
|
|
|
|