文/IT可达鸭

图/IT可达鸭、网络

写在前面

这次不蹭热点了，简简单单分享一个录屏软件的源码。我在阅读了近十多个录屏软件的源码后，发现这个是最简单，最好练手的一个源码。

为了便于理解，里面的一些不必要的代码我作了删减，主要是用来学习研究。

环境配置

python版本： 3.6.0

编辑器： pycharm

项目所需要的环境安装包

pip install numpypip install opencv-pythonpip install pillow需要注意的点

要先安装numpy，再安装opencv-python。

第一步：导包

time: 是一个时间包，其中time.sleep(n) 表示主线程睡眠 n 秒钟，子线程可以并发执行。

threading: 是一个关于线程操作的包，包括常用的线程函数、线程对象、锁、事件等。

cv2: 是一个强大的图像处理库 opencv, 据说人脸识别、视频处理等操作都可以做到。

numpy: 是一个扩展包，支持大量的维度数组与矩阵运算，包含大量数学函数。

PIL: 是一个图像处理标准库，一般和opencv一起用。

进程和线程简单介绍

什么是进程？举个例子，音乐播放器是一个进程，浏览器也是一个进程，互相不干扰。

什么是线程？音乐播放器中，音频播放是一个线程、歌词播放也是一个线程，它们之间有交流，保证歌词和音频同步播放。

第二步：录屏线程编写

写一个屏幕录制线程工具类，继承threading.Tread。重写__init__()方法，调用threading.Thread的__init__()方法。重写run()方法，不断截屏，并保存到video 对象中。利用 b_record 来控制是否停止录屏。这是一个线程开发的基本写法，只要涉及多线程，都可以这么写。

后续如果想加一个录制麦克风声音，可以写一个声音录制的线程类。

第三步：主函数

首先设定好录屏保存的路径，新建一个录屏对象。start()启动录制线程，time.sleep(10)表示主线程停留10秒，但是录屏线程依旧运行。直到主线程运行十秒后，运行 stop_record()，修改b_record 变量为Flase，录屏线程检测变化到就会立刻停止录屏，并输出保存视频文件。

完整代码# encoding:utf-8import timeimport threadingfrom cv2 import cv2import numpy as npfrom PIL import ImageGrabclass VideoCapThread(threading.Thread): def __init__(self, video_file='record.avi'): threading.Thread.__init__(self) self.b_record = True self.video = cv2.VideoWriter(video_file, cv2.VideoWriter_fourcc(*'XVID'), 32, ImageGrab.grab().size) # 帧率为32，可以调节 def run(self): while self.b_record: im = ImageGrab.grab() imm = cv2.cvtColor(np.array(im), cv2.COLOR_RGB2BGR) self.video.write(imm) self.video.release() cv2.destroyAllWindows() def stop_record(self): self.b_record = Falseif __name__ == '__main__': avi_file = 'data/avi_tmp.avi' t1 = VideoCapThread(avi_file) t1.start() time.sleep(10) # 录制10s t1.stop_re

最后，附上完整代码。仅有三十多行，是不是非常简洁？其实，编程也没那么难。

结语

当然，距离完整的录屏软件还有很多东西可以做。如果有兴趣深究下去，可以在这个源码的基础上做优化和改进。例如，录制屏幕的分辨率设置、录屏位置坐标自由设置、增加UI界面(QtPython)、增加声音或者麦克风录制、视频音频合并(FFmpeg)等等。