ffmpeg_opt: remove casts for OPT_FUNC2
[ffmpeg.git] / cmdutils.c
1 /*
2  * Various utilities for command line tools
3  * Copyright (c) 2000-2003 Fabrice Bellard
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include <string.h>
23 #include <stdlib.h>
24 #include <errno.h>
25 #include <math.h>
26
27 /* Include only the enabled headers since some compilers (namely, Sun
28    Studio) will not omit unused inline functions and create undefined
29    references to libraries that are not being built. */
30
31 #include "config.h"
32 #include "libavformat/avformat.h"
33 #include "libavfilter/avfilter.h"
34 #include "libavdevice/avdevice.h"
35 #include "libavresample/avresample.h"
36 #include "libswscale/swscale.h"
37 #include "libswresample/swresample.h"
38 #if CONFIG_POSTPROC
39 #include "libpostproc/postprocess.h"
40 #endif
41 #include "libavutil/avassert.h"
42 #include "libavutil/avstring.h"
43 #include "libavutil/mathematics.h"
44 #include "libavutil/imgutils.h"
45 #include "libavutil/parseutils.h"
46 #include "libavutil/pixdesc.h"
47 #include "libavutil/eval.h"
48 #include "libavutil/dict.h"
49 #include "libavutil/opt.h"
50 #include "cmdutils.h"
51 #include "version.h"
52 #if CONFIG_NETWORK
53 #include "libavformat/network.h"
54 #endif
55 #if HAVE_SYS_RESOURCE_H
56 #include <sys/resource.h>
57 #endif
58
59 struct SwsContext *sws_opts;
60 SwrContext *swr_opts;
61 AVDictionary *format_opts, *codec_opts;
62
63 const int this_year = 2012;
64
65 static FILE *report_file;
66
67 void init_opts(void)
68 {
69
70     if(CONFIG_SWSCALE)
71         sws_opts = sws_getContext(16, 16, 0, 16, 16, 0, SWS_BICUBIC,
72                               NULL, NULL, NULL);
73
74     if(CONFIG_SWRESAMPLE)
75         swr_opts = swr_alloc();
76 }
77
78 void uninit_opts(void)
79 {
80 #if CONFIG_SWSCALE
81     sws_freeContext(sws_opts);
82     sws_opts = NULL;
83 #endif
84
85     if(CONFIG_SWRESAMPLE)
86         swr_free(&swr_opts);
87
88     av_dict_free(&format_opts);
89     av_dict_free(&codec_opts);
90 }
91
92 void log_callback_help(void *ptr, int level, const char *fmt, va_list vl)
93 {
94     vfprintf(stdout, fmt, vl);
95 }
96
97 static void log_callback_report(void *ptr, int level, const char *fmt, va_list vl)
98 {
99     va_list vl2;
100     char line[1024];
101     static int print_prefix = 1;
102
103     va_copy(vl2, vl);
104     av_log_default_callback(ptr, level, fmt, vl);
105     av_log_format_line(ptr, level, fmt, vl2, line, sizeof(line), &print_prefix);
106     va_end(vl2);
107     fputs(line, report_file);
108     fflush(report_file);
109 }
110
111 double parse_number_or_die(const char *context, const char *numstr, int type,
112                            double min, double max)
113 {
114     char *tail;
115     const char *error;
116     double d = av_strtod(numstr, &tail);
117     if (*tail)
118         error = "Expected number for %s but found: %s\n";
119     else if (d < min || d > max)
120         error = "The value for %s was %s which is not within %f - %f\n";
121     else if (type == OPT_INT64 && (int64_t)d != d)
122         error = "Expected int64 for %s but found %s\n";
123     else if (type == OPT_INT && (int)d != d)
124         error = "Expected int for %s but found %s\n";
125     else
126         return d;
127     av_log(NULL, AV_LOG_FATAL, error, context, numstr, min, max);
128     exit_program(1);
129     return 0;
130 }
131
132 int64_t parse_time_or_die(const char *context, const char *timestr,
133                           int is_duration)
134 {
135     int64_t us;
136     if (av_parse_time(&us, timestr, is_duration) < 0) {
137         av_log(NULL, AV_LOG_FATAL, "Invalid %s specification for %s: %s\n",
138                is_duration ? "duration" : "date", context, timestr);
139         exit_program(1);
140     }
141     return us;
142 }
143
144 void show_help_options(const OptionDef *options, const char *msg, int mask,
145                        int value)
146 {
147     const OptionDef *po;
148     int first;
149
150     first = 1;
151     for (po = options; po->name != NULL; po++) {
152         char buf[64];
153         if ((po->flags & mask) == value) {
154             if (first) {
155                 printf("%s", msg);
156                 first = 0;
157             }
158             av_strlcpy(buf, po->name, sizeof(buf));
159             if (po->flags & HAS_ARG) {
160                 av_strlcat(buf, " ", sizeof(buf));
161                 av_strlcat(buf, po->argname, sizeof(buf));
162             }
163             printf("-%-17s  %s\n", buf, po->help);
164         }
165     }
166 }
167
168 void show_help_children(const AVClass *class, int flags)
169 {
170     const AVClass *child = NULL;
171     if (class->option) {
172         av_opt_show2(&class, NULL, flags, 0);
173         printf("\n");
174     }
175
176     while (child = av_opt_child_class_next(class, child))
177         show_help_children(child, flags);
178 }
179
180 static const OptionDef *find_option(const OptionDef *po, const char *name)
181 {
182     const char *p = strchr(name, ':');
183     int len = p ? p - name : strlen(name);
184
185     while (po->name != NULL) {
186         if (!strncmp(name, po->name, len) && strlen(po->name) == len)
187             break;
188         po++;
189     }
190     return po;
191 }
192
193 #if defined(_WIN32) && !defined(__MINGW32CE__)
194 #include <windows.h>
195 #include <shellapi.h>
196 /* Will be leaked on exit */
197 static char** win32_argv_utf8 = NULL;
198 static int win32_argc = 0;
199
200 /**
201  * Prepare command line arguments for executable.
202  * For Windows - perform wide-char to UTF-8 conversion.
203  * Input arguments should be main() function arguments.
204  * @param argc_ptr Arguments number (including executable)
205  * @param argv_ptr Arguments list.
206  */
207 static void prepare_app_arguments(int *argc_ptr, char ***argv_ptr)
208 {
209     char *argstr_flat;
210     wchar_t **argv_w;
211     int i, buffsize = 0, offset = 0;
212
213     if (win32_argv_utf8) {
214         *argc_ptr = win32_argc;
215         *argv_ptr = win32_argv_utf8;
216         return;
217     }
218
219     win32_argc = 0;
220     argv_w = CommandLineToArgvW(GetCommandLineW(), &win32_argc);
221     if (win32_argc <= 0 || !argv_w)
222         return;
223
224     /* determine the UTF-8 buffer size (including NULL-termination symbols) */
225     for (i = 0; i < win32_argc; i++)
226         buffsize += WideCharToMultiByte(CP_UTF8, 0, argv_w[i], -1,
227                                         NULL, 0, NULL, NULL);
228
229     win32_argv_utf8 = av_mallocz(sizeof(char *) * (win32_argc + 1) + buffsize);
230     argstr_flat     = (char *)win32_argv_utf8 + sizeof(char *) * (win32_argc + 1);
231     if (win32_argv_utf8 == NULL) {
232         LocalFree(argv_w);
233         return;
234     }
235
236     for (i = 0; i < win32_argc; i++) {
237         win32_argv_utf8[i] = &argstr_flat[offset];
238         offset += WideCharToMultiByte(CP_UTF8, 0, argv_w[i], -1,
239                                       &argstr_flat[offset],
240                                       buffsize - offset, NULL, NULL);
241     }
242     win32_argv_utf8[i] = NULL;
243     LocalFree(argv_w);
244
245     *argc_ptr = win32_argc;
246     *argv_ptr = win32_argv_utf8;
247 }
248 #else
249 static inline void prepare_app_arguments(int *argc_ptr, char ***argv_ptr)
250 {
251     /* nothing to do */
252 }
253 #endif /* WIN32 && !__MINGW32CE__ */
254
255 int parse_option(void *optctx, const char *opt, const char *arg,
256                  const OptionDef *options)
257 {
258     const OptionDef *po;
259     int bool_val = 1;
260     int *dstcount;
261     void *dst;
262
263     po = find_option(options, opt);
264     if (!po->name && opt[0] == 'n' && opt[1] == 'o') {
265         /* handle 'no' bool option */
266         po = find_option(options, opt + 2);
267         if ((po->name && (po->flags & OPT_BOOL)))
268             bool_val = 0;
269     }
270     if (!po->name)
271         po = find_option(options, "default");
272     if (!po->name) {
273         av_log(NULL, AV_LOG_ERROR, "Unrecognized option '%s'\n", opt);
274         return AVERROR(EINVAL);
275     }
276     if (po->flags & HAS_ARG && !arg) {
277         av_log(NULL, AV_LOG_ERROR, "Missing argument for option '%s'\n", opt);
278         return AVERROR(EINVAL);
279     }
280
281     /* new-style options contain an offset into optctx, old-style address of
282      * a global var*/
283     dst = po->flags & (OPT_OFFSET | OPT_SPEC) ? (uint8_t *)optctx + po->u.off
284                                               : po->u.dst_ptr;
285
286     if (po->flags & OPT_SPEC) {
287         SpecifierOpt **so = dst;
288         char *p = strchr(opt, ':');
289
290         dstcount = (int *)(so + 1);
291         *so = grow_array(*so, sizeof(**so), dstcount, *dstcount + 1);
292         (*so)[*dstcount - 1].specifier = av_strdup(p ? p + 1 : "");
293         dst = &(*so)[*dstcount - 1].u;
294     }
295
296     if (po->flags & OPT_STRING) {
297         char *str;
298         str = av_strdup(arg);
299         *(char **)dst = str;
300     } else if (po->flags & OPT_BOOL) {
301         *(int *)dst = bool_val;
302     } else if (po->flags & OPT_INT) {
303         *(int *)dst = parse_number_or_die(opt, arg, OPT_INT64, INT_MIN, INT_MAX);
304     } else if (po->flags & OPT_INT64) {
305         *(int64_t *)dst = parse_number_or_die(opt, arg, OPT_INT64, INT64_MIN, INT64_MAX);
306     } else if (po->flags & OPT_TIME) {
307         *(int64_t *)dst = parse_time_or_die(opt, arg, 1);
308     } else if (po->flags & OPT_FLOAT) {
309         *(float *)dst = parse_number_or_die(opt, arg, OPT_FLOAT, -INFINITY, INFINITY);
310     } else if (po->flags & OPT_DOUBLE) {
311         *(double *)dst = parse_number_or_die(opt, arg, OPT_DOUBLE, -INFINITY, INFINITY);
312     } else if (po->u.func_arg) {
313         int ret = po->flags & OPT_FUNC2 ? po->u.func2_arg(optctx, opt, arg)
314                                         : po->u.func_arg(opt, arg);
315         if (ret < 0) {
316             av_log(NULL, AV_LOG_ERROR,
317                    "Failed to set value '%s' for option '%s'\n", arg, opt);
318             return ret;
319         }
320     }
321     if (po->flags & OPT_EXIT)
322         exit_program(0);
323     return !!(po->flags & HAS_ARG);
324 }
325
326 void parse_options(void *optctx, int argc, char **argv, const OptionDef *options,
327                    void (*parse_arg_function)(void *, const char*))
328 {
329     const char *opt;
330     int optindex, handleoptions = 1, ret;
331
332     /* perform system-dependent conversions for arguments list */
333     prepare_app_arguments(&argc, &argv);
334
335     /* parse options */
336     optindex = 1;
337     while (optindex < argc) {
338         opt = argv[optindex++];
339
340         if (handleoptions && opt[0] == '-' && opt[1] != '\0') {
341             if (opt[1] == '-' && opt[2] == '\0') {
342                 handleoptions = 0;
343                 continue;
344             }
345             opt++;
346
347             if ((ret = parse_option(optctx, opt, argv[optindex], options)) < 0)
348                 exit_program(1);
349             optindex += ret;
350         } else {
351             if (parse_arg_function)
352                 parse_arg_function(optctx, opt);
353         }
354     }
355 }
356
357 int locate_option(int argc, char **argv, const OptionDef *options,
358                   const char *optname)
359 {
360     const OptionDef *po;
361     int i;
362
363     for (i = 1; i < argc; i++) {
364         const char *cur_opt = argv[i];
365
366         if (*cur_opt++ != '-')
367             continue;
368
369         po = find_option(options, cur_opt);
370         if (!po->name && cur_opt[0] == 'n' && cur_opt[1] == 'o')
371             po = find_option(options, cur_opt + 2);
372
373         if ((!po->name && !strcmp(cur_opt, optname)) ||
374              (po->name && !strcmp(optname, po->name)))
375             return i;
376
377         if (!po || po->flags & HAS_ARG)
378             i++;
379     }
380     return 0;
381 }
382
383 static void dump_argument(const char *a)
384 {
385     const unsigned char *p;
386
387     for (p = a; *p; p++)
388         if (!((*p >= '+' && *p <= ':') || (*p >= '@' && *p <= 'Z') ||
389               *p == '_' || (*p >= 'a' && *p <= 'z')))
390             break;
391     if (!*p) {
392         fputs(a, report_file);
393         return;
394     }
395     fputc('"', report_file);
396     for (p = a; *p; p++) {
397         if (*p == '\\' || *p == '"' || *p == '$' || *p == '`')
398             fprintf(report_file, "\\%c", *p);
399         else if (*p < ' ' || *p > '~')
400             fprintf(report_file, "\\x%02x", *p);
401         else
402             fputc(*p, report_file);
403     }
404     fputc('"', report_file);
405 }
406
407 void parse_loglevel(int argc, char **argv, const OptionDef *options)
408 {
409     int idx = locate_option(argc, argv, options, "loglevel");
410     if (!idx)
411         idx = locate_option(argc, argv, options, "v");
412     if (idx && argv[idx + 1])
413         opt_loglevel("loglevel", argv[idx + 1]);
414     idx = locate_option(argc, argv, options, "report");
415     if (idx || getenv("FFREPORT")) {
416         opt_report("report");
417         if (report_file) {
418             int i;
419             fprintf(report_file, "Command line:\n");
420             for (i = 0; i < argc; i++) {
421                 dump_argument(argv[i]);
422                 fputc(i < argc - 1 ? ' ' : '\n', report_file);
423             }
424             fflush(report_file);
425         }
426     }
427 }
428
429 #define FLAGS (o->type == AV_OPT_TYPE_FLAGS) ? AV_DICT_APPEND : 0
430 int opt_default(const char *opt, const char *arg)
431 {
432     const AVOption *o;
433     char opt_stripped[128];
434     const char *p;
435     const AVClass *cc = avcodec_get_class(), *fc = avformat_get_class(), *sc, *swr_class;
436
437     if (!(p = strchr(opt, ':')))
438         p = opt + strlen(opt);
439     av_strlcpy(opt_stripped, opt, FFMIN(sizeof(opt_stripped), p - opt + 1));
440
441     if ((o = av_opt_find(&cc, opt_stripped, NULL, 0,
442                          AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)) ||
443         ((opt[0] == 'v' || opt[0] == 'a' || opt[0] == 's') &&
444          (o = av_opt_find(&cc, opt + 1, NULL, 0, AV_OPT_SEARCH_FAKE_OBJ))))
445         av_dict_set(&codec_opts, opt, arg, FLAGS);
446     else if ((o = av_opt_find(&fc, opt, NULL, 0,
447                               AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ)))
448         av_dict_set(&format_opts, opt, arg, FLAGS);
449 #if CONFIG_SWSCALE
450     sc = sws_get_class();
451     if (!o && (o = av_opt_find(&sc, opt, NULL, 0,
452                          AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {
453         // XXX we only support sws_flags, not arbitrary sws options
454         int ret = av_opt_set(sws_opts, opt, arg, 0);
455         if (ret < 0) {
456             av_log(NULL, AV_LOG_ERROR, "Error setting option %s.\n", opt);
457             return ret;
458         }
459     }
460 #endif
461 #if CONFIG_SWRESAMPLE
462     swr_class = swr_get_class();
463     if (!o && (o = av_opt_find(&swr_class, opt, NULL, 0,
464                                AV_OPT_SEARCH_CHILDREN | AV_OPT_SEARCH_FAKE_OBJ))) {
465         int ret = av_opt_set(swr_opts, opt, arg, 0);
466         if (ret < 0) {
467             av_log(NULL, AV_LOG_ERROR, "Error setting option %s.\n", opt);
468             return ret;
469         }
470     }
471 #endif
472
473     if (o)
474         return 0;
475     av_log(NULL, AV_LOG_ERROR, "Unrecognized option '%s'\n", opt);
476     return AVERROR_OPTION_NOT_FOUND;
477 }
478
479 int opt_loglevel(const char *opt, const char *arg)
480 {
481     const struct { const char *name; int level; } log_levels[] = {
482         { "quiet"  , AV_LOG_QUIET   },
483         { "panic"  , AV_LOG_PANIC   },
484         { "fatal"  , AV_LOG_FATAL   },
485         { "error"  , AV_LOG_ERROR   },
486         { "warning", AV_LOG_WARNING },
487         { "info"   , AV_LOG_INFO    },
488         { "verbose", AV_LOG_VERBOSE },
489         { "debug"  , AV_LOG_DEBUG   },
490     };
491     char *tail;
492     int level;
493     int i;
494
495     for (i = 0; i < FF_ARRAY_ELEMS(log_levels); i++) {
496         if (!strcmp(log_levels[i].name, arg)) {
497             av_log_set_level(log_levels[i].level);
498             return 0;
499         }
500     }
501
502     level = strtol(arg, &tail, 10);
503     if (*tail) {
504         av_log(NULL, AV_LOG_FATAL, "Invalid loglevel \"%s\". "
505                "Possible levels are numbers or:\n", arg);
506         for (i = 0; i < FF_ARRAY_ELEMS(log_levels); i++)
507             av_log(NULL, AV_LOG_FATAL, "\"%s\"\n", log_levels[i].name);
508         exit_program(1);
509     }
510     av_log_set_level(level);
511     return 0;
512 }
513
514 int opt_report(const char *opt)
515 {
516     char filename[64];
517     time_t now;
518     struct tm *tm;
519
520     if (report_file) /* already opened */
521         return 0;
522     time(&now);
523     tm = localtime(&now);
524     snprintf(filename, sizeof(filename), "%s-%04d%02d%02d-%02d%02d%02d.log",
525              program_name,
526              tm->tm_year + 1900, tm->tm_mon + 1, tm->tm_mday,
527              tm->tm_hour, tm->tm_min, tm->tm_sec);
528     report_file = fopen(filename, "w");
529     if (!report_file) {
530         av_log(NULL, AV_LOG_ERROR, "Failed to open report \"%s\": %s\n",
531                filename, strerror(errno));
532         return AVERROR(errno);
533     }
534     av_log_set_callback(log_callback_report);
535     av_log(NULL, AV_LOG_INFO,
536            "%s started on %04d-%02d-%02d at %02d:%02d:%02d\n"
537            "Report written to \"%s\"\n",
538            program_name,
539            tm->tm_year + 1900, tm->tm_mon + 1, tm->tm_mday,
540            tm->tm_hour, tm->tm_min, tm->tm_sec,
541            filename);
542     av_log_set_level(FFMAX(av_log_get_level(), AV_LOG_VERBOSE));
543     return 0;
544 }
545
546 int opt_max_alloc(const char *opt, const char *arg)
547 {
548     char *tail;
549     size_t max;
550
551     max = strtol(arg, &tail, 10);
552     if (*tail) {
553         av_log(NULL, AV_LOG_FATAL, "Invalid max_alloc \"%s\".\n", arg);
554         exit_program(1);
555     }
556     av_max_alloc(max);
557     return 0;
558 }
559
560 int opt_cpuflags(const char *opt, const char *arg)
561 {
562     int ret;
563     unsigned flags = av_get_cpu_flags();
564
565     if ((ret = av_parse_cpu_caps(&flags, arg)) < 0)
566         return ret;
567
568     av_force_cpu_flags(flags);
569     return 0;
570 }
571
572 int opt_codec_debug(const char *opt, const char *arg)
573 {
574     av_log_set_level(AV_LOG_DEBUG);
575     return opt_default(opt, arg);
576 }
577
578 int opt_timelimit(const char *opt, const char *arg)
579 {
580 #if HAVE_SETRLIMIT
581     int lim = parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX);
582     struct rlimit rl = { lim, lim + 1 };
583     if (setrlimit(RLIMIT_CPU, &rl))
584         perror("setrlimit");
585 #else
586     av_log(NULL, AV_LOG_WARNING, "-%s not implemented on this OS\n", opt);
587 #endif
588     return 0;
589 }
590
591 void print_error(const char *filename, int err)
592 {
593     char errbuf[128];
594     const char *errbuf_ptr = errbuf;
595
596     if (av_strerror(err, errbuf, sizeof(errbuf)) < 0)
597         errbuf_ptr = strerror(AVUNERROR(err));
598     av_log(NULL, AV_LOG_ERROR, "%s: %s\n", filename, errbuf_ptr);
599 }
600
601 static int warned_cfg = 0;
602
603 #define INDENT        1
604 #define SHOW_VERSION  2
605 #define SHOW_CONFIG   4
606 #define SHOW_COPYRIGHT 8
607
608 #define PRINT_LIB_INFO(libname, LIBNAME, flags, level)                  \
609     if (CONFIG_##LIBNAME) {                                             \
610         const char *indent = flags & INDENT? "  " : "";                 \
611         if (flags & SHOW_VERSION) {                                     \
612             unsigned int version = libname##_version();                 \
613             av_log(NULL, level,                                         \
614                    "%slib%-11s %2d.%3d.%3d / %2d.%3d.%3d\n",            \
615                    indent, #libname,                                    \
616                    LIB##LIBNAME##_VERSION_MAJOR,                        \
617                    LIB##LIBNAME##_VERSION_MINOR,                        \
618                    LIB##LIBNAME##_VERSION_MICRO,                        \
619                    version >> 16, version >> 8 & 0xff, version & 0xff); \
620         }                                                               \
621         if (flags & SHOW_CONFIG) {                                      \
622             const char *cfg = libname##_configuration();                \
623             if (strcmp(FFMPEG_CONFIGURATION, cfg)) {                    \
624                 if (!warned_cfg) {                                      \
625                     av_log(NULL, level,                                 \
626                             "%sWARNING: library configuration mismatch\n", \
627                             indent);                                    \
628                     warned_cfg = 1;                                     \
629                 }                                                       \
630                 av_log(NULL, level, "%s%-11s configuration: %s\n",      \
631                         indent, #libname, cfg);                         \
632             }                                                           \
633         }                                                               \
634     }                                                                   \
635
636 static void print_all_libs_info(int flags, int level)
637 {
638     PRINT_LIB_INFO(avutil,   AVUTIL,   flags, level);
639     PRINT_LIB_INFO(avcodec,  AVCODEC,  flags, level);
640     PRINT_LIB_INFO(avformat, AVFORMAT, flags, level);
641     PRINT_LIB_INFO(avdevice, AVDEVICE, flags, level);
642     PRINT_LIB_INFO(avfilter, AVFILTER, flags, level);
643 //    PRINT_LIB_INFO(avresample, AVRESAMPLE, flags, level);
644     PRINT_LIB_INFO(swscale,  SWSCALE,  flags, level);
645     PRINT_LIB_INFO(swresample,SWRESAMPLE,  flags, level);
646 #if CONFIG_POSTPROC
647     PRINT_LIB_INFO(postproc, POSTPROC, flags, level);
648 #endif
649 }
650
651 static void print_program_info(int flags, int level)
652 {
653     const char *indent = flags & INDENT? "  " : "";
654
655     av_log(NULL, level, "%s version " FFMPEG_VERSION, program_name);
656     if (flags & SHOW_COPYRIGHT)
657         av_log(NULL, level, " Copyright (c) %d-%d the FFmpeg developers",
658                program_birth_year, this_year);
659     av_log(NULL, level, "\n");
660     av_log(NULL, level, "%sbuilt on %s %s with %s\n",
661            indent, __DATE__, __TIME__, CC_IDENT);
662
663     av_log(NULL, level, "%sconfiguration: " FFMPEG_CONFIGURATION "\n", indent);
664 }
665
666 void show_banner(int argc, char **argv, const OptionDef *options)
667 {
668     int idx = locate_option(argc, argv, options, "version");
669     if (idx)
670         return;
671
672     print_program_info (INDENT|SHOW_COPYRIGHT, AV_LOG_INFO);
673     print_all_libs_info(INDENT|SHOW_CONFIG,  AV_LOG_INFO);
674     print_all_libs_info(INDENT|SHOW_VERSION, AV_LOG_INFO);
675 }
676
677 int show_version(const char *opt, const char *arg)
678 {
679     av_log_set_callback(log_callback_help);
680     print_program_info (0           , AV_LOG_INFO);
681     print_all_libs_info(SHOW_VERSION, AV_LOG_INFO);
682
683     return 0;
684 }
685
686 int show_license(const char *opt, const char *arg)
687 {
688     printf(
689 #if CONFIG_NONFREE
690     "This version of %s has nonfree parts compiled in.\n"
691     "Therefore it is not legally redistributable.\n",
692     program_name
693 #elif CONFIG_GPLV3
694     "%s is free software; you can redistribute it and/or modify\n"
695     "it under the terms of the GNU General Public License as published by\n"
696     "the Free Software Foundation; either version 3 of the License, or\n"
697     "(at your option) any later version.\n"
698     "\n"
699     "%s is distributed in the hope that it will be useful,\n"
700     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
701     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
702     "GNU General Public License for more details.\n"
703     "\n"
704     "You should have received a copy of the GNU General Public License\n"
705     "along with %s.  If not, see <http://www.gnu.org/licenses/>.\n",
706     program_name, program_name, program_name
707 #elif CONFIG_GPL
708     "%s is free software; you can redistribute it and/or modify\n"
709     "it under the terms of the GNU General Public License as published by\n"
710     "the Free Software Foundation; either version 2 of the License, or\n"
711     "(at your option) any later version.\n"
712     "\n"
713     "%s is distributed in the hope that it will be useful,\n"
714     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
715     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
716     "GNU General Public License for more details.\n"
717     "\n"
718     "You should have received a copy of the GNU General Public License\n"
719     "along with %s; if not, write to the Free Software\n"
720     "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n",
721     program_name, program_name, program_name
722 #elif CONFIG_LGPLV3
723     "%s is free software; you can redistribute it and/or modify\n"
724     "it under the terms of the GNU Lesser General Public License as published by\n"
725     "the Free Software Foundation; either version 3 of the License, or\n"
726     "(at your option) any later version.\n"
727     "\n"
728     "%s is distributed in the hope that it will be useful,\n"
729     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
730     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
731     "GNU Lesser General Public License for more details.\n"
732     "\n"
733     "You should have received a copy of the GNU Lesser General Public License\n"
734     "along with %s.  If not, see <http://www.gnu.org/licenses/>.\n",
735     program_name, program_name, program_name
736 #else
737     "%s is free software; you can redistribute it and/or\n"
738     "modify it under the terms of the GNU Lesser General Public\n"
739     "License as published by the Free Software Foundation; either\n"
740     "version 2.1 of the License, or (at your option) any later version.\n"
741     "\n"
742     "%s is distributed in the hope that it will be useful,\n"
743     "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
744     "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\n"
745     "Lesser General Public License for more details.\n"
746     "\n"
747     "You should have received a copy of the GNU Lesser General Public\n"
748     "License along with %s; if not, write to the Free Software\n"
749     "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n",
750     program_name, program_name, program_name
751 #endif
752     );
753
754     return 0;
755 }
756
757 int show_formats(const char *opt, const char *arg)
758 {
759     AVInputFormat *ifmt  = NULL;
760     AVOutputFormat *ofmt = NULL;
761     const char *last_name;
762
763     printf("File formats:\n"
764            " D. = Demuxing supported\n"
765            " .E = Muxing supported\n"
766            " --\n");
767     last_name = "000";
768     for (;;) {
769         int decode = 0;
770         int encode = 0;
771         const char *name      = NULL;
772         const char *long_name = NULL;
773
774         while ((ofmt = av_oformat_next(ofmt))) {
775             if ((name == NULL || strcmp(ofmt->name, name) < 0) &&
776                 strcmp(ofmt->name, last_name) > 0) {
777                 name      = ofmt->name;
778                 long_name = ofmt->long_name;
779                 encode    = 1;
780             }
781         }
782         while ((ifmt = av_iformat_next(ifmt))) {
783             if ((name == NULL || strcmp(ifmt->name, name) < 0) &&
784                 strcmp(ifmt->name, last_name) > 0) {
785                 name      = ifmt->name;
786                 long_name = ifmt->long_name;
787                 encode    = 0;
788             }
789             if (name && strcmp(ifmt->name, name) == 0)
790                 decode = 1;
791         }
792         if (name == NULL)
793             break;
794         last_name = name;
795
796         printf(" %s%s %-15s %s\n",
797                decode ? "D" : " ",
798                encode ? "E" : " ",
799                name,
800             long_name ? long_name:" ");
801     }
802     return 0;
803 }
804
805 static char get_media_type_char(enum AVMediaType type)
806 {
807     switch (type) {
808         case AVMEDIA_TYPE_VIDEO:    return 'V';
809         case AVMEDIA_TYPE_AUDIO:    return 'A';
810         case AVMEDIA_TYPE_DATA:     return 'D';
811         case AVMEDIA_TYPE_SUBTITLE: return 'S';
812         case AVMEDIA_TYPE_ATTACHMENT:return 'T';
813         default:                    return '?';
814     }
815 }
816
817 static const AVCodec *next_codec_for_id(enum AVCodecID id, const AVCodec *prev,
818                                         int encoder)
819 {
820     while ((prev = av_codec_next(prev))) {
821         if (prev->id == id &&
822             (encoder ? av_codec_is_encoder(prev) : av_codec_is_decoder(prev)))
823             return prev;
824     }
825     return NULL;
826 }
827
828 static void print_codecs_for_id(enum AVCodecID id, int encoder)
829 {
830     const AVCodec *codec = NULL;
831
832     printf(" (%s: ", encoder ? "encoders" : "decoders");
833
834     while ((codec = next_codec_for_id(id, codec, encoder)))
835         printf("%s ", codec->name);
836
837     printf(")");
838 }
839
840 int show_codecs(const char *opt, const char *arg)
841 {
842     const AVCodecDescriptor *desc = NULL;
843
844     printf("Codecs:\n"
845            " D... = Decoding supported\n"
846            " .E.. = Encoding supported\n"
847            " ..V. = Video codec\n"
848            " ..A. = Audio codec\n"
849            " ..S. = Subtitle codec\n"
850            " ...I = Intra frame-only codec\n"
851            " -----\n");
852     while ((desc = avcodec_descriptor_next(desc))) {
853         const AVCodec *codec = NULL;
854
855         printf(avcodec_find_decoder(desc->id) ? "D" : ".");
856         printf(avcodec_find_encoder(desc->id) ? "E" : ".");
857
858         printf("%c", get_media_type_char(desc->type));
859         printf((desc->props & AV_CODEC_PROP_INTRA_ONLY) ? "I" : ".");
860
861         printf(" %-20s %s", desc->name, desc->long_name ? desc->long_name : "");
862
863         /* print decoders/encoders when there's more than one or their
864          * names are different from codec name */
865         while ((codec = next_codec_for_id(desc->id, codec, 0))) {
866             if (strcmp(codec->name, desc->name)) {
867                 print_codecs_for_id(desc->id, 0);
868                 break;
869             }
870         }
871         codec = NULL;
872         while ((codec = next_codec_for_id(desc->id, codec, 1))) {
873             if (strcmp(codec->name, desc->name)) {
874                 print_codecs_for_id(desc->id, 1);
875                 break;
876             }
877         }
878
879         printf("\n");
880     }
881     return 0;
882 }
883
884 static void print_codecs(int encoder)
885 {
886     const AVCodecDescriptor *desc = NULL;
887
888     printf("%s:\n"
889            " V..... = Video\n"
890            " A..... = Audio\n"
891            " S..... = Subtitle\n"
892            " .F.... = Frame-level multithreading\n"
893            " ..S... = Slice-level multithreading\n"
894            " ...X.. = Codec is experimental\n"
895            " ....B. = Supports draw_horiz_band\n"
896            " .....D = Supports direct rendering method 1\n"
897            " ------\n",
898            encoder ? "Encoders" : "Decoders");
899     while ((desc = avcodec_descriptor_next(desc))) {
900         const AVCodec *codec = NULL;
901
902         while ((codec = next_codec_for_id(desc->id, codec, encoder))) {
903             printf("%c", get_media_type_char(desc->type));
904             printf((codec->capabilities & CODEC_CAP_FRAME_THREADS) ? "F" : ".");
905             printf((codec->capabilities & CODEC_CAP_SLICE_THREADS) ? "S" : ".");
906             printf((codec->capabilities & CODEC_CAP_EXPERIMENTAL)  ? "X" : ".");
907             printf((codec->capabilities & CODEC_CAP_DRAW_HORIZ_BAND)?"B" : ".");
908             printf((codec->capabilities & CODEC_CAP_DR1)           ? "D" : ".");
909
910             printf(" %-20s %s", codec->name, codec->long_name ? codec->long_name : "");
911             if (strcmp(codec->name, desc->name))
912                 printf(" (codec %s)", desc->name);
913
914             printf("\n");
915         }
916     }
917 }
918
919 int show_decoders(const char *opt, const char *arg)
920 {
921     print_codecs(0);
922     return 0;
923 }
924
925 int show_encoders(const char *opt, const char *arg)
926 {
927     print_codecs(1);
928     return 0;
929 }
930
931 int show_bsfs(const char *opt, const char *arg)
932 {
933     AVBitStreamFilter *bsf = NULL;
934
935     printf("Bitstream filters:\n");
936     while ((bsf = av_bitstream_filter_next(bsf)))
937         printf("%s\n", bsf->name);
938     printf("\n");
939     return 0;
940 }
941
942 int show_protocols(const char *opt, const char *arg)
943 {
944     void *opaque = NULL;
945     const char *name;
946
947     printf("Supported file protocols:\n"
948            "Input:\n");
949     while ((name = avio_enum_protocols(&opaque, 0)))
950         printf("%s\n", name);
951     printf("Output:\n");
952     while ((name = avio_enum_protocols(&opaque, 1)))
953         printf("%s\n", name);
954     return 0;
955 }
956
957 int show_filters(const char *opt, const char *arg)
958 {
959     AVFilter av_unused(**filter) = NULL;
960     char descr[64], *descr_cur;
961     int i, j;
962     const AVFilterPad *pad;
963
964     printf("Filters:\n");
965 #if CONFIG_AVFILTER
966     while ((filter = av_filter_next(filter)) && *filter) {
967         descr_cur = descr;
968         for (i = 0; i < 2; i++) {
969             if (i) {
970                 *(descr_cur++) = '-';
971                 *(descr_cur++) = '>';
972             }
973             pad = i ? (*filter)->outputs : (*filter)->inputs;
974             for (j = 0; pad[j].name; j++) {
975                 if (descr_cur >= descr + sizeof(descr) - 4)
976                     break;
977                 *(descr_cur++) = get_media_type_char(pad[j].type);
978             }
979             if (!j)
980                 *(descr_cur++) = '|';
981         }
982         *descr_cur = 0;
983         printf("%-16s %-10s %s\n", (*filter)->name, descr, (*filter)->description);
984     }
985 #endif
986     return 0;
987 }
988
989 int show_pix_fmts(const char *opt, const char *arg)
990 {
991     enum PixelFormat pix_fmt;
992
993     printf("Pixel formats:\n"
994            "I.... = Supported Input  format for conversion\n"
995            ".O... = Supported Output format for conversion\n"
996            "..H.. = Hardware accelerated format\n"
997            "...P. = Paletted format\n"
998            "....B = Bitstream format\n"
999            "FLAGS NAME            NB_COMPONENTS BITS_PER_PIXEL\n"
1000            "-----\n");
1001
1002 #if !CONFIG_SWSCALE
1003 #   define sws_isSupportedInput(x)  0
1004 #   define sws_isSupportedOutput(x) 0
1005 #endif
1006
1007     for (pix_fmt = 0; pix_fmt < PIX_FMT_NB; pix_fmt++) {
1008         const AVPixFmtDescriptor *pix_desc = &av_pix_fmt_descriptors[pix_fmt];
1009         if(!pix_desc->name)
1010             continue;
1011         printf("%c%c%c%c%c %-16s       %d            %2d\n",
1012                sws_isSupportedInput (pix_fmt)      ? 'I' : '.',
1013                sws_isSupportedOutput(pix_fmt)      ? 'O' : '.',
1014                pix_desc->flags & PIX_FMT_HWACCEL   ? 'H' : '.',
1015                pix_desc->flags & PIX_FMT_PAL       ? 'P' : '.',
1016                pix_desc->flags & PIX_FMT_BITSTREAM ? 'B' : '.',
1017                pix_desc->name,
1018                pix_desc->nb_components,
1019                av_get_bits_per_pixel(pix_desc));
1020     }
1021     return 0;
1022 }
1023
1024 int show_sample_fmts(const char *opt, const char *arg)
1025 {
1026     int i;
1027     char fmt_str[128];
1028     for (i = -1; i < AV_SAMPLE_FMT_NB; i++)
1029         printf("%s\n", av_get_sample_fmt_string(fmt_str, sizeof(fmt_str), i));
1030     return 0;
1031 }
1032
1033 int read_yesno(void)
1034 {
1035     int c = getchar();
1036     int yesno = (toupper(c) == 'Y');
1037
1038     while (c != '\n' && c != EOF)
1039         c = getchar();
1040
1041     return yesno;
1042 }
1043
1044 int cmdutils_read_file(const char *filename, char **bufptr, size_t *size)
1045 {
1046     int ret;
1047     FILE *f = fopen(filename, "rb");
1048
1049     if (!f) {
1050         av_log(NULL, AV_LOG_ERROR, "Cannot read file '%s': %s\n", filename,
1051                strerror(errno));
1052         return AVERROR(errno);
1053     }
1054     fseek(f, 0, SEEK_END);
1055     *size = ftell(f);
1056     fseek(f, 0, SEEK_SET);
1057     *bufptr = av_malloc(*size + 1);
1058     if (!*bufptr) {
1059         av_log(NULL, AV_LOG_ERROR, "Could not allocate file buffer\n");
1060         fclose(f);
1061         return AVERROR(ENOMEM);
1062     }
1063     ret = fread(*bufptr, 1, *size, f);
1064     if (ret < *size) {
1065         av_free(*bufptr);
1066         if (ferror(f)) {
1067             av_log(NULL, AV_LOG_ERROR, "Error while reading file '%s': %s\n",
1068                    filename, strerror(errno));
1069             ret = AVERROR(errno);
1070         } else
1071             ret = AVERROR_EOF;
1072     } else {
1073         ret = 0;
1074         (*bufptr)[*size++] = '\0';
1075     }
1076
1077     fclose(f);
1078     return ret;
1079 }
1080
1081 FILE *get_preset_file(char *filename, size_t filename_size,
1082                       const char *preset_name, int is_path,
1083                       const char *codec_name)
1084 {
1085     FILE *f = NULL;
1086     int i;
1087     const char *base[3] = { getenv("FFMPEG_DATADIR"),
1088                             getenv("HOME"),
1089                             FFMPEG_DATADIR, };
1090
1091     if (is_path) {
1092         av_strlcpy(filename, preset_name, filename_size);
1093         f = fopen(filename, "r");
1094     } else {
1095 #ifdef _WIN32
1096         char datadir[MAX_PATH], *ls;
1097         base[2] = NULL;
1098
1099         if (GetModuleFileNameA(GetModuleHandleA(NULL), datadir, sizeof(datadir) - 1))
1100         {
1101             for (ls = datadir; ls < datadir + strlen(datadir); ls++)
1102                 if (*ls == '\\') *ls = '/';
1103
1104             if (ls = strrchr(datadir, '/'))
1105             {
1106                 *ls = 0;
1107                 strncat(datadir, "/ffpresets",  sizeof(datadir) - 1 - strlen(datadir));
1108                 base[2] = datadir;
1109             }
1110         }
1111 #endif
1112         for (i = 0; i < 3 && !f; i++) {
1113             if (!base[i])
1114                 continue;
1115             snprintf(filename, filename_size, "%s%s/%s.ffpreset", base[i],
1116                      i != 1 ? "" : "/.ffmpeg", preset_name);
1117             f = fopen(filename, "r");
1118             if (!f && codec_name) {
1119                 snprintf(filename, filename_size,
1120                          "%s%s/%s-%s.ffpreset",
1121                          base[i], i != 1 ? "" : "/.ffmpeg", codec_name,
1122                          preset_name);
1123                 f = fopen(filename, "r");
1124             }
1125         }
1126     }
1127
1128     return f;
1129 }
1130
1131 int check_stream_specifier(AVFormatContext *s, AVStream *st, const char *spec)
1132 {
1133     int ret = avformat_match_stream_specifier(s, st, spec);
1134     if (ret < 0)
1135         av_log(s, AV_LOG_ERROR, "Invalid stream specifier: %s.\n", spec);
1136     return ret;
1137 }
1138
1139 AVDictionary *filter_codec_opts(AVDictionary *opts, enum AVCodecID codec_id,
1140                                 AVFormatContext *s, AVStream *st, AVCodec *codec)
1141 {
1142     AVDictionary    *ret = NULL;
1143     AVDictionaryEntry *t = NULL;
1144     int            flags = s->oformat ? AV_OPT_FLAG_ENCODING_PARAM
1145                                       : AV_OPT_FLAG_DECODING_PARAM;
1146     char          prefix = 0;
1147     const AVClass    *cc = avcodec_get_class();
1148
1149     if (!codec)
1150         codec            = s->oformat ? avcodec_find_encoder(codec_id)
1151                                       : avcodec_find_decoder(codec_id);
1152     if (!codec)
1153         return NULL;
1154
1155     switch (codec->type) {
1156     case AVMEDIA_TYPE_VIDEO:
1157         prefix  = 'v';
1158         flags  |= AV_OPT_FLAG_VIDEO_PARAM;
1159         break;
1160     case AVMEDIA_TYPE_AUDIO:
1161         prefix  = 'a';
1162         flags  |= AV_OPT_FLAG_AUDIO_PARAM;
1163         break;
1164     case AVMEDIA_TYPE_SUBTITLE:
1165         prefix  = 's';
1166         flags  |= AV_OPT_FLAG_SUBTITLE_PARAM;
1167         break;
1168     }
1169
1170     while (t = av_dict_get(opts, "", t, AV_DICT_IGNORE_SUFFIX)) {
1171         char *p = strchr(t->key, ':');
1172
1173         /* check stream specification in opt name */
1174         if (p)
1175             switch (check_stream_specifier(s, st, p + 1)) {
1176             case  1: *p = 0; break;
1177             case  0:         continue;
1178             default:         return NULL;
1179             }
1180
1181         if (av_opt_find(&cc, t->key, NULL, flags, AV_OPT_SEARCH_FAKE_OBJ) ||
1182             (codec && codec->priv_class &&
1183              av_opt_find(&codec->priv_class, t->key, NULL, flags,
1184                          AV_OPT_SEARCH_FAKE_OBJ)))
1185             av_dict_set(&ret, t->key, t->value, 0);
1186         else if (t->key[0] == prefix &&
1187                  av_opt_find(&cc, t->key + 1, NULL, flags,
1188                              AV_OPT_SEARCH_FAKE_OBJ))
1189             av_dict_set(&ret, t->key + 1, t->value, 0);
1190
1191         if (p)
1192             *p = ':';
1193     }
1194     return ret;
1195 }
1196
1197 AVDictionary **setup_find_stream_info_opts(AVFormatContext *s,
1198                                            AVDictionary *codec_opts)
1199 {
1200     int i;
1201     AVDictionary **opts;
1202
1203     if (!s->nb_streams)
1204         return NULL;
1205     opts = av_mallocz(s->nb_streams * sizeof(*opts));
1206     if (!opts) {
1207         av_log(NULL, AV_LOG_ERROR,
1208                "Could not alloc memory for stream options.\n");
1209         return NULL;
1210     }
1211     for (i = 0; i < s->nb_streams; i++)
1212         opts[i] = filter_codec_opts(codec_opts, s->streams[i]->codec->codec_id,
1213                                     s, s->streams[i], NULL);
1214     return opts;
1215 }
1216
1217 void *grow_array(void *array, int elem_size, int *size, int new_size)
1218 {
1219     if (new_size >= INT_MAX / elem_size) {
1220         av_log(NULL, AV_LOG_ERROR, "Array too big.\n");
1221         exit_program(1);
1222     }
1223     if (*size < new_size) {
1224         uint8_t *tmp = av_realloc(array, new_size*elem_size);
1225         if (!tmp) {
1226             av_log(NULL, AV_LOG_ERROR, "Could not alloc buffer.\n");
1227             exit_program(1);
1228         }
1229         memset(tmp + *size*elem_size, 0, (new_size-*size) * elem_size);
1230         *size = new_size;
1231         return tmp;
1232     }
1233     return array;
1234 }
1235
1236 static int alloc_buffer(FrameBuffer **pool, AVCodecContext *s, FrameBuffer **pbuf)
1237 {
1238     FrameBuffer  *buf = av_mallocz(sizeof(*buf));
1239     int i, ret;
1240     const int pixel_size = av_pix_fmt_descriptors[s->pix_fmt].comp[0].step_minus1+1;
1241     int h_chroma_shift, v_chroma_shift;
1242     int edge = 32; // XXX should be avcodec_get_edge_width(), but that fails on svq1
1243     int w = s->width, h = s->height;
1244
1245     if (!buf)
1246         return AVERROR(ENOMEM);
1247
1248     avcodec_align_dimensions(s, &w, &h);
1249
1250     if (!(s->flags & CODEC_FLAG_EMU_EDGE)) {
1251         w += 2*edge;
1252         h += 2*edge;
1253     }
1254
1255     if ((ret = av_image_alloc(buf->base, buf->linesize, w, h,
1256                               s->pix_fmt, 32)) < 0) {
1257         av_freep(&buf);
1258         av_log(s, AV_LOG_ERROR, "alloc_buffer: av_image_alloc() failed\n");
1259         return ret;
1260     }
1261     /* XXX this shouldn't be needed, but some tests break without this line
1262      * those decoders are buggy and need to be fixed.
1263      * the following tests fail:
1264      * cdgraphics, ansi, aasc, fraps-v1, qtrle-1bit
1265      */
1266     memset(buf->base[0], 128, ret);
1267
1268     avcodec_get_chroma_sub_sample(s->pix_fmt, &h_chroma_shift, &v_chroma_shift);
1269     for (i = 0; i < FF_ARRAY_ELEMS(buf->data); i++) {
1270         const int h_shift = i==0 ? 0 : h_chroma_shift;
1271         const int v_shift = i==0 ? 0 : v_chroma_shift;
1272         if ((s->flags & CODEC_FLAG_EMU_EDGE) || !buf->linesize[i] || !buf->base[i])
1273             buf->data[i] = buf->base[i];
1274         else
1275             buf->data[i] = buf->base[i] +
1276                            FFALIGN((buf->linesize[i]*edge >> v_shift) +
1277                                    (pixel_size*edge >> h_shift), 32);
1278     }
1279     buf->w       = s->width;
1280     buf->h       = s->height;
1281     buf->pix_fmt = s->pix_fmt;
1282     buf->pool    = pool;
1283
1284     *pbuf = buf;
1285     return 0;
1286 }
1287
1288 int codec_get_buffer(AVCodecContext *s, AVFrame *frame)
1289 {
1290     FrameBuffer **pool = s->opaque;
1291     FrameBuffer *buf;
1292     int ret, i;
1293
1294     if(av_image_check_size(s->width, s->height, 0, s) || s->pix_fmt<0) {
1295         av_log(s, AV_LOG_ERROR, "codec_get_buffer: image parameters invalid\n");
1296         return -1;
1297     }
1298
1299     if (!*pool && (ret = alloc_buffer(pool, s, pool)) < 0)
1300         return ret;
1301
1302     buf              = *pool;
1303     *pool            = buf->next;
1304     buf->next        = NULL;
1305     if (buf->w != s->width || buf->h != s->height || buf->pix_fmt != s->pix_fmt) {
1306         av_freep(&buf->base[0]);
1307         av_free(buf);
1308         if ((ret = alloc_buffer(pool, s, &buf)) < 0)
1309             return ret;
1310     }
1311     av_assert0(!buf->refcount);
1312     buf->refcount++;
1313
1314     frame->opaque        = buf;
1315     frame->type          = FF_BUFFER_TYPE_USER;
1316     frame->extended_data = frame->data;
1317     frame->pkt_pts       = s->pkt ? s->pkt->pts : AV_NOPTS_VALUE;
1318     frame->width         = buf->w;
1319     frame->height        = buf->h;
1320     frame->format        = buf->pix_fmt;
1321     frame->sample_aspect_ratio = s->sample_aspect_ratio;
1322
1323     for (i = 0; i < FF_ARRAY_ELEMS(buf->data); i++) {
1324         frame->base[i]     = buf->base[i];  // XXX h264.c uses base though it shouldn't
1325         frame->data[i]     = buf->data[i];
1326         frame->linesize[i] = buf->linesize[i];
1327     }
1328
1329     return 0;
1330 }
1331
1332 static void unref_buffer(FrameBuffer *buf)
1333 {
1334     FrameBuffer **pool = buf->pool;
1335
1336     av_assert0(buf->refcount > 0);
1337     buf->refcount--;
1338     if (!buf->refcount) {
1339         FrameBuffer *tmp;
1340         for(tmp= *pool; tmp; tmp= tmp->next)
1341             av_assert1(tmp != buf);
1342
1343         buf->next = *pool;
1344         *pool = buf;
1345     }
1346 }
1347
1348 void codec_release_buffer(AVCodecContext *s, AVFrame *frame)
1349 {
1350     FrameBuffer *buf = frame->opaque;
1351     int i;
1352
1353     if(frame->type!=FF_BUFFER_TYPE_USER) {
1354         avcodec_default_release_buffer(s, frame);
1355         return;
1356     }
1357
1358     for (i = 0; i < FF_ARRAY_ELEMS(frame->data); i++)
1359         frame->data[i] = NULL;
1360
1361     unref_buffer(buf);
1362 }
1363
1364 void filter_release_buffer(AVFilterBuffer *fb)
1365 {
1366     FrameBuffer *buf = fb->priv;
1367     av_free(fb);
1368     unref_buffer(buf);
1369 }
1370
1371 void free_buffer_pool(FrameBuffer **pool)
1372 {
1373     FrameBuffer *buf = *pool;
1374     while (buf) {
1375         *pool = buf->next;
1376         av_freep(&buf->base[0]);
1377         av_free(buf);
1378         buf = *pool;
1379     }
1380 }