@@ -63,16 +63,17 @@ def xtract_cells_decl(dir, fout):
6363 parser .add_argument ('gowin_dir' , nargs = '?' , default = '/opt/gowin/' )
6464 args = parser .parse_args ()
6565
66- dirs = [
67- os .path .join (args .gowin_dir , 'IDE/simlib/gw1n/' ),
68- os .path .join (args .gowin_dir , 'IDE/simlib/gw2a/' ),
69- os .path .join (args .gowin_dir , 'IDE/simlib/gw5a/' ),
70- ]
66+ families = {
67+ 'gw1n' : os .path .join (args .gowin_dir , 'IDE/simlib/gw1n/' ),
68+ 'gw2a' : os .path .join (args .gowin_dir , 'IDE/simlib/gw2a/' ),
69+ 'gw5a' : os .path .join (args .gowin_dir , 'IDE/simlib/gw5a/' ),
70+ }
7171
72- with open ('cells_xtra.v' , 'w' ) as fout :
73- fout .write ('// Created by cells_xtra.py\n ' )
74- fout .write ('\n ' )
75- for dir in dirs :
76- if not os .path .isdir (dir ):
77- print (f'{ dir } is not a directory' )
78- xtract_cells_decl (dir , fout )
72+ for family , dir in families .items ():
73+ if not os .path .isdir (dir ):
74+ print (f'{ dir } is not a directory' )
75+ else :
76+ with open (f'cells_xtra_{ family } .v' , 'w' ) as fout :
77+ fout .write ('// Created by cells_xtra.py\n ' )
78+ fout .write ('\n ' )
79+ xtract_cells_decl (dir , fout )
0 commit comments