ytp.py

from pathHelper import *
from os import path, makedirs, system, remove, rename
from shutil import rmtree
from subprocess import getoutput
from math import floor
from random import uniform as r

def constrain(val, min_val, max_val):
    if val == None:
        return None
    if type(val)     == str:
        val     = float(val    )
    if type(min_val) == str:
        min_val = float(min_val)
    if type(max_val) == str:
        max_val = float(max_val)
    return min(max_val, max(min_val, val))

def ytp(name, itters, hasAudio):
	itters = max(1, itters)

	pat = getDir(name)
	e = path.splitext(name)
	e0 = getName(pat)

	t = round(eval(getoutput(["ffprobe", "-v", "error", "-select_streams", "v", "-of", "default=noprint_wrappers=1:nokey=1", "-show_entries", "stream=r_frame_rate", name])))
	d = round(eval(getoutput(["ffprobe", "-v", "error", "-show_entries", "format=duration", "-of", "default=noprint_wrappers=1:nokey=1", name])))

	dirPath = f"{pat}/YTP{e0}"
	if path.isdir(dirPath):
		rmtree(dirPath)
	makedirs(dirPath)

	points = []
	MT = 0.1

	inc = d / itters
	for i in range(itters * 2):
		time   = r(1, 6) * r(0.1, r(0.1, r(0.2, 0.6)) * r(r(0, min(10, d / 16)), min(10, d / 6)))
		startPoint = r(MT, MT + inc * 2)
		if startPoint + time > d:
			break
		points.append([startPoint, time])
		MT = startPoint + time

	points = sorted(points)

	t, index = 0, 0
	n = []

	for i in points:
		getoutput(["ffmpeg", "-hide_banner", "-loglevel", "error", "-y", "-i", name, "-ss", str(t), "-to", str(i[0]), "-break_non_keyframes", "1", f"{pat}/YTP{e0}/a{index}.mp4"])
		getoutput(["ffmpeg", "-hide_banner", "-loglevel", "error", "-y", "-i", name, "-ss", str(i[0]), "-to", str(i[0] + i[1]), "-break_non_keyframes", "1", f"{pat}/YTP{e0}/b{index}.mp4"])
		args = ["ffmpeg", "-hide_banner", "-loglevel", "error", "-y", "-i", f"{pat}/YTP{e0}/b{index}.mp4", "-vf", "reverse", f"{pat}/YTP{e0}/r{index}.mp4"]
		if hasAudio:
			args[9:9] = ["-af", "areverse"]
		getoutput(args)

		n.append(f"a{index}.mp4")
		n.append(f"b{index}.mp4")
		n.append(f"r{index}.mp4")
		n.append(f"b{index}.mp4")
		t = i[0] + i[1]
		index += 1
	getoutput(["ffmpeg", "-hide_banner", "-loglevel", "-y", "-i", name, "-ss", str(t), "-t", str(d), "-break_non_keyframes", "1", f"{pat}/YTP{e0}/e.mp4"])
	n.append('e.mp4')

	with open(f'{pat}/YTP{e0}/temp.txt','w') as f:
		for i in n:
			f.write(f"file '{i}'\n")

	getoutput(["ffmpeg", "-hide_banner", "-loglevel", "error", "-f", "concat", "-safe", "0", "-i", f"{pat}/YTP{e0}/temp.txt", "-c", "copy", f"{pat}/_{e0}.mp4"])
	remove(name)
	rename(f"{pat}/_{e0}.mp4", name)
	#shutil.rmtree(e[0])