video analysis2.0拆分,ffmpeg封装go接口库
zhangmeng
2021-05-25 48c8a8ce082ebdb992cb52cdd766298e8ca19011
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
package goffmpeg
 
/*
#include <stdlib.h>
#include "libcffmpeg.h"
*/
import "C"
import "unsafe"
 
// BuildDecoder build decoder
func (h *GoFFMPEG) BuildDecoder() {
    C.wrap_fn_decoder(unsafe.Pointer(libcffmpeg), h.ffmpeg)
}
 
// GetYUV get yuv data
func (h *GoFFMPEG) GetYUV() ([]byte, int, int, int64) {
    var fid C.long
    var length C.int
    var srcW, srcH, srcF C.int
 
    var yuv *C.uchar
    r := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &yuv, &srcW, &srcH, &srcF, &length, &fid)
 
    if r < 0 {
        return nil, int(r), int(r), int64(r)
    }
 
    if yuv == nil || length == 0 || srcW == 0 || srcH == 0 {
        return nil, 0, 0, 0
    }
 
    defer C.free(unsafe.Pointer(yuv))
 
    return C.GoBytes(unsafe.Pointer(yuv), length), int(srcW), int(srcH), int64(fid)
}
 
// GetYUV2 get yuv data
func (h *GoFFMPEG) GetYUV2() (unsafe.Pointer, []byte, int, int, int64) {
    var fid C.long
    var length C.int
    var srcW, srcH, srcF C.int
 
    var yuv *C.uchar
    r := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &yuv, &srcW, &srcH, &srcF, &length, &fid)
 
    if r < 0 {
        return nil, nil, int(r), int(r), int64(r)
    }
 
    if yuv == nil || length == 0 || srcW == 0 || srcH == 0 {
        return nil, nil, 0, 0, 0
    }
 
    const maxLen = 0x7fffffff
    size := int(length)
    data := (*[maxLen]byte)(unsafe.Pointer(yuv))[:size:size]
 
    return unsafe.Pointer(yuv), data, int(srcW), int(srcH), int64(fid)
}
 
// SetDecMinDuration min duration
// 主要是如果解码文件时长太短,可能在sdk收到解码数据前解码进程运行结束,所以设定一个最短的duration
func (h *GoFFMPEG) SetDecMinDuration(min, max int) {
    C.wrap_fn_rec_duration(unsafe.Pointer(libcffmpeg), h.ffmpeg, C.int(min), C.int(max))
}