-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
format_fmp4_segment.go
96 lines (77 loc) · 1.76 KB
/
format_fmp4_segment.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
package record
import (
"io"
"os"
"time"
"github.com/bluenviron/mediacommon/pkg/formats/fmp4"
"github.com/bluenviron/mediacommon/pkg/formats/fmp4/seekablebuffer"
"github.com/bluenviron/mediamtx/internal/logger"
)
func writeInit(f io.Writer, tracks []*formatFMP4Track) error {
fmp4Tracks := make([]*fmp4.InitTrack, len(tracks))
for i, track := range tracks {
fmp4Tracks[i] = track.initTrack
}
init := fmp4.Init{
Tracks: fmp4Tracks,
}
var buf seekablebuffer.Buffer
err := init.Marshal(&buf)
if err != nil {
return err
}
_, err = f.Write(buf.Bytes())
return err
}
type formatFMP4Segment struct {
f *formatFMP4
startDTS time.Duration
startNTP time.Time
path string
fi *os.File
curPart *formatFMP4Part
}
func (s *formatFMP4Segment) initialize() {
}
func (s *formatFMP4Segment) close() error {
var err error
if s.curPart != nil {
err = s.curPart.close()
}
if s.fi != nil {
s.f.a.agent.Log(logger.Debug, "closing segment %s", s.path)
err2 := s.fi.Close()
if err == nil {
err = err2
}
if err2 == nil {
s.f.a.agent.OnSegmentComplete(s.path)
}
}
return err
}
func (s *formatFMP4Segment) record(track *formatFMP4Track, sample *sample) error {
if s.curPart == nil {
s.curPart = &formatFMP4Part{
s: s,
sequenceNumber: s.f.nextSequenceNumber,
startDTS: sample.dts,
}
s.curPart.initialize()
s.f.nextSequenceNumber++
} else if s.curPart.duration() >= s.f.a.agent.PartDuration {
err := s.curPart.close()
s.curPart = nil
if err != nil {
return err
}
s.curPart = &formatFMP4Part{
s: s,
sequenceNumber: s.f.nextSequenceNumber,
startDTS: sample.dts,
}
s.curPart.initialize()
s.f.nextSequenceNumber++
}
return s.curPart.record(track, sample)
}