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 |  124 ++++++++++++++++++++++++++++++++++------
 1 files changed, 104 insertions(+), 20 deletions(-)

diff --git a/goffmpeg.go b/goffmpeg.go
index d7676d2..bca75d2 100644
--- a/goffmpeg.go
+++ b/goffmpeg.go
@@ -12,15 +12,48 @@
 	"unsafe"
 )
 
+const (
+	// ScaleNone self add no scale raw frame data
+	ScaleNone = 0
+	// ScaleFastBilinear SWS_FAST_BILINEAR
+	ScaleFastBilinear = 1
+	// ScaleBilinear SWS_BILINEAR
+	ScaleBilinear = 2
+	// ScaleBicubic SWS_BICUBIC
+	ScaleBicubic = 4
+	// ScaleX SWS_X
+	ScaleX = 8
+	// ScalePoint SWS_POINT
+	ScalePoint = 0x10
+	// ScaleArea SWS_AREA
+	ScaleArea = 0x20
+	// ScaleBicublin SWS_BICUBLIN
+	ScaleBicublin = 0x40
+	// ScaleGauss SWS_GAUSS
+	ScaleGauss = 0x80
+	// ScaleSinc SWS_SINC
+	ScaleSinc = 0x100
+	// ScaleLancZos SWS_LANCZOS
+	ScaleLancZos = 0x200
+	// ScaleSpline SWS_SPLINE
+	ScaleSpline = 0x400
+)
+
+// SrcFormat format NV
+const SrcFormat = 23
+
+// DstFormat format
+const DstFormat = 3
+
 var libcffmpeg C.libcffmpeg
 
 // InitFFmpeg init ffmepg
-func InitFFmpeg() error {
-	soFile := C.CString("./runtime/libcffmpeg.so")
+func InitFFmpeg(soFileGo string) error {
+	soFile := C.CString(soFileGo)
 	defer C.free(unsafe.Pointer(soFile))
 	lib := C.init_libcffmpeg(soFile)
 	if lib == nil {
-		fmt.Println("open libcffmpeg.so error")
+		fmt.Println("open error: ", soFileGo)
 		return errors.New("init ffmpeg error")
 	}
 	libcffmpeg = lib
@@ -39,16 +72,52 @@
 	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,
+	}
+}
+
+// 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,
 	}
 }
 
 // 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
@@ -56,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