← Back to branch summary

~ubuntu-branches/debian/squeeze/python-imaging/squeeze

« back to all changes in this revision

Viewing changes to PIL/ImageShow.py

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-11-20 19:22:59 UTC
  • mfrom: (1.1.3 upstream)
  • Revision ID: james.westby@ubuntu.com-20091120192259-n3iy0f17n5akogom
Tags: 1.1.7-1
New upstream version.

Show diffs side-by-side

added added

removed removed

Lines of Context:
PIL/ImageShow.py
 
1
#
 
2
# The Python Imaging Library.
 
3
# $Id$
 
4
#
 
5
# im.show() drivers
 
6
#
 
7
# History:
 
8
# 2008-04-06 fl   Created
 
9
#
 
10
# Copyright (c) Secret Labs AB 2008.
 
11
#
 
12
# See the README file for information on usage and redistribution.
 
13
#
 
14
 
 
15
import Image
 
16
import os, sys
 
17
 
 
18
_viewers = []
 
19
 
 
20
def register(viewer, order=1):
 
21
    try:
 
22
        if issubclass(viewer, Viewer):
 
23
            viewer = viewer()
 
24
    except TypeError:
 
25
        pass # raised if viewer wasn't a class
 
26
    if order > 0:
 
27
        _viewers.append(viewer)
 
28
    elif order < 0:
 
29
        _viewers.insert(0, viewer)
 
30
 
 
31
##
 
32
# Displays a given image.
 
33
#
 
34
# @param image An image object.
 
35
# @param title Optional title.  Not all viewers can display the title.
 
36
# @param **options Additional viewer options.
 
37
# @return True if a suitable viewer was found, false otherwise.
 
38
 
 
39
def show(image, title=None, **options):
 
40
    for viewer in _viewers:
 
41
        if viewer.show(image, title=title, **options):
 
42
            return 1
 
43
    return 0
 
44
 
 
45
##
 
46
# Base class for viewers.
 
47
 
 
48
class Viewer:
 
49
 
 
50
    # main api
 
51
 
 
52
    def show(self, image, **options):
 
53
 
 
54
        # save temporary image to disk
 
55
        if image.mode[:4] == "I;16":
 
56
            # @PIL88 @PIL101
 
57
            # "I;16" isn't an 'official' mode, but we still want to
 
58
            # provide a simple way to show 16-bit images.
 
59
            base = "L"
 
60
            # FIXME: auto-contrast if max() > 255?
 
61
        else:
 
62
            base = Image.getmodebase(image.mode)
 
63
        if base != image.mode and image.mode != "1":
 
64
            image = image.convert(base)
 
65
 
 
66
        self.show_image(image, **options)
 
67
 
 
68
    # hook methods
 
69
 
 
70
    format = None
 
71
 
 
72
    def get_format(self, image):
 
73
        # return format name, or None to save as PGM/PPM
 
74
        return self.format
 
75
 
 
76
    def get_command(self, file, **options):
 
77
        raise NotImplementedError
 
78
 
 
79
    def save_image(self, image):
 
80
        # save to temporary file, and return filename
 
81
        return image._dump(format=self.get_format(image))
 
82
 
 
83
    def show_image(self, image, **options):
 
84
        # display given image
 
85
        return self.show_file(self.save_image(image), **options)
 
86
 
 
87
    def show_file(self, file, **options):
 
88
        # display given file
 
89
        os.system(self.get_command(file, **options))
 
90
        return 1
 
91
 
 
92
# --------------------------------------------------------------------
 
93
 
 
94
if sys.platform == "win32":
 
95
 
 
96
    class WindowsViewer(Viewer):
 
97
        format = "BMP"
 
98
        def get_command(self, file, **options):
 
99
            return "start /wait %s && del /f %s" % (file, file)
 
100
 
 
101
    register(WindowsViewer)
 
102
 
 
103
elif sys.platform == "darwin":
 
104
 
 
105
    class MacViewer(Viewer):
 
106
        format = "BMP"
 
107
        def get_command(self, file, **options):
 
108
            # on darwin open returns immediately resulting in the temp
 
109
            # file removal while app is opening
 
110
            command = "open -a /Applications/Preview.app"
 
111
            command = "(%s %s; sleep 20; rm -f %s)&" % (command, file, file)
 
112
            return command
 
113
 
 
114
    register(MacViewer)
 
115
 
 
116
else:
 
117
 
 
118
    # unixoids
 
119
 
 
120
    def which(executable):
 
121
        path = os.environ.get("PATH")
 
122
        if not path:
 
123
            return None
 
124
        for dirname in path.split(os.pathsep):
 
125
            filename = os.path.join(dirname, executable)
 
126
            if os.path.isfile(filename):
 
127
                # FIXME: make sure it's executable
 
128
                return filename
 
129
        return None
 
130
 
 
131
    class UnixViewer(Viewer):
 
132
        def show_file(self, file, **options):
 
133
            command, executable = self.get_command_ex(file, **options)
 
134
            command = "(%s %s; rm -f %s)&" % (command, file, file)
 
135
            os.system(command)
 
136
            return 1
 
137
 
 
138
    # implementations
 
139
 
 
140
    class DisplayViewer(UnixViewer):
 
141
        def get_command_ex(self, file, **options):
 
142
            command = executable = "display"
 
143
            return command, executable
 
144
 
 
145
    if which("display"):
 
146
        register(DisplayViewer)
 
147
 
 
148
    class XVViewer(UnixViewer):
 
149
        def get_command_ex(self, file, title=None, **options):
 
150
            # note: xv is pretty outdated.  most modern systems have
 
151
            # imagemagick's display command instead.
 
152
            command = executable = "xv"
 
153
            if title:
 
154
                # FIXME: do full escaping
 
155
                command = command + " -name \"%s\"" % title
 
156
            return command, executable
 
157
 
 
158
    if which("xv"):
 
159
        register(XVViewer)
 
160
 
 
161
if __name__ == "__main__":
 
162
    # usage: python ImageShow.py imagefile [title]
 
163
    print show(Image.open(sys.argv[1]), *sys.argv[2:])