main.py

import os
import tkinter as tk
from tkinter import filedialog
from tkinter import messagebox
import ffmpeg
from subprocess import check_output
import re
from PIL import ImageTk, Image
import sys


def resource_path(relative_path):  # embedded files nonsense
    if hasattr(sys, '_MEIPASS'):
        return os.path.join(sys._MEIPASS, relative_path)
    return os.path.join(os.path.abspath("."), relative_path)


def sort_human(l):  # sort frame sequences in a sane way
    def convert(text): return float(text) if text.isdigit() else text
    def alphanum(key): return [convert(c)
                               for c in re.split('([-+]?[0-9]*\.?[0-9]*)', key)]
    l.sort(key=alphanum)
    return l


# Return the aspect ratio as an actual ratio string instead of a decimal.
# This is never used in our actual resizing calculations; it's just because
# "4:3" is a lot prettier (and easier to understand) than "1.33333...".
def aspect_ratio_str(width: int, height: int) -> str:

    temp = 0

    def gcd(a, b):
        return a if b == 0 else gcd(b, a % b)

    if width == height:
        return "1:1"

    if width < height:
        temp = width
        width = height
        height = temp

    divisor = gcd(width, height)

    x = int(width / divisor) if not temp else int(height / divisor)
    y = int(height / divisor) if not temp else int(width / divisor)

    return f"{x}:{y}"


root = tk.Tk()
root.title('JunieTool')
#root.iconbitmap(resource_path('icon.ico'))

frame_select = tk.Frame(root, bd=1, padx=12,
                        pady=3, relief="raised", width=480)
frame_select.grid(row=0, column=0, sticky="nesw")

selected_seq_info = tk.Message(
    frame_select, text="No images selected", width=480)
selected_seq_info.grid(row=1, sticky="ew")

selected_img_info = tk.Message(frame_select, text="", width=480)
selected_img_info.grid(row=2, sticky="ew")

# placeholder values, not 0,0 because otherwise i get yelled at for dividing by zero
sel_w, sel_h = (1, 1)
files = []
file_open = False
indir = ''
parent_indir = ''
seq_dirs = []
custom_outdir = ''

def select_meta():
    if parent_mode.get() == 1:
        select_parent_indir()
    else:
        select_frame_sequence()

def select_parent_indir():
    global sel_w
    global sel_h
    global file_open
    global parent_indir
    global seq_dirs
    parent_indir = filedialog.askdirectory(parent=root, mustexist=True, title='Choose the directory containing your frame sequences.')
    if not parent_indir:
        return None
    seq_dirs = [parent_indir + os.sep + x for x in os.listdir(parent_indir) if not os.path.isfile(x)]
    selected_seq_info.config(  # update info labels
        text="Selected parent " + parent_indir + " (" + str(len(seq_dirs)) + " total directories)")
    im = Image.open(seq_dirs[0] + os.sep + os.listdir(seq_dirs[0])[0])
    sel_w, sel_h = im.size
    selected_img_info.config(text="")
    file_open = True
    # check if we already have a desired size before enabling render button
    if entry_h.get() != "" and entry_w.get() != "":
        button_ffmpeg.config(state="normal")

def select_frame_sequence():
    global sel_w
    global sel_h
    global indir
    global files
    global file_open
    indir = filedialog.askdirectory(parent=root, mustexist=True, title='Choose the directory your sequence is in')
    if not indir:
        return None
    files = [x for x in os.listdir(indir) if x.endswith(".png")]
    files = sort_human(files)
    selected_seq_info.config(  # update info labels
        text="Selected " + files[0] + " (" + str(len(files)) + " total images)")
    im = Image.open(indir + os.sep + files[0])
    sel_w, sel_h = im.size
    selected_img_info.config(
        text=str(sel_w)+"x"+str(sel_h)+", "+aspect_ratio_str(sel_w, sel_h))
    file_open = True
    # check if we already have a desired size before enabling render button
    if entry_h.get() != "" and entry_w.get() != "":
        button_ffmpeg.config(state="normal")

def select_custom_outdir(): # browse for a custom output folder
    global custom_outdir
    custom_outdir = filedialog.askdirectory(parent=root, title='Choose an output directory', mustexist=True)
    if not custom_outdir:
        custom_outdir = ''

button_browse = tk.Button(
    frame_select, text='Browse...', command=select_meta, width=15)
button_browse.grid(row=0, sticky="ew")  # ew, sticky


def update_desired_ratio():  # called whenever the w/h entries are edited, keep this light
    disp_temp = "Ratio: " + \
        str(aspect_ratio_str(int(entry_w.get()), int(entry_h.get())))
    ratio_display.config(text=disp_temp)


# no, an "intvar" and an int var are not the same thing
# yes, i am annoyed about it
parent_mode = tk.IntVar()
parent_mode.set(0)
crop_h = tk.IntVar()
crop_h.set(1)
overwrite_og = tk.IntVar()
overwrite_og.set(0)
use_custom_outdir = tk.IntVar()
use_custom_outdir.set(0)

frame_options = tk.Frame(root, bd=1, relief="raised",
                         padx=12, pady=3)
frame_options.grid(row=2, column=0, sticky="nesw")
check_parent_mode = tk.Checkbutton(frame_options, text="Parent mode", variable=parent_mode, onvalue=True, offvalue=False)
check_parent_mode.pack(anchor="w")
check_overwrite_og = tk.Checkbutton(
    frame_options, text='Overwrite source images', variable=overwrite_og, onvalue=True, offvalue=False)
check_overwrite_og.pack(anchor="w")
radio_crop_h = tk.Radiobutton(
    frame_options, text="Crop by height", variable=crop_h, value=1).pack(anchor="w")
radio_crop_w = tk.Radiobutton(
    frame_options, text="Crop by width", variable=crop_h, value=0).pack(anchor="w")
check_custom_outdir = tk.Checkbutton(
    frame_options, text='Use custom output folder', variable=use_custom_outdir, onvalue=True, offvalue=False)
check_custom_outdir.pack(anchor="w")

browse_custom_outdir = tk.Button(frame_options, text="Browse...", command=select_custom_outdir, width=15)
browse_custom_outdir.pack(anchor="w")

def ffmpeg_export_meta():
    if parent_mode.get() == 1:
        for x in seq_dirs:
            ffmpeg_export(x)
    else:
        ffmpeg_export(indir)

def ffmpeg_export(p_indir):  # the real meat, this is where i struggle with ffmpeg-python and occasionally succeed
    des_w = int(entry_w.get())
    des_h = int(entry_h.get())
    if (des_w > sel_w) or (des_h > sel_h):
        messagebox.showerror(
            title="Error", message="Desired size is larger than source size!")
        return
    sel_ratio = sel_w / sel_h
    des_ratio = des_w / des_h
    # safe placeholder values for when src and output have the same ratio
    x_offset = 0
    y_offset = 0
    adj_w = sel_w
    adj_h = sel_h
    if (crop_h.get() == 1) and (sel_ratio != des_ratio):
        adj_w = des_ratio * sel_h  # get the new width for the desired aspect ratio
        x_offset = (sel_w - adj_w) / 2  # centering math
    elif (crop_h.get() == 0) and (sel_ratio != des_ratio):
        adj_h = des_ratio * sel_w
        y_offset = (sel_h - adj_h) / 2
    for x in files:
        x = p_indir + os.sep + x
        progress_ffmpeg.config(text='Rendering: ' + os.path.split(x)[1])
        frame_ffmpeg.update()
        # ffmpeg complains if we try to output to the same file as our input
        # so we output to a different file and replace the input afterwards
        outdir = x + '~.png'
        if overwrite_og.get() == 0 and use_custom_outdir.get() == 0:
            newdir = os.path.dirname(str(x)) + '_' + \
                str(des_w) + 'x' + str(
                    des_h)  # results in {old directory}_{width}x{height} in the old directory's parent dir
            outdir = newdir + os.sep + str(os.path.split(x)[1])
            if not os.path.isdir(newdir):
                os.mkdir(newdir)
        elif use_custom_outdir.get() == 1:
            outdir = custom_outdir + os.sep + str(os.path.split(x)[1])
        stream = ffmpeg.input(str(x), nostdin=None)
        stream = ffmpeg.crop(stream, x_offset, y_offset, adj_w, adj_h)
        stream = ffmpeg.filter(stream, "scale", des_w,
                               des_h, flags="bilinear")  # TODO: allow user to choose algorithm
        stream = ffmpeg.output(
            stream, outdir, hide_banner=None)  # TODO: find a way to stop making a new shell for each op
        stream = ffmpeg.overwrite_output(stream)
        ffmpeg.run_async(stream)
        if overwrite_og.get() == 1:  # check again because overwriting when we're not supposed to is bad mkay
            os.remove(x)
            os.rename(x + '~.png', x)
    progress_ffmpeg.config(text='Rendering: Done!')


frame_ffmpeg = tk.Frame(root, bd=1, padx=12, pady=3,
                        relief="raised")
frame_ffmpeg.grid(row=3, column=0, sticky="nesw")

button_ffmpeg = tk.Button(
    frame_ffmpeg, text='Render', command=ffmpeg_export_meta, state="disabled", width=15, height=3)
button_ffmpeg.grid(row=0, sticky="ew")
progress_ffmpeg = tk.Message(frame_ffmpeg, text='Rendering: N/A', width=400)
progress_ffmpeg.grid(row=1, sticky="ew")

frame_entry = tk.Frame(root, bd=1, relief="raised",
                       padx=12, pady=3, width=1400)
frame_entry.grid(row=1, column=0, sticky="nesw")

tk.Label(frame_entry, text='Enter desired size:').grid(row=0, column=0)
tk.Label(frame_entry, text='Width').grid(row=1, column=0, sticky="e")
tk.Label(frame_entry, text='Height').grid(row=2, column=0, sticky="e")

# no, stringvar and string var are also not the same thing
# yes, i am still annoyed
sv_w = tk.StringVar()
sv_h = tk.StringVar()


def sv_edited(var, indx, mode):
    if entry_h.get() == "" or entry_w.get() == "":
        button_ffmpeg.config(state="disabled")
        return
    else:
        if file_open:
            button_ffmpeg.config(state="normal")
        update_desired_ratio()


sv_w.trace_add("write", sv_edited)
sv_h.trace_add("write", sv_edited)

entry_w = tk.Entry(frame_entry, textvariable=sv_w)
entry_h = tk.Entry(frame_entry, textvariable=sv_h)
entry_w.grid(row=1, column=1, padx=1, pady=1, sticky="w")
entry_h.grid(row=2, column=1, padx=1, pady=1, sticky="w")


ratio_display = tk.Message(frame_entry, text="Ratio: N/A", width=500)
ratio_display.grid(row=0, column=1, sticky="e")
root.mainloop()