added floodfill network visualization script

This commit is contained in:
Kevin Froman 2020-10-26 06:57:22 +00:00
parent 4a831a4a4d
commit f80d58e5b8
1 changed files with 33 additions and 0 deletions

View File

@ -0,0 +1,33 @@
import networkx as nx
import matplotlib.pyplot as plt
import sys
import os
import subprocess
import base64
if not os.path.exists('onionr.sh'):
os.chdir('../')
sys.path.append("src/")
from streamfill import identify_neighbors
G = nx.Graph()
size = 20
onions = []
p = subprocess.Popen(["scripts/generate-onions.py", str(size)],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
for line in iter(p.stdout.readline, b''):
line = line.decode().strip()
onions.append(line)
G.add_node(line[:4])
for onion in onions:
neighbors = identify_neighbors(onion, onions, 0.25 * size)
for neighbor in neighbors:
G.add_edge(onion[:4], neighbor[:4])
#nx.draw(G, with_labels=True, font_weight='bold')
#nx.draw_shell(G, with_labels=True)
#nx.draw_random(G, with_labels=True)
nx.draw_kamada_kawai(G, with_labels=True)
plt.savefig("graph.png")