From 5cf96ee0da4f38e8f9fb53e0b4ce0dd2257a70a6 Mon Sep 17 00:00:00 2001
From: Louis Navarre <louisnavarre@hotmail.com>
Date: Mon, 23 Jan 2023 22:44:35 +0100
Subject: [PATCH] Add seed to random graph generation

---
 create_graph.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/create_graph.py b/create_graph.py
index 584c307..42b35cb 100644
--- a/create_graph.py
+++ b/create_graph.py
@@ -8,6 +8,8 @@ def create_random_graph(args):
     nb_nodes = args.nodes
     nb_links = args.links
 
+    random.seed(args.seed)
+
     nodes = dict()
     for _ in range(nb_links):
         node_a = random.randint(0, nb_nodes - 1)
@@ -88,8 +90,9 @@ if __name__ == "__main__":
                         help="Number of nodes. Unused if '--ntf'", default=5)
     parser.add_argument("-l", "--links", type=int,
                         help="Number of links. Unused if '--ntf'", default=10)
-    parser.add_argument("--c-header", action="store_true",
+    parser.add_argument("-c", "--c-header", action="store_true",
                         help="Writes the graph as a C header file (.h) instead of a binary file")
+    parser.add_argument("-s", "--seed", type=int, help="Seed for random generation of the graph", default=42)
     args = parser.parse_args()
 
     if args.ntf:
-- 
GitLab