package goffmpeg
|
|
/*
|
#cgo LDFLAGS: -ldl
|
#include <stdlib.h>
|
#include "libcffmpeg.h"
|
*/
|
import "C"
|
import (
|
"errors"
|
"fmt"
|
"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
|
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 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{
|
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() {
|
if h.ffmpeg != nil {
|
C.wrap_fn_destroy(unsafe.Pointer(libcffmpeg), h.ffmpeg)
|
}
|
}
|
|
// Run ffmpeg
|
func (h *GoFFMPEG) Run(input string) {
|
in := C.CString(input)
|
defer C.free(unsafe.Pointer(in))
|
|
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
|
}
|
