脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 软件编程 → C 语言 → OpenCV录屏功能

OpenCV实现简单录屏功能

 更新时间：2022年01月19日 15:16:42   作者：蓬 蒿 人  
这篇文章主要为大家详细介绍了OpenCV实现简单录屏功能，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

本文实例为大家分享了OpenCV实现简单录屏功能的具体代码，供大家参考，具体内容如下

OpenCV中VideoCapture和VideoWriter用于读写视频文件，这里的录屏功能用到VideoWriter，用于将捕获的屏幕的每一帧数据保存到视频文件。

VideoWriter写视频文件的步骤

1、bool open(const String& filename, int fourcc, double fps,Size frameSize, bool isColor = true);
2、void write(InputArray image);或者VideoWriter& operator << (const Mat& image);
3、void release();

下列代码用于获取屏幕的截图

int width = GetSystemMetrics(SM_CXSCREEN);
int height = GetSystemMetrics(SM_CYSCREEN);
HDC hdcScreen = GetDC(NULL);
HDC hdcMemDC = CreateCompatibleDC(hdcScreen);
HBITMAP hbmScreen = CreateCompatibleBitmap(hdcScreen, width, height);
 
BITMAPINFO bi;
bi.bmiHeader.biSize = sizeof(bi.bmiHeader);
bi.bmiHeader.biWidth = width;
bi.bmiHeader.biHeight = height;
bi.bmiHeader.biPlanes = 1;
bi.bmiHeader.biBitCount = 24;
bi.bmiHeader.biCompression = BI_RGB;
bi.bmiHeader.biSizeImage = 0;
bi.bmiHeader.biXPelsPerMeter = 0;
bi.bmiHeader.biYPelsPerMeter = 0;
bi.bmiHeader.biClrUsed = 0;
bi.bmiHeader.biClrImportant = 0;
SelectObject(hdcMemDC, hbmScreen);
int lineBytes = ((width * bi.bmiHeader.biBitCount + 31) / 32) * 4;//每行字节数必须是4字节的整数倍
int bmpSize = lineBytes * height;
char* lpbitmap = new char[bmpSize];
BitBlt(hdcMemDC, 0, 0, width, height, hdcScreen, 0, 0, SRCCOPY);
GetDIBits(hdcMemDC, hbmScreen, 0, height, lpbitmap, &bi, DIB_RGB_COLORS);

lpbitmap为屏幕的像素颜色数据，下列代码将lpbitmap作为一帧写到视频中（假设VideoWriter为已正常打开的VideoWriter实例）

cv::Mat bmpMat(height, width, CV_8UC3);
for (int i = 0; i < height; i++)
{
    int srcIndex = (height-i-1) * lineBytes;
    int destIndex = i * width * 3;
    memcpy(&bmpMat.data[destIndex],&lpbitmap[srcIndex],width*3);
}
videoWriter.write(bmpMat);//或videoWriter << bmpMat;

因为lpbitmap中的数据是从左下角到右上角排列，而视频帧图像的数据是从左上角到右下角排列，所以要将数据按行上下翻转，即lpbitmap第一行对应视频图像的最后一行。另外BMP图像数据每行的字节数必须是4字节的整数倍，而写入视频的Mat数据没有这个要求，即每行数据大小是图像实际宽度乘以每个颜色占用的字节数，所以实际每行拷贝的数据是width*3节字。

下面是一段测试代码，这里只录制100帧，实际使用中可通过命令行参数、快捷键或按钮等自行决定开始和结束时间，帧率这里也设为固定的25，其实也应该根据具体形况设定合适的值。最后别忘了将opencv_ffmpegXXX.dll文件放到可执行文件目录下。

#include<windows.h>
#include"opencv2/opencv.hpp"
int main()
{
 
    cv::VideoWriter videoWriter;
    double fps = 25;
    int codec = cv::VideoWriter::fourcc('m', 'p', '4', 'v');
    int width =  GetSystemMetrics(SM_CXSCREEN);
    int height = GetSystemMetrics(SM_CYSCREEN);
 
    time_t seconds = time(0);
    int s = seconds % 60;
    int m = (seconds % 3600) / 60;
    int h = (seconds % (3600 * 24)) / 3600 + 8;
    char timeBuf[128] = { 0 };
    sprintf_s(timeBuf, "CaptureScreen-%d-%d-%d.mp4", h, m, s);
    cv::String filePath = timeBuf;
    videoWriter.open(filePath, codec, fps, cv::Size(width, height), true);
    if (!videoWriter.isOpened())
    {
        return -1;
    }
 
    HDC hdcScreen = GetDC(NULL);
    HDC hdcMemDC = CreateCompatibleDC(hdcScreen);
    HBITMAP hbmScreen = CreateCompatibleBitmap(hdcScreen, width, height);
 
    BITMAPINFO bi;
    bi.bmiHeader.biSize = sizeof(bi.bmiHeader);
    bi.bmiHeader.biWidth = width;
    bi.bmiHeader.biHeight = height;
    bi.bmiHeader.biPlanes = 1;
    bi.bmiHeader.biBitCount = 24;
    bi.bmiHeader.biCompression = BI_RGB;
    bi.bmiHeader.biSizeImage = 0;
    bi.bmiHeader.biXPelsPerMeter = 0;
    bi.bmiHeader.biYPelsPerMeter = 0;
    bi.bmiHeader.biClrUsed = 0;
    bi.bmiHeader.biClrImportant = 0;
    SelectObject(hdcMemDC, hbmScreen);
 
    int lineBytes = ((width * bi.bmiHeader.biBitCount + 31) / 32) * 4;
    int bmpSize = lineBytes * height;
    char* lpbitmap = new char[bmpSize];
    cv::Mat bmpMat(height, width, CV_8UC3);
    for (int i=0;i<100;i++)
    {
        if (BitBlt(hdcMemDC, 0, 0, width, height, hdcScreen, 0, 0, SRCCOPY))
        {
            GetDIBits(hdcMemDC, hbmScreen, 0, height, lpbitmap, &bi, DIB_RGB_COLORS);
            for (int i = 0; i < height; i++)
            {
                int srcIndex = (height-i-1) * lineBytes;
                int destIndex = i * width * 3;
                memcpy(&bmpMat.data[destIndex],&lpbitmap[srcIndex],width*3);
            }
            videoWriter.write(bmpMat);//videoWriter << bmpMat;
        }
    }
    delete[] lpbitmap;
    if (videoWriter.isOpened())
    {
        videoWriter.release();
    }
    return 0;
}

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • 基于C语言实现UDP客户端

    基于C语言实现UDP客户端

    UDP是一种面向无连接的传输层协议,广泛应用于实时性要求较高的场景,本文将介绍如何使用C语言实现一个简单的UDP客户端程序,有需要的可以参考下
    2024-10-10
  • C++获得本机所有网卡的IP和MAC地址信息的实现方法

    C++获得本机所有网卡的IP和MAC地址信息的实现方法

    下面小编就为大家带来一篇C++获得本机所有网卡的IP和MAC地址信息的实现方法。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2016-10-10
  • C++使用宏函数实现单例模板详解

    C++使用宏函数实现单例模板详解

    在我们日常开发中，无可避免需要使用单例模式进行设计类对象。这篇文章主要介绍了如何使用宏函数实现单例模板，感兴趣的小伙伴可以了解一下
    2023-02-02
  • C++数据结构与算法之双缓存队列实现方法详解

    C++数据结构与算法之双缓存队列实现方法详解

    这篇文章主要介绍了C++数据结构与算法之双缓存队列实现方法,结合实例形式分析了双缓存队列的原理、实现方法与相关注意事项,需要的朋友可以参考下
    2017-08-08
  • C++ 自由存储区是否等价于堆你知道吗

    C++ 自由存储区是否等价于堆你知道吗

    自由存储是C++中通过new与delete动态分配和释放对象的抽象概念，而堆（heap）是C语言和操作系统的术语，是操作系统维护的一块动态分配内存
    2021-08-08
  • C++List容器常用函数接口刨析

    C++List容器常用函数接口刨析

    最近我学习了C++中的STL库中的list容器，对于常用容器，我们不仅要会使用其常用的函数接口，我们还有明白这些接口在其底层是如何实现的。所以特意整理出来一篇博客供我们学习
    2022-08-08
  • C++智能指针shared_ptr

    C++智能指针shared_ptr

    这篇文章主要介绍了C++智能指针shared_ptr，C++11中包括shared_ptr在内的多种指针，都是模板类型，意味着使用者可以指定想要操作的类型下文从shared_ptr创建方式展开全文，介绍详细具有一的参考价值，需要的小伙伴可以参考一下
    2022-03-03
  • C语言中字符串的两种定义方式详解

    C语言中字符串的两种定义方式详解

    这篇文章主要为大家详细介绍了C语言中字符串的两种定义方式，小编觉得这篇文章写的还不错，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2021-10-10
  • JetBrains CLion永久激活超详细教程(最新激活方法)

    JetBrains CLion永久激活超详细教程(最新激活方法)

    JetBrains Clion 是一款专为 C/C++ 开发所设计的跨平台 IDE，本文适用 JetBrains CLion v2019.3/3.1/3.2/3.3 永久激活，附破解补丁和激活码，可以永久激活 Windows、MAC、Linux 下的 CLion，下面给大家分享JetBrains CLion永久激活超详细教程，感兴趣的朋友一起看看吧
    2023-01-01
  • 关于C语言函数strstr()的分析以及实现

    关于C语言函数strstr()的分析以及实现

    以下是对C语言中strstr()函数的使用进行了详细的分析介绍，需要的朋友可以参考下
    2013-07-07

最新评论