ext/ffmpeg/gstffmpegcodecmap.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  * This file:
   4  * Copyright (c) 2002-2004 Ronald Bultje <rbultje@ronald.bitfreak.net>
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Library General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Library General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Library General Public
  17  * License along with this library; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include "config.h"
  24 #endif
  25 #include <gst/gst.h>
  26 #ifdef HAVE_FFMPEG_UNINSTALLED
  27 #include <avcodec.h>
  28 #else
  29 #include <libavcodec/avcodec.h>
  30 #endif
  31 #include <string.h>
  32
  33 #include "gstffmpeg.h"
  34 #include "gstffmpegcodecmap.h"
  35
  36 #include <gst/video/video.h>
  37 #include <gst/audio/audio.h>
  38 #include <gst/pbutils/codec-utils.h>
  39
  40 /*
  41  * Read a palette from a caps.
  42  */
  43
  44 static void
  45 gst_ffmpeg_get_palette (const GstCaps * caps, AVCodecContext * context)
  46 {
  47   GstStructure *str = gst_caps_get_structure (caps, 0);
  48   const GValue *palette_v;
  49   GstBuffer *palette;
  50
  51   /* do we have a palette? */
  52   if ((palette_v = gst_structure_get_value (str, "palette_data")) && context) {
  53     palette = gst_value_get_buffer (palette_v);
  54     if (gst_buffer_get_size (palette) >= AVPALETTE_SIZE) {
  55       if (context->palctrl)
  56         av_free (context->palctrl);
  57       context->palctrl = av_malloc (sizeof (AVPaletteControl));
  58       context->palctrl->palette_changed = 1;
  59       gst_buffer_extract (palette, 0, context->palctrl->palette,
  60           AVPALETTE_SIZE);
  61     }
  62   }
  63 }
  64
  65 static void
  66 gst_ffmpeg_set_palette (GstCaps * caps, AVCodecContext * context)
  67 {
  68   if (context->palctrl) {
  69     GstBuffer *palette = gst_buffer_new_and_alloc (AVPALETTE_SIZE);
  70
  71     gst_buffer_fill (palette, 0, context->palctrl->palette, AVPALETTE_SIZE);
  72     gst_caps_set_simple (caps, "palette_data", GST_TYPE_BUFFER, palette, NULL);
  73   }
  74 }
  75
  76 /* IMPORTANT: Keep this sorted by the ffmpeg channel masks */
  77 static const struct
  78 {
  79   guint64 ff;
  80   GstAudioChannelPosition gst;
  81 } _ff_to_gst_layout[] = {
  82   {
  83   CH_FRONT_LEFT, GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT}, {
  84   CH_FRONT_RIGHT, GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT}, {
  85   CH_FRONT_CENTER, GST_AUDIO_CHANNEL_POSITION_FRONT_CENTER}, {
  86   CH_LOW_FREQUENCY, GST_AUDIO_CHANNEL_POSITION_LFE}, {
  87   CH_BACK_LEFT, GST_AUDIO_CHANNEL_POSITION_REAR_LEFT}, {
  88   CH_BACK_RIGHT, GST_AUDIO_CHANNEL_POSITION_REAR_RIGHT}, {
  89   CH_FRONT_LEFT_OF_CENTER, GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT_OF_CENTER}, {
  90   CH_FRONT_RIGHT_OF_CENTER, GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT_OF_CENTER}, {
  91   CH_BACK_CENTER, GST_AUDIO_CHANNEL_POSITION_REAR_CENTER}, {
  92   CH_SIDE_LEFT, GST_AUDIO_CHANNEL_POSITION_SIDE_LEFT}, {
  93   CH_SIDE_RIGHT, GST_AUDIO_CHANNEL_POSITION_SIDE_RIGHT}, {
  94   CH_TOP_CENTER, GST_AUDIO_CHANNEL_POSITION_NONE}, {
  95   CH_TOP_FRONT_LEFT, GST_AUDIO_CHANNEL_POSITION_NONE}, {
  96   CH_TOP_FRONT_CENTER, GST_AUDIO_CHANNEL_POSITION_NONE}, {
  97   CH_TOP_FRONT_RIGHT, GST_AUDIO_CHANNEL_POSITION_NONE}, {
  98   CH_TOP_BACK_LEFT, GST_AUDIO_CHANNEL_POSITION_NONE}, {
  99   CH_TOP_BACK_CENTER, GST_AUDIO_CHANNEL_POSITION_NONE}, {
 100   CH_TOP_BACK_RIGHT, GST_AUDIO_CHANNEL_POSITION_NONE}, {
 101   CH_STEREO_LEFT, GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT}, {
 102   CH_STEREO_RIGHT, GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT}
 103 };
 104
 105 static GstAudioChannelPosition *
 106 gst_ff_channel_layout_to_gst (guint64 channel_layout, guint channels)
 107 {
 108   guint nchannels = 0, i, j;
 109   GstAudioChannelPosition *pos = NULL;
 110   gboolean none_layout = FALSE;
 111
 112   for (i = 0; i < 64; i++) {
 113     if ((channel_layout & (G_GUINT64_CONSTANT (1) << i)) != 0) {
 114       nchannels++;
 115     }
 116   }
 117
 118   if (channel_layout == 0) {
 119     nchannels = channels;
 120     none_layout = TRUE;
 121   }
 122
 123   if (nchannels != channels) {
 124     GST_ERROR ("Number of channels is different (%u != %u)", channels,
 125         nchannels);
 126     return NULL;
 127   }
 128
 129   pos = g_new (GstAudioChannelPosition, nchannels);
 130
 131   for (i = 0, j = 0; i < G_N_ELEMENTS (_ff_to_gst_layout); i++) {
 132     if ((channel_layout & _ff_to_gst_layout[i].ff) != 0) {
 133       pos[j++] = _ff_to_gst_layout[i].gst;
 134
 135       if (_ff_to_gst_layout[i].gst == GST_AUDIO_CHANNEL_POSITION_NONE)
 136         none_layout = TRUE;
 137     }
 138   }
 139
 140   if (j != nchannels) {
 141     GST_WARNING ("Unknown channels in channel layout - assuming NONE layout");
 142     none_layout = TRUE;
 143   }
 144
 145   if (!none_layout && !gst_audio_check_channel_positions (pos, nchannels)) {
 146     GST_ERROR ("Invalid channel layout %" G_GUINT64_FORMAT
 147         " - assuming NONE layout", channel_layout);
 148     none_layout = TRUE;
 149   }
 150
 151   if (none_layout) {
 152     if (nchannels == 1) {
 153       pos[0] = GST_AUDIO_CHANNEL_POSITION_FRONT_MONO;
 154     } else if (nchannels == 2) {
 155       pos[0] = GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT;
 156       pos[1] = GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT;
 157     } else if (channel_layout == 0) {
 158       g_free (pos);
 159       pos = NULL;
 160     } else {
 161       for (i = 0; i < nchannels; i++)
 162         pos[i] = GST_AUDIO_CHANNEL_POSITION_NONE;
 163     }
 164   }
 165
 166   if (nchannels == 1 && pos[0] == GST_AUDIO_CHANNEL_POSITION_FRONT_CENTER) {
 167     GST_DEBUG ("mono common case; won't set channel positions");
 168     g_free (pos);
 169     pos = NULL;
 170   } else if (nchannels == 2 && pos[0] == GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT
 171       && pos[1] == GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT) {
 172     GST_DEBUG ("stereo common case; won't set channel positions");
 173     g_free (pos);
 174     pos = NULL;
 175   }
 176
 177   return pos;
 178 }
 179
 180 /* this macro makes a caps width fixed or unfixed width/height
 181  * properties depending on whether we've got a context.
 182  *
 183  * See below for why we use this.
 184  *
 185  * We should actually do this stuff at the end, like in riff-media.c,
 186  * but I'm too lazy today. Maybe later.
 187  */
 188 static GstCaps *
 189 gst_ff_vid_caps_new (AVCodecContext * context, enum CodecID codec_id,
 190     const char *mimetype, const char *fieldname, ...)
 191 {
 192   GstStructure *structure = NULL;
 193   GstCaps *caps = NULL;
 194   va_list var_args;
 195   gint i;
 196
 197   GST_LOG ("context:%p, codec_id:%d, mimetype:%s", context, codec_id, mimetype);
 198
 199   /* fixed, non probing context */
 200   if (context != NULL && context->width != -1) {
 201     gint num, denom;
 202
 203     caps = gst_caps_new_simple (mimetype,
 204         "width", G_TYPE_INT, context->width,
 205         "height", G_TYPE_INT, context->height, NULL);
 206
 207     num = context->time_base.den / context->ticks_per_frame;
 208     denom = context->time_base.num;
 209
 210     if (!denom) {
 211       GST_LOG ("invalid framerate: %d/0, -> %d/1", num, num);
 212       denom = 1;
 213     }
 214     if (gst_util_fraction_compare (num, denom, 1000, 1) > 0) {
 215       GST_LOG ("excessive framerate: %d/%d, -> 0/1", num, denom);
 216       num = 0;
 217       denom = 1;
 218     }
 219     GST_LOG ("setting framerate: %d/%d", num, denom);
 220     gst_caps_set_simple (caps,
 221         "framerate", GST_TYPE_FRACTION, num, denom, NULL);
 222   } else {
 223     /* so we are after restricted caps in this case */
 224     switch (codec_id) {
 225       case CODEC_ID_H261:
 226       {
 227         caps = gst_caps_new_simple (mimetype,
 228             "width", G_TYPE_INT, 352,
 229             "height", G_TYPE_INT, 288,
 230             "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL);
 231         gst_caps_append (caps, gst_caps_new_simple (mimetype,
 232                 "width", G_TYPE_INT, 176,
 233                 "height", G_TYPE_INT, 144,
 234                 "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL));
 235         break;
 236       }
 237       case CODEC_ID_H263:
 238       {
 239         /* 128x96, 176x144, 352x288, 704x576, and 1408x1152. slightly reordered
 240          * because we want automatic negotiation to go as close to 320x240 as
 241          * possible. */
 242         const static gint widths[] = { 352, 704, 176, 1408, 128 };
 243         const static gint heights[] = { 288, 576, 144, 1152, 96 };
 244         GstCaps *temp;
 245         gint n_sizes = G_N_ELEMENTS (widths);
 246
 247         caps = gst_caps_new_empty ();
 248         for (i = 0; i < n_sizes; i++) {
 249           temp = gst_caps_new_simple (mimetype,
 250               "width", G_TYPE_INT, widths[i],
 251               "height", G_TYPE_INT, heights[i],
 252               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL);
 253
 254           gst_caps_append (caps, temp);
 255         }
 256         break;
 257       }
 258       case CODEC_ID_DVVIDEO:
 259       {
 260         static struct
 261         {
 262           guint32 csp;
 263           gint width, height;
 264           gint par_n, par_d;
 265           gint framerate_n, framerate_d;
 266         } profiles[] = {
 267           {
 268           GST_MAKE_FOURCC ('Y', '4', '1', 'B'), 720, 480, 10, 11, 30000, 1001}, {
 269           GST_MAKE_FOURCC ('Y', '4', '1', 'B'), 720, 480, 40, 33, 30000, 1001}, {
 270           GST_MAKE_FOURCC ('I', '4', '2', '0'), 720, 576, 59, 54, 25, 1}, {
 271           GST_MAKE_FOURCC ('I', '4', '2', '0'), 720, 576, 118, 81, 25, 1}, {
 272           GST_MAKE_FOURCC ('Y', '4', '1', 'B'), 720, 576, 59, 54, 25, 1}, {
 273           GST_MAKE_FOURCC ('Y', '4', '1', 'B'), 720, 576, 118, 81, 25, 1}
 274         };
 275         GstCaps *temp;
 276         gint n_sizes = G_N_ELEMENTS (profiles);
 277
 278         caps = gst_caps_new_empty ();
 279         for (i = 0; i < n_sizes; i++) {
 280           temp = gst_caps_new_simple (mimetype,
 281               "width", G_TYPE_INT, profiles[i].width,
 282               "height", G_TYPE_INT, profiles[i].height,
 283               "framerate", GST_TYPE_FRACTION, profiles[i].framerate_n,
 284               profiles[i].framerate_d, "pixel-aspect-ratio", GST_TYPE_FRACTION,
 285               profiles[i].par_n, profiles[i].par_d, NULL);
 286
 287           gst_caps_append (caps, temp);
 288         }
 289         break;
 290       }
 291       case CODEC_ID_DNXHD:
 292       {
 293         caps = gst_caps_new_simple (mimetype,
 294             "width", G_TYPE_INT, 1920,
 295             "height", G_TYPE_INT, 1080,
 296             "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL);
 297         gst_caps_append (caps, gst_caps_new_simple (mimetype,
 298                 "width", G_TYPE_INT, 1280,
 299                 "height", G_TYPE_INT, 720,
 300                 "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL));
 301         break;
 302       }
 303       default:
 304         break;
 305     }
 306   }
 307
 308   /* no fixed caps or special restrictions applied;
 309    * default unfixed setting */
 310   if (!caps) {
 311     GST_DEBUG ("Creating default caps");
 312     caps = gst_caps_new_simple (mimetype,
 313         "width", GST_TYPE_INT_RANGE, 16, 4096,
 314         "height", GST_TYPE_INT_RANGE, 16, 4096,
 315         "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL);
 316   }
 317
 318   for (i = 0; i < gst_caps_get_size (caps); i++) {
 319     va_start (var_args, fieldname);
 320     structure = gst_caps_get_structure (caps, i);
 321     gst_structure_set_valist (structure, fieldname, var_args);
 322     va_end (var_args);
 323   }
 324
 325   return caps;
 326 }
 327
 328 /* same for audio - now with channels/sample rate
 329  */
 330 static GstCaps *
 331 gst_ff_aud_caps_new (AVCodecContext * context, enum CodecID codec_id,
 332     const char *mimetype, const char *fieldname, ...)
 333 {
 334   GstCaps *caps = NULL;
 335   GstStructure *structure = NULL;
 336   gint i;
 337   va_list var_args;
 338
 339   /* fixed, non-probing context */
 340   if (context != NULL && context->channels != -1) {
 341     GstAudioChannelPosition *pos;
 342     guint64 channel_layout = context->channel_layout;
 343
 344     if (channel_layout == 0) {
 345       const guint64 default_channel_set[] = {
 346         0, 0, CH_LAYOUT_SURROUND, CH_LAYOUT_QUAD, CH_LAYOUT_5POINT0,
 347         CH_LAYOUT_5POINT1, 0, CH_LAYOUT_7POINT1
 348       };
 349
 350       switch (codec_id) {
 351         case CODEC_ID_EAC3:
 352         case CODEC_ID_AC3:
 353         case CODEC_ID_DTS:
 354           if (context->channels > 0
 355               && context->channels < G_N_ELEMENTS (default_channel_set))
 356             channel_layout = default_channel_set[context->channels - 1];
 357           break;
 358         default:
 359           break;
 360       }
 361     }
 362
 363     caps = gst_caps_new_simple (mimetype,
 364         "rate", G_TYPE_INT, context->sample_rate,
 365         "channels", G_TYPE_INT, context->channels, NULL);
 366
 367     pos = gst_ff_channel_layout_to_gst (channel_layout, context->channels);
 368     if (pos != NULL) {
 369       gst_audio_set_channel_positions (gst_caps_get_structure (caps, 0), pos);
 370       g_free (pos);
 371     }
 372   } else {
 373     gint maxchannels = 2;
 374     const gint *rates = NULL;
 375     gint n_rates = 0;
 376
 377     if (context) {
 378       /* so we must be after restricted caps in this particular case */
 379       switch (codec_id) {
 380         case CODEC_ID_MP2:
 381         {
 382           const static gint l_rates[] =
 383               { 48000, 44100, 32000, 24000, 22050, 16000 };
 384           n_rates = G_N_ELEMENTS (l_rates);
 385           rates = l_rates;
 386           break;
 387         }
 388         case CODEC_ID_EAC3:
 389         case CODEC_ID_AC3:
 390         {
 391           const static gint l_rates[] = { 48000, 44100, 32000 };
 392           n_rates = G_N_ELEMENTS (l_rates);
 393           rates = l_rates;
 394           break;
 395         }
 396         case CODEC_ID_ADPCM_SWF:
 397         {
 398           const static gint l_rates[] = { 11025, 22050, 44100 };
 399           n_rates = G_N_ELEMENTS (l_rates);
 400           rates = l_rates;
 401           break;
 402         }
 403         case CODEC_ID_ROQ_DPCM:
 404         {
 405           const static gint l_rates[] = { 22050 };
 406           n_rates = G_N_ELEMENTS (l_rates);
 407           rates = l_rates;
 408           break;
 409         }
 410         case CODEC_ID_ADPCM_G726:
 411           maxchannels = 1;
 412           break;
 413         case CODEC_ID_AMR_NB:
 414         {
 415           const static gint l_rates[] = { 8000 };
 416           maxchannels = 1;
 417           n_rates = G_N_ELEMENTS (l_rates);
 418           rates = l_rates;
 419           break;
 420         }
 421         case CODEC_ID_AMR_WB:
 422         {
 423           const static gint l_rates[] = { 16000 };
 424           maxchannels = 1;
 425           n_rates = G_N_ELEMENTS (l_rates);
 426           rates = l_rates;
 427           break;
 428         }
 429         default:
 430           break;
 431       }
 432
 433       /* TODO: handle context->channel_layouts here to set
 434        * the list of channel layouts supported by the encoder.
 435        * Unfortunately no encoder uses this yet....
 436        */
 437     }
 438 #if 0
 439     /* regardless of encode/decode, open up channels if applicable */
 440     /* Until decoders/encoders expose the maximum number of channels
 441      * they support, we whitelist them here. */
 442     switch (codec_id) {
 443       case CODEC_ID_AC3:
 444       case CODEC_ID_EAC3:
 445       case CODEC_ID_AAC:
 446       case CODEC_ID_AAC_LATM:
 447       case CODEC_ID_DTS:
 448         maxchannels = 6;
 449         break;
 450       case CODEC_ID_WMAPRO:
 451       case CODEC_ID_TRUEHD:
 452         maxchannels = 8;
 453         break;
 454       default:
 455         break;
 456     }
 457 #endif
 458
 459     if (maxchannels == 1)
 460       caps = gst_caps_new_simple (mimetype,
 461           "channels", G_TYPE_INT, maxchannels, NULL);
 462     else
 463       caps = gst_caps_new_simple (mimetype,
 464           "channels", GST_TYPE_INT_RANGE, 1, maxchannels, NULL);
 465     if (n_rates) {
 466       GValue list = { 0, };
 467       GstStructure *structure;
 468
 469       g_value_init (&list, GST_TYPE_LIST);
 470       for (i = 0; i < n_rates; i++) {
 471         GValue v = { 0, };
 472
 473         g_value_init (&v, G_TYPE_INT);
 474         g_value_set_int (&v, rates[i]);
 475         gst_value_list_append_value (&list, &v);
 476         g_value_unset (&v);
 477       }
 478       structure = gst_caps_get_structure (caps, 0);
 479       gst_structure_set_value (structure, "rate", &list);
 480       g_value_unset (&list);
 481     } else
 482       gst_caps_set_simple (caps, "rate", GST_TYPE_INT_RANGE, 4000, 96000, NULL);
 483   }
 484
 485   for (i = 0; i < gst_caps_get_size (caps); i++) {
 486     va_start (var_args, fieldname);
 487     structure = gst_caps_get_structure (caps, i);
 488     gst_structure_set_valist (structure, fieldname, var_args);
 489     va_end (var_args);
 490   }
 491
 492   return caps;
 493 }
 494
 495 /* Convert a FFMPEG codec ID and optional AVCodecContext
 496  * to a GstCaps. If the context is ommitted, no fixed values
 497  * for video/audio size will be included in the GstCaps
 498  *
 499  * CodecID is primarily meant for compressed data GstCaps!
 500  *
 501  * encode is a special parameter. gstffmpegdec will say
 502  * FALSE, gstffmpegenc will say TRUE. The output caps
 503  * depends on this, in such a way that it will be very
 504  * specific, defined, fixed and correct caps for encoders,
 505  * yet very wide, "forgiving" caps for decoders. Example
 506  * for mp3: decode: audio/mpeg,mpegversion=1,layer=[1-3]
 507  * but encode: audio/mpeg,mpegversion=1,layer=3,bitrate=x,
 508  * rate=x,channels=x.
 509  */
 510
 511 GstCaps *
 512 gst_ffmpeg_codecid_to_caps (enum CodecID codec_id,
 513     AVCodecContext * context, gboolean encode)
 514 {
 515   GstCaps *caps = NULL;
 516   gboolean buildcaps = FALSE;
 517
 518   GST_LOG ("codec_id:%d, context:%p, encode:%d", codec_id, context, encode);
 519
 520   switch (codec_id) {
 521     case CODEC_ID_MPEG1VIDEO:
 522       /* FIXME: bitrate */
 523       caps = gst_ff_vid_caps_new (context, codec_id, "video/mpeg",
 524           "mpegversion", G_TYPE_INT, 1,
 525           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 526       break;
 527
 528     case CODEC_ID_MPEG2VIDEO:
 529       if (encode) {
 530         /* FIXME: bitrate */
 531         caps = gst_ff_vid_caps_new (context, codec_id, "video/mpeg",
 532             "mpegversion", G_TYPE_INT, 2,
 533             "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 534       } else {
 535         /* decode both MPEG-1 and MPEG-2; width/height/fps are all in
 536          * the MPEG video stream headers, so may be omitted from caps. */
 537         caps = gst_caps_new_simple ("video/mpeg",
 538             "mpegversion", GST_TYPE_INT_RANGE, 1, 2,
 539             "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 540       }
 541       break;
 542
 543     case CODEC_ID_MPEG2VIDEO_XVMC:
 544       /* this is a special ID - don't need it in GStreamer, I think */
 545       break;
 546
 547     case CODEC_ID_H263:
 548       if (encode) {
 549         caps = gst_ff_vid_caps_new (context, codec_id, "video/x-h263",
 550             "variant", G_TYPE_STRING, "itu",
 551             "h263version", G_TYPE_STRING, "h263", NULL);
 552       } else {
 553         /* don't pass codec_id, we can decode other variants with the H263
 554          * decoder that don't have specific size requirements
 555          */
 556         caps = gst_ff_vid_caps_new (context, CODEC_ID_NONE, "video/x-h263",
 557             "variant", G_TYPE_STRING, "itu", NULL);
 558       }
 559       break;
 560
 561     case CODEC_ID_H263P:
 562       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-h263",
 563           "variant", G_TYPE_STRING, "itu",
 564           "h263version", G_TYPE_STRING, "h263p", NULL);
 565       if (encode && context) {
 566
 567         gst_caps_set_simple (caps,
 568             "annex-f", G_TYPE_BOOLEAN, context->flags & CODEC_FLAG_4MV,
 569             "annex-j", G_TYPE_BOOLEAN, context->flags & CODEC_FLAG_LOOP_FILTER,
 570             "annex-i", G_TYPE_BOOLEAN, context->flags & CODEC_FLAG_AC_PRED,
 571             "annex-t", G_TYPE_BOOLEAN, context->flags & CODEC_FLAG_AC_PRED,
 572             NULL);
 573       }
 574       break;
 575
 576     case CODEC_ID_H263I:
 577       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-intel-h263",
 578           "variant", G_TYPE_STRING, "intel", NULL);
 579       break;
 580
 581     case CODEC_ID_H261:
 582       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-h261", NULL);
 583       break;
 584
 585     case CODEC_ID_RV10:
 586     case CODEC_ID_RV20:
 587     case CODEC_ID_RV30:
 588     case CODEC_ID_RV40:
 589     {
 590       gint version;
 591
 592       switch (codec_id) {
 593         case CODEC_ID_RV40:
 594           version = 4;
 595           break;
 596         case CODEC_ID_RV30:
 597           version = 3;
 598           break;
 599         case CODEC_ID_RV20:
 600           version = 2;
 601           break;
 602         default:
 603           version = 1;
 604           break;
 605       }
 606
 607       /* FIXME: context->sub_id must be filled in during decoding */
 608       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-pn-realvideo",
 609           "systemstream", G_TYPE_BOOLEAN, FALSE,
 610           "rmversion", G_TYPE_INT, version, NULL);
 611       if (context) {
 612         gst_caps_set_simple (caps, "format", G_TYPE_INT, context->sub_id, NULL);
 613         if (context->extradata_size >= 8) {
 614           gst_caps_set_simple (caps,
 615               "subformat", G_TYPE_INT, GST_READ_UINT32_BE (context->extradata),
 616               NULL);
 617         }
 618       }
 619     }
 620       break;
 621
 622     case CODEC_ID_MP1:
 623       /* FIXME: bitrate */
 624       caps = gst_ff_aud_caps_new (context, codec_id, "audio/mpeg",
 625           "mpegversion", G_TYPE_INT, 1, "layer", G_TYPE_INT, 1, NULL);
 626       break;
 627
 628     case CODEC_ID_MP2:
 629       /* FIXME: bitrate */
 630       caps = gst_ff_aud_caps_new (context, codec_id, "audio/mpeg",
 631           "mpegversion", G_TYPE_INT, 1, "layer", G_TYPE_INT, 2, NULL);
 632       break;
 633
 634     case CODEC_ID_MP3:
 635       if (encode) {
 636         /* FIXME: bitrate */
 637         caps = gst_ff_aud_caps_new (context, codec_id, "audio/mpeg",
 638             "mpegversion", G_TYPE_INT, 1, "layer", G_TYPE_INT, 3, NULL);
 639       } else {
 640         /* Decodes MPEG-1 layer 1/2/3. Samplerate, channels et al are
 641          * in the MPEG audio header, so may be omitted from caps. */
 642         caps = gst_caps_new_simple ("audio/mpeg",
 643             "mpegversion", G_TYPE_INT, 1,
 644             "layer", GST_TYPE_INT_RANGE, 1, 3, NULL);
 645       }
 646       break;
 647
 648     case CODEC_ID_MUSEPACK7:
 649       caps =
 650           gst_ff_aud_caps_new (context, codec_id,
 651           "audio/x-ffmpeg-parsed-musepack", "streamversion", G_TYPE_INT, 7,
 652           NULL);
 653       break;
 654
 655     case CODEC_ID_MUSEPACK8:
 656       caps =
 657           gst_ff_aud_caps_new (context, codec_id,
 658           "audio/x-ffmpeg-parsed-musepack", "streamversion", G_TYPE_INT, 8,
 659           NULL);
 660       break;
 661
 662     case CODEC_ID_AC3:
 663       /* FIXME: bitrate */
 664       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-ac3", NULL);
 665       break;
 666
 667     case CODEC_ID_EAC3:
 668       /* FIXME: bitrate */
 669       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-eac3", NULL);
 670       break;
 671
 672     case CODEC_ID_TRUEHD:
 673       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-true-hd", NULL);
 674       break;
 675
 676     case CODEC_ID_ATRAC1:
 677       caps =
 678           gst_ff_aud_caps_new (context, codec_id, "audio/x-vnd.sony.atrac1",
 679           NULL);
 680       break;
 681
 682     case CODEC_ID_ATRAC3:
 683       caps =
 684           gst_ff_aud_caps_new (context, codec_id, "audio/x-vnd.sony.atrac3",
 685           NULL);
 686       break;
 687
 688     case CODEC_ID_DTS:
 689       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-dts", NULL);
 690       break;
 691
 692     case CODEC_ID_APE:
 693       caps =
 694           gst_ff_aud_caps_new (context, codec_id, "audio/x-ffmpeg-parsed-ape",
 695           NULL);
 696       if (context) {
 697         gst_caps_set_simple (caps,
 698             "depth", G_TYPE_INT, context->bits_per_coded_sample, NULL);
 699       }
 700       break;
 701
 702     case CODEC_ID_MLP:
 703       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-mlp", NULL);
 704       break;
 705
 706     case CODEC_ID_IMC:
 707       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-imc", NULL);
 708       break;
 709
 710       /* MJPEG is normal JPEG, Motion-JPEG and Quicktime MJPEG-A. MJPEGB
 711        * is Quicktime's MJPEG-B. LJPEG is lossless JPEG. I don't know what
 712        * sp5x is, but it's apparently something JPEG... We don't separate
 713        * between those in GStreamer. Should we (at least between MJPEG,
 714        * MJPEG-B and sp5x decoding...)? */
 715     case CODEC_ID_MJPEG:
 716     case CODEC_ID_LJPEG:
 717       caps = gst_ff_vid_caps_new (context, codec_id, "image/jpeg", NULL);
 718       break;
 719
 720     case CODEC_ID_SP5X:
 721       caps = gst_ff_vid_caps_new (context, codec_id, "video/sp5x", NULL);
 722       break;
 723
 724     case CODEC_ID_MJPEGB:
 725       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-mjpeg-b", NULL);
 726       break;
 727
 728     case CODEC_ID_MPEG4:
 729       if (encode && context != NULL) {
 730         /* I'm not exactly sure what ffmpeg outputs... ffmpeg itself uses
 731          * the AVI fourcc 'DIVX', but 'mp4v' for Quicktime... */
 732         switch (context->codec_tag) {
 733           case GST_MAKE_FOURCC ('D', 'I', 'V', 'X'):
 734             caps = gst_ff_vid_caps_new (context, codec_id, "video/x-divx",
 735                 "divxversion", G_TYPE_INT, 5, NULL);
 736             break;
 737           case GST_MAKE_FOURCC ('m', 'p', '4', 'v'):
 738           default:
 739             /* FIXME: bitrate */
 740             caps = gst_ff_vid_caps_new (context, codec_id, "video/mpeg",
 741                 "systemstream", G_TYPE_BOOLEAN, FALSE,
 742                 "mpegversion", G_TYPE_INT, 4, NULL);
 743             break;
 744         }
 745       } else {
 746         /* The trick here is to separate xvid, divx, mpeg4, 3ivx et al */
 747         caps = gst_ff_vid_caps_new (context, codec_id, "video/mpeg",
 748             "mpegversion", G_TYPE_INT, 4,
 749             "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 750         if (encode) {
 751           gst_caps_append (caps, gst_ff_vid_caps_new (context, codec_id,
 752                   "video/x-divx", "divxversion", G_TYPE_INT, 5, NULL));
 753         } else {
 754           gst_caps_append (caps, gst_ff_vid_caps_new (context, codec_id,
 755                   "video/x-divx", "divxversion", GST_TYPE_INT_RANGE, 4, 5,
 756                   NULL));
 757           gst_caps_append (caps, gst_ff_vid_caps_new (context, codec_id,
 758                   "video/x-xvid", NULL));
 759           gst_caps_append (caps, gst_ff_vid_caps_new (context, codec_id,
 760                   "video/x-3ivx", NULL));
 761         }
 762       }
 763       break;
 764
 765     case CODEC_ID_RAWVIDEO:
 766       caps =
 767           gst_ffmpeg_codectype_to_caps (AVMEDIA_TYPE_VIDEO, context, codec_id,
 768           encode);
 769       break;
 770
 771     case CODEC_ID_MSMPEG4V1:
 772     case CODEC_ID_MSMPEG4V2:
 773     case CODEC_ID_MSMPEG4V3:
 774     {
 775       gint version = 41 + codec_id - CODEC_ID_MSMPEG4V1;
 776
 777       /* encode-FIXME: bitrate */
 778       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-msmpeg",
 779           "msmpegversion", G_TYPE_INT, version, NULL);
 780       if (!encode && codec_id == CODEC_ID_MSMPEG4V3) {
 781         gst_caps_append (caps, gst_ff_vid_caps_new (context, codec_id,
 782                 "video/x-divx", "divxversion", G_TYPE_INT, 3, NULL));
 783       }
 784     }
 785       break;
 786
 787     case CODEC_ID_WMV1:
 788     case CODEC_ID_WMV2:
 789     {
 790       gint version = (codec_id == CODEC_ID_WMV1) ? 1 : 2;
 791
 792       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-wmv",
 793           "wmvversion", G_TYPE_INT, version, NULL);
 794     }
 795       break;
 796
 797     case CODEC_ID_FLV1:
 798       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-flash-video",
 799           "flvversion", G_TYPE_INT, 1, NULL);
 800       break;
 801
 802     case CODEC_ID_SVQ1:
 803       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-svq",
 804           "svqversion", G_TYPE_INT, 1, NULL);
 805       break;
 806
 807     case CODEC_ID_SVQ3:
 808       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-svq",
 809           "svqversion", G_TYPE_INT, 3, NULL);
 810       break;
 811
 812     case CODEC_ID_DVAUDIO:
 813       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-dv", NULL);
 814       break;
 815
 816     case CODEC_ID_DVVIDEO:
 817     {
 818       if (encode && context) {
 819         const gchar *format;
 820
 821         switch (context->pix_fmt) {
 822           case PIX_FMT_YUYV422:
 823             format = "YUY2";
 824             break;
 825           case PIX_FMT_YUV420P:
 826             format = "I420";
 827             break;
 828           case PIX_FMT_YUVA420P:
 829             format = "A420";
 830             break;
 831           case PIX_FMT_YUV411P:
 832             format = "Y41B";
 833             break;
 834           case PIX_FMT_YUV422P:
 835             format = "Y42B";
 836             break;
 837           case PIX_FMT_YUV410P:
 838             format = "YUV9";
 839             break;
 840           default:
 841             GST_WARNING
 842                 ("Couldnt' find format for pixfmt %d, defaulting to I420",
 843                 context->pix_fmt);
 844             format = "I420";
 845             break;
 846         }
 847         caps = gst_ff_vid_caps_new (context, codec_id, "video/x-dv",
 848             "systemstream", G_TYPE_BOOLEAN, FALSE,
 849             "format", G_TYPE_STRING, format, NULL);
 850       } else {
 851         caps = gst_ff_vid_caps_new (context, codec_id, "video/x-dv",
 852             "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 853       }
 854     }
 855       break;
 856
 857     case CODEC_ID_WMAV1:
 858     case CODEC_ID_WMAV2:
 859     {
 860       gint version = (codec_id == CODEC_ID_WMAV1) ? 1 : 2;
 861
 862       if (context) {
 863         caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-wma",
 864             "wmaversion", G_TYPE_INT, version,
 865             "block_align", G_TYPE_INT, context->block_align,
 866             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
 867       } else {
 868         caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-wma",
 869             "wmaversion", G_TYPE_INT, version,
 870             "block_align", GST_TYPE_INT_RANGE, 0, G_MAXINT,
 871             "bitrate", GST_TYPE_INT_RANGE, 0, G_MAXINT, NULL);
 872       }
 873     }
 874       break;
 875     case CODEC_ID_WMAPRO:
 876     {
 877       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-wma",
 878           "wmaversion", G_TYPE_INT, 3, NULL);
 879       break;
 880     }
 881
 882     case CODEC_ID_WMAVOICE:
 883     {
 884       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-wms", NULL);
 885       break;
 886     }
 887
 888     case CODEC_ID_MACE3:
 889     case CODEC_ID_MACE6:
 890     {
 891       gint version = (codec_id == CODEC_ID_MACE3) ? 3 : 6;
 892
 893       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-mace",
 894           "maceversion", G_TYPE_INT, version, NULL);
 895     }
 896       break;
 897
 898     case CODEC_ID_HUFFYUV:
 899       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-huffyuv", NULL);
 900       if (context) {
 901         gst_caps_set_simple (caps,
 902             "bpp", G_TYPE_INT, context->bits_per_coded_sample, NULL);
 903       }
 904       break;
 905
 906     case CODEC_ID_CYUV:
 907       caps =
 908           gst_ff_vid_caps_new (context, codec_id, "video/x-compressed-yuv",
 909           NULL);
 910       break;
 911
 912     case CODEC_ID_H264:
 913       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-h264", NULL);
 914       break;
 915
 916     case CODEC_ID_INDEO5:
 917       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-indeo",
 918           "indeoversion", G_TYPE_INT, 5, NULL);
 919       break;
 920
 921     case CODEC_ID_INDEO3:
 922       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-indeo",
 923           "indeoversion", G_TYPE_INT, 3, NULL);
 924       break;
 925
 926     case CODEC_ID_INDEO2:
 927       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-indeo",
 928           "indeoversion", G_TYPE_INT, 2, NULL);
 929       break;
 930
 931     case CODEC_ID_FLASHSV:
 932       caps =
 933           gst_ff_vid_caps_new (context, codec_id, "video/x-flash-screen", NULL);
 934       break;
 935
 936     case CODEC_ID_VP3:
 937       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-vp3", NULL);
 938       break;
 939
 940     case CODEC_ID_VP5:
 941       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-vp5", NULL);
 942       break;
 943
 944     case CODEC_ID_VP6:
 945       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-vp6", NULL);
 946       break;
 947
 948     case CODEC_ID_VP6F:
 949       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-vp6-flash", NULL);
 950       break;
 951
 952     case CODEC_ID_VP6A:
 953       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-vp6-alpha", NULL);
 954       break;
 955
 956     case CODEC_ID_VP8:
 957       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-vp8", NULL);
 958       break;
 959
 960     case CODEC_ID_THEORA:
 961       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-theora", NULL);
 962       break;
 963
 964     case CODEC_ID_AAC:
 965     {
 966       caps = gst_ff_aud_caps_new (context, codec_id, "audio/mpeg", NULL);
 967
 968       if (!encode) {
 969         GValue arr = { 0, };
 970         GValue item = { 0, };
 971
 972         g_value_init (&arr, GST_TYPE_LIST);
 973         g_value_init (&item, G_TYPE_INT);
 974         g_value_set_int (&item, 2);
 975         gst_value_list_append_value (&arr, &item);
 976         g_value_set_int (&item, 4);
 977         gst_value_list_append_value (&arr, &item);
 978         g_value_unset (&item);
 979
 980         gst_caps_set_value (caps, "mpegversion", &arr);
 981         g_value_unset (&arr);
 982
 983         g_value_init (&arr, GST_TYPE_LIST);
 984         g_value_init (&item, G_TYPE_STRING);
 985         g_value_set_string (&item, "raw");
 986         gst_value_list_append_value (&arr, &item);
 987         g_value_set_string (&item, "adts");
 988         gst_value_list_append_value (&arr, &item);
 989         g_value_set_string (&item, "adif");
 990         gst_value_list_append_value (&arr, &item);
 991         g_value_unset (&item);
 992
 993         gst_caps_set_value (caps, "stream-format", &arr);
 994         g_value_unset (&arr);
 995       } else {
 996         gst_caps_set_simple (caps, "mpegversion", G_TYPE_INT, 4,
 997             "stream-format", G_TYPE_STRING, "raw",
 998             "base-profile", G_TYPE_STRING, "lc", NULL);
 999
1000         if (context && context->extradata_size > 0)
1001           gst_codec_utils_aac_caps_set_level_and_profile (caps,
1002               context->extradata, context->extradata_size);
1003       }
1004
1005       break;
1006     }
1007     case CODEC_ID_AAC_LATM:    /* LATM/LOAS AAC syntax */
1008       caps = gst_ff_aud_caps_new (context, codec_id, "audio/mpeg",
1009           "mpegversion", G_TYPE_INT, 4, "stream-format", G_TYPE_STRING, "loas",
1010           NULL);
1011       break;
1012
1013     case CODEC_ID_ASV1:
1014       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-asus",
1015           "asusversion", G_TYPE_INT, 1, NULL);
1016       break;
1017     case CODEC_ID_ASV2:
1018       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-asus",
1019           "asusversion", G_TYPE_INT, 2, NULL);
1020       break;
1021
1022     case CODEC_ID_FFV1:
1023       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-ffv",
1024           "ffvversion", G_TYPE_INT, 1, NULL);
1025       break;
1026
1027     case CODEC_ID_4XM:
1028       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-4xm", NULL);
1029       break;
1030
1031     case CODEC_ID_XAN_WC3:
1032     case CODEC_ID_XAN_WC4:
1033       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-xan",
1034           "wcversion", G_TYPE_INT, 3 - CODEC_ID_XAN_WC3 + codec_id, NULL);
1035       break;
1036
1037     case CODEC_ID_CLJR:
1038       caps =
1039           gst_ff_vid_caps_new (context, codec_id,
1040           "video/x-cirrus-logic-accupak", NULL);
1041       break;
1042
1043     case CODEC_ID_FRAPS:
1044       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-fraps", NULL);
1045       break;
1046
1047     case CODEC_ID_MDEC:
1048     case CODEC_ID_ROQ:
1049     case CODEC_ID_INTERPLAY_VIDEO:
1050       buildcaps = TRUE;
1051       break;
1052
1053     case CODEC_ID_VCR1:
1054       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-ati-vcr",
1055           "vcrversion", G_TYPE_INT, 1, NULL);
1056       break;
1057
1058     case CODEC_ID_RPZA:
1059       caps =
1060           gst_ff_vid_caps_new (context, codec_id, "video/x-apple-video", NULL);
1061       break;
1062
1063     case CODEC_ID_CINEPAK:
1064       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-cinepak", NULL);
1065       break;
1066
1067       /* WS_VQA belogns here (order) */
1068
1069     case CODEC_ID_MSRLE:
1070       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-rle",
1071           "layout", G_TYPE_STRING, "microsoft", NULL);
1072       if (context) {
1073         gst_caps_set_simple (caps,
1074             "depth", G_TYPE_INT, (gint) context->bits_per_coded_sample, NULL);
1075       } else {
1076         gst_caps_set_simple (caps, "depth", GST_TYPE_INT_RANGE, 1, 64, NULL);
1077       }
1078       break;
1079
1080     case CODEC_ID_QTRLE:
1081       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-rle",
1082           "layout", G_TYPE_STRING, "quicktime", NULL);
1083       if (context) {
1084         gst_caps_set_simple (caps,
1085             "depth", G_TYPE_INT, (gint) context->bits_per_coded_sample, NULL);
1086       } else {
1087         gst_caps_set_simple (caps, "depth", GST_TYPE_INT_RANGE, 1, 64, NULL);
1088       }
1089       break;
1090
1091     case CODEC_ID_MSVIDEO1:
1092       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-msvideocodec",
1093           "msvideoversion", G_TYPE_INT, 1, NULL);
1094       break;
1095
1096     case CODEC_ID_WMV3:
1097       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-wmv",
1098           "wmvversion", G_TYPE_INT, 3, NULL);
1099       break;
1100     case CODEC_ID_VC1:
1101       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-wmv",
1102           "wmvversion", G_TYPE_INT, 3, "format", G_TYPE_STRING, "WVC1", NULL);
1103       break;
1104     case CODEC_ID_QDM2:
1105       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-qdm2", NULL);
1106       break;
1107
1108     case CODEC_ID_MSZH:
1109       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-mszh", NULL);
1110       break;
1111
1112     case CODEC_ID_ZLIB:
1113       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-zlib", NULL);
1114       break;
1115
1116     case CODEC_ID_TRUEMOTION1:
1117       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-truemotion",
1118           "trueversion", G_TYPE_INT, 1, NULL);
1119       break;
1120     case CODEC_ID_TRUEMOTION2:
1121       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-truemotion",
1122           "trueversion", G_TYPE_INT, 2, NULL);
1123       break;
1124
1125     case CODEC_ID_ULTI:
1126       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-ultimotion",
1127           NULL);
1128       break;
1129
1130     case CODEC_ID_TSCC:
1131       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-camtasia", NULL);
1132       if (context) {
1133         gst_caps_set_simple (caps,
1134             "depth", G_TYPE_INT, (gint) context->bits_per_coded_sample, NULL);
1135       } else {
1136         gst_caps_set_simple (caps, "depth", GST_TYPE_INT_RANGE, 8, 32, NULL);
1137       }
1138       break;
1139
1140     case CODEC_ID_KMVC:
1141       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-kmvc", NULL);
1142       break;
1143
1144     case CODEC_ID_NUV:
1145       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-nuv", NULL);
1146       break;
1147
1148     case CODEC_ID_GIF:
1149       caps = gst_ff_vid_caps_new (context, codec_id, "image/gif", NULL);
1150       break;
1151
1152     case CODEC_ID_PNG:
1153       caps = gst_ff_vid_caps_new (context, codec_id, "image/png", NULL);
1154       break;
1155
1156     case CODEC_ID_PPM:
1157       caps = gst_ff_vid_caps_new (context, codec_id, "image/ppm", NULL);
1158       break;
1159
1160     case CODEC_ID_PBM:
1161       caps = gst_ff_vid_caps_new (context, codec_id, "image/pbm", NULL);
1162       break;
1163
1164     case CODEC_ID_PAM:
1165       caps =
1166           gst_ff_vid_caps_new (context, codec_id, "image/x-portable-anymap",
1167           NULL);
1168       break;
1169
1170     case CODEC_ID_PGM:
1171       caps =
1172           gst_ff_vid_caps_new (context, codec_id, "image/x-portable-graymap",
1173           NULL);
1174       break;
1175
1176     case CODEC_ID_PCX:
1177       caps = gst_ff_vid_caps_new (context, codec_id, "image/x-pcx", NULL);
1178       break;
1179
1180     case CODEC_ID_SGI:
1181       caps = gst_ff_vid_caps_new (context, codec_id, "image/x-sgi", NULL);
1182       break;
1183
1184     case CODEC_ID_TARGA:
1185       caps = gst_ff_vid_caps_new (context, codec_id, "image/x-tga", NULL);
1186       break;
1187
1188     case CODEC_ID_TIFF:
1189       caps = gst_ff_vid_caps_new (context, codec_id, "image/tiff", NULL);
1190       break;
1191
1192     case CODEC_ID_SUNRAST:
1193       caps =
1194           gst_ff_vid_caps_new (context, codec_id, "image/x-sun-raster", NULL);
1195       break;
1196
1197     case CODEC_ID_SMC:
1198       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-smc", NULL);
1199       break;
1200
1201     case CODEC_ID_QDRAW:
1202       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-qdrw", NULL);
1203       break;
1204
1205     case CODEC_ID_DNXHD:
1206       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-dnxhd", NULL);
1207       break;
1208
1209     case CODEC_ID_MIMIC:
1210       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-mimic", NULL);
1211       break;
1212
1213     case CODEC_ID_VMNC:
1214       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-vmnc", NULL);
1215       break;
1216
1217     case CODEC_ID_TRUESPEECH:
1218       caps =
1219           gst_ff_aud_caps_new (context, codec_id, "audio/x-truespeech", NULL);
1220       break;
1221
1222     case CODEC_ID_QCELP:
1223       caps = gst_ff_aud_caps_new (context, codec_id, "audio/qcelp", NULL);
1224       break;
1225
1226     case CODEC_ID_AMV:
1227       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-amv", NULL);
1228       break;
1229
1230     case CODEC_ID_AASC:
1231       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-aasc", NULL);
1232       break;
1233
1234     case CODEC_ID_LOCO:
1235       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-loco", NULL);
1236       break;
1237
1238     case CODEC_ID_ZMBV:
1239       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-zmbv", NULL);
1240       break;
1241
1242     case CODEC_ID_LAGARITH:
1243       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-lagarith", NULL);
1244       break;
1245
1246     case CODEC_ID_CSCD:
1247       caps = gst_ff_vid_caps_new (context, codec_id, "video/x-camstudio", NULL);
1248       if (context) {
1249         gst_caps_set_simple (caps,
1250             "depth", G_TYPE_INT, (gint) context->bits_per_coded_sample, NULL);
1251       } else {
1252         gst_caps_set_simple (caps, "depth", GST_TYPE_INT_RANGE, 8, 32, NULL);
1253       }
1254       break;
1255
1256     case CODEC_ID_WS_VQA:
1257     case CODEC_ID_IDCIN:
1258     case CODEC_ID_8BPS:
1259     case CODEC_ID_FLIC:
1260     case CODEC_ID_VMDVIDEO:
1261     case CODEC_ID_VMDAUDIO:
1262     case CODEC_ID_SONIC:
1263     case CODEC_ID_SONIC_LS:
1264     case CODEC_ID_SNOW:
1265     case CODEC_ID_VIXL:
1266     case CODEC_ID_QPEG:
1267     case CODEC_ID_PGMYUV:
1268     case CODEC_ID_FFVHUFF:
1269     case CODEC_ID_WNV1:
1270     case CODEC_ID_MP3ADU:
1271     case CODEC_ID_MP3ON4:
1272     case CODEC_ID_WESTWOOD_SND1:
1273     case CODEC_ID_MMVIDEO:
1274     case CODEC_ID_AVS:
1275     case CODEC_ID_CAVS:
1276       buildcaps = TRUE;
1277       break;
1278
1279       /* weird quasi-codecs for the demuxers only */
1280     case CODEC_ID_PCM_S16LE:
1281     case CODEC_ID_PCM_S16BE:
1282     case CODEC_ID_PCM_U16LE:
1283     case CODEC_ID_PCM_U16BE:
1284     case CODEC_ID_PCM_S8:
1285     case CODEC_ID_PCM_U8:
1286     {
1287       GstAudioFormat format;
1288
1289       switch (codec_id) {
1290         case CODEC_ID_PCM_S16LE:
1291           format = GST_AUDIO_FORMAT_S16LE;
1292           break;
1293         case CODEC_ID_PCM_S16BE:
1294           format = GST_AUDIO_FORMAT_S16BE;
1295           break;
1296         case CODEC_ID_PCM_U16LE:
1297           format = GST_AUDIO_FORMAT_U16LE;
1298           break;
1299         case CODEC_ID_PCM_U16BE:
1300           format = GST_AUDIO_FORMAT_U16BE;
1301           break;
1302         case CODEC_ID_PCM_S8:
1303           format = GST_AUDIO_FORMAT_S8;
1304           break;
1305         case CODEC_ID_PCM_U8:
1306           format = GST_AUDIO_FORMAT_U8;
1307           break;
1308         default:
1309           g_assert (0);         /* don't worry, we never get here */
1310           break;
1311       }
1312
1313       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-raw",
1314           "format", G_TYPE_STRING, gst_audio_format_to_string (format), NULL);
1315     }
1316       break;
1317
1318     case CODEC_ID_PCM_MULAW:
1319       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-mulaw", NULL);
1320       break;
1321
1322     case CODEC_ID_PCM_ALAW:
1323       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-alaw", NULL);
1324       break;
1325
1326     case CODEC_ID_ADPCM_G722:
1327       caps = gst_ff_aud_caps_new (context, codec_id, "audio/G722", NULL);
1328       if (context)
1329         gst_caps_set_simple (caps,
1330             "block_align", G_TYPE_INT, context->block_align,
1331             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1332       break;
1333
1334     case CODEC_ID_ADPCM_G726:
1335     {
1336       /* the G726 decoder can also handle G721 */
1337       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-adpcm",
1338           "layout", G_TYPE_STRING, "g726", NULL);
1339       if (context)
1340         gst_caps_set_simple (caps,
1341             "block_align", G_TYPE_INT, context->block_align,
1342             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1343
1344       if (!encode) {
1345         gst_caps_append (caps, gst_caps_new_simple ("audio/x-adpcm",
1346                 "layout", G_TYPE_STRING, "g721",
1347                 "channels", G_TYPE_INT, 1, "rate", G_TYPE_INT, 8000, NULL));
1348       }
1349       break;
1350     }
1351     case CODEC_ID_ADPCM_IMA_QT:
1352     case CODEC_ID_ADPCM_IMA_WAV:
1353     case CODEC_ID_ADPCM_IMA_DK3:
1354     case CODEC_ID_ADPCM_IMA_DK4:
1355     case CODEC_ID_ADPCM_IMA_WS:
1356     case CODEC_ID_ADPCM_IMA_SMJPEG:
1357     case CODEC_ID_ADPCM_IMA_AMV:
1358     case CODEC_ID_ADPCM_IMA_ISS:
1359     case CODEC_ID_ADPCM_IMA_EA_EACS:
1360     case CODEC_ID_ADPCM_IMA_EA_SEAD:
1361     case CODEC_ID_ADPCM_MS:
1362     case CODEC_ID_ADPCM_4XM:
1363     case CODEC_ID_ADPCM_XA:
1364     case CODEC_ID_ADPCM_ADX:
1365     case CODEC_ID_ADPCM_EA:
1366     case CODEC_ID_ADPCM_CT:
1367     case CODEC_ID_ADPCM_SWF:
1368     case CODEC_ID_ADPCM_YAMAHA:
1369     case CODEC_ID_ADPCM_SBPRO_2:
1370     case CODEC_ID_ADPCM_SBPRO_3:
1371     case CODEC_ID_ADPCM_SBPRO_4:
1372     case CODEC_ID_ADPCM_EA_R1:
1373     case CODEC_ID_ADPCM_EA_R2:
1374     case CODEC_ID_ADPCM_EA_R3:
1375     case CODEC_ID_ADPCM_EA_MAXIS_XA:
1376     case CODEC_ID_ADPCM_EA_XAS:
1377     case CODEC_ID_ADPCM_THP:
1378     {
1379       const gchar *layout = NULL;
1380
1381       switch (codec_id) {
1382         case CODEC_ID_ADPCM_IMA_QT:
1383           layout = "quicktime";
1384           break;
1385         case CODEC_ID_ADPCM_IMA_WAV:
1386           layout = "dvi";
1387           break;
1388         case CODEC_ID_ADPCM_IMA_DK3:
1389           layout = "dk3";
1390           break;
1391         case CODEC_ID_ADPCM_IMA_DK4:
1392           layout = "dk4";
1393           break;
1394         case CODEC_ID_ADPCM_IMA_WS:
1395           layout = "westwood";
1396           break;
1397         case CODEC_ID_ADPCM_IMA_SMJPEG:
1398           layout = "smjpeg";
1399           break;
1400         case CODEC_ID_ADPCM_IMA_AMV:
1401           layout = "amv";
1402           break;
1403         case CODEC_ID_ADPCM_IMA_ISS:
1404           layout = "iss";
1405           break;
1406         case CODEC_ID_ADPCM_IMA_EA_EACS:
1407           layout = "ea-eacs";
1408           break;
1409         case CODEC_ID_ADPCM_IMA_EA_SEAD:
1410           layout = "ea-sead";
1411           break;
1412         case CODEC_ID_ADPCM_MS:
1413           layout = "microsoft";
1414           break;
1415         case CODEC_ID_ADPCM_4XM:
1416           layout = "4xm";
1417           break;
1418         case CODEC_ID_ADPCM_XA:
1419           layout = "xa";
1420           break;
1421         case CODEC_ID_ADPCM_ADX:
1422           layout = "adx";
1423           break;
1424         case CODEC_ID_ADPCM_EA:
1425           layout = "ea";
1426           break;
1427         case CODEC_ID_ADPCM_CT:
1428           layout = "ct";
1429           break;
1430         case CODEC_ID_ADPCM_SWF:
1431           layout = "swf";
1432           break;
1433         case CODEC_ID_ADPCM_YAMAHA:
1434           layout = "yamaha";
1435           break;
1436         case CODEC_ID_ADPCM_SBPRO_2:
1437           layout = "sbpro2";
1438           break;
1439         case CODEC_ID_ADPCM_SBPRO_3:
1440           layout = "sbpro3";
1441           break;
1442         case CODEC_ID_ADPCM_SBPRO_4:
1443           layout = "sbpro4";
1444           break;
1445         case CODEC_ID_ADPCM_EA_R1:
1446           layout = "ea-r1";
1447           break;
1448         case CODEC_ID_ADPCM_EA_R2:
1449           layout = "ea-r3";
1450           break;
1451         case CODEC_ID_ADPCM_EA_R3:
1452           layout = "ea-r3";
1453           break;
1454         case CODEC_ID_ADPCM_EA_MAXIS_XA:
1455           layout = "ea-maxis-xa";
1456           break;
1457         case CODEC_ID_ADPCM_EA_XAS:
1458           layout = "ea-xas";
1459           break;
1460         case CODEC_ID_ADPCM_THP:
1461           layout = "thp";
1462           break;
1463         default:
1464           g_assert (0);         /* don't worry, we never get here */
1465           break;
1466       }
1467
1468       /* FIXME: someone please check whether we need additional properties
1469        * in this caps definition. */
1470       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-adpcm",
1471           "layout", G_TYPE_STRING, layout, NULL);
1472       if (context)
1473         gst_caps_set_simple (caps,
1474             "block_align", G_TYPE_INT, context->block_align,
1475             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1476     }
1477       break;
1478
1479     case CODEC_ID_AMR_NB:
1480       caps = gst_ff_aud_caps_new (context, codec_id, "audio/AMR", NULL);
1481       break;
1482
1483     case CODEC_ID_AMR_WB:
1484       caps = gst_ff_aud_caps_new (context, codec_id, "audio/AMR-WB", NULL);
1485       break;
1486
1487     case CODEC_ID_GSM:
1488       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-gsm", NULL);
1489       break;
1490
1491     case CODEC_ID_GSM_MS:
1492       caps = gst_ff_aud_caps_new (context, codec_id, "audio/ms-gsm", NULL);
1493       break;
1494
1495     case CODEC_ID_NELLYMOSER:
1496       caps =
1497           gst_ff_aud_caps_new (context, codec_id, "audio/x-nellymoser", NULL);
1498       break;
1499
1500     case CODEC_ID_SIPR:
1501     {
1502       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-sipro", NULL);
1503       if (context) {
1504         gst_caps_set_simple (caps,
1505             "leaf_size", G_TYPE_INT, context->block_align,
1506             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1507       }
1508     }
1509       break;
1510
1511     case CODEC_ID_RA_144:
1512     case CODEC_ID_RA_288:
1513     case CODEC_ID_COOK:
1514     {
1515       gint version = 0;
1516
1517       switch (codec_id) {
1518         case CODEC_ID_RA_144:
1519           version = 1;
1520           break;
1521         case CODEC_ID_RA_288:
1522           version = 2;
1523           break;
1524         case CODEC_ID_COOK:
1525           version = 8;
1526           break;
1527         default:
1528           break;
1529       }
1530
1531       /* FIXME: properties? */
1532       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-pn-realaudio",
1533           "raversion", G_TYPE_INT, version, NULL);
1534       if (context) {
1535         gst_caps_set_simple (caps,
1536             "leaf_size", G_TYPE_INT, context->block_align,
1537             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1538       }
1539     }
1540       break;
1541
1542     case CODEC_ID_ROQ_DPCM:
1543     case CODEC_ID_INTERPLAY_DPCM:
1544     case CODEC_ID_XAN_DPCM:
1545     case CODEC_ID_SOL_DPCM:
1546     {
1547       const gchar *layout = NULL;
1548
1549       switch (codec_id) {
1550         case CODEC_ID_ROQ_DPCM:
1551           layout = "roq";
1552           break;
1553         case CODEC_ID_INTERPLAY_DPCM:
1554           layout = "interplay";
1555           break;
1556         case CODEC_ID_XAN_DPCM:
1557           layout = "xan";
1558           break;
1559         case CODEC_ID_SOL_DPCM:
1560           layout = "sol";
1561           break;
1562         default:
1563           g_assert (0);         /* don't worry, we never get here */
1564           break;
1565       }
1566
1567       /* FIXME: someone please check whether we need additional properties
1568        * in this caps definition. */
1569       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-dpcm",
1570           "layout", G_TYPE_STRING, layout, NULL);
1571       if (context)
1572         gst_caps_set_simple (caps,
1573             "block_align", G_TYPE_INT, context->block_align,
1574             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1575     }
1576       break;
1577
1578     case CODEC_ID_SHORTEN:
1579       caps = gst_caps_new_simple ("audio/x-shorten", NULL);
1580       break;
1581
1582     case CODEC_ID_ALAC:
1583       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-alac", NULL);
1584       if (context) {
1585         gst_caps_set_simple (caps,
1586             "samplesize", G_TYPE_INT, context->bits_per_coded_sample, NULL);
1587       }
1588       break;
1589
1590     case CODEC_ID_FLAC:
1591       /* Note that ffmpeg has no encoder yet, but just for safety. In the
1592        * encoder case, we want to add things like samplerate, channels... */
1593       if (!encode) {
1594         caps = gst_caps_new_simple ("audio/x-flac", NULL);
1595       }
1596       break;
1597
1598     case CODEC_ID_DVD_SUBTITLE:
1599     case CODEC_ID_DVB_SUBTITLE:
1600       caps = NULL;
1601       break;
1602     case CODEC_ID_BMP:
1603       caps = gst_caps_new_simple ("image/bmp", NULL);
1604       break;
1605     case CODEC_ID_TTA:
1606       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-tta", NULL);
1607       if (context) {
1608         gst_caps_set_simple (caps,
1609             "samplesize", G_TYPE_INT, context->bits_per_coded_sample, NULL);
1610       }
1611       break;
1612     case CODEC_ID_TWINVQ:
1613       caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-twin-vq", NULL);
1614       break;
1615     default:
1616       GST_DEBUG ("Unknown codec ID %d, please add mapping here", codec_id);
1617       break;
1618   }
1619
1620   if (buildcaps) {
1621     AVCodec *codec;
1622
1623     if ((codec = avcodec_find_decoder (codec_id)) ||
1624         (codec = avcodec_find_encoder (codec_id))) {
1625       gchar *mime = NULL;
1626
1627       GST_LOG ("Could not create stream format caps for %s", codec->name);
1628
1629       switch (codec->type) {
1630         case AVMEDIA_TYPE_VIDEO:
1631           mime = g_strdup_printf ("video/x-gst_ff-%s", codec->name);
1632           caps = gst_ff_vid_caps_new (context, codec_id, mime, NULL);
1633           g_free (mime);
1634           break;
1635         case AVMEDIA_TYPE_AUDIO:
1636           mime = g_strdup_printf ("audio/x-gst_ff-%s", codec->name);
1637           caps = gst_ff_aud_caps_new (context, codec_id, mime, NULL);
1638           if (context)
1639             gst_caps_set_simple (caps,
1640                 "block_align", G_TYPE_INT, context->block_align,
1641                 "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1642           g_free (mime);
1643           break;
1644         default:
1645           break;
1646       }
1647     }
1648   }
1649
1650   if (caps != NULL) {
1651
1652     /* set private data */
1653     if (context && context->extradata_size > 0) {
1654       GstBuffer *data = gst_buffer_new_and_alloc (context->extradata_size);
1655
1656       gst_buffer_fill (data, 0, context->extradata, context->extradata_size);
1657       gst_caps_set_simple (caps, "codec_data", GST_TYPE_BUFFER, data, NULL);
1658       gst_buffer_unref (data);
1659     }
1660
1661     /* palette */
1662     if (context) {
1663       gst_ffmpeg_set_palette (caps, context);
1664     }
1665
1666     GST_LOG ("caps for codec_id=%d: %" GST_PTR_FORMAT, codec_id, caps);
1667
1668   } else {
1669     GST_LOG ("No caps found for codec_id=%d", codec_id);
1670   }
1671
1672   return caps;
1673 }
1674
1675 /* Convert a FFMPEG Pixel Format to a GStreamer VideoFormat */
1676 GstVideoFormat
1677 gst_ffmpeg_pixfmt_to_video_format (enum PixelFormat pix_fmt)
1678 {
1679   GstVideoFormat fmt;
1680
1681   switch (pix_fmt) {
1682     case PIX_FMT_YUVJ420P:
1683     case PIX_FMT_YUV420P:
1684       fmt = GST_VIDEO_FORMAT_I420;
1685       break;
1686     case PIX_FMT_YUVA420P:
1687       fmt = GST_VIDEO_FORMAT_A420;
1688       break;
1689     case PIX_FMT_YUYV422:
1690       fmt = GST_VIDEO_FORMAT_YUY2;
1691       break;
1692     case PIX_FMT_RGB24:
1693       fmt = GST_VIDEO_FORMAT_RGB;
1694       break;
1695     case PIX_FMT_BGR24:
1696       fmt = GST_VIDEO_FORMAT_BGR;
1697       break;
1698     case PIX_FMT_YUVJ422P:
1699     case PIX_FMT_YUV422P:
1700       fmt = GST_VIDEO_FORMAT_Y42B;
1701       break;
1702     case PIX_FMT_YUVJ444P:
1703     case PIX_FMT_YUV444P:
1704       fmt = GST_VIDEO_FORMAT_Y444;
1705       break;
1706     case PIX_FMT_RGB32:
1707 #if (G_BYTE_ORDER == G_BIG_ENDIAN)
1708       fmt = GST_VIDEO_FORMAT_xRGB;
1709 #else
1710       fmt = GST_VIDEO_FORMAT_BGRx;
1711 #endif
1712       break;
1713     case PIX_FMT_YUV410P:
1714       fmt = GST_VIDEO_FORMAT_YUV9;
1715       break;
1716     case PIX_FMT_YUV411P:
1717       fmt = GST_VIDEO_FORMAT_Y41B;
1718       break;
1719     case PIX_FMT_RGB565:
1720       fmt = GST_VIDEO_FORMAT_RGB16;
1721       break;
1722     case PIX_FMT_RGB555:
1723       fmt = GST_VIDEO_FORMAT_RGB15;
1724       break;
1725     case PIX_FMT_PAL8:
1726       fmt = GST_VIDEO_FORMAT_RGB8_PALETTED;
1727       break;
1728     case PIX_FMT_GRAY8:
1729       fmt = GST_VIDEO_FORMAT_GRAY8;
1730       break;
1731     default:
1732       /* give up ... */
1733       fmt = GST_VIDEO_FORMAT_UNKNOWN;
1734       break;
1735   }
1736   return fmt;
1737 }
1738
1739 /* Convert a FFMPEG Pixel Format and optional AVCodecContext
1740  * to a GstCaps. If the context is ommitted, no fixed values
1741  * for video/audio size will be included in the GstCaps
1742  *
1743  * See below for usefullness
1744  */
1745
1746 GstCaps *
1747 gst_ffmpeg_pixfmt_to_caps (enum PixelFormat pix_fmt, AVCodecContext * context,
1748     enum CodecID codec_id)
1749 {
1750   GstCaps *caps = NULL;
1751   GstVideoFormat format;
1752
1753   format = gst_ffmpeg_pixfmt_to_video_format (pix_fmt);
1754
1755   if (format != GST_VIDEO_FORMAT_UNKNOWN) {
1756     caps = gst_ff_vid_caps_new (context, codec_id, "video/x-raw",
1757         "format", G_TYPE_STRING, gst_video_format_to_string (format), NULL);
1758   }
1759
1760   if (caps != NULL) {
1761     GST_DEBUG ("caps for pix_fmt=%d: %" GST_PTR_FORMAT, pix_fmt, caps);
1762   } else {
1763     GST_LOG ("No caps found for pix_fmt=%d", pix_fmt);
1764   }
1765
1766   return caps;
1767 }
1768
1769 /* Convert a FFMPEG Sample Format and optional AVCodecContext
1770  * to a GstCaps. If the context is ommitted, no fixed values
1771  * for video/audio size will be included in the GstCaps
1772  *
1773  * See below for usefullness
1774  */
1775
1776 static GstCaps *
1777 gst_ffmpeg_smpfmt_to_caps (enum SampleFormat sample_fmt,
1778     AVCodecContext * context, enum CodecID codec_id)
1779 {
1780   GstCaps *caps = NULL;
1781   GstAudioFormat format;
1782
1783   switch (sample_fmt) {
1784     case SAMPLE_FMT_S16:
1785       format = GST_AUDIO_FORMAT_S16;
1786       break;
1787     case SAMPLE_FMT_S32:
1788       format = GST_AUDIO_FORMAT_S32;
1789       break;
1790     case SAMPLE_FMT_FLT:
1791       format = GST_AUDIO_FORMAT_F32;
1792       break;
1793     case SAMPLE_FMT_DBL:
1794       format = GST_AUDIO_FORMAT_F64;
1795       break;
1796     default:
1797       /* .. */
1798       format = GST_AUDIO_FORMAT_UNKNOWN;
1799       break;
1800   }
1801
1802   if (format != GST_AUDIO_FORMAT_UNKNOWN) {
1803     caps = gst_ff_aud_caps_new (context, codec_id, "audio/x-raw",
1804         "format", G_TYPE_STRING, gst_audio_format_to_string (format), NULL);
1805     GST_LOG ("caps for sample_fmt=%d: %" GST_PTR_FORMAT, sample_fmt, caps);
1806   } else {
1807     GST_LOG ("No caps found for sample_fmt=%d", sample_fmt);
1808   }
1809
1810   return caps;
1811 }
1812
1813 GstCaps *
1814 gst_ffmpeg_codectype_to_audio_caps (AVCodecContext * context,
1815     enum CodecID codec_id, gboolean encode, AVCodec * codec)
1816 {
1817   GstCaps *caps = NULL;
1818
1819   GST_DEBUG ("context:%p, codec_id:%d, encode:%d, codec:%p",
1820       context, codec_id, encode, codec);
1821   if (codec)
1822     GST_DEBUG ("sample_fmts:%p, samplerates:%p",
1823         codec->sample_fmts, codec->supported_samplerates);
1824
1825   if (context) {
1826     /* Specific codec context */
1827     caps = gst_ffmpeg_smpfmt_to_caps (context->sample_fmt, context, codec_id);
1828   } else if (codec && codec->sample_fmts) {
1829     GstCaps *temp;
1830     int i;
1831
1832     caps = gst_caps_new_empty ();
1833     for (i = 0; codec->sample_fmts[i] != -1; i++) {
1834       temp =
1835           gst_ffmpeg_smpfmt_to_caps (codec->sample_fmts[i], context, codec_id);
1836       if (temp != NULL)
1837         gst_caps_append (caps, temp);
1838     }
1839   } else {
1840     GstCaps *temp;
1841     enum SampleFormat i;
1842     AVCodecContext ctx = { 0, };
1843
1844     ctx.channels = -1;
1845     caps = gst_caps_new_empty ();
1846     for (i = 0; i <= SAMPLE_FMT_DBL; i++) {
1847       temp = gst_ffmpeg_smpfmt_to_caps (i, encode ? &ctx : NULL, codec_id);
1848       if (temp != NULL) {
1849         gst_caps_append (caps, temp);
1850       }
1851     }
1852   }
1853   return caps;
1854 }
1855
1856 GstCaps *
1857 gst_ffmpeg_codectype_to_video_caps (AVCodecContext * context,
1858     enum CodecID codec_id, gboolean encode, AVCodec * codec)
1859 {
1860   GstCaps *caps;
1861
1862   GST_LOG ("context:%p, codec_id:%d, encode:%d, codec:%p",
1863       context, codec_id, encode, codec);
1864
1865   if (context) {
1866     caps = gst_ffmpeg_pixfmt_to_caps (context->pix_fmt, context, codec_id);
1867   } else {
1868     GstCaps *temp;
1869     enum PixelFormat i;
1870     AVCodecContext ctx = { 0, };
1871
1872     caps = gst_caps_new_empty ();
1873     for (i = 0; i < PIX_FMT_NB; i++) {
1874       ctx.width = -1;
1875       ctx.pix_fmt = i;
1876       temp = gst_ffmpeg_pixfmt_to_caps (i, encode ? &ctx : NULL, codec_id);
1877       if (temp != NULL) {
1878         gst_caps_append (caps, temp);
1879       }
1880     }
1881   }
1882   return caps;
1883 }
1884
1885 /* Convert a FFMPEG codec Type and optional AVCodecContext
1886  * to a GstCaps. If the context is ommitted, no fixed values
1887  * for video/audio size will be included in the GstCaps
1888  *
1889  * AVMediaType is primarily meant for uncompressed data GstCaps!
1890  */
1891
1892 GstCaps *
1893 gst_ffmpeg_codectype_to_caps (enum AVMediaType codec_type,
1894     AVCodecContext * context, enum CodecID codec_id, gboolean encode)
1895 {
1896   GstCaps *caps;
1897
1898   switch (codec_type) {
1899     case AVMEDIA_TYPE_VIDEO:
1900       caps =
1901           gst_ffmpeg_codectype_to_video_caps (context, codec_id, encode, NULL);
1902       break;
1903     case AVMEDIA_TYPE_AUDIO:
1904       caps =
1905           gst_ffmpeg_codectype_to_audio_caps (context, codec_id, encode, NULL);
1906       break;
1907     default:
1908       caps = NULL;
1909       break;
1910   }
1911
1912   return caps;
1913 }
1914
1915 /* Convert a GstCaps (audio/raw) to a FFMPEG SampleFmt
1916  * and other audio properties in a AVCodecContext.
1917  *
1918  * For usefullness, see below
1919  */
1920
1921 static void
1922 gst_ffmpeg_caps_to_smpfmt (const GstCaps * caps,
1923     AVCodecContext * context, gboolean raw)
1924 {
1925   GstStructure *structure;
1926   const gchar *fmt;
1927   GstAudioFormat format = GST_AUDIO_FORMAT_UNKNOWN;
1928
1929   g_return_if_fail (gst_caps_get_size (caps) == 1);
1930
1931   structure = gst_caps_get_structure (caps, 0);
1932
1933   gst_structure_get_int (structure, "channels", &context->channels);
1934   gst_structure_get_int (structure, "rate", &context->sample_rate);
1935   gst_structure_get_int (structure, "block_align", &context->block_align);
1936   gst_structure_get_int (structure, "bitrate", &context->bit_rate);
1937
1938   if (!raw)
1939     return;
1940
1941   if (gst_structure_has_name (structure, "audio/x-raw")) {
1942     if ((fmt = gst_structure_get_string (structure, "format"))) {
1943       format = gst_audio_format_from_string (fmt);
1944     }
1945   }
1946
1947   switch (format) {
1948     case GST_AUDIO_FORMAT_F32:
1949       context->sample_fmt = SAMPLE_FMT_FLT;
1950       break;
1951     case GST_AUDIO_FORMAT_F64:
1952       context->sample_fmt = SAMPLE_FMT_DBL;
1953       break;
1954     case GST_AUDIO_FORMAT_S32:
1955       context->sample_fmt = SAMPLE_FMT_S32;
1956       break;
1957     case GST_AUDIO_FORMAT_S16:
1958       context->sample_fmt = SAMPLE_FMT_S16;
1959       break;
1960     default:
1961       break;
1962   }
1963 }
1964
1965 /* Convert a GstCaps (video/raw) to a FFMPEG PixFmt
1966  * and other video properties in a AVCodecContext.
1967  *
1968  * For usefullness, see below
1969  */
1970
1971 static void
1972 gst_ffmpeg_caps_to_pixfmt (const GstCaps * caps,
1973     AVCodecContext * context, gboolean raw)
1974 {
1975   GstStructure *structure;
1976   const GValue *fps;
1977   const GValue *par = NULL;
1978   const gchar *fmt;
1979   GstVideoFormat format = GST_VIDEO_FORMAT_UNKNOWN;
1980
1981   GST_DEBUG ("converting caps %" GST_PTR_FORMAT, caps);
1982   g_return_if_fail (gst_caps_get_size (caps) == 1);
1983   structure = gst_caps_get_structure (caps, 0);
1984
1985   gst_structure_get_int (structure, "width", &context->width);
1986   gst_structure_get_int (structure, "height", &context->height);
1987   gst_structure_get_int (structure, "bpp", &context->bits_per_coded_sample);
1988
1989   fps = gst_structure_get_value (structure, "framerate");
1990   if (fps != NULL && GST_VALUE_HOLDS_FRACTION (fps)) {
1991
1992     /* somehow these seem mixed up.. */
1993     context->time_base.den = gst_value_get_fraction_numerator (fps);
1994     context->time_base.num = gst_value_get_fraction_denominator (fps);
1995     context->ticks_per_frame = 1;
1996
1997     GST_DEBUG ("setting framerate %d/%d = %lf",
1998         context->time_base.den, context->time_base.num,
1999         1. * context->time_base.den / context->time_base.num);
2000   }
2001
2002   par = gst_structure_get_value (structure, "pixel-aspect-ratio");
2003   if (par && GST_VALUE_HOLDS_FRACTION (par)) {
2004
2005     context->sample_aspect_ratio.num = gst_value_get_fraction_numerator (par);
2006     context->sample_aspect_ratio.den = gst_value_get_fraction_denominator (par);
2007
2008     GST_DEBUG ("setting pixel-aspect-ratio %d/%d = %lf",
2009         context->sample_aspect_ratio.den, context->sample_aspect_ratio.num,
2010         1. * context->sample_aspect_ratio.den /
2011         context->sample_aspect_ratio.num);
2012   }
2013
2014   if (!raw)
2015     return;
2016
2017   g_return_if_fail (fps != NULL && GST_VALUE_HOLDS_FRACTION (fps));
2018
2019   if (gst_structure_has_name (structure, "video/x-raw")) {
2020     if ((fmt = gst_structure_get_string (structure, "format"))) {
2021       format = gst_video_format_from_string (fmt);
2022     }
2023   }
2024
2025   switch (format) {
2026     case GST_VIDEO_FORMAT_YUY2:
2027       context->pix_fmt = PIX_FMT_YUYV422;
2028       break;
2029     case GST_VIDEO_FORMAT_I420:
2030       context->pix_fmt = PIX_FMT_YUV420P;
2031       break;
2032     case GST_VIDEO_FORMAT_A420:
2033       context->pix_fmt = PIX_FMT_YUVA420P;
2034       break;
2035     case GST_VIDEO_FORMAT_Y41B:
2036       context->pix_fmt = PIX_FMT_YUV411P;
2037       break;
2038     case GST_VIDEO_FORMAT_Y42B:
2039       context->pix_fmt = PIX_FMT_YUV422P;
2040       break;
2041     case GST_VIDEO_FORMAT_YUV9:
2042       context->pix_fmt = PIX_FMT_YUV410P;
2043       break;
2044     case GST_VIDEO_FORMAT_Y444:
2045       context->pix_fmt = PIX_FMT_YUV444P;
2046       break;
2047     case GST_VIDEO_FORMAT_GRAY8:
2048       context->pix_fmt = PIX_FMT_GRAY8;
2049       break;
2050     case GST_VIDEO_FORMAT_xRGB:
2051 #if (G_BYTE_ORDER == G_BIG_ENDIAN)
2052       context->pix_fmt = PIX_FMT_RGB32;
2053 #endif
2054       break;
2055     case GST_VIDEO_FORMAT_BGRx:
2056 #if (G_BYTE_ORDER == G_LITTLE_ENDIAN)
2057       context->pix_fmt = PIX_FMT_RGB32;
2058 #endif
2059       break;
2060     case GST_VIDEO_FORMAT_RGB:
2061       context->pix_fmt = PIX_FMT_RGB24;
2062       break;
2063     case GST_VIDEO_FORMAT_BGR:
2064       context->pix_fmt = PIX_FMT_BGR24;
2065       break;
2066     case GST_VIDEO_FORMAT_RGB16:
2067       context->pix_fmt = PIX_FMT_RGB565;
2068       break;
2069     case GST_VIDEO_FORMAT_RGB15:
2070       context->pix_fmt = PIX_FMT_RGB555;
2071       break;
2072     case GST_VIDEO_FORMAT_RGB8_PALETTED:
2073       context->pix_fmt = PIX_FMT_PAL8;
2074       gst_ffmpeg_get_palette (caps, context);
2075       break;
2076     default:
2077       break;
2078   }
2079 }
2080
2081 /* Convert a GstCaps and a FFMPEG codec Type to a
2082  * AVCodecContext. If the context is ommitted, no fixed values
2083  * for video/audio size will be included in the context
2084  *
2085  * AVMediaType is primarily meant for uncompressed data GstCaps!
2086  */
2087
2088 void
2089 gst_ffmpeg_caps_with_codectype (enum AVMediaType type,
2090     const GstCaps * caps, AVCodecContext * context)
2091 {
2092   if (context == NULL)
2093     return;
2094
2095   switch (type) {
2096     case AVMEDIA_TYPE_VIDEO:
2097       gst_ffmpeg_caps_to_pixfmt (caps, context, TRUE);
2098       break;
2099
2100     case AVMEDIA_TYPE_AUDIO:
2101       gst_ffmpeg_caps_to_smpfmt (caps, context, TRUE);
2102       break;
2103
2104     default:
2105       /* unknown */
2106       break;
2107   }
2108 }
2109
2110 #if 0
2111 static void
2112 nal_escape (guint8 * dst, guint8 * src, guint size, guint * destsize)
2113 {
2114   guint8 *dstp = dst;
2115   guint8 *srcp = src;
2116   guint8 *end = src + size;
2117   gint count = 0;
2118
2119   while (srcp < end) {
2120     if (count == 2 && *srcp <= 0x03) {
2121       GST_DEBUG ("added escape code");
2122       *dstp++ = 0x03;
2123       count = 0;
2124     }
2125     if (*srcp == 0)
2126       count++;
2127     else
2128       count = 0;
2129
2130     GST_DEBUG ("copy %02x, count %d", *srcp, count);
2131     *dstp++ = *srcp++;
2132   }
2133   *destsize = dstp - dst;
2134 }
2135
2136 /* copy the config, escaping NAL units as we iterate them, if something fails we
2137  * copy everything and hope for the best. */
2138 static void
2139 copy_config (guint8 * dst, guint8 * src, guint size, guint * destsize)
2140 {
2141   guint8 *dstp = dst;
2142   guint8 *srcp = src;
2143   gint cnt, i;
2144   guint nalsize, esize;
2145
2146   /* check size */
2147   if (size < 7)
2148     goto full_copy;
2149
2150   /* check version */
2151   if (*srcp != 1)
2152     goto full_copy;
2153
2154   cnt = *(srcp + 5) & 0x1f;     /* Number of sps */
2155
2156   GST_DEBUG ("num SPS %d", cnt);
2157
2158   memcpy (dstp, srcp, 6);
2159   srcp += 6;
2160   dstp += 6;
2161
2162   for (i = 0; i < cnt; i++) {
2163     GST_DEBUG ("copy SPS %d", i);
2164     nalsize = (srcp[0] << 8) | srcp[1];
2165     nal_escape (dstp + 2, srcp + 2, nalsize, &esize);
2166     dstp[0] = esize >> 8;
2167     dstp[1] = esize & 0xff;
2168     dstp += esize + 2;
2169     srcp += nalsize + 2;
2170   }
2171
2172   cnt = *(dstp++) = *(srcp++);  /* Number of pps */
2173
2174   GST_DEBUG ("num PPS %d", cnt);
2175
2176   for (i = 0; i < cnt; i++) {
2177     GST_DEBUG ("copy PPS %d", i);
2178     nalsize = (srcp[0] << 8) | srcp[1];
2179     nal_escape (dstp + 2, srcp + 2, nalsize, &esize);
2180     dstp[0] = esize >> 8;
2181     dstp[1] = esize & 0xff;
2182     dstp += esize + 2;
2183     srcp += nalsize + 2;
2184   }
2185   *destsize = dstp - dst;
2186
2187   return;
2188
2189 full_copy:
2190   {
2191     GST_DEBUG ("something unexpected, doing full copy");
2192     memcpy (dst, src, size);
2193     *destsize = size;
2194     return;
2195   }
2196 }
2197 #endif
2198
2199 /*
2200  * caps_with_codecid () transforms a GstCaps for a known codec
2201  * ID into a filled-in context.
2202  * codec_data from caps will override possible extradata already in the context
2203  */
2204
2205 void
2206 gst_ffmpeg_caps_with_codecid (enum CodecID codec_id,
2207     enum AVMediaType codec_type, const GstCaps * caps, AVCodecContext * context)
2208 {
2209   GstStructure *str;
2210   const GValue *value;
2211   GstBuffer *buf;
2212
2213   GST_LOG ("codec_id:%d, codec_type:%d, caps:%" GST_PTR_FORMAT " context:%p",
2214       codec_id, codec_type, caps, context);
2215
2216   if (!context || !gst_caps_get_size (caps))
2217     return;
2218
2219   str = gst_caps_get_structure (caps, 0);
2220
2221   /* extradata parsing (esds [mpeg4], wma/wmv, msmpeg4v1/2/3, etc.) */
2222   if ((value = gst_structure_get_value (str, "codec_data"))) {
2223     gsize size;
2224     guint8 *data;
2225
2226     buf = gst_value_get_buffer (value);
2227     data = gst_buffer_map (buf, &size, NULL, GST_MAP_READ);
2228
2229     /* free the old one if it is there */
2230     if (context->extradata)
2231       av_free (context->extradata);
2232
2233 #if 0
2234     if (codec_id == CODEC_ID_H264) {
2235       guint extrasize;
2236
2237       GST_DEBUG ("copy, escaping codec_data %d", size);
2238       /* ffmpeg h264 expects the codec_data to be escaped, there is no real
2239        * reason for this but let's just escape it for now. Start by allocating
2240        * enough space, x2 is more than enough.
2241        *
2242        * FIXME, we disabled escaping because some file already contain escaped
2243        * codec_data and then we escape twice and fail. It's better to leave it
2244        * as is, as that is what most players do. */
2245       context->extradata =
2246           av_mallocz (GST_ROUND_UP_16 (size * 2 +
2247               FF_INPUT_BUFFER_PADDING_SIZE));
2248       copy_config (context->extradata, data, size, &extrasize);
2249       GST_DEBUG ("escaped size: %d", extrasize);
2250       context->extradata_size = extrasize;
2251     } else
2252 #endif
2253     {
2254       /* allocate with enough padding */
2255       GST_DEBUG ("copy codec_data");
2256       context->extradata =
2257           av_mallocz (GST_ROUND_UP_16 (size + FF_INPUT_BUFFER_PADDING_SIZE));
2258       memcpy (context->extradata, data, size);
2259       context->extradata_size = size;
2260     }
2261
2262     /* Hack for VC1. Sometimes the first (length) byte is 0 for some files */
2263     if (codec_id == CODEC_ID_VC1 && size > 0 && data[0] == 0) {
2264       context->extradata[0] = (guint8) size;
2265     }
2266
2267     GST_DEBUG ("have codec data of size %d", size);
2268
2269     gst_buffer_unmap (buf, data, size);
2270   } else if (context->extradata == NULL && codec_id != CODEC_ID_AAC_LATM &&
2271       codec_id != CODEC_ID_FLAC) {
2272     /* no extradata, alloc dummy with 0 sized, some codecs insist on reading
2273      * extradata anyway which makes then segfault. */
2274     context->extradata =
2275         av_mallocz (GST_ROUND_UP_16 (FF_INPUT_BUFFER_PADDING_SIZE));
2276     context->extradata_size = 0;
2277     GST_DEBUG ("no codec data");
2278   }
2279
2280   switch (codec_id) {
2281     case CODEC_ID_MPEG4:
2282     {
2283       const gchar *mime = gst_structure_get_name (str);
2284
2285       if (!strcmp (mime, "video/x-divx"))
2286         context->codec_tag = GST_MAKE_FOURCC ('D', 'I', 'V', 'X');
2287       else if (!strcmp (mime, "video/x-xvid"))
2288         context->codec_tag = GST_MAKE_FOURCC ('X', 'V', 'I', 'D');
2289       else if (!strcmp (mime, "video/x-3ivx"))
2290         context->codec_tag = GST_MAKE_FOURCC ('3', 'I', 'V', '1');
2291       else if (!strcmp (mime, "video/mpeg"))
2292         context->codec_tag = GST_MAKE_FOURCC ('m', 'p', '4', 'v');
2293     }
2294       break;
2295
2296     case CODEC_ID_SVQ3:
2297       /* FIXME: this is a workaround for older gst-plugins releases
2298        * (<= 0.8.9). This should be removed at some point, because
2299        * it causes wrong decoded frame order. */
2300       if (!context->extradata) {
2301         gint halfpel_flag, thirdpel_flag, low_delay, unknown_svq3_flag;
2302         guint16 flags;
2303
2304         if (gst_structure_get_int (str, "halfpel_flag", &halfpel_flag) ||
2305             gst_structure_get_int (str, "thirdpel_flag", &thirdpel_flag) ||
2306             gst_structure_get_int (str, "low_delay", &low_delay) ||
2307             gst_structure_get_int (str, "unknown_svq3_flag",
2308                 &unknown_svq3_flag)) {
2309           context->extradata = (guint8 *) av_mallocz (0x64);
2310           g_stpcpy ((gchar *) context->extradata, "SVQ3");
2311           flags = 1 << 3;
2312           flags |= low_delay;
2313           flags = flags << 2;
2314           flags |= unknown_svq3_flag;
2315           flags = flags << 6;
2316           flags |= halfpel_flag;
2317           flags = flags << 1;
2318           flags |= thirdpel_flag;
2319           flags = flags << 3;
2320
2321           flags = GUINT16_FROM_LE (flags);
2322
2323           memcpy ((gchar *) context->extradata + 0x62, &flags, 2);
2324           context->extradata_size = 0x64;
2325         }
2326       }
2327       break;
2328
2329     case CODEC_ID_MSRLE:
2330     case CODEC_ID_QTRLE:
2331     case CODEC_ID_TSCC:
2332     case CODEC_ID_CSCD:
2333     case CODEC_ID_APE:
2334     {
2335       gint depth;
2336
2337       if (gst_structure_get_int (str, "depth", &depth)) {
2338         context->bits_per_coded_sample = depth;
2339       } else {
2340         GST_WARNING ("No depth field in caps %" GST_PTR_FORMAT, caps);
2341       }
2342
2343     }
2344       break;
2345
2346     case CODEC_ID_RV10:
2347     case CODEC_ID_RV20:
2348     case CODEC_ID_RV30:
2349     case CODEC_ID_RV40:
2350     {
2351       gint format;
2352
2353       if (gst_structure_get_int (str, "format", &format))
2354         context->sub_id = format;
2355
2356       break;
2357     }
2358     case CODEC_ID_COOK:
2359     case CODEC_ID_RA_288:
2360     case CODEC_ID_RA_144:
2361     case CODEC_ID_SIPR:
2362     {
2363       gint leaf_size;
2364       gint bitrate;
2365
2366       if (gst_structure_get_int (str, "leaf_size", &leaf_size))
2367         context->block_align = leaf_size;
2368       if (gst_structure_get_int (str, "bitrate", &bitrate))
2369         context->bit_rate = bitrate;
2370     }
2371     case CODEC_ID_ALAC:
2372       gst_structure_get_int (str, "samplesize",
2373           &context->bits_per_coded_sample);
2374       break;
2375
2376     case CODEC_ID_DVVIDEO:
2377     {
2378       const gchar *format;
2379
2380       format = gst_structure_get_string (str, "format");
2381
2382       if (g_str_equal (format, "YUY2"))
2383         context->pix_fmt = PIX_FMT_YUYV422;
2384       else if (g_str_equal (format, "I420"))
2385         context->pix_fmt = PIX_FMT_YUV420P;
2386       else if (g_str_equal (format, "A420"))
2387         context->pix_fmt = PIX_FMT_YUVA420P;
2388       else if (g_str_equal (format, "Y41B"))
2389         context->pix_fmt = PIX_FMT_YUV411P;
2390       else if (g_str_equal (format, "Y42B"))
2391         context->pix_fmt = PIX_FMT_YUV422P;
2392       else if (g_str_equal (format, "YUV9"))
2393         context->pix_fmt = PIX_FMT_YUV410P;
2394       else {
2395         GST_WARNING ("couldn't convert format %s" " to a pixel format", format);
2396       }
2397       break;
2398     }
2399     case CODEC_ID_H263P:
2400     {
2401       gboolean val;
2402
2403       if (!gst_structure_get_boolean (str, "annex-f", &val) || val)
2404         context->flags |= CODEC_FLAG_4MV;
2405       else
2406         context->flags &= ~CODEC_FLAG_4MV;
2407       if ((!gst_structure_get_boolean (str, "annex-i", &val) || val) &&
2408           (!gst_structure_get_boolean (str, "annex-t", &val) || val))
2409         context->flags |= CODEC_FLAG_AC_PRED;
2410       else
2411         context->flags &= ~CODEC_FLAG_AC_PRED;
2412       if (!gst_structure_get_boolean (str, "annex-j", &val) || val)
2413         context->flags |= CODEC_FLAG_LOOP_FILTER;
2414       else
2415         context->flags &= ~CODEC_FLAG_LOOP_FILTER;
2416       break;
2417     }
2418     case CODEC_ID_ADPCM_G726:
2419     {
2420       const gchar *layout;
2421
2422       if ((layout = gst_structure_get_string (str, "layout"))) {
2423         if (!strcmp (layout, "g721")) {
2424           context->sample_rate = 8000;
2425           context->channels = 1;
2426           context->bit_rate = 32000;
2427         }
2428       }
2429       break;
2430     }
2431     default:
2432       break;
2433   }
2434
2435   if (!gst_caps_is_fixed (caps))
2436     return;
2437
2438   /* common properties (width, height, fps) */
2439   switch (codec_type) {
2440     case AVMEDIA_TYPE_VIDEO:
2441       gst_ffmpeg_caps_to_pixfmt (caps, context, codec_id == CODEC_ID_RAWVIDEO);
2442       gst_ffmpeg_get_palette (caps, context);
2443       break;
2444     case AVMEDIA_TYPE_AUDIO:
2445       gst_ffmpeg_caps_to_smpfmt (caps, context, FALSE);
2446       break;
2447     default:
2448       break;
2449   }
2450
2451   /* fixup of default settings */
2452   switch (codec_id) {
2453     case CODEC_ID_QCELP:
2454       /* QCELP is always mono, no matter what the caps say */
2455       context->channels = 1;
2456       break;
2457     default:
2458       break;
2459   }
2460 }
2461
2462 /* _formatid_to_caps () is meant for muxers/demuxers, it
2463  * transforms a name (ffmpeg way of ID'ing these, why don't
2464  * they have unique numerical IDs?) to the corresponding
2465  * caps belonging to that mux-format
2466  *
2467  * Note: we don't need any additional info because the caps
2468  * isn't supposed to contain any useful info besides the
2469  * media type anyway
2470  */
2471
2472 GstCaps *
2473 gst_ffmpeg_formatid_to_caps (const gchar * format_name)
2474 {
2475   GstCaps *caps = NULL;
2476
2477   if (!strcmp (format_name, "mpeg")) {
2478     caps = gst_caps_new_simple ("video/mpeg",
2479         "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
2480   } else if (!strcmp (format_name, "mpegts")) {
2481     caps = gst_caps_new_simple ("video/mpegts",
2482         "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
2483   } else if (!strcmp (format_name, "rm")) {
2484     caps = gst_caps_new_simple ("application/x-pn-realmedia",
2485         "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
2486   } else if (!strcmp (format_name, "asf")) {
2487     caps = gst_caps_new_simple ("video/x-ms-asf", NULL);
2488   } else if (!strcmp (format_name, "avi")) {
2489     caps = gst_caps_new_simple ("video/x-msvideo", NULL);
2490   } else if (!strcmp (format_name, "wav")) {
2491     caps = gst_caps_new_simple ("audio/x-wav", NULL);
2492   } else if (!strcmp (format_name, "ape")) {
2493     caps = gst_caps_new_simple ("application/x-ape", NULL);
2494   } else if (!strcmp (format_name, "swf")) {
2495     caps = gst_caps_new_simple ("application/x-shockwave-flash", NULL);
2496   } else if (!strcmp (format_name, "au")) {
2497     caps = gst_caps_new_simple ("audio/x-au", NULL);
2498   } else if (!strcmp (format_name, "dv")) {
2499     caps = gst_caps_new_simple ("video/x-dv",
2500         "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
2501   } else if (!strcmp (format_name, "4xm")) {
2502     caps = gst_caps_new_simple ("video/x-4xm", NULL);
2503   } else if (!strcmp (format_name, "matroska")) {
2504     caps = gst_caps_new_simple ("video/x-matroska", NULL);
2505   } else if (!strcmp (format_name, "mp3")) {
2506     caps = gst_caps_new_simple ("application/x-id3", NULL);
2507   } else if (!strcmp (format_name, "flic")) {
2508     caps = gst_caps_new_simple ("video/x-fli", NULL);
2509   } else if (!strcmp (format_name, "flv")) {
2510     caps = gst_caps_new_simple ("video/x-flv", NULL);
2511   } else if (!strcmp (format_name, "tta")) {
2512     caps = gst_caps_new_simple ("audio/x-ttafile", NULL);
2513   } else if (!strcmp (format_name, "aiff")) {
2514     caps = gst_caps_new_simple ("audio/x-aiff", NULL);
2515   } else if (!strcmp (format_name, "mov_mp4_m4a_3gp_3g2")) {
2516     caps =
2517         gst_caps_from_string
2518         ("application/x-3gp; video/quicktime; audio/x-m4a");
2519   } else if (!strcmp (format_name, "mov")) {
2520     caps = gst_caps_from_string ("video/quicktime,variant=(string)apple");
2521   } else if (!strcmp (format_name, "mp4")) {
2522     caps = gst_caps_from_string ("video/quicktime,variant=(string)iso");
2523   } else if (!strcmp (format_name, "3gp")) {
2524     caps = gst_caps_from_string ("video/quicktime,variant=(string)3gpp");
2525   } else if (!strcmp (format_name, "3g2")) {
2526     caps = gst_caps_from_string ("video/quicktime,variant=(string)3g2");
2527   } else if (!strcmp (format_name, "psp")) {
2528     caps = gst_caps_from_string ("video/quicktime,variant=(string)psp");
2529   } else if (!strcmp (format_name, "ipod")) {
2530     caps = gst_caps_from_string ("video/quicktime,variant=(string)ipod");
2531   } else if (!strcmp (format_name, "aac")) {
2532     caps = gst_caps_new_simple ("audio/mpeg",
2533         "mpegversion", G_TYPE_INT, 4, NULL);
2534   } else if (!strcmp (format_name, "gif")) {
2535     caps = gst_caps_from_string ("image/gif");
2536   } else if (!strcmp (format_name, "ogg")) {
2537     caps = gst_caps_from_string ("application/ogg");
2538   } else if (!strcmp (format_name, "mxf") || !strcmp (format_name, "mxf_d10")) {
2539     caps = gst_caps_from_string ("application/mxf");
2540   } else if (!strcmp (format_name, "gxf")) {
2541     caps = gst_caps_from_string ("application/gxf");
2542   } else if (!strcmp (format_name, "yuv4mpegpipe")) {
2543     caps = gst_caps_new_simple ("application/x-yuv4mpeg",
2544         "y4mversion", G_TYPE_INT, 2, NULL);
2545   } else if (!strcmp (format_name, "mpc")) {
2546     caps = gst_caps_from_string ("audio/x-musepack, streamversion = (int) 7");
2547   } else if (!strcmp (format_name, "vqf")) {
2548     caps = gst_caps_from_string ("audio/x-vqf");
2549   } else if (!strcmp (format_name, "nsv")) {
2550     caps = gst_caps_from_string ("video/x-nsv");
2551   } else if (!strcmp (format_name, "amr")) {
2552     caps = gst_caps_from_string ("audio/x-amr-nb-sh");
2553   } else if (!strcmp (format_name, "webm")) {
2554     caps = gst_caps_from_string ("video/webm");
2555   } else {
2556     gchar *name;
2557
2558     GST_LOG ("Could not create stream format caps for %s", format_name);
2559     name = g_strdup_printf ("application/x-gst_ff-%s", format_name);
2560     caps = gst_caps_new_simple (name, NULL);
2561     g_free (name);
2562   }
2563
2564   return caps;
2565 }
2566
2567 gboolean
2568 gst_ffmpeg_formatid_get_codecids (const gchar * format_name,
2569     enum CodecID ** video_codec_list, enum CodecID ** audio_codec_list,
2570     AVOutputFormat * plugin)
2571 {
2572   static enum CodecID tmp_vlist[] = {
2573     CODEC_ID_NONE,
2574     CODEC_ID_NONE
2575   };
2576   static enum CodecID tmp_alist[] = {
2577     CODEC_ID_NONE,
2578     CODEC_ID_NONE
2579   };
2580
2581   GST_LOG ("format_name : %s", format_name);
2582
2583   if (!strcmp (format_name, "mp4")) {
2584     static enum CodecID mp4_video_list[] = {
2585       CODEC_ID_MPEG4, CODEC_ID_H264,
2586       CODEC_ID_MJPEG,
2587       CODEC_ID_NONE
2588     };
2589     static enum CodecID mp4_audio_list[] = {
2590       CODEC_ID_AAC, CODEC_ID_MP3,
2591       CODEC_ID_NONE
2592     };
2593
2594     *video_codec_list = mp4_video_list;
2595     *audio_codec_list = mp4_audio_list;
2596   } else if (!strcmp (format_name, "mpeg")) {
2597     static enum CodecID mpeg_video_list[] = { CODEC_ID_MPEG1VIDEO,
2598       CODEC_ID_MPEG2VIDEO,
2599       CODEC_ID_H264,
2600       CODEC_ID_NONE
2601     };
2602     static enum CodecID mpeg_audio_list[] = { CODEC_ID_MP1,
2603       CODEC_ID_MP2,
2604       CODEC_ID_MP3,
2605       CODEC_ID_NONE
2606     };
2607
2608     *video_codec_list = mpeg_video_list;
2609     *audio_codec_list = mpeg_audio_list;
2610   } else if (!strcmp (format_name, "dvd")) {
2611     static enum CodecID mpeg_video_list[] = { CODEC_ID_MPEG2VIDEO,
2612       CODEC_ID_NONE
2613     };
2614     static enum CodecID mpeg_audio_list[] = { CODEC_ID_MP2,
2615       CODEC_ID_AC3,
2616       CODEC_ID_DTS,
2617       CODEC_ID_PCM_S16BE,
2618       CODEC_ID_NONE
2619     };
2620
2621     *video_codec_list = mpeg_video_list;
2622     *audio_codec_list = mpeg_audio_list;
2623   } else if (!strcmp (format_name, "mpegts")) {
2624     static enum CodecID mpegts_video_list[] = { CODEC_ID_MPEG1VIDEO,
2625       CODEC_ID_MPEG2VIDEO,
2626       CODEC_ID_H264,
2627       CODEC_ID_NONE
2628     };
2629     static enum CodecID mpegts_audio_list[] = { CODEC_ID_MP2,
2630       CODEC_ID_MP3,
2631       CODEC_ID_AC3,
2632       CODEC_ID_DTS,
2633       CODEC_ID_AAC,
2634       CODEC_ID_NONE
2635     };
2636
2637     *video_codec_list = mpegts_video_list;
2638     *audio_codec_list = mpegts_audio_list;
2639   } else if (!strcmp (format_name, "vob")) {
2640     static enum CodecID vob_video_list[] =
2641         { CODEC_ID_MPEG2VIDEO, CODEC_ID_NONE };
2642     static enum CodecID vob_audio_list[] = { CODEC_ID_MP2, CODEC_ID_AC3,
2643       CODEC_ID_DTS, CODEC_ID_NONE
2644     };
2645
2646     *video_codec_list = vob_video_list;
2647     *audio_codec_list = vob_audio_list;
2648   } else if (!strcmp (format_name, "flv")) {
2649     static enum CodecID flv_video_list[] = { CODEC_ID_FLV1, CODEC_ID_NONE };
2650     static enum CodecID flv_audio_list[] = { CODEC_ID_MP3, CODEC_ID_NONE };
2651
2652     *video_codec_list = flv_video_list;
2653     *audio_codec_list = flv_audio_list;
2654   } else if (!strcmp (format_name, "asf")) {
2655     static enum CodecID asf_video_list[] =
2656         { CODEC_ID_WMV1, CODEC_ID_WMV2, CODEC_ID_MSMPEG4V3, CODEC_ID_NONE };
2657     static enum CodecID asf_audio_list[] =
2658         { CODEC_ID_WMAV1, CODEC_ID_WMAV2, CODEC_ID_MP3, CODEC_ID_NONE };
2659
2660     *video_codec_list = asf_video_list;
2661     *audio_codec_list = asf_audio_list;
2662   } else if (!strcmp (format_name, "dv")) {
2663     static enum CodecID dv_video_list[] = { CODEC_ID_DVVIDEO, CODEC_ID_NONE };
2664     static enum CodecID dv_audio_list[] = { CODEC_ID_PCM_S16LE, CODEC_ID_NONE };
2665
2666     *video_codec_list = dv_video_list;
2667     *audio_codec_list = dv_audio_list;
2668   } else if (!strcmp (format_name, "mov")) {
2669     static enum CodecID mov_video_list[] = {
2670       CODEC_ID_SVQ1, CODEC_ID_SVQ3, CODEC_ID_MPEG4,
2671       CODEC_ID_H263, CODEC_ID_H263P,
2672       CODEC_ID_H264, CODEC_ID_DVVIDEO,
2673       CODEC_ID_MJPEG,
2674       CODEC_ID_NONE
2675     };
2676     static enum CodecID mov_audio_list[] = {
2677       CODEC_ID_PCM_MULAW, CODEC_ID_PCM_ALAW, CODEC_ID_ADPCM_IMA_QT,
2678       CODEC_ID_MACE3, CODEC_ID_MACE6, CODEC_ID_AAC,
2679       CODEC_ID_AMR_NB, CODEC_ID_AMR_WB,
2680       CODEC_ID_PCM_S16BE, CODEC_ID_PCM_S16LE,
2681       CODEC_ID_MP3, CODEC_ID_NONE
2682     };
2683
2684     *video_codec_list = mov_video_list;
2685     *audio_codec_list = mov_audio_list;
2686   } else if ((!strcmp (format_name, "3gp") || !strcmp (format_name, "3g2"))) {
2687     static enum CodecID tgp_video_list[] = {
2688       CODEC_ID_MPEG4, CODEC_ID_H263, CODEC_ID_H263P, CODEC_ID_H264,
2689       CODEC_ID_NONE
2690     };
2691     static enum CodecID tgp_audio_list[] = {
2692       CODEC_ID_AMR_NB, CODEC_ID_AMR_WB,
2693       CODEC_ID_AAC,
2694       CODEC_ID_NONE
2695     };
2696
2697     *video_codec_list = tgp_video_list;
2698     *audio_codec_list = tgp_audio_list;
2699   } else if (!strcmp (format_name, "mmf")) {
2700     static enum CodecID mmf_audio_list[] = {
2701       CODEC_ID_ADPCM_YAMAHA, CODEC_ID_NONE
2702     };
2703     *video_codec_list = NULL;
2704     *audio_codec_list = mmf_audio_list;
2705   } else if (!strcmp (format_name, "amr")) {
2706     static enum CodecID amr_audio_list[] = {
2707       CODEC_ID_AMR_NB, CODEC_ID_AMR_WB,
2708       CODEC_ID_NONE
2709     };
2710     *video_codec_list = NULL;
2711     *audio_codec_list = amr_audio_list;
2712   } else if (!strcmp (format_name, "gif")) {
2713     static enum CodecID gif_image_list[] = {
2714       CODEC_ID_RAWVIDEO, CODEC_ID_NONE
2715     };
2716     *video_codec_list = gif_image_list;
2717     *audio_codec_list = NULL;
2718   } else if ((plugin->audio_codec != CODEC_ID_NONE) ||
2719       (plugin->video_codec != CODEC_ID_NONE)) {
2720     tmp_vlist[0] = plugin->video_codec;
2721     tmp_alist[0] = plugin->audio_codec;
2722
2723     *video_codec_list = tmp_vlist;
2724     *audio_codec_list = tmp_alist;
2725   } else {
2726     GST_LOG ("Format %s not found", format_name);
2727     return FALSE;
2728   }
2729
2730   return TRUE;
2731 }
2732
2733 /* Convert a GstCaps to a FFMPEG codec ID. Size et all
2734  * are omitted, that can be queried by the user itself,
2735  * we're not eating the GstCaps or anything
2736  * A pointer to an allocated context is also needed for
2737  * optional extra info
2738  */
2739
2740 enum CodecID
2741 gst_ffmpeg_caps_to_codecid (const GstCaps * caps, AVCodecContext * context)
2742 {
2743   enum CodecID id = CODEC_ID_NONE;
2744   const gchar *mimetype;
2745   const GstStructure *structure;
2746   gboolean video = FALSE, audio = FALSE;        /* we want to be sure! */
2747
2748   g_return_val_if_fail (caps != NULL, CODEC_ID_NONE);
2749   g_return_val_if_fail (gst_caps_get_size (caps) == 1, CODEC_ID_NONE);
2750   structure = gst_caps_get_structure (caps, 0);
2751
2752   mimetype = gst_structure_get_name (structure);
2753
2754   if (!strcmp (mimetype, "video/x-raw")) {
2755     id = CODEC_ID_RAWVIDEO;
2756     video = TRUE;
2757   } else if (!strcmp (mimetype, "audio/x-raw")) {
2758     GstAudioInfo info;
2759
2760     if (gst_audio_info_from_caps (&info, caps)) {
2761       switch (GST_AUDIO_INFO_FORMAT (&info)) {
2762         case GST_AUDIO_FORMAT_S8:
2763           id = CODEC_ID_PCM_S8;
2764           break;
2765         case GST_AUDIO_FORMAT_U8:
2766           id = CODEC_ID_PCM_U8;
2767           break;
2768         case GST_AUDIO_FORMAT_S16LE:
2769           id = CODEC_ID_PCM_S16LE;
2770           break;
2771         case GST_AUDIO_FORMAT_S16BE:
2772           id = CODEC_ID_PCM_S16BE;
2773           break;
2774         case GST_AUDIO_FORMAT_U16LE:
2775           id = CODEC_ID_PCM_U16LE;
2776           break;
2777         case GST_AUDIO_FORMAT_U16BE:
2778           id = CODEC_ID_PCM_U16BE;
2779           break;
2780         default:
2781           break;
2782       }
2783       if (id != CODEC_ID_NONE)
2784         audio = TRUE;
2785     }
2786   } else if (!strcmp (mimetype, "audio/x-mulaw")) {
2787     id = CODEC_ID_PCM_MULAW;
2788     audio = TRUE;
2789   } else if (!strcmp (mimetype, "audio/x-alaw")) {
2790     id = CODEC_ID_PCM_ALAW;
2791     audio = TRUE;
2792   } else if (!strcmp (mimetype, "video/x-dv")) {
2793     gboolean sys_strm;
2794
2795     if (gst_structure_get_boolean (structure, "systemstream", &sys_strm) &&
2796         !sys_strm) {
2797       id = CODEC_ID_DVVIDEO;
2798       video = TRUE;
2799     }
2800   } else if (!strcmp (mimetype, "audio/x-dv")) {        /* ??? */
2801     id = CODEC_ID_DVAUDIO;
2802     audio = TRUE;
2803   } else if (!strcmp (mimetype, "video/x-h263")) {
2804     const gchar *h263version =
2805         gst_structure_get_string (structure, "h263version");
2806     if (h263version && !strcmp (h263version, "h263p"))
2807       id = CODEC_ID_H263P;
2808     else
2809       id = CODEC_ID_H263;
2810     video = TRUE;
2811   } else if (!strcmp (mimetype, "video/x-intel-h263")) {
2812     id = CODEC_ID_H263I;
2813     video = TRUE;
2814   } else if (!strcmp (mimetype, "video/x-h261")) {
2815     id = CODEC_ID_H261;
2816     video = TRUE;
2817   } else if (!strcmp (mimetype, "video/mpeg")) {
2818     gboolean sys_strm;
2819     gint mpegversion;
2820
2821     if (gst_structure_get_boolean (structure, "systemstream", &sys_strm) &&
2822         gst_structure_get_int (structure, "mpegversion", &mpegversion) &&
2823         !sys_strm) {
2824       switch (mpegversion) {
2825         case 1:
2826           id = CODEC_ID_MPEG1VIDEO;
2827           break;
2828         case 2:
2829           id = CODEC_ID_MPEG2VIDEO;
2830           break;
2831         case 4:
2832           id = CODEC_ID_MPEG4;
2833           break;
2834       }
2835     }
2836     if (id != CODEC_ID_NONE)
2837       video = TRUE;
2838   } else if (!strcmp (mimetype, "image/jpeg")) {
2839     id = CODEC_ID_MJPEG;        /* A... B... */
2840     video = TRUE;
2841   } else if (!strcmp (mimetype, "video/x-jpeg-b")) {
2842     id = CODEC_ID_MJPEGB;
2843     video = TRUE;
2844   } else if (!strcmp (mimetype, "video/x-wmv")) {
2845     gint wmvversion = 0;
2846
2847     if (gst_structure_get_int (structure, "wmvversion", &wmvversion)) {
2848       switch (wmvversion) {
2849         case 1:
2850           id = CODEC_ID_WMV1;
2851           break;
2852         case 2:
2853           id = CODEC_ID_WMV2;
2854           break;
2855         case 3:
2856         {
2857           const gchar *format;
2858
2859           /* WMV3 unless the fourcc exists and says otherwise */
2860           id = CODEC_ID_WMV3;
2861           format = gst_structure_get_string (structure, "format");
2862
2863           if (g_str_equal (format, "WVC1") || g_str_equal (format, "WMVA")) {
2864             id = CODEC_ID_VC1;
2865           }
2866           break;
2867         }
2868       }
2869     }
2870     if (id != CODEC_ID_NONE)
2871       video = TRUE;
2872   } else if (!strcmp (mimetype, "audio/x-vorbis")) {
2873     id = CODEC_ID_VORBIS;
2874     audio = TRUE;
2875   } else if (!strcmp (mimetype, "audio/x-qdm2")) {
2876     id = CODEC_ID_QDM2;
2877     audio = TRUE;
2878   } else if (!strcmp (mimetype, "audio/mpeg")) {
2879     gint layer = 0;
2880     gint mpegversion = 0;
2881
2882     if (gst_structure_get_int (structure, "mpegversion", &mpegversion)) {
2883       switch (mpegversion) {
2884         case 2:                /* ffmpeg uses faad for both... */
2885         case 4:
2886           id = CODEC_ID_AAC;
2887           break;
2888         case 1:
2889           if (gst_structure_get_int (structure, "layer", &layer)) {
2890             switch (layer) {
2891               case 1:
2892                 id = CODEC_ID_MP1;
2893                 break;
2894               case 2:
2895                 id = CODEC_ID_MP2;
2896                 break;
2897               case 3:
2898                 id = CODEC_ID_MP3;
2899                 break;
2900             }
2901           }
2902       }
2903     }
2904     if (id != CODEC_ID_NONE)
2905       audio = TRUE;
2906   } else if (!strcmp (mimetype, "audio/x-musepack")) {
2907     gint streamversion = -1;
2908
2909     if (gst_structure_get_int (structure, "streamversion", &streamversion)) {
2910       if (streamversion == 7)
2911         id = CODEC_ID_MUSEPACK7;
2912     } else {
2913       id = CODEC_ID_MUSEPACK7;
2914     }
2915   } else if (!strcmp (mimetype, "audio/x-wma")) {
2916     gint wmaversion = 0;
2917
2918     if (gst_structure_get_int (structure, "wmaversion", &wmaversion)) {
2919       switch (wmaversion) {
2920         case 1:
2921           id = CODEC_ID_WMAV1;
2922           break;
2923         case 2:
2924           id = CODEC_ID_WMAV2;
2925           break;
2926         case 3:
2927           id = CODEC_ID_WMAPRO;
2928           break;
2929       }
2930     }
2931     if (id != CODEC_ID_NONE)
2932       audio = TRUE;
2933   } else if (!strcmp (mimetype, "audio/x-wms")) {
2934     id = CODEC_ID_WMAVOICE;
2935     audio = TRUE;
2936   } else if (!strcmp (mimetype, "audio/x-ac3")) {
2937     id = CODEC_ID_AC3;
2938     audio = TRUE;
2939   } else if (!strcmp (mimetype, "audio/x-eac3")) {
2940     id = CODEC_ID_EAC3;
2941     audio = TRUE;
2942   } else if (!strcmp (mimetype, "audio/x-vnd.sony.atrac3") ||
2943       !strcmp (mimetype, "audio/atrac3")) {
2944     id = CODEC_ID_ATRAC3;
2945     audio = TRUE;
2946   } else if (!strcmp (mimetype, "audio/x-dts")) {
2947     id = CODEC_ID_DTS;
2948     audio = TRUE;
2949   } else if (!strcmp (mimetype, "application/x-ape")) {
2950     id = CODEC_ID_APE;
2951     audio = TRUE;
2952   } else if (!strcmp (mimetype, "video/x-msmpeg")) {
2953     gint msmpegversion = 0;
2954
2955     if (gst_structure_get_int (structure, "msmpegversion", &msmpegversion)) {
2956       switch (msmpegversion) {
2957         case 41:
2958           id = CODEC_ID_MSMPEG4V1;
2959           break;
2960         case 42:
2961           id = CODEC_ID_MSMPEG4V2;
2962           break;
2963         case 43:
2964           id = CODEC_ID_MSMPEG4V3;
2965           break;
2966       }
2967     }
2968     if (id != CODEC_ID_NONE)
2969       video = TRUE;
2970   } else if (!strcmp (mimetype, "video/x-svq")) {
2971     gint svqversion = 0;
2972
2973     if (gst_structure_get_int (structure, "svqversion", &svqversion)) {
2974       switch (svqversion) {
2975         case 1:
2976           id = CODEC_ID_SVQ1;
2977           break;
2978         case 3:
2979           id = CODEC_ID_SVQ3;
2980           break;
2981       }
2982     }
2983     if (id != CODEC_ID_NONE)
2984       video = TRUE;
2985   } else if (!strcmp (mimetype, "video/x-huffyuv")) {
2986     id = CODEC_ID_HUFFYUV;
2987     video = TRUE;
2988   } else if (!strcmp (mimetype, "audio/x-mace")) {
2989     gint maceversion = 0;
2990
2991     if (gst_structure_get_int (structure, "maceversion", &maceversion)) {
2992       switch (maceversion) {
2993         case 3:
2994           id = CODEC_ID_MACE3;
2995           break;
2996         case 6:
2997           id = CODEC_ID_MACE6;
2998           break;
2999       }
3000     }
3001     if (id != CODEC_ID_NONE)
3002       audio = TRUE;
3003   } else if (!strcmp (mimetype, "video/x-theora")) {
3004     id = CODEC_ID_THEORA;
3005     video = TRUE;
3006   } else if (!strcmp (mimetype, "video/x-vp3")) {
3007     id = CODEC_ID_VP3;
3008     video = TRUE;
3009   } else if (!strcmp (mimetype, "video/x-vp5")) {
3010     id = CODEC_ID_VP5;
3011     video = TRUE;
3012   } else if (!strcmp (mimetype, "video/x-vp6")) {
3013     id = CODEC_ID_VP6;
3014     video = TRUE;
3015   } else if (!strcmp (mimetype, "video/x-vp6-flash")) {
3016     id = CODEC_ID_VP6F;
3017     video = TRUE;
3018   } else if (!strcmp (mimetype, "video/x-vp6-alpha")) {
3019     id = CODEC_ID_VP6A;
3020     video = TRUE;
3021   } else if (!strcmp (mimetype, "video/x-vp8")) {
3022     id = CODEC_ID_VP8;
3023     video = TRUE;
3024   } else if (!strcmp (mimetype, "video/x-flash-screen")) {
3025     id = CODEC_ID_FLASHSV;
3026     video = TRUE;
3027   } else if (!strcmp (mimetype, "video/x-indeo")) {
3028     gint indeoversion = 0;
3029
3030     if (gst_structure_get_int (structure, "indeoversion", &indeoversion)) {
3031       switch (indeoversion) {
3032         case 5:
3033           id = CODEC_ID_INDEO5;
3034           break;
3035         case 3:
3036           id = CODEC_ID_INDEO3;
3037           break;
3038         case 2:
3039           id = CODEC_ID_INDEO2;
3040           break;
3041       }
3042       if (id != CODEC_ID_NONE)
3043         video = TRUE;
3044     }
3045   } else if (!strcmp (mimetype, "video/x-divx")) {
3046     gint divxversion = 0;
3047
3048     if (gst_structure_get_int (structure, "divxversion", &divxversion)) {
3049       switch (divxversion) {
3050         case 3:
3051           id = CODEC_ID_MSMPEG4V3;
3052           break;
3053         case 4:
3054         case 5:
3055           id = CODEC_ID_MPEG4;
3056           break;
3057       }
3058     }
3059     if (id != CODEC_ID_NONE)
3060       video = TRUE;
3061   } else if (!strcmp (mimetype, "video/x-3ivx")) {
3062     id = CODEC_ID_MPEG4;
3063     video = TRUE;
3064   } else if (!strcmp (mimetype, "video/x-xvid")) {
3065     id = CODEC_ID_MPEG4;
3066     video = TRUE;
3067   } else if (!strcmp (mimetype, "video/x-ffv")) {
3068     gint ffvversion = 0;
3069
3070     if (gst_structure_get_int (structure, "ffvversion", &ffvversion) &&
3071         ffvversion == 1) {
3072       id = CODEC_ID_FFV1;
3073       video = TRUE;
3074     }
3075   } else if (!strcmp (mimetype, "audio/x-adpcm")) {
3076     const gchar *layout;
3077
3078     layout = gst_structure_get_string (structure, "layout");
3079     if (layout == NULL) {
3080       /* break */
3081     } else if (!strcmp (layout, "quicktime")) {
3082       id = CODEC_ID_ADPCM_IMA_QT;
3083     } else if (!strcmp (layout, "microsoft")) {
3084       id = CODEC_ID_ADPCM_MS;
3085     } else if (!strcmp (layout, "dvi")) {
3086       id = CODEC_ID_ADPCM_IMA_WAV;
3087     } else if (!strcmp (layout, "4xm")) {
3088       id = CODEC_ID_ADPCM_4XM;
3089     } else if (!strcmp (layout, "smjpeg")) {
3090       id = CODEC_ID_ADPCM_IMA_SMJPEG;
3091     } else if (!strcmp (layout, "dk3")) {
3092       id = CODEC_ID_ADPCM_IMA_DK3;
3093     } else if (!strcmp (layout, "dk4")) {
3094       id = CODEC_ID_ADPCM_IMA_DK4;
3095     } else if (!strcmp (layout, "westwood")) {
3096       id = CODEC_ID_ADPCM_IMA_WS;
3097     } else if (!strcmp (layout, "iss")) {
3098       id = CODEC_ID_ADPCM_IMA_ISS;
3099     } else if (!strcmp (layout, "xa")) {
3100       id = CODEC_ID_ADPCM_XA;
3101     } else if (!strcmp (layout, "adx")) {
3102       id = CODEC_ID_ADPCM_ADX;
3103     } else if (!strcmp (layout, "ea")) {
3104       id = CODEC_ID_ADPCM_EA;
3105     } else if (!strcmp (layout, "g726")) {
3106       id = CODEC_ID_ADPCM_G726;
3107     } else if (!strcmp (layout, "g721")) {
3108       id = CODEC_ID_ADPCM_G726;
3109     } else if (!strcmp (layout, "ct")) {
3110       id = CODEC_ID_ADPCM_CT;
3111     } else if (!strcmp (layout, "swf")) {
3112       id = CODEC_ID_ADPCM_SWF;
3113     } else if (!strcmp (layout, "yamaha")) {
3114       id = CODEC_ID_ADPCM_YAMAHA;
3115     } else if (!strcmp (layout, "sbpro2")) {
3116       id = CODEC_ID_ADPCM_SBPRO_2;
3117     } else if (!strcmp (layout, "sbpro3")) {
3118       id = CODEC_ID_ADPCM_SBPRO_3;
3119     } else if (!strcmp (layout, "sbpro4")) {
3120       id = CODEC_ID_ADPCM_SBPRO_4;
3121     }
3122     if (id != CODEC_ID_NONE)
3123       audio = TRUE;
3124   } else if (!strcmp (mimetype, "video/x-4xm")) {
3125     id = CODEC_ID_4XM;
3126     video = TRUE;
3127   } else if (!strcmp (mimetype, "audio/x-dpcm")) {
3128     const gchar *layout;
3129
3130     layout = gst_structure_get_string (structure, "layout");
3131     if (!layout) {
3132       /* .. */
3133     } else if (!strcmp (layout, "roq")) {
3134       id = CODEC_ID_ROQ_DPCM;
3135     } else if (!strcmp (layout, "interplay")) {
3136       id = CODEC_ID_INTERPLAY_DPCM;
3137     } else if (!strcmp (layout, "xan")) {
3138       id = CODEC_ID_XAN_DPCM;
3139     } else if (!strcmp (layout, "sol")) {
3140       id = CODEC_ID_SOL_DPCM;
3141     }
3142     if (id != CODEC_ID_NONE)
3143       audio = TRUE;
3144   } else if (!strcmp (mimetype, "audio/x-flac")) {
3145     id = CODEC_ID_FLAC;
3146     audio = TRUE;
3147   } else if (!strcmp (mimetype, "audio/x-shorten")) {
3148     id = CODEC_ID_SHORTEN;
3149     audio = TRUE;
3150   } else if (!strcmp (mimetype, "audio/x-alac")) {
3151     id = CODEC_ID_ALAC;
3152     audio = TRUE;
3153   } else if (!strcmp (mimetype, "video/x-cinepak")) {
3154     id = CODEC_ID_CINEPAK;
3155     video = TRUE;
3156   } else if (!strcmp (mimetype, "video/x-pn-realvideo")) {
3157     gint rmversion;
3158
3159     if (gst_structure_get_int (structure, "rmversion", &rmversion)) {
3160       switch (rmversion) {
3161         case 1:
3162           id = CODEC_ID_RV10;
3163           break;
3164         case 2:
3165           id = CODEC_ID_RV20;
3166           break;
3167         case 3:
3168           id = CODEC_ID_RV30;
3169           break;
3170         case 4:
3171           id = CODEC_ID_RV40;
3172           break;
3173       }
3174     }
3175     if (id != CODEC_ID_NONE)
3176       video = TRUE;
3177   } else if (!strcmp (mimetype, "audio/x-sipro")) {
3178     id = CODEC_ID_SIPR;
3179     audio = TRUE;
3180   } else if (!strcmp (mimetype, "audio/x-pn-realaudio")) {
3181     gint raversion;
3182
3183     if (gst_structure_get_int (structure, "raversion", &raversion)) {
3184       switch (raversion) {
3185         case 1:
3186           id = CODEC_ID_RA_144;
3187           break;
3188         case 2:
3189           id = CODEC_ID_RA_288;
3190           break;
3191         case 8:
3192           id = CODEC_ID_COOK;
3193           break;
3194       }
3195     }
3196     if (id != CODEC_ID_NONE)
3197       audio = TRUE;
3198   } else if (!strcmp (mimetype, "video/x-rle")) {
3199     const gchar *layout;
3200
3201     if ((layout = gst_structure_get_string (structure, "layout"))) {
3202       if (!strcmp (layout, "microsoft")) {
3203         id = CODEC_ID_MSRLE;
3204         video = TRUE;
3205       }
3206     }
3207   } else if (!strcmp (mimetype, "video/x-xan")) {
3208     gint wcversion = 0;
3209
3210     if ((gst_structure_get_int (structure, "wcversion", &wcversion))) {
3211       switch (wcversion) {
3212         case 3:
3213           id = CODEC_ID_XAN_WC3;
3214           video = TRUE;
3215           break;
3216         case 4:
3217           id = CODEC_ID_XAN_WC4;
3218           video = TRUE;
3219           break;
3220         default:
3221           break;
3222       }
3223     }
3224   } else if (!strcmp (mimetype, "audio/AMR")) {
3225     audio = TRUE;
3226     id = CODEC_ID_AMR_NB;
3227   } else if (!strcmp (mimetype, "audio/AMR-WB")) {
3228     id = CODEC_ID_AMR_WB;
3229     audio = TRUE;
3230   } else if (!strcmp (mimetype, "audio/qcelp")) {
3231     id = CODEC_ID_QCELP;
3232     audio = TRUE;
3233   } else if (!strcmp (mimetype, "video/x-h264")) {
3234     id = CODEC_ID_H264;
3235     video = TRUE;
3236   } else if (!strcmp (mimetype, "video/x-flash-video")) {
3237     gint flvversion = 0;
3238
3239     if ((gst_structure_get_int (structure, "flvversion", &flvversion))) {
3240       switch (flvversion) {
3241         case 1:
3242           id = CODEC_ID_FLV1;
3243           video = TRUE;
3244           break;
3245         default:
3246           break;
3247       }
3248     }
3249
3250   } else if (!strcmp (mimetype, "audio/x-nellymoser")) {
3251     id = CODEC_ID_NELLYMOSER;
3252     audio = TRUE;
3253   } else if (!strncmp (mimetype, "audio/x-gst_ff-", 15)) {
3254     gchar ext[16];
3255     AVCodec *codec;
3256
3257     if (strlen (mimetype) <= 30 &&
3258         sscanf (mimetype, "audio/x-gst_ff-%s", ext) == 1) {
3259       if ((codec = avcodec_find_decoder_by_name (ext)) ||
3260           (codec = avcodec_find_encoder_by_name (ext))) {
3261         id = codec->id;
3262         audio = TRUE;
3263       }
3264     }
3265   } else if (!strncmp (mimetype, "video/x-gst_ff-", 15)) {
3266     gchar ext[16];
3267     AVCodec *codec;
3268
3269     if (strlen (mimetype) <= 30 &&
3270         sscanf (mimetype, "video/x-gst_ff-%s", ext) == 1) {
3271       if ((codec = avcodec_find_decoder_by_name (ext)) ||
3272           (codec = avcodec_find_encoder_by_name (ext))) {
3273         id = codec->id;
3274         video = TRUE;
3275       }
3276     }
3277   }
3278
3279   if (context != NULL) {
3280     if (video == TRUE) {
3281       context->codec_type = AVMEDIA_TYPE_VIDEO;
3282     } else if (audio == TRUE) {
3283       context->codec_type = AVMEDIA_TYPE_AUDIO;
3284     } else {
3285       context->codec_type = AVMEDIA_TYPE_UNKNOWN;
3286     }
3287     context->codec_id = id;
3288     gst_ffmpeg_caps_with_codecid (id, context->codec_type, caps, context);
3289   }
3290
3291   if (id != CODEC_ID_NONE) {
3292     GST_DEBUG ("The id=%d belongs to the caps %" GST_PTR_FORMAT, id, caps);
3293   } else {
3294     GST_WARNING ("Couldn't figure out the id for caps %" GST_PTR_FORMAT, caps);
3295   }
3296
3297   return id;
3298 }