| @@ -1,16 +1,17 @@ | | | @@ -1,16 +1,17 @@ |
1 | $NetBSD: patch-pyrtl_visualization.py,v 1.1 2022/05/19 15:59:04 ryoon Exp $ | | 1 | $NetBSD: patch-pyrtl_visualization.py,v 1.2 2022/05/21 15:01:00 ryoon Exp $ |
2 | | | 2 | |
3 | * Support SVG output with py-graphviz 0.19 or later too. | | 3 | * Support SVG output with py-graphviz 0.19 or later too. |
| | | 4 | https://github.com/UCSBarchlab/PyRTL/pull/416/ |
4 | | | 5 | |
5 | --- pyrtl/visualization.py.orig 2021-09-02 19:11:31.000000000 +0000 | | 6 | --- pyrtl/visualization.py.orig 2021-09-02 19:11:31.000000000 +0000 |
6 | +++ pyrtl/visualization.py | | 7 | +++ pyrtl/visualization.py |
7 | @@ -453,8 +453,16 @@ def block_to_svg(block=None, split_state | | 8 | @@ -453,8 +453,16 @@ def block_to_svg(block=None, split_state |
8 | """ | | 9 | """ |
9 | try: | | 10 | try: |
10 | from graphviz import Source | | 11 | from graphviz import Source |
11 | - return Source(block_to_graphviz_string(block, split_state=split_state, | | 12 | - return Source(block_to_graphviz_string(block, split_state=split_state, |
12 | - maintain_arg_order=maintain_arg_order))._repr_svg_() | | 13 | - maintain_arg_order=maintain_arg_order))._repr_svg_() |
13 | + src = Source(block_to_graphviz_string(block, split_state=split_state, | | 14 | + src = Source(block_to_graphviz_string(block, split_state=split_state, |
14 | + maintain_arg_order=maintain_arg_order)) | | 15 | + maintain_arg_order=maintain_arg_order)) |
15 | + try: | | 16 | + try: |
16 | + svg = src._repr_image_svg_xml() | | 17 | + svg = src._repr_image_svg_xml() |