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