Expose pointer to underlying AVCodecContext (#95)

* feat: expose pointer to underlying C object * feat: add ut
asticode · Nov 5, 2024 · 0d36224 · 0d36224
1 parent bc14852
commit 0d36224
Show file tree

Hide file tree

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

diff --git a/codec_context_test.go b/codec_context_test.go
@@ -2,6 +2,7 @@ package astiav
 
 import (
 	"testing"
+	"unsafe"
 
 	"github.com/stretchr/testify/require"
 )
@@ -40,6 +41,7 @@ func TestCodecContext(t *testing.T) {
 	require.Equal(t, 1, cc1.ThreadCount())
 	require.Equal(t, ThreadType(3), cc1.ThreadType())
 	require.Equal(t, 320, cc1.Width())
+	require.Equal(t, unsafe.Pointer(cc1.c), cc1.UnsafePointer())
 	cl := cc1.Class()
 	require.NotNil(t, cl)
 	require.Equal(t, "AVCodecContext", cl.Name())