43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package media
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/bluenviron/gortsplib/v4"
|
|
"github.com/bluenviron/gortsplib/v4/pkg/description"
|
|
"github.com/bluenviron/gortsplib/v4/pkg/format"
|
|
"github.com/bluenviron/gortsplib/v4/pkg/format/rtplpcm"
|
|
"github.com/pion/rtp"
|
|
"log"
|
|
)
|
|
|
|
// CheckG711Format finds the H264 media and format.
|
|
func CheckG711Format(desc *description.Session) (*format.G711, *description.Media, error) {
|
|
var g711Format *format.G711
|
|
g711Media := desc.FindFormat(&g711Format)
|
|
if g711Media == nil {
|
|
return nil, nil, errors.New("media G711 not found")
|
|
}
|
|
|
|
return g711Format, g711Media, nil
|
|
}
|
|
|
|
// ProcG711 processes G711 flow and returns PTS and AU.
|
|
func ProcG711(c *gortsplib.Client, g711Media *description.Media, g711RTPDec *rtplpcm.Decoder, pkt *rtp.Packet) (
|
|
int64, []byte, error) {
|
|
// Decode timestamp.
|
|
pts, ok := c.PacketPTS2(g711Media, pkt)
|
|
if !ok {
|
|
log.Printf("waiting for timestamp\n")
|
|
return 0, nil, nil
|
|
}
|
|
|
|
// Extract access unit from RTP packets.
|
|
au, err := g711RTPDec.Decode(pkt)
|
|
if err != nil {
|
|
return 0, nil, fmt.Errorf("decoding G711 RTP packet: %w", err)
|
|
}
|
|
|
|
return pts, au, nil
|
|
}
|