aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--preview.py89
-rw-r--r--vidslice.py19
2 files changed, 101 insertions, 7 deletions
diff --git a/preview.py b/preview.py
new file mode 100644
index 0000000..f22c6ba
--- /dev/null
+++ b/preview.py
@@ -0,0 +1,89 @@
+import subprocess
+import tempfile
+import threading
+from tkinter import *
+from tkinter import ttk
+
+from options import FFmpegOptions
+
+
+class PreviewPanel(ttk.LabelFrame):
+ def __init__(self, *args, get_ffmpeg_args=lambda: FFmpegOptions([], []), get_frame_count=lambda: 0, **kw):
+ super(PreviewPanel, self).__init__(*args, text='Preview', **kw)
+ self.input_path = None
+ self.get_ffmpeg_args = get_ffmpeg_args
+ self.get_frame_count = get_frame_count
+
+ def button(text, command, column):
+ ttk.Button(self, text=text, command=command).grid(column=column, row=0, sticky=(N, W, S, E))
+ self.columnconfigure(column, weight=1)
+
+ button("Preview Start", self.preview_start, 0)
+ button("Preview Middle", self.preview_middle, 1)
+ button("Preview End", self.preview_end, 2)
+
+ self.image = None
+ self.image_label = ttk.Label(self, anchor='center')
+ self.image_label.grid(column=0, row=1, columnspan=3, sticky=(N, W, S, E))
+ self.rowconfigure(1, weight=1)
+
+ self.enable(False)
+
+ def preview_at(self, offset):
+ offset = int(offset)
+ self.enable(False)
+ real_args = self.get_ffmpeg_args()
+ input_args = real_args.input
+ width = self.image_label.winfo_width()
+ height = self.image_label.winfo_height()
+ real_args.vf += [
+ rf'select=eq(n\,{offset})',
+ f'scale=w={width}:h={height}:force_original_aspect_ratio=decrease'
+ ]
+ real_args.output += ['-frames:v', '1']
+ output_args = real_args.output_with_vf()
+
+ def run():
+ _, output_path = tempfile.mkstemp(suffix='.png')
+ args = ['ffmpeg', '-hide_banner', '-v', 'warning', '-y'] + input_args + \
+ ['-i', self.input_path] + output_args + [output_path]
+ print(args)
+ # noinspection PyArgumentList
+ proc = subprocess.Popen(args, stdin=subprocess.DEVNULL, stdout=subprocess.PIPE,
+ stderr=subprocess.STDOUT, text=True, creationflags=subprocess.CREATE_NO_WINDOW)
+ while proc.poll() is None:
+ out_data = proc.stdout.readline()
+ if out_data != '':
+ print(out_data, end='')
+ self.enable(True)
+ self.image = PhotoImage(file=output_path)
+ self.image_label['image'] = self.image
+
+ threading.Thread(target=run).start()
+
+ def preview_start(self, *args):
+ self.preview_at(0)
+
+ def preview_middle(self, *args):
+ self.preview_at(self.get_frame_count() / 2)
+
+ def preview_end(self, *args):
+ self.preview_at(self.get_frame_count() - 1)
+
+ def enable(self, enabled):
+ state = 'disabled'
+ if enabled:
+ state = '!' + state
+ self.state([state])
+ for child in self.winfo_children():
+ try:
+ child.state([state])
+ except AttributeError:
+ pass
+
+ def set_input_path(self, path, data):
+ self.enable(data is not None)
+ if data is None:
+ self.input_path = None
+ else:
+ self.input_path = path
diff --git a/vidslice.py b/vidslice.py
index 632e426..eca036c 100644
--- a/vidslice.py
+++ b/vidslice.py
@@ -7,6 +7,7 @@ from tkinter import ttk
from options import OptionsPanel
from output import OutputPanel
+from preview import PreviewPanel
from sources import SourcesPanel, update_ytdl
VERSION = "1.6"
@@ -45,20 +46,24 @@ class VidsliceFrame:
root.columnconfigure(0, weight=1)
root.rowconfigure(0, weight=1)
- # set up sources panel
self.sources_panel = SourcesPanel(mainframe)
- self.sources_panel.grid(column=0, row=0, columnspan=2, sticky=(W, E), padx=5, pady=5)
+ self.sources_panel.grid(column=0, row=0, columnspan=2, sticky=(W, E, N, S), padx=5, pady=5)
- # set up options panel
self.options_panel = OptionsPanel(mainframe)
- self.options_panel.grid(column=0, row=1, sticky=(W, E, N), padx=5, pady=5)
- mainframe.rowconfigure(1, weight=1)
+ self.options_panel.grid(column=0, row=1, columnspan=2, sticky=(W, N, S), padx=5, pady=5)
self.sources_panel.on_update(self.options_panel.update_info)
- # set up output panel
+ self.preview_panel = PreviewPanel(mainframe, get_ffmpeg_args=self.options_panel.ffmpeg_opts,
+ get_frame_count=self.options_panel.frame_count)
+ self.preview_panel.grid(column=0, row=2, sticky=(W, E, N, S), padx=5, pady=5)
+ mainframe.rowconfigure(2, weight=1)
+ mainframe.columnconfigure(0, weight=2)
+ self.sources_panel.on_update(
+ lambda data: self.preview_panel.set_input_path(self.sources_panel.get_file(), data))
+
self.output_panel = OutputPanel(mainframe, get_ffmpeg_args=self.options_panel.ffmpeg_opts,
get_frame_count=self.options_panel.frame_count)
- self.output_panel.grid(column=1, row=1, sticky=(W, E, N, S), padx=5, pady=5)
+ self.output_panel.grid(column=1, row=2, sticky=(W, E, N, S), padx=5, pady=5)
mainframe.columnconfigure(1, weight=1)
self.sources_panel.on_update(lambda data: self.output_panel.set_input_path(self.sources_panel.get_file(), data))