Examples now ignore decoded frames with non monotonic PTS

Author: asticode

examples/demuxing_decoding/main.go

@@ -8,6 +8,7 @@ import (
 	"strings"
 
 	"github.com/asticode/go-astiav"
+	"github.com/asticode/go-astikit"
 )
 
 var (
@@ -17,6 +18,7 @@ var (
 type stream struct {
 	decCodec        *astiav.Codec
 	decCodecContext *astiav.CodecContext
+	decLastPTS      *int64
 	inputStream     *astiav.Stream
 }
 
@@ -146,6 +148,12 @@ func main() {
 					// Make sure to unreference the frame
 					defer f.Unref()
 
+					// Ignore frames with non monotonic PTS
+					if s.decLastPTS != nil && *s.decLastPTS >= f.Pts() {
+						return false
+					}
+					s.decLastPTS = astikit.Int64Ptr(f.Pts())
+
 					// Log
 					log.Printf("new %s frame: stream %d - pts: %d", s.inputStream.CodecParameters().MediaType(), pkt.StreamIndex(), f.Pts())
 					return false

examples/filtering/main.go

@@ -27,6 +27,7 @@ type stream struct {
 	decCodec          *astiav.Codec
 	decCodecContext   *astiav.CodecContext
 	decFrame          *astiav.Frame
+	decLastPTS        *int64
 	filterFrame       *astiav.Frame
 	filterGraph       *astiav.FilterGraph
 	inputStream       *astiav.Stream
@@ -112,6 +113,12 @@ func main() {
 					// Make sure to unreference the frame
 					defer s.decFrame.Unref()
 
+					// Ignore frames with non monotonic PTS
+					if s.decLastPTS != nil && *s.decLastPTS >= s.decFrame.Pts() {
+						return false
+					}
+					s.decLastPTS = astikit.Int64Ptr(s.decFrame.Pts())
+
 					// Filter frame
 					if err := filterFrame(s.decFrame, s); err != nil {
 						log.Fatal(fmt.Errorf("main: filtering frame failed: %w", err))

examples/hardware_decoding_filtering/main.go

@@ -25,6 +25,7 @@ var (
 	c                     = astikit.NewCloser()
 	decCodec              *astiav.Codec
 	decCodecContext       *astiav.CodecContext
+	decLastPTS            *int64
 	decodedHardwareFrame  *astiav.Frame
 	filterGraph           *astiav.FilterGraph
 	filteredHardwareFrame *astiav.Frame
@@ -214,6 +215,12 @@ func main() {
 					// Make sure to unreference hardware frame
 					defer decodedHardwareFrame.Unref()
 
+					// Ignore frames with non monotonic PTS
+					if decLastPTS != nil && *decLastPTS >= decodedHardwareFrame.Pts() {
+						return false
+					}
+					decLastPTS = astikit.Int64Ptr(decodedHardwareFrame.Pts())
+
 					// Invalid pixel format
 					if decodedHardwareFrame.PixelFormat() != hardwarePixelFormat {
 						log.Fatalf("main: invalid decoded pixel format %s, expected %s", decodedHardwareFrame.PixelFormat(), hardwarePixelFormat)

examples/resampling_audio/main.go

@@ -8,6 +8,7 @@ import (
 	"strings"
 
 	"github.com/asticode/go-astiav"
+	"github.com/asticode/go-astikit"
 )
 
 var (
@@ -16,6 +17,7 @@ var (
 
 var (
 	af             *astiav.AudioFifo
+	decLastPTS     *int64
 	decodedFrame   *astiav.Frame
 	finalFrame     *astiav.Frame
 	resampledFrame *astiav.Frame
@@ -186,6 +188,12 @@ func main() {
 					// Make sure to unreference the frame
 					defer decodedFrame.Unref()
 
+					// Ignore frames with non monotonic PTS
+					if decLastPTS != nil && *decLastPTS >= decodedFrame.Pts() {
+						return false
+					}
+					decLastPTS = astikit.Int64Ptr(decodedFrame.Pts())
+
 					// Log
 					log.Printf("new decoded frame: nb samples: %d", decodedFrame.NbSamples())
 

examples/transcoding/main.go

@@ -29,6 +29,7 @@ type stream struct {
 	decCodec          *astiav.Codec
 	decCodecContext   *astiav.CodecContext
 	decFrame          *astiav.Frame
+	decLastPTS        *int64
 	encCodec          *astiav.Codec
 	encCodecContext   *astiav.CodecContext
 	encPkt            *astiav.Packet
@@ -126,6 +127,12 @@ func main() {
 					// Make sure to unreference the frame
 					defer s.decFrame.Unref()
 
+					// Ignore frames with non monotonic PTS
+					if s.decLastPTS != nil && *s.decLastPTS >= s.decFrame.Pts() {
+						return false
+					}
+					s.decLastPTS = astikit.Int64Ptr(s.decFrame.Pts())
+
 					// Filter, encode and write frame
 					if err := filterEncodeWriteFrame(s.decFrame, s); err != nil {
 						log.Fatal(fmt.Errorf("main: filtering, encoding and writing frame failed: %w", err))