/
keyframe.go
102 lines (85 loc) · 2.03 KB
/
keyframe.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package mp4
import (
"io"
"github.com/AlexxIT/go2rtc/pkg/core"
"github.com/AlexxIT/go2rtc/pkg/h264"
"github.com/AlexxIT/go2rtc/pkg/h265"
"github.com/pion/rtp"
)
type Keyframe struct {
core.SuperConsumer
wr *core.WriteBuffer
muxer *Muxer
}
func NewKeyframe(medias []*core.Media) *Keyframe {
if medias == nil {
medias = []*core.Media{
{
Kind: core.KindVideo,
Direction: core.DirectionSendonly,
Codecs: []*core.Codec{
{Name: core.CodecH264},
{Name: core.CodecH265},
},
},
}
}
cons := &Keyframe{
muxer: &Muxer{},
wr: core.NewWriteBuffer(nil),
}
cons.Medias = medias
return cons
}
func (c *Keyframe) AddTrack(media *core.Media, _ *core.Codec, track *core.Receiver) error {
c.muxer.AddTrack(track.Codec)
init, err := c.muxer.GetInit()
if err != nil {
return err
}
handler := core.NewSender(media, track.Codec)
switch track.Codec.Name {
case core.CodecH264:
handler.Handler = func(packet *rtp.Packet) {
if !h264.IsKeyframe(packet.Payload) {
return
}
// important to use Mutex because right fragment order
b := c.muxer.GetPayload(0, packet)
b = append(init, b...)
if n, err := c.wr.Write(b); err == nil {
c.Send += n
}
}
if track.Codec.IsRTP() {
handler.Handler = h264.RTPDepay(track.Codec, handler.Handler)
} else {
handler.Handler = h264.RepairAVCC(track.Codec, handler.Handler)
}
case core.CodecH265:
handler.Handler = func(packet *rtp.Packet) {
if !h265.IsKeyframe(packet.Payload) {
return
}
// important to use Mutex because right fragment order
b := c.muxer.GetPayload(0, packet)
b = append(init, b...)
if n, err := c.wr.Write(b); err == nil {
c.Send += n
}
}
if track.Codec.IsRTP() {
handler.Handler = h265.RTPDepay(track.Codec, handler.Handler)
}
}
handler.HandleRTP(track)
c.Senders = append(c.Senders, handler)
return nil
}
func (c *Keyframe) WriteTo(wr io.Writer) (int64, error) {
return c.wr.WriteTo(wr)
}
func (c *Keyframe) Stop() error {
_ = c.SuperConsumer.Close()
return c.wr.Close()
}