1
mirror of git://git.acid.vegas/scroll.git synced 2024-11-21 23:56:39 +00:00

Added invert color effect

This commit is contained in:
Dionysus 2023-06-28 15:23:47 -04:00
parent a21e7e2f6a
commit 8fbc870f58
Signed by: acidvegas
GPG Key ID: EF4B922DB85DC9DE

View File

@ -14,7 +14,7 @@ pull request: https://github.com/ircart/scroll/pull/3
import io import io
try: try:
from PIL import Image, ImageEnhance from PIL import Image, ImageEnhance, ImageOps
except ImportError: except ImportError:
raise SystemExit('missing required \'pillow\' library (https://pypi.org/project/pillow/)') raise SystemExit('missing required \'pillow\' library (https://pypi.org/project/pillow/)')
@ -51,13 +51,15 @@ def convert(data, max_line_len, img_width=80, palette='RGB99', enhance=False, ef
raise Exception('invalid palette option') raise Exception('invalid palette option')
palette = palettes[palette] palette = palettes[palette]
image = Image.open(io.BytesIO(data)) image = Image.open(io.BytesIO(data))
del data
if enhance: if enhance:
image = ImageEnhance.Contrast(image) image = ImageEnhance.Contrast(image)
if effect == 'grey': if effect == 'greyscale':
image = image.convert("L") image = image.convert("L")
elif effect == 'black': elif effect == 'blackwhite':
image = image.convert("1") image = image.convert("1")
del data elif effect == 'invert':
image = ImageOps.invert(image)
return convert_image(image, max_line_len, img_width, palette) return convert_image(image, max_line_len, img_width, palette)
def convert_image(image, max_line_len, img_width, palette): def convert_image(image, max_line_len, img_width, palette):