add fig2image convenience utility function

pyllusion/image/__init__.py

@@ -2,6 +2,7 @@
 Pyllusion submodule.
 """
 
+from .fig2image import fig2image
 from .image_blob import image_blob, image_blobs
 from .image_circle import image_circle, image_circles
 from .image_line import image_line
@@ -24,4 +25,5 @@
     "image_mosaic",
     "image_scramble",
     "rescale",
+    "fig2image"
 ]

pyllusion/image/fig2image.py

@@ -0,0 +1,36 @@
+import io
+
+import PIL.Image
+
+
+def fig2img(fig):
+    """Matplotlib Figure to PIL Image
+
+    Convert a Matplotlib figure to a PIL Image
+
+    Parameters
+    ----------
+    fig : plt.figure
+        Matplotlib figure.
+
+    Returns
+    ----------
+    list
+        The rescaled values.
+
+
+    Examples
+    ----------
+    >>> import pyllusion
+    >>> import matplotlib.pyplot as plt
+    >>>
+    >>> plt.plot([1, 2, 3, 4, 5])
+    >>> fig = plt.gcf()
+    >>> fig2img(fig)
+
+    """
+    buffer = io.BytesIO()
+    fig.savefig(buffer)
+    buffer.seek(0)
+    img = PIL.Image.open(buffer)
+    return img