mirror of
https://github.com/langhuihui/monibuca.git
synced 2025-12-24 13:48:04 +08:00
8a9fffb987
- Refactor frame converter implementation - Update mp4 track to use ICodex - General refactoring and code improvements 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
86 lines
2.7 KiB
Go
86 lines
2.7 KiB
Go
package flv
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"net"
|
|
|
|
"m7s.live/v5"
|
|
. "m7s.live/v5/pkg"
|
|
rtmp "m7s.live/v5/plugin/rtmp/pkg"
|
|
)
|
|
|
|
type Live struct {
|
|
b [15]byte
|
|
Subscriber *m7s.Subscriber
|
|
WriteFlvTag func(net.Buffers) error
|
|
}
|
|
|
|
func (task *Live) WriteFlvHeader() (err error) {
|
|
audioCtx, videoCtx := task.Subscriber.Publisher.GetAudioCodecCtx(), task.Subscriber.Publisher.GetVideoCodecCtx()
|
|
hasAudio, hasVideo := audioCtx != nil && task.Subscriber.SubAudio, videoCtx != nil && task.Subscriber.SubVideo
|
|
var amf rtmp.AMF
|
|
amf.Marshal("onMetaData")
|
|
metaData := rtmp.EcmaArray{
|
|
"MetaDataCreator": "m7s" + m7s.Version,
|
|
"hasVideo": hasVideo,
|
|
"hasAudio": hasAudio,
|
|
"hasMatadata": true,
|
|
"canSeekToEnd": false,
|
|
"duration": 0,
|
|
"hasKeyFrames": 0,
|
|
"framerate": 0,
|
|
"videodatarate": 0,
|
|
"filesize": 0,
|
|
}
|
|
var flags byte
|
|
if hasAudio {
|
|
flags |= (1 << 2)
|
|
metaData["audiocodecid"] = int(rtmp.ParseAudioCodec(audioCtx.FourCC()))
|
|
ctx := audioCtx.(IAudioCodecCtx)
|
|
metaData["audiosamplerate"] = ctx.GetSampleRate()
|
|
metaData["audiosamplesize"] = ctx.GetSampleSize()
|
|
metaData["stereo"] = ctx.GetChannels() == 2
|
|
}
|
|
if hasVideo {
|
|
flags |= 1
|
|
metaData["videocodecid"] = int(rtmp.ParseVideoCodec(videoCtx.FourCC()))
|
|
ctx := videoCtx.(IVideoCodecCtx)
|
|
metaData["width"] = ctx.Width()
|
|
metaData["height"] = ctx.Height()
|
|
}
|
|
var data = amf.Marshal(metaData)
|
|
WriteFLVTagHead(FLV_TAG_TYPE_SCRIPT, 0, uint32(len(data)), task.b[4:])
|
|
defer binary.BigEndian.PutUint32(task.b[:4], uint32(len(data))+11)
|
|
return task.WriteFlvTag(net.Buffers{[]byte{'F', 'L', 'V', 0x01, flags, 0, 0, 0, 9, 0, 0, 0, 0}, task.b[4:], data})
|
|
}
|
|
|
|
func (task *Live) rtmpData2FlvTag(t byte, data *rtmp.RTMPData, ts uint32) error {
|
|
WriteFLVTagHead(t, ts, uint32(data.Size), task.b[4:])
|
|
defer binary.BigEndian.PutUint32(task.b[:4], uint32(data.Size)+11)
|
|
return task.WriteFlvTag(append(net.Buffers{task.b[:]}, data.Memory.Buffers...))
|
|
}
|
|
|
|
func (task *Live) WriteAudioTag(data *rtmp.AudioFrame, ts uint32) error {
|
|
return task.rtmpData2FlvTag(FLV_TAG_TYPE_AUDIO, (*rtmp.RTMPData)(data), ts)
|
|
}
|
|
|
|
func (task *Live) WriteVideoTag(data *rtmp.VideoFrame, ts uint32) error {
|
|
return task.rtmpData2FlvTag(FLV_TAG_TYPE_VIDEO, (*rtmp.RTMPData)(data), ts)
|
|
}
|
|
|
|
func (task *Live) Run() (err error) {
|
|
err = task.WriteFlvHeader()
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = m7s.PlayBlock(task.Subscriber, func(audio *rtmp.AudioFrame) error {
|
|
return task.WriteAudioTag(audio, task.Subscriber.AudioReader.AbsTime)
|
|
}, func(video *rtmp.VideoFrame) error {
|
|
return task.WriteVideoTag(video, task.Subscriber.VideoReader.AbsTime)
|
|
})
|
|
if err != nil {
|
|
return
|
|
}
|
|
return task.WriteFlvTag(net.Buffers{task.b[:4]})
|
|
}
|