logos-co
/
Waku-topology-test
mirror of https://github.com/logos-co/Waku-topology-test.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

gitignore & minor edits

This commit is contained in:
0xFugue 2022-12-30 19:09:53 +05:30
parent d4f2bd1df2
commit 23238e77a6
2 changed files with 21 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.gitignore vendored
View File

@ -3,6 +3,10 @@
*.swo *.swo
*~ *~
# dirs
Waku
WakuNetwork
# local json # local json
Topology.* Topology.*
topology.* topology.*

32
generate_network.py
View File

@ -10,10 +10,9 @@ import typer
from enum import Enum from enum import Enum
# Enums Consts # Enums & Consts
# To add a new node type, add appropriate entries to the nodeType and nodeTypeSwitch
# To add a new node type, add appropriate entries to the enum and switch
class nodeType(Enum): class nodeType(Enum):
DESKTOP = "desktop" # waku desktop config DESKTOP = "desktop" # waku desktop config
MOBILE = "mobile" # waku mobile config MOBILE = "mobile" # waku mobile config
@ -24,8 +23,8 @@ nodeTypeSwitch = {
} }
# To add a new network type, add appropriate entries to the enum and switch # To add a new network type, add appropriate entries to the networkType and networkTypeSwitch
# the networkTypeSwitch is before generate_network(); no fwd declaration with typer/python # the networkTypeSwitch is placed before generate_network(): fwd declaration mismatch with typer/python :/
class networkType(Enum): class networkType(Enum):
CONFIGMODEL = "configmodel" CONFIGMODEL = "configmodel"
SCALEFREE = "scalefree" # power law SCALEFREE = "scalefree" # power law
@ -34,11 +33,12 @@ class networkType(Enum):
BALANCEDTREE = "balancedtree" # committees? BALANCEDTREE = "balancedtree" # committees?
STAR = "star" # spof STAR = "star" # spof
NW_DATA_FNAME = "network_data.json" NW_DATA_FNAME = "network_data.json"
PREFIX = "waku_" PREFIX = "waku_"
### I/O related fns ########################################################### ### I/O related fns ##############################################################
# Dump to a json file # Dump to a json file
def write_json(dirname, json_dump): def write_json(dirname, json_dump):
@ -67,6 +67,7 @@ def read_json(fname):
jdata = json.load(f) jdata = json.load(f)
return nx.node_link_graph(jdata) return nx.node_link_graph(jdata)
def exists_and_nonempty(dirname): def exists_and_nonempty(dirname):
if not os.path.exists(dirname): if not os.path.exists(dirname):
return False return False
@ -79,7 +80,8 @@ def exists_and_nonempty(dirname):
else: else:
return False return False
### topics related fns ###########################################################
### topics related fns #############################################################
# Generate a random string of upper case chars # Generate a random string of upper case chars
def generate_random_string(n): def generate_random_string(n):
@ -104,7 +106,7 @@ def get_random_sublist(topics):
return sublist return sublist
### network processing related fns ########################################################### ### network processing related fns #################################################
# Network Types # Network Types
def generate_config_model(n): def generate_config_model(n):
@ -168,7 +170,7 @@ def generate_toml(topics, node_type=nodeType.DESKTOP):
# Generates network-wide json and per-node toml and writes them # Generates network-wide json and per-node toml and writes them
def generate_and_write_files(dirname, num_topics, H): def generate_and_write_files(dirname, num_topics, H):
topics = generate_topics(num_topics) topics = generate_topics(num_topics)
json_dump = {} json_dump = {}
for node in H.nodes: for node in H.nodes:
@ -180,26 +182,26 @@ def generate_and_write_files(dirname, num_topics, H):
write_json(dirname, json_dump) # network wide json write_json(dirname, json_dump) # network wide json
### the main ########################################################### ### the main ##########################################################################
def main( def main(
dirname: str = "Waku", num_nodes: int = 4, num_topics: int = 1, dirname: str = "WakuNetwork", num_nodes: int = 4, num_topics: int = 1,
nw_type: networkType = networkType.NEWMANWATTSSTROGATZ.value, nw_type: networkType = networkType.NEWMANWATTSSTROGATZ.value,
node_type: nodeType = nodeType.DESKTOP.value, node_type: nodeType = nodeType.DESKTOP.value,
num_partitions: int = 1): num_partitions: int = 1):
if num_partitions > 1: if num_partitions > 1:
print("-p",num_partitions, "Sorry, we do not yet support partitions") print(f"--num-partitions {num_partitions}, Sorry, we do not yet support partitions")
sys.exit(1) sys.exit(1)
# Generate the network and post-process # Generate the network
G = generate_network(num_nodes, nw_type) G = generate_network(num_nodes, nw_type)
# Generate file format specific data structs and write the files; optionally, draw the network # Refuse to overwrite non-empty dirs
if exists_and_nonempty(dirname) : if exists_and_nonempty(dirname) :
sys.exit(1) sys.exit(1)
os.makedirs(dirname, exist_ok=True) os.makedirs(dirname, exist_ok=True)
# write files and draw the network # Generate file format specific data structs and write the files; optionally, draw the network
generate_and_write_files(dirname, num_topics, G) generate_and_write_files(dirname, num_topics, G)
draw(dirname, G) draw(dirname, G)