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