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