66import yaml
77
88__all__ = [
9- ' ParsedProtoVLSIR' ,
10- ' generate_raw_netlist_dict_from_module' ,
11- ' generate_raw_yaml_from_module'
9+ " ParsedProtoVLSIR" ,
10+ " generate_raw_netlist_dict_from_module" ,
11+ " generate_raw_yaml_from_module" ,
1212]
1313
1414ParsedProtoVLSIR = dict
@@ -72,16 +72,18 @@ def _parse_connections(proto_dict: ParsedProtoVLSIR) -> dict:
7272 target_signal = connection ["target" ][0 ]["sig" ]
7373 connection_key = f"{ instance_name } ,{ portname } "
7474 # Find the target instance and port
75- target_instance_port = _find_target_instance_port (proto_dict , target_signal , instance_name )
75+ target_instance_port = _find_target_instance_port (
76+ proto_dict , target_signal , instance_name
77+ )
7678 if target_instance_port :
7779 connections [connection_key ] = target_instance_port
7880
7981 return connections
8082
8183
82- def _find_target_instance_port (proto_dict : ParsedProtoVLSIR ,
83- target_signal ,
84- current_instance_name ):
84+ def _find_target_instance_port (
85+ proto_dict : ParsedProtoVLSIR , target_signal , current_instance_name
86+ ):
8587 """
8688 Find the target instance and port of the target signal in the proto_dict.
8789 """
@@ -114,9 +116,9 @@ def _generate_top_level_connections(proto_dict: ParsedProtoVLSIR):
114116 top_level_connections = {}
115117
116118 # Iterate over the top-level module ports
117- for module in proto_dict .get (' modules' , []):
118- for port in module .get (' ports' , []):
119- port_signal = port [' signal' ]
119+ for module in proto_dict .get (" modules" , []):
120+ for port in module .get (" ports" , []):
121+ port_signal = port [" signal" ]
120122 connection = _find_port_connection (proto_dict , port_signal )
121123 if connection :
122124 top_level_connections [port_signal ] = connection
@@ -148,17 +150,16 @@ def _extract_instance_parameters(proto_dict: ParsedProtoVLSIR):
148150 for instance in module .get ("instances" , []):
149151 instance_name = instance ["name" ]
150152 instance_info = {
151-
152- 'component' : _extract_component_name (instance ),
153- 'info' : {},
154- 'settings' : {}
153+ "component" : _extract_component_name (instance ),
154+ "info" : {},
155+ "settings" : {},
155156 }
156157
157158 # Extract parameters into the settings
158- for parameter in instance .get (' parameters' , []):
159- param_name = parameter [' name' ]
160- param_value = _extract_parameter_value (parameter [' value' ])
161- instance_info [' settings' ][param_name ] = param_value
159+ for parameter in instance .get (" parameters" , []):
160+ param_name = parameter [" name" ]
161+ param_value = _extract_parameter_value (parameter [" value" ])
162+ instance_info [" settings" ][param_name ] = param_value
162163
163164 # Extract connections and add to settings
164165 instance_info ["settings" ]["ports" ] = {}
@@ -178,10 +179,10 @@ def _extract_component_name(instance):
178179 """
179180 external_modules = instance .get ("module" , [])
180181 if external_modules :
181- domain = external_modules [0 ].get (' external' , [{}])[0 ].get (' domain' , '' )
182- name = external_modules [0 ].get (' external' , [{}])[0 ].get (' name' , '' )
182+ domain = external_modules [0 ].get (" external" , [{}])[0 ].get (" domain" , "" )
183+ name = external_modules [0 ].get (" external" , [{}])[0 ].get (" name" , "" )
183184 return f"{ name } "
184- return ' unknown_component'
185+ return " unknown_component"
185186
186187
187188def _extract_parameter_value (value ):
@@ -208,13 +209,13 @@ def _generate_raw_netlist_dict_from_proto_dict(proto_dict: ParsedProtoVLSIR):
208209 raw_netlist_dict ["name" ] = proto_dict ["modules" ][0 ].get ("name" , "" )
209210
210211 # Generate instances information
211- raw_netlist_dict [' instances' ] = _extract_instance_parameters (proto_dict )
212+ raw_netlist_dict [" instances" ] = _extract_instance_parameters (proto_dict )
212213
213214 # Generate connections
214215 raw_netlist_dict ["connections" ] = _parse_connections (proto_dict )
215216
216217 # Generate top-level connections
217- raw_netlist_dict [' ports' ] = _generate_top_level_connections (proto_dict )
218+ raw_netlist_dict [" ports" ] = _generate_top_level_connections (proto_dict )
218219
219220 return raw_netlist_dict
220221
0 commit comments