Skip to content

Instantly share code, notes, and snippets.

@phsamuel

phsamuel/main_trim.py

Created August 30, 2022 22:51
Show Gist options
  • Save phsamuel/6a33cdf1811d49c30f32d236572f3b2e to your computer and use it in GitHub Desktop.
Save phsamuel/6a33cdf1811d49c30f32d236572f3b2e to your computer and use it in GitHub Desktop.
Download ZIP
A video trimmer with Qt/ffmpeg/vlc
Raw
main_trim.py
import sys
from PyQt5.QtWidgets import QApplication, QDialog, QMainWindow, QMessageBox
from PyQt5.uic import loadUi
from PyQt5.QtCore import pyqtSlot as pyQtSlot
from PyQt5.QtCore import QTimer,QDateTime,Qt,QRect
from PyQt5.QtGui import QPainter, QPen, QFont
from PyQt5.QtWidgets import QFileDialog
from time import sleep
from trim import Ui_MainWindow
import os
from glob import glob
import vlc
from json import load as json_load
from json import dumps as json_dumps
from random import randint
# want to play the transition parts but doesn't quite work
def verify_cut(file,lens):
media = vlc.MediaPlayer(file)
media.play()
sleep(0.25)
media.pause()
acc = 0
alens=[]
for l in lens:
acc = acc + l
alens.append(acc)
halfplaytime=1 # 1 sec before and after transition
betweenpausetime=2 # 1 sec between transition
for l in alens:
print(f'before: {media.get_time()}\nSetting time {l-halfplaytime*1000}\n')
media.play()
media.set_time(l-halfplaytime*1000)
print(f'after: {media.get_time()}')
sleep(halfplaytime*2)
media.pause()
sleep(betweenpausetime)
media.stop()
# pass
def get_length(media):
# media = vlc.MediaPlayer(fname)
media.play()
sleep(0.25)
media.pause()
media.set_position(1)
sleep(0.25)
t=media.get_time()
media.stop()
return t
def update_timestamps(ts,T1,T2,is_keep):
timestamps=ts.copy()
timestamps.append(sys.maxsize)
for i,t in enumerate(timestamps):
if T1<t:
break
new_timestamps=timestamps[:i]
if (i+is_keep)%2==0:
new_timestamps.append(T1)
for j in range(i,len(timestamps)):
if T2< timestamps[j]:
break
if (j+is_keep)%2==0:
new_timestamps.append(T2)
return new_timestamps+timestamps[j:-1]
def convertMillis(millis):
# millis=float(millis)
seconds=int(millis/1000)%60
minutes=int(millis/(1000*60))%60
hours=int(millis/(1000*60*60))%24
return str(hours)+':'+str(minutes)+':'+str(seconds)+'.'+str(millis%1000)
class Window(QMainWindow, Ui_MainWindow):
def __init__(self, parent=None):
super().__init__(parent)
self.setupUi(self)
self.media = None
self.T1=self.T2=0
self.timestamps =[] #[20000,50000,80000,100000,150000,200000]
self.timer = QTimer(self)
self.timer.timeout.connect(self.update)
self.timer.start(100)
self.trim = self.load_trim()
self.L = 1 # length of video, shouldn't be zero
# self.connectSignalsSlots()
# def connectSignalsSlots(self):
# self.lineEdit.textChanged.connect(self.update_and_print)
def load_trim(self):
trim=dict()
if os.path.exists('trim.json'):
with open('trim.json') as f:
trim=json_load(f)
return trim
def save_trim(self):
j = json_dumps(self.trim)
with open('trim.json','w') as f:
f.write(j)
@pyQtSlot()
def on_cutButton_pressed(self):
self.timestamps=update_timestamps(self.timestamps,self.T1,self.T2,False)
@pyQtSlot()
def on_keepButton_pressed(self):
self.timestamps=update_timestamps(self.timestamps,self.T1,self.T2,True)
@pyQtSlot()
def on_T1Button_pressed(self):
if self.media != None:
self.T1=self.media.get_time()
@pyQtSlot()
def on_T2Button_pressed(self):
if self.media != None:
self.T2=self.media.get_time()
@pyQtSlot()
def on_speedupButton_pressed(self):
# print('plus')
if self.media != None:
self.media.set_rate(self.media.get_rate() + 0.1)
print(self.media.get_rate())
@pyQtSlot()
def on_speeddownButton_pressed(self):
# print('minus')
if self.media != None:
r = self.media.get_rate() - 0.1
if r > 0.05:
self.media.set_rate(r)
print(self.media.get_rate())
@pyQtSlot()
def on_trimButton_pressed(self):
if self.media != None:
fname = self.label.text()
self.trim[fname]=self.timestamps
self.save_trim()
r='-'+str(randint(0,10000000))
ts = self.timestamps.copy()
ts.append(self.L)
i=1 if ts[0]==0 else 0 # cut beginning if ts[0]=0
fnames=[]
print(ts,len(ts))
while i+1 < len(ts):
cmd = f'ffmpeg -ss {float(ts[i])/1000.} -i {fname} -t {float(ts[i+1]-ts[i])/1000.} -acodec copy -vcodec copy {str(i)+r}.mkv'
# print(cmd)
os.system(cmd)
fnames.append(f'{str(i)+r}.mkv')
i=i+2
outfile = f'o{r}.mkv'
vlens=concat_files(fnames,outfile)
# delete cut files
for f in fnames:
os.remove(f)
# verify_cut(outfile,[vlens[f] for f in fnames])
# print("trim")
@pyQtSlot()
def update(self):
self.repaint()
if self.media !=None:
self.timeLabel.setText(convertMillis(self.media.get_time()))
self.horizontalSlider.setValue(int(self.media.get_position()*self.horizontalSlider.maximum()))
@pyQtSlot()
def on_workaround_triggered(self):
options = QFileDialog.Options()
options |= QFileDialog.ShowDirsOnly # does not seem to work
options |= QFileDialog.DontResolveSymlinks
folder = QFileDialog.getExistingDirectory(self, "Select folder",
"/home/phsamuel/Videos/Webcam/",
QFileDialog.ShowDirsOnly) #options=options)
# files = glob('/home/phsamuel/Videos/Webcam/*.webm')
files = glob(os.path.join(folder,'*.webm'))
files = [f for f in files if os.stat(f).st_size > 20*1000*1000] # keep files larger than 20 M
self.files_selector.clear()
for f in files:
self.files_selector.addItem(f)
# print(files)
@pyQtSlot(str)
def on_files_selector_currentIndexChanged(self, t):
self.label.setText(t)
self.media = vlc.MediaPlayer(t)
self.L=get_length(self.media)
# self.timestamps=[] #[10000,20000,30000,40000]
self.timestamps = self.trim[t] if t in self.trim else []
print(self.L)
self.T1=0
self.T2=self.L
@pyQtSlot()
def on_playpauseButton_pressed(self):
print("pressed")
print (self.media)
if self.media != None:
if self.media.is_playing()==0:
self.media.play()
self.playpauseButton.setText('Pause')
else:
self.media.pause()
self.playpauseButton.setText('Play')
@pyQtSlot()
def on_stopButton_pressed(self):
if self.media != None:
self.media.stop()
self.playpauseButton.setText('Play')
@pyQtSlot(int)
def on_horizontalSlider_sliderMoved(self,i):
# def on_horizontalSlider_valueChanged(self,i):
# print("value changed")
print(float(i)/self.horizontalSlider.maximum())
if self.media != None:
self.media.set_position(float(i)/self.horizontalSlider.maximum())
# self.label.setText(str(i))
def paintEvent(self,e):
width=self.horizontalSlider.geometry().width()
X=self.horizontalSlider.geometry().x()
# print(self.L)
with QPainter(self) as painter:
painter.setRenderHint(QPainter.Antialiasing)
painter.setPen(QPen(Qt.blue, 1))
# painter.drawLine(self._cell_width(), 0, self._cell_width(), self.height())
painter.drawRect(QRect(X,200,width,30)) #Qt.SolidPattern)
# painter.fillRect(QRect(int(self.width()/4),int(self.height()/4),int(self.width()/2),int(self.height()/3)),Qt.darkGreen) #Qt.SolidPattern)
rect = QRect(X,200,width,30)
# painter.fillRect(rect,Qt.blue)
# if self.media == None:
if self.media != None:
# L=self.media.get_length()
drawTimestamps(painter,self.timestamps,self.L,QRect(X,200,width,30))
# draw T1
painter.setFont(QFont("Times", 12, QFont.Bold))
painter.setPen(QPen(Qt.red, 2))
x=int(X+self.T1*width/self.L)
painter.drawLine(x, 230, x, 190)
painter.drawText(x-20, 180, "T1");
# draw T2
painter.setPen(QPen(Qt.red, 2))
x=int(X+self.T2*width/self.L)
painter.drawLine(x, 230, x, 190)
painter.drawText(x+2, 180, "T2");
def keyPressEvent(self, event):
if event.key() == Qt.Key_Plus:
if self.media != None:
self.media.set_time(self.media.get_time()+2000)
# print ("Killing")
elif event.key() == Qt.Key_Minus:
if self.media != None:
self.media.set_time(self.media.get_time()-2000)
# print("Minus")
event.accept()
def _cell_width(self):
return self.width() / 3
def _cell_height(self):
return self.height() / 3
def drawTimestamps(painter,timestamps,L,rect):
# print('try drawing')
if timestamps==[]:
painter.fillRect(rect,Qt.blue)
else:
ts=timestamps.copy()
ts.insert(0,0)
ts.append(L)
i=0
# print(ts)
# print(rect)
while i+1<len(ts):
# print(int(rect.x()+ts[i]*rect.width()/L))
# print(int(rect.x()+(ts[i+1]-ts[i])*rect.width()/L))
painter.fillRect(QRect(int(rect.x()+ts[i]*rect.width()/L), 200, int((ts[i+1]-ts[i])*rect.width()/L), 30), Qt.blue)
i=i+2
def concat_files(fnames,outfile):
vlens=dict()
with open('mylist.txt','w') as lf:
for f in fnames:
lf.write(f"file '{os.path.join(os.getcwd(),f)}'\n")
vlens[f]=get_length(vlc.MediaPlayer(f))
cmd = f'ffmpeg -f concat -safe 0 -i mylist.txt -c copy {outfile}'
os.system(cmd)
return vlens
if __name__ == "__main__":
app = QApplication(sys.argv)
win = Window()
win.show()
sys.exit(app.exec())
Raw
trim.py
# -*- coding: utf-8 -*-
# Form implementation generated from reading ui file 'trim.ui'
#
# Created by: PyQt5 UI code generator 5.9.2
#
# WARNING! All changes made in this file will be lost!
from PyQt5 import QtCore, QtGui, QtWidgets
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName("MainWindow")
MainWindow.resize(1024, 287)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
self.playpauseButton = QtWidgets.QPushButton(self.centralwidget)
self.playpauseButton.setGeometry(QtCore.QRect(750, 70, 80, 21))
self.playpauseButton.setObjectName("playpauseButton")
self.horizontalSlider = QtWidgets.QSlider(self.centralwidget)
self.horizontalSlider.setGeometry(QtCore.QRect(40, 20, 931, 16))
self.horizontalSlider.setMaximum(9999999)
self.horizontalSlider.setOrientation(QtCore.Qt.Horizontal)
self.horizontalSlider.setObjectName("horizontalSlider")
self.files_selector = QtWidgets.QComboBox(self.centralwidget)
self.files_selector.setGeometry(QtCore.QRect(70, 70, 79, 23))
self.files_selector.setObjectName("files_selector")
self.files_selector.addItem("")
self.files_selector.addItem("")
self.label = QtWidgets.QLabel(self.centralwidget)
self.label.setGeometry(QtCore.QRect(80, 120, 641, 20))
self.label.setObjectName("label")
self.line = QtWidgets.QFrame(self.centralwidget)
self.line.setGeometry(QtCore.QRect(170, 520, 118, 3))
self.line.setFrameShape(QtWidgets.QFrame.HLine)
self.line.setFrameShadow(QtWidgets.QFrame.Sunken)
self.line.setObjectName("line")
self.stopButton = QtWidgets.QPushButton(self.centralwidget)
self.stopButton.setGeometry(QtCore.QRect(840, 70, 80, 23))
self.stopButton.setObjectName("stopButton")
self.T1Button = QtWidgets.QPushButton(self.centralwidget)
self.T1Button.setGeometry(QtCore.QRect(180, 70, 31, 23))
self.T1Button.setObjectName("T1Button")
self.T2Button = QtWidgets.QPushButton(self.centralwidget)
self.T2Button.setGeometry(QtCore.QRect(220, 70, 31, 23))
self.T2Button.setObjectName("T2Button")
self.cutButton = QtWidgets.QPushButton(self.centralwidget)
self.cutButton.setGeometry(QtCore.QRect(310, 70, 80, 23))
self.cutButton.setObjectName("cutButton")
self.keepButton = QtWidgets.QPushButton(self.centralwidget)
self.keepButton.setGeometry(QtCore.QRect(400, 70, 80, 23))
self.keepButton.setObjectName("keepButton")
self.timeLabel = QtWidgets.QLabel(self.centralwidget)
self.timeLabel.setGeometry(QtCore.QRect(770, 120, 151, 20))
self.timeLabel.setObjectName("timeLabel")
self.speedupButton = QtWidgets.QPushButton(self.centralwidget)
self.speedupButton.setGeometry(QtCore.QRect(640, 70, 41, 23))
self.speedupButton.setObjectName("speedupButton")
self.speeddownButton = QtWidgets.QPushButton(self.centralwidget)
self.speeddownButton.setGeometry(QtCore.QRect(690, 70, 41, 23))
self.speeddownButton.setObjectName("speeddownButton")
self.trimButton = QtWidgets.QPushButton(self.centralwidget)
self.trimButton.setGeometry(QtCore.QRect(490, 70, 80, 23))
self.trimButton.setObjectName("trimButton")
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtWidgets.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 1024, 20))
self.menubar.setObjectName("menubar")
self.menuload_file_work_around = QtWidgets.QMenu(self.menubar)
self.menuload_file_work_around.setObjectName("menuload_file_work_around")
MainWindow.setMenuBar(self.menubar)
self.statusbar = QtWidgets.QStatusBar(MainWindow)
self.statusbar.setObjectName("statusbar")
MainWindow.setStatusBar(self.statusbar)
self.workaround = QtWidgets.QAction(MainWindow)
self.workaround.setObjectName("workaround")
self.menuload_file_work_around.addAction(self.workaround)
self.menubar.addAction(self.menuload_file_work_around.menuAction())
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
def retranslateUi(self, MainWindow):
_translate = QtCore.QCoreApplication.translate
MainWindow.setWindowTitle(_translate("MainWindow", "VideoTrimmer"))
self.playpauseButton.setText(_translate("MainWindow", "Play"))
self.files_selector.setItemText(0, _translate("MainWindow", "hello"))
self.files_selector.setItemText(1, _translate("MainWindow", "world"))
self.label.setText(_translate("MainWindow", "TextLabel"))
self.stopButton.setText(_translate("MainWindow", "Stop"))
self.T1Button.setText(_translate("MainWindow", "T1"))
self.T2Button.setText(_translate("MainWindow", "T2"))
self.cutButton.setText(_translate("MainWindow", "Cut"))
self.keepButton.setText(_translate("MainWindow", "Keep"))
self.timeLabel.setText(_translate("MainWindow", "TextLabel"))
self.speedupButton.setText(_translate("MainWindow", "+"))
self.speeddownButton.setText(_translate("MainWindow", "-"))
self.trimButton.setText(_translate("MainWindow", "Trim"))
self.menuload_file_work_around.setTitle(_translate("MainWindow", "File"))
self.workaround.setText(_translate("MainWindow", "Open folder"))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment