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 | 85 +++++++++++++++++++++++++++++++++++-------
1 files changed, 71 insertions(+), 14 deletions(-)
diff --git a/goffmpeg.go b/goffmpeg.go
index 0a2e4ee..b8739f8 100644
--- a/goffmpeg.go
+++ b/goffmpeg.go
@@ -40,12 +40,12 @@
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
@@ -59,24 +59,36 @@
}
}
+// 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 create handle
-func New() *GoFFMPEG {
- return &GoFFMPEG{
- ffmpeg: C.wrap_fn_create(),
- }
-}
-
-// NewWithScale scale out pic
-func NewWithScale(w, h int, flag int) *GoFFMPEG {
+// New 2nd new
+func New(conf Config) *GoFFMPEG {
f := C.wrap_fn_create()
- if f != nil {
- C.wrap_fn_scale(f, C.int(w), C.int(h), C.int(flag))
+ 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,
}
@@ -113,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