// Copyright (c) 2023 QRTC. All rights reserved.
// Use of this source code is governed by a MIT
// license that can be found in the LICENSE file.

package ffmpeg

/*
#include <libavutil/cpu.h>
*/
import "C"

const (
	AV_CPU_FLAG_FORCE    = C.AV_CPU_FLAG_FORCE
	AV_CPU_FLAG_MMX      = C.AV_CPU_FLAG_MMX
	AV_CPU_FLAG_MMXEXT   = C.AV_CPU_FLAG_MMXEXT
	AV_CPU_FLAG_MMX2     = C.AV_CPU_FLAG_MMX2
	AV_CPU_FLAG_3DNOW    = C.AV_CPU_FLAG_3DNOW
	AV_CPU_FLAG_SSE      = C.AV_CPU_FLAG_SSE
	AV_CPU_FLAG_SSE2     = C.AV_CPU_FLAG_SSE2
	AV_CPU_FLAG_SSE2SLOW = C.AV_CPU_FLAG_SSE2SLOW

	AV_CPU_FLAG_3DNOWEXT = C.AV_CPU_FLAG_3DNOWEXT
	AV_CPU_FLAG_SSE3     = C.AV_CPU_FLAG_SSE3
	AV_CPU_FLAG_SSE3SLOW = C.AV_CPU_FLAG_SSE3SLOW

	AV_CPU_FLAG_SSSE3       = C.AV_CPU_FLAG_SSSE3
	AV_CPU_FLAG_SSSE3SLOW   = C.AV_CPU_FLAG_SSSE3SLOW
	AV_CPU_FLAG_ATOM        = C.AV_CPU_FLAG_ATOM
	AV_CPU_FLAG_SSE4        = C.AV_CPU_FLAG_SSE4
	AV_CPU_FLAG_SSE42       = C.AV_CPU_FLAG_SSE42
	AV_CPU_FLAG_AESNI       = C.AV_CPU_FLAG_AESNI
	AV_CPU_FLAG_AVX         = C.AV_CPU_FLAG_AVX
	AV_CPU_FLAG_AVXSLOW     = C.AV_CPU_FLAG_AVXSLOW
	AV_CPU_FLAG_XOP         = C.AV_CPU_FLAG_XOP
	AV_CPU_FLAG_FMA4        = C.AV_CPU_FLAG_FMA4
	AV_CPU_FLAG_CMOV        = C.AV_CPU_FLAG_CMOV
	AV_CPU_FLAG_AVX2        = C.AV_CPU_FLAG_AVX2
	AV_CPU_FLAG_FMA3        = C.AV_CPU_FLAG_FMA3
	AV_CPU_FLAG_BMI1        = C.AV_CPU_FLAG_BMI1
	AV_CPU_FLAG_BMI2        = C.AV_CPU_FLAG_BMI2
	AV_CPU_FLAG_AVX512      = C.AV_CPU_FLAG_AVX512
	AV_CPU_FLAG_AVX512ICL   = C.AV_CPU_FLAG_AVX512ICL
	AV_CPU_FLAG_SLOW_GATHER = C.AV_CPU_FLAG_SLOW_GATHER

	AV_CPU_FLAG_ALTIVEC = C.AV_CPU_FLAG_ALTIVEC
	AV_CPU_FLAG_VSX     = C.AV_CPU_FLAG_VSX
	AV_CPU_FLAG_POWER8  = C.AV_CPU_FLAG_POWER8

	AV_CPU_FLAG_ARMV5TE = C.AV_CPU_FLAG_ARMV5TE
	AV_CPU_FLAG_ARMV6   = C.AV_CPU_FLAG_ARMV6
	AV_CPU_FLAG_ARMV6T2 = C.AV_CPU_FLAG_ARMV6T2
	AV_CPU_FLAG_VFP     = C.AV_CPU_FLAG_VFP
	AV_CPU_FLAG_VFPV3   = C.AV_CPU_FLAG_VFPV3
	AV_CPU_FLAG_NEON    = C.AV_CPU_FLAG_NEON
	AV_CPU_FLAG_ARMV8   = C.AV_CPU_FLAG_ARMV8
	AV_CPU_FLAG_VFP_VM  = C.AV_CPU_FLAG_VFP_VM
	AV_CPU_FLAG_SETEND  = C.AV_CPU_FLAG_SETEND

	AV_CPU_FLAG_MMI = C.AV_CPU_FLAG_MMI
	AV_CPU_FLAG_MSA = C.AV_CPU_FLAG_MSA

	AV_CPU_FLAG_LSX       = C.AV_CPU_FLAG_LSX
	AV_CPU_FLAG_LASX      = C.AV_CPU_FLAG_LASX
	AV_CPU_FLAG_RVI       = C.AV_CPU_FLAG_RVI
	AV_CPU_FLAG_RVF       = C.AV_CPU_FLAG_RVF
	AV_CPU_FLAG_RVD       = C.AV_CPU_FLAG_RVD
	AV_CPU_FLAG_RVV_I32   = C.AV_CPU_FLAG_RVV_I32
	AV_CPU_FLAG_RVV_F32   = C.AV_CPU_FLAG_RVV_F32
	AV_CPU_FLAG_RVV_I64   = C.AV_CPU_FLAG_RVV_I64
	AV_CPU_FLAG_RVV_F64   = C.AV_CPU_FLAG_RVV_F64
	AV_CPU_FLAG_RVB_BASIC = C.AV_CPU_FLAG_RVB_BASIC
)

// AvGetCpuFlags returns the flags which specify extensions supported by the CPU.
func AvGetCpuFlags() int32 {
	return (int32)(C.av_get_cpu_flags())
}

// AvForceCpuFlags disables cpu detection and forces the specified flags.
func AvForceCpuFlags(flags int32) {
	C.av_force_cpu_flags((C.int)(flags))
}

// AvParseCpuCaps parses CPU caps from a string and update the given AV_CPU_* flags based on that.
func AvParseCpuCaps(flags *uint32, str string) int32 {
	strPtr, strFunc := StringCasting(str)
	defer strFunc()
	return (int32)(C.av_parse_cpu_caps((*C.uint)(flags), (*C.char)(strPtr)))
}

// AvCpuCount returns the number of logical CPU cores present.
func AvCpuCount() int32 {
	return (int32)(C.av_cpu_count())
}

// AvCpuForceCount overrides cpu count detection and forces the specified count.
func AvCpuForceCount(count int32) {
	C.av_cpu_force_count((C.int)(count))
}

// AvCpuMaxAlign gets the maximum data alignment that may be required by FFmpeg.
func AvCpuMaxAlign() int32 {
	return (int32)(C.av_cpu_max_align())
}