当前位置:   article > 正文

Python3 & OpenCV 视频转字符动画_frame = self.convert(rawframe, os.get_terminal_siz

frame = self.convert(rawframe, os.get_terminal_size(), fill=true)

Python3 & OpenCV 视频转字符动画

一、实验简介

1.1. 知识点

  • OpenCV 编译
  • 使用 OpenCV 处理图片、视频
  • 图片转字符画原理
  • 守护线程
  • 光标定位转义编码

1.2. 效果展示

(由于是在线环境,流畅度是不及本地环境的)

播放停止后的效果,注意终端中并无残留的动画字符:

此处输入图片的描述

二、实验步骤

2.1. 编译安装 OpenCV

由于安装过程需要等待很长时间,实验楼已经在环境中为大家安装OpenCV,可直接使用,以下安装步骤仅作参考

本课程的实验中使用了 OpenCV 3.1,因此我们需要编译安装它。

首先我们需要处理一个问题:当前实验楼的环境中 python3 命令使用的 python 版本为 3.5,但源中没有 python3.5-dev 的包,这会导致编译 numpy、OpenCV 出错,以及可能会出现其它问题。总而言之,我们需要将 python3 命令使用的 python 版本切换为 3.4。

$ sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.4 70 --slave /usr/bin/python3m python3m /usr/bin/python3.4m

然后安装一些依赖的包:

  1. $ sudo apt-get update
  2. $ sudo apt-get install python3-dev
  3. $ sudo pip3 install numpy
  4. $ sudo apt-get install cmake libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev

现在可以开始编译 OpenCV 3.1 了,不过这个编译过程极其耗费时间(2个小时多),我相信各位同学们是没有耐心等待编译完成的,所以我在后面提供了编译后的二进制文件下载链接,大家直接使用就可以了。

下面是在实验楼环境中编译 OpenCV 3.1 所需的命令,其他环境中的编译请参考官网:http://docs.opencv.org/master/d7/d9f/tutorial_linux_install.html

  1. $ wget https://github.com/Itseez/opencv/archive/3.1.0.zip
  2. $ unzip 3.1.0.zip && cd opencv-3.1.0/
  3. $ mkdir build && cd build
  4. $ cmake -D CMAKE_BUILD_TYPE=Release \
  5. -D CMAKE_INSTALL_PREFIX=/usr/local \
  6. PYTHON3_EXECUTABLE=/usr/bin/python3 \
  7. PYTHON_INCLUDE_DIR=/usr/include/python3.4 \
  8. PYTHON_LIBRARY=/usr/lib/x86_64-linux-gnu/libpython3.4m.so \
  9. PYTHON3_NUMPY_INCLUDE_DIRS=/usr/local/lib/python3.4/dist-packages/numpy/core/include ..
  10. $ make -j4

不想自己编译的同学请下载编译好的二进制文件,然后解压并进入 opencv-3.1.0/build 目录:

  1. $ wget http://labfile.oss.aliyuncs.com/courses/637/opencv-3.1.0.tar.gz
  2. $ tar xzvf opencv-3.1.0.tar.gz
  3. $ cd opencv-3.1.0/build

然后我们开始安装:

$ sudo make install

2.2. 程序原理

大家应该都明白视频其实可以看作一系列图片组成的,因此视频转字符动画最基本的便是图片转字符画,这一部分内容也在Python 图片转字符画 课程中有讲过。

在这里简单的说一下图片转字符画的原理:首先将图片转为灰度图,每个像素都只有亮度信息(用 0~255 表示)。然后我们构建一个有限字符集合,其中的每一个字符都与一段亮度范围对应,我们便可以根据此对应关系以及像素的亮度信息把每一个像素用对应的字符表示,这样字符画就形成了。

字符动画要能播放才有意义。最最简单粗暴的,用文本编辑器打开字符动画文本文件,然后狂按 PageDown 键就能播放。然而这真的太简单太粗暴了,一点都不优雅。

我们还是在终端里面播放字符动画,只需要一帧一帧输出就能达到动画的效果了,然而这却有一个很大的弊端:播放时,你会发现终端右边的滚动条会越来越小(如果有的话);播放完毕后,在终端中往上翻页,全是之前输出的字符画,播放前的命令历史全部被挤占掉了。在本实验后面提供了这个问题的解决办法。

2.3. 程序实现

为了避免忘记,先导入会使用到的包:

  1. import sys
  2. import os
  3. import time
  4. import threading
  5. import termios
  6. import tty
  7. import cv2
  8. import pyprind

最后一个 pyprind 模块提供显示用的进度条。因为从视频转为字符动画是一个耗时的过程,我们需要一个进度条来查看任务进度以及大致剩余时间,能更直观的了解程序状态。这样安装它:

$ sudo pip3 install pyprind

本实验将编写的程序除了能将视频文件转为字符动画并播放外,还顺便把图片文件转字符画的功能也添加上了,因此我们的程序设计有三个类:CharFrameI2CharV2Char,后两个类继承自第一个类。

CharFrame 类:

  1. class CharFrame:
  2. ascii_char = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. "
  3. # 像素映射到字符
  4. def pixelToChar(self, luminance):
  5. return self.ascii_char[int(luminance/256*len(self.ascii_char))]
  6. # 将普通帧转为 ASCII 字符帧
  7. def convert(self, img, limitSize=-1, fill=False, wrap=False):
  8. if limitSize != -1 and (img.shape[0] > limitSize[1] or img.shape[1] > limitSize[0]):
  9. img = cv2.resize(img, limitSize, interpolation=cv2.INTER_AREA)
  10. ascii_frame = ''
  11. blank = ''
  12. if fill:
  13. blank += ' '*(limitSize[0]-img.shape[1])
  14. if wrap:
  15. blank += '\n'
  16. for i in range(img.shape[0]):
  17. for j in range(img.shape[1]):
  18. ascii_frame += self.pixelToChar(img[i,j])
  19. ascii_frame += blank
  20. return ascii_frame

属性 ascii_char 可以根据你要转换的视频进行针对性调整。

pixelToChar() 方法只有一个参数,其接收像素的亮度信息。需要注意的是,方法的 return 语句中的表达式使用了一个数值 256,虽然像素的亮度范围是 0~255,但若是把 256 更改为 255,那么这个表达式将有可能引发 IndexError 异常。

convert() 方法有一个位置参数三个可选参数,参数 img 接收一个对象,这个对象类型是 numpy.ndarray,也就是 OpenCV 打开图片返回的对象,同样,之后用 OpenCV 得到的视频的帧也是这个对象。limitSize 参数接受一个元组,表示图片的限宽限高。fill 表示是否用空格填充图片宽度至限宽,wrap 表示是否在行末添加换行符。

img.shape 返回一个元组,含有图片的行数(高),列数(宽),以及颜色通道数。如果图片为灰度图则不包含颜色通道数。

cv2.resize() 函数用于缩放图片大小,第一个参数为 numpy.ndarray 对象,第二个是将缩放的宽高,interpolation 参数为插值方法。有几个插值方法可用,引用 OpenCV 官网的说明:

Preferable interpolation methods are cv2.INTER_AREA for shrinking and cv2.INTER_CUBIC (slow) &cv2.INTER_LINEAR for zooming. By default, interpolation method used is cv2.INTER_LINEAR for all resizing purposes.

img[i,j] 返回图片第 i 行第 j 列像素 BGR 值构成的列表,灰度图片则为直接返回对应像素的亮度值。

下面是继承自 CharFrame 的 I2Char 类:

  1. class I2Char(CharFrame):
  2. result = None
  3. def __init__(self, path, limitSize=-1, fill=False, wrap=False):
  4. self.genCharImage(path, limitSize, fill, wrap)
  5. def genCharImage(self, path, limitSize=-1, fill=False, wrap=False):
  6. img = cv2.imread(path, cv2.IMREAD_GRAYSCALE)
  7. if img is None:
  8. return
  9. self.result = self.convert(img, limitSize, fill, wrap)
  10. def show(self, stream = 2):
  11. if self.result is None:
  12. return
  13. if stream == 1 and os.isatty(sys.stdout.fileno()):
  14. self.streamOut = sys.stdout.write
  15. self.streamFlush = sys.stdout.flush
  16. elif stream == 2 and os.isatty(sys.stderr.fileno()):
  17. self.streamOut = sys.stderr.write
  18. self.streamFlush = sys.stderr.flush
  19. elif hasattr(stream, 'write'):
  20. self.streamOut = stream.write
  21. self.streamFlush = stream.flush
  22. self.streamOut(self.result)
  23. self.streamFlush()
  24. self.streamOut('\n')

cv2.imread() 读取一个图片,返回一个 numpy.ndarray 对象。第一个参数为要打开的图片路径,第二个参数指定图片的打开方式,可以具有下面三个值:

  • cv2.IMREAD_COLOR : 加载彩色图片,忽略透明通道。
  • cv2.IMREAD_GRAYSCALE : 以灰度模式加载 图片。
  • cv2.IMREAD_UNCHANGED : 加载包含透明通道的图片。

show() 方法接受一个参数,表示使用哪种输出流。1 代表输出流使用标准输出 sys.stdout2 代表输出流使用标准错误输出 sys.stderr。其中 sys.stdout.fileno() 和 sys.stderr.fileno() 分别返回标准输出和标准错误输出的文件描述符os.isatty(fd) 返回一个布尔值,当文件描述符 fd 是打开并连接到 tty 设备时返回真。

然后便是我们的重点了,继承自 CharFrame 类的 V2Char 类。

先设想一下我们的类,类的其中一个属性为 charVideo ,这是一个列表,用来存放字符动画全部数据。

然后是两个主要方法:一个是从视频文件转为字符动画的方法 genCharVideo(),一个是播放字符动画的方法 play()

另外由于从视频转到字符动画是一个耗时耗力的过程,所以我们可以把 charVideo 中的字符动画数据导出来,方便下次读取播放,这就意味着要有导出和读取方法,即 export() 方法和 load() 方法。

类还要有一个初始化方法,参数为要读取的文件路径,文件若是导出的 txt,则调用 load() 方法把数据加载到属性 charVideo 里。否则视为视频文件,调用 genCharVideo() 方法将视频转化为字符动画存放到属性 charVideo 里:

  1. def __init__(self, path):
  2. if path.endswith('txt'):
  3. self.load(path)
  4. else:
  5. self.genCharVideo(path)

接下来是 genCharVideo() 方法:

  1. def genCharVideo(self, filepath):
  2. self.charVideo = []
  3. cap = cv2.VideoCapture(filepath)
  4. self.timeInterval = round(1/cap.get(5), 3)
  5. nf = int(cap.get(7))
  6. print('Generate char video, please wait...')
  7. for i in pyprind.prog_bar(range(nf)):
  8. rawFrame = cv2.cvtColor(cap.read()[1], cv2.COLOR_BGR2GRAY)
  9. frame = self.convert(rawFrame, os.get_terminal_size(), fill=True)
  10. self.charVideo.append(frame)
  11. cap.release()

用 cv2.VideoCapture() 方法读取视频文件,生成的对象我们赋值给 cap

通过 cap.get() 方法我们可以获得视频的属性信息,比如 cap.get(3) 和 cap.get(4) 分别返回视频的宽高信息,cap.get(5) 则返回视频的帧率,cap.get(7) 返回视频的总帧数。

timeInterval 存放播放时间间隔,用来让之后播放字符动画的帧率与原视频相同。

pyprind.prog_bar() 是一个生成器,使用这个生成器进行迭代会自然在终端中输出进度条。

cap.read() 读取视频的下一帧,其返回一个两元素的元组,第一个元素为 bool 值,指示帧是否被正确读取,第二个元素为numpy.ndarray ,其存放的便是帧的数据。

cv2.cvtColor() 用来转换图像的颜色空间,第一个参数为图像对象,第二个参数指示转换类型,OpenCV 中有超过 150 个颜色空间转换,这里我们使用的是彩色转灰度 cv2.COLOR_BGR2GRAY

os.get_terminal_size() 方法返回当前终端的列数(宽),行数(高)。这里我们将 fill 参数设置为 True,未设置 wrap 参数,其值为默认的 False。在终端里如果打印的字符超过一行的宽度,终端会自动进行显示上的换行。

最后别忘了用 cap.release() 释放资源。

然后是 play() 方法,这就要接着之前说了,要想终端不在播放字符动画后充满没用的字符,可以使用光标定位转义编码。

我们可以这样:每输出一帧,就将光标移动到播放开始处,下一帧会从这个位置输出,并自动覆盖掉之前的内容,如此往复循环,播放完毕时,清除输出的最后一帧,这样终端就不会被字符画挤满了。

这里是一系列用来定位光标的转义编码(某些终端不支持一些转义编码),引自 The Linux Command Line

转义编码行动
\033[l;cH把光标移到第 l 行,第 c 列。
\033[nA把光标向上移动 n 行。
\033[nB把光标向下移动 n 行。
\033[nC把光标向前移动 n 个字符。
\033[nD把光标向后移动 n 个字符。
\033[2J清空屏幕,把光标移到左上角(第零行,第零列)。
\033[K清空从光标位置到当前行末的内容。
\033[s存储当前光标位置。
\033[u唤醒之前存储的光标位置。

还有一个问题,如何在中途停止播放?当然你可以按 Ctrl + C,但这样程序是没办法做扫尾工作的,终端会留下一堆没用的乱七八糟的字符。

我们这样设计,字符动画开始播放时启动一个守护线程,进程启动时便开始等待用户输入,一旦接收到输入便停止播放字符动画。

这里有两个地方需要注意:

  1. 不要使用 input() 方法接收字符输入
  2. 不能使用普通线程

对于第一点,如果是想按任意字符终止播放的话,那么就不应该使用 input(),否则要么你按回车停止播放,要么按其他字符再敲回车停止播放,总而言之,使用 input() 并不舒服。最好的办法是,使用类似于 C 语言中的 getchar() 方法,然而 python 并没有提供类似方法,这在后面的代码中会给出替代方案。

对于第二点,我们要明白如果任何一个派生线程仍在运行中,主线程是不会退出的,除非派生线程被设定为守护线程。所以使用普通线程的话,若用户并没有在中途停止播放,等到播放完毕后,这个线程会永远的运行下去,直到用户输入任意字符。如果我们可以在播放完毕时手动 kill 掉这个派生线程,这也不是问题,然而 python 并没有提供 kill 掉线程的方法。所以,只能把这个派生线程设为守护线程。等主线程退出后,程序只剩下一个守护线程在运行,守护线程会自动被 kill 掉,程序退出。

下面类 play() 方法的代码:

  1. def play(self, stream = 1):
  2. # Bug:
  3. # 光标定位转义编码不兼容 Windows
  4. if not self.charVideo:
  5. return
  6. if stream == 1 and os.isatty(sys.stdout.fileno()):
  7. self.streamOut = sys.stdout.write
  8. self.streamFlush = sys.stdout.flush
  9. elif stream == 2 and os.isatty(sys.stderr.fileno()):
  10. self.streamOut = sys.stderr.write
  11. self.streamFlush = sys.stderr.flush
  12. elif hasattr(stream, 'write'):
  13. self.streamOut = stream.write
  14. self.streamFlush = stream.flush
  15. old_settings = None
  16. breakflag = None
  17. # 获得标准输入的文件描述符
  18. fd = sys.stdin.fileno()
  19. def getChar():
  20. nonlocal breakflag
  21. nonlocal old_settings
  22. # 保存标准输入的属性
  23. old_settings = termios.tcgetattr(fd)
  24. # 设置标准输入为原始模式
  25. tty.setraw(sys.stdin.fileno())
  26. # 读取一个字符
  27. ch = sys.stdin.read(1)
  28. breakflag = True if ch else False
  29. # 创建线程
  30. getchar = threading.Thread(target=getChar)
  31. # 设置为守护线程
  32. getchar.daemon = True
  33. # 启动守护线程
  34. getchar.start()
  35. # 输出的字符画行数
  36. rows = len(self.charVideo[0])//os.get_terminal_size()[0]
  37. for frame in self.charVideo:
  38. # 接收到输入则退出循环
  39. if breakflag is True:
  40. break
  41. self.streamOut(frame)
  42. self.streamFlush()
  43. time.sleep(self.timeInterval)
  44. # 共 rows 行,光标上移 rows-1 行回到开始处
  45. self.streamOut('\033[{}A\r'.format(rows-1))
  46. # 恢复标准输入为原来的属性
  47. termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
  48. # 光标下移 rows-1 行到最后一行,清空最后一行
  49. self.streamOut('\033[{}B\033[K'.format(rows-1))
  50. # 清空最后一帧的所有行(从倒数第二行起)
  51. for i in range(rows-1):
  52. # 光标上移一行
  53. self.streamOut('\033[1A')
  54. # 清空光标所在行
  55. self.streamOut('\r\033[K')
  56. info = 'User interrupt!\n' if breakflag else 'Finished!\n'
  57. self.streamOut(info)

一些函数的文档链接:

termios.tcgetattr(fd)termios.tcsetattr(fd, when, attributes)链接

tty.setraw(fd, when=termios.TCSAFLUSH)链接

V2Char 类的完整代码如下:

  1. class V2Char(CharFrame):
  2. charVideo = []
  3. timeInterval = 0.033
  4. def __init__(self, path):
  5. if path.endswith('txt'):
  6. self.load(path)
  7. else:
  8. self.genCharVideo(path)
  9. def genCharVideo(self, filepath):
  10. self.charVideo = []
  11. cap = cv2.VideoCapture(filepath)
  12. self.timeInterval = round(1/cap.get(5), 3)
  13. nf = int(cap.get(7))
  14. print('Generate char video, please wait...')
  15. for i in pyprind.prog_bar(range(nf)):
  16. rawFrame = cv2.cvtColor(cap.read()[1], cv2.COLOR_BGR2GRAY)
  17. frame = self.convert(rawFrame, os.get_terminal_size(), fill=True)
  18. self.charVideo.append(frame)
  19. cap.release()
  20. def export(self, filepath):
  21. if not self.charVideo:
  22. return
  23. with open(filepath,'w') as f:
  24. for frame in self.charVideo:
  25. # 加一个换行符用以分隔每一帧
  26. f.write(frame + '\n')
  27. def load(self, filepath):
  28. self.charVideo = []
  29. # 一行即为一帧
  30. for i in open(filepath):
  31. self.charVideo.append(i[:-1])
  32. def play(self, stream = 1):
  33. # Bug:
  34. # 光标定位转义编码不兼容 Windows
  35. if not self.charVideo:
  36. return
  37. if stream == 1 and os.isatty(sys.stdout.fileno()):
  38. self.streamOut = sys.stdout.write
  39. self.streamFlush = sys.stdout.flush
  40. elif stream == 2 and os.isatty(sys.stderr.fileno()):
  41. self.streamOut = sys.stderr.write
  42. self.streamFlush = sys.stderr.flush
  43. elif hasattr(stream, 'write'):
  44. self.streamOut = stream.write
  45. self.streamFlush = stream.flush
  46. old_settings = None
  47. breakflag = None
  48. # 获得标准输入的文件描述符
  49. fd = sys.stdin.fileno()
  50. def getChar():
  51. nonlocal breakflag
  52. nonlocal old_settings
  53. # 保存标准输入的属性
  54. old_settings = termios.tcgetattr(fd)
  55. # 设置标准输入为原始模式
  56. tty.setraw(sys.stdin.fileno())
  57. # 读取一个字符
  58. ch = sys.stdin.read(1)
  59. breakflag = True if ch else False
  60. # 创建线程
  61. getchar = threading.Thread(target=getChar)
  62. # 设置为守护线程
  63. getchar.daemon = True
  64. # 启动守护线程
  65. getchar.start()
  66. # 输出的字符画行数
  67. rows = len(self.charVideo[0])//os.get_terminal_size()[0]
  68. for frame in self.charVideo:
  69. # 接收到输入则退出循环
  70. if breakflag is True:
  71. break
  72. self.streamOut(frame)
  73. self.streamFlush()
  74. time.sleep(self.timeInterval)
  75. # 共 rows 行,光标上移 rows-1 行回到开始处
  76. self.streamOut('\033[{}A\r'.format(rows-1))
  77. # 恢复标准输入为原来的属性
  78. termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
  79. # 光标下移 rows-1 行到最后一行,清空最后一行
  80. self.streamOut('\033[{}B\033[K'.format(rows-1))
  81. # 清空最后一帧的所有行(从倒数第二行起)
  82. for i in range(rows-1):
  83. # 光标上移一行
  84. self.streamOut('\033[1A')
  85. # 清空光标所在行
  86. self.streamOut('\r\033[K')
  87. info = 'User interrupt!\n' if breakflag else 'Finished!\n'
  88. self.streamOut(info)

2.4. 测试效果

把下面的代码键入到之前的代码后面,这份代码就可以用做视频转字符画的脚本文件了。

  1. if __name__ == '__main__':
  2. import argparse
  3. # 设置命令行参数
  4. parser = argparse.ArgumentParser()
  5. parser.add_argument('file',
  6. help='Video file or charvideo file')
  7. parser.add_argument('-e', '--export', nargs = '?', const = 'charvideo.txt',
  8. help='Export charvideo file')
  9. # 获取参数
  10. args = parser.parse_args()
  11. v2char = V2Char(args.file)
  12. if args.export:
  13. v2char.export(args.export)
  14. v2char.play()

用如下命令把测试用视频下载下来:

$ wget http://labfile.oss.aliyuncs.com/courses/637/BadApple.mp4

假设你的代码文件为 CLIPlayVideo.py,输入如下命令将 BadApple.mp4 文件转为字符动画并导出为文件,同时播放转换后的字符动画。

$ python3 CLIPlayVideo.py BadApple.mp4 -e

之后再次播放不必重新转换一次,可以直接读取导出的字符画,假设这个文件是 charvideo.txt:

$ python3 CLIPlayVideo.py charvideo.txt

三、实验总结

本实验使用了 OpenCV 处理图片及视频,虽然编译 OpenCV 的过程很恼人,但这也是一个必要的过程。这里的 OpenCV 的操作对于大家来说应该只是一个开始,如果想深入学习应该多看看官方文档。使用了守护线程后,相信大家能够明白守护线程是什么以及与普通线程有什么区别。最后还了解了光标定位转义编码,虽然这个用处可能并不大,但也是一个有趣的东西,它还可以做很多事情,大家可以多玩玩。

四、完整代码

可以直接把代码下载下来:

$ wget http://labfile.oss.aliyuncs.com/courses/637/CLIPlayVideo.py

下面贴出了完整的代码:

  1. import sys
  2. import os
  3. import time
  4. import threading
  5. import termios
  6. import tty
  7. import cv2
  8. import pyprind
  9. class CharFrame:
  10. ascii_char = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. "
  11. # 像素映射到字符
  12. def pixelToChar(self, luminance):
  13. return self.ascii_char[int(luminance/256*len(self.ascii_char))]
  14. # 将普通帧转为 ASCII 字符帧
  15. def convert(self, img, limitSize=-1, fill=False, wrap=False):
  16. if limitSize != -1 and (img.shape[0] > limitSize[1] or img.shape[1] > limitSize[0]):
  17. img = cv2.resize(img, limitSize, interpolation=cv2.INTER_AREA)
  18. ascii_frame = ''
  19. blank = ''
  20. if fill:
  21. blank += ' '*(limitSize[0]-img.shape[1])
  22. if wrap:
  23. blank += '\n'
  24. for i in range(img.shape[0]):
  25. for j in range(img.shape[1]):
  26. ascii_frame += self.pixelToChar(img[i,j])
  27. ascii_frame += blank
  28. return ascii_frame
  29. class I2Char(CharFrame):
  30. result = None
  31. def __init__(self, path, limitSize=-1, fill=False, wrap=False):
  32. self.genCharImage(path, limitSize, fill, wrap)
  33. def genCharImage(self, path, limitSize=-1, fill=False, wrap=False):
  34. img = cv2.imread(path, cv2.IMREAD_GRAYSCALE)
  35. if img is None:
  36. return
  37. self.result = self.convert(img, limitSize, fill, wrap)
  38. def show(self, stream = 2):
  39. if self.result is None:
  40. return
  41. if stream == 1 and os.isatty(sys.stdout.fileno()):
  42. self.streamOut = sys.stdout.write
  43. self.streamFlush = sys.stdout.flush
  44. elif stream == 2 and os.isatty(sys.stderr.fileno()):
  45. self.streamOut = sys.stderr.write
  46. self.streamFlush = sys.stderr.flush
  47. elif hasattr(stream, 'write'):
  48. self.streamOut = stream.write
  49. self.streamFlush = stream.flush
  50. self.streamOut(self.result)
  51. self.streamFlush()
  52. self.streamOut('\n')
  53. class V2Char(CharFrame):
  54. charVideo = []
  55. timeInterval = 0.033
  56. def __init__(self, path):
  57. if path.endswith('txt'):
  58. self.load(path)
  59. else:
  60. self.genCharVideo(path)
  61. def genCharVideo(self, filepath):
  62. self.charVideo = []
  63. cap = cv2.VideoCapture(filepath)
  64. self.timeInterval = round(1/cap.get(5), 3)
  65. nf = int(cap.get(7))
  66. print('Generate char video, please wait...')
  67. for i in pyprind.prog_bar(range(nf)):
  68. rawFrame = cv2.cvtColor(cap.read()[1], cv2.COLOR_BGR2GRAY)
  69. frame = self.convert(rawFrame, os.get_terminal_size(), fill=True)
  70. self.charVideo.append(frame)
  71. cap.release()
  72. def export(self, filepath):
  73. if not self.charVideo:
  74. return
  75. with open(filepath,'w') as f:
  76. for frame in self.charVideo:
  77. # 加一个换行符用以分隔每一帧
  78. f.write(frame + '\n')
  79. def load(self, filepath):
  80. self.charVideo = []
  81. # 一行即为一帧
  82. for i in open(filepath):
  83. self.charVideo.append(i[:-1])
  84. def play(self, stream = 1):
  85. # Bug:
  86. # 光标定位转义编码不兼容 Windows
  87. if not self.charVideo:
  88. return
  89. if stream == 1 and os.isatty(sys.stdout.fileno()):
  90. self.streamOut = sys.stdout.write
  91. self.streamFlush = sys.stdout.flush
  92. elif stream == 2 and os.isatty(sys.stderr.fileno()):
  93. self.streamOut = sys.stderr.write
  94. self.streamFlush = sys.stderr.flush
  95. elif hasattr(stream, 'write'):
  96. self.streamOut = stream.write
  97. self.streamFlush = stream.flush
  98. old_settings = None
  99. breakflag = None
  100. # 获得标准输入的文件描述符
  101. fd = sys.stdin.fileno()
  102. def getChar():
  103. nonlocal breakflag
  104. nonlocal old_settings
  105. # 保存标准输入的属性
  106. old_settings = termios.tcgetattr(fd)
  107. # 设置标准输入为原始模式
  108. tty.setraw(sys.stdin.fileno())
  109. # 读取一个字符
  110. ch = sys.stdin.read(1)
  111. breakflag = True if ch else False
  112. # 创建线程
  113. getchar = threading.Thread(target=getChar)
  114. # 设置为守护线程
  115. getchar.daemon = True
  116. # 启动守护线程
  117. getchar.start()
  118. # 输出的字符画行数
  119. rows = len(self.charVideo[0])//os.get_terminal_size()[0]
  120. for frame in self.charVideo:
  121. # 接收到输入则退出循环
  122. if breakflag is True:
  123. break
  124. self.streamOut(frame)
  125. self.streamFlush()
  126. time.sleep(self.timeInterval)
  127. # 共 rows 行,光标上移 rows-1 行回到开始处
  128. self.streamOut('\033[{}A\r'.format(rows-1))
  129. # 恢复标准输入为原来的属性
  130. termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
  131. # 光标下移 rows-1 行到最后一行,清空最后一行
  132. self.streamOut('\033[{}B\033[K'.format(rows-1))
  133. # 清空最后一帧的所有行(从倒数第二行起)
  134. for i in range(rows-1):
  135. # 光标上移一行
  136. self.streamOut('\033[1A')
  137. # 清空光标所在行
  138. self.streamOut('\r\033[K')
  139. info = 'User interrupt!\n' if breakflag else 'Finished!\n'
  140. self.streamOut(info)
  141. if __name__ == '__main__':
  142. import argparse
  143. # 设置命令行参数
  144. parser = argparse.ArgumentParser()
  145. parser.add_argument('file',
  146. help='Video file or charvideo file')
  147. parser.add_argument('-e', '--export', nargs = '?', const = 'charvideo.txt',
  148. help='Export charvideo file')
  149. # 获取参数
  150. args = parser.parse_args()
  151. v2char = V2Char(args.file)
  152. if args.export:
  153. v2char.export(args.export)
  154. v2char.play()
声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号