just updates
This commit is contained in:
144
quality_test_new_threading.py
Normal file
144
quality_test_new_threading.py
Normal file
@@ -0,0 +1,144 @@
|
||||
import subprocess
|
||||
import sys
|
||||
from pathlib import Path
|
||||
import os
|
||||
from time import sleep
|
||||
from threading import Thread
|
||||
import threading
|
||||
import re
|
||||
|
||||
|
||||
|
||||
max_threads=4
|
||||
i=0
|
||||
output_folder_name ="/bulk_convert_out/"
|
||||
|
||||
#droppedFile = sys.argv[1]
|
||||
#droppedName = Path(droppedFile).name
|
||||
#originalfiles = sys.orig_argv[2:]
|
||||
|
||||
#long path shenanigans
|
||||
#joined = (' '.join(originalfiles))
|
||||
#re.sub (' +', ' ', joined)
|
||||
#listed = re.split(r'(.+?mkv)', joined)
|
||||
#listed = list(filter(None, listed))
|
||||
#listed = [i.lstrip() for i in listed]
|
||||
|
||||
listed= ['test.mkv0out.mkv', "test.mkv1out.mkv","test.mkv2out.mkv","test.mkv3out.mkv","test.mkv4out.mkv","test.mkv5out.mkv"]
|
||||
#get path of script location and remove the tail
|
||||
head_tail = os.path.split(sys.argv[0])
|
||||
output_path = head_tail[0]
|
||||
|
||||
|
||||
#skip first part of sys.argv since it points to script location
|
||||
#arguments = sys.argv[1:]
|
||||
#arguments = droppedFile
|
||||
|
||||
#check if output folder exists, otherwise create it
|
||||
isExist = os.path.exists(output_path+output_folder_name)
|
||||
if not isExist:
|
||||
os.makedirs(output_path+output_folder_name)
|
||||
|
||||
org_count = len(listed)
|
||||
finished_count = int()
|
||||
|
||||
def subproc_ffmpeg(file="",crf_value="", output_path="", output_folder_name=""):
|
||||
basename = os.path.basename(file)
|
||||
print(basename)
|
||||
print(file)
|
||||
print("printing output path: ", output_path)
|
||||
print("printing folder_name: ", output_folder_name)
|
||||
return 'ffmpeg.exe -i "{file}" -map 0:v -c:v libx265 -preset slow -crf "{crf_value}" -x265-params "no-sao=1:bframes=8:psy-rd=2:psy-rdoq=1.5:aq-mode=3:ref=6" -pix_fmt yuv420p10le -map 0:a -c:a libopus -b:a 192k -f matroska "{output_path}{output_folder_name}crf-{crf_value}_{basename}"'.format(file=file, crf_value=crf_value, output_path=output_path, output_folder_name=output_folder_name, basename=basename)
|
||||
|
||||
|
||||
|
||||
print("CRF granuality? 1 or 0.1: ")
|
||||
while True:
|
||||
crf_gran=float(input("> "))
|
||||
if crf_gran == 1:
|
||||
large_gran = True
|
||||
print("Stepping in whole values")
|
||||
break
|
||||
if crf_gran == 0.1:
|
||||
large_gran = False
|
||||
print("stepping in decimal values")
|
||||
break
|
||||
print("Invalid choice, try again")
|
||||
|
||||
|
||||
|
||||
|
||||
crf_range = [int(z) for z in input ("CRF range to test? Seperate with dash (eg: 15-17): ").split('-')]
|
||||
|
||||
|
||||
class CustomThread(Thread):
|
||||
# constructor
|
||||
def __init__(self):
|
||||
|
||||
# execute the base constructor
|
||||
Thread.__init__(self)
|
||||
# set a default value
|
||||
|
||||
|
||||
def run(self):
|
||||
|
||||
#sleep(1)
|
||||
global i
|
||||
global finished_count
|
||||
global large_gran
|
||||
global crf_range
|
||||
file=listed.pop(0)
|
||||
basename = os.path.basename(file)
|
||||
print(file)
|
||||
if large_gran == True:
|
||||
while crf_range[0] <= crf_range[1]:
|
||||
crf_value = crf_range[0]
|
||||
for file in listed:
|
||||
|
||||
subprocess.run(subproc_ffmpeg(file, crf_value, output_path, output_folder_name))
|
||||
crf_range[0] += 1
|
||||
if large_gran == False:
|
||||
while crf_range[0] < crf_range[1]:
|
||||
crf_value = round(crf_range[0], 1)
|
||||
for file in listed:
|
||||
|
||||
subprocess.run(subproc_ffmpeg(file, crf_value, output_path, output_folder_name))
|
||||
crf_range[0] += 0.1
|
||||
|
||||
|
||||
#args = 'ffmpeg.exe -i "{file}" -fps_mode passthrough -map_metadata -1 -map 0:v:0 -c:v libx265 -preset slow -crf 14.5 -x265-params "no-sao=1:bframes=8:psy-rd=2:psy-rdoq=1.5:aq-mode=3:ref=6:deblock=-1,-1" -pix_fmt yuv420p10le -map 0:a -c:a libopus -ac 6 -b:a 576k -map 0:s -c:s copy -f matroska "{output_path}{output_folder_name}{basename}"'.format(file=file, output_path=output_path, output_folder_name=output_folder_name, basename=basename)
|
||||
# print(args)
|
||||
#results = subprocess.Popen(args, stdout=subprocess.PIPE)
|
||||
#streeamdata = results.communicate()[0]
|
||||
#data = results.wait()
|
||||
|
||||
|
||||
if data == 0:
|
||||
i=i-1
|
||||
finished_count=finished_count+1
|
||||
|
||||
|
||||
|
||||
while listed != False:
|
||||
if i < max_threads:
|
||||
global self
|
||||
|
||||
if bool(listed) != False:
|
||||
t1 = CustomThread()
|
||||
t1.start()
|
||||
if bool(listed) == False:
|
||||
while bool(threading.active_count() > 1):
|
||||
print("Finished",finished_count,"out of", org_count," ")
|
||||
sleep(1)
|
||||
t1.join()
|
||||
print("Completed",finished_count,"out of",org_count, "inputs Successfully")
|
||||
input("waiting")
|
||||
exit()
|
||||
|
||||
print("incrementing i in loop")
|
||||
i=i+1
|
||||
|
||||
else:
|
||||
print("Finished ",finished_count,"out of ", org_count)
|
||||
sleep(1)
|
||||
print("continuing ")
|
||||
Reference in New Issue
Block a user