FFmpeg для Windows – создание своей Dll / FFmpeg / Recog.ru - Распознавание образов для программистов


FFmpeg для Windows – создание своей Dll

Для того, чтобы использовать FFmpeg для Windows необходимо создать Dll, которая будет экспортировать ваши функции. Чтобы не усложнять задачу, я взял пример decoding_encoding.c, поставляемый вместе с библиотекой, выбрал оттуда функцию декодирования видео и создал собственную Dll.
Первое, что нужно сделать, это создать файл, который будет содержать экспортируемую функцию (-и). В нашем примере out_dll.c. И его содержимое представлено ниже.
#define UINT64_C(val) val##ULL
#define FFMPEG_API __declspec(dllexport)

#include "libavcodec/avcodec.h"
#include <string.h>

#define INBUF_SIZE 4096


static void pgm_save(unsigned char *buf, int wrap, int xsize, int ysize,
                     char *filename)
{
    FILE *f;
    int i;

    f=fopen(filename,"w");
    fprintf(f,"P5\n%d %d\n%d\n",xsize,ysize,255);
    for(i=0;i<ysize;i++)
        fwrite(buf + i * wrap,1,xsize,f);
    fclose(f);
}


FFMPEG_API void video_decode_example(const char *outfilename, const char *filename)
{
    avcodec_register_all();AVCodec *codec;
    AVCodecContext *c= NULL;
    int frame, got_picture, len;
    FILE *f;
    AVFrame *picture;
    uint8_t inbuf[INBUF_SIZE + FF_INPUT_BUFFER_PADDING_SIZE];
    char buf[1024];
    AVPacket avpkt;

    av_init_packet(&avpkt);

    /* set end of buffer to 0 (this ensures that no overreading happens for damaged mpeg streams) */
    memset(inbuf + INBUF_SIZE, 0, FF_INPUT_BUFFER_PADDING_SIZE);

    printf("Decode video file %s\n", filename);

    /* find the mpeg1 video decoder */
    codec = avcodec_find_decoder(CODEC_ID_MPEG1VIDEO);
    if (!codec) {
        fprintf(stderr, "codec not found\n");
        exit(1);
    }

    c = avcodec_alloc_context3(codec);
    picture= avcodec_alloc_frame();

    if(codec->capabilities&CODEC_CAP_TRUNCATED)
        c->flags|= CODEC_FLAG_TRUNCATED; /* we do not send complete frames */

    /* For some codecs, such as msmpeg4 and mpeg4, width and height
       MUST be initialized there because this information is not
       available in the bitstream. */

    /* open it */
    if (avcodec_open2(c, codec, NULL) < 0) {
        fprintf(stderr, "could not open codec\n");
        exit(1);
    }

    /* the codec gives us the frame size, in samples */

    f = fopen(filename, "rb");
    if (!f) {
        fprintf(stderr, "could not open %s\n", filename);
        exit(1);
    }

    frame = 0;
    for(;;) {
        avpkt.size = fread(inbuf, 1, INBUF_SIZE, f);
        if (avpkt.size == 0)
            break;

        /* NOTE1: some codecs are stream based (mpegvideo, mpegaudio)
           and this is the only method to use them because you cannot
           know the compressed data size before analysing it.

           BUT some other codecs (msmpeg4, mpeg4) are inherently frame
           based, so you must call them with all the data for one
           frame exactly. You must also initialize 'width' and
           'height' before initializing them. */

        /* NOTE2: some codecs allow the raw parameters (frame size,
           sample rate) to be changed at any frame. We handle this, so
           you should also take care of it */

        /* here, we use a stream based decoder (mpeg1video), so we
           feed decoder and see if it could decode a frame */
        avpkt.data = inbuf;
        while (avpkt.size > 0) {
            len = avcodec_decode_video2(c, picture, &got_picture, &avpkt);
            if (len < 0) {
                fprintf(stderr, "Error while decoding frame %d\n", frame);
                exit(1);
            }
            if (got_picture) {
                printf("saving frame %3d\n", frame);
                fflush(stdout);

                /* the picture is allocated by the decoder. no need to
                   free it */
                snprintf(buf, sizeof(buf), outfilename, frame);
                pgm_save(picture->data[0], picture->linesize[0],
                         c->width, c->height, buf);
                frame++;
            }
            avpkt.size -= len;
            avpkt.data += len;
        }
    }

    /* some codecs, such as MPEG, transmit the I and P frame with a
       latency of one frame. You must do the following to have a
       chance to get the last frame of the video */
    avpkt.data = NULL;
    avpkt.size = 0;
    len = avcodec_decode_video2(c, picture, &got_picture, &avpkt);
    if (got_picture) {
        printf("saving last frame %3d\n", frame);
        fflush(stdout);

        /* the picture is allocated by the decoder. no need to
           free it */
        snprintf(buf, sizeof(buf), outfilename, frame);
        pgm_save(picture->data[0], picture->linesize[0],
                 c->width, c->height, buf);
        frame++;
    }

    fclose(f);

    avcodec_close©;
    av_free©;
    av_free(picture);
    printf("\n");
}


Экспортируется функция video_decode_example, которая открывает файл filename, закодированный кодеком MPEG и декодирует фреймы, сохраняя результат в графический файл с помощью функции pgm_save. Зачем это нужно – умолчим, поскольку нам главное проверить работоспособность библиотеки.
После создания файла в каталоге build (см. предыдущий пост) запускаем в MINGW командной строке:
gcc out_dll.c -Wall -shared -o ffmpeg.dll -O2 -x c++ -I../include -L../lib -lavformat -lavcodec -lavdevice -lswscale -lavutil -lwsock32
Через некоторое время создается библиотека (у меня она занимала более 7MB). Чтобы использовать данную функцию, можно воспользоваться следующим примером:
#include <windows.h>
#include <string.h>

void (__cdecl *video_decode_example)(const char*,const char*);	

int main(int argc, char* argv[])
{
	if (argc < 2) return 1;

	char fileout[256];
	strcpy_s(fileout,256,argv[1]);
	strcat(fileout,"_res.pgm");

	HINSTANCE dllInstance=LoadLibrary("ffmpeg.dll");
	(FARPROC&)video_decode_example=GetProcAddress(dllInstance,"video_decode_example");


	video_decode_example(fileout, argv[1]);

	return 0;
}

Компилируем, смотрим работоспособность – все работает на видео, закодированных MPEG.
  • 0
  • 05 июля 2012, 18:04
  • vidikon

Комментарии (0)

RSS свернуть / развернуть

Только зарегистрированные и авторизованные пользователи могут оставлять комментарии.