统一解析处理1078和28181流

2025-10-04 06:46:24 +08:00 · 2024-07-14 17:22:03 +08:00
parent 27da874258
commit ac39deea33
6 changed files with 359 additions and 291 deletions
--- a/stream/source_utils.go
+++ b/stream/source_utils.go
@@ -1,7 +1,11 @@
 package stream

 import (
+	"encoding/hex"
 	"fmt"
+	"github.com/lkmio/avformat/libavc"
+	"github.com/lkmio/avformat/libhevc"
+	"github.com/lkmio/avformat/utils"
 	"github.com/lkmio/lkm/log"
 	"net/url"
 	"strings"
@@ -69,3 +73,93 @@ func ParseUrl(name string) (string, url.Values) {

 	return name, nil
 }
+
+func ExtractVideoPacket(codec utils.AVCodecID, key, extractStream bool, data []byte, pts, dts int64, index, timebase int) (utils.AVStream, utils.AVPacket, error) {
+	var stream utils.AVStream
+
+	if utils.AVCodecIdH264 == codec {
+		//从关键帧中解析出sps和pps
+		if key && extractStream {
+			sps, pps, err := libavc.ParseExtraDataFromKeyNALU(data)
+			if err != nil {
+				log.Sugar.Errorf("从关键帧中解析sps pps失败 data:%s", hex.EncodeToString(data))
+				return nil, nil, err
+			}
+
+			codecData, err := utils.NewAVCCodecData(sps, pps)
+			if err != nil {
+				log.Sugar.Errorf("解析sps pps失败 data:%s sps:%s, pps:%s", hex.EncodeToString(data), hex.EncodeToString(sps), hex.EncodeToString(pps))
+				return nil, nil, err
+			}
+
+			stream = utils.NewAVStream(utils.AVMediaTypeVideo, 0, codec, codecData.AnnexBExtraData(), codecData)
+		}
+
+	} else if utils.AVCodecIdH265 == codec {
+		if key && extractStream {
+			vps, sps, pps, err := libhevc.ParseExtraDataFromKeyNALU(data)
+			if err != nil {
+				log.Sugar.Errorf("从关键帧中解析vps sps pps失败  data:%s", hex.EncodeToString(data))
+				return nil, nil, err
+			}
+
+			codecData, err := utils.NewHEVCCodecData(vps, sps, pps)
+			if err != nil {
+				log.Sugar.Errorf("解析sps pps失败 data:%s vps:%s sps:%s, pps:%s", hex.EncodeToString(data), hex.EncodeToString(vps), hex.EncodeToString(sps), hex.EncodeToString(pps))
+				return nil, nil, err
+			}
+
+			stream = utils.NewAVStream(utils.AVMediaTypeVideo, 0, codec, codecData.AnnexBExtraData(), codecData)
+		}
+
+	}
+
+	packet := utils.NewVideoPacket(data, dts, pts, key, utils.PacketTypeAnnexB, codec, index, timebase)
+	return stream, packet, nil
+}
+
+func ExtractAudioPacket(codec utils.AVCodecID, extractStream bool, data []byte, pts, dts int64, index, timebase int) (utils.AVStream, utils.AVPacket, error) {
+	var stream utils.AVStream
+	var packet utils.AVPacket
+	if utils.AVCodecIdAAC == codec {
+		//必须包含ADTSHeader
+		if len(data) < 7 {
+			return nil, nil, fmt.Errorf("need more data")
+		}
+
+		var skip int
+		header, err := utils.ReadADtsFixedHeader(data)
+		if err != nil {
+			log.Sugar.Errorf("读取ADTSHeader失败 data:%s", hex.EncodeToString(data[:7]))
+			return nil, nil, err
+		} else {
+			skip = 7
+			//跳过ADtsHeader长度
+			if header.ProtectionAbsent() == 0 {
+				skip += 2
+			}
+		}
+
+		if extractStream {
+			configData, err := utils.ADtsHeader2MpegAudioConfigData(header)
+			config, err := utils.ParseMpeg4AudioConfig(configData)
+			println(config)
+			if err != nil {
+				log.Sugar.Errorf("adt头转m4ac失败 data:%s", hex.EncodeToString(data[:7]))
+				return nil, nil, err
+			}
+
+			stream = utils.NewAVStream(utils.AVMediaTypeAudio, index, codec, configData, nil)
+		}
+
+		packet = utils.NewAudioPacket(data[skip:], dts, pts, codec, index, timebase)
+	} else if utils.AVCodecIdPCMALAW == codec || utils.AVCodecIdPCMMULAW == codec {
+		if extractStream {
+			stream = utils.NewAVStream(utils.AVMediaTypeAudio, index, codec, nil, nil)
+		}
+
+		packet = utils.NewAudioPacket(data, dts, pts, codec, index, timebase)
+	}
+
+	return stream, packet, nil
+}