4c1068bef339e21ce4f71bd4e0ec3d378e0cd3e8
[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, got_picture;
58
59     while (pkt->size > 0) {
60         ret = avcodec_decode_video2(dec_ctx, frame, &got_picture, pkt);
61         if (ret < 0) {
62             fprintf(stderr, "Error while decoding frame %d\n", dec_ctx->frame_number);
63             exit(1);
64         }
65         if (got_picture) {
66             printf("saving frame %3d\n", dec_ctx->frame_number);
67             fflush(stdout);
68
69             /* the picture is allocated by the decoder. no need to
70                free it */
71             snprintf(buf, sizeof(buf), filename, dec_ctx->frame_number);
72             pgm_save(frame->data[0], frame->linesize[0],
73                      frame->width, frame->height, buf);
74         }
75         pkt->size -= ret;
76         pkt->data += ret;
77     }
78 }
79
80 int main(int argc, char **argv)
81 {
82     const char *filename, *outfilename;
83     const AVCodec *codec;
84     AVCodecParserContext *parser;
85     AVCodecContext *c= NULL;
86     FILE *f;
87     AVFrame *picture;
88     uint8_t inbuf[INBUF_SIZE + AV_INPUT_BUFFER_PADDING_SIZE];
89     uint8_t *data;
90     size_t   data_size;
91     int ret;
92     AVPacket avpkt;
93
94     if (argc <= 2) {
95         fprintf(stderr, "Usage: %s <input file> <output file>\n", argv[0]);
96         exit(0);
97     }
98     filename    = argv[1];
99     outfilename = argv[2];
100
101     avcodec_register_all();
102
103     av_init_packet(&avpkt);
104
105     /* set end of buffer to 0 (this ensures that no overreading happens for damaged MPEG streams) */
106     memset(inbuf + INBUF_SIZE, 0, AV_INPUT_BUFFER_PADDING_SIZE);
107
108     /* find the MPEG-1 video decoder */
109     codec = avcodec_find_decoder(AV_CODEC_ID_MPEG1VIDEO);
110     if (!codec) {
111         fprintf(stderr, "codec not found\n");
112         exit(1);
113     }
114
115     parser = av_parser_init(codec->id);
116     if (!parser) {
117         fprintf(stderr, "parser not found\n");
118         exit(1);
119     }
120
121     c = avcodec_alloc_context3(codec);
122     picture = av_frame_alloc();
123
124     /* For some codecs, such as msmpeg4 and mpeg4, width and height
125        MUST be initialized there because this information is not
126        available in the bitstream. */
127
128     /* open it */
129     if (avcodec_open2(c, codec, NULL) < 0) {
130         fprintf(stderr, "could not open codec\n");
131         exit(1);
132     }
133
134     f = fopen(filename, "rb");
135     if (!f) {
136         fprintf(stderr, "could not open %s\n", filename);
137         exit(1);
138     }
139
140     while (!feof(f)) {
141         /* read raw data from the input file */
142         data_size = fread(inbuf, 1, INBUF_SIZE, f);
143         if (!data_size)
144             break;
145
146         /* use the parser to split the data into frames */
147         data = inbuf;
148         while (data_size > 0) {
149             ret = av_parser_parse2(parser, c, &avpkt.data, &avpkt.size,
150                                    data, data_size, AV_NOPTS_VALUE, AV_NOPTS_VALUE, 0);
151             if (ret < 0) {
152                 fprintf(stderr, "Error while parsing\n");
153                 exit(1);
154             }
155             data      += ret;
156             data_size -= ret;
157
158             if (avpkt.size)
159                 decode(c, picture, &avpkt, outfilename);
160         }
161     }
162
163     /* flush the decoder */
164     avpkt.data = NULL;
165     avpkt.size = 0;
166     decode(c, picture, &avpkt, outfilename);
167
168     fclose(f);
169
170     av_parser_close(parser);
171     avcodec_free_context(&c);
172     av_frame_free(&picture);
173
174     return 0;
175 }