135 lines
4.7 KiB
Python
135 lines
4.7 KiB
Python
import os
|
|
import random
|
|
from tkinter import *
|
|
from PIL import ImageTk, Image, ImageOps
|
|
from util import Util
|
|
|
|
|
|
class ImageWindow:
|
|
def __init__(self, app):
|
|
self.app = app
|
|
self.window = Toplevel(app.root)
|
|
self.window.title("Image")
|
|
self.window.geometry("1280x1024")
|
|
self.option = {
|
|
"bw": False,
|
|
"mirror": False
|
|
}
|
|
self.images = []
|
|
self.current_image = None
|
|
self.image_label = None
|
|
toolbar = Frame(self.window, bd=1, relief=RAISED)
|
|
toolbar.pack(side=BOTTOM, fill=X)
|
|
self.timer = 0
|
|
self.timer_label = None
|
|
self.timer_check = None
|
|
|
|
next_button = Button(toolbar, relief=FLAT, compound=LEFT, text="next ➡️", command=self.next_image)
|
|
next_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
self.bw_button = Button(toolbar, relief=FLAT, compound=LEFT, text="black&white",
|
|
command=self.toggle_black_white)
|
|
self.bw_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
self.mirror_button = Button(toolbar, relief=FLAT, compound=LEFT, command=self.toggle_mirror, text="mirror")
|
|
self.mirror_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
self.open_folder_button = Button(toolbar, relief=FLAT, compound=LEFT, command=self.open_folder
|
|
, text="open folder")
|
|
self.open_folder_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
self.timer_button = Button(toolbar, relief=FLAT, compound=LEFT, command=self.toggle_timer, text="timer")
|
|
self.timer_button.pack(side=LEFT, padx=0, pady=0)
|
|
|
|
self.image_label = Label(self.window)
|
|
self.image_label.pack(side=TOP, fill=BOTH, expand=1)
|
|
|
|
self.timer_label = Label(toolbar, text=self.timer)
|
|
self.timer_label.pack(side=RIGHT, ipadx=20)
|
|
|
|
def update_timer(self, current):
|
|
current -= 1
|
|
if current > 0:
|
|
self.timer_label.configure(text=Util.format_seconds(current))
|
|
self.timer_check = self.window.after(1000, self.update_timer, current)
|
|
else:
|
|
self.window.after_cancel(self.timer_check)
|
|
if len(self.images) == 0:
|
|
self.window.destroy()
|
|
self.window.update()
|
|
return
|
|
self.next_image()
|
|
|
|
def next_image(self):
|
|
if self.timer_check is not None:
|
|
self.window.after_cancel(self.timer_check)
|
|
self.display_new_image()
|
|
self.display_new_timer()
|
|
|
|
def toggle_black_white(self):
|
|
self.option["bw"] = not self.option["bw"]
|
|
self.bw_button.config(bg="blue" if self.option["bw"] else "grey85")
|
|
|
|
def toggle_mirror(self):
|
|
self.option["mirror"] = not self.option["mirror"]
|
|
self.mirror_button.config(bg="blue" if self.option["mirror"] else "grey85")
|
|
|
|
def toggle_timer(self):
|
|
try:
|
|
self.timer_label.pack_info()
|
|
self.timer_label.pack_forget()
|
|
except TclError:
|
|
self.timer_label.pack(side=RIGHT, ipadx=20)
|
|
|
|
def open_folder(self):
|
|
os.startfile(self.app.selected_folder)
|
|
|
|
def lets_draw(self, images, timer):
|
|
random.shuffle(images)
|
|
self.images = images
|
|
self.timer = timer
|
|
|
|
# canvas = Canvas(self.window)
|
|
# canvas.pack(fill=BOTH, expand=1)
|
|
|
|
self.next_image()
|
|
|
|
def display_new_timer(self):
|
|
self.timer_label.configure(text=self.timer)
|
|
self.timer_check = self.window.after(1000, self.update_timer, self.timer)
|
|
|
|
def display_new_image(self):
|
|
image_path = self.images[0]
|
|
self.images.pop(0)
|
|
self.current_image = Image.open(image_path)
|
|
|
|
self.resize_image()
|
|
|
|
if self.option["bw"]:
|
|
self.current_image = self.current_image.convert("L")
|
|
|
|
if self.option["mirror"]:
|
|
self.current_image = ImageOps.mirror(self.current_image)
|
|
|
|
image_to_display = ImageTk.PhotoImage(self.current_image)
|
|
|
|
self.image_label.configure(image=image_to_display)
|
|
self.image_label.bind('<Configure>', lambda event: self.resize_image())
|
|
self.image_label.image = image_to_display
|
|
|
|
def resize_image(self):
|
|
ma = self.window.winfo_toplevel()
|
|
w, h = ma.winfo_width(), ma.winfo_height()
|
|
|
|
# if self.current_image.size[0] > w or self.current_image.size[1] > h:
|
|
if w < 21 or h < 21: # too small
|
|
w = 1280
|
|
h = 1024
|
|
self.current_image.thumbnail((w - 20, h - 20), Image.ANTIALIAS)
|
|
|
|
print("resized: win %s >= img %s", (w, h), self.current_image.size)
|
|
|
|
image_to_display = ImageTk.PhotoImage(self.current_image)
|
|
self.image_label.configure(image=image_to_display)
|
|
self.image_label.image = image_to_display
|