w33b/Upscale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor & Rewrite

Browse Source
This commit is contained in:
w33b
2026-02-08 12:10:28 +01:00
parent 9c51eee3bc
commit 1dc3be801b
9 changed files with 493 additions and 385 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
Start.py
View File

@@ -1,43 +1,50 @@
import os
import re
from utils.encodeCDN import EncodeCDN
from utils.encodeDDL import EncodeDDL
from utils.interpolate import Interpolate
from utils.interpolate4k import Interpolate4K
from utils.encode_downloads import encode_downloads
from utils.encode_stream import encode_streams
from utils.interpolate import interpolate
from utils.upcale import upscale
from utils.mediainfo import get_aspect_ratio
INTERPOLATE_4K = False
MAX_INPUT_WIDTH = '720'
def create_folder(folder_path):
if not os.path.exists(folder_path):
os.makedirs(folder_path)
for filename in os.listdir('0-Source'):
input_file = os.path.join('0-Source', filename)
if not os.path.isfile(input_file):
continue
if filename == '.gitignore':
continue
input_file = os.path.join('0-Source', filename)
if not os.path.isfile(input_file):
continue
# Parse File Name
temp_name = re.sub(r'\[.*?\]|\(.*?\)', "", filename).rsplit('.', 1)[0].strip()
folder_name = re.sub(r'[^A-Za-z ]+', '', temp_name).strip()
episode_number = re.findall(r'\d+', temp_name)[-1]
try:
hentai_name = re.sub(r'\[.*?\]|\(.*?\)', "", filename).rsplit('.', 1)[0].strip()
folder_name = re.sub(r'[^A-Za-z ]+', '', hentai_name).strip()
episode_number = re.findall(r'\d+', hentai_name)[-1]
except:
print(f"Error Parsing name for: {filename}")
continue
cdn_folder_name = folder_name.replace(" ", ".")
cdn_folder = os.path.join('2-Out', folder_name, cdn_folder_name, 'E' + episode_number)
cdn_folder = os.path.join('2-Out', folder_name, cdn_folder_name, f'E{episode_number}')
muxed_folder = os.path.join('2-Out', folder_name, 'Muxed')
upscale_output_folder = os.path.join('2-Out', folder_name, folder_name + ' [2160p]')
upscale_output = os.path.join(upscale_output_folder, temp_name + ' [4k][HEVC].mkv')
interpolate_output = os.path.join(upscale_output_folder, temp_name + ' [1080p][48fps][HEVC].mkv')
interpolate_4k_output = os.path.join(upscale_output_folder, temp_name + ' [2160p][48fps][HEVC].mkv')
# Output of Upscale and Interpolation
# Hentai Name [2160p]
# -> "Hentai Name [4k][HEVC].mkv"
# -> "Hentai Name [1080p][48fps][HEVC].mkv"
# -> "Hentai Name [2160p][48fps][HEVC].mkv"
upscale_output_folder = os.path.join('2-Out', folder_name, f'{folder_name} [2160p]')
upscale_output = os.path.join(upscale_output_folder, f'{hentai_name} [4k][HEVC].mkv')
interpolate_output = os.path.join(upscale_output_folder, f'{hentai_name} [1080p][48fps][HEVC].mkv')
interpolate_4k_output = os.path.join(upscale_output_folder, f'{hentai_name} [2160p][48fps][HEVC].mkv')
print('Parsed Name: ' + temp_name)
print('Parsed Name: ' + hentai_name)
create_folder(cdn_folder)
create_folder(muxed_folder)
@@ -45,8 +52,14 @@ for filename in os.listdir('0-Source'):
aspect_ratio = get_aspect_ratio(input_file)
upscale(input_file, upscale_output, MAX_INPUT_WIDTH, aspect_ratio)
Interpolate(interpolate_output, upscale_output, temp_name, aspect_ratio)
Interpolate4K(interpolate_4k_output, upscale_output, INTERPOLATE_4K, temp_name)
EncodeDDL(input_file, cdn_folder, folder_name, temp_name, upscale_output, aspect_ratio, interpolate_output, INTERPOLATE_4K, interpolate_4k_output)
EncodeCDN(input_file, cdn_folder, aspect_ratio, upscale_output, interpolate_output, INTERPOLATE_4K, interpolate_4k_output)
upscale(input_file, upscale_output, aspect_ratio)
interpolate(upscale_output, interpolate_output, hentai_name, "1080p", aspect_ratio)
if INTERPOLATE_4K:
interpolate(upscale_output, interpolate_4k_output, hentai_name, "2160p", aspect_ratio)
encode_downloads(folder_name, hentai_name, input_file, upscale_output, interpolate_output, interpolate_4k_output, aspect_ratio)
encode_streams(cdn_folder, input_file, upscale_output, interpolate_output, interpolate_4k_output, aspect_ratio)
else:
encode_downloads(folder_name, hentai_name, input_file, upscale_output, interpolate_output, None, aspect_ratio)
encode_streams(cdn_folder, input_file, upscale_output, interpolate_output, None, aspect_ratio)