examples: add scaling_video example
[ffmpeg.git] / doc / examples / scaling_video.c
1 /*
2  * Copyright (c) 2012 Stefano Sabatini
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 #include <libavutil/imgutils.h>
24 #include <libavutil/parseutils.h>
25 #include <libswscale/swscale.h>
26
27 static void fill_yuv_image(uint8_t *data[4], int linesize[4],
28                            int width, int height, int frame_index)
29 {
30     int x, y, i;
31
32     i = frame_index;
33
34     /* Y */
35     for (y = 0; y < height; y++)
36         for (x = 0; x < width; x++)
37             data[0][y * linesize[0] + x] = x + y + i * 3;
38
39     /* Cb and Cr */
40     for (y = 0; y < height / 2; y++) {
41         for (x = 0; x < width / 2; x++) {
42             data[1][y * linesize[1] + x] = 128 + y + i * 2;
43             data[2][y * linesize[2] + x] = 64 + x + i * 5;
44         }
45     }
46 }
47
48 static void save_pgm(const char *filename,
49                      const uint8_t *data, int linesize, int width, int height)
50 {
51     FILE *f;
52     int i;
53
54     f = fopen(filename, "w");
55     fprintf(f, "P5\n%d %d\n%d\n", width, height, 255);
56     for (i = 0; i < height; i++)
57         fwrite(data + i * linesize, 1, width, f);
58     fclose(f);
59 }
60
61 int main(int argc, char **argv)
62 {
63     uint8_t *src_data[4], *dst_data[4];
64     int src_linesize[4], dst_linesize[4];
65     int src_w = 320, src_h = 240, dst_w, dst_h;
66     enum PixelFormat src_pix_fmt = PIX_FMT_YUV420P, dst_pix_fmt = PIX_FMT_GRAY8;
67     const char *dst_size = argv[1];
68     struct SwsContext *sws_ctx;
69     int i, ret;
70
71     if (argc != 2) {
72         fprintf(stderr, "Usage: %s output_size\n"
73                 "API example program to show how to scale an image with libswscale.\n"
74                 "This program generates a series of pictures, rescales them to the given "
75                 "<output size> and finally saves the rescaled pictures as PGM files named "
76                 "like outscale<frame number>.pgm.\n"
77                 "\n", argv[0]);
78         exit(1);
79     }
80
81     if (av_parse_video_size(&dst_w, &dst_h, dst_size) < 0) {
82         av_log(NULL, AV_LOG_ERROR,
83                "Invalid size '%s', must be in the form WxH or a valid size abbreviation\n",
84                dst_size);
85         exit(1);
86     }
87
88     /* create scaling context */
89     sws_ctx = sws_getContext(src_w, src_h, src_pix_fmt,
90                              dst_w, dst_h, dst_pix_fmt,
91                              SWS_BILINEAR, NULL, NULL, NULL);
92     if (!sws_ctx) {
93         av_log(NULL, AV_LOG_ERROR,
94                "Impossible to create scale context for the conversion "
95                "fmt:%s s:%dx%d -> fmt:%s s:%dx%d\n",
96                av_get_pix_fmt_name(src_pix_fmt), src_w, src_h,
97                av_get_pix_fmt_name(dst_pix_fmt), dst_w, dst_h);
98         ret = AVERROR(EINVAL);
99         goto end;
100     }
101
102     /* allocate source and destination image buffers */
103     if ((ret = av_image_alloc(src_data, src_linesize,
104                               src_w, src_h, src_pix_fmt, 16)) < 0)
105         goto end;
106
107     if ((ret = av_image_alloc(dst_data, dst_linesize,
108                               dst_w, dst_h, dst_pix_fmt, 16)) < 0)
109         goto end;
110
111     for (i = 0; i < 100; i++) {
112         char filename[1024];
113
114         /* generate synthetic video */
115         fill_yuv_image(src_data, src_linesize, src_w, src_h, i);
116
117         /* convert to destination format */
118         sws_scale(sws_ctx, (const uint8_t * const*)src_data,
119                   src_linesize, 0, src_h, dst_data, dst_linesize);
120
121         /* write Y plane to to output PGM file */
122         snprintf(filename, sizeof(filename), "outscale%02d.pgm", i);
123         save_pgm(filename, dst_data[0], dst_linesize[0], dst_w, dst_h);
124     }
125
126 end:
127     av_freep(&src_data[0]);
128     av_freep(&dst_data[0]);
129     sws_freeContext(sws_ctx);
130     return ret < 0;
131 }