diff --git a/encode.go b/encode.go index c8e04b9..e045e6a 100644 --- a/encode.go +++ b/encode.go @@ -57,6 +57,7 @@ type Encoder struct { csp int32 pts int64 + dts int64 nnals int32 nals []*x264c.Nal @@ -229,9 +230,16 @@ func (e *Encoder) Encode(im image.Image) (err error) { } } + e.dts = picOut.IDts + return } +// GetTimestamp returns the current PTS and DTS. +func (e *Encoder) GetTimestamp() (int64, int64) { + return e.pts, e.dts +} + // Flush flushes encoder. func (e *Encoder) Flush() (err error) { var picOut x264c.Picture