Improve content rendering

b_asic/architecture.py

@@ -8,7 +8,6 @@ from typing import Dict, Iterable, Iterator, List, Optional, Set, Tuple, Union,
 import matplotlib.pyplot as plt
 from graphviz import Digraph
 
-from b_asic.mpl_utils import FigureWrapper
 from b_asic.port import InputPort, OutputPort
 from b_asic.process import MemoryVariable, OperatorProcess, PlainMemoryVariable
 from b_asic.resources import ProcessCollection
@@ -192,16 +191,16 @@ class Resource(HardwareBlock):
         return self._assignment is not None
 
     @property
-    def content(self) -> FigureWrapper:
+    def content(self) -> plt.Figure:
         """
         Return a graphical representation of the content.
 
-        This is visible in enriched shells, but the object in itself has no further
-        meaning.
+        This is visible in enriched shells, but the object itself has no further
+        meaning (it is a Matplotlib Figure).
         """
         fig, ax = plt.subplots()
         self.plot_content(ax)
-        return FigureWrapper(fig)
+        return fig
 
 
 class ProcessingElement(Resource):

b_asic/mpl_utils.py

-"""
-B-ASIC utilities for Matplotlib integration.
-"""
-import io
-
-from matplotlib.figure import Figure
-
-
-class FigureWrapper:
-    """Simple wrapper to display a figure inline in enriched shells."""
-
-    def __init__(self, figure: Figure):
-        self._figure = figure
-
-    def _repr_svg_(self):
-        buffer = io.StringIO()
-        self._figure.savefig(buffer, format="svg")
-        return buffer.getvalue()

docs_sphinx/api/index.rst

@@ -8,7 +8,6 @@ API
     architecture.rst
     core_operations.rst
     graph_component.rst
-    mpl_utils.rst
     operation.rst
     port.rst
     process.rst

docs_sphinx/api/mpl_utils.rst

-********************
-``b_asic.mpl_utils``
-********************
-
-.. automodule:: b_asic.mpl_utils
-   :members:
-   :undoc-members: