From f6aa35eedbc76e7467a68dbe3aba63ec8cb352a5 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 26 七月 2019 10:51:03 +0800
Subject: [PATCH] add log

---
 goffmpeg.go |  130 +++++++++++++++++++++++++++++++++++++++----
 1 files changed, 118 insertions(+), 12 deletions(-)

diff --git a/goffmpeg.go b/goffmpeg.go
index b497fac..b8739f8 100644
--- a/goffmpeg.go
+++ b/goffmpeg.go
@@ -7,35 +7,96 @@
 */
 import "C"
 import (
+	"errors"
 	"fmt"
 	"unsafe"
 )
 
-// GoFFMPEG handle for c
-type GoFFMPEG struct {
-	lib    C.libcffmpeg
-	ffmpeg C.cffmpeg
-}
+const (
+	// 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
+)
 
-// New create handle
-func New() *GoFFMPEG {
-	soFile := C.CString("./runtime/libcffmpeg.so")
+var libcffmpeg C.libcffmpeg
+
+// InitFFmpeg init ffmepg
+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
+	return nil
+}
+
+// FreeFFmpeg free ffmpeg
+func FreeFFmpeg() {
+	if libcffmpeg != nil {
+		C.release_libcffmpeg(libcffmpeg)
+	}
+}
+
+// Config config
+type Config struct {
+	Scale  int
+	Width  int
+	Height int
+	GB     bool
+	CPU    bool
+}
+
+// GoFFMPEG handle for c
+type GoFFMPEG struct {
+	ffmpeg C.cffmpeg
+}
+
+// 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{
-		lib:    lib,
-		ffmpeg: C.wrap_fn_create(),
+		ffmpeg: f,
 	}
 }
 
 // Free free handle
 func (h *GoFFMPEG) Free() {
 	C.wrap_fn_destroy(h.ffmpeg)
-	C.release_libcffmpeg(h.lib)
 }
 
 // Run ffmpeg
@@ -64,3 +125,48 @@
 	}
 	return nil, 0, 0
 }
+
+///////////////for encoder
+
+// GoEncoder encoder
+type GoEncoder struct {
+	enc C.cencoder
+}
+
+// NewEncoder encoder
+func NewEncoder(w, h, fps, br, sFlag, gi int) *GoEncoder {
+	if w <= 0 || h <= 0 {
+		return nil
+	}
+
+	return &GoEncoder{
+		enc: C.wrap_fn_create_encoder(C.int(w), C.int(h), C.int(fps), C.int(br), C.int(sFlag), C.int(gi)),
+	}
+}
+
+// Free free
+func (e *GoEncoder) Free() {
+	C.wrap_fn_destroy_encoder(e.enc)
+}
+
+// Encode pic
+func (e *GoEncoder) Encode(in []byte, w, h int) ([]byte, int, bool) {
+
+	var size C.int
+	var key C.int
+	cin := C.CBytes(in)
+	defer C.free(cin)
+
+	p := C.wrap_fn_encode(e.enc, cin, C.int(w), C.int(h), &size, &key)
+	defer C.free(p)
+	if p != nil && size > 0 {
+		b := C.GoBytes(p, size)
+
+		isKey := false
+		if key > 0 {
+			isKey = true
+		}
+		return b, int(size), isKey
+	}
+	return nil, 0, false
+}

--
Gitblit v1.8.0