保存触发id之后的视频作为触发视频

zhangmeng

2020-07-24 f93ee1a42e8c47e472332287b7350b66a6b0fa11

 goffmpeg.go

@@ -13,6 +13,8 @@
)

const (
   // ScaleNone self add no scale raw frame data
   ScaleNone = 0
   // ScaleFastBilinear SWS_FAST_BILINEAR
   ScaleFastBilinear = 1
   // ScaleBilinear SWS_BILINEAR
@@ -36,6 +38,12 @@
   // ScaleSpline SWS_SPLINE
   ScaleSpline = 0x400
)

// SrcFormat format NV
const SrcFormat = 23

// DstFormat format
const DstFormat = 3

var libcffmpeg C.libcffmpeg

@@ -64,19 +72,42 @@
   ffmpeg C.cffmpeg
}

// New create handle
func New() *GoFFMPEG {
// 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: C.wrap_fn_create(),
      ffmpeg: f,
   }
}

// NewWithScale scale out pic
func NewWithScale(w, h int, flag int) *GoFFMPEG {
   f := C.wrap_fn_create()
   if f != nil {
      C.wrap_fn_scale(f, C.int(w), C.int(h), C.int(flag))
// 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,
   }
@@ -84,7 +115,9 @@

// Free free handle
func (h *GoFFMPEG) Free() {
   C.wrap_fn_destroy(h.ffmpeg)
   if h.ffmpeg != nil {
      C.wrap_fn_destroy(unsafe.Pointer(libcffmpeg), h.ffmpeg)
   }
}

// Run ffmpeg
@@ -92,24 +125,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) []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)
   defer C.free(unsafe.Pointer(pic))

   retJpg := C.GoBytes(unsafe.Pointer(pic), jpgLen)
   return retJpg
}