【杂项】使用gocv库实现视频转终端字符画

1
package main
2

3
import (
4
  "bytes"
5
  "fmt"
6
  "gocv.io/x/gocv"
7
  "golang.org/x/crypto/ssh/terminal"
8
  "image"
9
  "os"
10
  "os/exec"
11
  "time"
12
)
13

14
const pixels = ` .,-'` + `:!1+*abcdefghijklmnopqrstuvwxyz<>()\/{}[]?234567890ABCDEFGHIJKLMNOPQRSTUVWXYZ%&@#$`
15

16
func main() {
17
  // 打开文件
18
  c, err := gocv.VideoCaptureFile("../1.mp4")
19
  // 获取帧数
20
  fps := c.Get(gocv.VideoCaptureFPS)
21
  if err != nil {
22
    panic(err)
23
  }
24

25
  mat := gocv.NewMat()
26
  w, h := 0, 0
27
  // 放大缩小倍率
28
  b := 0
29
  // 每一帧之间时长
30
  s := int64((1 / fps) * 1e6)
31

32
  var t time.Time
33
  // 每一帧时长与每一帧计算时长之差
34
  var sub int64
35

36
  var buf bytes.Buffer
37
  // 转化大小
38
  point := image.Point{Y: 0, X: 0}
39
  for c.Read(&mat) {
40
    t = time.Now()
41
    // 获取终端大小
42
    wl, hl, _ := terminal.GetSize(int(os.Stdout.Fd()))
43
    // 如果终端大小发生变化则重新计算放大缩小倍率
44
    if wl != w || hl != h {
45
      w = wl
46
      h = hl
47
      x, y := mat.Rows(), mat.Cols()
48
      if hl > wl {
49
        if x < wl {
50
          b = wl / x
51
          x *= b
52
          y *= b
53
        } else {
54
          b = x / wl
55
          x /= b
56
          y /= b
57
        }
58
      } else {
59
        if y < hl {
60
          b = hl / y
61
          x *= b
62
          y *= b
63
        } else {
64
          b = y / hl
65
          x /= b
66
          y /= b
67
        }
68
      }
69
      point.X = x
70
      point.Y = y
71
    }
72
    // 转灰度
73
    gocv.CvtColor(mat, &mat, gocv.ColorBGRToGray)
74
    // resize图像大小
75
    gocv.Resize(mat, &mat, point, 0, 0, gocv.InterpolationLinear)
76
    // 图像归一化
77
    mat.DivideFloat(255 / float32(len(pixels)-1))
78
    cols, rows := mat.Cols(), mat.Rows()
79
    buf.Reset()
80
    for i := 0; i < rows; i++ {
81
      for j := 0; j < cols; j++ {
82

83
        // 图像转化字符
84
        buf.WriteByte(pixels[mat.GetSCharAt(i, j)])
85
      }
86
      buf.WriteByte('\n')
87
    }
88
    sub= time.Now().Sub(t).Microseconds()
89
    // 如果比较计算时间与每一帧图像时间
90
    if sub < s {
91
      // 休眠，使字符显示时间靠近源视频每一帧时间
92
      time.Sleep(time.Duration(s-sub) * time.Microsecond)
93
    }
94
    // 清屏
95
    cmd := exec.Command("bash", "-c", "clear")
96
    cmd.Stdout = os.Stdout
97
    cmd.Run()
98
    fmt.Print(buf.String())
99
  }
100
}