(engine_json_fname: str)
| 32 | |
| 33 | |
| 34 | def draw_engine(engine_json_fname: str): |
| 35 | graphviz_is_installed = shutil.which("dot") is not None |
| 36 | if not graphviz_is_installed: |
| 37 | print("graphviz is required but it is not installed.\n") |
| 38 | print("To install on Ubuntu:") |
| 39 | print("sudo apt --yes install graphviz") |
| 40 | exit() |
| 41 | |
| 42 | plan = EnginePlan(engine_json_fname) |
| 43 | formatter = layer_type_formatter |
| 44 | display_regions = True |
| 45 | expand_layer_details = False |
| 46 | |
| 47 | graph = to_dot(plan, formatter, |
| 48 | display_regions=display_regions, |
| 49 | expand_layer_details=expand_layer_details) |
| 50 | render_dot(graph, engine_json_fname, 'svg') |
| 51 | |
| 52 | |
| 53 | if __name__ == "__main__": |
no test coverage detected