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