自制录屏软件,不到30行代码(仅供学习研究)

文/IT可达鸭

图/IT可达鸭、 络

写在前面

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

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

  • 环境配置
  • python版本: 3.6.0

    项目所需要的环境安装包

    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)等等。


    如果有疑问想获取源码,可以关注后,在后台私信我,回复:python录屏。 我把源码发你。持续关注”IT可达鸭” ,每天分享有趣Python源码。最后,感谢大家的阅读,祝大家工作生活愉快!

    声明:本站部分文章及图片源自用户投稿,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!

    上一篇 2020年3月23日
    下一篇 2020年3月23日

    相关推荐