#
# The Python Imaging Library.
# $Id$
#
# im.show() drivers
#
# History:
# 2008-04-06 fl   Created
#
# Copyright (c) Secret Labs AB 2008.
#
# See the README file for information on usage and redistribution.
#

import Image
import os, sys

_viewers = []

def register(viewer, order=1):
    try:
        if issubclass(viewer, Viewer):
            viewer = viewer()
    except TypeError:
        pass # raised if viewer wasn't a class
    if order > 0:
        _viewers.append(viewer)
    elif order < 0:
        _viewers.insert(0, viewer)

def which(executable):
    path = os.environ.get("PATH")
    if not path:
        return None
    for dirname in path.split(os.pathsep):
        filename = os.path.join(dirname, executable)
        if os.path.isfile(filename):
            # FIXME: make sure it's executable
            return filename
    return None

##
# Displays a given image.
#
# @param image An image object.
# @param title Optional title.  Not all viewers can display the title.
# @param **options Additional viewer options.
# @return True if a suitable viewer was found, false otherwise.

def show(image, title=None, **options):
    for viewer in _viewers:
        if viewer.show(image, title=title, **options):
            return 1
    return 0

##
# Base class for viewers.

class Viewer(object):

    # main api

    def show(self, image, **options):

        # convert to 'robust' mode (RGB, L, or 1)
        if image.mode == "I;16":
            # @PIL88 @PIL101
            # "I;16" isn't an 'official' mode, but we still want to
            # provide a simple way to show 16-bit images.
            base = "L"
        else:
            base = Image.getmodebase(image.mode)
        if base != image.mode and image.mode != "1":
            image = image.convert(base)

        return self.show_image(image, **options)

    # hook methods

    format = None

    def get_format(self, image):
        # return format name, or None to save as PGM/PPM
        return self.format

    def get_command(self, file, **options):
        raise NotImplementedError

    def save_image(self, image):
        # save to temporary file, and return filename
        return image._dump(format=self.get_format(image))

    def show_image(self, image, **options):
        # display given image
        return self.show_file(self.save_image(image), **options)

    def show_file(self, file, **options):
        # display given file
        os.system(self.get_command(file, **options))
        return 1

# --------------------------------------------------------------------

if sys.platform == "win32":

    class WindowsViewer(Viewer):
        format = "BMP"
        def get_command(self, file, **options):
            return "start /wait %s && del /f %s" % (file, file)

    register(WindowsViewer)

elif sys.platform == "darwin":

    class MacViewer(Viewer):
        # FIXME: pick better format for "P" images? (use get_format)
        format = "JPEG"
        def get_command(self, file, **options):
            # on darwin open returns immediately resulting in the temp
            # file removal while app is opening
            command = "open -a /Applications/Preview.app"
            command = "(%s %s; sleep 20; rm -f %s)&" % (command, file, file)
            return command

    register(MacViewer)

else:

    # unixoids

    class UnixViewer(Viewer):
        executable = None # must override
        def show_file(self, file, **options):
            command = self.get_command(file, **options)
            # FIXME: do runtime which check?
            command = "(%s %s; rm -f %s)&" % (command, file, file)
            os.system(command)
            return 1 # assume success if we get this far

    def register_if_present(viewer):
        if which(viewer.executable):
            register(viewer)

    # implementations

    class DisplayViewer(UnixViewer):
        executable = "display"
        def get_command(self, file, title=None, **options):
            command = self.executable + " -immutable"
            if title:
                # FIXME: do full escaping
                command = command + " -title \"%s\"" % title
            return command

    register_if_present(DisplayViewer)

    class XVViewer(UnixViewer):
        executable = "xv"
        def get_command(self, file, title=None, **options):
            # note: xv is pretty outdated.  most modern systems have
            # imagemagick's display command instead (see above).
            command = self.executable
            if title:
                # FIXME: do full escaping
                command = command + " -name \"%s\"" % title
            return command

    register_if_present(XVViewer)

if __name__ == "__main__":
    # usage: python ImageShow.py imagefile [title]
    print show(Image.open(sys.argv[1]), *sys.argv[2:])