From 7950b86a4bf821e126d5e5659b772ce32faa1445 Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期二, 25 五月 2021 11:32:24 +0800 Subject: [PATCH] bug fixed declaration --- goffmpeg.go | 126 ++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 126 insertions(+), 0 deletions(-) diff --git a/goffmpeg.go b/goffmpeg.go index 4fb54af..485e568 100644 --- a/goffmpeg.go +++ b/goffmpeg.go @@ -12,6 +12,39 @@ "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 @@ -59,6 +92,65 @@ } } +// 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) @@ -94,3 +186,37 @@ C.wrap_fn_run(unsafe.Pointer(libcffmpeg), h.ffmpeg, in) } + +// Run2 ffmpeg +func (h *GoFFMPEG) Run2(input string, minDuration int) { + in := C.CString(input) + defer C.free(unsafe.Pointer(in)) + + C.wrap_fn_run(unsafe.Pointer(libcffmpeg), h.ffmpeg, in) +} + +// 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 + } + 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