qt ffmpeg 视频文件

#include "xvideothread2.h" XVideoThread2::XVideoThread2(QObject *parent) : QObject(parent) { qDebug() << "XVideoThread2"; } XVideoThread2::~XVideoThread2() { } void XVideoThread2::setRunflag(int flag) { runFlag = flag; } void XVideoThread2::handleResults() { } std::string XVideoThread2::strPath() const { return m_strPath; } void XVideoThread2::setStrPath(const std::string &strPath) { runFlag = 0; m_strPath = strPath; } void XVideoThread2::start() { qDebug() << "start()"; int nGotPic = 0; av_register_all(); avformat_network_init(); m_pInFmtCtx = avformat_alloc_context(); runFlag = 1; if (runFlag == 0) { return; } qDebug() << "m_strPath" << QString::fromStdString(m_strPath); if(avformat_open_input(&m_pInFmtCtx, m_strPath.c_str(), NULL, NULL))// TODO 测试 // if(avformat_open_input(&m_pInFmtCtx, "D:/27a434687f7d420ab46057ba377621610.wmv", NULL, NULL)) { return; } if(avformat_find_stream_info(m_pInFmtCtx, NULL) < 0) { return; } int nVideoIndex = -1; for(int i = 0; i < m_pInFmtCtx->nb_streams; i++) { if(m_pInFmtCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) { nVideoIndex = i; break; } } if(nVideoIndex == -1) { return; } m_pCodecCtx = m_pInFmtCtx->streams[nVideoIndex]->codec; m_pCodec = avcodec_find_decoder(m_pCodecCtx->codec_id); if(!m_pCodec) { return; } if (avcodec_open2(m_pCodecCtx, m_pCodec, NULL) < 0) { return; } m_pFrame = av_frame_alloc(); m_pFrameRGB = av_frame_alloc(); m_pOutBuf = (uint8_t*)av_malloc(avpicture_get_size(AV_PIX_FMT_RGB32, m_pCodecCtx->width, m_pCodecCtx->height)); avpicture_fill((AVPicture*)m_pFrameRGB, m_pOutBuf, AV_PIX_FMT_RGB32, m_pCodecCtx->width, m_pCodecCtx->height); pImgCtx = sws_getContext(m_pCodecCtx->width, m_pCodecCtx->height, m_pCodecCtx->pix_fmt, m_pCodecCtx->width, m_pCodecCtx->height, AV_PIX_FMT_RGB32, SWS_BICUBIC, NULL, NULL, NULL); int nSize = m_pCodecCtx->width * m_pCodecCtx->height; m_pPacket = (AVPacket *)av_malloc(sizeof(AVPacket)); if(av_new_packet(m_pPacket, nSize) != 0) { } while (runFlag == 1) { if(av_read_frame(m_pInFmtCtx, m_pPacket) >= 0) { if(m_pPacket->stream_index == nVideoIndex) { nGotPic = 0; if(avcodec_decode_video2(m_pCodecCtx, m_pFrame, &nGotPic, m_pPacket) < 0) { return; } if(nGotPic) { sws_scale(pImgCtx, (const uint8_t* const*)m_pFrame->data, m_pFrame->linesize, 0, m_pCodecCtx->height, m_pFrameRGB->data, m_pFrameRGB->linesize); pImage = QImage((uchar*)m_pOutBuf, m_pCodecCtx->width, m_pCodecCtx->height, QImage::Format_RGB32); // qDebug() << "width" << QString::number(m_pCodecCtx->width) << "height"<< QString::number(m_pCodecCtx->height); if (runFlag == 1) { emit sig_SendOneFrame(pImage); } } } } av_packet_unref(m_pPacket); QThread::msleep(20); } qDebug() << "av_free_packet(m_pPacket)111"; av_packet_unref(m_pPacket); av_free_packet(m_pPacket); av_free(m_pOutBuf); av_frame_free(&m_pFrame); av_frame_free(&m_pFrameRGB); av_free(m_pCodec); avcodec_close(m_pCodecCtx); avcodec_free_context(&m_pCodecCtx); avformat_close_input(&m_pInFmtCtx); avformat_free_context(m_pInFmtCtx); sws_freeContext(pImgCtx); qDebug() << "av_free_packet(m_pPacket)000"; } QImage XVideoThread2::Mat2QImage(const Mat &mat) { if(mat.type()==CV_8UC1) { // cout<<"1"<<endl; // Set the color table (used to translate colour indexes to qRgb values) QVector<QRgb> colorTable; for (int i=0; i<256; i++) colorTable.push_back(qRgb(i,i,i)); // Copy input Mat const uchar *qImageBuffer = (const uchar*)mat.data; // Create QImage with same dimensions as input Mat QImage img(qImageBuffer, mat.cols, mat.rows, mat.step, QImage::Format_Indexed8); img.setColorTable(colorTable); return img; } // 8-bits unsigned, NO. OF CHANNELS=3 if(mat.type()==CV_8UC3) { // cout<<"3"<<endl; // Copy input Mat const uchar *qImageBuffer = (const uchar*)mat.data; // Create QImage with same dimensions as input Mat QImage img(qImageBuffer, mat.cols, mat.rows, mat.step, QImage::Format_RGB888); return img.rgbSwapped(); } else { qDebug() << "ERROR: Mat could not be converted to QImage."; return QImage(); } }