ffmpeg 비디오 디코더(decoder) API 사용법 - how to use ffmpeg video decoder API

#define FFMPEG_0_8   // ffmpeg-1.0 이전 버전

extern "C" {
#include "libavcodec\avcodec.h"
}

#define VIDEO_CODEC_H264 0
#define VIDEO_CODEC_MPEG4 1

// 코덱관련 변수들 선언

class VideoDecoder {

    AVCodec *m_pCodec;

    AVCodecContext *m_pCodecContext;

    AVFrame *m_pPicture;

    AVPacket m_avPacket;

...

};

static HANDLE hMutex;

static int g_numCPU = 1;

< 디코더 초기화 - initialize decoder >

void VideoDecoder::initCodec()

{

    static int init = 0;    // 코덱 초기화는 프로그램 돌때 한번만 해주면됨

    if (init == 0) {
        SYSTEM_INFO sysinfo;

GetSystemInfo(&sysinfo);

g_numCPU = sysinfo.dwNumberOfProcessors;    // cpu 코어 갯수

#ifdef FFMPEG_0_8

avcodec_init();
#endif

avcodec_register_all();

        hMutex = CreateMutex(NULL, FALSE, NULL);

        init = 1;

    }
}

< 디코더 열기 - open decoder >

int VideoDecoder::openDecoder(int codec)

{

WaitForSingleObject(hMutex, INFINITE);    // 뮤텍스 처리 (thread-safe 하지않음)

av_init_packet(&m_avPacket);

if (codec == VIDEO_CODEC_H264)
m_pCodec = avcodec_find_decoder(CODEC_ID_H264); else if (codec == VIDEO_CODEC_MPEG4)
m_pCodec = avcodec_find_decoder(CODEC_ID_MPEG4);
else {
ReleaseMutex(hMutex);
return -1;
}

#ifdef FFMPEG_0_8

m_pCodecContext = avcodec_alloc_context();

#else
    m_pCodecContext = avcodec_alloc_context3(m_pCodec);
#endif

//m_pCodecContext->thread_type = FF_THREAD_FRAME; m_pCodecContext->thread_type = FF_THREAD_SLICE;
m_pCodecContext->thread_count = g_numCPU;

m_pPicture = avcodec_alloc_frame();

if (codec == VIDEO_CODEC_H264) // MPEG4('mp4v') 일때 영상깨짐 {
if (m_pCodec->capabilities&CODEC_CAP_TRUNCATED)
m_pCodecContext->flags |= CODEC_FLAG_TRUNCATED; }

#ifdef FFMPEG_0_8

if (avcodec_open(m_pCodecContext, m_pCodec) < 0) {
#else
    if (avcodec_open2(m_pCodecContext, m_pCodec, NULL) < 0)
#endif
  ReleaseMutex(hMutex);

  return -1;

 }

ReleaseMutex(hMutex);
    return 0;

}

< 디코더 닫기 - close decoder >

int VideoDecoder::closeDecoder()

{

WaitForSingleObject(hMutex, INFINITE);

if (m_pCodecContext)

  avcodec_close(m_pCodecContext);

av_free(m_pCodecContext);

av_free(m_pPicture);

ReleaseMutex(hMutex);

return 0;

}

< 디코더 사용 - decode >

int VideoDecoder::Decode(char *inBuf, int inLen)

{

int retLen, got_picture;


    char *buf = (char *)malloc(inLen+FF_INPUT_BUFFER_PADDING_SIZE);
    memset(&buf[inLen], 0, FF_INPUT_BUFFER_PADDING_SIZE);
    memcpy(buf, inBuf, inLen);
    m_avPacket.data = (uint8_t *)buf;
    m_avPacket.size = inLen;

retLen = avcodec_decode_video2(m_pCodecContext, m_pPicture, &got_picture, &m_avPacket);    // 디코딩 => m_pPicture->data[0], data[1], data[2] 이 각각 YCbCr 값을 가짐 (YUV420)

if (retLen < 0) {

  goto exit;

 }

if (!got_picture) {    // 한 프레임을 만들지 못함 -> 한 프레임을 잘라서 주지 못할경우 여기에 대한 처리가 필요

  goto exit;

}

...
exit:

    if (buf)
        free(buf);

    return retLen;

}

사용법은 클래스 생성후 initCodec()(프로그램 실행처음 한번만 하면됨) -> openDecoder() -> Decode() -> closeDecoder() 해주면 된다.

ffmpeg 디코더 라이브러리 => ffmpeg 디코더 참고소스