add log

zhangmeng

2019-07-26 f6aa35eedbc76e7467a68dbe3aba63ec8cb352a5

 goffmpeg.go

@@ -12,15 +12,40 @@
   "unsafe"
)

const (
   // 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
)

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
@@ -34,15 +59,38 @@
   }
}

// Config config
type Config struct {
   Scale  int
   Width  int
   Height int
   GB     bool
   CPU    bool
}

// GoFFMPEG handle for c
type GoFFMPEG struct {
   ffmpeg C.cffmpeg
}

// New create handle
func New() *GoFFMPEG {
// New 2nd new
func New(conf Config) *GoFFMPEG {
   f := C.wrap_fn_create()
   if f == nil {
      return nil
   }
   if conf.Scale != 0 && conf.Width != 0 && conf.Height != 0 {
      C.wrap_fn_scale(f, C.int(conf.Width), C.int(conf.Height), C.int(conf.Scale))
   }
   if conf.GB {
      C.wrap_fn_run_gb28181(f)
   }
   if conf.CPU {
      C.wrap_fn_use_cpu(f)
   }

   return &GoFFMPEG{
      ffmpeg: C.wrap_fn_create(),
      ffmpeg: f,
   }
}

@@ -77,3 +125,48 @@
   }
   return nil, 0, 0
}

///////////////for encoder

// GoEncoder encoder
type GoEncoder struct {
   enc C.cencoder
}

// NewEncoder encoder
func NewEncoder(w, h, fps, br, sFlag, gi int) *GoEncoder {
   if w <= 0 || h <= 0 {
      return nil
   }

   return &GoEncoder{
      enc: C.wrap_fn_create_encoder(C.int(w), C.int(h), C.int(fps), C.int(br), C.int(sFlag), C.int(gi)),
   }
}

// Free free
func (e *GoEncoder) Free() {
   C.wrap_fn_destroy_encoder(e.enc)
}

// Encode pic
func (e *GoEncoder) Encode(in []byte, w, h int) ([]byte, int, bool) {

   var size C.int
   var key C.int
   cin := C.CBytes(in)
   defer C.free(cin)

   p := C.wrap_fn_encode(e.enc, cin, C.int(w), C.int(h), &size, &key)
   defer C.free(p)
   if p != nil && size > 0 {
      b := C.GoBytes(p, size)

      isKey := false
      if key > 0 {
         isKey = true
      }
      return b, int(size), isKey
   }
   return nil, 0, false
}