mirror of
https://github.com/Akascape/Datamosher-Pro.git
synced 2025-12-12 03:09:59 +01:00
Delete Datamosher Pro.py
This commit is contained in:
Akascape
GitHub
@@ -1,414 +0,0 @@
|
|||||||
import os
|
|
||||||
from tkinter import *
|
|
||||||
from tkinter import ttk, messagebox, filedialog
|
|
||||||
import tkinter
|
|
||||||
import sys
|
|
||||||
import imageio
|
|
||||||
import random
|
|
||||||
import re
|
|
||||||
import struct
|
|
||||||
import time
|
|
||||||
import webbrowser
|
|
||||||
from itertools import chain, repeat
|
|
||||||
def resource_path0(relative_path):
|
|
||||||
base_path = getattr(
|
|
||||||
sys,
|
|
||||||
'_MEIPASS',
|
|
||||||
os.path.dirname(os.path.abspath(__file__)))
|
|
||||||
return os.path.join(base_path, relative_path)
|
|
||||||
def openfile():
|
|
||||||
global file
|
|
||||||
file=tkinter.filedialog.askopenfilename(filetypes =[('MP4', '*.mp4'),('AVI', '*.avi'),('GIF','*.gif'),('MOV','*.mov')])
|
|
||||||
if(len(file)>1):
|
|
||||||
LocationError.config(text=file, fg="green")
|
|
||||||
OpeningFile['text']='Open Again'
|
|
||||||
OpeningFile['bg']='#D0CECE'
|
|
||||||
else:
|
|
||||||
LocationError.config(text="Choose Video To Datamosh", fg="red")
|
|
||||||
OpeningFile['text']='OPEN'
|
|
||||||
OpeningFile['bg']='#82CC6C'
|
|
||||||
def convert(inputpath,targetformat):
|
|
||||||
global outputpath
|
|
||||||
try:
|
|
||||||
Wait.config(text="Converting the Video...", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
outputpath=os.path.splitext(inputpath)[0]+'_datamoshed'+targetformat
|
|
||||||
reader=imageio.get_reader(inputpath)
|
|
||||||
fps=reader.get_meta_data()['fps']
|
|
||||||
writer= imageio.get_writer(outputpath, fps=fps)
|
|
||||||
for frames in reader:
|
|
||||||
writer.append_data(frames)
|
|
||||||
#print(f'Frame {frames}')
|
|
||||||
#print("Converted")
|
|
||||||
writer.close()
|
|
||||||
try:
|
|
||||||
Datamosh(outputpath)
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
except Exception:
|
|
||||||
messagebox.showerror("FAILED","The video file or the input data have \n some issues!")
|
|
||||||
Wait.place_forget()
|
|
||||||
os.remove(outputpath)
|
|
||||||
datamoshbtn['state']=NORMAL
|
|
||||||
def Step1():
|
|
||||||
try:
|
|
||||||
if (len(file)>=1):
|
|
||||||
Wait.place(relx=0.5,rely=0.85,anchor='center')
|
|
||||||
root.update()
|
|
||||||
extension=os.path.splitext(file)[1]
|
|
||||||
datamoshbtn['state']=DISABLED
|
|
||||||
datamoshbtn['cursor']='watch'
|
|
||||||
if extension==".avi":
|
|
||||||
Datamosh(file)
|
|
||||||
else:
|
|
||||||
targetformat=".avi"
|
|
||||||
convert(file,targetformat)
|
|
||||||
else:
|
|
||||||
messagebox.showerror("OOPS","Please choose the video again!")
|
|
||||||
except:
|
|
||||||
messagebox.showerror("OOPS","Please choose the video!")
|
|
||||||
datamoshbtn['state']=NORMAL
|
|
||||||
datamoshbtn['cursor']=''
|
|
||||||
def Datamosh(Inputfile):
|
|
||||||
if os.path.splitext(outputpath)[1]==".avi":
|
|
||||||
filein = Inputfile
|
|
||||||
countframes = int(Countframe.get())
|
|
||||||
positframes = int(Positframe.get())
|
|
||||||
firstframe = int(firstframes.get())
|
|
||||||
choice = modechoices.get()
|
|
||||||
kill = float(killframe.get())
|
|
||||||
if(choice==modes[0]):
|
|
||||||
mode="bloom"
|
|
||||||
elif(choice==modes[1]):
|
|
||||||
mode="invert"
|
|
||||||
elif(choice==modes[2]):
|
|
||||||
mode="jiggle"
|
|
||||||
elif(choice==modes[3]):
|
|
||||||
mode="overlap"
|
|
||||||
elif(choice==modes[4]):
|
|
||||||
mode="pulse"
|
|
||||||
elif(choice==modes[5]):
|
|
||||||
mode="reverse"
|
|
||||||
elif(choice==modes[6]):
|
|
||||||
mode="random"
|
|
||||||
else:
|
|
||||||
messagebox.showerror("OOPS","Please Select the Correct Mode!")
|
|
||||||
os.remove(outputpath)
|
|
||||||
os.kill(Datamosh())
|
|
||||||
Wait.place_forget()
|
|
||||||
datamoshbtn['state']=NORMAL
|
|
||||||
datamoshbtn['cursor']=''
|
|
||||||
if filein is None or os.path.exists(filein) == False:
|
|
||||||
messagebox.showerror("ERROR","Input File is Missing!")
|
|
||||||
Wait.place_forget()
|
|
||||||
os.remove(outputpath)
|
|
||||||
datamoshbtn['state']=NORMAL
|
|
||||||
os.kill(Datamosh())
|
|
||||||
datamoshbtn['cursor']=''
|
|
||||||
#define temp directory and files
|
|
||||||
temp_nb = random.randint(10000, 99999)
|
|
||||||
temp_dir = "temp-" + str(temp_nb)
|
|
||||||
temp_hdrl = temp_dir +"/hdrl.bin"
|
|
||||||
temp_movi = temp_dir +"/movi.bin"
|
|
||||||
temp_idx1 = temp_dir +"/idx1.bin"
|
|
||||||
os.mkdir(temp_dir)
|
|
||||||
#Define constrain function for jiggle :3
|
|
||||||
def constrain(val, min_val, max_val):
|
|
||||||
return min(max_val, max(min_val, val))
|
|
||||||
def bstream_until_marker(bfilein, bfileout, marker=0, startpos=0):
|
|
||||||
chunk = 1024
|
|
||||||
filesize = os.path.getsize(bfilein)
|
|
||||||
if marker :
|
|
||||||
marker = str.encode(marker)
|
|
||||||
|
|
||||||
with open(bfilein,'rb') as rd:
|
|
||||||
with open(bfileout,'ab') as wr:
|
|
||||||
for pos in range(startpos, filesize, chunk):
|
|
||||||
rd.seek(pos)
|
|
||||||
buffer = rd.read(chunk)
|
|
||||||
|
|
||||||
if marker:
|
|
||||||
if buffer.find(marker) > 0 :
|
|
||||||
marker_pos = re.search(marker, buffer).start() # position is relative to buffer glitchedframes
|
|
||||||
marker_pos = marker_pos + pos # position should be absolute now
|
|
||||||
split = buffer.split(marker, 1)
|
|
||||||
wr.write(split[0])
|
|
||||||
return marker_pos
|
|
||||||
else:
|
|
||||||
wr.write(buffer)
|
|
||||||
else:
|
|
||||||
wr.write(buffer)
|
|
||||||
#make 3 files, 1 for each chunk
|
|
||||||
movi_marker_pos = bstream_until_marker(filein, temp_hdrl, "movi")
|
|
||||||
idx1_marker_pos = bstream_until_marker(filein, temp_movi, "idx1", movi_marker_pos)
|
|
||||||
bstream_until_marker(filein, temp_idx1, 0, idx1_marker_pos)
|
|
||||||
with open(temp_movi,'rb') as rd:
|
|
||||||
chunk = 1024
|
|
||||||
filesize = os.path.getsize(temp_movi)
|
|
||||||
frame_table = []
|
|
||||||
|
|
||||||
for pos in range(0, filesize, chunk):
|
|
||||||
rd.seek(pos)
|
|
||||||
buffer = rd.read(chunk)
|
|
||||||
#build first list with all adresses
|
|
||||||
for m in (re.finditer(b'\x30\x31\x77\x62', buffer)): # find iframes
|
|
||||||
if audio : frame_table.append([m.start() + pos, 'sound'])
|
|
||||||
for m in (re.finditer(b'\x30\x30\x64\x63', buffer)): # find b frames
|
|
||||||
frame_table.append([m.start() + pos, 'video'])
|
|
||||||
#then remember to sort the list
|
|
||||||
frame_table.sort(key=lambda tup: tup[0])
|
|
||||||
l = []
|
|
||||||
l.append([0,0, 'void'])
|
|
||||||
max_frame_size = 0
|
|
||||||
#build tuples for each frame index with frame sizes
|
|
||||||
for n in range(len(frame_table)):
|
|
||||||
if n + 1 < len(frame_table):
|
|
||||||
frame_size = frame_table[n + 1][0] - frame_table[n][0]
|
|
||||||
else:
|
|
||||||
frame_size = filesize - frame_table[n][0]
|
|
||||||
max_frame_size = max(max_frame_size, frame_size)
|
|
||||||
l.append([frame_table[n][0],frame_size, frame_table[n][1]])
|
|
||||||
# variables that make shit work
|
|
||||||
clean = []
|
|
||||||
final = []
|
|
||||||
# keep first video frame or not
|
|
||||||
if firstframe :
|
|
||||||
for x in l :
|
|
||||||
if x[2] == 'video':
|
|
||||||
clean.append(x)
|
|
||||||
break
|
|
||||||
# clean the list by killing "big" frames
|
|
||||||
for x in l:
|
|
||||||
if x[1] <= (max_frame_size * kill) :
|
|
||||||
clean.append(x)
|
|
||||||
# FX modes
|
|
||||||
if mode == "random":
|
|
||||||
Wait.config(text="Applying Effect: Random", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
final = random.sample(clean,len(clean))
|
|
||||||
if mode == "reverse":
|
|
||||||
Wait.config(text="Applying Effect: Reverse", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
final = clean[::-1]
|
|
||||||
if mode == "invert":
|
|
||||||
Wait.config(text="Applying Effect: Invert", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
final = sum(zip(clean[1::2], clean[::2]), ())
|
|
||||||
if mode == 'bloom':
|
|
||||||
Wait.config(text="Applying Effect: Bloom", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
repeat = int(countframes)
|
|
||||||
frame = int(positframes)
|
|
||||||
## split list
|
|
||||||
lista = clean[:frame]
|
|
||||||
listb = clean[frame:]
|
|
||||||
## rejoin list with bloom
|
|
||||||
final = lista + ([clean[frame]]*repeat) + listb
|
|
||||||
if mode == 'pulse':
|
|
||||||
Wait.config(text="Applying Effect: Pulse", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
pulselen = int(countframes)
|
|
||||||
pulseryt = int(positframes)
|
|
||||||
j = 0
|
|
||||||
for x in clean:
|
|
||||||
i = 0
|
|
||||||
if(j % pulselen == 0):
|
|
||||||
while i < pulselen :
|
|
||||||
final.append(x)
|
|
||||||
i = i + 1
|
|
||||||
else:
|
|
||||||
final.append(x)
|
|
||||||
j = j + 1
|
|
||||||
if mode == "jiggle":
|
|
||||||
Wait.config(text="Applying Effect: Jiggle", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
amount = int(positframes)
|
|
||||||
final = [clean[constrain(x+int(random.gauss(0,amount)),0,len(clean)-1)] for x in range(0,len(clean))]
|
|
||||||
if mode == "overlap":
|
|
||||||
Wait.config(text="Applying Effect: Overlap", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
pulselen = int(countframes)
|
|
||||||
pulseryt = int(positframes)
|
|
||||||
clean = [clean[i:i+pulselen] for i in range(0,len(clean),pulseryt)]
|
|
||||||
final = [item for sublist in clean for item in sublist]
|
|
||||||
time.sleep(2)
|
|
||||||
#name new file
|
|
||||||
cname = '-c' + str(countframes) if int(countframes) > 1 else ''
|
|
||||||
pname = '-n' + str(positframes) if int(positframes) > 1 else ''
|
|
||||||
fileout= filein[:-4] + '-' + mode + cname + pname + '.avi'
|
|
||||||
#delete old file
|
|
||||||
if os.path.exists(fileout):
|
|
||||||
os.remove(fileout)
|
|
||||||
bstream_until_marker(temp_hdrl, fileout)
|
|
||||||
with open(temp_movi,'rb') as rd:
|
|
||||||
filesize = os.path.getsize(temp_movi)
|
|
||||||
with open(fileout,'ab') as wr:
|
|
||||||
wr.write(struct.pack('<4s', b'movi'))
|
|
||||||
for x in final:
|
|
||||||
if x[0] != 0 and x[1] != 0:
|
|
||||||
rd.seek(x[0])
|
|
||||||
wr.write(rd.read(x[1]))
|
|
||||||
bstream_until_marker(temp_idx1, fileout)
|
|
||||||
export=exportbox.get()
|
|
||||||
def removefileout():
|
|
||||||
if os.path.exists(fileout):
|
|
||||||
os.remove(fileout)
|
|
||||||
messagebox.showinfo("DONE","Datamoshed video is ready!")
|
|
||||||
Wait.place_forget()
|
|
||||||
datamoshbtn['state']=NORMAL
|
|
||||||
datamoshbtn['cursor']=''
|
|
||||||
else:
|
|
||||||
pass
|
|
||||||
if os.path.exists(outputpath):
|
|
||||||
os.remove(outputpath)
|
|
||||||
else:
|
|
||||||
pass
|
|
||||||
os.remove(temp_hdrl)
|
|
||||||
os.remove(temp_movi)
|
|
||||||
os.remove(temp_idx1)
|
|
||||||
os.rmdir(temp_dir)
|
|
||||||
if(export==exportchoices[0]):
|
|
||||||
Wait.config(text="Moshed Video is Ready!", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
messagebox.showinfo("DONE","Datamoshed video is ready!")
|
|
||||||
Wait.place_forget()
|
|
||||||
datamoshbtn['state']=NORMAL
|
|
||||||
datamoshbtn['cursor']=''
|
|
||||||
elif(export==exportchoices[1]):
|
|
||||||
targetformat=".mp4"
|
|
||||||
convert(fileout,targetformat)
|
|
||||||
removefileout()
|
|
||||||
elif(export==exportchoices[2]):
|
|
||||||
targetformat=".gif"
|
|
||||||
convert(fileout,targetformat)
|
|
||||||
removefileout()
|
|
||||||
elif(export==exportchoices[3]):
|
|
||||||
targetformat=".mov"
|
|
||||||
convert(fileout,targetformat)
|
|
||||||
removefileout()
|
|
||||||
elif(export==exportchoices[4]):
|
|
||||||
targetformat=".mkv"
|
|
||||||
convert(fileout,targetformat)
|
|
||||||
removefileout()
|
|
||||||
elif(export==exportchoices[5]):
|
|
||||||
targetformat=".wmv"
|
|
||||||
convert(fileout,targetformat)
|
|
||||||
removefileout()
|
|
||||||
else:
|
|
||||||
messagebox.showerror("OOPS","Please Select the Correct Format!")
|
|
||||||
os.remove(outputpath)
|
|
||||||
os.kill(Datamosh())
|
|
||||||
Wait.place_forget()
|
|
||||||
datamoshbtn['state']=NORMAL
|
|
||||||
datamoshbtn['cursor']=''
|
|
||||||
else:
|
|
||||||
Wait.config(text="Moshed Video is Ready!", fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
root.update()
|
|
||||||
def toggle():
|
|
||||||
if var.get() == 0:
|
|
||||||
Disableadvanced()
|
|
||||||
else:
|
|
||||||
Enableadvanced()
|
|
||||||
def Enableadvanced():
|
|
||||||
countlabel['fg']="green"
|
|
||||||
positlabel['fg']="green"
|
|
||||||
firstflabel['fg']="green"
|
|
||||||
killlabel['fg']="green"
|
|
||||||
Countframe['state']=NORMAL
|
|
||||||
Positframe['state']=NORMAL
|
|
||||||
firstframes['state']=NORMAL
|
|
||||||
killframe['state']=NORMAL
|
|
||||||
def Disableadvanced():
|
|
||||||
countlabel['fg']="grey"
|
|
||||||
positlabel['fg']="grey"
|
|
||||||
firstflabel['fg']="grey"
|
|
||||||
killlabel['fg']="grey"
|
|
||||||
Countframe['state']=DISABLED
|
|
||||||
Positframe['state']=DISABLED
|
|
||||||
firstframes['state']=DISABLED
|
|
||||||
killframe['state']=DISABLED
|
|
||||||
def info():
|
|
||||||
messagebox.showinfo("Help",
|
|
||||||
"Datamosher Pro is made for those who want to datamosh their video files and achieve that glitch effect."
|
|
||||||
"\nHow To Use:\n➤First input the video you want to datamosh."
|
|
||||||
"\n➤Choose the desired datamosh mode, then select the export format."
|
|
||||||
"\n➤Use advance options for different modes (For more info about the advanced options, go to our Github page)."
|
|
||||||
"\n➤Then just click on the datamosh button, then wait for a few seconds."
|
|
||||||
"\n➤After converting, your video will be moshed and you can find the video in the directory."
|
|
||||||
"\n➤Note that if you mosh the same files in the same location again, then the new moshed file will replace the old file."
|
|
||||||
"\n\nDeveloper: Akash Bora (a.k.a. Akascape)\nIf you have any issue then contact me on Github.")
|
|
||||||
def callback(url):
|
|
||||||
webbrowser.open_new_tab("https://github.com/Akascape/Datamosher-Pro-GUI-.git")
|
|
||||||
root=Tk()
|
|
||||||
root.title("Datamosher Pro")
|
|
||||||
root.resizable(width=False, height=False)
|
|
||||||
root.geometry("500x450")
|
|
||||||
root.configure(bg='#FFFFFF')
|
|
||||||
root.columnconfigure(0,weight=1)
|
|
||||||
icopath=resource_path0("icon.ico")
|
|
||||||
root.wm_iconbitmap(icopath)
|
|
||||||
path=resource_path0("label.png")
|
|
||||||
LabelIMG=PhotoImage(file=path)
|
|
||||||
headlabel=Label(root,image=LabelIMG,borderwidth=0, highlightthickness=0, padx=0,pady=0)
|
|
||||||
headlabel.grid()
|
|
||||||
LocationError=Label(root,text="Choose Video To Datamosh",fg="red",bg='#FFFFFF')
|
|
||||||
LocationError.grid()
|
|
||||||
OpeningFile= Button(root, width=60,bg="#82CC6C",fg="white", text="OPEN",highlightthickness=1,borderwidth=0.2,relief="groove",command=openfile)
|
|
||||||
OpeningFile.grid()
|
|
||||||
chooselabel=Label(root,text="Select Mode:",font="Aharoni",bg='#FFFFFF')
|
|
||||||
chooselabel.place(x=80,y=115,anchor='center')
|
|
||||||
modes=["Bloom","Invert","Jiggle", "Overlap","Pulse", "Reverse", "Random"]
|
|
||||||
modechoices=ttk.Combobox(root,values=modes, font="Verdana 12", width=7)
|
|
||||||
modechoices.current(0)
|
|
||||||
modechoices.place(x=130,y=104)
|
|
||||||
exportchoices=["AVI(RAW)","MP4","GIF","MOV","MKV","WMV"]
|
|
||||||
exportlabel=Label(root,text="Export Format:",font="Aharoni",bg='#FFFFFF')
|
|
||||||
exportlabel.place(x=270,y=105)
|
|
||||||
exportbox=ttk.Combobox(root,values=exportchoices, font="Verdana 10", width=8)
|
|
||||||
exportbox.current(1)
|
|
||||||
exportbox.place(x=380,y=104)
|
|
||||||
var = IntVar()
|
|
||||||
var.set(0)
|
|
||||||
advancedbox=Checkbutton(root, text="Advanced",bg="#FFFFFF", command=toggle,variable=var,onvalue=1, offvalue=0)
|
|
||||||
advancedbox.place(x=70,y=150, anchor='center')
|
|
||||||
var1=IntVar()
|
|
||||||
var1.set(1)
|
|
||||||
countlabel=Label(root,text="Glitch Freqency",fg="grey",bg='#FFFFFF')
|
|
||||||
countlabel.place(x=40,y=165)
|
|
||||||
Countframe=Entry(root,bg="light blue",width=10,borderwidth=3, textvariable=var1,state=DISABLED)
|
|
||||||
Countframe.place(x=140,y=165)
|
|
||||||
var2=IntVar()
|
|
||||||
var2.set(1)
|
|
||||||
positlabel=Label(root,text="Frames Frequency",fg="grey",bg='#FFFFFF')
|
|
||||||
positlabel.place(x=270,y=165)
|
|
||||||
Positframe=Entry(root,bg="light blue",width=10,borderwidth=3, textvariable=var2, state=DISABLED)
|
|
||||||
Positframe.place(x=380,y=165)
|
|
||||||
var3=IntVar()
|
|
||||||
var3.set(1)
|
|
||||||
firstflabel=Label(root,text="Ignored Frames",fg="grey",bg='#FFFFFF')
|
|
||||||
firstflabel.place(x=40,y=200)
|
|
||||||
firstframes=Entry(root,bg="light blue",width=10,borderwidth=3, textvariable=var3, state=DISABLED)
|
|
||||||
firstframes.place(x=140,y=200)
|
|
||||||
var4=DoubleVar()
|
|
||||||
var4.set(0.7)
|
|
||||||
killlabel=Label(root,text="Kill Frames",fg="grey",bg='#FFFFFF')
|
|
||||||
killlabel.place(x=270,y=200)
|
|
||||||
killframe=Entry(root,bg="light blue",width=10,borderwidth=3, textvariable=var4, state=DISABLED)
|
|
||||||
killframe.place(x=380,y=200)
|
|
||||||
global Wait
|
|
||||||
Wait=Label(root,text="",fg="#6D76CD",bg='#FFFFFF', font=("Aharoni",15))
|
|
||||||
infobtn=Button(root, width=2,bg="#FFFFFF",fg="black", text="ⓘ",font=(10),relief="sunken",cursor='hand2', highlightthickness=0,borderwidth=0,padx=0,pady=0,command=info)
|
|
||||||
infobtn.place(x=475,y=425)
|
|
||||||
dev=Label(root, text='Developed by Akascape | For more info, visit:',bg='#FFFFFF',fg="#6D76CD", font=("Impact",10))
|
|
||||||
dev.place(x=5,y=430)
|
|
||||||
link=Label(root, text="Datamosher Pro Github",font=('Impact',10),bg='#FFFFFF',fg="#6D76CD", cursor="hand2")
|
|
||||||
link.place(x=245,y=430)
|
|
||||||
link.bind("<Button-1>", lambda e:
|
|
||||||
callback("https://github.com/Akascape/Datamosher-Pro-GUI-.git"))
|
|
||||||
path0=resource_path0("button.png")
|
|
||||||
buttonIMG=PhotoImage(file=path0)
|
|
||||||
datamoshbtn=Button(root,image=buttonIMG,borderwidth=0, highlightthickness=0, padx=0,pady=0,command=Step1)
|
|
||||||
datamoshbtn.place(x=100,y=270)
|
|
||||||
root.mainloop()
|
|
||||||
#DEVELOPER: AKASH BORA (a.k.a Akascape)
|
|
||||||
#Version=1.0
|
|
||||||
Reference in New Issue
Block a user