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

added comments

Browse Source
This commit is contained in:
0xFugue 2023-12-12 18:28:24 +05:30
parent 0ebc6b7f03
commit 0471251edf
1 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
scripts/analysis.py
View File

@ -12,33 +12,40 @@ import matplotlib.pyplot as plt
from os import walk
from collections import defaultdict
# read a json and return the dict
def read_json(fname):
with open(fname) as f:
cdata = json.load(f)
return cdata
# write the json from a dict
def write_json(dic, fname):
dic_str = {str(k): str(v) for k,v in dic.items()}
jdump = json.dumps(dic_str, indent=4, sort_keys=True)
with open(fname, 'w') as f:
f.write(jdump)
# read a serialised python dict
def read_dict(fname):
with open(fname, 'r') as f:
return eval(f.read())
# write a serialised python fict
def write_dict(dic, fname):
with open(fname, 'w') as f:
return f.write(str(dic))
# read the output csv and return a panadas dataframe
def read_csv(fname):
df = pd.read_csv(fname, header=0, comment='#', skipinitialspace = True )
return df
# write pandas dataframe as csv
def write_csv(df, fname):
df.to_csv(fname)
# compute the steps it took to compute the views, and tag the tree depth
def compute_view_finalisation_times(df, conf, oprefix, simtype, tag="tag", plot=False):
if simtype == "tree":
num_nodes = conf["node_count"]
@ -52,7 +59,7 @@ def compute_view_finalisation_times(df, conf, oprefix, simtype, tag="tag", plot=
two3rd = math.floor(conf["node_count"] * 2/3) + 1
#two3rd = math.floor(conf["node_count"] * 3/3)
# for k+i^th view, last two views will need to be omiiited
# for view_offset^th view, last view_offset number of views will need to be omitted
view_offset = 1
#views, view2fin_time = df.current_view.unique()[:-2], {}
views, view2fin_time = df.current_view.unique()[:-view_offset], {}
@ -86,6 +93,8 @@ def compute_view_finalisation_times(df, conf, oprefix, simtype, tag="tag", plot=
plt.show()
plt.savefig(f'{oprefix}-view-finalisation-times.pdf', format="pdf", bbox_inches="tight")
# iterate over the different networks/overlay type and collect view finalisation times
def compute_view_times(path, oprefix, otype):
nwsize2vfins = {}
#conf_fnames = next(walk(f'{path}/configs'), (None, None, []))[2]
@ -122,6 +131,8 @@ def compute_view_times(path, oprefix, otype):
nwsize2vfins[num_nodes] = [(simtype, max_depth, view2fin, tag)]
return nwsize2vfins
# plot the view times, add log plots for comparison
def plot_view_times(nwsize2vfins, simtype, oprefix, otype):
logbands = {}
logbands[simtype] = {}
@ -174,6 +185,8 @@ def plot_view_times(nwsize2vfins, simtype, oprefix, otype):
plt.close()
return data
# plot tree vs branch against the number of nodes; works only when #tree sims = # branch sims
def plot_tree_vs_branch(tree, branch, oprefix):
print(tree, branch)
@ -210,6 +223,7 @@ def plot_tree_vs_branch(tree, branch, oprefix):
app = typer.Typer()
@app.command()
def view(ctx: typer.Context,
data_file: Path = typer.Option("config.json",
@ -227,6 +241,7 @@ def view(ctx: typer.Context,
conf, df = read_json(config_file), read_csv(data_file)
compute_view_finalisation_times(df, conf, oprefix, simtype, tag, plot=True)
@app.command()
def views(ctx: typer.Context,
path: Path = typer.Option("../",
@ -249,6 +264,7 @@ def views(ctx: typer.Context,
plot_tree_vs_branch(tree, branch, oprefix)
@app.command()
def test():
pass