Source code for openalea.phenomenal.display.image

# -*- python -*-
#
#       Copyright INRIA - CIRAD - INRA
#
#       Distributed under the Cecill-C License.
#       See accompanying file LICENSE.txt or copy at
#           http://www.cecill.info/licences/Licence_CeCILL-C_V1-en.html
#
# ==============================================================================
"""
Module to display image and binarization result
"""

# ==============================================================================
import math
import numpy
from PIL import Image
from matplotlib import pyplot as plt
# ==============================================================================

__all__ = ["show_image", "show_images"]

# ==============================================================================




def show_image(image, title=""):
    plt.title(title)
    plt.axis('off')
    plt.imshow(Image.fromarray(image).convert("RGB"))
    plt.show()





def show_images(images, title=None):

    if isinstance(title, str):
        plt.suptitle(title)
        title = None

    nb_col = min(len(images), 4)
    nb_row = int(math.ceil(len(images) / float(nb_col)))

    for i, image in enumerate(images, 1):
        image = numpy.array(numpy.round(image), dtype=numpy.uint8)

        ax = plt.subplot(nb_row, nb_col, i)
        ax.axis("off")

        if title is not None:
            ax.set_title(title[i - 1])

        ax.imshow(Image.fromarray(image).convert("RGB"))

    plt.show()