You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
lal/pkg/logic/trans.go

71 lines
1.8 KiB
Go

// Copyright 2019, Chef. All rights reserved.
// https://github.com/q191201771/lal
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)
package logic
import (
"github.com/q191201771/lal/pkg/httpflv"
"github.com/q191201771/lal/pkg/rtmp"
)
var Trans trans
type trans struct {
}
func (t trans) FLVTagHeader2RTMPHeader(in httpflv.TagHeader) (out rtmp.Header) {
out.MsgLen = in.DataSize
out.MsgTypeID = in.Type
out.MsgStreamID = rtmp.MSID1
switch in.Type {
case httpflv.TagTypeMetadata:
out.CSID = rtmp.CSIDAMF
case httpflv.TagTypeAudio:
out.CSID = rtmp.CSIDAudio
case httpflv.TagTypeVideo:
out.CSID = rtmp.CSIDVideo
}
out.Timestamp = in.Timestamp
out.TimestampAbs = in.Timestamp
return
}
func (t trans) MakeDefaultRTMPHeader(in rtmp.Header) (out rtmp.Header) {
out.MsgLen = in.MsgLen
out.Timestamp = in.Timestamp
out.TimestampAbs = in.TimestampAbs
out.MsgTypeID = in.MsgTypeID
out.MsgStreamID = rtmp.MSID1
switch in.MsgTypeID {
case rtmp.TypeidDataMessageAMF0:
out.CSID = rtmp.CSIDAMF
case rtmp.TypeidAudio:
out.CSID = rtmp.CSIDAudio
case rtmp.TypeidVideo:
out.CSID = rtmp.CSIDVideo
}
return
}
// 音视频内存块不发生拷贝
func (t trans) FLVTag2RTMPMsg(tag httpflv.Tag) (msg rtmp.AVMsg) {
msg.Header = t.FLVTagHeader2RTMPHeader(tag.Header)
msg.Payload = tag.Raw[11 : 11+msg.Header.MsgLen]
return
}
// 音视频内存块发生拷贝
func (t trans) RTMPMsg2FLVTag(msg rtmp.AVMsg) *httpflv.Tag {
var tag httpflv.Tag
tag.Header.Type = msg.Header.MsgTypeID
tag.Header.DataSize = msg.Header.MsgLen
tag.Header.Timestamp = msg.Header.TimestampAbs
tag.Raw = httpflv.PackHTTPFLVTag(msg.Header.MsgTypeID, msg.Header.TimestampAbs, msg.Payload)
return &tag
}