git.sesse.net Git - ffmpeg/blob - libavformat/format.c

   1 /*
   2  * Format register and lookup
   3  * Copyright (c) 2000, 2001, 2002 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 "avformat.h"
  23 #include "internal.h"
  24 #include "libavutil/atomic.h"
  25 #include "libavutil/avstring.h"
  26
  27 /**
  28  * @file
  29  * Format register and lookup
  30  */
  31 /** head of registered input format linked list */
  32 static AVInputFormat *first_iformat = NULL;
  33 /** head of registered output format linked list */
  34 static AVOutputFormat *first_oformat = NULL;
  35
  36 AVInputFormat *av_iformat_next(AVInputFormat *f)
  37 {
  38     if (f)
  39         return f->next;
  40     else
  41         return first_iformat;
  42 }
  43
  44 AVOutputFormat *av_oformat_next(AVOutputFormat *f)
  45 {
  46     if (f)
  47         return f->next;
  48     else
  49         return first_oformat;
  50 }
  51
  52 void av_register_input_format(AVInputFormat *format)
  53 {
  54     AVInputFormat **p = &first_iformat;
  55
  56     format->next = NULL;
  57     while(avpriv_atomic_ptr_cas((void * volatile *)p, NULL, format))
  58         p = &(*p)->next;
  59 }
  60
  61 void av_register_output_format(AVOutputFormat *format)
  62 {
  63     AVOutputFormat **p = &first_oformat;
  64
  65     format->next = NULL;
  66     while(avpriv_atomic_ptr_cas((void * volatile *)p, NULL, format))
  67         p = &(*p)->next;
  68 }
  69
  70 int av_match_ext(const char *filename, const char *extensions)
  71 {
  72     const char *ext, *p;
  73     char ext1[32], *q;
  74
  75     if (!filename)
  76         return 0;
  77
  78     ext = strrchr(filename, '.');
  79     if (ext) {
  80         ext++;
  81         p = extensions;
  82         for (;;) {
  83             q = ext1;
  84             while (*p != '\0' && *p != ','  && q - ext1 < sizeof(ext1) - 1)
  85                 *q++ = *p++;
  86             *q = '\0';
  87             if (!av_strcasecmp(ext1, ext))
  88                 return 1;
  89             if (*p == '\0')
  90                 break;
  91             p++;
  92         }
  93     }
  94     return 0;
  95 }
  96
  97 static int match_format(const char *name, const char *names)
  98 {
  99     const char *p;
 100     int len, namelen;
 101
 102     if (!name || !names)
 103         return 0;
 104
 105     namelen = strlen(name);
 106     while ((p = strchr(names, ','))) {
 107         len = FFMAX(p - names, namelen);
 108         if (!av_strncasecmp(name, names, len))
 109             return 1;
 110         names = p + 1;
 111     }
 112     return !av_strcasecmp(name, names);
 113 }
 114
 115 AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
 116                                 const char *mime_type)
 117 {
 118     AVOutputFormat *fmt = NULL, *fmt_found;
 119     int score_max, score;
 120
 121     /* specific test for image sequences */
 122 #if CONFIG_IMAGE2_MUXER
 123     if (!short_name && filename &&
 124         av_filename_number_test(filename) &&
 125         ff_guess_image2_codec(filename) != AV_CODEC_ID_NONE) {
 126         return av_guess_format("image2", NULL, NULL);
 127     }
 128 #endif
 129     /* Find the proper file type. */
 130     fmt_found = NULL;
 131     score_max = 0;
 132     while ((fmt = av_oformat_next(fmt))) {
 133         score = 0;
 134         if (fmt->name && short_name && match_format(short_name, fmt->name))
 135             score += 100;
 136         if (fmt->mime_type && mime_type && !strcmp(fmt->mime_type, mime_type))
 137             score += 10;
 138         if (filename && fmt->extensions &&
 139             av_match_ext(filename, fmt->extensions)) {
 140             score += 5;
 141         }
 142         if (score > score_max) {
 143             score_max = score;
 144             fmt_found = fmt;
 145         }
 146     }
 147     return fmt_found;
 148 }
 149
 150 enum AVCodecID av_guess_codec(AVOutputFormat *fmt, const char *short_name,
 151                               const char *filename, const char *mime_type,
 152                               enum AVMediaType type)
 153 {
 154     if (!strcmp(fmt->name, "segment") || !strcmp(fmt->name, "ssegment")) {
 155         fmt = av_guess_format(NULL, filename, NULL);
 156     }
 157
 158     if (type == AVMEDIA_TYPE_VIDEO) {
 159         enum AVCodecID codec_id = AV_CODEC_ID_NONE;
 160
 161 #if CONFIG_IMAGE2_MUXER
 162         if (!strcmp(fmt->name, "image2") || !strcmp(fmt->name, "image2pipe")) {
 163             codec_id = ff_guess_image2_codec(filename);
 164         }
 165 #endif
 166         if (codec_id == AV_CODEC_ID_NONE)
 167             codec_id = fmt->video_codec;
 168         return codec_id;
 169     } else if (type == AVMEDIA_TYPE_AUDIO)
 170         return fmt->audio_codec;
 171     else if (type == AVMEDIA_TYPE_SUBTITLE)
 172         return fmt->subtitle_codec;
 173     else
 174         return AV_CODEC_ID_NONE;
 175 }
 176
 177 AVInputFormat *av_find_input_format(const char *short_name)
 178 {
 179     AVInputFormat *fmt = NULL;
 180     while ((fmt = av_iformat_next(fmt)))
 181         if (match_format(short_name, fmt->name))
 182             return fmt;
 183     return NULL;
 184 }