mirror of
https://github.com/aler9/gortsplib
synced 2025-10-04 23:02:45 +08:00
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"os"
|
|
|
|
"github.com/bluenviron/gortsplib/v4/pkg/format"
|
|
"github.com/bluenviron/mediacommon/v2/pkg/formats/mpegts"
|
|
)
|
|
|
|
func multiplyAndDivide(v, m, d int64) int64 {
|
|
secs := v / d
|
|
dec := v % d
|
|
return (secs*m + dec*m/d)
|
|
}
|
|
|
|
// mpegtsMuxer allows to save a MPEG-4 audio stream into a MPEG-TS file.
|
|
type mpegtsMuxer struct {
|
|
fileName string
|
|
format format.Format
|
|
track *mpegts.Track
|
|
|
|
f *os.File
|
|
b *bufio.Writer
|
|
w *mpegts.Writer
|
|
}
|
|
|
|
// initialize initializes a mpegtsMuxer.
|
|
func (e *mpegtsMuxer) initialize() error {
|
|
var err error
|
|
e.f, err = os.Create(e.fileName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
e.b = bufio.NewWriter(e.f)
|
|
|
|
e.w = &mpegts.Writer{W: e.b, Tracks: []*mpegts.Track{e.track}}
|
|
err = e.w.Initialize()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// close closes all the mpegtsMuxer resources.
|
|
func (e *mpegtsMuxer) close() {
|
|
e.b.Flush() //nolint:errcheck
|
|
e.f.Close()
|
|
}
|
|
|
|
// writeMPEG4Audio writes MPEG-4 audio access units into MPEG-TS.
|
|
func (e *mpegtsMuxer) writeMPEG4Audio(aus [][]byte, pts int64) error {
|
|
return e.w.WriteMPEG4Audio(e.track, multiplyAndDivide(pts, 90000, int64(e.format.ClockRate())), aus)
|
|
}
|