From 18a05d269516a5e33d8460291c2f93e73d95adce Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期二, 26 十二月 2023 10:45:31 +0800
Subject: [PATCH] GetYUV format is NV12
---
goffmpeg.go | 208 +++++++++++++++++++++++++++++++++++++++++++++------
1 files changed, 182 insertions(+), 26 deletions(-)
diff --git a/goffmpeg.go b/goffmpeg.go
index b497fac..485e568 100644
--- a/goffmpeg.go
+++ b/goffmpeg.go
@@ -7,35 +7,176 @@
*/
import "C"
import (
+ "errors"
"fmt"
"unsafe"
)
-// GoFFMPEG handle for c
-type GoFFMPEG struct {
- lib C.libcffmpeg
- ffmpeg C.cffmpeg
-}
+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
+)
-// New create handle
-func New() *GoFFMPEG {
- soFile := C.CString("./runtime/libcffmpeg.so")
+// SrcFormat format NV
+const SrcFormat = 23
+
+// DstFormat format
+const DstFormat = 3
+
+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)
+ }
+}
+
+// GoFFMPEG handle for c
+type GoFFMPEG struct {
+ ffmpeg C.cffmpeg
+}
+
+// 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{
- lib: lib,
- ffmpeg: C.wrap_fn_create(),
+ ffmpeg: f,
+ }
+}
+
+// NewWithDevID 2nd new
+func NewWithDevID(GB, CPU bool, devID int) *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)
+ } else if devID > -1 {
+ r := C.wrap_fn_set_devid(unsafe.Pointer(libcffmpeg), f, C.int(devID))
+ if r != 0 {
+ if f != nil {
+ C.wrap_fn_destroy(unsafe.Pointer(libcffmpeg), f)
+ }
+ FreeFFmpeg()
+ return nil
+ }
+ }
+
+ return &GoFFMPEG{
+ ffmpeg: f,
+ }
+}
+
+// NewWithLogAndDevID log
+func NewWithLogAndDevID(GB, CPU bool, devID int, 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)
+ } else if devID > -1 {
+ r := C.wrap_fn_set_devid(unsafe.Pointer(libcffmpeg), f, C.int(devID))
+ if r != 0 {
+ if f != nil {
+ C.wrap_fn_destroy(unsafe.Pointer(libcffmpeg), f)
+ }
+ FreeFFmpeg()
+ return nil
+ }
+ }
+
+ return &GoFFMPEG{
+ 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)
- C.release_libcffmpeg(h.lib)
+ if h.ffmpeg != nil {
+ C.wrap_fn_destroy(unsafe.Pointer(libcffmpeg), h.ffmpeg)
+ }
}
// Run ffmpeg
@@ -43,24 +184,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, maxTry int) []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, C.int(maxTry))
+ defer C.free(unsafe.Pointer(pic))
+
+ retJpg := C.GoBytes(unsafe.Pointer(pic), jpgLen)
+ return retJpg
}
--
Gitblit v1.8.0