From 18a05d269516a5e33d8460291c2f93e73d95adce Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 26 十二月 2023 10:45:31 +0800
Subject: [PATCH] GetYUV format is NV12

---
 godec.go |   36 ++++++++++++++++++++++++++++--------
 1 files changed, 28 insertions(+), 8 deletions(-)

diff --git a/godec.go b/godec.go
index 25baa88..50f5ea2 100644
--- a/godec.go
+++ b/godec.go
@@ -18,13 +18,20 @@
 	var length C.int
 	var srcW, srcH, srcF C.int
 
-	p := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &srcW, &srcH, &srcF, &length, &fid)
-	if p == nil || length == 0 || srcW == 0 || srcH == 0 {
+	var yuv *C.uchar
+	r := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &yuv, &srcW, &srcH, &srcF, &length, &fid)
+
+	if r < 0 {
+		return nil, int(r), int(r), int64(r)
+	}
+
+	if yuv == nil || length == 0 || srcW == 0 || srcH == 0 {
 		return nil, 0, 0, 0
 	}
-	defer C.free(unsafe.Pointer(p))
 
-	return C.GoBytes(p, length), int(srcW), int(srcH), int64(fid)
+	defer C.free(unsafe.Pointer(yuv))
+
+	return C.GoBytes(unsafe.Pointer(yuv), length), int(srcW), int(srcH), int64(fid)
 }
 
 // GetYUV2 get yuv data
@@ -33,13 +40,26 @@
 	var length C.int
 	var srcW, srcH, srcF C.int
 
-	p := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &srcW, &srcH, &srcF, &length, &fid)
-	if p == nil || length == 0 || srcW == 0 || srcH == 0 {
+	var yuv *C.uchar
+	r := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &yuv, &srcW, &srcH, &srcF, &length, &fid)
+
+	if r < 0 {
+		return nil, nil, int(r), int(r), int64(r)
+	}
+
+	if yuv == nil || length == 0 || srcW == 0 || srcH == 0 {
 		return nil, nil, 0, 0, 0
 	}
+
 	const maxLen = 0x7fffffff
 	size := int(length)
-	data := (*[maxLen]byte)(unsafe.Pointer(p))[:size:size]
+	data := (*[maxLen]byte)(unsafe.Pointer(yuv))[:size:size]
 
-	return unsafe.Pointer(p), data, int(srcW), int(srcH), int64(fid)
+	return unsafe.Pointer(yuv), data, int(srcW), int(srcH), int64(fid)
+}
+
+// SetDecMinDuration min duration
+// 涓昏鏄鏋滆В鐮佹枃浠舵椂闀垮お鐭紝鍙兘鍦╯dk鏀跺埌瑙g爜鏁版嵁鍓嶈В鐮佽繘绋嬭繍琛岀粨鏉燂紝鎵�浠ヨ瀹氫竴涓渶鐭殑duration
+func (h *GoFFMPEG) SetDecMinDuration(min, max int) {
+	C.wrap_fn_rec_duration(unsafe.Pointer(libcffmpeg), h.ffmpeg, C.int(min), C.int(max))
 }

--
Gitblit v1.8.0