ffmpeg_video.py

import subprocess
import io
import cv2

class FfmpegVideo():
    
    def __init__(self, output_path, framerate=30):
        self.output_path = output_path
        self.closed = False
        self.ffmpeg_proc = subprocess.Popen(f"ffmpeg -y -loglevel error -framerate {framerate} -f image2pipe -c:v mjpeg -i - -c:v libx264 -b:v 10M -preset ultrafast -r {framerate} {output_path}", shell=True, stdin=subprocess.PIPE)#, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
        
    def add_image(self, data):
        if self.closed: return
        
        try:
            self.ffmpeg_proc.stdin.write(data)
        except (ValueError, BrokenPipeError):
            print(f"Broken pipe {self.output_path}")
            self.closed = True
            
    def add_pil_image(self, pil_image):
        with io.BytesIO() as output:
            pil_image.save(output, format="JPEG")
            self.add_image(output.getvalue())
            
    def add_opencv_image(self, opencv_image):
        _, buffer = cv2.imencode(".jpg", opencv_image)
        self.add_image(buffer)
        
    def stop(self):
        if not self.closed:
            self.ffmpeg_proc.stdin.close()
            self.ffmpeg_proc.wait()
            self.closed = True