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