mirror of
https://github.com/kkroening/ffmpeg-python.git
synced 2025-04-05 04:22:51 +08:00
55 lines
2.3 KiB
Python
55 lines
2.3 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import random
|
|
import re
|
|
import subprocess
|
|
from builtins import bytes, range, str
|
|
|
|
import ffmpeg
|
|
import pytest
|
|
from ffmpeg import nodes
|
|
|
|
try:
|
|
import mock # python 2
|
|
except ImportError:
|
|
from unittest import mock # python 3
|
|
|
|
|
|
def pip_to_rtmp(url):
|
|
""" Capture Facetime camera and screen in macOS and stream it to facebook
|
|
"""
|
|
facetime_camera_input = (
|
|
ffmpeg
|
|
.input('FaceTime:1', format='avfoundation', pix_fmt='uyvy422', framerate=30, s='320x240', probesize='200M')
|
|
)
|
|
|
|
audio = facetime_camera_input.audio
|
|
|
|
# ffmpeg in MacOS AVFoundation MUST increase the thread_queue_size in order to handle PIP
|
|
video = (
|
|
ffmpeg
|
|
.header(thread_queue_size='512', vsync='2')
|
|
.input('1:1', format='avfoundation',
|
|
pix_fmt='uyvy422', framerate=30, probesize='200M')
|
|
.overlay(facetime_camera_input)
|
|
|
|
)
|
|
output = (
|
|
ffmpeg
|
|
.output(video, audio, url, vsync='2', s='1280x720', pix_fmt='yuv420p', video_bitrate='1500000', f='flv',
|
|
vcodec='libx264', preset='fast', x264opts='keyint=15', g='30', ac='2', ar='48000', acodec="aac", audio_bitrate="128000")
|
|
)
|
|
|
|
return output.overwrite_output().compile()
|
|
|
|
|
|
def test_rtmp():
|
|
expected_result = ['ffmpeg', '-thread_queue_size', '512', '-vsync', '2', '-f', 'avfoundation', '-framerate', '30', '-pix_fmt', 'uyvy422', '-probesize', '200M', '-i', '1:1', '-f', 'avfoundation', '-framerate', '30', '-pix_fmt', 'uyvy422', '-probesize', '200M', '-s', '320x240', '-i', 'FaceTime:1', '-filter_complex', '[0][1]overlay=eof_action=repeat[s0]',
|
|
'-map', '[s0]', '-map', '1:a', '-f', 'flv', '-b:v', '1500000', '-b:a', '128000', '-ac', '2', '-acodec', 'aac', '-ar', '48000', '-g', '30', '-pix_fmt', 'yuv420p', '-preset', 'fast', '-s', '1280x720', '-vcodec', 'libx264', '-vsync', '2', '-x264opts', 'keyint=15', 'rtmps://live-api-s.facebook.com:443/rtmp/input_your_facebook_stream_key_here', '-y']
|
|
# Your facebook key should look like: 123456789012345?s_bl=1&s_ps=1&s_sml=1&s_sw=0&s_vt=api-s&a=AbCdEfGhiJK12345
|
|
your_facebook_key = 'input_your_facebook_stream_key_here'
|
|
facebook_stream_url = 'rtmps://live-api-s.facebook.com:443/rtmp/{}'.format(your_facebook_key)
|
|
result = pip_to_rtmp(facebook_stream_url)
|
|
assert result == expected_result
|