lal/pkg/base/avpacket.go

// Copyright 2020, 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 base

import (
	"encoding/hex"
	"fmt"
	"github.com/q191201771/naza/pkg/nazabytes"
)

// ---------------------------------------------------------------------------------------------------------------------

type AvPacketPt int

const (
	AvPacketPtUnknown AvPacketPt = -1
	AvPacketPtAvc     AvPacketPt = 96 // h264
	AvPacketPtHevc    AvPacketPt = 98 // h265
	AvPacketPtAac     AvPacketPt = 97
)

func (a AvPacketPt) ReadableString() string {
	switch a {
	case AvPacketPtUnknown:
		return "unknown"
	case AvPacketPtAvc:
		return "h264"
	case AvPacketPtHevc:
		return "h265"
	case AvPacketPtAac:
		return "aac"
	}
	return ""
}

// ---------------------------------------------------------------------------------------------------------------------

// AvPacket
//
// 不同场景使用时，字段含义可能不同。
// 使用AvPacket的地方，应注明各字段的含义。
//
//
//
type AvPacket struct {
	PayloadType AvPacketPt
	Timestamp   int64 // 如无特殊说明，此字段是Dts
	Pts         int64
	Payload     []byte
}

func (packet *AvPacket) IsAudio() bool {
	return packet.PayloadType == AvPacketPtAac
}

func (packet *AvPacket) IsVideo() bool {
	return packet.PayloadType == AvPacketPtAvc || packet.PayloadType == AvPacketPtHevc
}

func (packet *AvPacket) DebugString() string {
	return fmt.Sprintf("[%p] type=%s, timestamp=%d, pts=%d, len=%d, payload=%s",
		packet, packet.PayloadType.ReadableString(), packet.Timestamp, packet.Pts, len(packet.Payload), hex.Dump(nazabytes.Prefix(packet.Payload, 64)))
}

// ---------------------------------------------------------------------------------------------------------------------

type OnAvPacketFunc func(packet *AvPacket)