add new capture method: i-frame to jpg

chenshijun

2019-12-04 4f7ab713e2ccc9a1525bd72dc45df7f98a9e85a6

add new capture method: i-frame to jpg

 capture.go

@@ -39,6 +39,31 @@
)

// Capture pic
func CaptureGB28181(soFile string, url string, maxTry int) ([]byte, error) {

    var ret []byte
    var err error

    err = goffmpeg.InitFFmpeg(soFile)
    if err != nil {
        return nil, err
    }

    reterr := errors.New(fmt.Sprintf("try %d times to capture image, is url correct?", maxTry))
    for i := 0; i < maxTry; i++ {
        ret = goffmpeg.GetGBJpg(url)
        if len(ret) > 0 {
            reterr = nil
            break
        }
    }

    goffmpeg.FreeFFmpeg()

    return ret, reterr
}

// Capture pic
func Capture(soFile string, m StreamMode, url string, ext PicExt, w, h, maxTry int) ([]byte, error) {
    if m != GB28181 && m != Rtsp {
        return nil, errors.New("there is no this mode, try capture.Rtsp/capture.GB28181")