diff --git a/lib/render_workers/ffmpeg_worker.py b/lib/render_workers/ffmpeg_worker.py index fee773f..33ecdf8 100644 --- a/lib/render_workers/ffmpeg_worker.py +++ b/lib/render_workers/ffmpeg_worker.py @@ -1,27 +1,9 @@ #!/usr/bin/env python3 import re -import time from .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) - - stream_info = subprocess.check_output([self.renderer_path(), "-i", # https://stackoverflow.com/a/61604105 - input_path, "-map", "0:v:0", "-c", "copy", "-f", "null", "-y", - "/dev/null"], stderr=subprocess.STDOUT).decode('utf-8') - found_frames = re.findall('frame=\s*(\d+)', stream_info) - self.total_frames = found_frames[-1] if found_frames else '-1' - self.frame = 0 - - # Stats - self.current_frame = -1 +class FFMPEG(BaseRenderEngine): @classmethod def version(cls): @@ -35,18 +17,58 @@ class FFMPEGRenderWorker(BaseRenderWorker): logger.error("Failed to get FFMPEG version: {}".format(e)) return version + @classmethod + def get_encoders(cls): + encoders_raw = subprocess.check_output([cls.renderer_path(), '-encoders'], stderr=subprocess.DEVNULL).decode('utf-8') + pattern = '(?P[VASFXBD.]{6})\s+(?P\S{2,})\s+(?P.*)' + encoders = [m.groupdict() for m in re.finditer(pattern, encoders_raw)] + return encoders + + @classmethod + def get_formats(cls): + formats_raw = subprocess.check_output([cls.renderer_path(), '-formats'], stderr=subprocess.DEVNULL).decode('utf-8') + pattern = '(?P[DE]{1,2})\s+(?P\S{2,})\s+(?P.*)' + formats = [m.groupdict() for m in re.finditer(pattern, formats_raw)] + return formats + + @classmethod + def full_report(cls): + return {'version': cls.version(), + 'help_text': cls.get_help(), + 'encoders': cls.get_encoders(), + 'formats': cls.get_formats()} + + +class FFMPEGRenderWorker(BaseRenderWorker): + + 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) + + stream_info = subprocess.check_output([self.engine.renderer_path(), "-i", # https://stackoverflow.com/a/61604105 + input_path, "-map", "0:v:0", "-c", "copy", "-f", "null", "-y", + "/dev/null"], stderr=subprocess.DEVNULL).decode('utf-8') + found_frames = re.findall('frame=\s*(\d+)', stream_info) + self.total_frames = found_frames[-1] if found_frames else '-1' + self.frame = 0 + + # Stats + self.current_frame = -1 + def generate_worker_subprocess(self): - cmd = [self.renderer_path(), '-y', '-stats', '-i', self.input_path] + cmd = [self.engine.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.get_raw_args() + raw_args = self.args.get('raw', None) if raw_args: - cmd.extend(raw_args) + cmd.extend(raw_args.split(' ')) # Close with output path cmd.append(self.output_path) @@ -67,12 +89,15 @@ class FFMPEGRenderWorker(BaseRenderWorker): 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' + print(FFMPEG.full_report()) - 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) + # 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)