Add -timelimit option
[libav.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 "libswscale/swscale.h"
36 #include "libpostproc/postprocess.h"
37 #include "libavutil/avstring.h"
38 #include "libavcodec/opt.h"
39 #include "cmdutils.h"
40 #include "version.h"
41 #if CONFIG_NETWORK
42 #include "libavformat/network.h"
43 #endif
44 #if HAVE_SYS_RESOURCE_H
45 #include <sys/resource.h>
46 #endif
47
48 #undef exit
49
50 const char **opt_names;
51 static int opt_name_count;
52 AVCodecContext *avcodec_opts[CODEC_TYPE_NB];
53 AVFormatContext *avformat_opts;
54 struct SwsContext *sws_opts;
55
56 const int this_year = 2010;
57
58 double parse_number_or_die(const char *context, const char *numstr, int type, double min, double max)
59 {
60 char *tail;
61 const char *error;
62 double d = strtod(numstr, &tail);
63 if (*tail)
64 error= "Expected number for %s but found: %s\n";
65 else if (d < min || d > max)
66 error= "The value for %s was %s which is not within %f - %f\n";
67 else if(type == OPT_INT64 && (int64_t)d != d)
68 error= "Expected int64 for %s but found %s\n";
69 else
70 return d;
71 fprintf(stderr, error, context, numstr, min, max);
72 exit(1);
73 }
74
75 int64_t parse_time_or_die(const char *context, const char *timestr, int is_duration)
76 {
77 int64_t us = parse_date(timestr, is_duration);
78 if (us == INT64_MIN) {
79 fprintf(stderr, "Invalid %s specification for %s: %s\n",
80 is_duration ? "duration" : "date", context, timestr);
81 exit(1);
82 }
83 return us;
84 }
85
86 void show_help_options(const OptionDef *options, const char *msg, int mask, int value)
87 {
88 const OptionDef *po;
89 int first;
90
91 first = 1;
92 for(po = options; po->name != NULL; po++) {
93 char buf[64];
94 if ((po->flags & mask) == value) {
95 if (first) {
96 printf("%s", msg);
97 first = 0;
98 }
99 av_strlcpy(buf, po->name, sizeof(buf));
100 if (po->flags & HAS_ARG) {
101 av_strlcat(buf, " ", sizeof(buf));
102 av_strlcat(buf, po->argname, sizeof(buf));
103 }
104 printf("-%-17s %s\n", buf, po->help);
105 }
106 }
107 }
108
109 static const OptionDef* find_option(const OptionDef *po, const char *name){
110 while (po->name != NULL) {
111 if (!strcmp(name, po->name))
112 break;
113 po++;
114 }
115 return po;
116 }
117
118 void parse_options(int argc, char **argv, const OptionDef *options,
119 void (* parse_arg_function)(const char*))
120 {
121 const char *opt, *arg;
122 int optindex, handleoptions=1;
123 const OptionDef *po;
124
125 /* parse options */
126 optindex = 1;
127 while (optindex < argc) {
128 opt = argv[optindex++];
129
130 if (handleoptions && opt[0] == '-' && opt[1] != '\0') {
131 int bool_val = 1;
132 if (opt[1] == '-' && opt[2] == '\0') {
133 handleoptions = 0;
134 continue;
135 }
136 opt++;
137 po= find_option(options, opt);
138 if (!po->name && opt[0] == 'n' && opt[1] == 'o') {
139 /* handle 'no' bool option */
140 po = find_option(options, opt + 2);
141 if (!(po->name && (po->flags & OPT_BOOL)))
142 goto unknown_opt;
143 bool_val = 0;
144 }
145 if (!po->name)
146 po= find_option(options, "default");
147 if (!po->name) {
148 unknown_opt:
149 fprintf(stderr, "%s: unrecognized option '%s'\n", argv[0], opt);
150 exit(1);
151 }
152 arg = NULL;
153 if (po->flags & HAS_ARG) {
154 arg = argv[optindex++];
155 if (!arg) {
156 fprintf(stderr, "%s: missing argument for option '%s'\n", argv[0], opt);
157 exit(1);
158 }
159 }
160 if (po->flags & OPT_STRING) {
161 char *str;
162 str = av_strdup(arg);
163 *po->u.str_arg = str;
164 } else if (po->flags & OPT_BOOL) {
165 *po->u.int_arg = bool_val;
166 } else if (po->flags & OPT_INT) {
167 *po->u.int_arg = parse_number_or_die(opt, arg, OPT_INT64, INT_MIN, INT_MAX);
168 } else if (po->flags & OPT_INT64) {
169 *po->u.int64_arg = parse_number_or_die(opt, arg, OPT_INT64, INT64_MIN, INT64_MAX);
170 } else if (po->flags & OPT_FLOAT) {
171 *po->u.float_arg = parse_number_or_die(opt, arg, OPT_FLOAT, -1.0/0.0, 1.0/0.0);
172 } else if (po->flags & OPT_FUNC2) {
173 if(po->u.func2_arg(opt, arg)<0)
174 goto unknown_opt;
175 } else {
176 po->u.func_arg(arg);
177 }
178 if(po->flags & OPT_EXIT)
179 exit(0);
180 } else {
181 if (parse_arg_function)
182 parse_arg_function(opt);
183 }
184 }
185 }
186
187 int opt_default(const char *opt, const char *arg){
188 int type;
189 int ret= 0;
190 const AVOption *o= NULL;
191 int opt_types[]={AV_OPT_FLAG_VIDEO_PARAM, AV_OPT_FLAG_AUDIO_PARAM, 0, AV_OPT_FLAG_SUBTITLE_PARAM, 0};
192
193 for(type=0; type<CODEC_TYPE_NB && ret>= 0; type++){
194 const AVOption *o2 = av_find_opt(avcodec_opts[0], opt, NULL, opt_types[type], opt_types[type]);
195 if(o2)
196 ret = av_set_string3(avcodec_opts[type], opt, arg, 1, &o);
197 }
198 if(!o)
199 ret = av_set_string3(avformat_opts, opt, arg, 1, &o);
200 if(!o)
201 ret = av_set_string3(sws_opts, opt, arg, 1, &o);
202 if(!o){
203 if(opt[0] == 'a')
204 ret = av_set_string3(avcodec_opts[CODEC_TYPE_AUDIO], opt+1, arg, 1, &o);
205 else if(opt[0] == 'v')
206 ret = av_set_string3(avcodec_opts[CODEC_TYPE_VIDEO], opt+1, arg, 1, &o);
207 else if(opt[0] == 's')
208 ret = av_set_string3(avcodec_opts[CODEC_TYPE_SUBTITLE], opt+1, arg, 1, &o);
209 }
210 if (o && ret < 0) {
211 fprintf(stderr, "Invalid value '%s' for option '%s'\n", arg, opt);
212 exit(1);
213 }
214 if(!o)
215 return -1;
216
217 // av_log(NULL, AV_LOG_ERROR, "%s:%s: %f 0x%0X\n", opt, arg, av_get_double(avcodec_opts, opt, NULL), (int)av_get_int(avcodec_opts, opt, NULL));
218
219 //FIXME we should always use avcodec_opts, ... for storing options so there will not be any need to keep track of what i set over this
220 opt_names= av_realloc(opt_names, sizeof(void*)*(opt_name_count+1));
221 opt_names[opt_name_count++]= o->name;
222
223 if(avcodec_opts[0]->debug || avformat_opts->debug)
224 av_log_set_level(AV_LOG_DEBUG);
225 return 0;
226 }
227
228 int opt_loglevel(const char *opt, const char *arg)
229 {
230 const struct { const char *name; int level; } log_levels[] = {
231 { "quiet" , AV_LOG_QUIET },
232 { "panic" , AV_LOG_PANIC },
233 { "fatal" , AV_LOG_FATAL },
234 { "error" , AV_LOG_ERROR },
235 { "warning", AV_LOG_WARNING },
236 { "info" , AV_LOG_INFO },
237 { "verbose", AV_LOG_VERBOSE },
238 { "debug" , AV_LOG_DEBUG },
239 };
240 char *tail;
241 int level;
242 int i;
243
244 for (i = 0; i < FF_ARRAY_ELEMS(log_levels); i++) {
245 if (!strcmp(log_levels[i].name, arg)) {
246 av_log_set_level(log_levels[i].level);
247 return 0;
248 }
249 }
250
251 level = strtol(arg, &tail, 10);
252 if (*tail) {
253 fprintf(stderr, "Invalid loglevel \"%s\". "
254 "Possible levels are numbers or:\n", arg);
255 for (i = 0; i < FF_ARRAY_ELEMS(log_levels); i++)
256 fprintf(stderr, "\"%s\"\n", log_levels[i].name);
257 exit(1);
258 }
259 av_log_set_level(level);
260 return 0;
261 }
262
263 int opt_timelimit(const char *opt, const char *arg)
264 {
265 #if HAVE_SYS_RESOURCE_H
266 int lim = parse_number_or_die(opt, arg, OPT_INT64, 0, INT_MAX);
267 struct rlimit rl = { lim, lim + 1 };
268 if (setrlimit(RLIMIT_CPU, &rl))
269 perror("setrlimit");
270 #else
271 fprintf(stderr, "Warning: -%s not implemented on this OS\n", opt);
272 #endif
273 return 0;
274 }
275
276 void set_context_opts(void *ctx, void *opts_ctx, int flags)
277 {
278 int i;
279 for(i=0; i<opt_name_count; i++){
280 char buf[256];
281 const AVOption *opt;
282 const char *str= av_get_string(opts_ctx, opt_names[i], &opt, buf, sizeof(buf));
283 /* if an option with name opt_names[i] is present in opts_ctx then str is non-NULL */
284 if(str && ((opt->flags & flags) == flags))
285 av_set_string3(ctx, opt_names[i], str, 1, NULL);
286 }
287 }
288
289 void print_error(const char *filename, int err)
290 {
291 switch(err) {
292 case AVERROR_NUMEXPECTED:
293 fprintf(stderr, "%s: Incorrect image filename syntax.\n"
294 "Use '%%d' to specify the image number:\n"
295 " for img1.jpg, img2.jpg, ..., use 'img%%d.jpg';\n"
296 " for img001.jpg, img002.jpg, ..., use 'img%%03d.jpg'.\n",
297 filename);
298 break;
299 case AVERROR_INVALIDDATA:
300 fprintf(stderr, "%s: Error while parsing header\n", filename);
301 break;
302 case AVERROR_NOFMT:
303 fprintf(stderr, "%s: Unknown format\n", filename);
304 break;
305 case AVERROR(EIO):
306 fprintf(stderr, "%s: I/O error occurred\n"
307 "Usually that means that input file is truncated and/or corrupted.\n",
308 filename);
309 break;
310 case AVERROR(ENOMEM):
311 fprintf(stderr, "%s: memory allocation error occurred\n", filename);
312 break;
313 case AVERROR(ENOENT):
314 fprintf(stderr, "%s: no such file or directory\n", filename);
315 break;
316 #if CONFIG_NETWORK
317 case AVERROR(FF_NETERROR(EPROTONOSUPPORT)):
318 fprintf(stderr, "%s: Unsupported network protocol\n", filename);
319 break;
320 #endif
321 default:
322 fprintf(stderr, "%s: Error while opening file\n", filename);
323 break;
324 }
325 }
326
327 #define PRINT_LIB_VERSION(outstream,libname,LIBNAME,indent) \
328 version= libname##_version(); \
329 fprintf(outstream, "%slib%-10s %2d.%2d.%2d / %2d.%2d.%2d\n", indent? " " : "", #libname, \
330 LIB##LIBNAME##_VERSION_MAJOR, LIB##LIBNAME##_VERSION_MINOR, LIB##LIBNAME##_VERSION_MICRO, \
331 version >> 16, version >> 8 & 0xff, version & 0xff);
332
333 static void print_all_lib_versions(FILE* outstream, int indent)
334 {
335 unsigned int version;
336 PRINT_LIB_VERSION(outstream, avutil, AVUTIL, indent);
337 PRINT_LIB_VERSION(outstream, avcodec, AVCODEC, indent);
338 PRINT_LIB_VERSION(outstream, avformat, AVFORMAT, indent);
339 PRINT_LIB_VERSION(outstream, avdevice, AVDEVICE, indent);
340 #if CONFIG_AVFILTER
341 PRINT_LIB_VERSION(outstream, avfilter, AVFILTER, indent);
342 #endif
343 PRINT_LIB_VERSION(outstream, swscale, SWSCALE, indent);
344 #if CONFIG_POSTPROC
345 PRINT_LIB_VERSION(outstream, postproc, POSTPROC, indent);
346 #endif
347 }
348
349 void show_banner(void)
350 {
351 fprintf(stderr, "%s version " FFMPEG_VERSION ", Copyright (c) %d-%d Fabrice Bellard, et al.\n",
352 program_name, program_birth_year, this_year);
353 fprintf(stderr, " built on %s %s with %s %s\n",
354 __DATE__, __TIME__, CC_TYPE, CC_VERSION);
355 fprintf(stderr, " configuration: " FFMPEG_CONFIGURATION "\n");
356 print_all_lib_versions(stderr, 1);
357 }
358
359 void show_version(void) {
360 printf("%s " FFMPEG_VERSION "\n", program_name);
361 print_all_lib_versions(stdout, 0);
362 }
363
364 void show_license(void)
365 {
366 printf(
367 #if CONFIG_NONFREE
368 "This version of %s has nonfree parts compiled in.\n"
369 "Therefore it is not legally redistributable.\n",
370 program_name
371 #elif CONFIG_GPLV3
372 "%s is free software; you can redistribute it and/or modify\n"
373 "it under the terms of the GNU General Public License as published by\n"
374 "the Free Software Foundation; either version 3 of the License, or\n"
375 "(at your option) any later version.\n"
376 "\n"
377 "%s is distributed in the hope that it will be useful,\n"
378 "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
379 "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n"
380 "GNU General Public License for more details.\n"
381 "\n"
382 "You should have received a copy of the GNU General Public License\n"
383 "along with %s. If not, see <http://www.gnu.org/licenses/>.\n",
384 program_name, program_name, program_name
385 #elif CONFIG_GPL
386 "%s is free software; you can redistribute it and/or modify\n"
387 "it under the terms of the GNU General Public License as published by\n"
388 "the Free Software Foundation; either version 2 of the License, or\n"
389 "(at your option) any later version.\n"
390 "\n"
391 "%s is distributed in the hope that it will be useful,\n"
392 "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
393 "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n"
394 "GNU General Public License for more details.\n"
395 "\n"
396 "You should have received a copy of the GNU General Public License\n"
397 "along with %s; if not, write to the Free Software\n"
398 "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n",
399 program_name, program_name, program_name
400 #elif CONFIG_LGPLV3
401 "%s is free software; you can redistribute it and/or modify\n"
402 "it under the terms of the GNU Lesser General Public License as published by\n"
403 "the Free Software Foundation; either version 3 of the License, or\n"
404 "(at your option) any later version.\n"
405 "\n"
406 "%s is distributed in the hope that it will be useful,\n"
407 "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
408 "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n"
409 "GNU Lesser General Public License for more details.\n"
410 "\n"
411 "You should have received a copy of the GNU Lesser General Public License\n"
412 "along with %s. If not, see <http://www.gnu.org/licenses/>.\n",
413 program_name, program_name, program_name
414 #else
415 "%s is free software; you can redistribute it and/or\n"
416 "modify it under the terms of the GNU Lesser General Public\n"
417 "License as published by the Free Software Foundation; either\n"
418 "version 2.1 of the License, or (at your option) any later version.\n"
419 "\n"
420 "%s is distributed in the hope that it will be useful,\n"
421 "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
422 "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU\n"
423 "Lesser General Public License for more details.\n"
424 "\n"
425 "You should have received a copy of the GNU Lesser General Public\n"
426 "License along with %s; if not, write to the Free Software\n"
427 "Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA\n",
428 program_name, program_name, program_name
429 #endif
430 );
431 }
432
433 void list_fmts(void (*get_fmt_string)(char *buf, int buf_size, int fmt), int nb_fmts)
434 {
435 int i;
436 char fmt_str[128];
437 for (i=-1; i < nb_fmts; i++) {
438 get_fmt_string (fmt_str, sizeof(fmt_str), i);
439 fprintf(stdout, "%s\n", fmt_str);
440 }
441 }
442
443 void show_formats(void)
444 {
445 AVInputFormat *ifmt=NULL;
446 AVOutputFormat *ofmt=NULL;
447 const char *last_name;
448
449 printf(
450 "File formats:\n"
451 " D. = Demuxing supported\n"
452 " .E = Muxing supported\n"
453 " --\n");
454 last_name= "000";
455 for(;;){
456 int decode=0;
457 int encode=0;
458 const char *name=NULL;
459 const char *long_name=NULL;
460
461 while((ofmt= av_oformat_next(ofmt))) {
462 if((name == NULL || strcmp(ofmt->name, name)<0) &&
463 strcmp(ofmt->name, last_name)>0){
464 name= ofmt->name;
465 long_name= ofmt->long_name;
466 encode=1;
467 }
468 }
469 while((ifmt= av_iformat_next(ifmt))) {
470 if((name == NULL || strcmp(ifmt->name, name)<0) &&
471 strcmp(ifmt->name, last_name)>0){
472 name= ifmt->name;
473 long_name= ifmt->long_name;
474 encode=0;
475 }
476 if(name && strcmp(ifmt->name, name)==0)
477 decode=1;
478 }
479 if(name==NULL)
480 break;
481 last_name= name;
482
483 printf(
484 " %s%s %-15s %s\n",
485 decode ? "D":" ",
486 encode ? "E":" ",
487 name,
488 long_name ? long_name:" ");
489 }
490 }
491
492 void show_codecs(void)
493 {
494 AVCodec *p=NULL, *p2;
495 const char *last_name;
496 printf(
497 "Codecs:\n"
498 " D..... = Decoding supported\n"
499 " .E.... = Encoding supported\n"
500 " ..V... = Video codec\n"
501 " ..A... = Audio codec\n"
502 " ..S... = Subtitle codec\n"
503 " ...S.. = Supports draw_horiz_band\n"
504 " ....D. = Supports direct rendering method 1\n"
505 " .....T = Supports weird frame truncation\n"
506 " ------\n");
507 last_name= "000";
508 for(;;){
509 int decode=0;
510 int encode=0;
511 int cap=0;
512 const char *type_str;
513
514 p2=NULL;
515 while((p= av_codec_next(p))) {
516 if((p2==NULL || strcmp(p->name, p2->name)<0) &&
517 strcmp(p->name, last_name)>0){
518 p2= p;
519 decode= encode= cap=0;
520 }
521 if(p2 && strcmp(p->name, p2->name)==0){
522 if(p->decode) decode=1;
523 if(p->encode) encode=1;
524 cap |= p->capabilities;
525 }
526 }
527 if(p2==NULL)
528 break;
529 last_name= p2->name;
530
531 switch(p2->type) {
532 case CODEC_TYPE_VIDEO:
533 type_str = "V";
534 break;
535 case CODEC_TYPE_AUDIO:
536 type_str = "A";
537 break;
538 case CODEC_TYPE_SUBTITLE:
539 type_str = "S";
540 break;
541 default:
542 type_str = "?";
543 break;
544 }
545 printf(
546 " %s%s%s%s%s%s %-15s %s",
547 decode ? "D": (/*p2->decoder ? "d":*/" "),
548 encode ? "E":" ",
549 type_str,
550 cap & CODEC_CAP_DRAW_HORIZ_BAND ? "S":" ",
551 cap & CODEC_CAP_DR1 ? "D":" ",
552 cap & CODEC_CAP_TRUNCATED ? "T":" ",
553 p2->name,
554 p2->long_name ? p2->long_name : "");
555 /* if(p2->decoder && decode==0)
556 printf(" use %s for decoding", p2->decoder->name);*/
557 printf("\n");
558 }
559 printf("\n");
560 printf(
561 "Note, the names of encoders and decoders do not always match, so there are\n"
562 "several cases where the above table shows encoder only or decoder only entries\n"
563 "even though both encoding and decoding are supported. For example, the h263\n"
564 "decoder corresponds to the h263 and h263p encoders, for file formats it is even\n"
565 "worse.\n");
566 }
567
568 void show_bsfs(void)
569 {
570 AVBitStreamFilter *bsf=NULL;
571
572 printf("Bitstream filters:\n");
573 while((bsf = av_bitstream_filter_next(bsf)))
574 printf("%s\n", bsf->name);
575 printf("\n");
576 }
577
578 void show_protocols(void)
579 {
580 URLProtocol *up=NULL;
581
582 printf("Supported file protocols:\n");
583 while((up = av_protocol_next(up)))
584 printf("%s\n", up->name);
585 printf("\n");
586
587 printf("Frame size, frame rate abbreviations:\n ntsc pal qntsc qpal sntsc spal film ntsc-film sqcif qcif cif 4cif\n");
588 }
589
590 void show_filters(void)
591 {
592 AVFilter av_unused(**filter) = NULL;
593
594 printf("Filters:\n");
595 #if CONFIG_AVFILTER
596 while ((filter = av_filter_next(filter)) && *filter)
597 printf("%-16s %s\n", (*filter)->name, (*filter)->description);
598 #endif
599 }
600
601 void show_pix_fmts(void)
602 {
603 list_fmts(avcodec_pix_fmt_string, PIX_FMT_NB);
604 }
605
606 int read_yesno(void)
607 {
608 int c = getchar();
609 int yesno = (toupper(c) == 'Y');
610
611 while (c != '\n' && c != EOF)
612 c = getchar();
613
614 return yesno;
615 }