first commit

avformat/avpacket.go

@@ -0,0 +1,87 @@
+package avformat
+
+import (
+	"sync"
+
+	"github.com/Monibuca/engine/pool"
+)
+
+var (
+	AVPacketPool = &sync.Pool{
+		New: func() interface{} {
+			return new(AVPacket)
+		},
+	}
+	SendPacketPool = &sync.Pool{
+		New: func() interface{} {
+			return new(SendPacket)
+		},
+	}
+)
+
+// Video or Audio
+type AVPacket struct {
+	Timestamp     uint32
+	Type          byte //8 audio,9 video
+	IsAACSequence bool
+	IsADTS        bool
+	// Video
+	VideoFrameType byte //4bit
+	IsAVCSequence  bool
+	Payload        []byte
+	RefCount       int //Payload的引用次数
+}
+
+func (av *AVPacket) IsKeyFrame() bool {
+	return av.VideoFrameType == 1 || av.VideoFrameType == 4
+}
+func (av *AVPacket) ADTS2ASC() (tagPacket *AVPacket) {
+	tagPacket = NewAVPacket(FLV_TAG_TYPE_AUDIO)
+	tagPacket.Payload = ADTSToAudioSpecificConfig(av.Payload)
+	tagPacket.IsAACSequence = true
+	ADTSLength := 7 + ((1 - int(av.Payload[1]&1)) << 1)
+	if len(av.Payload) > ADTSLength {
+		av.Payload[0] = 0xAF
+		av.Payload[1] = 0x01 //raw AAC
+		copy(av.Payload[2:], av.Payload[ADTSLength:])
+		av.Payload = av.Payload[:(len(av.Payload) - ADTSLength + 2)]
+	}
+	return
+}
+func (av *AVPacket) Recycle() {
+	if av.RefCount == 0 {
+		return
+	} else if av.RefCount == 1 {
+		av.RefCount = 0
+		pool.RecycleSlice(av.Payload)
+		AVPacketPool.Put(av)
+	} else {
+		av.RefCount--
+	}
+}
+func NewAVPacket(avType byte) (p *AVPacket) {
+	p = AVPacketPool.Get().(*AVPacket)
+	p.Type = avType
+	p.IsAVCSequence = false
+	p.VideoFrameType = 0
+	p.Timestamp = 0
+	p.IsAACSequence = false
+	p.IsADTS = false
+	return
+}
+
+type SendPacket struct {
+	Timestamp uint32
+	Packet    *AVPacket
+}
+
+func (packet *SendPacket) Recycle() {
+	packet.Packet.Recycle()
+	SendPacketPool.Put(packet)
+}
+func NewSendPacket(p *AVPacket, timestamp uint32) (result *SendPacket) {
+	result = SendPacketPool.Get().(*SendPacket)
+	result.Packet = p
+	result.Timestamp = timestamp
+	return
+}