#include "videodec.h"

VideoDec::VideoDec(QObject *parent) :
    QObject(parent)
{
}
void VideoDec::init()
{
    BUFFSIZE = 0;
    avcodec_register_all();
    av_register_all();
    pFormatCtx = avformat_alloc_context();
    pCodecCtx = avcodec_alloc_context3(pCodec);
    if(avformat_open_input(&pFormatCtx,FileName,NULL,NULL)!=0){
        qDebug()<<"OpenFail";
    }
    if(avformat_find_stream_info(pFormatCtx,NULL)<0){
        qDebug()<<"FindFail";
    }
    videoindex = -1;
    for(int i=0;pFormatCtx->nb_streams;i++){
        if(pFormatCtx->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO){
            videoindex = i;
            break;
        }
    }
    if(videoindex == -1){
        qDebug()<<"Din't find video stream";
    }
    pCodecCtx = pFormatCtx->streams[videoindex]->codec;
    pCodec = avcodec_find_decoder(pCodecCtx->codec_id);
    if(pCodec == NULL){
        qDebug()<<"codec not find";
    }
    if(avcodec_open2(pCodecCtx,pCodec,NULL)<0){
        qDebug()<<"can't open codec";
    }
//    pFrame = avcodec_alloc_frame();
//    pFrameRGB = avcodec_alloc_frame();
    pFrame = av_frame_alloc();
    pFrameRGB = av_frame_alloc();
    play();
}
void VideoDec::play()
{
    int got_picture;
    av_new_packet(&packet, pCodecCtx->width*pCodecCtx->height);

    uint8_t *out_buffer;
    out_buffer = new uint8_t[avpicture_get_size(AV_PIX_FMT_RGB32, pCodecCtx->width, pCodecCtx->height)];//分配AVFrame所需内存
    avpicture_fill((AVPicture *)pFrameRGB, out_buffer, AV_PIX_FMT_RGB32, pCodecCtx->width, pCodecCtx->height);//填充AVFrame
    avpicture_fill((AVPicture *)pFrame, out_buffer, AV_PIX_FMT_RGB32, pCodecCtx->width, pCodecCtx->height);//填充AVFrame
    while(1){
       // QThread::msleep(1);
        if(iSopflag==1)
            return;
        if(BUFFSIZE>51) continue;
        if(iSopflag==1)
            return;
        if(!(av_read_frame(pFormatCtx,&packet)>=0)) break;
        if(packet.stream_index == videoindex){
            int rec = avcodec_decode_video2(pCodecCtx,pFrame,&got_picture,&packet);
            if(rec>0){
                SwsContext *convertCtx = sws_getContext(pCodecCtx->width,pCodecCtx->height
                                                        ,pCodecCtx->pix_fmt,pCodecCtx->width,pCodecCtx->height
                                                        ,AV_PIX_FMT_RGB32,SWS_BICUBIC,NULL,NULL,NULL);
                sws_scale(convertCtx,(const uint8_t*  const*)pFrame->data,pFrame->linesize,0
                          ,pCodecCtx->height,pFrameRGB->data,pFrameRGB->linesize);
                QImage Img((uchar *)pFrameRGB->data[0],pCodecCtx->width,pCodecCtx->height,QImage::Format_RGB32);
                VideoImg.append(Img.copy());
                BUFFSIZE++;
                sws_freeContext(convertCtx);
            }

            av_free_packet(&packet);
        }
        if(iSopflag==1)
            return;
    }
}
void VideoDec::stop()
{
    iSopflag=1;
}