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

利用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.csdn.net/huangqi734044860/article/details/60956995

 

五：结论

终于搞定了。