add RTP/Opus decoder and encoder (#151)

This commit is contained in:
Alessandro Ros
2022-11-14 17:12:58 +01:00
committed by GitHub
parent 7951b2e4af
commit 0ad09c2184
7 changed files with 325 additions and 0 deletions

View File

@@ -0,0 +1,81 @@
package main
import (
"log"
"github.com/aler9/gortsplib"
"github.com/aler9/gortsplib/pkg/rtpopus"
"github.com/aler9/gortsplib/pkg/url"
)
// This example shows how to
// 1. connect to a RTSP server and read all tracks on a path
// 2. check if there's an Opus track
// 3. get Opus packets of that track
func main() {
c := gortsplib.Client{}
// parse URL
u, err := url.Parse("rtsp://localhost:8554/mystream")
if err != nil {
panic(err)
}
// connect to the server
err = c.Start(u.Scheme, u.Host)
if err != nil {
panic(err)
}
defer c.Close()
// find published tracks
tracks, baseURL, _, err := c.Describe(u)
if err != nil {
panic(err)
}
// find the Opus track
opusTrack, opusTrackID := func() (*gortsplib.TrackOpus, int) {
for i, track := range tracks {
if tt, ok := track.(*gortsplib.TrackOpus); ok {
return tt, i
}
}
return nil, -1
}()
if opusTrack == nil {
panic("Opus track not found")
}
// setup decoder
dec := &rtpopus.Decoder{
SampleRate: opusTrack.SampleRate,
}
dec.Init()
// called when a RTP packet arrives
c.OnPacketRTP = func(ctx *gortsplib.ClientOnPacketRTPCtx) {
if ctx.TrackID != opusTrackID {
return
}
// decode an Opus packet from the RTP packet
op, _, err := dec.Decode(ctx.Packet)
if err != nil {
return
}
// print
log.Printf("received Opus packet of size %d\n", len(op))
}
// setup and read all tracks
err = c.SetupAndPlay(tracks, baseURL)
if err != nil {
panic(err)
}
// wait until a fatal error
panic(c.Wait())
}