From e878e92811a2dbfb6b4d3f7b2c357435f56e28db Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期日, 29 九月 2019 10:27:21 +0800
Subject: [PATCH] add trytime

---
 goffmpeg.go |   80 +++++++++++++++++++++++++++++----------
 1 files changed, 59 insertions(+), 21 deletions(-)

diff --git a/goffmpeg.go b/goffmpeg.go
index 1efcb8d..d23bf80 100644
--- a/goffmpeg.go
+++ b/goffmpeg.go
@@ -75,7 +75,33 @@
 
 // New 2nd new
 func New(conf Config) *GoFFMPEG {
+
 	f := C.wrap_fn_create()
+
+	if f == nil {
+		return nil
+	}
+	if conf.Scale != 0 && conf.Width != 0 && conf.Height != 0 {
+		C.wrap_fn_scale(f, C.int(conf.Width), C.int(conf.Height), C.int(conf.Scale))
+	}
+	if conf.GB {
+		C.wrap_fn_run_gb28181(f)
+	}
+	if conf.CPU {
+		C.wrap_fn_use_cpu(f)
+	}
+
+	return &GoFFMPEG{
+		ffmpeg: f,
+	}
+}
+
+// NewWithLog log
+func NewWithLog(conf Config, logfile string) *GoFFMPEG {
+	lf := C.CString(logfile)
+	defer C.free(unsafe.Pointer(lf))
+
+	f := C.wrap_fn_create2(lf)
 	if f == nil {
 		return nil
 	}
@@ -107,25 +133,6 @@
 	C.wrap_fn_run(h.ffmpeg, in)
 }
 
-// DecodeJPEG decode jpeg file
-func DecodeJPEG(input string) ([]byte, int, 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(in, &width, &height)
-	defer C.free(p)
-
-	if width > 0 && height > 0 {
-		data := C.GoBytes(p, width*height*3)
-		wid := int(width)
-		hei := int(height)
-		return data, wid, hei
-	}
-	return nil, 0, 0
-}
-
 // FireRecorder fire recorder
 func (h *GoFFMPEG) FireRecorder(sid string, id int64) {
 	csid := C.CString(sid)
@@ -134,13 +141,17 @@
 }
 
 // BuildRecorder build recorder
-func (h *GoFFMPEG) BuildRecorder(sid, output string, mind, maxd int) {
+func (h *GoFFMPEG) BuildRecorder(sid, output string, mind, maxd int, audio bool) {
 	out := C.CString(output)
 	defer C.free(unsafe.Pointer(out))
 	csid := C.CString(sid)
 	defer C.free(unsafe.Pointer(csid))
 
-	C.wrap_fn_recorder(h.ffmpeg, csid, out, C.int(mind), C.int(maxd))
+	a := 0
+	if audio {
+		a = 1
+	}
+	C.wrap_fn_recorder(h.ffmpeg, csid, out, C.int(mind), C.int(maxd), C.int(a))
 }
 
 // GetInfoRecorder info
@@ -207,6 +218,33 @@
 	return d, s, k
 }
 
+/////////////// for decoder
+
+// Decode decode jpeg file
+// return val: -1 open error; -2, find stream error; -3, converter create error
+func Decode(input string, gb bool) ([]byte, int, int) {
+	in := C.CString(input)
+	defer C.free(unsafe.Pointer(in))
+
+	withGB := 0
+	if gb {
+		withGB = 1
+	}
+
+	var width C.int
+	var height C.int
+	p := C.wrap_fn_decode(in, C.int(withGB), &width, &height)
+	defer C.free(p)
+
+	if width > 0 && height > 0 {
+		data := C.GoBytes(p, width*height*3)
+		wid := int(width)
+		hei := int(height)
+		return data, wid, hei
+	}
+	return nil, int(width), int(height)
+}
+
 ///////////////for encoder
 
 // GoEncoder encoder

--
Gitblit v1.8.0