Spaces:
Sleeping
Sleeping
| # | |
| # The Python Imaging Library. | |
| # $Id$ | |
| # | |
| # image enhancement classes | |
| # | |
| # For a background, see "Image Processing By Interpolation and | |
| # Extrapolation", Paul Haeberli and Douglas Voorhies. Available | |
| # at http://www.graficaobscura.com/interp/index.html | |
| # | |
| # History: | |
| # 1996-03-23 fl Created | |
| # 2009-06-16 fl Fixed mean calculation | |
| # | |
| # Copyright (c) Secret Labs AB 1997. | |
| # Copyright (c) Fredrik Lundh 1996. | |
| # | |
| # See the README file for information on usage and redistribution. | |
| # | |
| from __future__ import annotations | |
| from . import Image, ImageFilter, ImageStat | |
| class _Enhance: | |
| image: Image.Image | |
| degenerate: Image.Image | |
| def enhance(self, factor: float) -> Image.Image: | |
| """ | |
| Returns an enhanced image. | |
| :param factor: A floating point value controlling the enhancement. | |
| Factor 1.0 always returns a copy of the original image, | |
| lower factors mean less color (brightness, contrast, | |
| etc), and higher values more. There are no restrictions | |
| on this value. | |
| :rtype: :py:class:`~PIL.Image.Image` | |
| """ | |
| return Image.blend(self.degenerate, self.image, factor) | |
| class Color(_Enhance): | |
| """Adjust image color balance. | |
| This class can be used to adjust the colour balance of an image, in | |
| a manner similar to the controls on a colour TV set. An enhancement | |
| factor of 0.0 gives a black and white image. A factor of 1.0 gives | |
| the original image. | |
| """ | |
| def __init__(self, image: Image.Image) -> None: | |
| self.image = image | |
| self.intermediate_mode = "L" | |
| if "A" in image.getbands(): | |
| self.intermediate_mode = "LA" | |
| self.degenerate = image.convert(self.intermediate_mode).convert(image.mode) | |
| class Contrast(_Enhance): | |
| """Adjust image contrast. | |
| This class can be used to control the contrast of an image, similar | |
| to the contrast control on a TV set. An enhancement factor of 0.0 | |
| gives a solid gray image. A factor of 1.0 gives the original image. | |
| """ | |
| def __init__(self, image: Image.Image) -> None: | |
| self.image = image | |
| mean = int(ImageStat.Stat(image.convert("L")).mean[0] + 0.5) | |
| self.degenerate = Image.new("L", image.size, mean).convert(image.mode) | |
| if "A" in image.getbands(): | |
| self.degenerate.putalpha(image.getchannel("A")) | |
| class Brightness(_Enhance): | |
| """Adjust image brightness. | |
| This class can be used to control the brightness of an image. An | |
| enhancement factor of 0.0 gives a black image. A factor of 1.0 gives the | |
| original image. | |
| """ | |
| def __init__(self, image: Image.Image) -> None: | |
| self.image = image | |
| self.degenerate = Image.new(image.mode, image.size, 0) | |
| if "A" in image.getbands(): | |
| self.degenerate.putalpha(image.getchannel("A")) | |
| class Sharpness(_Enhance): | |
| """Adjust image sharpness. | |
| This class can be used to adjust the sharpness of an image. An | |
| enhancement factor of 0.0 gives a blurred image, a factor of 1.0 gives the | |
| original image, and a factor of 2.0 gives a sharpened image. | |
| """ | |
| def __init__(self, image: Image.Image) -> None: | |
| self.image = image | |
| self.degenerate = image.filter(ImageFilter.SMOOTH) | |
| if "A" in image.getbands(): | |
| self.degenerate.putalpha(image.getchannel("A")) | |