树莓派使用python-librtmp实现rtmp推流h264的方法
作者:西红柿炖番茄 发布时间:2021-10-24 13:50:15
标签:python,librtmp,rtmp,h264
目的是能使用Python进行rtmp推流,方便在h264帧里加入弹幕等操作。
librtmp使用的是0.3.0,使用树莓派noir官方摄像头适配的。
通过wireshark抓ffmpeg的包一点点改动,最终可以在red5和斗鱼上推流了。
没怎么写过python,有不恰当的地方请包涵。
上代码:
# -- 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可以通过一个类文件的对象来输出,实现write方法即可
conn = None # rtmp连接
sps = None # 记录sps帧,发过以后就不需要再发了(抓包看到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帧间隔符 完成
# 通过间隔符个数确定类型,树莓派摄像头的第一帧是sps+pps同时发的
if len(indexs) == 1: # 非sps pps帧
buf = data[4: len(data)] # 裁掉原来的头(00 00 00 01),把帧内容拿出来
buf_len = len(buf)
type = ord(buf[0]) & 0x1f
if type == 0x05: # 关键帧,根据wire shark抓包结果,需要拼装sps pps 帧内容 三部分,长度都用4个字节表示
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 # 第一次发出的时候,发时间戳0,此后发真时间戳
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: # 非关键帧
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): # 返回两字节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()
# 拼装视频格式的数据包
meta_body_array = [get_meta_string('@setDataFrame'), get_meta_string('onMetaData'),
bytes(bytearray([0x08, 0x00, 0x00, 0x00, 0x06])), # 两个字符串和ECMA array头,共计6个元素,注释掉了音频相关数据
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) # 开始录制,数据输出到Writer的对象里
while True:#永远不停止
time.sleep(60)
camera.stop_recording()
camera.stop_preview()
来源:https://blog.csdn.net/luhanglei/article/details/78403900
0
投稿
猜你喜欢
- A 定义数组有两种方式:DIM和REDIM。DIM定义的是固定个数、数据类型的数组;而REDIM则不同,它可以定义不同类型的数据,也可以定义
- 相比于2018年,在ICLR2019提交论文中,提及不同框架的论文数量发生了极大变化,网友发现,提及tensorflow的论文数量从2018
- Softmax回归函数是用于将分类结果归一化。但它不同于一般的按照比例归一化的方法,它通过对数变换来进行归一化,这样实现了较大的值在归一化过
- 如下所示:<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional
- 1、代码from aip import AipFaceimport cv2import timeimport base64from PIL
- N久没有开始写博客了,总觉得要随便记点东西,岁月蹉跎,曾经搞得一些东西、技术、工具,说丢也就丢了,点点滴滴还是要记录一下吧。。。在windo
- 本文实例讲述了PHP使用自定义key实现对数据加密解密的方法。分享给大家供大家参考,具体如下:客户端和服务端通信时,有个场景很常见,通过一个
- 最近将Jesse James Garrett的《用户体验的要素》一书读了两遍,做一些简要的摘录并添加一些个人注释。当然,一本好书绝对不是简单
- 问题:因为有的友情连接的网站关闭或者网络连接较慢导致连接的LOGO图片显示不出来或者显示很慢.在IE下面老是提示剩下几项没打开,看起来很不舒
- 前言:perl是什么,干什么用的?perl原来设计者的意图是用来处理 字符的,80%的强项是处理字符,当然其它的很多都可以。现在很多网页也是
- 有时候我们需要判断某一个IP地址是否属于一个网段,以决定该用户能否访问系统.比如用户登录的IP是218.6.7.7,而我们的程序必须判断他是
- 实现步骤:图像灰度化边缘检测根据Canny检测得出来的Mat寻找轮廓算出最大轮廓周长or面积根据获取到的最大轮廓下标进行轮廓绘制画出最大矩形
- 一、数据预处理实验数据来自genki4k提取含有完整人脸的图片def init_file(): num = 0&n
- 虚拟环境管理创建虚拟环境#默认路径下创建虚拟环境conda create -n pythonVirtual python=x.x # -n:
- 本文实例讲述了使用symfony命令创建项目的方法。分享给大家供大家参考,具体如下:概况这一章节描述一个Symfony项目的合理结构框架,并
- Jquery中的一些东西学习一下子,补充完善一下,毕竟有些时候没有使用到这个方式很有用,在使用bootstrap table的时候,选择当前
- 前言因为写好了测试xmind脑图后,然后再编写测试用例,实在是太麻烦了,所以我写了一点测试用例后,就网上百度了下,怎么直接把xmind脑图转
- 代码如下:SELECT * FROM Orders WHERE OrderGUID IN('BC71D821-9E25-
- 文件提交页面既已生成,下面任务就很明确了:将提交的文件内容保存到服务器上。 下面我们用两种方法来实现这个功能: 1. 用 PHP 来保存:
- 描述:下午快下班的时候公司供应链部门的同事跑过来问我能不能以程序的方法帮他解决一些excel表格每周都需要手工重复做的事情,Excel 是数