Merge commit 'fee0f1de2c6a9924acb74013436dbea8f2bd1ecb'
[ffmpeg.git] / doc / examples / encode_video.c
1 /*
2  * Copyright (c) 2001 Fabrice Bellard
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a copy
5  * of this software and associated documentation files (the "Software"), to deal
6  * in the Software without restriction, including without limitation the rights
7  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8  * copies of the Software, and to permit persons to whom the Software is
9  * furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
20  * THE SOFTWARE.
21  */
22
23 /**
24  * @file
25  * video encoding with libavcodec API example
26  *
27  * @example encode_video.c
28  */
29
30 #include <stdio.h>
31 #include <stdlib.h>
32 #include <string.h>
33
34 #include <libavcodec/avcodec.h>
35
36 #include <libavutil/opt.h>
37 #include <libavutil/imgutils.h>
38
39 int main(int argc, char **argv)
40 {
41     const char *filename, *codec_name;
42     const AVCodec *codec;
43     AVCodecContext *c= NULL;
44     int i, ret, x, y, got_output;
45     FILE *f;
46     AVFrame *frame;
47     AVPacket pkt;
48     uint8_t endcode[] = { 0, 0, 1, 0xb7 };
49
50     if (argc <= 2) {
51         fprintf(stderr, "Usage: %s <output file> <codec name>\n", argv[0]);
52         exit(0);
53     }
54     filename = argv[1];
55     codec_name = argv[2];
56
57     avcodec_register_all();
58
59     /* find the mpeg1video encoder */
60     codec = avcodec_find_encoder_by_name(codec_name);
61     if (!codec) {
62         fprintf(stderr, "Codec not found\n");
63         exit(1);
64     }
65
66     c = avcodec_alloc_context3(codec);
67     if (!c) {
68         fprintf(stderr, "Could not allocate video codec context\n");
69         exit(1);
70     }
71
72     /* put sample parameters */
73     c->bit_rate = 400000;
74     /* resolution must be a multiple of two */
75     c->width = 352;
76     c->height = 288;
77     /* frames per second */
78     c->time_base = (AVRational){1, 25};
79     c->framerate = (AVRational){25, 1};
80
81     /* emit one intra frame every ten frames
82      * check frame pict_type before passing frame
83      * to encoder, if frame->pict_type is AV_PICTURE_TYPE_I
84      * then gop_size is ignored and the output of encoder
85      * will always be I frame irrespective to gop_size
86      */
87     c->gop_size = 10;
88     c->max_b_frames = 1;
89     c->pix_fmt = AV_PIX_FMT_YUV420P;
90
91     if (codec->id == AV_CODEC_ID_H264)
92         av_opt_set(c->priv_data, "preset", "slow", 0);
93
94     /* open it */
95     if (avcodec_open2(c, codec, NULL) < 0) {
96         fprintf(stderr, "Could not open codec\n");
97         exit(1);
98     }
99
100     f = fopen(filename, "wb");
101     if (!f) {
102         fprintf(stderr, "Could not open %s\n", filename);
103         exit(1);
104     }
105
106     frame = av_frame_alloc();
107     if (!frame) {
108         fprintf(stderr, "Could not allocate video frame\n");
109         exit(1);
110     }
111     frame->format = c->pix_fmt;
112     frame->width  = c->width;
113     frame->height = c->height;
114
115     ret = av_frame_get_buffer(frame, 32);
116     if (ret < 0) {
117         fprintf(stderr, "Could not allocate the video frame data\n");
118         exit(1);
119     }
120
121     /* encode 1 second of video */
122     for (i = 0; i < 25; i++) {
123         av_init_packet(&pkt);
124         pkt.data = NULL;    // packet data will be allocated by the encoder
125         pkt.size = 0;
126
127         fflush(stdout);
128
129         /* make sure the frame data is writable */
130         ret = av_frame_make_writable(frame);
131         if (ret < 0)
132             exit(1);
133
134         /* prepare a dummy image */
135         /* Y */
136         for (y = 0; y < c->height; y++) {
137             for (x = 0; x < c->width; x++) {
138                 frame->data[0][y * frame->linesize[0] + x] = x + y + i * 3;
139             }
140         }
141
142         /* Cb and Cr */
143         for (y = 0; y < c->height/2; y++) {
144             for (x = 0; x < c->width/2; x++) {
145                 frame->data[1][y * frame->linesize[1] + x] = 128 + y + i * 2;
146                 frame->data[2][y * frame->linesize[2] + x] = 64 + x + i * 5;
147             }
148         }
149
150         frame->pts = i;
151
152         /* encode the image */
153         ret = avcodec_encode_video2(c, &pkt, frame, &got_output);
154         if (ret < 0) {
155             fprintf(stderr, "Error encoding frame\n");
156             exit(1);
157         }
158
159         if (got_output) {
160             printf("Write frame %3d (size=%5d)\n", i, pkt.size);
161             fwrite(pkt.data, 1, pkt.size, f);
162             av_packet_unref(&pkt);
163         }
164     }
165
166     /* get the delayed frames */
167     for (got_output = 1; got_output; i++) {
168         fflush(stdout);
169
170         ret = avcodec_encode_video2(c, &pkt, NULL, &got_output);
171         if (ret < 0) {
172             fprintf(stderr, "Error encoding frame\n");
173             exit(1);
174         }
175
176         if (got_output) {
177             printf("Write frame %3d (size=%5d)\n", i, pkt.size);
178             fwrite(pkt.data, 1, pkt.size, f);
179             av_packet_unref(&pkt);
180         }
181     }
182
183     /* add sequence end code to have a real MPEG file */
184     fwrite(endcode, 1, sizeof(endcode), f);
185     fclose(f);
186
187     avcodec_free_context(&c);
188     av_frame_free(&frame);
189
190     return 0;
191 }