video analysis2.0拆分,ffmpeg封装go接口库
zhangmeng
2019-10-10 cd3fe8cc1ae9028acb4f630ed16c12f4fb327f3c
goffmpeg.go
@@ -15,12 +15,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
@@ -39,16 +39,52 @@
   ffmpeg C.cffmpeg
}
// New create handle
func New() *GoFFMPEG {
// New 2nd new
func New(GB, CPU bool) *GoFFMPEG {
   f := C.wrap_fn_create()
   if f == nil {
      return nil
   }
   if GB {
      C.wrap_fn_run_gb28181(f)
   }
   if CPU {
      C.wrap_fn_use_cpu(f)
   }
   return &GoFFMPEG{
      ffmpeg: C.wrap_fn_create(),
      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(lf)
   if f == nil {
      return nil
   }
   if GB {
      C.wrap_fn_run_gb28181(f)
   }
   if CPU {
      C.wrap_fn_use_cpu(f)
   }
   return &GoFFMPEG{
      ffmpeg: f,
   }
}
// Free free handle
func (h *GoFFMPEG) Free() {
   C.wrap_fn_destroy(h.ffmpeg)
   if h.ffmpeg != nil {
      C.wrap_fn_destroy(h.ffmpeg)
   }
}
// Run ffmpeg
@@ -57,23 +93,4 @@
   defer C.free(unsafe.Pointer(in))
   C.wrap_fn_run(h.ffmpeg, in)
}
// DecodeJPEG decode jpeg file
func (h *GoFFMPEG) DecodeJPEG(input string) ([]byte, int, 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)
   if width > 0 && height > 0 {
      data := C.GoBytes(p, width*height*3)
      wid := int(width)
      hei := int(height)
      return data, wid, hei
   }
   return nil, 0, 0
}