mirror of https://github.com/YosysHQ/yosys.git
35 lines
881 B
Python
35 lines
881 B
Python
from pyosys import libyosys as ys
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
__file_dir__ = Path(__file__).absolute().parent
|
|
|
|
class CellStatsPass(ys.Pass):
|
|
def __init__(self):
|
|
super().__init__(
|
|
"cell_stats",
|
|
"dumps cell statistics in JSON format"
|
|
)
|
|
|
|
def execute(self, args, design):
|
|
ys.log_header(design, "Dumping cell stats\n")
|
|
ys.log_push()
|
|
cell_stats = {}
|
|
for module in design.all_selected_whole_modules():
|
|
for cell in module.selected_cells():
|
|
if cell.type.str() in cell_stats:
|
|
cell_stats[cell.type.str()] += 1
|
|
else:
|
|
cell_stats[cell.type.str()] = 1
|
|
ys.log(json.dumps(cell_stats))
|
|
ys.log_pop()
|
|
|
|
p = CellStatsPass() # registration
|
|
|
|
design = ys.Design()
|
|
ys.run_pass(f"read_verilog {__file_dir__.parent / 'simple' / 'fiedler-cooley.v'}", design)
|
|
ys.run_pass("prep", design)
|
|
ys.run_pass("opt -full", design)
|
|
ys.run_pass("cell_stats", design)
|