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

转载自：http://blog.youkuaiyun.com/huangqi734044860/article/details/61194287

---

利用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）效果图

注意：这样就把视频给截取出来了。完全和视频一致。
关于FFMPEG的搭建，见
http://blog.youkuaiyun.com/huangqi734044860/article/details/60956995

五：结论
终于搞定了。