mirror of https://bitbucket.org/ausocean/av.git
120 lines
2.5 KiB
Go
120 lines
2.5 KiB
Go
package flv
|
|
|
|
import (
|
|
"../tools"
|
|
"fmt"
|
|
)
|
|
|
|
const (
|
|
headerLength = 72
|
|
version = 0x01
|
|
maxVideoTagSize = 10000
|
|
maxAudioTagSize = 10000
|
|
)
|
|
|
|
const (
|
|
VideoTagType = 9
|
|
AudioTagType = 8
|
|
KeyFrameType = 1
|
|
InterFrameType = 2
|
|
H264 = 7
|
|
AVCNALU = 1
|
|
SequenceHeader = 0
|
|
DataHeaderLength = 5
|
|
NoTimestampExtension = 0
|
|
AACAudioFormat = 10
|
|
PCMAudioFormat = 0
|
|
)
|
|
|
|
type Header struct {
|
|
AudioFlag bool
|
|
VideoFlag bool
|
|
}
|
|
|
|
func (h *Header) ToByteSlice() (output []byte) {
|
|
output = make([]byte, 0, headerLength)
|
|
output = append(output, []byte{0x46, 0x4C, 0x56,
|
|
version,
|
|
0x00 | tools.BoolToByte(h.AudioFlag)<<2 | tools.BoolToByte(h.VideoFlag),
|
|
0x00, 0x00, 0x00, byte(9),
|
|
}...)
|
|
fmt.Println(output)
|
|
return
|
|
}
|
|
|
|
type VideoTag struct {
|
|
TagType uint8
|
|
DataSize uint32
|
|
Timestamp uint32
|
|
TimestampExtended uint32
|
|
FrameType byte
|
|
Codec byte
|
|
PacketType byte
|
|
CompositionTime uint32
|
|
Data []byte
|
|
PrevTagSize uint32
|
|
}
|
|
|
|
func (t *VideoTag) ToByteSlice() (output []byte) {
|
|
output = make([]byte, 0, maxVideoTagSize)
|
|
output = append(output, []byte{
|
|
byte(t.TagType),
|
|
byte(t.DataSize >> 16),
|
|
byte(t.DataSize >> 8),
|
|
byte(t.DataSize),
|
|
byte(t.Timestamp >> 16),
|
|
byte(t.Timestamp >> 8),
|
|
byte(t.Timestamp),
|
|
byte(t.TimestampExtended),
|
|
0x00, 0x00, 0x00,
|
|
0x00 | byte(t.FrameType << 4) | byte(t.Codec),
|
|
t.PacketType,
|
|
byte(t.CompositionTime >> 16),byte(t.CompositionTime >> 8),byte(t.CompositionTime),
|
|
}...)
|
|
output = append(output, t.Data...)
|
|
output = append(output, []byte{
|
|
byte(t.PrevTagSize >> 24),
|
|
byte(t.PrevTagSize >> 16),
|
|
byte(t.PrevTagSize >> 8),
|
|
byte(t.PrevTagSize),
|
|
}...)
|
|
return
|
|
}
|
|
|
|
type AudioTag struct {
|
|
TagType uint8
|
|
DataSize uint32
|
|
Timestamp uint32
|
|
TimestampExtended uint32
|
|
SoundFormat uint8
|
|
SoundRate uint8
|
|
SoundSize bool
|
|
SoundType bool
|
|
Data []byte
|
|
PrevTagSize uint32
|
|
}
|
|
|
|
func (t *AudioTag) ToByteSlice() (output []byte) {
|
|
output = make([]byte, 0, maxVideoTagSize)
|
|
output = append(output, []byte{
|
|
byte(t.TagType),
|
|
byte(t.DataSize >> 16),
|
|
byte(t.DataSize >> 8),
|
|
byte(t.DataSize),
|
|
byte(t.Timestamp >> 16),
|
|
byte(t.Timestamp >> 8),
|
|
byte(t.Timestamp),
|
|
byte(t.TimestampExtended),
|
|
0x00, 0x00, 0x00,
|
|
byte(t.SoundFormat << 4) | byte(t.SoundRate<<2) | byte(tools.BoolToByte(t.SoundSize)<<1) | byte(tools.BoolToByte(t.SoundType)),
|
|
}...)
|
|
output = append(output, t.Data...)
|
|
output = append(output, []byte{
|
|
byte(t.PrevTagSize >> 24),
|
|
byte(t.PrevTagSize >> 16),
|
|
byte(t.PrevTagSize >> 8),
|
|
byte(t.PrevTagSize),
|
|
}...)
|
|
return
|
|
}
|