UPDATE: json-creator.py

made the code not case-sensitive and made the connection adding, more permissive.

Author: EmilienChinsy

src/json_creator/json-creator.py

@@ -15,19 +15,19 @@ def add_connection(elements, connections, input_id, output_id, timing, input_por
     # Determine element types from the dictionaries
     print(input_io)
     print(output_io)
-    if input_id in elements["LUTs"]:
+    if input_id in elements["luts"]:
         input_type = "lut"
-    elif input_id in elements["FlipFlops"]:
+    elif input_id in elements["flipflops"]:
         input_type = "DFF"
     else:
-        input_type = elements["IOs"].get(input_id, {}).get("name")
+        input_type = elements["ios"].get(input_id, {}).get("name")
 
-    if output_id in elements["LUTs"]:
+    if output_id in elements["luts"]:
         output_type = "lut"
-    elif output_id in elements["FlipFlops"]:
+    elif output_id in elements["flipflops"]:
         output_type = "DFF"
     else:
-        output_type = elements["IOs"].get(output_id, {}).get("name")
+        output_type = elements["ios"].get(output_id, {}).get("name")
     if input_type is None or output_type is None:
         print(f"Error: One of the elements ({input_id}, {output_id}) does not exist.")
         return
@@ -50,15 +50,16 @@ def add_connection(elements, connections, input_id, output_id, timing, input_por
     connections.append(connection)
 
 def user_input():
-    elements = {"LUTs": {}, "FlipFlops": {}, "IOs": {}}
+    elements = {"luts": {}, "flipflops": {}, "ios": {}}
     connections = []
 
     # Creating elements
     while True:
         element_type = input("Enter element type (LUT, FlipFlop, IO) or 'done' to finish: ")
-        if element_type.lower() == 'done':
+        element_type=element_type.lower()
+        if element_type == 'done':
             break
-        if element_type != "IO":
+        if element_type != "io":
             while True:
                 element_id = input("Enter element ID (numeric): ")
                 if element_id.isdigit():
@@ -67,19 +68,19 @@ def user_input():
                     print("Invalid input. Please enter a numeric ID.")
             key = element_id
             elements[element_type + "s"][key] = create_element(element_type, key)
-            if element_type == "LUT":
+            if element_type == "lut":
                 # For LUTs, let the user add port definitions
                 while True:
                     add_port = input(f"Do you want to add a port for {element_type} with ID {element_id}? (yes/no): ")
-                    if add_port.lower() != "yes":
+                    if add_port != "yes":
                         break
                     port_io = input("Enter port's io (input/output): ")
                     port_id = input("Enter port id: ")
                     elements[element_type + "s"][key]["connections"].append({
                         "io": port_io,
                         "id": port_id
                     })
-            elif element_type == "FlipFlop":
+            elif element_type == "flipflop":
                 # For FlipFlops, automatically add the three ports.
                 elements[element_type + "s"][key]["connections"] = [
                     {"port": "clock"},
@@ -89,7 +90,7 @@ def user_input():
         else:
             name = input("Enter name (for IO element): ")
             key = name
-            elements["IOs"][key] = create_element("IO", key, name)
+            elements["ios"][key] = create_element("IO", key, name)
 
     # Adding global connections between elements
     while True:
@@ -98,29 +99,59 @@ def user_input():
             break
 
         input_id = input("Enter input element ID or IO name: ")
-        output_id = input("Enter output element ID or IO name: ")
+        print(input_id)
 
         # If the input element is a LUT or a DFF, prompt for its output port.
         input_port = "0"
         input_io = "0"
-        if input_id in elements["LUTs"]:
+        if input_id in elements["luts"]:
+            input_port = input("Enter the port number for the LUT output: ")
+            if input_port == "":
+                input_port = "0"
+        if input_id.lower()=="lut":
+            input_id = input("Enter the id number for the LUT: ")
+            while input_id not in elements["luts"]:
+               input_id = input("This id does not exist, please try again.")
             input_port = input("Enter the port number for the LUT output: ")
             if input_port == "":
                 input_port = "0"
-        if input_id in elements["FlipFlops"]:
+        if input_id in elements["flipflops"]:
             input_io = input("Enter the port number for the FlipFlop output: ")
             if input_io == "":
                 input_io = "0"
+        if input_id.lower()=="flipflop":
+            input_id = input("Enter the id number for the FlipFlop: ")
+            while input_id not in elements["flipflops"]:
+               input_id = input("This id does not exist, please try again.")
+            input_io = input("Enter the port number for the FlipFlop output: ")
+            if input_io == "":
+                input_io = "0"
+
+        output_id = input("Enter output element ID or IO name: ")
+        print(output_id)
 
         # If the output element is a LUT or a DFF, prompt for its input port.
         output_port = "0"
         output_io = "0"
-        if output_id in elements["LUTs"]:
+        if output_id in elements["luts"]:
+            output_port = input("Enter the port number for the LUT input: ")
+            if output_port == "":
+                output_port = "0"
+        if output_id.lower()=="lut":
+            output_id = input("Enter the id number for the LUT: ")
+            while output_id not in elements["luts"]:
+                output_id = input("This id does not exist, please try again.")
             output_port = input("Enter the port number for the LUT input: ")
             if output_port == "":
                 output_port = "0"
-        if output_id in elements["FlipFlops"]:
-            print("test")
+        if output_id in elements["flipflops"]:
+            output_io = input("Enter the port number for the FlipFlop input: ")
+            if output_io == "":
+                output_io = "0"
+        if output_id.lower()=="flipflop":
+            output_id = input("Enter the id number for the FlipFlop: ")
+            while output_id not in elements["flipflops"]:
+                output_id = input("This id does not exist, please try again.")
             output_io = input("Enter the port number for the FlipFlop input: ")
             if output_io == "":
                 output_io = "0"
@@ -133,9 +164,9 @@ def user_input():
         add_connection(elements, connections, input_id, output_id, timing, input_port, output_port, input_io, output_io)
 
     json_output = {
-        "LUTs": list(elements["LUTs"].values()),
+        "LUTs": list(elements["luts"].values()),
         "FlipFlops": list(elements["FlipFlops"].values()),
-        "IOs": list(elements["IOs"].values()),
+        "IOs": list(elements["ios"].values()),
         "Connections": connections
     }