From f93ee1a42e8c47e472332287b7350b66a6b0fa11 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 24 七月 2020 18:28:57 +0800
Subject: [PATCH] 保存触发id之后的视频作为触发视频

---
 goffmpeg.go |   92 +++++++++++++++++++++++++++++++++++-----------
 1 files changed, 70 insertions(+), 22 deletions(-)

diff --git a/goffmpeg.go b/goffmpeg.go
index e33de3c..bca75d2 100644
--- a/goffmpeg.go
+++ b/goffmpeg.go
@@ -13,6 +13,8 @@
 )
 
 const (
+	// ScaleNone self add no scale raw frame data
+	ScaleNone = 0
 	// ScaleFastBilinear SWS_FAST_BILINEAR
 	ScaleFastBilinear = 1
 	// ScaleBilinear SWS_BILINEAR
@@ -36,6 +38,12 @@
 	// ScaleSpline SWS_SPLINE
 	ScaleSpline = 0x400
 )
+
+// SrcFormat format NV
+const SrcFormat = 23
+
+// DstFormat format
+const DstFormat = 3
 
 var libcffmpeg C.libcffmpeg
 
@@ -64,19 +72,42 @@
 	ffmpeg C.cffmpeg
 }
 
-// New create handle
-func New() *GoFFMPEG {
+// New 2nd new
+func New(GB, CPU bool) *GoFFMPEG {
+
+	f := C.wrap_fn_create(unsafe.Pointer(libcffmpeg))
+
+	if f == nil {
+		return nil
+	}
+	if GB {
+		C.wrap_fn_run_gb28181(unsafe.Pointer(libcffmpeg), f)
+	}
+	if CPU {
+		C.wrap_fn_use_cpu(unsafe.Pointer(libcffmpeg), f)
+	}
+
 	return &GoFFMPEG{
-		ffmpeg: C.wrap_fn_create(),
+		ffmpeg: f,
 	}
 }
 
-// NewWithScale scale out pic
-func NewWithScale(w, h int, flag int) *GoFFMPEG {
-	f := C.wrap_fn_create()
-	if f != nil {
-		C.wrap_fn_scale(f, C.int(w), C.int(h), C.int(flag))
+// NewWithLog log
+func NewWithLog(GB, CPU bool, ffmpegLog string) *GoFFMPEG {
+	lf := C.CString(ffmpegLog)
+	defer C.free(unsafe.Pointer(lf))
+
+	f := C.wrap_fn_create2(unsafe.Pointer(libcffmpeg), lf)
+	if f == nil {
+		return nil
 	}
+	if GB {
+		C.wrap_fn_run_gb28181(unsafe.Pointer(libcffmpeg), f)
+	}
+	if CPU {
+		C.wrap_fn_use_cpu(unsafe.Pointer(libcffmpeg), f)
+	}
+
 	return &GoFFMPEG{
 		ffmpeg: f,
 	}
@@ -84,7 +115,9 @@
 
 // Free free handle
 func (h *GoFFMPEG) Free() {
-	C.wrap_fn_destroy(h.ffmpeg)
+	if h.ffmpeg != nil {
+		C.wrap_fn_destroy(unsafe.Pointer(libcffmpeg), h.ffmpeg)
+	}
 }
 
 // Run ffmpeg
@@ -92,24 +125,39 @@
 	in := C.CString(input)
 	defer C.free(unsafe.Pointer(in))
 
-	C.wrap_fn_run(h.ffmpeg, in)
+	C.wrap_fn_run(unsafe.Pointer(libcffmpeg), h.ffmpeg, in)
 }
 
-// DecodeJPEG decode jpeg file
-func (h *GoFFMPEG) DecodeJPEG(input string) ([]byte, int, int) {
+// Run2 ffmpeg
+func (h *GoFFMPEG) Run2(input string, minDuration int) {
 	in := C.CString(input)
 	defer C.free(unsafe.Pointer(in))
 
-	var width C.int
-	var height C.int
-	p := C.wrap_fn_decode_jpeg(h.ffmpeg, in, &width, &height)
-	defer C.free(p)
+	C.wrap_fn_run(unsafe.Pointer(libcffmpeg), h.ffmpeg, in)
+}
 
-	if width > 0 && height > 0 {
-		data := C.GoBytes(p, width*height*3)
-		wid := int(width)
-		hei := int(height)
-		return data, wid, hei
+// FPS fps
+func (h *GoFFMPEG) FPS() int {
+	return int(C.wrap_fn_fps(unsafe.Pointer(libcffmpeg), h.ffmpeg))
+}
+
+// ReleaseC release c memory
+func ReleaseC(p unsafe.Pointer) {
+	if p == nil {
+		return
 	}
-	return nil, 0, 0
+	C.free(unsafe.Pointer(p))
+}
+
+// GetGBJpg Get GB28181 Jpg
+func GetGBJpg(rtspUrl string) []byte {
+	rtsp := C.CString(rtspUrl)
+	defer C.free(unsafe.Pointer(rtsp))
+	var jpgLen C.int
+
+	pic := C.wrap_fn_get_gb28181_pic(unsafe.Pointer(libcffmpeg), rtsp, &jpgLen)
+	defer C.free(unsafe.Pointer(pic))
+
+	retJpg := C.GoBytes(unsafe.Pointer(pic), jpgLen)
+	return retJpg
 }

--
Gitblit v1.8.0