2022-08-14 11:01:06 +00:00
|
|
|
package core
|
|
|
|
|
|
|
|
import (
|
2022-11-02 19:52:12 +00:00
|
|
|
"fmt"
|
|
|
|
|
2022-08-14 11:01:06 +00:00
|
|
|
"github.com/aler9/gortsplib"
|
2022-11-15 22:46:40 +00:00
|
|
|
"github.com/aler9/gortsplib/pkg/rtpcodecs/rtpmpeg4audio"
|
2022-08-14 11:01:06 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type streamTrackMPEG4Audio struct {
|
2022-11-02 19:52:12 +00:00
|
|
|
track *gortsplib.TrackMPEG4Audio
|
|
|
|
encoder *rtpmpeg4audio.Encoder
|
|
|
|
decoder *rtpmpeg4audio.Decoder
|
2022-08-14 11:01:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func newStreamTrackMPEG4Audio(
|
|
|
|
track *gortsplib.TrackMPEG4Audio,
|
2022-11-11 10:59:52 +00:00
|
|
|
allocateEncoder bool,
|
2022-08-14 11:01:06 +00:00
|
|
|
) *streamTrackMPEG4Audio {
|
|
|
|
t := &streamTrackMPEG4Audio{
|
2022-11-02 19:52:12 +00:00
|
|
|
track: track,
|
2022-08-14 11:01:06 +00:00
|
|
|
}
|
|
|
|
|
2022-11-11 10:59:52 +00:00
|
|
|
if allocateEncoder {
|
2022-11-15 22:46:40 +00:00
|
|
|
t.encoder = track.CreateEncoder()
|
2022-08-14 11:01:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return t
|
|
|
|
}
|
|
|
|
|
2022-11-02 19:52:12 +00:00
|
|
|
func (t *streamTrackMPEG4Audio) generateRTPPackets(tdata *dataMPEG4Audio) error {
|
|
|
|
pkts, err := t.encoder.Encode(tdata.aus, tdata.pts)
|
2022-08-14 11:01:06 +00:00
|
|
|
if err != nil {
|
2022-11-02 19:52:12 +00:00
|
|
|
return err
|
2022-08-14 11:01:06 +00:00
|
|
|
}
|
|
|
|
|
2022-11-02 19:52:12 +00:00
|
|
|
tdata.rtpPackets = pkts
|
|
|
|
return nil
|
2022-08-14 11:01:06 +00:00
|
|
|
}
|
|
|
|
|
2022-11-02 19:52:12 +00:00
|
|
|
func (t *streamTrackMPEG4Audio) onData(dat data, hasNonRTSPReaders bool) error {
|
|
|
|
tdata := dat.(*dataMPEG4Audio)
|
|
|
|
|
|
|
|
if tdata.rtpPackets != nil {
|
|
|
|
pkt := tdata.rtpPackets[0]
|
|
|
|
|
|
|
|
// remove padding
|
|
|
|
pkt.Header.Padding = false
|
|
|
|
pkt.PaddingSize = 0
|
|
|
|
|
|
|
|
if pkt.MarshalSize() > maxPacketSize {
|
|
|
|
return fmt.Errorf("payload size (%d) is greater than maximum allowed (%d)",
|
|
|
|
pkt.MarshalSize(), maxPacketSize)
|
|
|
|
}
|
|
|
|
|
|
|
|
// decode from RTP
|
|
|
|
if hasNonRTSPReaders {
|
|
|
|
if t.decoder == nil {
|
2022-11-15 22:46:40 +00:00
|
|
|
t.decoder = t.track.CreateDecoder()
|
2022-11-02 19:52:12 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
aus, pts, err := t.decoder.Decode(pkt)
|
|
|
|
if err != nil {
|
2022-11-03 15:05:46 +00:00
|
|
|
if err == rtpmpeg4audio.ErrMorePacketsNeeded {
|
|
|
|
return nil
|
|
|
|
}
|
2022-11-02 19:52:12 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
tdata.aus = aus
|
|
|
|
tdata.pts = pts
|
|
|
|
}
|
|
|
|
|
|
|
|
// route packet as is
|
|
|
|
return nil
|
2022-08-14 11:01:06 +00:00
|
|
|
}
|
2022-11-02 19:52:12 +00:00
|
|
|
|
|
|
|
return t.generateRTPPackets(tdata)
|
2022-08-14 11:01:06 +00:00
|
|
|
}
|