examples/decode_video: switch to the new decoding API
[ffmpeg.git] / doc / examples / decode_video.c
1 /*
2  * copyright (c) 2001 Fabrice Bellard
3  *
4  * This file is part of Libav.
5  *
6  * Libav is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * Libav is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with Libav; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 /**
22  * @file
23  * video decoding with libavcodec API example
24  *
25  * @example decode_video.c
26  */
27
28 #include <stdio.h>
29 #include <stdlib.h>
30 #include <string.h>
31
32 #include "libavcodec/avcodec.h"
33
34 #include "libavutil/common.h"
35 #include "libavutil/imgutils.h"
36 #include "libavutil/mathematics.h"
37
38 #define INBUF_SIZE 4096
39
40 static void pgm_save(unsigned char *buf, int wrap, int xsize, int ysize,
41                      char *filename)
42 {
43     FILE *f;
44     int i;
45
46     f=fopen(filename,"w");
47     fprintf(f,"P5\n%d %d\n%d\n",xsize,ysize,255);
48     for(i=0;i<ysize;i++)
49         fwrite(buf + i * wrap,1,xsize,f);
50     fclose(f);
51 }
52
53 static void decode(AVCodecContext *dec_ctx, AVFrame *frame, AVPacket *pkt,
54                    const char *filename)
55 {
56     char buf[1024];
57     int ret;
58
59     ret = avcodec_send_packet(dec_ctx, pkt);
60     if (ret < 0) {
61         fprintf(stderr, "Error sending a packet for decoding\n");
62         exit(1);
63     }
64
65     while (ret >= 0) {
66         ret = avcodec_receive_frame(dec_ctx, frame);
67         if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
68             return;
69         else if (ret < 0) {
70             fprintf(stderr, "Error during decoding\n");
71             exit(1);
72         }
73
74         printf("saving frame %3d\n", dec_ctx->frame_number);
75         fflush(stdout);
76
77         /* the picture is allocated by the decoder. no need to
78            free it */
79         snprintf(buf, sizeof(buf), filename, dec_ctx->frame_number);
80         pgm_save(frame->data[0], frame->linesize[0],
81                  frame->width, frame->height, buf);
82     }
83 }
84
85 int main(int argc, char **argv)
86 {
87     const char *filename, *outfilename;
88     const AVCodec *codec;
89     AVCodecParserContext *parser;
90     AVCodecContext *c= NULL;
91     FILE *f;
92     AVFrame *picture;
93     uint8_t inbuf[INBUF_SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
94     uint8_t *data;
95     size_t   data_size;
96     int ret;
97     AVPacket avpkt;
98
99     if (argc <= 2) {
100         fprintf(stderr, "Usage: %s <input file> <output file>\n", argv[0]);
101         exit(0);
102     }
103     filename    = argv[1];
104     outfilename = argv[2];
105
106     avcodec_register_all();
107
108     av_init_packet(&avpkt);
109
110     /* set end of buffer to 0 (this ensures that no overreading happens for damaged MPEG streams) */
111     memset(inbuf + INBUF_SIZE, 0, AV_INPUT_BUFFER_PADDING_SIZE);
112
113     /* find the MPEG-1 video decoder */
114     codec = avcodec_find_decoder(AV_CODEC_ID_MPEG1VIDEO);
115     if (!codec) {
116         fprintf(stderr, "codec not found\n");
117         exit(1);
118     }
119
120     parser = av_parser_init(codec->id);
121     if (!parser) {
122         fprintf(stderr, "parser not found\n");
123         exit(1);
124     }
125
126     c = avcodec_alloc_context3(codec);
127     picture = av_frame_alloc();
128
129     /* For some codecs, such as msmpeg4 and mpeg4, width and height
130        MUST be initialized there because this information is not
131        available in the bitstream. */
132
133     /* open it */
134     if (avcodec_open2(c, codec, NULL) < 0) {
135         fprintf(stderr, "could not open codec\n");
136         exit(1);
137     }
138
139     f = fopen(filename, "rb");
140     if (!f) {
141         fprintf(stderr, "could not open %s\n", filename);
142         exit(1);
143     }
144
145     while (!feof(f)) {
146         /* read raw data from the input file */
147         data_size = fread(inbuf, 1, INBUF_SIZE, f);
148         if (!data_size)
149             break;
150
151         /* use the parser to split the data into frames */
152         data = inbuf;
153         while (data_size > 0) {
154             ret = av_parser_parse2(parser, c, &avpkt.data, &avpkt.size,
155                                    data, data_size, AV_NOPTS_VALUE, AV_NOPTS_VALUE, 0);
156             if (ret < 0) {
157                 fprintf(stderr, "Error while parsing\n");
158                 exit(1);
159             }
160             data      += ret;
161             data_size -= ret;
162
163             if (avpkt.size)
164                 decode(c, picture, &avpkt, outfilename);
165         }
166     }
167
168     /* flush the decoder */
169     decode(c, picture, NULL, outfilename);
170
171     fclose(f);
172
173     av_parser_close(parser);
174     avcodec_free_context(&c);
175     av_frame_free(&picture);
176
177     return 0;
178 }