From f29f269d999fc4bb8a9bfa85629f08b89971a2ef Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期二, 03 十二月 2019 16:00:49 +0800
Subject: [PATCH] Merge branch 'master' of ssh://192.168.5.5:29418/valib/goffmpeg

---
 godec.go |   19 ++++++++++++++++++-
 1 files changed, 18 insertions(+), 1 deletions(-)

diff --git a/godec.go b/godec.go
index aab8c7a..25baa88 100644
--- a/godec.go
+++ b/godec.go
@@ -19,10 +19,27 @@
 	var srcW, srcH, srcF C.int
 
 	p := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &srcW, &srcH, &srcF, &length, &fid)
-	if srcW == 0 || srcH == 0 {
+	if p == 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)
 }
+
+// GetYUV2 get yuv data
+func (h *GoFFMPEG) GetYUV2() (unsafe.Pointer, []byte, int, int, int64) {
+	var fid C.long
+	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 {
+		return nil, nil, 0, 0, 0
+	}
+	const maxLen = 0x7fffffff
+	size := int(length)
+	data := (*[maxLen]byte)(unsafe.Pointer(p))[:size:size]
+
+	return unsafe.Pointer(p), data, int(srcW), int(srcH), int64(fid)
+}

--
Gitblit v1.8.0