package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-05 23:26:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

simplify examples

Browse Source
This commit is contained in:
aler9
2021-11-12 17:50:34 +01:00
committed by Alessandro Ros
parent 922d74b940
commit 075f84de28
2 changed files with 63 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
examples/client-read-h264-save-to-disk/main.go
View File

@@ -13,37 +13,7 @@ import (
// 3. save the content of the H264 track to a file in MPEG-TS format // 3. save the content of the H264 track to a file in MPEG-TS format
func main() { func main() {
dec := rtph264.NewDecoder() c := gortsplib.Client{}
var h264Track int
var enc *mpegtsEncoder
c := gortsplib.Client{
// called when a RTP packet arrives
OnPacketRTP: func(trackID int, payload []byte) {
if trackID != h264Track {
return
}
// parse RTP packet
var pkt rtp.Packet
err := pkt.Unmarshal(payload)
if err != nil {
return
}
// decode H264 NALUs from RTP packets
nalus, pts, err := dec.DecodeUntilMarker(&pkt)
if err != nil {
return
}
// encode H264 NALUs into MPEG-TS
err = enc.encode(nalus, pts)
if err != nil {
return
}
},
}
// parse URL // parse URL
u, err := base.ParseURL("rtsp://localhost:8554/mystream") u, err := base.ParseURL("rtsp://localhost:8554/mystream")
@@ -70,7 +40,7 @@ func main() {
} }
// find the H264 track // find the H264 track
h264Track = func() int { h264Track := func() int {
for i, track := range tracks { for i, track := range tracks {
if track.IsH264() { if track.IsH264() {
return i return i
@@ -88,12 +58,41 @@ func main() {
panic(err) panic(err)
} }
// setup the encoder // setup decoder
enc, err = newMPEGTSEncoder(h264Conf) dec := rtph264.NewDecoder()
// setup encoder
enc, err := newMPEGTSEncoder(h264Conf)
if err != nil { if err != nil {
panic(err) panic(err)
} }
// called when a RTP packet arrives
c.OnPacketRTP = func(trackID int, payload []byte) {
if trackID != h264Track {
return
}
// parse RTP packet
var pkt rtp.Packet
err := pkt.Unmarshal(payload)
if err != nil {
return
}
// decode H264 NALUs from RTP packets
nalus, pts, err := dec.DecodeUntilMarker(&pkt)
if err != nil {
return
}
// encode H264 NALUs into MPEG-TS
err = enc.encode(nalus, pts)
if err != nil {
return
}
}
// setup all tracks // setup all tracks
for _, t := range tracks { for _, t := range tracks {
_, err := c.Setup(true, baseURL, t, 0, 0) _, err := c.Setup(true, baseURL, t, 0, 0)

60
examples/client-read-h264/main.go
View File

@@ -15,35 +15,7 @@ import (
// 3. get H264 NALUs of that track // 3. get H264 NALUs of that track
func main() { func main() {
var h264Track int c := gortsplib.Client{}
dec := rtph264.NewDecoder()
c := gortsplib.Client{
// called when a RTP packet arrives
OnPacketRTP: func(trackID int, payload []byte) {
if trackID != h264Track {
return
}
// parse RTP packet
var pkt rtp.Packet
err := pkt.Unmarshal(payload)
if err != nil {
return
}
// decode H264 NALUs from RTP packets
nalus, _, err := dec.Decode(&pkt)
if err != nil {
return
}
// print NALUs
for _, nalu := range nalus {
fmt.Printf("received H264 NALU of size %d\n", len(nalu))
}
},
}
// parse URL // parse URL
u, err := base.ParseURL("rtsp://localhost:8554/mystream") u, err := base.ParseURL("rtsp://localhost:8554/mystream")
@@ -70,7 +42,7 @@ func main() {
} }
// find the H264 track // find the H264 track
h264Track = func() int { h264Track := func() int {
for i, track := range tracks { for i, track := range tracks {
if track.IsH264() { if track.IsH264() {
return i return i
@@ -82,6 +54,34 @@ func main() {
panic("H264 track not found") panic("H264 track not found")
} }
// setup decoder
dec := rtph264.NewDecoder()
// called when a RTP packet arrives
c.OnPacketRTP = func(trackID int, payload []byte) {
if trackID != h264Track {
return
}
// parse RTP packet
var pkt rtp.Packet
err := pkt.Unmarshal(payload)
if err != nil {
return
}
// decode H264 NALUs from RTP packets
nalus, _, err := dec.Decode(&pkt)
if err != nil {
return
}
// print NALUs
for _, nalu := range nalus {
fmt.Printf("received H264 NALU of size %d\n", len(nalu))
}
}
// setup all tracks // setup all tracks
for _, t := range tracks { for _, t := range tracks {
_, err := c.Setup(true, baseURL, t, 0, 0) _, err := c.Setup(true, baseURL, t, 0, 0)