184 lines
7.3 KiB
Python
184 lines
7.3 KiB
Python
import os
|
|
import subprocess
|
|
import shutil
|
|
import platform
|
|
|
|
segment = ' -init_seg_name chunks/init-stream$RepresentationID$.m4s -media_seg_name chunks/chunk-stream$RepresentationID$-$Number%05d$.m4s'
|
|
if platform.system() == 'Linux':
|
|
segment = ' -init_seg_name chunks/init-stream\$RepresentationID\$.m4s -media_seg_name chunks/chunk-stream\$RepresentationID\$-\$Number%05d\$.m4s'
|
|
|
|
def changeM4SToHTML(mpdpath, chunkspath):
|
|
for filename in os.listdir(chunkspath):
|
|
infilename = os.path.join(chunkspath, filename)
|
|
if not os.path.isfile(infilename): continue
|
|
newname = infilename.replace('.m4s', '.webp')
|
|
_ = os.rename(infilename, newname)
|
|
|
|
# Modify manifest
|
|
with open(mpdpath, 'r') as file :
|
|
filedata = file.read()
|
|
# Replace the target string
|
|
filedata = filedata.replace('.m4s', '.webp')
|
|
# Write the file out again
|
|
with open(mpdpath, 'w') as file:
|
|
file.write(filedata)
|
|
|
|
def createFolder(cdnFolder, resolution):
|
|
if platform.system() == 'Linux':
|
|
if not os.path.exists(os.path.join(cdnFolder, resolution, 'chunks')):
|
|
os.makedirs(os.path.join(cdnFolder, resolution, 'chunks'))
|
|
else:
|
|
if not os.path.exists('chunks'):
|
|
os.makedirs('chunks')
|
|
|
|
if not os.path.exists(os.path.join(cdnFolder, resolution)):
|
|
os.makedirs(os.path.join(cdnFolder, resolution))
|
|
|
|
def create_sprites(cdn_folder):
|
|
"""
|
|
Creates video player sprites
|
|
"""
|
|
video_file = os.path.join(cdn_folder, 'x264.720p.mp4')
|
|
# Generating Sprites
|
|
if not os.path.exists(os.path.join(cdn_folder, 'thumbs.vtt')) and os.path.exists(video_file):
|
|
os.system(f'python makesprites.py "{video_file}"')
|
|
os.rename("thumbs.vtt", os.path.join(cdn_folder, 'thumbs.vtt'))
|
|
os.rename("sprite.jpg", os.path.join(cdn_folder, 'sprite.jpg'))
|
|
shutil.rmtree('thumbs')
|
|
shutil.rmtree('logs')
|
|
return
|
|
|
|
print('Skipped Sprites')
|
|
|
|
def encode_720p_fallback(cdn_folder, video_source, upscale_output, aspect_ratio):
|
|
"""
|
|
Fallback video stream for apple devices
|
|
"""
|
|
output = os.path.join(cdn_folder, 'x264.720p.mp4')
|
|
|
|
if os.path.exists(output):
|
|
print('Skipped 720p Encode')
|
|
return
|
|
|
|
command = (
|
|
f'ffmpeg -v quiet -stats -i "{upscale_output}" -i "{video_source}" '
|
|
'-map 0:v:0 -map 1:a:0 '
|
|
'-c:v libx264 -crf 22 -pix_fmt yuv420p '
|
|
f'-vf scale=1280:720,setsar=1:1 -aspect {aspect_ratio} '
|
|
'-c:a aac -b:a 128k '
|
|
f'-sn -map_metadata -1 -movflags +faststart "{output}"'
|
|
)
|
|
|
|
subprocess.call(command, shell=True)
|
|
|
|
def encode_720p(cdn_folder, video_source, upscale_output, aspect_ratio):
|
|
output = os.path.join(cdn_folder, '720', 'manifest.mpd')
|
|
|
|
if os.path.exists(output):
|
|
print('Skipped 720p DASH Encode')
|
|
return
|
|
|
|
print('Encoding 720p')
|
|
|
|
createFolder(cdn_folder, '720')
|
|
|
|
command = (
|
|
f'ffmpeg -v quiet -stats -i "{upscale_output}" -i "{video_source}" '
|
|
'-map 0:v:0 -map 1:a:0 '
|
|
'-c:v libx264 -crf 22 -preset medium -pix_fmt yuv420p -g 24 -keyint_min 24 -sc_threshold 0 -x264-params keyint=24:min-keyint=24:scenecut=0 '
|
|
f'-vf scale=1280:720,setsar=1:1 -aspect {aspect_ratio} '
|
|
f'{segment} "{output}"'
|
|
)
|
|
|
|
subprocess.call(command, shell=True)
|
|
|
|
if platform.system() != 'Linux':
|
|
shutil.move('chunks', os.path.join(cdn_folder, '720', 'chunks'))
|
|
|
|
changeM4SToHTML(output, os.path.join(cdn_folder, '720', 'chunks'))
|
|
|
|
def EncodeCDN(f, cdnFolder, inputAspect, upscaleOut, interpolateOut, INTERPOLATE_4K, interpolate_4k_output):
|
|
|
|
out1080mpd = os.path.join(cdnFolder, '1080', 'manifest.mpd')
|
|
out1080mpd48 = os.path.join(cdnFolder, '1080i', 'manifest.mpd')
|
|
out2160mpd = os.path.join(cdnFolder, '2160', 'manifest.mpd')
|
|
out2160mpd48 = os.path.join(cdnFolder, '2160i', 'manifest.mpd')
|
|
|
|
# 720p
|
|
encode_720p(cdnFolder, f, upscaleOut, inputAspect)
|
|
encode_720p_fallback(cdnFolder, f, upscaleOut, inputAspect)
|
|
create_sprites(cdnFolder)
|
|
|
|
# 1080p
|
|
if not os.path.exists(out1080mpd):
|
|
print('Encoding 1080p')
|
|
|
|
createFolder(cdnFolder, '1080')
|
|
|
|
subprocess.call('ffmpeg -v quiet -stats -i "' + upscaleOut + '" -i "' + f + '" -map 0:v:0 -map 1:a:0'
|
|
+ ' -c:v libsvtav1 -preset 6 -crf 26 -pix_fmt yuv420p -svtav1-params keyint=1s:tune=0 -vf scale=1920:1080,setsar=1:1 -aspect ' + inputAspect
|
|
+ ' -c:a aac -b:a 128k -ac 2 -sn -map_metadata -1 -seg_duration 10 -use_template 1 -use_timeline 1'
|
|
+ segment + ' "' + out1080mpd + '"', shell=True)
|
|
|
|
if platform.system() != 'Linux':
|
|
shutil.move('chunks', os.path.join(cdnFolder, '1080', 'chunks'))
|
|
|
|
changeM4SToHTML(out1080mpd, os.path.join(cdnFolder, '1080', 'chunks'))
|
|
else:
|
|
print('Skipped 1080p Encode')
|
|
|
|
# 1080p 48fps
|
|
if not os.path.exists(out1080mpd48):
|
|
print('Encoding 1080p 48fps')
|
|
|
|
createFolder(cdnFolder, '1080i')
|
|
|
|
# DASH av1 1080p
|
|
subprocess.call('ffmpeg -v quiet -stats -i "' + interpolateOut + '" -i "' + f + '" -map 0:v:0 -map 1:a:0'
|
|
+ ' -c:v libsvtav1 -preset 6 -crf 26 -pix_fmt yuv420p -svtav1-params keyint=1s:tune=0 -vf scale=1920:1080,setsar=1:1 -aspect ' + inputAspect
|
|
+ ' -c:a aac -b:a 128k -ac 2 -sn -map_metadata -1 -seg_duration 10 -use_template 1 -use_timeline 1'
|
|
+ segment + ' "' + out1080mpd48 + '"', shell=True)
|
|
|
|
if platform.system() != 'Linux':
|
|
shutil.move('chunks', os.path.join(cdnFolder, '1080i', 'chunks'))
|
|
|
|
changeM4SToHTML(out1080mpd48, os.path.join(cdnFolder, '1080i', 'chunks'))
|
|
else:
|
|
print('Skipped 1080p 48fps Encode')
|
|
|
|
# 2160p
|
|
if not os.path.exists(out2160mpd):
|
|
print('Encoding 2160p')
|
|
|
|
createFolder(cdnFolder, '2160')
|
|
|
|
subprocess.call('ffmpeg -v quiet -stats -i "' + upscaleOut + '" -i "' + f + '" -map 0:v:0 -map 1:a:0'
|
|
+ ' -c:v libsvtav1 -preset 6 -crf 26 -pix_fmt yuv420p -svtav1-params keyint=1s:tune=0 -vf "scale=\'min(3840,iw)\':-2,setsar=1:1" -aspect ' + inputAspect
|
|
+ ' -c:a aac -b:a 128k -ac 2 -sn -map_metadata -1 -seg_duration 10 -use_template 1 -use_timeline 1'
|
|
+ segment + ' "' + out2160mpd + '"', shell=True)
|
|
|
|
if platform.system() != 'Linux':
|
|
shutil.move('chunks', os.path.join(cdnFolder, '2160', 'chunks'))
|
|
|
|
changeM4SToHTML(out2160mpd, os.path.join(cdnFolder, '2160', 'chunks'))
|
|
else:
|
|
print('Skipped 2160p Encode')
|
|
|
|
# 2160p 48fps
|
|
if not os.path.exists(out2160mpd48) and INTERPOLATE_4K:
|
|
print('Encoding 2160p 48fps')
|
|
|
|
createFolder(cdnFolder, '2160i')
|
|
|
|
subprocess.call('ffmpeg -v quiet -stats -i "' + interpolate_4k_output + '" -i "' + f + '" -map 0:v:0 -map 1:a:0'
|
|
+ ' -c:v libsvtav1 -preset 6 -crf 26 -pix_fmt yuv420p -svtav1-params keyint=1s:tune=0 -vf "scale=\'min(3840,iw)\':-2,setsar=1:1" -aspect ' + inputAspect
|
|
+ ' -c:a aac -b:a 128k -ac 2 -sn -map_metadata -1 -seg_duration 10 -use_template 1 -use_timeline 1'
|
|
+ segment + ' "' + out2160mpd48 + '"', shell=True)
|
|
|
|
if platform.system() != 'Linux':
|
|
shutil.move('chunks', os.path.join(cdnFolder, '2160i', 'chunks'))
|
|
|
|
changeM4SToHTML(out2160mpd48, os.path.join(cdnFolder, '2160i', 'chunks'))
|
|
else:
|
|
print('Skipped 2160p 48fps Encode')
|