generic-library/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

api-example: use new video encoding API.

Browse Source
This commit is contained in:
Anton Khirnov 2012-07-31 16:35:18 +02:00
parent 29cfdd3767
commit 661454aa28
1 changed files with 44 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
libavcodec/api-example.c
View File

@ -38,6 +38,7 @@
#include "libavcodec/avcodec.h" #include "libavcodec/avcodec.h"
#include "libavutil/audioconvert.h" #include "libavutil/audioconvert.h"
#include "libavutil/imgutils.h"
#include "libavutil/mathematics.h" #include "libavutil/mathematics.h"
#include "libavutil/samplefmt.h" #include "libavutil/samplefmt.h"
@ -316,10 +317,11 @@ static void video_encode_example(const char *filename)
{ {
AVCodec *codec; AVCodec *codec;
AVCodecContext *c= NULL; AVCodecContext *c= NULL;
int i, out_size, size, x, y, outbuf_size; int i, ret, x, y, got_output;
FILE *f; FILE *f;
AVFrame *picture; AVFrame *picture;
uint8_t *outbuf, *picture_buf; AVPacket pkt;
uint8_t endcode[] = { 0, 0, 1, 0xb7 };
printf("Video encoding\n"); printf("Video encoding\n");
@ -356,21 +358,22 @@ static void video_encode_example(const char *filename)
exit(1); exit(1);
} }
/* alloc image and output buffer */ ret = av_image_alloc(picture->data, picture->linesize, c->width, c->height,
outbuf_size = 100000; c->pix_fmt, 32);
outbuf = malloc(outbuf_size); if (ret < 0) {
size = c->width * c->height; fprintf(stderr, "could not alloc raw picture buffer\n");
picture_buf = malloc((size * 3) / 2); /* size for YUV 420 */ exit(1);
}
picture->data[0] = picture_buf; picture->format = c->pix_fmt;
picture->data[1] = picture->data[0] + size; picture->width = c->width;
picture->data[2] = picture->data[1] + size / 4; picture->height = c->height;
picture->linesize[0] = c->width;
picture->linesize[1] = c->width / 2;
picture->linesize[2] = c->width / 2;
/* encode 1 second of video */ /* encode 1 second of video */
for(i=0;i<25;i++) { for(i=0;i<25;i++) {
av_init_packet(&pkt);
pkt.data = NULL; // packet data will be allocated by the encoder
pkt.size = 0;
fflush(stdout); fflush(stdout);
/* prepare a dummy image */ /* prepare a dummy image */
/* Y */ /* Y */
@ -388,33 +391,46 @@ static void video_encode_example(const char *filename)
} }
} }
picture->pts = i;
/* encode the image */ /* encode the image */
out_size = avcodec_encode_video(c, outbuf, outbuf_size, picture); ret = avcodec_encode_video2(c, &pkt, picture, &got_output);
printf("encoding frame %3d (size=%5d)\n", i, out_size); if (ret < 0) {
fwrite(outbuf, 1, out_size, f); fprintf(stderr, "error encoding frame\n");
exit(1);
}
if (got_output) {
printf("encoding frame %3d (size=%5d)\n", i, pkt.size);
fwrite(pkt.data, 1, pkt.size, f);
av_free_packet(&pkt);
}
} }
/* get the delayed frames */ /* get the delayed frames */
for(; out_size; i++) { for (got_output = 1; got_output; i++) {
fflush(stdout); fflush(stdout);
out_size = avcodec_encode_video(c, outbuf, outbuf_size, NULL); ret = avcodec_encode_video2(c, &pkt, NULL, &got_output);
printf("write frame %3d (size=%5d)\n", i, out_size); if (ret < 0) {
fwrite(outbuf, 1, out_size, f); fprintf(stderr, "error encoding frame\n");
exit(1);
}
if (got_output) {
printf("encoding frame %3d (size=%5d)\n", i, pkt.size);
fwrite(pkt.data, 1, pkt.size, f);
av_free_packet(&pkt);
}
} }
/* add sequence end code to have a real mpeg file */ /* add sequence end code to have a real mpeg file */
outbuf[0] = 0x00; fwrite(endcode, 1, sizeof(endcode), f);
outbuf[1] = 0x00;
outbuf[2] = 0x01;
outbuf[3] = 0xb7;
fwrite(outbuf, 1, 4, f);
fclose(f); fclose(f);
free(picture_buf);
free(outbuf);
avcodec_close(c); avcodec_close(c);
av_free(c); av_free(c);
av_freep(&picture->data[0]);
av_free(picture); av_free(picture);
printf("\n"); printf("\n");
} }