From dc65b072a13994ff02cc7b54f3b00ba087e7f86a Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期三, 20 十一月 2019 14:05:52 +0800
Subject: [PATCH] add zerocopy

---
 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