Spaces:
Running
Running
| # | |
| # The Python Imaging Library. | |
| # $Id$ | |
| # | |
| # a Tk display interface | |
| # | |
| # History: | |
| # 96-04-08 fl Created | |
| # 96-09-06 fl Added getimage method | |
| # 96-11-01 fl Rewritten, removed image attribute and crop method | |
| # 97-05-09 fl Use PyImagingPaste method instead of image type | |
| # 97-05-12 fl Minor tweaks to match the IFUNC95 interface | |
| # 97-05-17 fl Support the "pilbitmap" booster patch | |
| # 97-06-05 fl Added file= and data= argument to image constructors | |
| # 98-03-09 fl Added width and height methods to Image classes | |
| # 98-07-02 fl Use default mode for "P" images without palette attribute | |
| # 98-07-02 fl Explicitly destroy Tkinter image objects | |
| # 99-07-24 fl Support multiple Tk interpreters (from Greg Couch) | |
| # 99-07-26 fl Automatically hook into Tkinter (if possible) | |
| # 99-08-15 fl Hook uses _imagingtk instead of _imaging | |
| # | |
| # Copyright (c) 1997-1999 by Secret Labs AB | |
| # Copyright (c) 1996-1997 by Fredrik Lundh | |
| # | |
| # See the README file for information on usage and redistribution. | |
| # | |
| from __future__ import annotations | |
| import tkinter | |
| from io import BytesIO | |
| from . import Image | |
| # -------------------------------------------------------------------- | |
| # Check for Tkinter interface hooks | |
| _pilbitmap_ok = None | |
| def _pilbitmap_check(): | |
| global _pilbitmap_ok | |
| if _pilbitmap_ok is None: | |
| try: | |
| im = Image.new("1", (1, 1)) | |
| tkinter.BitmapImage(data=f"PIL:{im.im.id}") | |
| _pilbitmap_ok = 1 | |
| except tkinter.TclError: | |
| _pilbitmap_ok = 0 | |
| return _pilbitmap_ok | |
| def _get_image_from_kw(kw): | |
| source = None | |
| if "file" in kw: | |
| source = kw.pop("file") | |
| elif "data" in kw: | |
| source = BytesIO(kw.pop("data")) | |
| if source: | |
| return Image.open(source) | |
| def _pyimagingtkcall(command, photo, id): | |
| tk = photo.tk | |
| try: | |
| tk.call(command, photo, id) | |
| except tkinter.TclError: | |
| # activate Tkinter hook | |
| # may raise an error if it cannot attach to Tkinter | |
| from . import _imagingtk | |
| _imagingtk.tkinit(tk.interpaddr()) | |
| tk.call(command, photo, id) | |
| # -------------------------------------------------------------------- | |
| # PhotoImage | |
| class PhotoImage: | |
| """ | |
| A Tkinter-compatible photo image. This can be used | |
| everywhere Tkinter expects an image object. If the image is an RGBA | |
| image, pixels having alpha 0 are treated as transparent. | |
| The constructor takes either a PIL image, or a mode and a size. | |
| Alternatively, you can use the ``file`` or ``data`` options to initialize | |
| the photo image object. | |
| :param image: Either a PIL image, or a mode string. If a mode string is | |
| used, a size must also be given. | |
| :param size: If the first argument is a mode string, this defines the size | |
| of the image. | |
| :keyword file: A filename to load the image from (using | |
| ``Image.open(file)``). | |
| :keyword data: An 8-bit string containing image data (as loaded from an | |
| image file). | |
| """ | |
| def __init__(self, image=None, size=None, **kw): | |
| # Tk compatibility: file or data | |
| if image is None: | |
| image = _get_image_from_kw(kw) | |
| if hasattr(image, "mode") and hasattr(image, "size"): | |
| # got an image instead of a mode | |
| mode = image.mode | |
| if mode == "P": | |
| # palette mapped data | |
| image.apply_transparency() | |
| image.load() | |
| try: | |
| mode = image.palette.mode | |
| except AttributeError: | |
| mode = "RGB" # default | |
| size = image.size | |
| kw["width"], kw["height"] = size | |
| else: | |
| mode = image | |
| image = None | |
| if mode not in ["1", "L", "RGB", "RGBA"]: | |
| mode = Image.getmodebase(mode) | |
| self.__mode = mode | |
| self.__size = size | |
| self.__photo = tkinter.PhotoImage(**kw) | |
| self.tk = self.__photo.tk | |
| if image: | |
| self.paste(image) | |
| def __del__(self): | |
| name = self.__photo.name | |
| self.__photo.name = None | |
| try: | |
| self.__photo.tk.call("image", "delete", name) | |
| except Exception: | |
| pass # ignore internal errors | |
| def __str__(self): | |
| """ | |
| Get the Tkinter photo image identifier. This method is automatically | |
| called by Tkinter whenever a PhotoImage object is passed to a Tkinter | |
| method. | |
| :return: A Tkinter photo image identifier (a string). | |
| """ | |
| return str(self.__photo) | |
| def width(self): | |
| """ | |
| Get the width of the image. | |
| :return: The width, in pixels. | |
| """ | |
| return self.__size[0] | |
| def height(self): | |
| """ | |
| Get the height of the image. | |
| :return: The height, in pixels. | |
| """ | |
| return self.__size[1] | |
| def paste(self, im): | |
| """ | |
| Paste a PIL image into the photo image. Note that this can | |
| be very slow if the photo image is displayed. | |
| :param im: A PIL image. The size must match the target region. If the | |
| mode does not match, the image is converted to the mode of | |
| the bitmap image. | |
| """ | |
| # convert to blittable | |
| im.load() | |
| image = im.im | |
| if image.isblock() and im.mode == self.__mode: | |
| block = image | |
| else: | |
| block = image.new_block(self.__mode, im.size) | |
| image.convert2(block, image) # convert directly between buffers | |
| _pyimagingtkcall("PyImagingPhoto", self.__photo, block.id) | |
| # -------------------------------------------------------------------- | |
| # BitmapImage | |
| class BitmapImage: | |
| """ | |
| A Tkinter-compatible bitmap image. This can be used everywhere Tkinter | |
| expects an image object. | |
| The given image must have mode "1". Pixels having value 0 are treated as | |
| transparent. Options, if any, are passed on to Tkinter. The most commonly | |
| used option is ``foreground``, which is used to specify the color for the | |
| non-transparent parts. See the Tkinter documentation for information on | |
| how to specify colours. | |
| :param image: A PIL image. | |
| """ | |
| def __init__(self, image=None, **kw): | |
| # Tk compatibility: file or data | |
| if image is None: | |
| image = _get_image_from_kw(kw) | |
| self.__mode = image.mode | |
| self.__size = image.size | |
| if _pilbitmap_check(): | |
| # fast way (requires the pilbitmap booster patch) | |
| image.load() | |
| kw["data"] = f"PIL:{image.im.id}" | |
| self.__im = image # must keep a reference | |
| else: | |
| # slow but safe way | |
| kw["data"] = image.tobitmap() | |
| self.__photo = tkinter.BitmapImage(**kw) | |
| def __del__(self): | |
| name = self.__photo.name | |
| self.__photo.name = None | |
| try: | |
| self.__photo.tk.call("image", "delete", name) | |
| except Exception: | |
| pass # ignore internal errors | |
| def width(self): | |
| """ | |
| Get the width of the image. | |
| :return: The width, in pixels. | |
| """ | |
| return self.__size[0] | |
| def height(self): | |
| """ | |
| Get the height of the image. | |
| :return: The height, in pixels. | |
| """ | |
| return self.__size[1] | |
| def __str__(self): | |
| """ | |
| Get the Tkinter bitmap image identifier. This method is automatically | |
| called by Tkinter whenever a BitmapImage object is passed to a Tkinter | |
| method. | |
| :return: A Tkinter bitmap image identifier (a string). | |
| """ | |
| return str(self.__photo) | |
| def getimage(photo): | |
| """Copies the contents of a PhotoImage to a PIL image memory.""" | |
| im = Image.new("RGBA", (photo.width(), photo.height())) | |
| block = im.im | |
| _pyimagingtkcall("PyImagingPhotoGet", photo, block.id) | |
| return im | |
| def _show(image, title): | |
| """Helper for the Image.show method.""" | |
| class UI(tkinter.Label): | |
| def __init__(self, master, im): | |
| if im.mode == "1": | |
| self.image = BitmapImage(im, foreground="white", master=master) | |
| else: | |
| self.image = PhotoImage(im, master=master) | |
| super().__init__(master, image=self.image, bg="black", bd=0) | |
| if not tkinter._default_root: | |
| msg = "tkinter not initialized" | |
| raise OSError(msg) | |
| top = tkinter.Toplevel() | |
| if title: | |
| top.title(title) | |
| UI(top, image).pack() | |