mirror of
https://github.com/blw1138/Zordon.git
synced 2025-12-17 16:58:12 +00:00
83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
import re
|
|
import time
|
|
import ffmpeg
|
|
from utilities.render_worker import *
|
|
|
|
|
|
class FFMPEGRenderWorker(BaseRenderWorker):
|
|
|
|
renderer = 'ffmpeg'
|
|
render_engine = 'ffmpeg'
|
|
|
|
def __init__(self, input_path, output_path, args=None):
|
|
super(FFMPEGRenderWorker, self).__init__(input_path=input_path, output_path=output_path, ignore_extensions=True,
|
|
args=args)
|
|
|
|
self.total_frames = -1
|
|
if os.path.exists(input_path):
|
|
media_stats = ffmpeg.probe(input_path)
|
|
for stream in media_stats['streams']:
|
|
if stream['codec_type'] == 'video':
|
|
self.total_frames = stream['nb_frames']
|
|
break
|
|
|
|
self.frame = 0
|
|
|
|
# Stats
|
|
self.current_frame = -1
|
|
|
|
@classmethod
|
|
def version(cls):
|
|
version = None
|
|
try:
|
|
ver_out = subprocess.check_output([cls.renderer_path(), '-version']).decode('utf-8')
|
|
match = re.match(".*version\s*(\S+)\s*Copyright", ver_out)
|
|
if match:
|
|
version = match.groups()[0]
|
|
except Exception as e:
|
|
logger.error("Failed to get FFMPEG version: {}".format(e))
|
|
return version
|
|
|
|
def _generate_subprocess(self):
|
|
|
|
cmd = [self.renderer_path(), '-y', '-stats', '-i', self.input_path]
|
|
|
|
# Resize frame
|
|
if self.args.get('x_resolution', None) and self.args.get('y_resolution', None):
|
|
cmd.extend(['-vf', f"scale={self.args['x_resolution']}:{self.args['y_resolution']}"])
|
|
|
|
# Convert raw args from string if available
|
|
raw_args = self.args.get('raw', None)
|
|
if raw_args:
|
|
cmd.extend(raw_args.split(' '))
|
|
|
|
# Close with output path
|
|
cmd.append(self.output_path)
|
|
return cmd
|
|
|
|
def percent_complete(self):
|
|
return max(float(self.current_frame) / float(self.total_frames), 0.0)
|
|
|
|
def _parse_stdout(self, line):
|
|
pattern = re.compile(r'frame=\s*(?P<current_frame>\d+)\s*fps.*time=(?P<time_elapsed>\S+)')
|
|
found = pattern.search(line)
|
|
if found:
|
|
stats = found.groupdict()
|
|
self.current_frame = stats['current_frame']
|
|
time_elapsed = stats['time_elapsed']
|
|
elif "not found" in line:
|
|
self.errors.append(line)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
logging.basicConfig(format='%(asctime)s - %(message)s', datefmt='%d-%b-%y %H:%M:%S', level=logging.DEBUG)
|
|
|
|
test_movie = '/Users/brettwilliams/Desktop/dark_knight_rises.mp4'
|
|
|
|
r = FFMPEGRenderWorker(test_movie, '/Users/brettwilliams/Desktop/test-ffmpeg.mp4', args=['-c:v', 'libx265', '-vtag', 'hvc1'])
|
|
# r = FFMPEGRenderer(test_movie, '/Users/brettwilliams/Desktop/dark_knight_rises-output.mp4')
|
|
r.start()
|
|
while r.is_running():
|
|
time.sleep(1)
|