樹莓派使用python-librtmp實(shí)現(xiàn)rtmp推流h264的方法
更新時(shí)間：2019年07月22日 10:23:04 作者：西紅柿燉番茄
今天小編就為大家分享一篇樹莓派使用python-librtmp實(shí)現(xiàn)rtmp推流h264的方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
目的是能使用Python進(jìn)行rtmp推流，方便在h264幀里加入彈幕等操作。
librtmp使用的是0.3.0，使用樹莓派noir官方攝像頭適配的。
通過wireshark抓ffmpeg的包一點(diǎn)點(diǎn)改動(dòng)，最終可以在red5和斗魚上推流了。
沒怎么寫過python，有不恰當(dāng)?shù)牡胤秸埌?/p>
上代碼：
# -- coding: utf-8 --
# http://blog.csdn.net/luhanglei
import picamera
import time
import traceback
import ctypes
from librtmp import *
global meta_packet
global start_time
class Writer(): # camera可以通過一個(gè)類文件的對象來輸出，實(shí)現(xiàn)write方法即可
conn = None # rtmp連接
sps = None # 記錄sps幀，發(fā)過以后就不需要再發(fā)了（抓包看到ffmpeg是這樣的）
pps = None # 同上
sps_len = 0 # 同上
pps_len = 0 # 同上
time_stamp = 0
def __init__(self, conn):
self.conn = conn
def write(self, data):
try:
# 尋找h264幀間隔符
indexs = []
index = 0
data_len = len(data)
while index < data_len - 3:
if ord(data[index]) == 0x00 and ord(data[index + 1]) == 0x00 and ord(
data[index + 2]) == 0x00 and ord(data[index + 3]) == 0x01:
indexs.append(index)
index = index + 3
index = index + 1
# 尋找h264幀間隔符 完成
# 通過間隔符個(gè)數(shù)確定類型，樹莓派攝像頭的第一幀是sps+pps同時(shí)發(fā)的
if len(indexs) == 1: # 非sps pps幀
buf = data[4: len(data)] # 裁掉原來的頭（00 00 00 01），把幀內(nèi)容拿出來
buf_len = len(buf)
type = ord(buf[0]) & 0x1f
if type == 0x05: # 關(guān)鍵幀，根據(jù)wire shark抓包結(jié)果，需要拼裝sps pps 幀內(nèi)容 三部分，長度都用4個(gè)字節(jié)表示
body0 = 0x17
data_body_array = [bytes(bytearray(
[body0, 0x01, 0x00, 0x00, 0x00, (self.sps_len >> 24) & 0xff, (self.sps_len >> 16) & 0xff,
(self.sps_len >> 8) & 0xff,
self.sps_len & 0xff])), self.sps,
bytes(bytearray(
[(self.pps_len >> 24) & 0xff, (self.pps_len >> 16) & 0xff, (self.pps_len >> 8) & 0xff,
self.pps_len & 0xff])),
self.pps,
bytes(bytearray(
[(buf_len >> 24) & 0xff, (buf_len >> 16) & 0xff, (buf_len >> 8) & 0xff, (buf_len) & 0xff])),
buf
]
mbody = ''.join(data_body_array)
time_stamp = 0 # 第一次發(fā)出的時(shí)候，發(fā)時(shí)間戳0，此后發(fā)真時(shí)間戳
if self.time_stamp != 0:
time_stamp = int((time.time() - start_time) * 1000)
packet_body = RTMPPacket(type=PACKET_TYPE_VIDEO, format=PACKET_SIZE_LARGE, channel=0x06,
timestamp=time_stamp, body=mbody)
packet_body.packet.m_nInfoField2 = 1
else: # 非關(guān)鍵幀
body0 = 0x27
data_body_array = [bytes(bytearray(
[body0, 0x01, 0x00, 0x00, 0x00, (buf_len >> 24) & 0xff, (buf_len >> 16) & 0xff,
(buf_len >> 8) & 0xff,
(buf_len) & 0xff])), buf]
mbody = ''.join(data_body_array)
# if (self.time_stamp == 0):
self.time_stamp = int((time.time() - start_time) * 1000)
packet_body = RTMPPacket(type=PACKET_TYPE_VIDEO, format=PACKET_SIZE_MEDIUM, channel=0x06,
timestamp=self.time_stamp, body=mbody)
self.conn.send_packet(packet_body)
elif len(indexs) == 2: # sps pps幀
if self.sps is not None:
return
data_body_array = [bytes(bytearray([0x17, 0x00, 0x00, 0x00, 0x00, 0x01]))]
sps = data[indexs[0] + 4: indexs[1]]
sps_len = len(sps)
pps = data[indexs[1] + 4: len(data)]
pps_len = len(pps)
self.sps = sps
self.sps_len = sps_len
self.pps = pps
self.pps_len = pps_len
data_body_array.append(sps[1:4])
data_body_array.append(bytes(bytearray([0xff, 0xe1, (sps_len >> 8) & 0xff, sps_len & 0xff])))
data_body_array.append(sps)
data_body_array.append(bytes(bytearray([0x01, (pps_len >> 8) & 0xff, pps_len & 0xff])))
data_body_array.append(pps)
data_body = ''.join(data_body_array)
body_packet = RTMPPacket(type=PACKET_TYPE_VIDEO, format=PACKET_SIZE_LARGE, channel=0x06,
timestamp=0, body=data_body)
body_packet.packet.m_nInfoField2 = 1
self.conn.send_packet(meta_packet, queue=True)
self.conn.send_packet(body_packet, queue=True)
except Exception, e:
traceback.print_exc()
def flush(self):
pass
def get_property_string(string): # 返回兩字節(jié)string長度及string
length = len(string)
return ''.join([chr((length >> 8) & 0xff), chr(length & 0xff), string])
def get_meta_string(string): # 按照meta packet要求格式返回bytes,帶02前綴
return ''.join([chr(0x02), get_property_string(string)])
def get_meta_double(db):
nums = [0x00]
fp = ctypes.pointer(ctypes.c_double(db))
cp = ctypes.cast(fp, ctypes.POINTER(ctypes.c_longlong))
for i in range(7, -1, -1):
nums.append((cp.contents.value >> (i * 8)) & 0xff)
return ''.join(bytes(bytearray(nums)))
def get_meta_boolean(isTrue):
nums = [0x01]
if (isTrue):
nums.append(0x01)
else:
nums.append(0x00)
return ''.join(bytes(bytearray(nums)))
conn = RTMP(
'rtmp://192.168.199.154/oflaDemo/test', # 推流地址
live=True)
librtmp.RTMP_EnableWrite(conn.rtmp)
conn.connect()
start_time = time.time()
# 拼裝視頻格式的數(shù)據(jù)包
meta_body_array = [get_meta_string('@setDataFrame'), get_meta_string('onMetaData'),
bytes(bytearray([0x08, 0x00, 0x00, 0x00, 0x06])), # 兩個(gè)字符串和ECMA array頭，共計(jì)6個(gè)元素,注釋掉了音頻相關(guān)數(shù)據(jù)
get_property_string('width'), get_meta_double(640.0),
get_property_string('height'), get_meta_double(480.0),
get_property_string('videodatarate'), get_meta_double(0.0),
get_property_string('framerate'), get_meta_double(25.0),
get_property_string('videocodecid'), get_meta_double(7.0),
# get_property_string('audiodatarate'), get_meta_double(125.0),
# get_property_string('audiosamplerate'), get_meta_double(44100.0),
# get_property_string('audiosamplesize'), get_meta_double(16.0),
# get_property_string('stereo'), get_meta_boolean(True),
# get_property_string('audiocodecid'), get_meta_double(10.0),
get_property_string('encoder'), get_meta_string('Lavf57.56.101'),
bytes(bytearray([0x00, 0x00, 0x09]))
]
meta_body = ''.join(meta_body_array)
print meta_body.encode('hex')
meta_packet = RTMPPacket(type=PACKET_TYPE_INFO, format=PACKET_SIZE_LARGE, channel=0x04,
timestamp=0, body=meta_body)
meta_packet.packet.m_nInfoField2 = 1 # 修改stream id
stream = conn.create_stream(writeable=True)
with picamera.PiCamera() as camera:
camera.start_preview()
time.sleep(2)
camera.start_recording(Writer(conn), format='h264', resize=(640, 480), intra_period=25,
quality=25) # 開始錄制，數(shù)據(jù)輸出到Writer的對象里
while True:#永遠(yuǎn)不停止
time.sleep(60)
camera.stop_recording()
camera.stop_preview()
以上這篇樹莓派使用python-librtmp實(shí)現(xiàn)rtmp推流h264的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
最近學(xué)習(xí)了python遍歷目錄,下面這篇文章主要給大家介紹了關(guān)于Pandas快速合并多張excel表格的兩種方法,文中通過實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2023-01-01
這篇文章主要介紹了推薦8款常用的Python GUI圖形界面開發(fā)框架,需要的朋友可以參考下2020-02-02
這篇文章主要介紹了python 利用棧和隊(duì)列模擬遞歸的過程,文中并通過兩段代碼給大家介紹了下遞歸和非遞歸的區(qū)別，需要的朋友可以參考下2018-05-05
想要從文字海洋中捕獲珍珠般的古詩文嗎?通過Python爬取古詩文網(wǎng),你將掌握直搗黃龍的技能,簡單幾步,讓經(jīng)典文學(xué)盡收囊中,為你的項(xiàng)目增添靈氣,需要的朋友可以參考下2024-03-03
教你如何用python開發(fā)一款數(shù)字推盤小游戲
這篇文章主要介紹了教你如何用python開發(fā)一款數(shù)字推盤小游戲,文中有非常詳細(xì)的代碼示例,喜對歡玩小游戲的或者正在學(xué)習(xí)python小游戲開發(fā)的小伙伴們有很好的幫助,需要的朋友可以參考下2021-04-04
Word文檔中的表格可以用于組織和展示數(shù)據(jù),在實(shí)際應(yīng)用過程中,有時(shí)為了調(diào)整表格的結(jié)構(gòu)或適應(yīng)不同的數(shù)據(jù)展示需求,我們可能會(huì)需要插入、刪除行或列,本文提供了幾種使用Python在Word表格中插入或刪除行、列的方法供參考2024-07-07
基于np.arange與np.linspace細(xì)微區(qū)別(數(shù)據(jù)溢出問題)
這篇文章主要介紹了基于np.arange與np.linspace細(xì)微區(qū)別(數(shù)據(jù)溢出問題)，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-05-05
SOLID原則是由Robert C. Martin提出的以首字母縮寫命名的編碼準(zhǔn)則,它代表了五種不同的編碼習(xí)慣,下面這篇文章主要給大家介紹了關(guān)于Python中SOLID原則的相關(guān)資料,需要的朋友可以參考下2023-02-02
python同時(shí)給兩個(gè)收件人發(fā)送郵件的方法
這篇文章主要介紹了python同時(shí)給兩個(gè)收件人發(fā)送郵件的方法,涉及Python使用smtplib包發(fā)送郵件的相關(guān)技巧,需要的朋友可以參考下2015-04-04
Python?watchdog靈活監(jiān)控文件和目錄的變化
Python?Watchdog是一個(gè)強(qiáng)大的Python庫,它提供了簡單而靈活的方式來監(jiān)控文件系統(tǒng)的變化,本文將詳細(xì)介紹Python?Watchdog的用法和功能,包括安裝、基本用法、事件處理以及實(shí)際應(yīng)用場景,并提供豐富的示例代碼2024-01-01