-
Notifications
You must be signed in to change notification settings - Fork 0
/
ffmpeg.go
55 lines (51 loc) · 1.52 KB
/
ffmpeg.go
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
package util
import (
"bytes"
"fmt"
"os"
"github.com/disintegration/imaging"
ffmpeg "github.com/u2takey/ffmpeg-go"
)
// 参考github示例 https://github.com/u2takey/ffmpeg-go
//从视频中抽取第 frameNum 帧作为封面
func GetVideoFrameQiNiu(inFileName string, frameNum int) (string, Response) {
// 使用ffmpeg
buf := bytes.NewBuffer(nil)
err := ffmpeg.Input(inFileName).
Filter("select", ffmpeg.Args{fmt.Sprintf("gte(n,%d)", frameNum)}).
Output("pipe:", ffmpeg.KwArgs{"vframes": 1, "format": "image2", "vcodec": "mjpeg"}).
WithOutput(buf, os.Stdout).
Run()
if err != nil { // ffmpeg获取数据流
return "Error", ReadFrameErr
}
// 上传到七牛云 ???待解决
url, resp := UploadImgToQiNiu(buf, int64(len(buf.Bytes())), GenFileName())
if url == "Error" {
return "Error", ImgUploadErr
}
return url, resp
}
//从视频中抽取第 frameNum 帧作为封面
func GetVideoFrame(inFileName, outFileName string, frameNum int) (string, Response) {
// 使用ffmpeg
buf := bytes.NewBuffer(nil)
err := ffmpeg.Input(inFileName).
Filter("select", ffmpeg.Args{fmt.Sprintf("gte(n,%d)", frameNum)}).
Output("pipe:", ffmpeg.KwArgs{"vframes": 1, "format": "image2", "vcodec": "mjpeg"}).
WithOutput(buf, os.Stdout).
Run()
if err != nil { // ffmpeg获取数据流
return "Error", ReadFrameErr
}
reader := buf
img, err := imaging.Decode(reader)
if err != nil {
return "Error", DecodeBufErr
}
err = imaging.Save(img, outFileName)
if err != nil {
return "Error", ImgUploadErr
}
return outFileName, Success
}