Enhance Codec Iteration and Checking of HardwareConfigMethodFlags

codec.go

@@ -13,6 +13,8 @@ type Codec struct {
 	c *C.struct_AVCodec
 }
 
+type CodecProcessor func(*Codec)
+
 func newCodecFromC(c *C.struct_AVCodec) *Codec {
 	if c == nil {
 		return nil
@@ -28,6 +30,10 @@ func (c *Codec) String() string {
 	return c.Name()
 }
 
+func (c *Codec) ID() CodecID {
+	return CodecID(c.c.id)
+}
+
 func (c *Codec) ChannelLayouts() (o []ChannelLayout) {
 	if c.c.ch_layouts == nil {
 		return nil
@@ -113,3 +119,33 @@ func (c *Codec) HardwareConfigs(dt HardwareDeviceType) (configs []CodecHardwareC
 	}
 	return
 }
+
+func (c *Codec) HasHardwareConfigMethodFlag(chcmf CodecHardwareConfigMethodFlag) bool {
+	var i C.int
+	for {
+		config := C.avcodec_get_hw_config(c.c, i)
+		if config == nil {
+			break
+		}
+
+		if (CodecHardwareConfig{c: config}.MethodFlags().Has(chcmf)) {
+			return true
+		}
+
+		i++
+	}
+	return false
+}
+
+func IterateCodecs(processor CodecProcessor) {
+	var opq *C.void = nil
+	for {
+		c := C.av_codec_iterate((*unsafe.Pointer)(unsafe.Pointer(&opq)))
+		if c == nil {
+			break
+		}
+
+		codec := newCodecFromC(c)
+		processor(codec)
+	}
+}

codec_test.go

@@ -66,4 +66,13 @@ func TestCodec(t *testing.T) {
 
 	c = astiav.FindDecoderByName("invalid")
 	require.Nil(t, c)
+
+	var decoders []*astiav.Codec
+	processor := func(c *astiav.Codec) {
+		if c.IsDecoder() && c.ID() == astiav.CodecIDMjpeg {
+			decoders = append(decoders, c)
+		}
+	}
+	astiav.IterateCodecs(processor)
+	require.Equal(t, decoders[0].ID(), astiav.CodecIDMjpeg)
 }