mirror of
https://github.com/Akascape/Datamosher-Pro.git
synced 2025-12-05 15:59:59 +01:00
726 lines
32 KiB
Python
726 lines
32 KiB
Python
#Author: Akash Bora
|
|
#License: MIT | Copyright (c) 2022 Akash Bora
|
|
currentversion=1.8
|
|
|
|
#Import required modules
|
|
import tkinter
|
|
import tkinter.messagebox
|
|
import customtkinter
|
|
from tkinter import ttk, messagebox, filedialog
|
|
import sys
|
|
import os
|
|
import imageio_ffmpeg
|
|
import subprocess
|
|
import imageio
|
|
from PIL import Image, ImageTk #Upgrade pillow if you are facing any import error with PIL (pip install pillow --upgrade)
|
|
from RangeSlider.RangeSlider import RangeSliderH, RangeSliderV
|
|
import threading
|
|
import ctypes
|
|
import webbrowser
|
|
import requests
|
|
import time
|
|
|
|
#Import the local datamosh library
|
|
from DatamoshLib.Tomato import tomato
|
|
from DatamoshLib.Original import classic, repeat, pymodes
|
|
from DatamoshLib.FFG_effects import basic_modes, external_script
|
|
|
|
#Resource Finder
|
|
def resource(relative_path):
|
|
base_path = getattr(
|
|
sys,
|
|
'_MEIPASS',
|
|
os.path.dirname(os.path.abspath(__file__)))
|
|
return os.path.join(base_path, relative_path)
|
|
|
|
#Main Window
|
|
WIDTH = 780
|
|
HEIGHT = 520
|
|
try:
|
|
ctypes.windll.shcore.SetProcessDpiAwareness(0)
|
|
except:
|
|
pass
|
|
customtkinter.set_appearance_mode("Dark")
|
|
customtkinter.set_default_color_theme("blue")
|
|
self=customtkinter.CTk()
|
|
self.title("Datamosher Pro (python version)")
|
|
self.geometry(f"{WIDTH}x{HEIGHT}")
|
|
self.bind("<1>", lambda event: event.widget.focus_set())
|
|
self.grid_columnconfigure(1, weight=1)
|
|
self.grid_rowconfigure(0, weight=1)
|
|
self.resizable(width=False, height=False)
|
|
frame_left = customtkinter.CTkFrame(master=self,width=180,corner_radius=0)
|
|
frame_left.grid(row=0, column=0, sticky="nswe")
|
|
frame_right = customtkinter.CTkFrame(master=self)
|
|
frame_right.grid(row=0, column=1, sticky="nswe", padx=20, pady=20)
|
|
icopath=ImageTk.PhotoImage(file=resource("Assets/Icons/Program_icon.png"))
|
|
self.iconphoto(False, icopath)
|
|
|
|
#FFMPEG path (using the imageio ffmpeg plugin)
|
|
ffdir=os.path.dirname(imageio_ffmpeg.__file__).replace(os.sep, '/')+"/binaries/"
|
|
fileff=''.join([idx for idx in os.listdir(ffdir) if idx[0].lower()=='f'.lower()])
|
|
ffmpeg=resource(ffdir+fileff)
|
|
|
|
#Effect List
|
|
modelist=sorted(["Bloom", "Invert", "Jiggle", "Overlap", "Pulse", "Reverse",
|
|
"Random", "Classic", "Glide", "Sort", "Echo", "Void",
|
|
"Fluid", "Stretch", "Motion Transfer", "Repeat", "Shear", "Delay", "Sink",
|
|
"Mirror", "Vibrate", "Slam Zoom", "Zoom","Invert-Reverse", "Shift",
|
|
"Noise", "Stop", "Buffer", "Slice", "Shuffle", "Rise", "Custom Script"])
|
|
current=modelist[0]
|
|
|
|
#Making the top widgets for changing the modes dynamically
|
|
def ChangeModeRight():
|
|
global current
|
|
modelist.append(modelist.pop(0))
|
|
current=modelist[0]
|
|
mode_info.configure(text=current)
|
|
dynamic()
|
|
def ChangeModeLeft():
|
|
global current
|
|
modelist.insert(0, modelist.pop())
|
|
current=modelist[0]
|
|
mode_info.configure(text=current)
|
|
dynamic()
|
|
frame_info = customtkinter.CTkFrame(master=frame_right, width=520, height=100)
|
|
frame_info.place(x=20,y=20)
|
|
mode_info = customtkinter.CTkLabel(master=frame_info,text=current, corner_radius=10, width=320,
|
|
height=50,
|
|
fg_color=("white", "gray38"))
|
|
mode_info.place(x=100,y=25)
|
|
play_icon =Image.open(resource("Assets/Icons/right_icon.png")).resize((20, 20), Image.Resampling.LANCZOS)
|
|
|
|
left_but = customtkinter.CTkButton(master=frame_info, image=ImageTk.PhotoImage(play_icon.transpose(Image.Transpose.FLIP_LEFT_RIGHT)), text="", width=50, height=50,
|
|
corner_radius=10, fg_color="gray40", hover_color="gray25", command=ChangeModeLeft)
|
|
left_but.place(x=20,y=25)
|
|
|
|
right_but = customtkinter.CTkButton(master=frame_info, image=ImageTk.PhotoImage(play_icon), text="", width=50, height=50,
|
|
corner_radius=10, fg_color="gray40", hover_color="gray25", command=ChangeModeRight)
|
|
right_but.place(x=450,y=25)
|
|
|
|
#Open video function
|
|
previous=""
|
|
def open_function():
|
|
global ofile, vid_image2, previous, duration, vid
|
|
ofile=tkinter.filedialog.askopenfilename(filetypes =[('Video', ['*.mp4','*.avi','*.mov','*.mkv','*gif']),('All Files', '*.*')])
|
|
#Check the video type
|
|
supported=["mp4","avi","mov","gif","mkv","wmv","m4v"]
|
|
if ofile:
|
|
previous=ofile
|
|
pass
|
|
else:
|
|
ofile=previous
|
|
return
|
|
if ofile[-3:].lower() in supported:
|
|
pass
|
|
else:
|
|
print("This file type is not supported!")
|
|
return
|
|
if len(os.path.basename(ofile))>=20:
|
|
showinput=os.path.basename(ofile)[:10]+"..."+os.path.basename(ofile)[-3:]
|
|
else:
|
|
showinput=os.path.basename(ofile)[:20]
|
|
#Change the thumbnail
|
|
button_open.configure(fg_color='grey', text=showinput)
|
|
outpng="Assets/thumbnail_cache/vid_thumbnail.jpg"
|
|
button_thumbnail.configure(image=vid_image)
|
|
if os.path.exists("Assets/thumbnail_cache/vid_thumbnail.jpg"):
|
|
os.remove("Assets/thumbnail_cache/vid_thumbnail.jpg")
|
|
subprocess.call(f'"{ffmpeg}" -loglevel quiet -ss 00:00:01 -t 00:00:01 -i "{ofile}" -qscale:v 2 -r 10.0 "{outpng}"', shell=True)
|
|
vid_image2 = ImageTk.PhotoImage(Image.open(outpng).resize((167, 100), Image.Resampling.LANCZOS))
|
|
button_thumbnail.configure(image=vid_image2)
|
|
vid=imageio.get_reader(ofile, 'ffmpeg')
|
|
#Update the widget parameters
|
|
position_frame.configure(to=vid.count_frames(), number_of_steps=vid.count_frames())
|
|
duration= vid.get_meta_data()['duration']
|
|
rangebar.max_val=duration
|
|
label_seconds2.configure(text="End: "+str(int(duration))+"s")
|
|
rangebar2.max_val=vid.count_frames()
|
|
label_showframe2.configure(text="End: "+str(vid.count_frames()))
|
|
shuf_slider.configure(to=vid.count_frames(), number_of_steps=vid.count_frames())
|
|
end_mosh.set(duration)
|
|
end_frame_mosh.set(vid.count_frames())
|
|
|
|
#Left Frame Widgets
|
|
label_appname = customtkinter.CTkLabel(master=frame_left,text="DATAMOSHER PRO")
|
|
label_appname.place(x=20,y=10)
|
|
|
|
vid_image = ImageTk.PhotoImage(Image.open("Assets/thumbnail_cache/offline_image.png").resize((167, 100), Image.Resampling.LANCZOS))
|
|
|
|
button_thumbnail = tkinter.Label(master=frame_left, image=vid_image, width=167, height=100, text="", bg="grey")
|
|
button_thumbnail.place(x=5,y=40)
|
|
|
|
add_video_image = ImageTk.PhotoImage(Image.open(resource("Assets/Icons/video_icon.png")).resize((20, 15), Image.Resampling.LANCZOS))
|
|
button_open = customtkinter.CTkButton(master=frame_left, image=add_video_image, text="IMPORT VIDEO", width=160, height=35,
|
|
compound="right", command=open_function)
|
|
button_open.place(x=10,y=170)
|
|
|
|
label_export = customtkinter.CTkLabel(master=frame_left,anchor="w",text="Export Format")
|
|
label_export.place(x=12,y=215)
|
|
|
|
optionmenu_1 = customtkinter.CTkOptionMenu(master=frame_left, fg_color="#4d4d4d",width=160, height=35, values=["mp4","avi","mov","mkv"])
|
|
optionmenu_1.set("mp4")
|
|
optionmenu_1.place(x=10,y=250)
|
|
|
|
settingpng = ImageTk.PhotoImage(Image.open(resource("Assets/Icons/settings.png")).resize((20, 20), Image.Resampling.LANCZOS))
|
|
|
|
#Setting Window
|
|
def close_top3():
|
|
window_UI.destroy()
|
|
uisetting.configure(state=tkinter.NORMAL)
|
|
def changeUI():
|
|
global window_UI
|
|
window_UI = customtkinter.CTkToplevel(self)
|
|
window_UI.geometry("410x200")
|
|
window_UI.resizable(width=False, height=False)
|
|
window_UI.title("App Preferences")
|
|
window_UI.iconphoto(False, icopath)
|
|
uisetting.configure(state=tkinter.DISABLED)
|
|
window_UI.wm_transient(self)
|
|
def check():
|
|
URL = "https://raw.githubusercontent.com/Akascape/Datamosher-Pro/Miscellaneous/VERSIONPY.txt"
|
|
try:
|
|
response = requests.get(URL)
|
|
open("Assets\\version\\VERSIONPY.txt", "wb").write(response.content)
|
|
except:
|
|
tkinter.messagebox.showinfo("Unable to connect!", "Unable to get information, please check your internet connection or visit the github repository.")
|
|
return
|
|
time.sleep(2)
|
|
with open("Assets\\version\\VERSIONPY.txt", 'r') as uf:
|
|
nver=float(uf.read())
|
|
if nver>currentversion:
|
|
tkinter.messagebox.showinfo("New Version available!","A new version "+ str(nver) +
|
|
" is available, \nPlease visit the github repository or the original download page!")
|
|
else:
|
|
tkinter.messagebox.showinfo("No Updates!", "You are on the latest version!")
|
|
def docs():
|
|
webbrowser.open_new_tab("https://github.com/Akascape/Datamosher-Pro/wiki")
|
|
def repo():
|
|
webbrowser.open_new_tab("https://github.com/Akascape/Datamosher-Pro")
|
|
logo_image = ImageTk.PhotoImage(Image.open(resource("Assets/Icons/Logo.png")).resize((210, 200), Image.Resampling.LANCZOS))
|
|
logo=customtkinter.CTkButton(master=window_UI, image=logo_image, width=205, height=105, border_width=0,
|
|
corner_radius=1, border_color="grey20", text="",fg_color=customtkinter.ThemeManager.theme["color"]["frame_low"][1],
|
|
hover_color=customtkinter.ThemeManager.theme["color"]["frame_low"][1])
|
|
logo.place(x=200,y=0)
|
|
visit=customtkinter.CTkButton(window_UI, text="Visit Repo", width=150, height=35,
|
|
corner_radius=10, fg_color=customtkinter.ThemeManager.theme["color"]["button"][1], hover_color="gray25",command=repo)
|
|
visit.place(x=20,y=30)
|
|
checkupdate=customtkinter.CTkButton(window_UI, text="Check For Updates", width=150, height=35,
|
|
corner_radius=10, fg_color=customtkinter.ThemeManager.theme["color"]["button"][1], hover_color="gray25",command=check)
|
|
checkupdate.place(x=20,y=80)
|
|
helpbutton=customtkinter.CTkButton(window_UI, text="Help", width=150, height=35,
|
|
corner_radius=10, fg_color=customtkinter.ThemeManager.theme["color"]["button"][1], hover_color="gray25",command=docs)
|
|
helpbutton.place(x=20,y=130)
|
|
version_label=customtkinter.CTkLabel(window_UI,anchor="w",width=1,fg_color=customtkinter.ThemeManager.theme["color"]["frame_low"][1],
|
|
text="v"+str(currentversion), bg_color=customtkinter.ThemeManager.theme["color"]["frame_low"][1])
|
|
version_label.place(x=365, y=2)
|
|
dvname=customtkinter.CTkLabel(window_UI,anchor="w",width=1,fg_color=customtkinter.ThemeManager.theme["color"]["frame_low"][1],
|
|
text="Developer: Akash Bora", bg_color=customtkinter.ThemeManager.theme["color"]["frame_low"][1])
|
|
dvname.place(x=30,y=170)
|
|
window_UI.protocol("WM_DELETE_WINDOW", close_top3)
|
|
uisetting = customtkinter.CTkButton(master=frame_left, image=settingpng, text="", width=40, height=40,
|
|
corner_radius=10, fg_color=customtkinter.ThemeManager.theme["color"]["text_disabled"][1],
|
|
hover_color="gray25", command=changeUI)
|
|
uisetting.place(x=10,y=470)
|
|
|
|
#Validation for entries
|
|
def only_numbers(char):
|
|
if ((char.isdigit()) or (char=="")) and (len(char)<=6):
|
|
return True
|
|
else:
|
|
return False
|
|
validation = self.register(only_numbers)
|
|
|
|
### Dynamimc widgets that change with modes ###
|
|
|
|
#Kill Frame Widget
|
|
def changekill(value):
|
|
label_kill.configure(text="Kill Frame Size: "+str(round(value,4)))
|
|
label_kill = customtkinter.CTkLabel(master=frame_right,anchor="w",text="Kill Frame Size: 0.6")
|
|
slider_kill= customtkinter.CTkSlider(master=frame_right, width=500,
|
|
from_=1,
|
|
to=0,
|
|
number_of_steps=100, command=changekill)
|
|
slider_kill.set(0.6)
|
|
|
|
#N-frameWidget
|
|
varn=tkinter.IntVar()
|
|
label_ctime = customtkinter.CTkLabel(master=frame_right,anchor='w',text="Frame Count:",width=1)
|
|
ctime=customtkinter.CTkEntry(frame_right,textvariable=varn, validate='key', validatecommand=(validation, '%P'),
|
|
placeholder_text="1",
|
|
width=100,
|
|
placeholder_text_color="grey70",
|
|
height=30,
|
|
border_width=2,
|
|
corner_radius=10)
|
|
varn.set(1)
|
|
|
|
#Keep frame & Keep audio widgets
|
|
keepaudio = customtkinter.CTkCheckBox(master=frame_right, text="Keep Audio", onvalue=1, offvalue=0)
|
|
keepframe = customtkinter.CTkCheckBox(master=frame_right, text="Keep First Frame", onvalue=1, offvalue=0)
|
|
|
|
#Count Frame widget
|
|
label_frame1 = customtkinter.CTkLabel(master=frame_right,anchor='w',text="Position Frame: 1",width=1)
|
|
def framework(value):
|
|
try:
|
|
if ofile:
|
|
pass
|
|
except:
|
|
return
|
|
label_frame1.configure(text="Position Frame: "+str(int(value)))
|
|
position_frame=customtkinter.CTkSlider(master=frame_right, width=500,progress_color="black", fg_color="black",
|
|
from_=1,
|
|
to=0,
|
|
number_of_steps=1, command=framework)
|
|
|
|
position_frame.set(1)
|
|
|
|
#Classic Rangebar
|
|
start_mosh = tkinter.DoubleVar()
|
|
end_mosh = tkinter.DoubleVar()
|
|
rangebar = RangeSliderH(frame_right, [start_mosh, end_mosh], Width=510, Height=63,
|
|
bgColor=customtkinter.ThemeManager.theme["color"]["frame_low"][1],line_color="black",
|
|
bar_color_outer=customtkinter.ThemeManager.theme["color"]["button"][0],
|
|
bar_color_inner=customtkinter.ThemeManager.theme["color"]["checkmark"][1],min_val=0, max_val=1, show_value= False,
|
|
line_s_color=customtkinter.ThemeManager.theme["color"]["button"][0])
|
|
label_seconds1 = customtkinter.CTkLabel(master=frame_right,anchor='w',text="Start: 0s",width=1)
|
|
label_seconds2 = customtkinter.CTkLabel(master=frame_right,anchor='w',text="End: 0s",width=1)
|
|
label_segment= customtkinter.CTkLabel(master=frame_right,anchor='w',text="Choose Segment:",width=1)
|
|
def show2(*args):
|
|
try:
|
|
if ofile:
|
|
pass
|
|
except:
|
|
return
|
|
label_seconds2.configure(text="End: "+str(int(end_mosh.get()))+"s")
|
|
def show(*args):
|
|
try:
|
|
if ofile:
|
|
pass
|
|
except:
|
|
return
|
|
label_seconds1.configure(text="Start: "+str(int(start_mosh.get()))+"s")
|
|
start_mosh.trace_add('write', show)
|
|
end_mosh.trace_add('write', show2)
|
|
|
|
#Delta entry widget
|
|
label_p = customtkinter.CTkLabel(master=frame_right,anchor='w',text="P-frames (Delta):",width=1)
|
|
label_segment= customtkinter.CTkLabel(master=frame_right,anchor='w',text="Choose Segment:",width=1)
|
|
|
|
varp=tkinter.IntVar()
|
|
delta=customtkinter.CTkEntry(frame_right,
|
|
placeholder_text="1",validate='key', validatecommand=(validation, '%P'),
|
|
width=100, textvariable=varp,
|
|
placeholder_text_color="grey70",
|
|
height=30,
|
|
border_width=2,
|
|
corner_radius=10)
|
|
varp.set(1)
|
|
|
|
#Frame Rangebar for repeat and rise modes
|
|
start_frame_mosh = tkinter.DoubleVar()
|
|
end_frame_mosh = tkinter.DoubleVar()
|
|
rangebar2 = RangeSliderH(frame_right, [start_frame_mosh, end_frame_mosh], Width=510, Height=63,
|
|
bgColor=customtkinter.ThemeManager.theme["color"]["frame_low"][1],line_color="black",
|
|
bar_color_outer=customtkinter.ThemeManager.theme["color"]["button"][0],
|
|
bar_color_inner=customtkinter.ThemeManager.theme["color"]["checkmark"][1],min_val=0, max_val=1, show_value= False,
|
|
line_s_color=customtkinter.ThemeManager.theme["color"]["button"][0])
|
|
label_showframe1 = customtkinter.CTkLabel(master=frame_right,anchor='w',text="Start Frame: 0",width=1)
|
|
label_showframe2 = customtkinter.CTkLabel(master=frame_right,anchor='w',text="End: 0",width=1)
|
|
label_segment2= customtkinter.CTkLabel(master=frame_right,anchor='w',text="Choose Frame Segment:",width=1)
|
|
def show3(*args):
|
|
try:
|
|
if ofile:
|
|
pass
|
|
except:
|
|
return
|
|
label_showframe2.configure(text="End: "+str(int(end_frame_mosh.get())))
|
|
def show4(*args):
|
|
try:
|
|
if ofile:
|
|
pass
|
|
except:
|
|
return
|
|
label_showframe1.configure(text="Start Frame: "+str(int(start_frame_mosh.get())))
|
|
start_frame_mosh.trace_add('write', show4)
|
|
end_frame_mosh.trace_add('write', show3)
|
|
|
|
#slider for echo mode
|
|
def midwork(value):
|
|
label_mid.configure(text="Start Point: "+str(round(value,1)))
|
|
label_mid = customtkinter.CTkLabel(master=frame_right,anchor='w',text="Start Point: 0.5",width=1)
|
|
mid_point=customtkinter.CTkSlider(master=frame_right, width=500,progress_color="black", fg_color="black",
|
|
from_=0,
|
|
to=1,
|
|
number_of_steps=10,command=midwork)
|
|
mid_point.set(0.5)
|
|
|
|
#Some options for sort mode
|
|
keepsort = customtkinter.CTkCheckBox(master=frame_right, text="Keep First Frames", onvalue=0, offvalue=1)
|
|
reversesort=customtkinter.CTkCheckBox(master=frame_right, text="Reverse", onvalue=False, offvalue=True)
|
|
|
|
#Options for ffglitch modes
|
|
hw_auto=customtkinter.CTkSwitch(master=frame_right,text="HW Acceleration \n(Auto)",onvalue=1, offvalue=0)
|
|
labelk=customtkinter.CTkLabel(master=frame_right,anchor='w',text="Keyframes:",width=1)
|
|
kf=customtkinter.CTkComboBox(master=frame_right,height=30, width=150,
|
|
values=["1000","100","10", "1"])
|
|
#Widget for Shuffle mde
|
|
def changeshuf(value):
|
|
shuf_label.configure(text="Chunk Size: "+str(int(value)))
|
|
shuf_label=customtkinter.CTkLabel(master=frame_right,anchor='w',text="Chunk Size: 1",width=1)
|
|
shuf_slider=customtkinter.CTkSlider(master=frame_right, width=500,
|
|
from_=1,
|
|
to=0,
|
|
number_of_steps=1, command=changeshuf)
|
|
shuf_slider.set(1)
|
|
|
|
#Widget for Fluid mode
|
|
def changefluid(value):
|
|
fluid_label.configure(text="Amount: "+str(int(value)))
|
|
fluid_label=customtkinter.CTkLabel(master=frame_right,anchor='w',text="Amount: 5",width=1)
|
|
slider_fluid=customtkinter.CTkSlider(master=frame_right, width=500,
|
|
from_=1,
|
|
to=20,
|
|
number_of_steps=100, command=changefluid)
|
|
slider_fluid.set(5)
|
|
#Stretch mode widget
|
|
v_h=customtkinter.CTkSwitch(frame_right,text="Horizontal Stretch", onvalue=1, offvalue=0)
|
|
|
|
#Button for motion transfer mode
|
|
def open_MT():
|
|
global vfile
|
|
vfile=tkinter.filedialog.askopenfilename(filetypes =[('Vector File', ['*.mp4','*.avi','*.mov','*.mkv']),('All Files', '*.*')])
|
|
if vfile:
|
|
mt_button.configure(fg_color='grey', text=os.path.basename(vfile))
|
|
else:
|
|
return
|
|
mt_button=customtkinter.CTkButton(master=frame_right, text="OPEN VIDEO", width=520, height=30,
|
|
compound="right",command=open_MT)
|
|
|
|
#Button for custom script mode
|
|
scriptfile=''
|
|
def open_script():
|
|
global scriptfile
|
|
scriptfile=tkinter.filedialog.askopenfilename(filetypes =[('Script File', ['*.js','*.py']),('All Files', '*.*')])
|
|
if scriptfile:
|
|
scriptbutton.configure(fg_color='grey', text=os.path.basename(scriptfile))
|
|
else:
|
|
scriptbutton.configure(fg_color=customtkinter.ThemeManager.theme["color"]["button"], text='OPEN SCRIPT')
|
|
scriptfile=''
|
|
scriptbutton=customtkinter.CTkButton(master=frame_right, text="OPEN SCRIPT", width=520, height=30,
|
|
compound="right",command=open_script)
|
|
|
|
#Dynamic UI functions for each widget
|
|
def rangeslider(x):
|
|
if x==1:
|
|
rangebar.place(x=20,y=210)
|
|
label_seconds1.place(x=25,y=200)
|
|
label_seconds2.place(x=470,y=200)
|
|
label_segment.place(x=25,y=170)
|
|
else:
|
|
rangebar.place_forget()
|
|
label_segment.place_forget()
|
|
label_seconds1.place_forget()
|
|
label_seconds2.place_forget()
|
|
def rangeslider2(x):
|
|
if x==1:
|
|
if (current=="Rise"):
|
|
rangebar2.place(x=20,y=260)
|
|
label_showframe1.place(x=25,y=250)
|
|
label_showframe2.place(x=470,y=250)
|
|
label_segment2.place(x=25,y=220)
|
|
else:
|
|
rangebar2.place(x=20,y=210)
|
|
label_showframe1.place(x=25,y=200)
|
|
label_showframe2.place(x=470,y=200)
|
|
label_segment2.place(x=25,y=170)
|
|
else:
|
|
rangebar2.place_forget()
|
|
label_showframe1.place_forget()
|
|
label_showframe2.place_forget()
|
|
label_segment2.place_forget()
|
|
def mid(x):
|
|
if x==1:
|
|
mid_point.place(x=20,y=210)
|
|
label_mid.place(x=25,y=170)
|
|
else:
|
|
mid_point.place_forget()
|
|
label_mid.place_forget()
|
|
def killoption(x):
|
|
if x==1 or x==2 or x==3:
|
|
label_kill.place(x=25,y=170)
|
|
slider_kill.place(x=20,y=200)
|
|
else:
|
|
label_kill.place_forget()
|
|
slider_kill.place_forget()
|
|
def positionslider(x):
|
|
if x==1:
|
|
label_frame1.place(x=25,y=230)
|
|
position_frame.place(x=20,y=260)
|
|
else:
|
|
label_frame1.place_forget()
|
|
position_frame.place_forget()
|
|
def framekeep(x):
|
|
if x==1:
|
|
keepframe.place(x=250,y=300)
|
|
elif x==2:
|
|
keepframe.place(x=250,y=240)
|
|
else:
|
|
keepframe.place_forget()
|
|
def audiokeep(x):
|
|
if x==1:
|
|
keepaudio.place(x=400,y=300)
|
|
elif x==2:
|
|
keepaudio.place(x=400,y=240)
|
|
else:
|
|
keepaudio.place_forget()
|
|
def ctimes(x):
|
|
if x==1:
|
|
ctime.place(x=110,y=300)
|
|
label_ctime.place(x=25,y=300)
|
|
else:
|
|
ctime.place_forget()
|
|
label_ctime.place_forget()
|
|
def pdelta(x):
|
|
if x==1:
|
|
delta.place(x=135,y=275)
|
|
label_p.place(x=25,y=275)
|
|
if current=="Repeat":
|
|
varp.set(5)
|
|
elif x==2:
|
|
delta.place(x=135,y=170)
|
|
label_p.place(x=25,y=170)
|
|
if current=="Glide":
|
|
varp.set(5)
|
|
else:
|
|
delta.place_forget()
|
|
label_p.place_forget()
|
|
def sortoptions(x):
|
|
if x==1:
|
|
keepsort.place(x=30, y=170)
|
|
reversesort.place(x=300, y=170)
|
|
else:
|
|
keepsort.place_forget()
|
|
reversesort.place_forget()
|
|
def ffgassist(x):
|
|
if x==1:
|
|
hw_auto.place(x=25, y=170)
|
|
labelk.place(x=300,y=170)
|
|
kf.place(x=380,y=170)
|
|
else:
|
|
hw_auto.place_forget()
|
|
labelk.place_forget()
|
|
kf.place_forget()
|
|
def shuf(x):
|
|
if x==1:
|
|
shuf_label.place(x=25,y=220)
|
|
shuf_slider.place(x=20,y=250)
|
|
else:
|
|
shuf_label.place_forget()
|
|
shuf_slider.place_forget()
|
|
def fluidwidget(x):
|
|
if x==1:
|
|
fluid_label.place(x=25,y=220)
|
|
slider_fluid.place(x=20,y=250)
|
|
else:
|
|
fluid_label.place_forget()
|
|
slider_fluid.place_forget()
|
|
def h_v(x):
|
|
if x==1:
|
|
v_h.place(x=25,y=220)
|
|
else:
|
|
v_h.place_forget()
|
|
def mtwid(x):
|
|
if x==1:
|
|
mt_button.place(x=20,y=230)
|
|
else:
|
|
mt_button.place_forget()
|
|
def custom(x):
|
|
if x==1:
|
|
scriptbutton.place(x=20,y=230)
|
|
else:
|
|
scriptbutton.place_forget()
|
|
|
|
#Main Function to update the widgets
|
|
def dynamic():
|
|
global current, showwidgets
|
|
allwidgets=[audiokeep, positionslider, killoption, framekeep,
|
|
ctimes, pdelta, rangeslider, rangeslider2, mid, sortoptions, ffgassist, fluidwidget, h_v, custom, mtwid, shuf]
|
|
for i in allwidgets:
|
|
i(0)
|
|
showwidgets=[]
|
|
if (current=="Bloom") or (current=="Pulse") or (current=="Pulse") or(current=="Overlap"):
|
|
showwidgets=[audiokeep, positionslider, killoption, framekeep, ctimes]
|
|
u=1
|
|
elif (current=="Jiggle"):
|
|
showwidgets=[positionslider, audiokeep, killoption, framekeep]
|
|
u=1
|
|
elif (current=="Void") or (current=="Reverse") or (current=="Invert") or (current=="Random"):
|
|
showwidgets=[killoption,audiokeep, framekeep]
|
|
u=2
|
|
elif (current=="Classic"):
|
|
showwidgets=[rangeslider, pdelta]
|
|
u=1
|
|
elif (current=="Glide"):
|
|
showwidgets=[pdelta]
|
|
u=2
|
|
elif (current=="Repeat") or (current=="Rise"):
|
|
if (current=="Rise"):
|
|
showwidgets=[rangeslider2, ffgassist]
|
|
else:
|
|
showwidgets=[rangeslider2, pdelta]
|
|
u=1
|
|
elif (current=="Echo"):
|
|
showwidgets=[mid]
|
|
u=1
|
|
elif (current=="Sort"):
|
|
showwidgets=[sortoptions]
|
|
u=1
|
|
elif ((current=="Buffer") or (current=="Sink") or (current=="Mirror") or (current=="Shear") or (current=="Noise")
|
|
or (current=="Delay") or (current=="Slam Zoom") or (current=="Invert-Reverse") or (current=="Shift") or (current=="Zoom")
|
|
or (current=="Slice")or (current=="Vibrate") or (current=="Stop")):
|
|
showwidgets=[ffgassist]
|
|
u=1
|
|
elif (current=="Fluid"):
|
|
showwidgets=[ffgassist, fluidwidget]
|
|
u=1
|
|
elif (current=="Stretch"):
|
|
showwidgets=[ffgassist, h_v]
|
|
u=1
|
|
elif (current=="Motion Transfer"):
|
|
showwidgets=[ffgassist, mtwid]
|
|
u=1
|
|
elif (current=="Custom Script"):
|
|
showwidgets=[ffgassist, custom]
|
|
u=1
|
|
elif (current=="Shuffle"):
|
|
showwidgets=[ffgassist, shuf]
|
|
u=1
|
|
for widgets in showwidgets:
|
|
widgets(u)
|
|
dynamic()
|
|
keepframe.select()
|
|
|
|
#autosave video function
|
|
def savename():
|
|
global sfile
|
|
if ofile:
|
|
try:
|
|
sfile=ofile[:-4]+"_datamoshed_"+current+'.'+optionmenu_1.get()
|
|
nf=0
|
|
while os.path.exists(sfile):
|
|
nf=nf+1
|
|
sfile=ofile[:-4]+"_datamoshed_"+current+'('+str(nf)+')'+'.'+optionmenu_1.get()
|
|
except:
|
|
sfile=""
|
|
#A function that will thread the main mosh function to separate the processes
|
|
def Threadthis():
|
|
global varp, varn
|
|
if delta.get()=='' or delta.get()<'1':
|
|
varp.set(1)
|
|
if ctime.get()=='' or ctime.get()<'1':
|
|
varn.set(1)
|
|
threading.Thread(target=Do_the_mosh).start()
|
|
#Converter function
|
|
def ffmpeg_convert(inputpath, parameters, outputpath, extra=''):
|
|
subprocess.call(f'"{ffmpeg}" {extra} -i "{inputpath}" {parameters} -y "{outputpath}"', shell=True)
|
|
|
|
#Main Function of the whole program
|
|
def Do_the_mosh():
|
|
global ofile, sfile, param, param2
|
|
button_mosh.configure(state=tkinter.DISABLED)
|
|
if previous=="":
|
|
tkinter.messagebox.showinfo("No Video imported!","Please import a video file!")
|
|
button_mosh.configure(state=tkinter.NORMAL)
|
|
return
|
|
try:
|
|
savename()
|
|
ProcessLabel.configure(text='STEP 1/3 CONVERTING...')
|
|
param="-c:v libx264 -preset medium -b:v 2M -minrate 2M -maxrate 2M -bufsize 2M" #Add other ffmpeg parameters in this line only
|
|
if ((current=="Bloom") or (current=="Pulse") or (current=="Pulse") or(current=="Overlap")
|
|
or (current=="Void") or (current=="Reverse") or (current=="Invert") or (current=="Random") or (current=="Jiggle")):
|
|
ifile=sfile[:-4]+".avi"
|
|
ffmpeg_convert(ofile,param,ifile)
|
|
ProcessLabel.configure(text='STEP 2/3 MOSHING...')
|
|
mfile=sfile[:-4]+"_corrupted.avi"
|
|
tomato.mosh(infile=ifile, outfile=mfile, m=current.lower(), c=varn.get(), n=int(position_frame.get()), k=round(slider_kill.get(),4), a=keepaudio.get(), f=keepframe.get())
|
|
time.sleep(1)
|
|
os.remove(ifile)
|
|
ProcessLabel.configure(text='STEP 3/3 FIXING THE CORRUPTED FILE...')
|
|
ffmpeg_convert(mfile,param,sfile)
|
|
os.remove(mfile)
|
|
elif ((current=="Classic") or (current=="Repeat") or (current=="Glide") or (current=="Sort") or (current=="Echo")):
|
|
param="-bf 0 -b 10000k" #Add other ffmpeg parameters in this line only for the above modes
|
|
ifile=sfile[:-4]+".avi"
|
|
ffmpeg_convert(ofile,param,ifile)
|
|
ProcessLabel.configure(text='STEP 2/3 MOSHING...')
|
|
mfile=sfile[:-4]+"_corrupted.avi"
|
|
if current=="Classic":
|
|
classic.Datamosh(ifile, mfile,s=int(start_mosh.get()),e=int(end_mosh.get()),p=varp.get(), fps=vid.get_meta_data()['fps'])
|
|
elif current=="Repeat":
|
|
repeat.Datamosh(ifile, mfile, s=int(start_frame_mosh.get()), e=int(end_frame_mosh.get()), p=varp.get(), fps=vid.get_meta_data()['fps'])
|
|
elif current=="Glide":
|
|
pymodes.library.glide(varp.get(), ifile, mfile)
|
|
elif current=="Sort":
|
|
pymodes.library.avi_sort(ifile, mfile, mode=keepsort.get(), rev=reversesort.get())
|
|
elif current=="Echo":
|
|
pymodes.library.process_streams(ifile, mfile, mid=round(mid_point.get(),1))
|
|
os.remove(ifile)
|
|
ProcessLabel.configure(text='STEP 3/3 FIXING THE CORRUPTED FILE...')
|
|
ffmpeg_convert(mfile,param,sfile)
|
|
os.remove(mfile)
|
|
else:
|
|
time.sleep(1)
|
|
ProcessLabel.configure(text='STEP 2/3 MOSHING...')
|
|
mfile=sfile[:-4]+"_corrupted.mpg"
|
|
if current=="Fluid":
|
|
basic_modes.library(ofile, mfile, mode=3, fluidity=int(slider_fluid.get()), gop=kf.get())
|
|
elif current=="Stretch":
|
|
basic_modes.library(ofile, mfile, mode=2, vh=v_h.get(), gop=kf.get())
|
|
elif current=="Motion Transfer":
|
|
if vfile:
|
|
basic_modes.library(ofile, mfile, mode=1, extract_from=vfile, gop=kf.get())
|
|
else:
|
|
tkinter.messagebox.showinfo("No Vector File imported!", "Please choose the video from where you want to extract the vector motion.")
|
|
button_mosh.configure(state=tkinter.NORMAL)
|
|
ProcessLabel.configure(text='Choose any secondary video file for transfering the vectors!')
|
|
return
|
|
elif current=="Shuffle":
|
|
basic_modes.library(ofile, mfile, mode=4, size=int(shuf_slider.get()), gop=kf.get())
|
|
elif current=="Rise":
|
|
basic_modes.library(ofile, mfile, mode=5, s=int(start_frame_mosh.get()), e=int(end_frame_mosh.get()-start_frame_mosh.get()), gop=kf.get())
|
|
elif current=="Custom Script":
|
|
external_script.mosh(ofile, mfile, mode=1, scriptfile=scriptfile, gop=kf.get())
|
|
else:
|
|
external_script.mosh(ofile, mfile, mode=2, effect=current, gop=kf.get())
|
|
ProcessLabel.configure(text='STEP 3/3 FIXING THE CORRUPTED FILE...')
|
|
if hw_auto.get()==1:
|
|
hw_type=' -hwaccel auto '
|
|
else:
|
|
hw_type=''
|
|
ffmpeg_convert(mfile,param,sfile,extra=hw_type)
|
|
os.remove(mfile)
|
|
except:
|
|
pass
|
|
#Check the result and complete the task
|
|
if os.path.exists(sfile):
|
|
tkinter.messagebox.showinfo("Exported!", "File exported successfully, \nFile Location:" +str(sfile))
|
|
ProcessLabel.configure(text="Last used: "+current)
|
|
button_mosh.configure(state=tkinter.NORMAL)
|
|
else:
|
|
tkinter.messagebox.showerror("Oops!", "Something went wrong! \nPlease recheck the settings and try again.")
|
|
ProcessLabel.configure(text='Recheck the settings and try again!')
|
|
button_mosh.configure(state=tkinter.NORMAL)
|
|
|
|
#Bottom Part
|
|
ProcessLabel = customtkinter.CTkLabel(master=frame_right,
|
|
width=400, height=30,corner_radius=10,
|
|
text="START DATAMOSHING!", fg_color=("white", "gray38"))
|
|
ProcessLabel.place(x=20,y=430)
|
|
button_mosh = customtkinter.CTkButton(master=frame_right, height=30,width=110,corner_radius=10,
|
|
text="MOSH", command=Threadthis)
|
|
button_mosh.place(x=430,y=430)
|
|
self.mainloop()
|
|
|
|
#--------------------------------------------------------------------#
|