feat: add setter and getter for extra_hw_frames

asticode · Mar 12, 2024 · a3a029b · a3a029b
1 parent d3d2df8
commit a3a029b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/codec_context.go b/codec_context.go
@@ -314,6 +314,14 @@ func (cc *CodecContext) SetHardwareDeviceContext(hdc *HardwareDeviceContext) {
 	}
 }
 
+func (cc *CodecContext) ExtraHwFrames() int {
+	return int(cc.c.extra_hw_frames)
+}
+
+func (cc *CodecContext) SetExtraHardwareFrames(n int) {
+	cc.c.extra_hw_frames = C.int(n)
+}
+
 type CodecContextPixelFormatCallback func(pfs []PixelFormat) PixelFormat
 
 var (

diff --git a/codec_context_test.go b/codec_context_test.go
@@ -97,6 +97,7 @@ func TestCodecContext(t *testing.T) {
 	cc4.SetThreadType(ThreadTypeSlice)
 	cc4.SetTimeBase(NewRational(15, 1))
 	cc4.SetWidth(16)
+	cc4.SetExtraHardwareFrames(4)
 	require.Equal(t, int64(1), cc4.BitRate())
 	require.True(t, cc4.ChannelLayout().Equal(ChannelLayout21))
 	require.Equal(t, 3, cc4.Channels())
@@ -115,6 +116,7 @@ func TestCodecContext(t *testing.T) {
 	require.Equal(t, ThreadTypeSlice, cc4.ThreadType())
 	require.Equal(t, NewRational(15, 1), cc4.TimeBase())
 	require.Equal(t, 16, cc4.Width())
+	require.Equal(t, 4, cc4.ExtraHwFrames())
 
 	// TODO Test ReceivePacket
 	// TODO Test SendPacket