利用QT实现截屏

admin

利用QT实现截屏的四种方法；

一：截取整个屏幕

（1）代码如下：

void Test::slotGrabFullScreen()

{

    QScreen *screen = QGuiApplication::primaryScreen();

    QString filePathName = "full-";

    filePathName += QDateTime::currentDateTime().toString("yyyy-MM-dd hh-mm-ss-zzz");

    filePathName += ".jpg";

    if(!screen->grabWindow(0).save(filePathName, "jpg"))

    {

        cout<<"save full screen failed"<<endl;

    }

}

（2）效果如下图：



注意：此方法用于保存整个屏幕。

二：截取某个控件（QWidget）

（1）代码如下：

void Test::slotGrabWidgetScreen()

{

    QRect rect = ui.vw->geometry();

    QPixmap p = this->grab(QRect(0, 0, 1000, 800));

    QString filePathName = "widget";

    filePathName += QDateTime::currentDateTime().toString("yyyy-MM-dd hh-mm-ss-zzz");

    filePathName += ".png";

    if(!p.save(filePathName,"png"))

    {

        cout<<"save widget screen failed"<<endl;

    }

}

调用QWidget的grab方法。

（2）效果图：



注意：此方法对截取播放视频的widget无效。播放视频的widget（QVideoWidget）

三：截取视频图片

方法：先截取整个屏幕，保存为图片，然后计算获取到视频在图片的位置。

（1）代码如下：

void Test::slotCutScreen()

{

// 保存整个屏幕为QPixmap

    QScreen *screen = QGuiApplication::primaryScreen();

    QString filePathName = "cut-";

    filePathName += QDateTime::currentDateTime().toString("yyyy-MM-dd hh-mm-ss-zzz");

    filePathName += ".png";



    QPixmap pixmap = screen->grabWindow(0);

    if(!pixmap.save(filePathName,"png"))

    {

        cout<<"cut save png failed"<<endl;

    }   



// 计算视频的位置和大小

    cout<<"Full pixmap width: "<<pixmap.width()<<" height: "<<pixmap.height()<<endl;

    QRect geo = this->geometry();

    QRect appGeo = geo; // 整个应用程序在图片中的位置。

    cout<<"App x: "<<geo.x()<<" y: "<<geo.y()<<" width: "<<geo.width()<<" height: "<<geo.height()<<endl;



    geo = ui.vw->geometry(); // 播放视频在图片中的位置。

    cout<<"VW x: "<<geo.x()<<" y: "<<geo.y()<<" width: "<<geo.width()<<" height: "<<geo.height()<<endl;



    QWidget *centerWidget = centralWidget(); // QMainWindow在应用程序的位置

    QRect centerRect = centerWidget->geometry();

    cout<<"center x: "<<centerRect.x()<<" y: "<<centerRect.y()<<" width: "<<centerRect.width()<<" height: "<<centerRect.height()<<endl;



    QRect copyGeo;

    copyGeo.setX(geo.x() + appGeo.x() + centerRect.x()); // x=三个x相加

    copyGeo.setY(geo.y() + appGeo.y() + centerRect.y());

    copyGeo.setWidth(geo.width());

    copyGeo.setHeight(geo.height());

    cout<<"VW1 x: "<<copyGeo.x()<<" y: "<<copyGeo.y()<<" width: "<<copyGeo.width()<<" height: "<<copyGeo.height()<<endl;



    QPixmap pixmapCopy = pixmap.copy(copyGeo); // copy图片

    filePathName.prepend("Copy+");

    if(!pixmapCopy.save(filePathName,"png"))

    {

        cout<<"copy cut save png failed"<<endl;

    }

}

（2）效果图：整个屏幕



（3）截取视频



注意：此方法有两个弊端

A：此方法截图的视频图片与窗口大小有关。有可能截取出来的像素非常低（播放视频的窗口很小）

B：如果视频的分辨率与窗口大小不一致，则截取出来的图片有黑边框。（看上图左右两边有黑色边框）

三、利用FFMPEG获取视频图片

处理步骤：

1：获取一帧数据

2：利用FFMPEG将YUV格式转换为RGB格式

3：保存图片

（1）源代码

// 以下是两个成员变量。一个是player，一个是侦测帧对象

QMediaPlayer *mPlayer;

    QVideoProbe *mVideoProbe;



void Test::slotGrabMediaScreenFFMPEG()

{

    connect(mVideoProbe, SIGNAL(videoFrameProbed(QVideoFrame)), this, SLOT(slotProcessFrameFFMPEG(QVideoFrame)));   

}



void Test::slotProcessFrameFFMPEG(const QVideoFrame & buffer)

{

    disconnect(mVideoProbe, SIGNAL(videoFrameProbed(QVideoFrame)), this, SLOT(slotProcessFrameFFMPEG(QVideoFrame)));

    if(!buffer.isValid()) // 数据是否有效

    {

        cout<<"frame is invalid"<<endl;

        connect(mVideoProbe, SIGNAL(videoFrameProbed(QVideoFrame)), this, SLOT(slotProcessFrame(QVideoFrame)));

        return;

    }



    QImage img;

    QVideoFrame frame(buffer); // 拷贝数据

    frame.map(QAbstractVideoBuffer::ReadOnly); // 将视频缓存映射到内存中



    int totalBytes = frame.width() * frame.height() * 3;

    uchar *imageBuffer = (uchar*)malloc(totalBytes);

    if(!YV12ToARGB24_FFmpeg(frame.bits(), imageBuffer, frame.width(), frame.height()))

    {

        cout<<"convert YUV to RGB failed"<<endl;

        return;

    }



    img = QImage(imageBuffer, frame.width(), frame.height(), //frame.bytesPerLine(),

        //imageFormat);

        QImage::Format_RGB888);



    QString filePathName = "convert-vedio-";

    filePathName += QDateTime::currentDateTime().toString("yyyy-MM-dd hh-mm-ss-zzz");

    filePathName += ".png";

    if(!img.save(filePathName,"png"))

    {

        cout<<"save convert vedio screen failed"<<endl;

    }

}





bool Test::YV12ToARGB24_FFmpeg(unsigned char* pYUV,unsigned char* pBGR24,int width,int height)

{   

    if (width < 1 || height < 1 || pYUV == NULL || pBGR24 == NULL)

        return false;

    AVPicture pFrameYUV, pFrameBGR;

    avpicture_fill(&pFrameYUV, pYUV, PIX_FMT_NV12, width, height);

    avpicture_fill(&pFrameBGR, pBGR24, AV_PIX_FMT_RGB24, width,height);



    struct SwsContext* imgCtx = NULL;

    imgCtx = sws_getContext(width, height, PIX_FMT_NV12, width, height, AV_PIX_FMT_RGB24, SWS_BICUBIC, 0, 0, 0);

    if (imgCtx != NULL){

        sws_scale(imgCtx, pFrameYUV.data, pFrameYUV.linesize, 0, height, pFrameBGR.data, pFrameBGR.linesize);

        if(imgCtx){

            sws_freeContext(imgCtx);

            imgCtx = NULL;

        }

        return true;

    }

    else{

        sws_freeContext(imgCtx);

        imgCtx = NULL;

        return false;

    }

}



（2）效果图
————————————————
版权声明：本文为CSDN博主「huangqi734044860」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/huangqi734044860/article/details/61194287