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