ext/libav/gstavcodecmap.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., 51 Franklin St, Fifth Floor,
  19  * Boston, MA 02110-1301, USA.
  20  */
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include "config.h"
  24 #endif
  25
  26 #include <string.h>
  27
  28 #include <gst/gst.h>
  29 #include <libavcodec/avcodec.h>
  30 #include <libavutil/channel_layout.h>
  31
  32 #include "gstav.h"
  33 #include "gstavcodecmap.h"
  34
  35 #include <gst/video/video.h>
  36 #include <gst/audio/audio.h>
  37 #include <gst/pbutils/codec-utils.h>
  38
  39 /* IMPORTANT: Keep this sorted by the ffmpeg channel masks */
  40 static const struct
  41 {
  42   guint64 ff;
  43   GstAudioChannelPosition gst;
  44 } _ff_to_gst_layout[] = {
  45   {
  46   AV_CH_FRONT_LEFT, GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT}, {
  47   AV_CH_FRONT_RIGHT, GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT}, {
  48   AV_CH_FRONT_CENTER, GST_AUDIO_CHANNEL_POSITION_FRONT_CENTER}, {
  49   AV_CH_LOW_FREQUENCY, GST_AUDIO_CHANNEL_POSITION_LFE1}, {
  50   AV_CH_BACK_LEFT, GST_AUDIO_CHANNEL_POSITION_REAR_LEFT}, {
  51   AV_CH_BACK_RIGHT, GST_AUDIO_CHANNEL_POSITION_REAR_RIGHT}, {
  52   AV_CH_FRONT_LEFT_OF_CENTER, GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT_OF_CENTER}, {
  53   AV_CH_FRONT_RIGHT_OF_CENTER,
  54         GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT_OF_CENTER}, {
  55   AV_CH_BACK_CENTER, GST_AUDIO_CHANNEL_POSITION_REAR_CENTER}, {
  56   AV_CH_SIDE_LEFT, GST_AUDIO_CHANNEL_POSITION_SIDE_LEFT}, {
  57   AV_CH_SIDE_RIGHT, GST_AUDIO_CHANNEL_POSITION_SIDE_RIGHT}, {
  58   AV_CH_TOP_CENTER, GST_AUDIO_CHANNEL_POSITION_TOP_CENTER}, {
  59   AV_CH_TOP_FRONT_LEFT, GST_AUDIO_CHANNEL_POSITION_TOP_FRONT_LEFT}, {
  60   AV_CH_TOP_FRONT_CENTER, GST_AUDIO_CHANNEL_POSITION_TOP_FRONT_CENTER}, {
  61   AV_CH_TOP_FRONT_RIGHT, GST_AUDIO_CHANNEL_POSITION_TOP_FRONT_RIGHT}, {
  62   AV_CH_TOP_BACK_LEFT, GST_AUDIO_CHANNEL_POSITION_TOP_REAR_LEFT}, {
  63   AV_CH_TOP_BACK_CENTER, GST_AUDIO_CHANNEL_POSITION_TOP_REAR_CENTER}, {
  64   AV_CH_TOP_BACK_RIGHT, GST_AUDIO_CHANNEL_POSITION_TOP_REAR_RIGHT}, {
  65   AV_CH_STEREO_LEFT, GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT}, {
  66   AV_CH_STEREO_RIGHT, GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT}
  67 };
  68
  69 static guint64
  70 gst_ffmpeg_channel_positions_to_layout (GstAudioChannelPosition * pos,
  71     gint channels)
  72 {
  73   gint i, j;
  74   guint64 ret = 0;
  75   gint channels_found = 0;
  76
  77   if (!pos)
  78     return 0;
  79
  80   for (i = 0; i < channels; i++) {
  81     for (j = 0; j < G_N_ELEMENTS (_ff_to_gst_layout); j++) {
  82       if (_ff_to_gst_layout[j].gst == pos[i]) {
  83         ret |= _ff_to_gst_layout[j].ff;
  84         channels_found++;
  85         break;
  86       }
  87     }
  88   }
  89
  90   if (channels_found != channels)
  91     return 0;
  92   return ret;
  93 }
  94
  95 gboolean
  96 gst_ffmpeg_channel_layout_to_gst (guint64 channel_layout, gint channels,
  97     GstAudioChannelPosition * pos)
  98 {
  99   guint nchannels = 0;
 100   gboolean none_layout = FALSE;
 101
 102   if (channel_layout == 0) {
 103     nchannels = channels;
 104     none_layout = TRUE;
 105   } else {
 106     guint i, j;
 107
 108     for (i = 0; i < 64; i++) {
 109       if ((channel_layout & (G_GUINT64_CONSTANT (1) << i)) != 0) {
 110         nchannels++;
 111       }
 112     }
 113
 114     if (nchannels != channels) {
 115       GST_ERROR ("Number of channels is different (%u != %u)", channels,
 116           nchannels);
 117       nchannels = channels;
 118       none_layout = TRUE;
 119     } else {
 120
 121       for (i = 0, j = 0; i < G_N_ELEMENTS (_ff_to_gst_layout); i++) {
 122         if ((channel_layout & _ff_to_gst_layout[i].ff) != 0) {
 123           pos[j++] = _ff_to_gst_layout[i].gst;
 124
 125           if (_ff_to_gst_layout[i].gst == GST_AUDIO_CHANNEL_POSITION_NONE)
 126             none_layout = TRUE;
 127         }
 128       }
 129
 130       if (j != nchannels) {
 131         GST_WARNING
 132             ("Unknown channels in channel layout - assuming NONE layout");
 133         none_layout = TRUE;
 134       }
 135     }
 136   }
 137
 138   if (!none_layout
 139       && !gst_audio_check_valid_channel_positions (pos, nchannels, FALSE)) {
 140     GST_ERROR ("Invalid channel layout %" G_GUINT64_FORMAT
 141         " - assuming NONE layout", channel_layout);
 142     none_layout = TRUE;
 143   }
 144
 145   if (none_layout) {
 146     if (nchannels == 1) {
 147       pos[0] = GST_AUDIO_CHANNEL_POSITION_MONO;
 148     } else if (nchannels == 2) {
 149       pos[0] = GST_AUDIO_CHANNEL_POSITION_FRONT_LEFT;
 150       pos[1] = GST_AUDIO_CHANNEL_POSITION_FRONT_RIGHT;
 151     } else {
 152       guint i;
 153
 154       for (i = 0; i < nchannels; i++)
 155         pos[i] = GST_AUDIO_CHANNEL_POSITION_NONE;
 156     }
 157   }
 158
 159   return TRUE;
 160 }
 161
 162 static void
 163 gst_ffmpeg_video_set_pix_fmts (GstCaps * caps, const enum AVPixelFormat *fmts)
 164 {
 165   GValue va = { 0, };
 166   GValue v = { 0, };
 167   GstVideoFormat format;
 168
 169   if (!fmts || fmts[0] == -1) {
 170     gint i;
 171
 172     g_value_init (&va, GST_TYPE_LIST);
 173     g_value_init (&v, G_TYPE_STRING);
 174     for (i = 0; i <= PIX_FMT_NB; i++) {
 175       format = gst_ffmpeg_pixfmt_to_videoformat (i);
 176       if (format == GST_VIDEO_FORMAT_UNKNOWN)
 177         continue;
 178       g_value_set_string (&v, gst_video_format_to_string (format));
 179       gst_value_list_append_value (&va, &v);
 180     }
 181     gst_caps_set_value (caps, "format", &va);
 182     g_value_unset (&v);
 183     g_value_unset (&va);
 184     return;
 185   }
 186
 187   /* Only a single format */
 188   g_value_init (&va, GST_TYPE_LIST);
 189   g_value_init (&v, G_TYPE_STRING);
 190   while (*fmts != -1) {
 191     format = gst_ffmpeg_pixfmt_to_videoformat (*fmts);
 192     if (format != GST_VIDEO_FORMAT_UNKNOWN) {
 193       g_value_set_string (&v, gst_video_format_to_string (format));
 194       gst_value_list_append_value (&va, &v);
 195     }
 196     fmts++;
 197   }
 198   if (gst_value_list_get_size (&va) == 1) {
 199     /* The single value is still in v */
 200     gst_caps_set_value (caps, "format", &v);
 201   } else if (gst_value_list_get_size (&va) > 1) {
 202     gst_caps_set_value (caps, "format", &va);
 203   }
 204   g_value_unset (&v);
 205   g_value_unset (&va);
 206 }
 207
 208 /* this macro makes a caps width fixed or unfixed width/height
 209  * properties depending on whether we've got a context.
 210  *
 211  * See below for why we use this.
 212  *
 213  * We should actually do this stuff at the end, like in riff-media.c,
 214  * but I'm too lazy today. Maybe later.
 215  */
 216 static GstCaps *
 217 gst_ff_vid_caps_new (AVCodecContext * context, AVCodec * codec,
 218     enum CodecID codec_id, gboolean encode, const char *mimetype,
 219     const char *fieldname, ...)
 220 {
 221   GstCaps *caps = NULL;
 222   va_list var_args;
 223   gint i;
 224
 225   GST_LOG ("context:%p, codec_id:%d, mimetype:%s", context, codec_id, mimetype);
 226
 227   /* fixed, non probing context */
 228   if (context != NULL && context->width != -1) {
 229     gint num, denom;
 230
 231     caps = gst_caps_new_simple (mimetype,
 232         "width", G_TYPE_INT, context->width,
 233         "height", G_TYPE_INT, context->height, NULL);
 234
 235     num = context->time_base.den / context->ticks_per_frame;
 236     denom = context->time_base.num;
 237
 238     if (!denom) {
 239       GST_LOG ("invalid framerate: %d/0, -> %d/1", num, num);
 240       denom = 1;
 241     }
 242     if (gst_util_fraction_compare (num, denom, 1000, 1) > 0) {
 243       GST_LOG ("excessive framerate: %d/%d, -> 0/1", num, denom);
 244       num = 0;
 245       denom = 1;
 246     }
 247     GST_LOG ("setting framerate: %d/%d", num, denom);
 248     gst_caps_set_simple (caps,
 249         "framerate", GST_TYPE_FRACTION, num, denom, NULL);
 250   } else if (encode) {
 251     /* so we are after restricted caps in this case */
 252     switch (codec_id) {
 253       case AV_CODEC_ID_H261:
 254       {
 255         caps = gst_caps_new_simple (mimetype,
 256             "width", G_TYPE_INT, 352,
 257             "height", G_TYPE_INT, 288,
 258             "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL);
 259         gst_caps_append (caps, gst_caps_new_simple (mimetype,
 260                 "width", G_TYPE_INT, 176,
 261                 "height", G_TYPE_INT, 144,
 262                 "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL));
 263         break;
 264       }
 265       case AV_CODEC_ID_H263:
 266       {
 267         /* 128x96, 176x144, 352x288, 704x576, and 1408x1152. slightly reordered
 268          * because we want automatic negotiation to go as close to 320x240 as
 269          * possible. */
 270         const static gint widths[] = { 352, 704, 176, 1408, 128 };
 271         const static gint heights[] = { 288, 576, 144, 1152, 96 };
 272         GstCaps *temp;
 273         gint n_sizes = G_N_ELEMENTS (widths);
 274
 275         caps = gst_caps_new_empty ();
 276         for (i = 0; i < n_sizes; i++) {
 277           temp = gst_caps_new_simple (mimetype,
 278               "width", G_TYPE_INT, widths[i],
 279               "height", G_TYPE_INT, heights[i],
 280               "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL);
 281
 282           gst_caps_append (caps, temp);
 283         }
 284         break;
 285       }
 286       case AV_CODEC_ID_DVVIDEO:
 287       {
 288         static struct
 289         {
 290           const gchar *csp;
 291           gint width, height;
 292           gint par_n, par_d;
 293           gint framerate_n, framerate_d;
 294         } profiles[] = {
 295           {
 296           "Y41B", 720, 480, 8, 9, 30000, 1001}, {
 297           "Y41B", 720, 480, 32, 27, 30000, 1001}, {
 298           "Y42B", 720, 480, 8, 9, 30000, 1001}, {
 299           "Y42B", 720, 480, 32, 27, 30000, 1001}, {
 300           "I420", 720, 576, 16, 15, 25, 1}, {
 301           "I420", 720, 576, 64, 45, 25, 1}, {
 302           "Y41B", 720, 576, 16, 15, 25, 1}, {
 303           "Y41B", 720, 576, 64, 45, 25, 1}, {
 304           "Y42B", 720, 576, 16, 15, 25, 1}, {
 305           "Y42B", 720, 576, 64, 45, 25, 1}, {
 306           "Y42B", 1280, 1080, 1, 1, 30000, 1001}, {
 307           "Y42B", 1280, 1080, 3, 2, 30000, 1001}, {
 308           "Y42B", 1440, 1080, 1, 1, 25, 1}, {
 309           "Y42B", 1440, 1080, 4, 3, 25, 1}, {
 310           "Y42B", 960, 720, 1, 1, 60000, 1001}, {
 311           "Y42B", 960, 720, 4, 3, 60000, 1001}, {
 312           "Y42B", 960, 720, 1, 1, 50, 1}, {
 313         "Y42B", 960, 720, 4, 3, 50, 1},};
 314         GstCaps *temp;
 315         gint n_sizes = G_N_ELEMENTS (profiles);
 316
 317         if (strcmp (mimetype, "video/x-raw") == 0) {
 318           caps = gst_caps_new_empty ();
 319           for (i = 0; i < n_sizes; i++) {
 320             temp = gst_caps_new_simple (mimetype,
 321                 "format", G_TYPE_STRING, profiles[i].csp,
 322                 "width", G_TYPE_INT, profiles[i].width,
 323                 "height", G_TYPE_INT, profiles[i].height,
 324                 "framerate", GST_TYPE_FRACTION, profiles[i].framerate_n,
 325                 profiles[i].framerate_d, "pixel-aspect-ratio",
 326                 GST_TYPE_FRACTION, profiles[i].par_n, profiles[i].par_d, NULL);
 327
 328             gst_caps_append (caps, temp);
 329           }
 330         } else {
 331           caps = gst_caps_new_empty ();
 332           for (i = 0; i < n_sizes; i++) {
 333             temp = gst_caps_new_simple (mimetype,
 334                 "width", G_TYPE_INT, profiles[i].width,
 335                 "height", G_TYPE_INT, profiles[i].height,
 336                 "framerate", GST_TYPE_FRACTION, profiles[i].framerate_n,
 337                 profiles[i].framerate_d, "pixel-aspect-ratio",
 338                 GST_TYPE_FRACTION, profiles[i].par_n, profiles[i].par_d, NULL);
 339
 340             gst_caps_append (caps, temp);
 341           }
 342         }
 343         break;
 344       }
 345       case AV_CODEC_ID_DNXHD:
 346       {
 347         caps = gst_caps_new_simple (mimetype,
 348             "width", G_TYPE_INT, 1920,
 349             "height", G_TYPE_INT, 1080,
 350             "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL);
 351         gst_caps_append (caps, gst_caps_new_simple (mimetype,
 352                 "width", G_TYPE_INT, 1280,
 353                 "height", G_TYPE_INT, 720,
 354                 "framerate", GST_TYPE_FRACTION_RANGE, 0, 1, G_MAXINT, 1, NULL));
 355         break;
 356       }
 357       default:
 358       {
 359         if (codec && codec->supported_framerates
 360             && codec->supported_framerates[0].num != 0
 361             && codec->supported_framerates[0].den != 0) {
 362           GValue va = { 0, };
 363           GValue v = { 0, };
 364           const AVRational *rates = codec->supported_framerates;
 365
 366           if (rates[1].num == 0 && rates[1].den == 0) {
 367             caps =
 368                 gst_caps_new_simple (mimetype, "framerate", GST_TYPE_FRACTION,
 369                 rates[0].num, rates[0].den, NULL);
 370           } else {
 371             g_value_init (&va, GST_TYPE_LIST);
 372             g_value_init (&v, GST_TYPE_FRACTION);
 373
 374             while (rates->num != 0 && rates->den != 0) {
 375               gst_value_set_fraction (&v, rates->num, rates->den);
 376               gst_value_list_append_value (&va, &v);
 377               rates++;
 378             }
 379
 380             caps = gst_caps_new_simple (mimetype, NULL, NULL, NULL);
 381             gst_caps_set_value (caps, "framerate", &va);
 382             g_value_unset (&va);
 383             g_value_unset (&v);
 384           }
 385
 386         } else {
 387           caps = gst_caps_new_empty_simple (mimetype);
 388         }
 389
 390         break;
 391       }
 392     }
 393   }
 394
 395   /* no fixed caps or special restrictions applied;
 396    * default unfixed setting */
 397   if (!caps) {
 398     GST_DEBUG ("Creating default caps");
 399     caps = gst_caps_new_empty_simple (mimetype);
 400   }
 401
 402   va_start (var_args, fieldname);
 403   gst_caps_set_simple_valist (caps, fieldname, var_args);
 404   va_end (var_args);
 405
 406   return caps;
 407 }
 408
 409 static gint
 410 get_nbits_set (guint64 n)
 411 {
 412   gint i, x;
 413
 414   x = 0;
 415   for (i = 0; i < 64; i++) {
 416     if ((n & (G_GUINT64_CONSTANT (1) << i)))
 417       x++;
 418   }
 419
 420   return x;
 421 }
 422
 423 static void
 424 gst_ffmpeg_audio_set_sample_fmts (GstCaps * caps,
 425     const enum AVSampleFormat *fmts)
 426 {
 427   GValue va = { 0, };
 428   GValue v = { 0, };
 429   GstAudioFormat format;
 430
 431   if (!fmts || fmts[0] == -1) {
 432     gint i;
 433
 434     g_value_init (&va, GST_TYPE_LIST);
 435     g_value_init (&v, G_TYPE_STRING);
 436     for (i = 0; i <= AV_SAMPLE_FMT_DBL; i++) {
 437       format = gst_ffmpeg_smpfmt_to_audioformat (i);
 438       if (format == GST_AUDIO_FORMAT_UNKNOWN)
 439         continue;
 440       g_value_set_string (&v, gst_audio_format_to_string (format));
 441       gst_value_list_append_value (&va, &v);
 442     }
 443     gst_caps_set_value (caps, "format", &va);
 444     g_value_unset (&v);
 445     g_value_unset (&va);
 446     return;
 447   }
 448
 449   g_value_init (&va, GST_TYPE_LIST);
 450   g_value_init (&v, G_TYPE_STRING);
 451   while (*fmts != -1) {
 452     format = gst_ffmpeg_smpfmt_to_audioformat (*fmts);
 453     if (format != GST_AUDIO_FORMAT_UNKNOWN) {
 454       g_value_set_string (&v, gst_audio_format_to_string (format));
 455       gst_value_list_append_value (&va, &v);
 456     }
 457     fmts++;
 458   }
 459   if (gst_value_list_get_size (&va) == 1) {
 460     /* The single value is still in v */
 461     gst_caps_set_value (caps, "format", &v);
 462   } else if (gst_value_list_get_size (&va) > 1) {
 463     gst_caps_set_value (caps, "format", &va);
 464   }
 465   g_value_unset (&v);
 466   g_value_unset (&va);
 467 }
 468
 469 /* same for audio - now with channels/sample rate
 470  */
 471 static GstCaps *
 472 gst_ff_aud_caps_new (AVCodecContext * context, AVCodec * codec,
 473     enum CodecID codec_id, gboolean encode, const char *mimetype,
 474     const char *fieldname, ...)
 475 {
 476   GstCaps *caps = NULL;
 477   gint i;
 478   va_list var_args;
 479
 480   /* fixed, non-probing context */
 481   if (context != NULL && context->channels != -1) {
 482     GstAudioChannelPosition pos[64];
 483
 484     caps = gst_caps_new_simple (mimetype,
 485         "rate", G_TYPE_INT, context->sample_rate,
 486         "channels", G_TYPE_INT, context->channels, NULL);
 487
 488     if (gst_ffmpeg_channel_layout_to_gst (context->channel_layout,
 489             context->channels, pos)) {
 490       guint64 mask;
 491
 492       if (gst_audio_channel_positions_to_mask (pos, context->channels, FALSE,
 493               &mask)) {
 494         gst_caps_set_simple (caps, "channel-mask", GST_TYPE_BITMASK, mask,
 495             NULL);
 496       }
 497     }
 498   } else if (encode) {
 499     gint maxchannels = 2;
 500     const gint *rates = NULL;
 501     gint n_rates = 0;
 502
 503     /* so we must be after restricted caps in this case */
 504     switch (codec_id) {
 505       case AV_CODEC_ID_AAC:
 506       case AV_CODEC_ID_AAC_LATM:
 507       case AV_CODEC_ID_DTS:
 508         maxchannels = 6;
 509         break;
 510       case AV_CODEC_ID_MP2:
 511       {
 512         const static gint l_rates[] =
 513             { 48000, 44100, 32000, 24000, 22050, 16000 };
 514         n_rates = G_N_ELEMENTS (l_rates);
 515         rates = l_rates;
 516         break;
 517       }
 518       case AV_CODEC_ID_EAC3:
 519       case AV_CODEC_ID_AC3:
 520       {
 521         const static gint l_rates[] = { 48000, 44100, 32000 };
 522         maxchannels = 6;
 523         n_rates = G_N_ELEMENTS (l_rates);
 524         rates = l_rates;
 525         break;
 526       }
 527       case AV_CODEC_ID_ADPCM_G722:
 528       {
 529         const static gint l_rates[] = { 16000 };
 530         n_rates = G_N_ELEMENTS (l_rates);
 531         rates = l_rates;
 532         maxchannels = 1;
 533         break;
 534       }
 535       case AV_CODEC_ID_ADPCM_G726:
 536       {
 537         const static gint l_rates[] = { 8000 };
 538         n_rates = G_N_ELEMENTS (l_rates);
 539         rates = l_rates;
 540         maxchannels = 1;
 541         break;
 542       }
 543       case AV_CODEC_ID_ADPCM_SWF:
 544       {
 545         const static gint l_rates[] = { 11025, 22050, 44100 };
 546         n_rates = G_N_ELEMENTS (l_rates);
 547         rates = l_rates;
 548         break;
 549       }
 550       case AV_CODEC_ID_ROQ_DPCM:
 551       {
 552         const static gint l_rates[] = { 22050 };
 553         n_rates = G_N_ELEMENTS (l_rates);
 554         rates = l_rates;
 555         break;
 556       }
 557       case AV_CODEC_ID_AMR_NB:
 558       {
 559         const static gint l_rates[] = { 8000 };
 560         maxchannels = 1;
 561         n_rates = G_N_ELEMENTS (l_rates);
 562         rates = l_rates;
 563         break;
 564       }
 565       case AV_CODEC_ID_AMR_WB:
 566       {
 567         const static gint l_rates[] = { 16000 };
 568         maxchannels = 1;
 569         n_rates = G_N_ELEMENTS (l_rates);
 570         rates = l_rates;
 571         break;
 572       }
 573       default:
 574         break;
 575     }
 576
 577     /* regardless of encode/decode, open up channels if applicable */
 578     /* Until decoders/encoders expose the maximum number of channels
 579      * they support, we whitelist them here. */
 580     switch (codec_id) {
 581       case AV_CODEC_ID_WMAPRO:
 582       case AV_CODEC_ID_TRUEHD:
 583         maxchannels = 8;
 584         break;
 585       default:
 586         break;
 587     }
 588
 589     if (codec && codec->channel_layouts) {
 590       const uint64_t *layouts = codec->channel_layouts;
 591       GstAudioChannelPosition pos[64];
 592
 593       caps = gst_caps_new_empty ();
 594       while (*layouts) {
 595         gint nbits_set = get_nbits_set (*layouts);
 596
 597         if (gst_ffmpeg_channel_layout_to_gst (*layouts, nbits_set, pos)) {
 598           guint64 mask;
 599
 600           if (gst_audio_channel_positions_to_mask (pos, nbits_set, FALSE,
 601                   &mask)) {
 602             GstCaps *tmp =
 603                 gst_caps_new_simple (mimetype, "channel-mask", GST_TYPE_BITMASK,
 604                 mask,
 605                 "channels", G_TYPE_INT, nbits_set, NULL);
 606
 607             gst_caps_append (caps, tmp);
 608           }
 609         }
 610         layouts++;
 611       }
 612     } else {
 613       if (maxchannels == 1)
 614         caps = gst_caps_new_simple (mimetype,
 615             "channels", G_TYPE_INT, maxchannels, NULL);
 616       else
 617         caps = gst_caps_new_simple (mimetype,
 618             "channels", GST_TYPE_INT_RANGE, 1, maxchannels, NULL);
 619     }
 620
 621     if (n_rates) {
 622       GValue list = { 0, };
 623
 624       g_value_init (&list, GST_TYPE_LIST);
 625       for (i = 0; i < n_rates; i++) {
 626         GValue v = { 0, };
 627
 628         g_value_init (&v, G_TYPE_INT);
 629         g_value_set_int (&v, rates[i]);
 630         gst_value_list_append_value (&list, &v);
 631         g_value_unset (&v);
 632       }
 633       gst_caps_set_value (caps, "rate", &list);
 634       g_value_unset (&list);
 635     } else if (codec && codec->supported_samplerates
 636         && codec->supported_samplerates[0]) {
 637       GValue va = { 0, };
 638       GValue v = { 0, };
 639
 640       if (!codec->supported_samplerates[1]) {
 641         gst_caps_set_simple (caps, "rate", G_TYPE_INT,
 642             codec->supported_samplerates[0], NULL);
 643       } else {
 644         const int *rates = codec->supported_samplerates;
 645
 646         g_value_init (&va, GST_TYPE_LIST);
 647         g_value_init (&v, G_TYPE_INT);
 648
 649         while (*rates) {
 650           g_value_set_int (&v, *rates);
 651           gst_value_list_append_value (&va, &v);
 652           rates++;
 653         }
 654         gst_caps_set_value (caps, "rate", &va);
 655         g_value_unset (&va);
 656         g_value_unset (&v);
 657       }
 658     } else {
 659       gst_caps_set_simple (caps, "rate", GST_TYPE_INT_RANGE, 4000, 96000, NULL);
 660     }
 661   } else {
 662     caps = gst_caps_new_empty_simple (mimetype);
 663   }
 664
 665   va_start (var_args, fieldname);
 666   gst_caps_set_simple_valist (caps, fieldname, var_args);
 667   va_end (var_args);
 668
 669   return caps;
 670 }
 671
 672 /* Convert a FFMPEG codec ID and optional AVCodecContext
 673  * to a GstCaps. If the context is ommitted, no fixed values
 674  * for video/audio size will be included in the GstCaps
 675  *
 676  * CodecID is primarily meant for compressed data GstCaps!
 677  *
 678  * encode is a special parameter. gstffmpegdec will say
 679  * FALSE, gstffmpegenc will say TRUE. The output caps
 680  * depends on this, in such a way that it will be very
 681  * specific, defined, fixed and correct caps for encoders,
 682  * yet very wide, "forgiving" caps for decoders. Example
 683  * for mp3: decode: audio/mpeg,mpegversion=1,layer=[1-3]
 684  * but encode: audio/mpeg,mpegversion=1,layer=3,bitrate=x,
 685  * rate=x,channels=x.
 686  */
 687
 688 GstCaps *
 689 gst_ffmpeg_codecid_to_caps (enum CodecID codec_id,
 690     AVCodecContext * context, gboolean encode)
 691 {
 692   GstCaps *caps = NULL;
 693   gboolean buildcaps = FALSE;
 694
 695   GST_LOG ("codec_id:%d, context:%p, encode:%d", codec_id, context, encode);
 696
 697   switch (codec_id) {
 698     case AV_CODEC_ID_MPEG1VIDEO:
 699       /* FIXME: bitrate */
 700       caps = gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/mpeg",
 701           "mpegversion", G_TYPE_INT, 1,
 702           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 703       break;
 704
 705     case AV_CODEC_ID_MPEG2VIDEO:
 706       if (encode) {
 707         /* FIXME: bitrate */
 708         caps =
 709             gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/mpeg",
 710             "mpegversion", G_TYPE_INT, 2, "systemstream", G_TYPE_BOOLEAN, FALSE,
 711             NULL);
 712       } else {
 713         /* decode both MPEG-1 and MPEG-2; width/height/fps are all in
 714          * the MPEG video stream headers, so may be omitted from caps. */
 715         caps = gst_caps_new_simple ("video/mpeg",
 716             "mpegversion", GST_TYPE_INT_RANGE, 1, 2,
 717             "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
 718       }
 719       break;
 720
 721     case AV_CODEC_ID_MPEG2VIDEO_XVMC:
 722       /* this is a special ID - don't need it in GStreamer, I think */
 723       break;
 724
 725     case AV_CODEC_ID_H263:
 726       if (encode) {
 727         caps =
 728             gst_ff_vid_caps_new (context, NULL, codec_id, encode,
 729             "video/x-h263", "variant", G_TYPE_STRING, "itu", "h263version",
 730             G_TYPE_STRING, "h263", NULL);
 731       } else {
 732         /* don't pass codec_id, we can decode other variants with the H263
 733          * decoder that don't have specific size requirements
 734          */
 735         caps =
 736             gst_ff_vid_caps_new (context, NULL, AV_CODEC_ID_NONE, encode,
 737             "video/x-h263", "variant", G_TYPE_STRING, "itu", NULL);
 738       }
 739       break;
 740
 741     case AV_CODEC_ID_H263P:
 742       caps =
 743           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-h263",
 744           "variant", G_TYPE_STRING, "itu", "h263version", G_TYPE_STRING,
 745           "h263p", NULL);
 746       if (encode && context) {
 747
 748         gst_caps_set_simple (caps,
 749             "annex-f", G_TYPE_BOOLEAN, context->flags & CODEC_FLAG_4MV,
 750             "annex-j", G_TYPE_BOOLEAN, context->flags & CODEC_FLAG_LOOP_FILTER,
 751             "annex-i", G_TYPE_BOOLEAN, context->flags & CODEC_FLAG_AC_PRED,
 752             "annex-t", G_TYPE_BOOLEAN, context->flags & CODEC_FLAG_AC_PRED,
 753             NULL);
 754       }
 755       break;
 756
 757     case AV_CODEC_ID_H263I:
 758       caps =
 759           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
 760           "video/x-intel-h263", "variant", G_TYPE_STRING, "intel", NULL);
 761       break;
 762
 763     case AV_CODEC_ID_H261:
 764       caps =
 765           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-h261",
 766           NULL);
 767       break;
 768
 769     case AV_CODEC_ID_RV10:
 770     case AV_CODEC_ID_RV20:
 771     case AV_CODEC_ID_RV30:
 772     case AV_CODEC_ID_RV40:
 773     {
 774       gint version;
 775
 776       switch (codec_id) {
 777         case AV_CODEC_ID_RV40:
 778           version = 4;
 779           break;
 780         case AV_CODEC_ID_RV30:
 781           version = 3;
 782           break;
 783         case AV_CODEC_ID_RV20:
 784           version = 2;
 785           break;
 786         default:
 787           version = 1;
 788           break;
 789       }
 790
 791       /* FIXME: context->sub_id must be filled in during decoding */
 792       caps =
 793           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
 794           "video/x-pn-realvideo", "rmversion", G_TYPE_INT, version, NULL);
 795       if (context) {
 796         gst_caps_set_simple (caps, "format", G_TYPE_INT, context->sub_id, NULL);
 797         if (context->extradata_size >= 8) {
 798           gst_caps_set_simple (caps,
 799               "subformat", G_TYPE_INT, GST_READ_UINT32_BE (context->extradata),
 800               NULL);
 801         }
 802       }
 803     }
 804       break;
 805
 806     case AV_CODEC_ID_MP1:
 807       /* FIXME: bitrate */
 808       caps = gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/mpeg",
 809           "mpegversion", G_TYPE_INT, 1, "layer", G_TYPE_INT, 1, NULL);
 810       break;
 811
 812     case AV_CODEC_ID_MP2:
 813       /* FIXME: bitrate */
 814       caps = gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/mpeg",
 815           "mpegversion", G_TYPE_INT, 1, "layer", G_TYPE_INT, 2, NULL);
 816       break;
 817
 818     case AV_CODEC_ID_MP3:
 819       if (encode) {
 820         /* FIXME: bitrate */
 821         caps =
 822             gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/mpeg",
 823             "mpegversion", G_TYPE_INT, 1, "layer", G_TYPE_INT, 3, NULL);
 824       } else {
 825         /* Decodes MPEG-1 layer 1/2/3. Samplerate, channels et al are
 826          * in the MPEG audio header, so may be omitted from caps. */
 827         caps = gst_caps_new_simple ("audio/mpeg",
 828             "mpegversion", G_TYPE_INT, 1,
 829             "layer", GST_TYPE_INT_RANGE, 1, 3, NULL);
 830       }
 831       break;
 832
 833     case AV_CODEC_ID_MUSEPACK7:
 834       caps =
 835           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
 836           "audio/x-ffmpeg-parsed-musepack", "streamversion", G_TYPE_INT, 7,
 837           NULL);
 838       break;
 839
 840     case AV_CODEC_ID_MUSEPACK8:
 841       caps =
 842           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
 843           "audio/x-ffmpeg-parsed-musepack", "streamversion", G_TYPE_INT, 8,
 844           NULL);
 845       break;
 846
 847     case AV_CODEC_ID_AC3:
 848       /* FIXME: bitrate */
 849       caps =
 850           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-ac3",
 851           NULL);
 852       break;
 853
 854     case AV_CODEC_ID_EAC3:
 855       /* FIXME: bitrate */
 856       caps =
 857           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-eac3",
 858           NULL);
 859       break;
 860
 861     case AV_CODEC_ID_TRUEHD:
 862       caps =
 863           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
 864           "audio/x-true-hd", NULL);
 865       break;
 866
 867     case AV_CODEC_ID_ATRAC1:
 868       caps =
 869           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
 870           "audio/x-vnd.sony.atrac1", NULL);
 871       break;
 872
 873     case AV_CODEC_ID_ATRAC3:
 874       caps =
 875           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
 876           "audio/x-vnd.sony.atrac3", NULL);
 877       break;
 878
 879     case AV_CODEC_ID_DTS:
 880       caps =
 881           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-dts",
 882           NULL);
 883       break;
 884
 885     case AV_CODEC_ID_APE:
 886       caps =
 887           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
 888           "audio/x-ffmpeg-parsed-ape", NULL);
 889       if (context) {
 890         gst_caps_set_simple (caps,
 891             "depth", G_TYPE_INT, context->bits_per_coded_sample, NULL);
 892       }
 893       break;
 894
 895     case AV_CODEC_ID_MLP:
 896       caps =
 897           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-mlp",
 898           NULL);
 899       break;
 900
 901     case AV_CODEC_ID_IMC:
 902       caps =
 903           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-imc",
 904           NULL);
 905       break;
 906
 907       /* MJPEG is normal JPEG, Motion-JPEG and Quicktime MJPEG-A. MJPEGB
 908        * is Quicktime's MJPEG-B. LJPEG is lossless JPEG. I don't know what
 909        * sp5x is, but it's apparently something JPEG... We don't separate
 910        * between those in GStreamer. Should we (at least between MJPEG,
 911        * MJPEG-B and sp5x decoding...)? */
 912     case AV_CODEC_ID_MJPEG:
 913     case AV_CODEC_ID_LJPEG:
 914       caps =
 915           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/jpeg",
 916           NULL);
 917       break;
 918
 919     case AV_CODEC_ID_SP5X:
 920       caps =
 921           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/sp5x",
 922           NULL);
 923       break;
 924
 925     case AV_CODEC_ID_MJPEGB:
 926       caps =
 927           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
 928           "video/x-mjpeg-b", NULL);
 929       break;
 930
 931     case AV_CODEC_ID_MPEG4:
 932       if (encode && context != NULL) {
 933         /* I'm not exactly sure what ffmpeg outputs... ffmpeg itself uses
 934          * the AVI fourcc 'DIVX', but 'mp4v' for Quicktime... */
 935         switch (context->codec_tag) {
 936           case GST_MAKE_FOURCC ('D', 'I', 'V', 'X'):
 937             caps =
 938                 gst_ff_vid_caps_new (context, NULL, codec_id, encode,
 939                 "video/x-divx", "divxversion", G_TYPE_INT, 5, NULL);
 940             break;
 941           case GST_MAKE_FOURCC ('m', 'p', '4', 'v'):
 942           default:
 943             /* FIXME: bitrate */
 944             caps =
 945                 gst_ff_vid_caps_new (context, NULL, codec_id, encode,
 946                 "video/mpeg", "systemstream", G_TYPE_BOOLEAN, FALSE,
 947                 "mpegversion", G_TYPE_INT, 4, NULL);
 948             break;
 949         }
 950       } else {
 951         /* The trick here is to separate xvid, divx, mpeg4, 3ivx et al */
 952         caps =
 953             gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/mpeg",
 954             "mpegversion", G_TYPE_INT, 4, "systemstream", G_TYPE_BOOLEAN, FALSE,
 955             NULL);
 956         if (encode) {
 957           gst_caps_append (caps, gst_ff_vid_caps_new (context, NULL, codec_id,
 958                   encode, "video/x-divx", "divxversion", G_TYPE_INT, 5, NULL));
 959         } else {
 960           gst_caps_append (caps, gst_ff_vid_caps_new (context, NULL, codec_id,
 961                   encode, "video/x-divx", "divxversion", GST_TYPE_INT_RANGE, 4,
 962                   5, NULL));
 963           gst_caps_append (caps, gst_ff_vid_caps_new (context, NULL, codec_id,
 964                   encode, "video/x-xvid", NULL));
 965           gst_caps_append (caps, gst_ff_vid_caps_new (context, NULL, codec_id,
 966                   encode, "video/x-3ivx", NULL));
 967         }
 968       }
 969       break;
 970
 971     case AV_CODEC_ID_RAWVIDEO:
 972       caps =
 973           gst_ffmpeg_codectype_to_video_caps (context, codec_id, encode, NULL);
 974       break;
 975
 976     case AV_CODEC_ID_MSMPEG4V1:
 977     case AV_CODEC_ID_MSMPEG4V2:
 978     case AV_CODEC_ID_MSMPEG4V3:
 979     {
 980       gint version = 41 + codec_id - AV_CODEC_ID_MSMPEG4V1;
 981
 982       /* encode-FIXME: bitrate */
 983       caps =
 984           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
 985           "video/x-msmpeg", "msmpegversion", G_TYPE_INT, version, NULL);
 986       if (!encode && codec_id == AV_CODEC_ID_MSMPEG4V3) {
 987         gst_caps_append (caps, gst_ff_vid_caps_new (context, NULL, codec_id,
 988                 encode, "video/x-divx", "divxversion", G_TYPE_INT, 3, NULL));
 989       }
 990     }
 991       break;
 992
 993     case AV_CODEC_ID_WMV1:
 994     case AV_CODEC_ID_WMV2:
 995     {
 996       gint version = (codec_id == AV_CODEC_ID_WMV1) ? 1 : 2;
 997
 998       caps =
 999           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-wmv",
1000           "wmvversion", G_TYPE_INT, version, NULL);
1001     }
1002       break;
1003
1004     case AV_CODEC_ID_FLV1:
1005       caps =
1006           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1007           "video/x-flash-video", "flvversion", G_TYPE_INT, 1, NULL);
1008       break;
1009
1010     case AV_CODEC_ID_SVQ1:
1011       caps =
1012           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-svq",
1013           "svqversion", G_TYPE_INT, 1, NULL);
1014       break;
1015
1016     case AV_CODEC_ID_SVQ3:
1017       caps =
1018           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-svq",
1019           "svqversion", G_TYPE_INT, 3, NULL);
1020       break;
1021
1022     case AV_CODEC_ID_DVAUDIO:
1023       caps =
1024           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-dv",
1025           NULL);
1026       break;
1027
1028     case AV_CODEC_ID_DVVIDEO:
1029     {
1030       if (encode && context) {
1031         const gchar *format;
1032
1033         switch (context->pix_fmt) {
1034           case PIX_FMT_YUYV422:
1035             format = "YUY2";
1036             break;
1037           case PIX_FMT_YUV420P:
1038             format = "I420";
1039             break;
1040           case PIX_FMT_YUVA420P:
1041             format = "A420";
1042             break;
1043           case PIX_FMT_YUV411P:
1044             format = "Y41B";
1045             break;
1046           case PIX_FMT_YUV422P:
1047             format = "Y42B";
1048             break;
1049           case PIX_FMT_YUV410P:
1050             format = "YUV9";
1051             break;
1052           default:
1053             GST_WARNING
1054                 ("Couldnt' find format for pixfmt %d, defaulting to I420",
1055                 context->pix_fmt);
1056             format = "I420";
1057             break;
1058         }
1059         caps =
1060             gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-dv",
1061             "systemstream", G_TYPE_BOOLEAN, FALSE, "format", G_TYPE_STRING,
1062             format, NULL);
1063       } else {
1064         caps =
1065             gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-dv",
1066             "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
1067       }
1068     }
1069       break;
1070
1071     case AV_CODEC_ID_WMAV1:
1072     case AV_CODEC_ID_WMAV2:
1073     {
1074       gint version = (codec_id == AV_CODEC_ID_WMAV1) ? 1 : 2;
1075
1076       if (context) {
1077         caps =
1078             gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-wma",
1079             "wmaversion", G_TYPE_INT, version, "block_align", G_TYPE_INT,
1080             context->block_align, "bitrate", G_TYPE_INT, context->bit_rate,
1081             NULL);
1082       } else {
1083         caps =
1084             gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-wma",
1085             "wmaversion", G_TYPE_INT, version, "block_align",
1086             GST_TYPE_INT_RANGE, 0, G_MAXINT, "bitrate", GST_TYPE_INT_RANGE, 0,
1087             G_MAXINT, NULL);
1088       }
1089     }
1090       break;
1091     case AV_CODEC_ID_WMAPRO:
1092     {
1093       caps =
1094           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-wma",
1095           "wmaversion", G_TYPE_INT, 3, NULL);
1096       break;
1097     }
1098     case AV_CODEC_ID_WMALOSSLESS:
1099     {
1100       /* libav only supports a depth of 16 at the moment */
1101       caps =
1102           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-wma",
1103           "wmaversion", G_TYPE_INT, 4, "depth", G_TYPE_INT, 16, NULL);
1104       break;
1105     }
1106     case AV_CODEC_ID_WMAVOICE:
1107     {
1108       caps =
1109           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-wms",
1110           NULL);
1111       break;
1112     }
1113
1114     case AV_CODEC_ID_MACE3:
1115     case AV_CODEC_ID_MACE6:
1116     {
1117       gint version = (codec_id == AV_CODEC_ID_MACE3) ? 3 : 6;
1118
1119       caps =
1120           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-mace",
1121           "maceversion", G_TYPE_INT, version, NULL);
1122     }
1123       break;
1124
1125     case AV_CODEC_ID_HUFFYUV:
1126       caps =
1127           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1128           "video/x-huffyuv", NULL);
1129       if (context) {
1130         gst_caps_set_simple (caps,
1131             "bpp", G_TYPE_INT, context->bits_per_coded_sample, NULL);
1132       }
1133       break;
1134
1135     case AV_CODEC_ID_CYUV:
1136       caps =
1137           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1138           "video/x-compressed-yuv", NULL);
1139       break;
1140
1141     case AV_CODEC_ID_H264:
1142       caps =
1143           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-h264",
1144           "alignment", G_TYPE_STRING, "au", NULL);
1145       break;
1146
1147     case AV_CODEC_ID_INDEO5:
1148       caps =
1149           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-indeo",
1150           "indeoversion", G_TYPE_INT, 5, NULL);
1151       break;
1152
1153     case AV_CODEC_ID_INDEO4:
1154       caps =
1155           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-indeo",
1156           "indeoversion", G_TYPE_INT, 4, NULL);
1157       break;
1158
1159     case AV_CODEC_ID_INDEO3:
1160       caps =
1161           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-indeo",
1162           "indeoversion", G_TYPE_INT, 3, NULL);
1163       break;
1164
1165     case AV_CODEC_ID_INDEO2:
1166       caps =
1167           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-indeo",
1168           "indeoversion", G_TYPE_INT, 2, NULL);
1169       break;
1170
1171     case AV_CODEC_ID_FLASHSV:
1172       caps =
1173           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1174           "video/x-flash-screen", NULL);
1175       break;
1176
1177     case AV_CODEC_ID_VP3:
1178       caps =
1179           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-vp3",
1180           NULL);
1181       break;
1182
1183     case AV_CODEC_ID_VP5:
1184       caps =
1185           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-vp5",
1186           NULL);
1187       break;
1188
1189     case AV_CODEC_ID_VP6:
1190       caps =
1191           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-vp6",
1192           NULL);
1193       break;
1194
1195     case AV_CODEC_ID_VP6F:
1196       caps =
1197           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1198           "video/x-vp6-flash", NULL);
1199       break;
1200
1201     case AV_CODEC_ID_VP6A:
1202       caps =
1203           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1204           "video/x-vp6-alpha", NULL);
1205       break;
1206
1207     case AV_CODEC_ID_VP8:
1208       caps =
1209           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-vp8",
1210           NULL);
1211       break;
1212
1213     case AV_CODEC_ID_THEORA:
1214       caps =
1215           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1216           "video/x-theora", NULL);
1217       break;
1218
1219     case AV_CODEC_ID_AAC:
1220     {
1221       caps =
1222           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/mpeg",
1223           NULL);
1224
1225       if (!encode) {
1226         GValue arr = { 0, };
1227         GValue item = { 0, };
1228
1229         g_value_init (&arr, GST_TYPE_LIST);
1230         g_value_init (&item, G_TYPE_INT);
1231         g_value_set_int (&item, 2);
1232         gst_value_list_append_value (&arr, &item);
1233         g_value_set_int (&item, 4);
1234         gst_value_list_append_value (&arr, &item);
1235         g_value_unset (&item);
1236
1237         gst_caps_set_value (caps, "mpegversion", &arr);
1238         g_value_unset (&arr);
1239
1240         g_value_init (&arr, GST_TYPE_LIST);
1241         g_value_init (&item, G_TYPE_STRING);
1242         g_value_set_string (&item, "raw");
1243         gst_value_list_append_value (&arr, &item);
1244         g_value_set_string (&item, "adts");
1245         gst_value_list_append_value (&arr, &item);
1246         g_value_set_string (&item, "adif");
1247         gst_value_list_append_value (&arr, &item);
1248         g_value_unset (&item);
1249
1250         gst_caps_set_value (caps, "stream-format", &arr);
1251         g_value_unset (&arr);
1252       } else {
1253         gst_caps_set_simple (caps, "mpegversion", G_TYPE_INT, 4,
1254             "stream-format", G_TYPE_STRING, "raw",
1255             "base-profile", G_TYPE_STRING, "lc", NULL);
1256
1257         if (context && context->extradata_size > 0)
1258           gst_codec_utils_aac_caps_set_level_and_profile (caps,
1259               context->extradata, context->extradata_size);
1260       }
1261
1262       break;
1263     }
1264     case AV_CODEC_ID_AAC_LATM: /* LATM/LOAS AAC syntax */
1265       caps = gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/mpeg",
1266           "mpegversion", G_TYPE_INT, 4, "stream-format", G_TYPE_STRING, "loas",
1267           NULL);
1268       break;
1269
1270     case AV_CODEC_ID_ASV1:
1271       caps =
1272           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-asus",
1273           "asusversion", G_TYPE_INT, 1, NULL);
1274       break;
1275     case AV_CODEC_ID_ASV2:
1276       caps =
1277           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-asus",
1278           "asusversion", G_TYPE_INT, 2, NULL);
1279       break;
1280
1281     case AV_CODEC_ID_FFV1:
1282       caps =
1283           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-ffv",
1284           "ffvversion", G_TYPE_INT, 1, NULL);
1285       break;
1286
1287     case AV_CODEC_ID_4XM:
1288       caps =
1289           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-4xm",
1290           NULL);
1291       break;
1292
1293     case AV_CODEC_ID_XAN_WC3:
1294     case AV_CODEC_ID_XAN_WC4:
1295       caps =
1296           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-xan",
1297           "wcversion", G_TYPE_INT, 3 - AV_CODEC_ID_XAN_WC3 + codec_id, NULL);
1298       break;
1299
1300     case AV_CODEC_ID_CLJR:
1301       caps =
1302           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1303           "video/x-cirrus-logic-accupak", NULL);
1304       break;
1305
1306     case AV_CODEC_ID_FRAPS:
1307       caps =
1308           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-fraps",
1309           NULL);
1310       break;
1311
1312     case AV_CODEC_ID_MDEC:
1313     case AV_CODEC_ID_ROQ:
1314     case AV_CODEC_ID_INTERPLAY_VIDEO:
1315       buildcaps = TRUE;
1316       break;
1317
1318     case AV_CODEC_ID_VCR1:
1319       caps =
1320           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1321           "video/x-ati-vcr", "vcrversion", G_TYPE_INT, 1, NULL);
1322       break;
1323
1324     case AV_CODEC_ID_RPZA:
1325       caps =
1326           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1327           "video/x-apple-video", NULL);
1328       break;
1329
1330     case AV_CODEC_ID_CINEPAK:
1331       caps =
1332           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1333           "video/x-cinepak", NULL);
1334       break;
1335
1336       /* WS_VQA belogns here (order) */
1337
1338     case AV_CODEC_ID_MSRLE:
1339       caps =
1340           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-rle",
1341           "layout", G_TYPE_STRING, "microsoft", NULL);
1342       if (context) {
1343         gst_caps_set_simple (caps,
1344             "depth", G_TYPE_INT, (gint) context->bits_per_coded_sample, NULL);
1345       } else {
1346         gst_caps_set_simple (caps, "depth", GST_TYPE_INT_RANGE, 1, 64, NULL);
1347       }
1348       break;
1349
1350     case AV_CODEC_ID_QTRLE:
1351       caps =
1352           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-rle",
1353           "layout", G_TYPE_STRING, "quicktime", NULL);
1354       if (context) {
1355         gst_caps_set_simple (caps,
1356             "depth", G_TYPE_INT, (gint) context->bits_per_coded_sample, NULL);
1357       } else {
1358         gst_caps_set_simple (caps, "depth", GST_TYPE_INT_RANGE, 1, 64, NULL);
1359       }
1360       break;
1361
1362     case AV_CODEC_ID_MSVIDEO1:
1363       caps =
1364           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1365           "video/x-msvideocodec", "msvideoversion", G_TYPE_INT, 1, NULL);
1366       break;
1367
1368     case AV_CODEC_ID_MSS1:
1369       caps =
1370           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-wmv",
1371           "wmvversion", G_TYPE_INT, 1, "format", G_TYPE_STRING, "MSS1", NULL);
1372       break;
1373
1374     case AV_CODEC_ID_MSS2:
1375       caps =
1376           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-wmv",
1377           "wmvversion", G_TYPE_INT, 3, "format", G_TYPE_STRING, "MSS2", NULL);
1378       break;
1379
1380     case AV_CODEC_ID_WMV3:
1381       caps =
1382           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-wmv",
1383           "wmvversion", G_TYPE_INT, 3, "format", G_TYPE_STRING, "WMV3", NULL);
1384       break;
1385     case AV_CODEC_ID_VC1:
1386       caps =
1387           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-wmv",
1388           "wmvversion", G_TYPE_INT, 3, "format", G_TYPE_STRING, "WVC1", NULL);
1389       break;
1390     case AV_CODEC_ID_QDM2:
1391       caps =
1392           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-qdm2",
1393           NULL);
1394       break;
1395
1396     case AV_CODEC_ID_MSZH:
1397       caps =
1398           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-mszh",
1399           NULL);
1400       break;
1401
1402     case AV_CODEC_ID_ZLIB:
1403       caps =
1404           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-zlib",
1405           NULL);
1406       break;
1407
1408     case AV_CODEC_ID_TRUEMOTION1:
1409       caps =
1410           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1411           "video/x-truemotion", "trueversion", G_TYPE_INT, 1, NULL);
1412       break;
1413     case AV_CODEC_ID_TRUEMOTION2:
1414       caps =
1415           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1416           "video/x-truemotion", "trueversion", G_TYPE_INT, 2, NULL);
1417       break;
1418
1419     case AV_CODEC_ID_ULTI:
1420       caps =
1421           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1422           "video/x-ultimotion", NULL);
1423       break;
1424
1425     case AV_CODEC_ID_TSCC:
1426       caps =
1427           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1428           "video/x-camtasia", NULL);
1429       if (context) {
1430         gst_caps_set_simple (caps,
1431             "depth", G_TYPE_INT, (gint) context->bits_per_coded_sample, NULL);
1432       } else {
1433         gst_caps_set_simple (caps, "depth", GST_TYPE_INT_RANGE, 8, 32, NULL);
1434       }
1435       break;
1436
1437     case AV_CODEC_ID_KMVC:
1438       caps =
1439           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-kmvc",
1440           NULL);
1441       break;
1442
1443     case AV_CODEC_ID_NUV:
1444       caps =
1445           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-nuv",
1446           NULL);
1447       break;
1448
1449     case AV_CODEC_ID_GIF:
1450       caps =
1451           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/gif",
1452           NULL);
1453       break;
1454
1455     case AV_CODEC_ID_PNG:
1456       caps =
1457           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/png",
1458           NULL);
1459       break;
1460
1461     case AV_CODEC_ID_PPM:
1462       caps =
1463           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/ppm",
1464           NULL);
1465       break;
1466
1467     case AV_CODEC_ID_PBM:
1468       caps =
1469           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/pbm",
1470           NULL);
1471       break;
1472
1473     case AV_CODEC_ID_PAM:
1474       caps =
1475           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1476           "image/x-portable-anymap", NULL);
1477       break;
1478
1479     case AV_CODEC_ID_PGM:
1480       caps =
1481           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1482           "image/x-portable-graymap", NULL);
1483       break;
1484
1485     case AV_CODEC_ID_PCX:
1486       caps =
1487           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/x-pcx",
1488           NULL);
1489       break;
1490
1491     case AV_CODEC_ID_SGI:
1492       caps =
1493           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/x-sgi",
1494           NULL);
1495       break;
1496
1497     case AV_CODEC_ID_TARGA:
1498       caps =
1499           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/x-tga",
1500           NULL);
1501       break;
1502
1503     case AV_CODEC_ID_TIFF:
1504       caps =
1505           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "image/tiff",
1506           NULL);
1507       break;
1508
1509     case AV_CODEC_ID_SUNRAST:
1510       caps =
1511           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1512           "image/x-sun-raster", NULL);
1513       break;
1514
1515     case AV_CODEC_ID_SMC:
1516       caps =
1517           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-smc",
1518           NULL);
1519       break;
1520
1521     case AV_CODEC_ID_QDRAW:
1522       caps =
1523           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-qdrw",
1524           NULL);
1525       break;
1526
1527     case AV_CODEC_ID_DNXHD:
1528       caps =
1529           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-dnxhd",
1530           NULL);
1531       break;
1532
1533     case AV_CODEC_ID_PRORES:
1534       caps =
1535           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1536           "video/x-prores", NULL);
1537       break;
1538
1539     case AV_CODEC_ID_MIMIC:
1540       caps =
1541           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-mimic",
1542           NULL);
1543       break;
1544
1545     case AV_CODEC_ID_VMNC:
1546       caps =
1547           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-vmnc",
1548           NULL);
1549       break;
1550
1551     case AV_CODEC_ID_TRUESPEECH:
1552       caps =
1553           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
1554           "audio/x-truespeech", NULL);
1555       break;
1556
1557     case AV_CODEC_ID_QCELP:
1558       caps =
1559           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/qcelp",
1560           NULL);
1561       break;
1562
1563     case AV_CODEC_ID_AMV:
1564       caps =
1565           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-amv",
1566           NULL);
1567       break;
1568
1569     case AV_CODEC_ID_AASC:
1570       caps =
1571           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-aasc",
1572           NULL);
1573       break;
1574
1575     case AV_CODEC_ID_LOCO:
1576       caps =
1577           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-loco",
1578           NULL);
1579       break;
1580
1581     case AV_CODEC_ID_ZMBV:
1582       caps =
1583           gst_ff_vid_caps_new (context, NULL, codec_id, encode, "video/x-zmbv",
1584           NULL);
1585       break;
1586
1587     case AV_CODEC_ID_LAGARITH:
1588       caps =
1589           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1590           "video/x-lagarith", NULL);
1591       break;
1592
1593     case AV_CODEC_ID_CSCD:
1594       caps =
1595           gst_ff_vid_caps_new (context, NULL, codec_id, encode,
1596           "video/x-camstudio", NULL);
1597       if (context) {
1598         gst_caps_set_simple (caps,
1599             "depth", G_TYPE_INT, (gint) context->bits_per_coded_sample, NULL);
1600       } else {
1601         gst_caps_set_simple (caps, "depth", GST_TYPE_INT_RANGE, 8, 32, NULL);
1602       }
1603       break;
1604
1605     case AV_CODEC_ID_WS_VQA:
1606     case AV_CODEC_ID_IDCIN:
1607     case AV_CODEC_ID_8BPS:
1608     case AV_CODEC_ID_FLIC:
1609     case AV_CODEC_ID_VMDVIDEO:
1610     case AV_CODEC_ID_VMDAUDIO:
1611     case AV_CODEC_ID_SNOW:
1612     case AV_CODEC_ID_VIXL:
1613     case AV_CODEC_ID_QPEG:
1614     case AV_CODEC_ID_PGMYUV:
1615     case AV_CODEC_ID_FFVHUFF:
1616     case AV_CODEC_ID_WNV1:
1617     case AV_CODEC_ID_MP3ADU:
1618     case AV_CODEC_ID_MP3ON4:
1619     case AV_CODEC_ID_WESTWOOD_SND1:
1620     case AV_CODEC_ID_MMVIDEO:
1621     case AV_CODEC_ID_AVS:
1622     case AV_CODEC_ID_CAVS:
1623       buildcaps = TRUE;
1624       break;
1625
1626       /* weird quasi-codecs for the demuxers only */
1627     case AV_CODEC_ID_PCM_S16LE:
1628     case AV_CODEC_ID_PCM_S16BE:
1629     case AV_CODEC_ID_PCM_U16LE:
1630     case AV_CODEC_ID_PCM_U16BE:
1631     case AV_CODEC_ID_PCM_S8:
1632     case AV_CODEC_ID_PCM_U8:
1633     {
1634       GstAudioFormat format;
1635
1636       switch (codec_id) {
1637         case AV_CODEC_ID_PCM_S16LE:
1638           format = GST_AUDIO_FORMAT_S16LE;
1639           break;
1640         case AV_CODEC_ID_PCM_S16BE:
1641           format = GST_AUDIO_FORMAT_S16BE;
1642           break;
1643         case AV_CODEC_ID_PCM_U16LE:
1644           format = GST_AUDIO_FORMAT_U16LE;
1645           break;
1646         case AV_CODEC_ID_PCM_U16BE:
1647           format = GST_AUDIO_FORMAT_U16BE;
1648           break;
1649         case AV_CODEC_ID_PCM_S8:
1650           format = GST_AUDIO_FORMAT_S8;
1651           break;
1652         case AV_CODEC_ID_PCM_U8:
1653           format = GST_AUDIO_FORMAT_U8;
1654           break;
1655         default:
1656           g_assert (0);         /* don't worry, we never get here */
1657           break;
1658       }
1659
1660       caps =
1661           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-raw",
1662           "format", G_TYPE_STRING, gst_audio_format_to_string (format),
1663           "layout", G_TYPE_STRING, "interleaved", NULL);
1664     }
1665       break;
1666
1667     case AV_CODEC_ID_PCM_MULAW:
1668       caps =
1669           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-mulaw",
1670           NULL);
1671       break;
1672
1673     case AV_CODEC_ID_PCM_ALAW:
1674       caps =
1675           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-alaw",
1676           NULL);
1677       break;
1678
1679     case AV_CODEC_ID_ADPCM_G722:
1680       caps =
1681           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/G722",
1682           NULL);
1683       if (context)
1684         gst_caps_set_simple (caps,
1685             "block_align", G_TYPE_INT, context->block_align,
1686             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1687       break;
1688
1689     case AV_CODEC_ID_ADPCM_G726:
1690     {
1691       /* the G726 decoder can also handle G721 */
1692       caps =
1693           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-adpcm",
1694           "layout", G_TYPE_STRING, "g726", NULL);
1695       if (context)
1696         gst_caps_set_simple (caps,
1697             "block_align", G_TYPE_INT, context->block_align,
1698             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1699
1700       if (!encode) {
1701         gst_caps_append (caps, gst_caps_new_simple ("audio/x-adpcm",
1702                 "layout", G_TYPE_STRING, "g721",
1703                 "channels", G_TYPE_INT, 1, "rate", G_TYPE_INT, 8000, NULL));
1704       }
1705       break;
1706     }
1707     case AV_CODEC_ID_ADPCM_IMA_QT:
1708     case AV_CODEC_ID_ADPCM_IMA_WAV:
1709     case AV_CODEC_ID_ADPCM_IMA_DK3:
1710     case AV_CODEC_ID_ADPCM_IMA_DK4:
1711     case AV_CODEC_ID_ADPCM_IMA_WS:
1712     case AV_CODEC_ID_ADPCM_IMA_SMJPEG:
1713     case AV_CODEC_ID_ADPCM_IMA_AMV:
1714     case AV_CODEC_ID_ADPCM_IMA_ISS:
1715     case AV_CODEC_ID_ADPCM_IMA_EA_EACS:
1716     case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
1717     case AV_CODEC_ID_ADPCM_MS:
1718     case AV_CODEC_ID_ADPCM_4XM:
1719     case AV_CODEC_ID_ADPCM_XA:
1720     case AV_CODEC_ID_ADPCM_ADX:
1721     case AV_CODEC_ID_ADPCM_EA:
1722     case AV_CODEC_ID_ADPCM_CT:
1723     case AV_CODEC_ID_ADPCM_SWF:
1724     case AV_CODEC_ID_ADPCM_YAMAHA:
1725     case AV_CODEC_ID_ADPCM_SBPRO_2:
1726     case AV_CODEC_ID_ADPCM_SBPRO_3:
1727     case AV_CODEC_ID_ADPCM_SBPRO_4:
1728     case AV_CODEC_ID_ADPCM_EA_R1:
1729     case AV_CODEC_ID_ADPCM_EA_R2:
1730     case AV_CODEC_ID_ADPCM_EA_R3:
1731     case AV_CODEC_ID_ADPCM_EA_MAXIS_XA:
1732     case AV_CODEC_ID_ADPCM_EA_XAS:
1733     case AV_CODEC_ID_ADPCM_THP:
1734     {
1735       const gchar *layout = NULL;
1736
1737       switch (codec_id) {
1738         case AV_CODEC_ID_ADPCM_IMA_QT:
1739           layout = "quicktime";
1740           break;
1741         case AV_CODEC_ID_ADPCM_IMA_WAV:
1742           layout = "dvi";
1743           break;
1744         case AV_CODEC_ID_ADPCM_IMA_DK3:
1745           layout = "dk3";
1746           break;
1747         case AV_CODEC_ID_ADPCM_IMA_DK4:
1748           layout = "dk4";
1749           break;
1750         case AV_CODEC_ID_ADPCM_IMA_WS:
1751           layout = "westwood";
1752           break;
1753         case AV_CODEC_ID_ADPCM_IMA_SMJPEG:
1754           layout = "smjpeg";
1755           break;
1756         case AV_CODEC_ID_ADPCM_IMA_AMV:
1757           layout = "amv";
1758           break;
1759         case AV_CODEC_ID_ADPCM_IMA_ISS:
1760           layout = "iss";
1761           break;
1762         case AV_CODEC_ID_ADPCM_IMA_EA_EACS:
1763           layout = "ea-eacs";
1764           break;
1765         case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
1766           layout = "ea-sead";
1767           break;
1768         case AV_CODEC_ID_ADPCM_MS:
1769           layout = "microsoft";
1770           break;
1771         case AV_CODEC_ID_ADPCM_4XM:
1772           layout = "4xm";
1773           break;
1774         case AV_CODEC_ID_ADPCM_XA:
1775           layout = "xa";
1776           break;
1777         case AV_CODEC_ID_ADPCM_ADX:
1778           layout = "adx";
1779           break;
1780         case AV_CODEC_ID_ADPCM_EA:
1781           layout = "ea";
1782           break;
1783         case AV_CODEC_ID_ADPCM_CT:
1784           layout = "ct";
1785           break;
1786         case AV_CODEC_ID_ADPCM_SWF:
1787           layout = "swf";
1788           break;
1789         case AV_CODEC_ID_ADPCM_YAMAHA:
1790           layout = "yamaha";
1791           break;
1792         case AV_CODEC_ID_ADPCM_SBPRO_2:
1793           layout = "sbpro2";
1794           break;
1795         case AV_CODEC_ID_ADPCM_SBPRO_3:
1796           layout = "sbpro3";
1797           break;
1798         case AV_CODEC_ID_ADPCM_SBPRO_4:
1799           layout = "sbpro4";
1800           break;
1801         case AV_CODEC_ID_ADPCM_EA_R1:
1802           layout = "ea-r1";
1803           break;
1804         case AV_CODEC_ID_ADPCM_EA_R2:
1805           layout = "ea-r3";
1806           break;
1807         case AV_CODEC_ID_ADPCM_EA_R3:
1808           layout = "ea-r3";
1809           break;
1810         case AV_CODEC_ID_ADPCM_EA_MAXIS_XA:
1811           layout = "ea-maxis-xa";
1812           break;
1813         case AV_CODEC_ID_ADPCM_EA_XAS:
1814           layout = "ea-xas";
1815           break;
1816         case AV_CODEC_ID_ADPCM_THP:
1817           layout = "thp";
1818           break;
1819         default:
1820           g_assert (0);         /* don't worry, we never get here */
1821           break;
1822       }
1823
1824       /* FIXME: someone please check whether we need additional properties
1825        * in this caps definition. */
1826       caps =
1827           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-adpcm",
1828           "layout", G_TYPE_STRING, layout, NULL);
1829       if (context)
1830         gst_caps_set_simple (caps,
1831             "block_align", G_TYPE_INT, context->block_align,
1832             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1833     }
1834       break;
1835
1836     case AV_CODEC_ID_AMR_NB:
1837       caps =
1838           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/AMR",
1839           NULL);
1840       break;
1841
1842     case AV_CODEC_ID_AMR_WB:
1843       caps =
1844           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/AMR-WB",
1845           NULL);
1846       break;
1847
1848     case AV_CODEC_ID_GSM:
1849       caps =
1850           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-gsm",
1851           NULL);
1852       break;
1853
1854     case AV_CODEC_ID_GSM_MS:
1855       caps =
1856           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/ms-gsm",
1857           NULL);
1858       break;
1859
1860     case AV_CODEC_ID_NELLYMOSER:
1861       caps =
1862           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
1863           "audio/x-nellymoser", NULL);
1864       break;
1865
1866     case AV_CODEC_ID_SIPR:
1867     {
1868       caps =
1869           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-sipro",
1870           NULL);
1871       if (context) {
1872         gst_caps_set_simple (caps,
1873             "leaf_size", G_TYPE_INT, context->block_align,
1874             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1875       }
1876     }
1877       break;
1878
1879     case AV_CODEC_ID_RA_144:
1880     case AV_CODEC_ID_RA_288:
1881     case AV_CODEC_ID_COOK:
1882     {
1883       gint version = 0;
1884
1885       switch (codec_id) {
1886         case AV_CODEC_ID_RA_144:
1887           version = 1;
1888           break;
1889         case AV_CODEC_ID_RA_288:
1890           version = 2;
1891           break;
1892         case AV_CODEC_ID_COOK:
1893           version = 8;
1894           break;
1895         default:
1896           break;
1897       }
1898
1899       /* FIXME: properties? */
1900       caps =
1901           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
1902           "audio/x-pn-realaudio", "raversion", G_TYPE_INT, version, NULL);
1903       if (context) {
1904         gst_caps_set_simple (caps,
1905             "leaf_size", G_TYPE_INT, context->block_align,
1906             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1907       }
1908     }
1909       break;
1910
1911     case AV_CODEC_ID_ROQ_DPCM:
1912     case AV_CODEC_ID_INTERPLAY_DPCM:
1913     case AV_CODEC_ID_XAN_DPCM:
1914     case AV_CODEC_ID_SOL_DPCM:
1915     {
1916       const gchar *layout = NULL;
1917
1918       switch (codec_id) {
1919         case AV_CODEC_ID_ROQ_DPCM:
1920           layout = "roq";
1921           break;
1922         case AV_CODEC_ID_INTERPLAY_DPCM:
1923           layout = "interplay";
1924           break;
1925         case AV_CODEC_ID_XAN_DPCM:
1926           layout = "xan";
1927           break;
1928         case AV_CODEC_ID_SOL_DPCM:
1929           layout = "sol";
1930           break;
1931         default:
1932           g_assert (0);         /* don't worry, we never get here */
1933           break;
1934       }
1935
1936       /* FIXME: someone please check whether we need additional properties
1937        * in this caps definition. */
1938       caps =
1939           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-dpcm",
1940           "layout", G_TYPE_STRING, layout, NULL);
1941       if (context)
1942         gst_caps_set_simple (caps,
1943             "block_align", G_TYPE_INT, context->block_align,
1944             "bitrate", G_TYPE_INT, context->bit_rate, NULL);
1945     }
1946       break;
1947
1948     case AV_CODEC_ID_SHORTEN:
1949       caps = gst_caps_new_empty_simple ("audio/x-shorten");
1950       break;
1951
1952     case AV_CODEC_ID_ALAC:
1953       caps =
1954           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-alac",
1955           NULL);
1956       if (context) {
1957         gst_caps_set_simple (caps,
1958             "samplesize", G_TYPE_INT, context->bits_per_coded_sample, NULL);
1959       }
1960       break;
1961
1962     case AV_CODEC_ID_FLAC:
1963       /* Note that ffmpeg has no encoder yet, but just for safety. In the
1964        * encoder case, we want to add things like samplerate, channels... */
1965       if (!encode) {
1966         caps = gst_caps_new_empty_simple ("audio/x-flac");
1967       }
1968       break;
1969
1970     case AV_CODEC_ID_S302M:
1971       caps = gst_caps_new_empty_simple ("audio/x-smpte-302m");
1972       break;
1973
1974     case AV_CODEC_ID_DVD_SUBTITLE:
1975     case AV_CODEC_ID_DVB_SUBTITLE:
1976       caps = NULL;
1977       break;
1978     case AV_CODEC_ID_BMP:
1979       caps = gst_caps_new_empty_simple ("image/bmp");
1980       break;
1981     case AV_CODEC_ID_TTA:
1982       caps =
1983           gst_ff_aud_caps_new (context, NULL, codec_id, encode, "audio/x-tta",
1984           NULL);
1985       if (context) {
1986         gst_caps_set_simple (caps,
1987             "samplesize", G_TYPE_INT, context->bits_per_coded_sample, NULL);
1988       }
1989       break;
1990     case AV_CODEC_ID_TWINVQ:
1991       caps =
1992           gst_ff_aud_caps_new (context, NULL, codec_id, encode,
1993           "audio/x-twin-vq", NULL);
1994       break;
1995     default:
1996       GST_DEBUG ("Unknown codec ID %d, please add mapping here", codec_id);
1997       break;
1998   }
1999
2000   if (buildcaps) {
2001     AVCodec *codec;
2002
2003     if ((codec = avcodec_find_decoder (codec_id)) ||
2004         (codec = avcodec_find_encoder (codec_id))) {
2005       gchar *mime = NULL;
2006
2007       GST_LOG ("Could not create stream format caps for %s", codec->name);
2008
2009       switch (codec->type) {
2010         case AVMEDIA_TYPE_VIDEO:
2011           mime = g_strdup_printf ("video/x-gst-av-%s", codec->name);
2012           caps =
2013               gst_ff_vid_caps_new (context, NULL, codec_id, encode, mime, NULL);
2014           g_free (mime);
2015           break;
2016         case AVMEDIA_TYPE_AUDIO:
2017           mime = g_strdup_printf ("audio/x-gst-av-%s", codec->name);
2018           caps =
2019               gst_ff_aud_caps_new (context, NULL, codec_id, encode, mime, NULL);
2020           if (context)
2021             gst_caps_set_simple (caps,
2022                 "block_align", G_TYPE_INT, context->block_align,
2023                 "bitrate", G_TYPE_INT, context->bit_rate, NULL);
2024           g_free (mime);
2025           break;
2026         default:
2027           break;
2028       }
2029     }
2030   }
2031
2032   if (caps != NULL) {
2033
2034     /* set private data */
2035     if (context && context->extradata_size > 0) {
2036       GstBuffer *data = gst_buffer_new_and_alloc (context->extradata_size);
2037
2038       gst_buffer_fill (data, 0, context->extradata, context->extradata_size);
2039       gst_caps_set_simple (caps, "codec_data", GST_TYPE_BUFFER, data, NULL);
2040       gst_buffer_unref (data);
2041     }
2042
2043     GST_LOG ("caps for codec_id=%d: %" GST_PTR_FORMAT, codec_id, caps);
2044
2045   } else {
2046     GST_LOG ("No caps found for codec_id=%d", codec_id);
2047   }
2048
2049   return caps;
2050 }
2051
2052 /* Convert a FFMPEG Pixel Format and optional AVCodecContext
2053  * to a GstCaps. If the context is ommitted, no fixed values
2054  * for video/audio size will be included in the GstCaps
2055  *
2056  * See below for usefullness
2057  */
2058
2059 static GstCaps *
2060 gst_ffmpeg_pixfmt_to_caps (enum PixelFormat pix_fmt, AVCodecContext * context,
2061     enum CodecID codec_id)
2062 {
2063   GstCaps *caps = NULL;
2064   GstVideoFormat format;
2065
2066   format = gst_ffmpeg_pixfmt_to_videoformat (pix_fmt);
2067
2068   if (format != GST_VIDEO_FORMAT_UNKNOWN) {
2069     caps = gst_ff_vid_caps_new (context, NULL, codec_id, TRUE, "video/x-raw",
2070         "format", G_TYPE_STRING, gst_video_format_to_string (format), NULL);
2071   }
2072
2073   if (caps != NULL) {
2074     GST_DEBUG ("caps for pix_fmt=%d: %" GST_PTR_FORMAT, pix_fmt, caps);
2075   } else {
2076     GST_LOG ("No caps found for pix_fmt=%d", pix_fmt);
2077   }
2078
2079   return caps;
2080 }
2081
2082 GstAudioFormat
2083 gst_ffmpeg_smpfmt_to_audioformat (enum AVSampleFormat sample_fmt)
2084 {
2085   switch (sample_fmt) {
2086     case AV_SAMPLE_FMT_U8:
2087     case AV_SAMPLE_FMT_U8P:
2088       return GST_AUDIO_FORMAT_U8;
2089       break;
2090     case AV_SAMPLE_FMT_S16:
2091     case AV_SAMPLE_FMT_S16P:
2092       return GST_AUDIO_FORMAT_S16;
2093       break;
2094     case AV_SAMPLE_FMT_S32:
2095     case AV_SAMPLE_FMT_S32P:
2096       return GST_AUDIO_FORMAT_S32;
2097       break;
2098     case AV_SAMPLE_FMT_FLT:
2099     case AV_SAMPLE_FMT_FLTP:
2100       return GST_AUDIO_FORMAT_F32;
2101       break;
2102     case AV_SAMPLE_FMT_DBL:
2103     case AV_SAMPLE_FMT_DBLP:
2104       return GST_AUDIO_FORMAT_F64;
2105       break;
2106     default:
2107       /* .. */
2108       return GST_AUDIO_FORMAT_UNKNOWN;
2109       break;
2110   }
2111 }
2112
2113 /* Convert a FFMPEG Sample Format and optional AVCodecContext
2114  * to a GstCaps. If the context is ommitted, no fixed values
2115  * for video/audio size will be included in the GstCaps
2116  *
2117  * See below for usefullness
2118  */
2119
2120 static GstCaps *
2121 gst_ffmpeg_smpfmt_to_caps (enum AVSampleFormat sample_fmt,
2122     AVCodecContext * context, AVCodec * codec, enum CodecID codec_id)
2123 {
2124   GstCaps *caps = NULL;
2125   GstAudioFormat format;
2126
2127   format = gst_ffmpeg_smpfmt_to_audioformat (sample_fmt);
2128
2129   if (format != GST_AUDIO_FORMAT_UNKNOWN) {
2130     caps = gst_ff_aud_caps_new (context, codec, codec_id, TRUE, "audio/x-raw",
2131         "format", G_TYPE_STRING, gst_audio_format_to_string (format),
2132         "layout", G_TYPE_STRING, "interleaved", NULL);
2133     GST_LOG ("caps for sample_fmt=%d: %" GST_PTR_FORMAT, sample_fmt, caps);
2134   } else {
2135     GST_LOG ("No caps found for sample_fmt=%d", sample_fmt);
2136   }
2137
2138   return caps;
2139 }
2140
2141 static gboolean
2142 caps_has_field (GstCaps * caps, const gchar * field)
2143 {
2144   guint i, n;
2145
2146   n = gst_caps_get_size (caps);
2147   for (i = 0; i < n; i++) {
2148     GstStructure *s = gst_caps_get_structure (caps, i);
2149
2150     if (gst_structure_has_field (s, field))
2151       return TRUE;
2152   }
2153
2154   return FALSE;
2155 }
2156
2157 GstCaps *
2158 gst_ffmpeg_codectype_to_audio_caps (AVCodecContext * context,
2159     enum CodecID codec_id, gboolean encode, AVCodec * codec)
2160 {
2161   GstCaps *caps = NULL;
2162
2163   GST_DEBUG ("context:%p, codec_id:%d, encode:%d, codec:%p",
2164       context, codec_id, encode, codec);
2165   if (codec)
2166     GST_DEBUG ("sample_fmts:%p, samplerates:%p",
2167         codec->sample_fmts, codec->supported_samplerates);
2168
2169   if (context) {
2170     /* Specific codec context */
2171     caps =
2172         gst_ffmpeg_smpfmt_to_caps (context->sample_fmt, context, codec,
2173         codec_id);
2174   } else {
2175     caps = gst_ff_aud_caps_new (context, codec, codec_id, encode, "audio/x-raw",
2176         "layout", G_TYPE_STRING, "interleaved", NULL);
2177     if (!caps_has_field (caps, "format"))
2178       gst_ffmpeg_audio_set_sample_fmts (caps,
2179           codec ? codec->sample_fmts : NULL);
2180   }
2181
2182   return caps;
2183 }
2184
2185 GstCaps *
2186 gst_ffmpeg_codectype_to_video_caps (AVCodecContext * context,
2187     enum CodecID codec_id, gboolean encode, AVCodec * codec)
2188 {
2189   GstCaps *caps;
2190
2191   GST_LOG ("context:%p, codec_id:%d, encode:%d, codec:%p",
2192       context, codec_id, encode, codec);
2193
2194   if (context) {
2195     caps = gst_ffmpeg_pixfmt_to_caps (context->pix_fmt, context, codec_id);
2196   } else {
2197     caps =
2198         gst_ff_vid_caps_new (context, codec, codec_id, encode, "video/x-raw",
2199         NULL);
2200     if (!caps_has_field (caps, "format"))
2201       gst_ffmpeg_video_set_pix_fmts (caps, codec ? codec->pix_fmts : NULL);
2202   }
2203   return caps;
2204 }
2205
2206 /* Convert a GstCaps (audio/raw) to a FFMPEG SampleFmt
2207  * and other audio properties in a AVCodecContext.
2208  *
2209  * For usefullness, see below
2210  */
2211
2212 static void
2213 gst_ffmpeg_caps_to_smpfmt (const GstCaps * caps,
2214     AVCodecContext * context, gboolean raw)
2215 {
2216   GstStructure *structure;
2217   const gchar *fmt;
2218   GstAudioFormat format = GST_AUDIO_FORMAT_UNKNOWN;
2219
2220   g_return_if_fail (gst_caps_get_size (caps) == 1);
2221
2222   structure = gst_caps_get_structure (caps, 0);
2223
2224   gst_structure_get_int (structure, "channels", &context->channels);
2225   gst_structure_get_int (structure, "rate", &context->sample_rate);
2226   gst_structure_get_int (structure, "block_align", &context->block_align);
2227   gst_structure_get_int (structure, "bitrate", &context->bit_rate);
2228
2229   if (!raw)
2230     return;
2231
2232   if (gst_structure_has_name (structure, "audio/x-raw")) {
2233     if ((fmt = gst_structure_get_string (structure, "format"))) {
2234       format = gst_audio_format_from_string (fmt);
2235     }
2236   }
2237
2238   switch (format) {
2239     case GST_AUDIO_FORMAT_F32:
2240       context->sample_fmt = AV_SAMPLE_FMT_FLT;
2241       break;
2242     case GST_AUDIO_FORMAT_F64:
2243       context->sample_fmt = AV_SAMPLE_FMT_DBL;
2244       break;
2245     case GST_AUDIO_FORMAT_S32:
2246       context->sample_fmt = AV_SAMPLE_FMT_S32;
2247       break;
2248     case GST_AUDIO_FORMAT_S16:
2249       context->sample_fmt = AV_SAMPLE_FMT_S16;
2250       break;
2251     default:
2252       break;
2253   }
2254 }
2255
2256 /* Convert a GstCaps (video/raw) to a FFMPEG PixFmt
2257  * and other video properties in a AVCodecContext.
2258  *
2259  * For usefullness, see below
2260  */
2261
2262 static void
2263 gst_ffmpeg_caps_to_pixfmt (const GstCaps * caps,
2264     AVCodecContext * context, gboolean raw)
2265 {
2266   GstStructure *structure;
2267   const GValue *fps;
2268   const GValue *par = NULL;
2269   const gchar *fmt;
2270   GstVideoFormat format = GST_VIDEO_FORMAT_UNKNOWN;
2271
2272   GST_DEBUG ("converting caps %" GST_PTR_FORMAT, caps);
2273   g_return_if_fail (gst_caps_get_size (caps) == 1);
2274   structure = gst_caps_get_structure (caps, 0);
2275
2276   gst_structure_get_int (structure, "width", &context->width);
2277   gst_structure_get_int (structure, "height", &context->height);
2278   gst_structure_get_int (structure, "bpp", &context->bits_per_coded_sample);
2279
2280   fps = gst_structure_get_value (structure, "framerate");
2281   if (fps != NULL && GST_VALUE_HOLDS_FRACTION (fps)) {
2282
2283     /* somehow these seem mixed up.. */
2284     context->time_base.den = gst_value_get_fraction_numerator (fps);
2285     context->time_base.num = gst_value_get_fraction_denominator (fps);
2286     context->ticks_per_frame = 1;
2287
2288     GST_DEBUG ("setting framerate %d/%d = %lf",
2289         context->time_base.den, context->time_base.num,
2290         1. * context->time_base.den / context->time_base.num);
2291   }
2292
2293   par = gst_structure_get_value (structure, "pixel-aspect-ratio");
2294   if (par && GST_VALUE_HOLDS_FRACTION (par)) {
2295
2296     context->sample_aspect_ratio.num = gst_value_get_fraction_numerator (par);
2297     context->sample_aspect_ratio.den = gst_value_get_fraction_denominator (par);
2298
2299     GST_DEBUG ("setting pixel-aspect-ratio %d/%d = %lf",
2300         context->sample_aspect_ratio.den, context->sample_aspect_ratio.num,
2301         1. * context->sample_aspect_ratio.den /
2302         context->sample_aspect_ratio.num);
2303   }
2304
2305   if (!raw)
2306     return;
2307
2308   g_return_if_fail (fps != NULL && GST_VALUE_HOLDS_FRACTION (fps));
2309
2310   if (gst_structure_has_name (structure, "video/x-raw")) {
2311     if ((fmt = gst_structure_get_string (structure, "format"))) {
2312       format = gst_video_format_from_string (fmt);
2313     }
2314   }
2315
2316   switch (format) {
2317     case GST_VIDEO_FORMAT_YUY2:
2318       context->pix_fmt = PIX_FMT_YUYV422;
2319       break;
2320     case GST_VIDEO_FORMAT_I420:
2321       context->pix_fmt = PIX_FMT_YUV420P;
2322       break;
2323     case GST_VIDEO_FORMAT_A420:
2324       context->pix_fmt = PIX_FMT_YUVA420P;
2325       break;
2326     case GST_VIDEO_FORMAT_Y41B:
2327       context->pix_fmt = PIX_FMT_YUV411P;
2328       break;
2329     case GST_VIDEO_FORMAT_Y42B:
2330       context->pix_fmt = PIX_FMT_YUV422P;
2331       break;
2332     case GST_VIDEO_FORMAT_YUV9:
2333       context->pix_fmt = PIX_FMT_YUV410P;
2334       break;
2335     case GST_VIDEO_FORMAT_Y444:
2336       context->pix_fmt = PIX_FMT_YUV444P;
2337       break;
2338     case GST_VIDEO_FORMAT_GRAY8:
2339       context->pix_fmt = PIX_FMT_GRAY8;
2340       break;
2341     case GST_VIDEO_FORMAT_xRGB:
2342 #if (G_BYTE_ORDER == G_BIG_ENDIAN)
2343       context->pix_fmt = PIX_FMT_RGB32;
2344 #endif
2345       break;
2346     case GST_VIDEO_FORMAT_BGRx:
2347 #if (G_BYTE_ORDER == G_LITTLE_ENDIAN)
2348       context->pix_fmt = PIX_FMT_RGB32;
2349 #endif
2350       break;
2351     case GST_VIDEO_FORMAT_RGB:
2352       context->pix_fmt = PIX_FMT_RGB24;
2353       break;
2354     case GST_VIDEO_FORMAT_BGR:
2355       context->pix_fmt = PIX_FMT_BGR24;
2356       break;
2357     case GST_VIDEO_FORMAT_RGB16:
2358       context->pix_fmt = PIX_FMT_RGB565;
2359       break;
2360     case GST_VIDEO_FORMAT_RGB15:
2361       context->pix_fmt = PIX_FMT_RGB555;
2362       break;
2363     case GST_VIDEO_FORMAT_RGB8P:
2364       context->pix_fmt = PIX_FMT_PAL8;
2365       break;
2366     default:
2367       break;
2368   }
2369 }
2370
2371 typedef struct
2372 {
2373   GstVideoFormat format;
2374   enum PixelFormat pixfmt;
2375 } PixToFmt;
2376
2377 /* FIXME : FILLME */
2378 static const PixToFmt pixtofmttable[] = {
2379   /* GST_VIDEO_FORMAT_I420, */
2380   {GST_VIDEO_FORMAT_I420, PIX_FMT_YUV420P},
2381   /* Note : this should use a different chroma placement */
2382   {GST_VIDEO_FORMAT_I420, PIX_FMT_YUVJ420P},
2383
2384   /* GST_VIDEO_FORMAT_YV12, */
2385   /* GST_VIDEO_FORMAT_YUY2, */
2386   {GST_VIDEO_FORMAT_YUY2, PIX_FMT_YUYV422},
2387   /* GST_VIDEO_FORMAT_UYVY, */
2388   {GST_VIDEO_FORMAT_UYVY, PIX_FMT_UYVY422},
2389   /* GST_VIDEO_FORMAT_AYUV, */
2390   /* GST_VIDEO_FORMAT_RGBx, */
2391   /* GST_VIDEO_FORMAT_BGRx, */
2392   /* GST_VIDEO_FORMAT_xRGB, */
2393   /* GST_VIDEO_FORMAT_xBGR, */
2394   /* GST_VIDEO_FORMAT_RGBA, */
2395   {GST_VIDEO_FORMAT_RGBA, PIX_FMT_RGBA},
2396   /* GST_VIDEO_FORMAT_BGRA, */
2397   {GST_VIDEO_FORMAT_BGRA, PIX_FMT_BGRA},
2398   /* GST_VIDEO_FORMAT_ARGB, */
2399   {GST_VIDEO_FORMAT_ARGB, PIX_FMT_ARGB},
2400   /* GST_VIDEO_FORMAT_ABGR, */
2401   {GST_VIDEO_FORMAT_ABGR, PIX_FMT_ABGR},
2402   /* GST_VIDEO_FORMAT_RGB, */
2403   {GST_VIDEO_FORMAT_RGB, PIX_FMT_RGB24},
2404   /* GST_VIDEO_FORMAT_BGR, */
2405   {GST_VIDEO_FORMAT_BGR, PIX_FMT_BGR24},
2406   /* GST_VIDEO_FORMAT_Y41B, */
2407   {GST_VIDEO_FORMAT_Y41B, PIX_FMT_YUV411P},
2408   /* GST_VIDEO_FORMAT_Y42B, */
2409   {GST_VIDEO_FORMAT_Y42B, PIX_FMT_YUV422P},
2410   {GST_VIDEO_FORMAT_Y42B, PIX_FMT_YUVJ422P},
2411   /* GST_VIDEO_FORMAT_YVYU, */
2412   /* GST_VIDEO_FORMAT_Y444, */
2413   {GST_VIDEO_FORMAT_Y444, PIX_FMT_YUV444P},
2414   {GST_VIDEO_FORMAT_Y444, PIX_FMT_YUVJ444P},
2415   /* GST_VIDEO_FORMAT_v210, */
2416   /* GST_VIDEO_FORMAT_v216, */
2417   /* GST_VIDEO_FORMAT_NV12, */
2418   {GST_VIDEO_FORMAT_NV12, PIX_FMT_NV12},
2419   /* GST_VIDEO_FORMAT_NV21, */
2420   {GST_VIDEO_FORMAT_NV21, PIX_FMT_NV21},
2421   /* GST_VIDEO_FORMAT_GRAY8, */
2422   {GST_VIDEO_FORMAT_GRAY8, PIX_FMT_GRAY8},
2423   /* GST_VIDEO_FORMAT_GRAY16_BE, */
2424   {GST_VIDEO_FORMAT_GRAY16_BE, PIX_FMT_GRAY16BE},
2425   /* GST_VIDEO_FORMAT_GRAY16_LE, */
2426   {GST_VIDEO_FORMAT_GRAY16_LE, PIX_FMT_GRAY16LE},
2427   /* GST_VIDEO_FORMAT_v308, */
2428   /* GST_VIDEO_FORMAT_Y800, */
2429   /* GST_VIDEO_FORMAT_Y16, */
2430   /* GST_VIDEO_FORMAT_RGB16, */
2431   {GST_VIDEO_FORMAT_RGB16, PIX_FMT_RGB565},
2432   /* GST_VIDEO_FORMAT_BGR16, */
2433   /* GST_VIDEO_FORMAT_RGB15, */
2434   {GST_VIDEO_FORMAT_RGB15, PIX_FMT_RGB555},
2435   /* GST_VIDEO_FORMAT_BGR15, */
2436   /* GST_VIDEO_FORMAT_UYVP, */
2437   /* GST_VIDEO_FORMAT_A420, */
2438   {GST_VIDEO_FORMAT_A420, PIX_FMT_YUVA420P},
2439   /* GST_VIDEO_FORMAT_RGB8_PALETTED, */
2440   {GST_VIDEO_FORMAT_RGB8P, PIX_FMT_PAL8},
2441   /* GST_VIDEO_FORMAT_YUV9, */
2442   {GST_VIDEO_FORMAT_YUV9, PIX_FMT_YUV410P},
2443   /* GST_VIDEO_FORMAT_YVU9, */
2444   /* GST_VIDEO_FORMAT_IYU1, */
2445   /* GST_VIDEO_FORMAT_ARGB64, */
2446   /* GST_VIDEO_FORMAT_AYUV64, */
2447   /* GST_VIDEO_FORMAT_r210, */
2448   {GST_VIDEO_FORMAT_I420_10LE, PIX_FMT_YUV420P10LE},
2449   {GST_VIDEO_FORMAT_I420_10BE, PIX_FMT_YUV420P10BE},
2450   {GST_VIDEO_FORMAT_I422_10LE, PIX_FMT_YUV422P10LE},
2451   {GST_VIDEO_FORMAT_I422_10BE, PIX_FMT_YUV422P10BE},
2452   {GST_VIDEO_FORMAT_Y444_10LE, PIX_FMT_YUV444P10LE},
2453   {GST_VIDEO_FORMAT_Y444_10BE, PIX_FMT_YUV444P10BE},
2454   {GST_VIDEO_FORMAT_GBR, PIX_FMT_GBRP},
2455   {GST_VIDEO_FORMAT_GBR_10LE, PIX_FMT_GBRP10LE},
2456   {GST_VIDEO_FORMAT_GBR_10BE, PIX_FMT_GBRP10BE},
2457 };
2458
2459 GstVideoFormat
2460 gst_ffmpeg_pixfmt_to_videoformat (enum PixelFormat pixfmt)
2461 {
2462   guint i;
2463
2464   for (i = 0; i < G_N_ELEMENTS (pixtofmttable); i++)
2465     if (pixtofmttable[i].pixfmt == pixfmt)
2466       return pixtofmttable[i].format;
2467
2468   GST_DEBUG ("Unknown pixel format %d", pixfmt);
2469   return GST_VIDEO_FORMAT_UNKNOWN;
2470 }
2471
2472 enum PixelFormat
2473 gst_ffmpeg_videoformat_to_pixfmt (GstVideoFormat format)
2474 {
2475   guint i;
2476
2477   for (i = 0; i < G_N_ELEMENTS (pixtofmttable); i++)
2478     if (pixtofmttable[i].format == format)
2479       return pixtofmttable[i].pixfmt;
2480   return PIX_FMT_NONE;
2481 }
2482
2483 void
2484 gst_ffmpeg_videoinfo_to_context (GstVideoInfo * info, AVCodecContext * context)
2485 {
2486   gint i, bpp = 0;
2487
2488   context->width = GST_VIDEO_INFO_WIDTH (info);
2489   context->height = GST_VIDEO_INFO_HEIGHT (info);
2490   for (i = 0; i < GST_VIDEO_INFO_N_COMPONENTS (info); i++)
2491     bpp += GST_VIDEO_INFO_COMP_DEPTH (info, i);
2492   context->bits_per_coded_sample = bpp;
2493
2494   context->ticks_per_frame = 1;
2495   context->time_base.den = GST_VIDEO_INFO_FPS_N (info);
2496   context->time_base.num = GST_VIDEO_INFO_FPS_D (info);
2497
2498   context->sample_aspect_ratio.num = GST_VIDEO_INFO_PAR_N (info);
2499   context->sample_aspect_ratio.den = GST_VIDEO_INFO_PAR_D (info);
2500
2501   context->pix_fmt =
2502       gst_ffmpeg_videoformat_to_pixfmt (GST_VIDEO_INFO_FORMAT (info));
2503 }
2504
2505 void
2506 gst_ffmpeg_audioinfo_to_context (GstAudioInfo * info, AVCodecContext * context)
2507 {
2508   const AVCodec *codec;
2509   const enum AVSampleFormat *smpl_fmts;
2510   enum AVSampleFormat smpl_fmt = -1;
2511
2512   context->channels = info->channels;
2513   context->sample_rate = info->rate;
2514   context->channel_layout =
2515       gst_ffmpeg_channel_positions_to_layout (info->position, info->channels);
2516
2517   codec = context->codec;
2518
2519   smpl_fmts = codec->sample_fmts;
2520
2521   switch (info->finfo->format) {
2522     case GST_AUDIO_FORMAT_F32:
2523       if (smpl_fmts) {
2524         while (*smpl_fmts != -1) {
2525           if (*smpl_fmts == AV_SAMPLE_FMT_FLT) {
2526             smpl_fmt = *smpl_fmts;
2527             break;
2528           } else if (*smpl_fmts == AV_SAMPLE_FMT_FLTP) {
2529             smpl_fmt = *smpl_fmts;
2530           }
2531
2532           smpl_fmts++;
2533         }
2534       } else {
2535         smpl_fmt = AV_SAMPLE_FMT_FLT;
2536       }
2537       break;
2538     case GST_AUDIO_FORMAT_F64:
2539       if (smpl_fmts) {
2540         while (*smpl_fmts != -1) {
2541           if (*smpl_fmts == AV_SAMPLE_FMT_DBL) {
2542             smpl_fmt = *smpl_fmts;
2543             break;
2544           } else if (*smpl_fmts == AV_SAMPLE_FMT_DBLP) {
2545             smpl_fmt = *smpl_fmts;
2546           }
2547
2548           smpl_fmts++;
2549         }
2550       } else {
2551         smpl_fmt = AV_SAMPLE_FMT_DBL;
2552       }
2553       break;
2554     case GST_AUDIO_FORMAT_S32:
2555       if (smpl_fmts) {
2556         while (*smpl_fmts != -1) {
2557           if (*smpl_fmts == AV_SAMPLE_FMT_S32) {
2558             smpl_fmt = *smpl_fmts;
2559             break;
2560           } else if (*smpl_fmts == AV_SAMPLE_FMT_S32P) {
2561             smpl_fmt = *smpl_fmts;
2562           }
2563
2564           smpl_fmts++;
2565         }
2566       } else {
2567         smpl_fmt = AV_SAMPLE_FMT_S32;
2568       }
2569       break;
2570     case GST_AUDIO_FORMAT_S16:
2571       if (smpl_fmts) {
2572         while (*smpl_fmts != -1) {
2573           if (*smpl_fmts == AV_SAMPLE_FMT_S16) {
2574             smpl_fmt = *smpl_fmts;
2575             break;
2576           } else if (*smpl_fmts == AV_SAMPLE_FMT_S16P) {
2577             smpl_fmt = *smpl_fmts;
2578           }
2579
2580           smpl_fmts++;
2581         }
2582       } else {
2583         smpl_fmt = AV_SAMPLE_FMT_S16;
2584       }
2585       break;
2586     case GST_AUDIO_FORMAT_U8:
2587       if (smpl_fmts) {
2588         while (*smpl_fmts != -1) {
2589           if (*smpl_fmts == AV_SAMPLE_FMT_U8) {
2590             smpl_fmt = *smpl_fmts;
2591             break;
2592           } else if (*smpl_fmts == AV_SAMPLE_FMT_U8P) {
2593             smpl_fmt = *smpl_fmts;
2594           }
2595
2596           smpl_fmts++;
2597         }
2598       } else {
2599         smpl_fmt = AV_SAMPLE_FMT_U8;
2600       }
2601       break;
2602     default:
2603       break;
2604   }
2605
2606   g_assert (smpl_fmt != -1);
2607
2608   context->sample_fmt = smpl_fmt;
2609 }
2610
2611 /* Convert a GstCaps and a FFMPEG codec Type to a
2612  * AVCodecContext. If the context is ommitted, no fixed values
2613  * for video/audio size will be included in the context
2614  *
2615  * AVMediaType is primarily meant for uncompressed data GstCaps!
2616  */
2617
2618 void
2619 gst_ffmpeg_caps_with_codectype (enum AVMediaType type,
2620     const GstCaps * caps, AVCodecContext * context)
2621 {
2622   if (context == NULL)
2623     return;
2624
2625   switch (type) {
2626     case AVMEDIA_TYPE_VIDEO:
2627       gst_ffmpeg_caps_to_pixfmt (caps, context, TRUE);
2628       break;
2629
2630     case AVMEDIA_TYPE_AUDIO:
2631       gst_ffmpeg_caps_to_smpfmt (caps, context, TRUE);
2632       break;
2633
2634     default:
2635       /* unknown */
2636       break;
2637   }
2638 }
2639
2640 #if 0
2641 static void
2642 nal_escape (guint8 * dst, guint8 * src, guint size, guint * destsize)
2643 {
2644   guint8 *dstp = dst;
2645   guint8 *srcp = src;
2646   guint8 *end = src + size;
2647   gint count = 0;
2648
2649   while (srcp < end) {
2650     if (count == 2 && *srcp <= 0x03) {
2651       GST_DEBUG ("added escape code");
2652       *dstp++ = 0x03;
2653       count = 0;
2654     }
2655     if (*srcp == 0)
2656       count++;
2657     else
2658       count = 0;
2659
2660     GST_DEBUG ("copy %02x, count %d", *srcp, count);
2661     *dstp++ = *srcp++;
2662   }
2663   *destsize = dstp - dst;
2664 }
2665
2666 /* copy the config, escaping NAL units as we iterate them, if something fails we
2667  * copy everything and hope for the best. */
2668 static void
2669 copy_config (guint8 * dst, guint8 * src, guint size, guint * destsize)
2670 {
2671   guint8 *dstp = dst;
2672   guint8 *srcp = src;
2673   gint cnt, i;
2674   guint nalsize, esize;
2675
2676   /* check size */
2677   if (size < 7)
2678     goto full_copy;
2679
2680   /* check version */
2681   if (*srcp != 1)
2682     goto full_copy;
2683
2684   cnt = *(srcp + 5) & 0x1f;     /* Number of sps */
2685
2686   GST_DEBUG ("num SPS %d", cnt);
2687
2688   memcpy (dstp, srcp, 6);
2689   srcp += 6;
2690   dstp += 6;
2691
2692   for (i = 0; i < cnt; i++) {
2693     GST_DEBUG ("copy SPS %d", i);
2694     nalsize = (srcp[0] << 8) | srcp[1];
2695     nal_escape (dstp + 2, srcp + 2, nalsize, &esize);
2696     dstp[0] = esize >> 8;
2697     dstp[1] = esize & 0xff;
2698     dstp += esize + 2;
2699     srcp += nalsize + 2;
2700   }
2701
2702   cnt = *(dstp++) = *(srcp++);  /* Number of pps */
2703
2704   GST_DEBUG ("num PPS %d", cnt);
2705
2706   for (i = 0; i < cnt; i++) {
2707     GST_DEBUG ("copy PPS %d", i);
2708     nalsize = (srcp[0] << 8) | srcp[1];
2709     nal_escape (dstp + 2, srcp + 2, nalsize, &esize);
2710     dstp[0] = esize >> 8;
2711     dstp[1] = esize & 0xff;
2712     dstp += esize + 2;
2713     srcp += nalsize + 2;
2714   }
2715   *destsize = dstp - dst;
2716
2717   return;
2718
2719 full_copy:
2720   {
2721     GST_DEBUG ("something unexpected, doing full copy");
2722     memcpy (dst, src, size);
2723     *destsize = size;
2724     return;
2725   }
2726 }
2727 #endif
2728
2729 /*
2730  * caps_with_codecid () transforms a GstCaps for a known codec
2731  * ID into a filled-in context.
2732  * codec_data from caps will override possible extradata already in the context
2733  */
2734
2735 void
2736 gst_ffmpeg_caps_with_codecid (enum CodecID codec_id,
2737     enum AVMediaType codec_type, const GstCaps * caps, AVCodecContext * context)
2738 {
2739   GstStructure *str;
2740   const GValue *value;
2741   GstBuffer *buf;
2742
2743   GST_LOG ("codec_id:%d, codec_type:%d, caps:%" GST_PTR_FORMAT " context:%p",
2744       codec_id, codec_type, caps, context);
2745
2746   if (!context || !gst_caps_get_size (caps))
2747     return;
2748
2749   str = gst_caps_get_structure (caps, 0);
2750
2751   /* extradata parsing (esds [mpeg4], wma/wmv, msmpeg4v1/2/3, etc.) */
2752   if ((value = gst_structure_get_value (str, "codec_data"))) {
2753     GstMapInfo map;
2754
2755     buf = gst_value_get_buffer (value);
2756     gst_buffer_map (buf, &map, GST_MAP_READ);
2757
2758     /* free the old one if it is there */
2759     if (context->extradata)
2760       av_free (context->extradata);
2761
2762 #if 0
2763     if (codec_id == AV_CODEC_ID_H264) {
2764       guint extrasize;
2765
2766       GST_DEBUG ("copy, escaping codec_data %d", size);
2767       /* ffmpeg h264 expects the codec_data to be escaped, there is no real
2768        * reason for this but let's just escape it for now. Start by allocating
2769        * enough space, x2 is more than enough.
2770        *
2771        * FIXME, we disabled escaping because some file already contain escaped
2772        * codec_data and then we escape twice and fail. It's better to leave it
2773        * as is, as that is what most players do. */
2774       context->extradata =
2775           av_mallocz (GST_ROUND_UP_16 (size * 2 +
2776               FF_INPUT_BUFFER_PADDING_SIZE));
2777       copy_config (context->extradata, data, size, &extrasize);
2778       GST_DEBUG ("escaped size: %d", extrasize);
2779       context->extradata_size = extrasize;
2780     } else
2781 #endif
2782     {
2783       /* allocate with enough padding */
2784       GST_DEBUG ("copy codec_data");
2785       context->extradata =
2786           av_mallocz (GST_ROUND_UP_16 (map.size +
2787               FF_INPUT_BUFFER_PADDING_SIZE));
2788       memcpy (context->extradata, map.data, map.size);
2789       context->extradata_size = map.size;
2790     }
2791
2792     /* Hack for VC1. Sometimes the first (length) byte is 0 for some files */
2793     if (codec_id == AV_CODEC_ID_VC1 && map.size > 0 && map.data[0] == 0) {
2794       context->extradata[0] = (guint8) map.size;
2795     }
2796
2797     GST_DEBUG ("have codec data of size %" G_GSIZE_FORMAT, map.size);
2798
2799     gst_buffer_unmap (buf, &map);
2800   } else if (context->extradata == NULL && codec_id != AV_CODEC_ID_AAC_LATM &&
2801       codec_id != AV_CODEC_ID_FLAC) {
2802     /* no extradata, alloc dummy with 0 sized, some codecs insist on reading
2803      * extradata anyway which makes then segfault. */
2804     context->extradata =
2805         av_mallocz (GST_ROUND_UP_16 (FF_INPUT_BUFFER_PADDING_SIZE));
2806     context->extradata_size = 0;
2807     GST_DEBUG ("no codec data");
2808   }
2809
2810   switch (codec_id) {
2811     case AV_CODEC_ID_MPEG4:
2812     {
2813       const gchar *mime = gst_structure_get_name (str);
2814
2815       if (!strcmp (mime, "video/x-divx"))
2816         context->codec_tag = GST_MAKE_FOURCC ('D', 'I', 'V', 'X');
2817       else if (!strcmp (mime, "video/x-xvid"))
2818         context->codec_tag = GST_MAKE_FOURCC ('X', 'V', 'I', 'D');
2819       else if (!strcmp (mime, "video/x-3ivx"))
2820         context->codec_tag = GST_MAKE_FOURCC ('3', 'I', 'V', '1');
2821       else if (!strcmp (mime, "video/mpeg"))
2822         context->codec_tag = GST_MAKE_FOURCC ('m', 'p', '4', 'v');
2823     }
2824       break;
2825
2826     case AV_CODEC_ID_SVQ3:
2827       /* FIXME: this is a workaround for older gst-plugins releases
2828        * (<= 0.8.9). This should be removed at some point, because
2829        * it causes wrong decoded frame order. */
2830       if (!context->extradata) {
2831         gint halfpel_flag, thirdpel_flag, low_delay, unknown_svq3_flag;
2832         guint16 flags;
2833
2834         if (gst_structure_get_int (str, "halfpel_flag", &halfpel_flag) ||
2835             gst_structure_get_int (str, "thirdpel_flag", &thirdpel_flag) ||
2836             gst_structure_get_int (str, "low_delay", &low_delay) ||
2837             gst_structure_get_int (str, "unknown_svq3_flag",
2838                 &unknown_svq3_flag)) {
2839           context->extradata = (guint8 *) av_mallocz (0x64);
2840           g_stpcpy ((gchar *) context->extradata, "SVQ3");
2841           flags = 1 << 3;
2842           flags |= low_delay;
2843           flags = flags << 2;
2844           flags |= unknown_svq3_flag;
2845           flags = flags << 6;
2846           flags |= halfpel_flag;
2847           flags = flags << 1;
2848           flags |= thirdpel_flag;
2849           flags = flags << 3;
2850
2851           flags = GUINT16_FROM_LE (flags);
2852
2853           memcpy ((gchar *) context->extradata + 0x62, &flags, 2);
2854           context->extradata_size = 0x64;
2855         }
2856       }
2857       break;
2858
2859     case AV_CODEC_ID_MSRLE:
2860     case AV_CODEC_ID_QTRLE:
2861     case AV_CODEC_ID_TSCC:
2862     case AV_CODEC_ID_CSCD:
2863     case AV_CODEC_ID_APE:
2864     {
2865       gint depth;
2866
2867       if (gst_structure_get_int (str, "depth", &depth)) {
2868         context->bits_per_coded_sample = depth;
2869       } else {
2870         GST_WARNING ("No depth field in caps %" GST_PTR_FORMAT, caps);
2871       }
2872
2873     }
2874       break;
2875
2876     case AV_CODEC_ID_RV10:
2877     case AV_CODEC_ID_RV20:
2878     case AV_CODEC_ID_RV30:
2879     case AV_CODEC_ID_RV40:
2880     {
2881       gint format;
2882
2883       if (gst_structure_get_int (str, "format", &format))
2884         context->sub_id = format;
2885
2886       break;
2887     }
2888     case AV_CODEC_ID_COOK:
2889     case AV_CODEC_ID_RA_288:
2890     case AV_CODEC_ID_RA_144:
2891     case AV_CODEC_ID_SIPR:
2892     {
2893       gint leaf_size;
2894       gint bitrate;
2895
2896       if (gst_structure_get_int (str, "leaf_size", &leaf_size))
2897         context->block_align = leaf_size;
2898       if (gst_structure_get_int (str, "bitrate", &bitrate))
2899         context->bit_rate = bitrate;
2900     }
2901     case AV_CODEC_ID_ALAC:
2902       gst_structure_get_int (str, "samplesize",
2903           &context->bits_per_coded_sample);
2904       break;
2905
2906     case AV_CODEC_ID_DVVIDEO:
2907     {
2908       const gchar *format;
2909
2910       if ((format = gst_structure_get_string (str, "format"))) {
2911
2912         if (g_str_equal (format, "YUY2"))
2913           context->pix_fmt = PIX_FMT_YUYV422;
2914         else if (g_str_equal (format, "I420"))
2915           context->pix_fmt = PIX_FMT_YUV420P;
2916         else if (g_str_equal (format, "A420"))
2917           context->pix_fmt = PIX_FMT_YUVA420P;
2918         else if (g_str_equal (format, "Y41B"))
2919           context->pix_fmt = PIX_FMT_YUV411P;
2920         else if (g_str_equal (format, "Y42B"))
2921           context->pix_fmt = PIX_FMT_YUV422P;
2922         else if (g_str_equal (format, "YUV9"))
2923           context->pix_fmt = PIX_FMT_YUV410P;
2924         else {
2925           GST_WARNING ("couldn't convert format %s" " to a pixel format",
2926               format);
2927         }
2928       } else
2929         GST_WARNING ("No specified format");
2930       break;
2931     }
2932     case AV_CODEC_ID_H263P:
2933     {
2934       gboolean val;
2935
2936       if (!gst_structure_get_boolean (str, "annex-f", &val) || val)
2937         context->flags |= CODEC_FLAG_4MV;
2938       else
2939         context->flags &= ~CODEC_FLAG_4MV;
2940       if ((!gst_structure_get_boolean (str, "annex-i", &val) || val) &&
2941           (!gst_structure_get_boolean (str, "annex-t", &val) || val))
2942         context->flags |= CODEC_FLAG_AC_PRED;
2943       else
2944         context->flags &= ~CODEC_FLAG_AC_PRED;
2945       if (!gst_structure_get_boolean (str, "annex-j", &val) || val)
2946         context->flags |= CODEC_FLAG_LOOP_FILTER;
2947       else
2948         context->flags &= ~CODEC_FLAG_LOOP_FILTER;
2949       break;
2950     }
2951     case AV_CODEC_ID_ADPCM_G726:
2952     {
2953       const gchar *layout;
2954
2955       if ((layout = gst_structure_get_string (str, "layout"))) {
2956         if (!strcmp (layout, "g721")) {
2957           context->sample_rate = 8000;
2958           context->channels = 1;
2959           context->bit_rate = 32000;
2960         }
2961       }
2962       break;
2963     }
2964     default:
2965       break;
2966   }
2967
2968   if (!gst_caps_is_fixed (caps))
2969     return;
2970
2971   /* common properties (width, height, fps) */
2972   switch (codec_type) {
2973     case AVMEDIA_TYPE_VIDEO:
2974       gst_ffmpeg_caps_to_pixfmt (caps, context,
2975           codec_id == AV_CODEC_ID_RAWVIDEO);
2976       break;
2977     case AVMEDIA_TYPE_AUDIO:
2978       gst_ffmpeg_caps_to_smpfmt (caps, context, FALSE);
2979       break;
2980     default:
2981       break;
2982   }
2983
2984   /* fixup of default settings */
2985   switch (codec_id) {
2986     case AV_CODEC_ID_QCELP:
2987       /* QCELP is always mono, no matter what the caps say */
2988       context->channels = 1;
2989       break;
2990     case AV_CODEC_ID_ADPCM_G726:
2991       if (context->sample_rate && context->bit_rate)
2992         context->bits_per_coded_sample =
2993             context->bit_rate / context->sample_rate;
2994       break;
2995     default:
2996       break;
2997   }
2998 }
2999
3000 /* _formatid_to_caps () is meant for muxers/demuxers, it
3001  * transforms a name (ffmpeg way of ID'ing these, why don't
3002  * they have unique numerical IDs?) to the corresponding
3003  * caps belonging to that mux-format
3004  *
3005  * Note: we don't need any additional info because the caps
3006  * isn't supposed to contain any useful info besides the
3007  * media type anyway
3008  */
3009
3010 GstCaps *
3011 gst_ffmpeg_formatid_to_caps (const gchar * format_name)
3012 {
3013   GstCaps *caps = NULL;
3014
3015   if (!strcmp (format_name, "mpeg")) {
3016     caps = gst_caps_new_simple ("video/mpeg",
3017         "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
3018   } else if (!strcmp (format_name, "mpegts")) {
3019     caps = gst_caps_new_simple ("video/mpegts",
3020         "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
3021   } else if (!strcmp (format_name, "rm")) {
3022     caps = gst_caps_new_simple ("application/x-pn-realmedia",
3023         "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
3024   } else if (!strcmp (format_name, "asf")) {
3025     caps = gst_caps_new_empty_simple ("video/x-ms-asf");
3026   } else if (!strcmp (format_name, "avi")) {
3027     caps = gst_caps_new_empty_simple ("video/x-msvideo");
3028   } else if (!strcmp (format_name, "wav")) {
3029     caps = gst_caps_new_empty_simple ("audio/x-wav");
3030   } else if (!strcmp (format_name, "ape")) {
3031     caps = gst_caps_new_empty_simple ("application/x-ape");
3032   } else if (!strcmp (format_name, "swf")) {
3033     caps = gst_caps_new_empty_simple ("application/x-shockwave-flash");
3034   } else if (!strcmp (format_name, "au")) {
3035     caps = gst_caps_new_empty_simple ("audio/x-au");
3036   } else if (!strcmp (format_name, "dv")) {
3037     caps = gst_caps_new_simple ("video/x-dv",
3038         "systemstream", G_TYPE_BOOLEAN, TRUE, NULL);
3039   } else if (!strcmp (format_name, "4xm")) {
3040     caps = gst_caps_new_empty_simple ("video/x-4xm");
3041   } else if (!strcmp (format_name, "matroska")) {
3042     caps = gst_caps_new_empty_simple ("video/x-matroska");
3043   } else if (!strcmp (format_name, "mp3")) {
3044     caps = gst_caps_new_empty_simple ("application/x-id3");
3045   } else if (!strcmp (format_name, "flic")) {
3046     caps = gst_caps_new_empty_simple ("video/x-fli");
3047   } else if (!strcmp (format_name, "flv")) {
3048     caps = gst_caps_new_empty_simple ("video/x-flv");
3049   } else if (!strcmp (format_name, "tta")) {
3050     caps = gst_caps_new_empty_simple ("audio/x-ttafile");
3051   } else if (!strcmp (format_name, "aiff")) {
3052     caps = gst_caps_new_empty_simple ("audio/x-aiff");
3053   } else if (!strcmp (format_name, "mov_mp4_m4a_3gp_3g2")) {
3054     caps =
3055         gst_caps_from_string
3056         ("application/x-3gp; video/quicktime; audio/x-m4a");
3057   } else if (!strcmp (format_name, "mov")) {
3058     caps = gst_caps_from_string ("video/quicktime,variant=(string)apple");
3059   } else if (!strcmp (format_name, "mp4")) {
3060     caps = gst_caps_from_string ("video/quicktime,variant=(string)iso");
3061   } else if (!strcmp (format_name, "3gp")) {
3062     caps = gst_caps_from_string ("video/quicktime,variant=(string)3gpp");
3063   } else if (!strcmp (format_name, "3g2")) {
3064     caps = gst_caps_from_string ("video/quicktime,variant=(string)3g2");
3065   } else if (!strcmp (format_name, "psp")) {
3066     caps = gst_caps_from_string ("video/quicktime,variant=(string)psp");
3067   } else if (!strcmp (format_name, "ipod")) {
3068     caps = gst_caps_from_string ("video/quicktime,variant=(string)ipod");
3069   } else if (!strcmp (format_name, "aac")) {
3070     caps = gst_caps_new_simple ("audio/mpeg",
3071         "mpegversion", G_TYPE_INT, 4, NULL);
3072   } else if (!strcmp (format_name, "gif")) {
3073     caps = gst_caps_from_string ("image/gif");
3074   } else if (!strcmp (format_name, "ogg")) {
3075     caps = gst_caps_from_string ("application/ogg");
3076   } else if (!strcmp (format_name, "mxf") || !strcmp (format_name, "mxf_d10")) {
3077     caps = gst_caps_from_string ("application/mxf");
3078   } else if (!strcmp (format_name, "gxf")) {
3079     caps = gst_caps_from_string ("application/gxf");
3080   } else if (!strcmp (format_name, "yuv4mpegpipe")) {
3081     caps = gst_caps_new_simple ("application/x-yuv4mpeg",
3082         "y4mversion", G_TYPE_INT, 2, NULL);
3083   } else if (!strcmp (format_name, "mpc")) {
3084     caps = gst_caps_from_string ("audio/x-musepack, streamversion = (int) 7");
3085   } else if (!strcmp (format_name, "vqf")) {
3086     caps = gst_caps_from_string ("audio/x-vqf");
3087   } else if (!strcmp (format_name, "nsv")) {
3088     caps = gst_caps_from_string ("video/x-nsv");
3089   } else if (!strcmp (format_name, "amr")) {
3090     caps = gst_caps_from_string ("audio/x-amr-nb-sh");
3091   } else if (!strcmp (format_name, "webm")) {
3092     caps = gst_caps_from_string ("video/webm");
3093   } else if (!strcmp (format_name, "voc")) {
3094     caps = gst_caps_from_string ("audio/x-voc");
3095   } else if (!strcmp (format_name, "pva")) {
3096     caps = gst_caps_from_string ("video/x-pva");
3097   } else {
3098     gchar *name;
3099
3100     GST_LOG ("Could not create stream format caps for %s", format_name);
3101     name = g_strdup_printf ("application/x-gst-av-%s", format_name);
3102     caps = gst_caps_new_empty_simple (name);
3103     g_free (name);
3104   }
3105
3106   return caps;
3107 }
3108
3109 gboolean
3110 gst_ffmpeg_formatid_get_codecids (const gchar * format_name,
3111     enum CodecID ** video_codec_list, enum CodecID ** audio_codec_list,
3112     AVOutputFormat * plugin)
3113 {
3114   static enum CodecID tmp_vlist[] = {
3115     AV_CODEC_ID_NONE,
3116     AV_CODEC_ID_NONE
3117   };
3118   static enum CodecID tmp_alist[] = {
3119     AV_CODEC_ID_NONE,
3120     AV_CODEC_ID_NONE
3121   };
3122
3123   GST_LOG ("format_name : %s", format_name);
3124
3125   if (!strcmp (format_name, "mp4")) {
3126     static enum CodecID mp4_video_list[] = {
3127       AV_CODEC_ID_MPEG4, AV_CODEC_ID_H264,
3128       AV_CODEC_ID_MJPEG,
3129       AV_CODEC_ID_NONE
3130     };
3131     static enum CodecID mp4_audio_list[] = {
3132       AV_CODEC_ID_AAC, AV_CODEC_ID_MP3,
3133       AV_CODEC_ID_NONE
3134     };
3135
3136     *video_codec_list = mp4_video_list;
3137     *audio_codec_list = mp4_audio_list;
3138   } else if (!strcmp (format_name, "mpeg")) {
3139     static enum CodecID mpeg_video_list[] = { AV_CODEC_ID_MPEG1VIDEO,
3140       AV_CODEC_ID_MPEG2VIDEO,
3141       AV_CODEC_ID_H264,
3142       AV_CODEC_ID_NONE
3143     };
3144     static enum CodecID mpeg_audio_list[] = { AV_CODEC_ID_MP1,
3145       AV_CODEC_ID_MP2,
3146       AV_CODEC_ID_MP3,
3147       AV_CODEC_ID_NONE
3148     };
3149
3150     *video_codec_list = mpeg_video_list;
3151     *audio_codec_list = mpeg_audio_list;
3152   } else if (!strcmp (format_name, "dvd")) {
3153     static enum CodecID mpeg_video_list[] = { AV_CODEC_ID_MPEG2VIDEO,
3154       AV_CODEC_ID_NONE
3155     };
3156     static enum CodecID mpeg_audio_list[] = { AV_CODEC_ID_MP2,
3157       AV_CODEC_ID_AC3,
3158       AV_CODEC_ID_DTS,
3159       AV_CODEC_ID_PCM_S16BE,
3160       AV_CODEC_ID_NONE
3161     };
3162
3163     *video_codec_list = mpeg_video_list;
3164     *audio_codec_list = mpeg_audio_list;
3165   } else if (!strcmp (format_name, "mpegts")) {
3166     static enum CodecID mpegts_video_list[] = { AV_CODEC_ID_MPEG1VIDEO,
3167       AV_CODEC_ID_MPEG2VIDEO,
3168       AV_CODEC_ID_H264,
3169       AV_CODEC_ID_NONE
3170     };
3171     static enum CodecID mpegts_audio_list[] = { AV_CODEC_ID_MP2,
3172       AV_CODEC_ID_MP3,
3173       AV_CODEC_ID_AC3,
3174       AV_CODEC_ID_DTS,
3175       AV_CODEC_ID_AAC,
3176       AV_CODEC_ID_NONE
3177     };
3178
3179     *video_codec_list = mpegts_video_list;
3180     *audio_codec_list = mpegts_audio_list;
3181   } else if (!strcmp (format_name, "vob")) {
3182     static enum CodecID vob_video_list[] =
3183         { AV_CODEC_ID_MPEG2VIDEO, AV_CODEC_ID_NONE };
3184     static enum CodecID vob_audio_list[] = { AV_CODEC_ID_MP2, AV_CODEC_ID_AC3,
3185       AV_CODEC_ID_DTS, AV_CODEC_ID_NONE
3186     };
3187
3188     *video_codec_list = vob_video_list;
3189     *audio_codec_list = vob_audio_list;
3190   } else if (!strcmp (format_name, "flv")) {
3191     static enum CodecID flv_video_list[] =
3192         { AV_CODEC_ID_FLV1, AV_CODEC_ID_NONE };
3193     static enum CodecID flv_audio_list[] =
3194         { AV_CODEC_ID_MP3, AV_CODEC_ID_NONE };
3195
3196     *video_codec_list = flv_video_list;
3197     *audio_codec_list = flv_audio_list;
3198   } else if (!strcmp (format_name, "asf")) {
3199     static enum CodecID asf_video_list[] =
3200         { AV_CODEC_ID_WMV1, AV_CODEC_ID_WMV2, AV_CODEC_ID_MSMPEG4V3,
3201       AV_CODEC_ID_NONE
3202     };
3203     static enum CodecID asf_audio_list[] =
3204         { AV_CODEC_ID_WMAV1, AV_CODEC_ID_WMAV2, AV_CODEC_ID_MP3,
3205       AV_CODEC_ID_NONE
3206     };
3207
3208     *video_codec_list = asf_video_list;
3209     *audio_codec_list = asf_audio_list;
3210   } else if (!strcmp (format_name, "dv")) {
3211     static enum CodecID dv_video_list[] =
3212         { AV_CODEC_ID_DVVIDEO, AV_CODEC_ID_NONE };
3213     static enum CodecID dv_audio_list[] =
3214         { AV_CODEC_ID_PCM_S16LE, AV_CODEC_ID_NONE };
3215
3216     *video_codec_list = dv_video_list;
3217     *audio_codec_list = dv_audio_list;
3218   } else if (!strcmp (format_name, "mov")) {
3219     static enum CodecID mov_video_list[] = {
3220       AV_CODEC_ID_SVQ1, AV_CODEC_ID_SVQ3, AV_CODEC_ID_MPEG4,
3221       AV_CODEC_ID_H263, AV_CODEC_ID_H263P,
3222       AV_CODEC_ID_H264, AV_CODEC_ID_DVVIDEO,
3223       AV_CODEC_ID_MJPEG,
3224       AV_CODEC_ID_NONE
3225     };
3226     static enum CodecID mov_audio_list[] = {
3227       AV_CODEC_ID_PCM_MULAW, AV_CODEC_ID_PCM_ALAW, AV_CODEC_ID_ADPCM_IMA_QT,
3228       AV_CODEC_ID_MACE3, AV_CODEC_ID_MACE6, AV_CODEC_ID_AAC,
3229       AV_CODEC_ID_AMR_NB, AV_CODEC_ID_AMR_WB,
3230       AV_CODEC_ID_PCM_S16BE, AV_CODEC_ID_PCM_S16LE,
3231       AV_CODEC_ID_MP3, AV_CODEC_ID_NONE
3232     };
3233
3234     *video_codec_list = mov_video_list;
3235     *audio_codec_list = mov_audio_list;
3236   } else if ((!strcmp (format_name, "3gp") || !strcmp (format_name, "3g2"))) {
3237     static enum CodecID tgp_video_list[] = {
3238       AV_CODEC_ID_MPEG4, AV_CODEC_ID_H263, AV_CODEC_ID_H263P, AV_CODEC_ID_H264,
3239       AV_CODEC_ID_NONE
3240     };
3241     static enum CodecID tgp_audio_list[] = {
3242       AV_CODEC_ID_AMR_NB, AV_CODEC_ID_AMR_WB,
3243       AV_CODEC_ID_AAC,
3244       AV_CODEC_ID_NONE
3245     };
3246
3247     *video_codec_list = tgp_video_list;
3248     *audio_codec_list = tgp_audio_list;
3249   } else if (!strcmp (format_name, "mmf")) {
3250     static enum CodecID mmf_audio_list[] = {
3251       AV_CODEC_ID_ADPCM_YAMAHA, AV_CODEC_ID_NONE
3252     };
3253     *video_codec_list = NULL;
3254     *audio_codec_list = mmf_audio_list;
3255   } else if (!strcmp (format_name, "amr")) {
3256     static enum CodecID amr_audio_list[] = {
3257       AV_CODEC_ID_AMR_NB, AV_CODEC_ID_AMR_WB,
3258       AV_CODEC_ID_NONE
3259     };
3260     *video_codec_list = NULL;
3261     *audio_codec_list = amr_audio_list;
3262   } else if (!strcmp (format_name, "gif")) {
3263     static enum CodecID gif_image_list[] = {
3264       AV_CODEC_ID_RAWVIDEO, AV_CODEC_ID_NONE
3265     };
3266     *video_codec_list = gif_image_list;
3267     *audio_codec_list = NULL;
3268   } else if ((plugin->audio_codec != AV_CODEC_ID_NONE) ||
3269       (plugin->video_codec != AV_CODEC_ID_NONE)) {
3270     tmp_vlist[0] = plugin->video_codec;
3271     tmp_alist[0] = plugin->audio_codec;
3272
3273     *video_codec_list = tmp_vlist;
3274     *audio_codec_list = tmp_alist;
3275   } else if ((!strcmp (format_name, "pva"))) {
3276     static enum CodecID tgp_video_list[] = {
3277       AV_CODEC_ID_MPEG2VIDEO,
3278       AV_CODEC_ID_NONE
3279     };
3280     static enum CodecID tgp_audio_list[] = {
3281       AV_CODEC_ID_MP2,
3282       AV_CODEC_ID_NONE
3283     };
3284
3285     *video_codec_list = tgp_video_list;
3286     *audio_codec_list = tgp_audio_list;
3287   } else {
3288     GST_LOG ("Format %s not found", format_name);
3289     return FALSE;
3290   }
3291
3292   return TRUE;
3293 }
3294
3295 /* Convert a GstCaps to a FFMPEG codec ID. Size et all
3296  * are omitted, that can be queried by the user itself,
3297  * we're not eating the GstCaps or anything
3298  * A pointer to an allocated context is also needed for
3299  * optional extra info
3300  */
3301
3302 enum CodecID
3303 gst_ffmpeg_caps_to_codecid (const GstCaps * caps, AVCodecContext * context)
3304 {
3305   enum CodecID id = AV_CODEC_ID_NONE;
3306   const gchar *mimetype;
3307   const GstStructure *structure;
3308   gboolean video = FALSE, audio = FALSE;        /* we want to be sure! */
3309
3310   g_return_val_if_fail (caps != NULL, AV_CODEC_ID_NONE);
3311   g_return_val_if_fail (gst_caps_get_size (caps) == 1, AV_CODEC_ID_NONE);
3312   structure = gst_caps_get_structure (caps, 0);
3313
3314   mimetype = gst_structure_get_name (structure);
3315
3316   if (!strcmp (mimetype, "video/x-raw")) {
3317     id = AV_CODEC_ID_RAWVIDEO;
3318     video = TRUE;
3319   } else if (!strcmp (mimetype, "audio/x-raw")) {
3320     GstAudioInfo info;
3321
3322     if (gst_audio_info_from_caps (&info, caps)) {
3323       switch (GST_AUDIO_INFO_FORMAT (&info)) {
3324         case GST_AUDIO_FORMAT_S8:
3325           id = AV_CODEC_ID_PCM_S8;
3326           break;
3327         case GST_AUDIO_FORMAT_U8:
3328           id = AV_CODEC_ID_PCM_U8;
3329           break;
3330         case GST_AUDIO_FORMAT_S16LE:
3331           id = AV_CODEC_ID_PCM_S16LE;
3332           break;
3333         case GST_AUDIO_FORMAT_S16BE:
3334           id = AV_CODEC_ID_PCM_S16BE;
3335           break;
3336         case GST_AUDIO_FORMAT_U16LE:
3337           id = AV_CODEC_ID_PCM_U16LE;
3338           break;
3339         case GST_AUDIO_FORMAT_U16BE:
3340           id = AV_CODEC_ID_PCM_U16BE;
3341           break;
3342         default:
3343           break;
3344       }
3345       if (id != AV_CODEC_ID_NONE)
3346         audio = TRUE;
3347     }
3348   } else if (!strcmp (mimetype, "audio/x-mulaw")) {
3349     id = AV_CODEC_ID_PCM_MULAW;
3350     audio = TRUE;
3351   } else if (!strcmp (mimetype, "audio/x-alaw")) {
3352     id = AV_CODEC_ID_PCM_ALAW;
3353     audio = TRUE;
3354   } else if (!strcmp (mimetype, "video/x-dv")) {
3355     gboolean sys_strm;
3356
3357     if (gst_structure_get_boolean (structure, "systemstream", &sys_strm) &&
3358         !sys_strm) {
3359       id = AV_CODEC_ID_DVVIDEO;
3360       video = TRUE;
3361     }
3362   } else if (!strcmp (mimetype, "audio/x-dv")) {        /* ??? */
3363     id = AV_CODEC_ID_DVAUDIO;
3364     audio = TRUE;
3365   } else if (!strcmp (mimetype, "video/x-h263")) {
3366     const gchar *h263version =
3367         gst_structure_get_string (structure, "h263version");
3368     if (h263version && !strcmp (h263version, "h263p"))
3369       id = AV_CODEC_ID_H263P;
3370     else
3371       id = AV_CODEC_ID_H263;
3372     video = TRUE;
3373   } else if (!strcmp (mimetype, "video/x-intel-h263")) {
3374     id = AV_CODEC_ID_H263I;
3375     video = TRUE;
3376   } else if (!strcmp (mimetype, "video/x-h261")) {
3377     id = AV_CODEC_ID_H261;
3378     video = TRUE;
3379   } else if (!strcmp (mimetype, "video/mpeg")) {
3380     gboolean sys_strm;
3381     gint mpegversion;
3382
3383     if (gst_structure_get_boolean (structure, "systemstream", &sys_strm) &&
3384         gst_structure_get_int (structure, "mpegversion", &mpegversion) &&
3385         !sys_strm) {
3386       switch (mpegversion) {
3387         case 1:
3388           id = AV_CODEC_ID_MPEG1VIDEO;
3389           break;
3390         case 2:
3391           id = AV_CODEC_ID_MPEG2VIDEO;
3392           break;
3393         case 4:
3394           id = AV_CODEC_ID_MPEG4;
3395           break;
3396       }
3397     }
3398     if (id != AV_CODEC_ID_NONE)
3399       video = TRUE;
3400   } else if (!strcmp (mimetype, "image/jpeg")) {
3401     id = AV_CODEC_ID_MJPEG;     /* A... B... */
3402     video = TRUE;
3403   } else if (!strcmp (mimetype, "video/x-jpeg-b")) {
3404     id = AV_CODEC_ID_MJPEGB;
3405     video = TRUE;
3406   } else if (!strcmp (mimetype, "video/x-wmv")) {
3407     gint wmvversion = 0;
3408
3409     if (gst_structure_get_int (structure, "wmvversion", &wmvversion)) {
3410       switch (wmvversion) {
3411         case 1:
3412           id = AV_CODEC_ID_WMV1;
3413           break;
3414         case 2:
3415           id = AV_CODEC_ID_WMV2;
3416           break;
3417         case 3:
3418         {
3419           const gchar *format;
3420
3421           /* WMV3 unless the fourcc exists and says otherwise */
3422           id = AV_CODEC_ID_WMV3;
3423
3424           if ((format = gst_structure_get_string (structure, "format")) &&
3425               (g_str_equal (format, "WVC1") || g_str_equal (format, "WMVA")))
3426             id = AV_CODEC_ID_VC1;
3427
3428           break;
3429         }
3430       }
3431     }
3432     if (id != AV_CODEC_ID_NONE)
3433       video = TRUE;
3434   } else if (!strcmp (mimetype, "audio/x-vorbis")) {
3435     id = AV_CODEC_ID_VORBIS;
3436     audio = TRUE;
3437   } else if (!strcmp (mimetype, "audio/x-qdm2")) {
3438     id = AV_CODEC_ID_QDM2;
3439     audio = TRUE;
3440   } else if (!strcmp (mimetype, "audio/mpeg")) {
3441     gint layer = 0;
3442     gint mpegversion = 0;
3443
3444     if (gst_structure_get_int (structure, "mpegversion", &mpegversion)) {
3445       switch (mpegversion) {
3446         case 2:                /* ffmpeg uses faad for both... */
3447         case 4:
3448           id = AV_CODEC_ID_AAC;
3449           break;
3450         case 1:
3451           if (gst_structure_get_int (structure, "layer", &layer)) {
3452             switch (layer) {
3453               case 1:
3454                 id = AV_CODEC_ID_MP1;
3455                 break;
3456               case 2:
3457                 id = AV_CODEC_ID_MP2;
3458                 break;
3459               case 3:
3460                 id = AV_CODEC_ID_MP3;
3461                 break;
3462             }
3463           }
3464       }
3465     }
3466     if (id != AV_CODEC_ID_NONE)
3467       audio = TRUE;
3468   } else if (!strcmp (mimetype, "audio/x-musepack")) {
3469     gint streamversion = -1;
3470
3471     if (gst_structure_get_int (structure, "streamversion", &streamversion)) {
3472       if (streamversion == 7)
3473         id = AV_CODEC_ID_MUSEPACK7;
3474     } else {
3475       id = AV_CODEC_ID_MUSEPACK7;
3476     }
3477   } else if (!strcmp (mimetype, "audio/x-wma")) {
3478     gint wmaversion = 0;
3479
3480     if (gst_structure_get_int (structure, "wmaversion", &wmaversion)) {
3481       switch (wmaversion) {
3482         case 1:
3483           id = AV_CODEC_ID_WMAV1;
3484           break;
3485         case 2:
3486           id = AV_CODEC_ID_WMAV2;
3487           break;
3488         case 3:
3489           id = AV_CODEC_ID_WMAPRO;
3490           break;
3491       }
3492     }
3493     if (id != AV_CODEC_ID_NONE)
3494       audio = TRUE;
3495   } else if (!strcmp (mimetype, "audio/x-wms")) {
3496     id = AV_CODEC_ID_WMAVOICE;
3497     audio = TRUE;
3498   } else if (!strcmp (mimetype, "audio/x-ac3")) {
3499     id = AV_CODEC_ID_AC3;
3500     audio = TRUE;
3501   } else if (!strcmp (mimetype, "audio/x-eac3")) {
3502     id = AV_CODEC_ID_EAC3;
3503     audio = TRUE;
3504   } else if (!strcmp (mimetype, "audio/x-vnd.sony.atrac3") ||
3505       !strcmp (mimetype, "audio/atrac3")) {
3506     id = AV_CODEC_ID_ATRAC3;
3507     audio = TRUE;
3508   } else if (!strcmp (mimetype, "audio/x-dts")) {
3509     id = AV_CODEC_ID_DTS;
3510     audio = TRUE;
3511   } else if (!strcmp (mimetype, "application/x-ape")) {
3512     id = AV_CODEC_ID_APE;
3513     audio = TRUE;
3514   } else if (!strcmp (mimetype, "video/x-msmpeg")) {
3515     gint msmpegversion = 0;
3516
3517     if (gst_structure_get_int (structure, "msmpegversion", &msmpegversion)) {
3518       switch (msmpegversion) {
3519         case 41:
3520           id = AV_CODEC_ID_MSMPEG4V1;
3521           break;
3522         case 42:
3523           id = AV_CODEC_ID_MSMPEG4V2;
3524           break;
3525         case 43:
3526           id = AV_CODEC_ID_MSMPEG4V3;
3527           break;
3528       }
3529     }
3530     if (id != AV_CODEC_ID_NONE)
3531       video = TRUE;
3532   } else if (!strcmp (mimetype, "video/x-svq")) {
3533     gint svqversion = 0;
3534
3535     if (gst_structure_get_int (structure, "svqversion", &svqversion)) {
3536       switch (svqversion) {
3537         case 1:
3538           id = AV_CODEC_ID_SVQ1;
3539           break;
3540         case 3:
3541           id = AV_CODEC_ID_SVQ3;
3542           break;
3543       }
3544     }
3545     if (id != AV_CODEC_ID_NONE)
3546       video = TRUE;
3547   } else if (!strcmp (mimetype, "video/x-huffyuv")) {
3548     id = AV_CODEC_ID_HUFFYUV;
3549     video = TRUE;
3550   } else if (!strcmp (mimetype, "audio/x-mace")) {
3551     gint maceversion = 0;
3552
3553     if (gst_structure_get_int (structure, "maceversion", &maceversion)) {
3554       switch (maceversion) {
3555         case 3:
3556           id = AV_CODEC_ID_MACE3;
3557           break;
3558         case 6:
3559           id = AV_CODEC_ID_MACE6;
3560           break;
3561       }
3562     }
3563     if (id != AV_CODEC_ID_NONE)
3564       audio = TRUE;
3565   } else if (!strcmp (mimetype, "video/x-theora")) {
3566     id = AV_CODEC_ID_THEORA;
3567     video = TRUE;
3568   } else if (!strcmp (mimetype, "video/x-vp3")) {
3569     id = AV_CODEC_ID_VP3;
3570     video = TRUE;
3571   } else if (!strcmp (mimetype, "video/x-vp5")) {
3572     id = AV_CODEC_ID_VP5;
3573     video = TRUE;
3574   } else if (!strcmp (mimetype, "video/x-vp6")) {
3575     id = AV_CODEC_ID_VP6;
3576     video = TRUE;
3577   } else if (!strcmp (mimetype, "video/x-vp6-flash")) {
3578     id = AV_CODEC_ID_VP6F;
3579     video = TRUE;
3580   } else if (!strcmp (mimetype, "video/x-vp6-alpha")) {
3581     id = AV_CODEC_ID_VP6A;
3582     video = TRUE;
3583   } else if (!strcmp (mimetype, "video/x-vp8")) {
3584     id = AV_CODEC_ID_VP8;
3585     video = TRUE;
3586   } else if (!strcmp (mimetype, "video/x-flash-screen")) {
3587     id = AV_CODEC_ID_FLASHSV;
3588     video = TRUE;
3589   } else if (!strcmp (mimetype, "video/x-indeo")) {
3590     gint indeoversion = 0;
3591
3592     if (gst_structure_get_int (structure, "indeoversion", &indeoversion)) {
3593       switch (indeoversion) {
3594         case 5:
3595           id = AV_CODEC_ID_INDEO5;
3596           break;
3597         case 4:
3598           id = AV_CODEC_ID_INDEO4;
3599           break;
3600         case 3:
3601           id = AV_CODEC_ID_INDEO3;
3602           break;
3603         case 2:
3604           id = AV_CODEC_ID_INDEO2;
3605           break;
3606       }
3607       if (id != AV_CODEC_ID_NONE)
3608         video = TRUE;
3609     }
3610   } else if (!strcmp (mimetype, "video/x-divx")) {
3611     gint divxversion = 0;
3612
3613     if (gst_structure_get_int (structure, "divxversion", &divxversion)) {
3614       switch (divxversion) {
3615         case 3:
3616           id = AV_CODEC_ID_MSMPEG4V3;
3617           break;
3618         case 4:
3619         case 5:
3620           id = AV_CODEC_ID_MPEG4;
3621           break;
3622       }
3623     }
3624     if (id != AV_CODEC_ID_NONE)
3625       video = TRUE;
3626   } else if (!strcmp (mimetype, "video/x-3ivx")) {
3627     id = AV_CODEC_ID_MPEG4;
3628     video = TRUE;
3629   } else if (!strcmp (mimetype, "video/x-xvid")) {
3630     id = AV_CODEC_ID_MPEG4;
3631     video = TRUE;
3632   } else if (!strcmp (mimetype, "video/x-ffv")) {
3633     gint ffvversion = 0;
3634
3635     if (gst_structure_get_int (structure, "ffvversion", &ffvversion) &&
3636         ffvversion == 1) {
3637       id = AV_CODEC_ID_FFV1;
3638       video = TRUE;
3639     }
3640   } else if (!strcmp (mimetype, "audio/x-adpcm")) {
3641     const gchar *layout;
3642
3643     layout = gst_structure_get_string (structure, "layout");
3644     if (layout == NULL) {
3645       /* break */
3646     } else if (!strcmp (layout, "quicktime")) {
3647       id = AV_CODEC_ID_ADPCM_IMA_QT;
3648     } else if (!strcmp (layout, "microsoft")) {
3649       id = AV_CODEC_ID_ADPCM_MS;
3650     } else if (!strcmp (layout, "dvi")) {
3651       id = AV_CODEC_ID_ADPCM_IMA_WAV;
3652     } else if (!strcmp (layout, "4xm")) {
3653       id = AV_CODEC_ID_ADPCM_4XM;
3654     } else if (!strcmp (layout, "smjpeg")) {
3655       id = AV_CODEC_ID_ADPCM_IMA_SMJPEG;
3656     } else if (!strcmp (layout, "dk3")) {
3657       id = AV_CODEC_ID_ADPCM_IMA_DK3;
3658     } else if (!strcmp (layout, "dk4")) {
3659       id = AV_CODEC_ID_ADPCM_IMA_DK4;
3660     } else if (!strcmp (layout, "westwood")) {
3661       id = AV_CODEC_ID_ADPCM_IMA_WS;
3662     } else if (!strcmp (layout, "iss")) {
3663       id = AV_CODEC_ID_ADPCM_IMA_ISS;
3664     } else if (!strcmp (layout, "xa")) {
3665       id = AV_CODEC_ID_ADPCM_XA;
3666     } else if (!strcmp (layout, "adx")) {
3667       id = AV_CODEC_ID_ADPCM_ADX;
3668     } else if (!strcmp (layout, "ea")) {
3669       id = AV_CODEC_ID_ADPCM_EA;
3670     } else if (!strcmp (layout, "g726")) {
3671       id = AV_CODEC_ID_ADPCM_G726;
3672     } else if (!strcmp (layout, "g721")) {
3673       id = AV_CODEC_ID_ADPCM_G726;
3674     } else if (!strcmp (layout, "ct")) {
3675       id = AV_CODEC_ID_ADPCM_CT;
3676     } else if (!strcmp (layout, "swf")) {
3677       id = AV_CODEC_ID_ADPCM_SWF;
3678     } else if (!strcmp (layout, "yamaha")) {
3679       id = AV_CODEC_ID_ADPCM_YAMAHA;
3680     } else if (!strcmp (layout, "sbpro2")) {
3681       id = AV_CODEC_ID_ADPCM_SBPRO_2;
3682     } else if (!strcmp (layout, "sbpro3")) {
3683       id = AV_CODEC_ID_ADPCM_SBPRO_3;
3684     } else if (!strcmp (layout, "sbpro4")) {
3685       id = AV_CODEC_ID_ADPCM_SBPRO_4;
3686     }
3687     if (id != AV_CODEC_ID_NONE)
3688       audio = TRUE;
3689   } else if (!strcmp (mimetype, "video/x-4xm")) {
3690     id = AV_CODEC_ID_4XM;
3691     video = TRUE;
3692   } else if (!strcmp (mimetype, "audio/x-dpcm")) {
3693     const gchar *layout;
3694
3695     layout = gst_structure_get_string (structure, "layout");
3696     if (!layout) {
3697       /* .. */
3698     } else if (!strcmp (layout, "roq")) {
3699       id = AV_CODEC_ID_ROQ_DPCM;
3700     } else if (!strcmp (layout, "interplay")) {
3701       id = AV_CODEC_ID_INTERPLAY_DPCM;
3702     } else if (!strcmp (layout, "xan")) {
3703       id = AV_CODEC_ID_XAN_DPCM;
3704     } else if (!strcmp (layout, "sol")) {
3705       id = AV_CODEC_ID_SOL_DPCM;
3706     }
3707     if (id != AV_CODEC_ID_NONE)
3708       audio = TRUE;
3709   } else if (!strcmp (mimetype, "audio/x-flac")) {
3710     id = AV_CODEC_ID_FLAC;
3711     audio = TRUE;
3712   } else if (!strcmp (mimetype, "audio/x-shorten")) {
3713     id = AV_CODEC_ID_SHORTEN;
3714     audio = TRUE;
3715   } else if (!strcmp (mimetype, "audio/x-alac")) {
3716     id = AV_CODEC_ID_ALAC;
3717     audio = TRUE;
3718   } else if (!strcmp (mimetype, "video/x-cinepak")) {
3719     id = AV_CODEC_ID_CINEPAK;
3720     video = TRUE;
3721   } else if (!strcmp (mimetype, "video/x-pn-realvideo")) {
3722     gint rmversion;
3723
3724     if (gst_structure_get_int (structure, "rmversion", &rmversion)) {
3725       switch (rmversion) {
3726         case 1:
3727           id = AV_CODEC_ID_RV10;
3728           break;
3729         case 2:
3730           id = AV_CODEC_ID_RV20;
3731           break;
3732         case 3:
3733           id = AV_CODEC_ID_RV30;
3734           break;
3735         case 4:
3736           id = AV_CODEC_ID_RV40;
3737           break;
3738       }
3739     }
3740     if (id != AV_CODEC_ID_NONE)
3741       video = TRUE;
3742   } else if (!strcmp (mimetype, "audio/x-sipro")) {
3743     id = AV_CODEC_ID_SIPR;
3744     audio = TRUE;
3745   } else if (!strcmp (mimetype, "audio/x-pn-realaudio")) {
3746     gint raversion;
3747
3748     if (gst_structure_get_int (structure, "raversion", &raversion)) {
3749       switch (raversion) {
3750         case 1:
3751           id = AV_CODEC_ID_RA_144;
3752           break;
3753         case 2:
3754           id = AV_CODEC_ID_RA_288;
3755           break;
3756         case 8:
3757           id = AV_CODEC_ID_COOK;
3758           break;
3759       }
3760     }
3761     if (id != AV_CODEC_ID_NONE)
3762       audio = TRUE;
3763   } else if (!strcmp (mimetype, "video/x-rle")) {
3764     const gchar *layout;
3765
3766     if ((layout = gst_structure_get_string (structure, "layout"))) {
3767       if (!strcmp (layout, "microsoft")) {
3768         id = AV_CODEC_ID_MSRLE;
3769         video = TRUE;
3770       }
3771     }
3772   } else if (!strcmp (mimetype, "video/x-xan")) {
3773     gint wcversion = 0;
3774
3775     if ((gst_structure_get_int (structure, "wcversion", &wcversion))) {
3776       switch (wcversion) {
3777         case 3:
3778           id = AV_CODEC_ID_XAN_WC3;
3779           video = TRUE;
3780           break;
3781         case 4:
3782           id = AV_CODEC_ID_XAN_WC4;
3783           video = TRUE;
3784           break;
3785         default:
3786           break;
3787       }
3788     }
3789   } else if (!strcmp (mimetype, "audio/AMR")) {
3790     audio = TRUE;
3791     id = AV_CODEC_ID_AMR_NB;
3792   } else if (!strcmp (mimetype, "audio/AMR-WB")) {
3793     id = AV_CODEC_ID_AMR_WB;
3794     audio = TRUE;
3795   } else if (!strcmp (mimetype, "audio/qcelp")) {
3796     id = AV_CODEC_ID_QCELP;
3797     audio = TRUE;
3798   } else if (!strcmp (mimetype, "video/x-h264")) {
3799     id = AV_CODEC_ID_H264;
3800     video = TRUE;
3801   } else if (!strcmp (mimetype, "video/x-flash-video")) {
3802     gint flvversion = 0;
3803
3804     if ((gst_structure_get_int (structure, "flvversion", &flvversion))) {
3805       switch (flvversion) {
3806         case 1:
3807           id = AV_CODEC_ID_FLV1;
3808           video = TRUE;
3809           break;
3810         default:
3811           break;
3812       }
3813     }
3814
3815   } else if (!strcmp (mimetype, "audio/x-nellymoser")) {
3816     id = AV_CODEC_ID_NELLYMOSER;
3817     audio = TRUE;
3818   } else if (!strncmp (mimetype, "audio/x-gst-av-", 15)) {
3819     gchar ext[16];
3820     AVCodec *codec;
3821
3822     if (strlen (mimetype) <= 30 &&
3823         sscanf (mimetype, "audio/x-gst-av-%s", ext) == 1) {
3824       if ((codec = avcodec_find_decoder_by_name (ext)) ||
3825           (codec = avcodec_find_encoder_by_name (ext))) {
3826         id = codec->id;
3827         audio = TRUE;
3828       }
3829     }
3830   } else if (!strncmp (mimetype, "video/x-gst-av-", 15)) {
3831     gchar ext[16];
3832     AVCodec *codec;
3833
3834     if (strlen (mimetype) <= 30 &&
3835         sscanf (mimetype, "video/x-gst-av-%s", ext) == 1) {
3836       if ((codec = avcodec_find_decoder_by_name (ext)) ||
3837           (codec = avcodec_find_encoder_by_name (ext))) {
3838         id = codec->id;
3839         video = TRUE;
3840       }
3841     }
3842   }
3843
3844   if (context != NULL) {
3845     if (video == TRUE) {
3846       context->codec_type = AVMEDIA_TYPE_VIDEO;
3847     } else if (audio == TRUE) {
3848       context->codec_type = AVMEDIA_TYPE_AUDIO;
3849     } else {
3850       context->codec_type = AVMEDIA_TYPE_UNKNOWN;
3851     }
3852     context->codec_id = id;
3853     gst_ffmpeg_caps_with_codecid (id, context->codec_type, caps, context);
3854   }
3855
3856   if (id != AV_CODEC_ID_NONE) {
3857     GST_DEBUG ("The id=%d belongs to the caps %" GST_PTR_FORMAT, id, caps);
3858   } else {
3859     GST_WARNING ("Couldn't figure out the id for caps %" GST_PTR_FORMAT, caps);
3860   }
3861
3862   return id;
3863 }