127 lines
4.1 KiB
Python
127 lines
4.1 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
Python converter for converting png, jpeg and bmp to usable .bmp files which
|
|
can be used by the E-Paper display. Currently, any image will be converted
|
|
to pure white, pure black and pure red pixels, without any shades of grey,
|
|
just as the displays expects them to be.
|
|
|
|
Copyright by Ace-Laboratory
|
|
"""
|
|
|
|
import glob, os, errno
|
|
from PIL import Image
|
|
import PIL.ImageOps
|
|
#--------------only change the following two lines-----------------#
|
|
input_folder = '/home/pi/conversion/input/'
|
|
output_folder = '/home/pi/conversion/output/'
|
|
#-----------------no need to change anything below----------------#
|
|
|
|
os.chdir(input_folder)
|
|
|
|
""" Creating lists with filenames first"""
|
|
|
|
pngs = []
|
|
for files in glob.glob("*.png"):
|
|
print('found this png: '+ os.path.splitext(os.path.basename(files))[0])
|
|
pngs.append(os.path.splitext(os.path.basename(files))[0])
|
|
|
|
bmps = []
|
|
for files in glob.glob("*.bmp"):
|
|
print('found this bmp: '+files)
|
|
bmps.append(files)
|
|
|
|
jpegs= []
|
|
for files in glob.glob("*.jpeg"):
|
|
print('found this jpeg: '+ os.path.splitext(os.path.basename(files))[0])
|
|
jpegs.append(os.path.splitext(os.path.basename(files))[0])
|
|
|
|
print('converting the images now')
|
|
|
|
"""First, a function. It's required for reducing colours"""
|
|
fn = lambda x : 179 if x == 179 else (255 if x > 179 else 0)
|
|
|
|
""" Conversion is done in two steps:
|
|
1) Converting the file from .png or .jpeg .bmp to a greyscale .bmp
|
|
2) Inverting the colours and rotating the image 90° anticlockwise
|
|
"""
|
|
|
|
""" Step 1 conversion for png files"""
|
|
if not os.path.exists(input_folder+'pngs/'):
|
|
print('temporary png folder does not exist, creating now...')
|
|
os.makedirs(input_folder+'pngs/')
|
|
|
|
print("attempting to convert pngs now...")
|
|
for files in pngs:
|
|
png=Image.open(input_folder+files+".png")
|
|
png.load()
|
|
background = Image.new("RGB", png.size, (255, 255, 255)) #removing alpha channel
|
|
background.paste(png, mask=png.split()[3]) #removing alpha channel
|
|
background.convert('L').point(fn).save((input_folder+'pngs/'+files+'.bmp'), 'BMP', quality=90)
|
|
|
|
""" Step 2 conversion for png files"""
|
|
png_bmp = []
|
|
os.chdir(input_folder+'pngs/')
|
|
for files in glob.glob("*.bmp"):
|
|
png_bmp.append(files)
|
|
for files in png_bmp:
|
|
(PIL.ImageOps.invert((Image.open(input_folder+'pngs/'+files).rotate(-90, expand=True)).convert('L'))).save(output_folder+files)
|
|
print("All png files have been converted.")
|
|
|
|
|
|
os.chdir(input_folder)
|
|
|
|
""" Step 1 conversion for jpeg files"""
|
|
if not os.path.exists(input_folder+'jpegs/'):
|
|
print('temporary jpeg folder does not exist, creating now...')
|
|
os.makedirs(input_folder+'jpegs/')
|
|
|
|
print("attempting to convert jpeg now...")
|
|
|
|
for files in jpegs:
|
|
(Image.open(input_folder+files+".jpeg").convert('L').point(fn).save((input_folder+'jpegs/'+files+".bmp"), 'BMP', quality=90))
|
|
|
|
""" Step 2 conversion for jpeg files"""
|
|
|
|
jpeg_bmp = []
|
|
os.chdir(input_folder+'jpegs/')
|
|
for files in glob.glob("*.bmp"):
|
|
print('These :'+files)
|
|
jpeg_bmp.append(files)
|
|
for files in jpeg_bmp:
|
|
(PIL.ImageOps.invert((Image.open(input_folder+'jpegs/'+files).rotate(-90, expand=True)).convert('L'))).save(output_folder+files)
|
|
|
|
|
|
""" Step 1 & 2 conversion for bmp files"""
|
|
for files in bmps:
|
|
PIL.ImageOps.invert(Image.open(input_folder+files).convert('L').point(fn)).save((output_folder+files), 'BMP', quality=90)
|
|
#(PIL.ImageOps.invert((Image.open(input_folder+files).rotate(-90, expand=True)).convert('L'))).save(output_folder+files)
|
|
|
|
print('All done')
|
|
|
|
"""
|
|
os.chdir(input_folder)
|
|
bmps= []
|
|
for files in glob.glob("*.bmp"):
|
|
print('found this bmp :'+os.path.splitext(os.path.basename(files))[0])
|
|
bmps.append(os.path.splitext(os.path.basename(files))[0])
|
|
|
|
print('converting...')
|
|
|
|
# 255 = white
|
|
# 179 = gray (which turns red on the display)
|
|
# 0 = black
|
|
#min_thresh = 175
|
|
#max_thresh = 180
|
|
#fn = lambda x : 255 if x >= max_thresh else (0 if x <= min_thresh else 179)
|
|
fn = lambda x : 255 if x > 179 else (0 if x < 178 else 179)
|
|
|
|
# BMP to BMP conversion -> in testing
|
|
for files in bmps:
|
|
(Image.open(input_folder+files+".bmp")).point(fn).save(output_folder+files+".bmp")
|
|
# (Image.open(input_folder+files+".jpeg").convert('L').save(output_folder+files+".bmp"))
|
|
|
|
print('All done')
|
|
"""
|