diff options
author | Edward Hervey <bilboed@bilboed.com> | 2009-04-16 11:00:46 +0200 |
---|---|---|
committer | Edward Hervey <bilboed@bilboed.com> | 2009-04-16 11:00:46 +0200 |
commit | cbea9841ecaa4ec82286ab307e3b8aff4b3fb162 (patch) | |
tree | 87e20a9a41aa4b91b388053e7dafe8f69ebf91dc | |
parent | cc7ce2639cd85450dc4c81256ef585afaa3cc678 (diff) |
gstffmpegutils: Add the files :)
-rw-r--r-- | ext/ffmpeg/gstffmpegutils.c | 398 | ||||
-rw-r--r-- | ext/ffmpeg/gstffmpegutils.h | 90 |
2 files changed, 488 insertions, 0 deletions
diff --git a/ext/ffmpeg/gstffmpegutils.c b/ext/ffmpeg/gstffmpegutils.c new file mode 100644 index 0000000..988cc1c --- /dev/null +++ b/ext/ffmpeg/gstffmpegutils.c @@ -0,0 +1,398 @@ +/* GStreamer + * Copyright (c) 2009 Edward Hervey <bilboed@bilboed.com> + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Library General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Library General Public License for more details. + * + * You should have received a copy of the GNU Library General Public + * License along with this library; if not, write to the + * Free Software Foundation, Inc., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ + +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif +#include "gstffmpegutils.h" + +G_CONST_RETURN gchar * +gst_ffmpeg_get_codecid_longname (enum CodecID codec_id) +{ + AVCodec *codec; + /* Let's use what ffmpeg can provide us */ + + if ((codec = avcodec_find_decoder (codec_id)) || + (codec = avcodec_find_encoder (codec_id))) + return codec->long_name; + return NULL; +} + +gint +av_smp_format_depth (enum SampleFormat smp_fmt) +{ + gint depth = -1; + switch (smp_fmt) { + case SAMPLE_FMT_U8: + depth = 1; + break; + case SAMPLE_FMT_S16: + depth = 2; + break; + case SAMPLE_FMT_S32: + case SAMPLE_FMT_FLT: + depth = 4; + break; + case SAMPLE_FMT_DBL: + depth = 8; + break; + default: + GST_ERROR ("UNHANDLED SAMPLE FORMAT !"); + break; + } + return depth; +} + + +/* + * Fill in pointers to memory in a AVPicture, where + * everything is aligned by 4 (as required by X). + * This is mostly a copy from imgconvert.c with some + * small changes. + */ + +#define FF_COLOR_RGB 0 /* RGB color space */ +#define FF_COLOR_GRAY 1 /* gray color space */ +#define FF_COLOR_YUV 2 /* YUV color space. 16 <= Y <= 235, 16 <= U, V <= 240 */ +#define FF_COLOR_YUV_JPEG 3 /* YUV color space. 0 <= Y <= 255, 0 <= U, V <= 255 */ + +#define FF_PIXEL_PLANAR 0 /* each channel has one component in AVPicture */ +#define FF_PIXEL_PACKED 1 /* only one components containing all the channels */ +#define FF_PIXEL_PALETTE 2 /* one components containing indexes for a palette */ + +typedef struct PixFmtInfo +{ + const char *name; + uint8_t nb_channels; /* number of channels (including alpha) */ + uint8_t color_type; /* color type (see FF_COLOR_xxx constants) */ + uint8_t pixel_type; /* pixel storage type (see FF_PIXEL_xxx constants) */ + uint8_t is_alpha:1; /* true if alpha can be specified */ + uint8_t x_chroma_shift; /* X chroma subsampling factor is 2 ^ shift */ + uint8_t y_chroma_shift; /* Y chroma subsampling factor is 2 ^ shift */ + uint8_t depth; /* bit depth of the color components */ +} PixFmtInfo; + + +/* this table gives more information about formats */ +static PixFmtInfo pix_fmt_info[PIX_FMT_NB]; +void +gst_ffmpeg_init_pix_fmt_info () +{ + /* YUV formats */ + pix_fmt_info[PIX_FMT_YUV420P].name = g_strdup ("yuv420p"); + pix_fmt_info[PIX_FMT_YUV420P].nb_channels = 3; + pix_fmt_info[PIX_FMT_YUV420P].color_type = FF_COLOR_YUV; + pix_fmt_info[PIX_FMT_YUV420P].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_YUV420P].depth = 8, + pix_fmt_info[PIX_FMT_YUV420P].x_chroma_shift = 1, + pix_fmt_info[PIX_FMT_YUV420P].y_chroma_shift = 1; + + pix_fmt_info[PIX_FMT_YUV422P].name = g_strdup ("yuv422p"); + pix_fmt_info[PIX_FMT_YUV422P].nb_channels = 3; + pix_fmt_info[PIX_FMT_YUV422P].color_type = FF_COLOR_YUV; + pix_fmt_info[PIX_FMT_YUV422P].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_YUV422P].depth = 8; + pix_fmt_info[PIX_FMT_YUV422P].x_chroma_shift = 1; + pix_fmt_info[PIX_FMT_YUV422P].y_chroma_shift = 0; + + pix_fmt_info[PIX_FMT_YUV444P].name = g_strdup ("yuv444p"); + pix_fmt_info[PIX_FMT_YUV444P].nb_channels = 3; + pix_fmt_info[PIX_FMT_YUV444P].color_type = FF_COLOR_YUV; + pix_fmt_info[PIX_FMT_YUV444P].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_YUV444P].depth = 8; + pix_fmt_info[PIX_FMT_YUV444P].x_chroma_shift = 0; + pix_fmt_info[PIX_FMT_YUV444P].y_chroma_shift = 0; + + pix_fmt_info[PIX_FMT_YUYV422].name = g_strdup ("yuv422"); + pix_fmt_info[PIX_FMT_YUYV422].nb_channels = 1; + pix_fmt_info[PIX_FMT_YUYV422].color_type = FF_COLOR_YUV; + pix_fmt_info[PIX_FMT_YUYV422].pixel_type = FF_PIXEL_PACKED; + pix_fmt_info[PIX_FMT_YUYV422].depth = 8; + pix_fmt_info[PIX_FMT_YUYV422].x_chroma_shift = 1; + pix_fmt_info[PIX_FMT_YUYV422].y_chroma_shift = 0; + + pix_fmt_info[PIX_FMT_YUV410P].name = g_strdup ("yuv410p"); + pix_fmt_info[PIX_FMT_YUV410P].nb_channels = 3; + pix_fmt_info[PIX_FMT_YUV410P].color_type = FF_COLOR_YUV; + pix_fmt_info[PIX_FMT_YUV410P].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_YUV410P].depth = 8; + pix_fmt_info[PIX_FMT_YUV410P].x_chroma_shift = 2; + pix_fmt_info[PIX_FMT_YUV410P].y_chroma_shift = 2; + + pix_fmt_info[PIX_FMT_YUV411P].name = g_strdup ("yuv411p"); + pix_fmt_info[PIX_FMT_YUV411P].nb_channels = 3; + pix_fmt_info[PIX_FMT_YUV411P].color_type = FF_COLOR_YUV; + pix_fmt_info[PIX_FMT_YUV411P].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_YUV411P].depth = 8; + pix_fmt_info[PIX_FMT_YUV411P].x_chroma_shift = 2; + pix_fmt_info[PIX_FMT_YUV411P].y_chroma_shift = 0; + + /* JPEG YUV */ + pix_fmt_info[PIX_FMT_YUVJ420P].name = g_strdup ("yuvj420p"); + pix_fmt_info[PIX_FMT_YUVJ420P].nb_channels = 3; + pix_fmt_info[PIX_FMT_YUVJ420P].color_type = FF_COLOR_YUV_JPEG; + pix_fmt_info[PIX_FMT_YUVJ420P].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_YUVJ420P].depth = 8; + pix_fmt_info[PIX_FMT_YUVJ420P].x_chroma_shift = 1; + pix_fmt_info[PIX_FMT_YUVJ420P].y_chroma_shift = 1; + + pix_fmt_info[PIX_FMT_YUVJ422P].name = g_strdup ("yuvj422p"); + pix_fmt_info[PIX_FMT_YUVJ422P].nb_channels = 3; + pix_fmt_info[PIX_FMT_YUVJ422P].color_type = FF_COLOR_YUV_JPEG; + pix_fmt_info[PIX_FMT_YUVJ422P].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_YUVJ422P].depth = 8; + pix_fmt_info[PIX_FMT_YUVJ422P].x_chroma_shift = 1; + pix_fmt_info[PIX_FMT_YUVJ422P].y_chroma_shift = 0; + + pix_fmt_info[PIX_FMT_YUVJ444P].name = g_strdup ("yuvj444p"); + pix_fmt_info[PIX_FMT_YUVJ444P].nb_channels = 3; + pix_fmt_info[PIX_FMT_YUVJ444P].color_type = FF_COLOR_YUV_JPEG; + pix_fmt_info[PIX_FMT_YUVJ444P].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_YUVJ444P].depth = 8; + pix_fmt_info[PIX_FMT_YUVJ444P].x_chroma_shift = 0; + pix_fmt_info[PIX_FMT_YUVJ444P].y_chroma_shift = 0; + + /* RGB formats */ + pix_fmt_info[PIX_FMT_RGB24].name = g_strdup ("rgb24"); + pix_fmt_info[PIX_FMT_RGB24].nb_channels = 3; + pix_fmt_info[PIX_FMT_RGB24].color_type = FF_COLOR_RGB; + pix_fmt_info[PIX_FMT_RGB24].pixel_type = FF_PIXEL_PACKED; + pix_fmt_info[PIX_FMT_RGB24].depth = 8; + pix_fmt_info[PIX_FMT_RGB24].x_chroma_shift = 0; + pix_fmt_info[PIX_FMT_RGB24].y_chroma_shift = 0; + + pix_fmt_info[PIX_FMT_BGR24].name = g_strdup ("bgr24"); + pix_fmt_info[PIX_FMT_BGR24].nb_channels = 3; + pix_fmt_info[PIX_FMT_BGR24].color_type = FF_COLOR_RGB; + pix_fmt_info[PIX_FMT_BGR24].pixel_type = FF_PIXEL_PACKED; + pix_fmt_info[PIX_FMT_BGR24].depth = 8; + pix_fmt_info[PIX_FMT_BGR24].x_chroma_shift = 0; + pix_fmt_info[PIX_FMT_BGR24].y_chroma_shift = 0; + + pix_fmt_info[PIX_FMT_RGB32].name = g_strdup ("rgba32"); + pix_fmt_info[PIX_FMT_RGB32].nb_channels = 4; + pix_fmt_info[PIX_FMT_RGB32].is_alpha = 1; + pix_fmt_info[PIX_FMT_RGB32].color_type = FF_COLOR_RGB; + pix_fmt_info[PIX_FMT_RGB32].pixel_type = FF_PIXEL_PACKED; + pix_fmt_info[PIX_FMT_RGB32].depth = 8; + pix_fmt_info[PIX_FMT_RGB32].x_chroma_shift = 0; + pix_fmt_info[PIX_FMT_RGB32].y_chroma_shift = 0; + + pix_fmt_info[PIX_FMT_RGB565].name = g_strdup ("rgb565"); + pix_fmt_info[PIX_FMT_RGB565].nb_channels = 3; + pix_fmt_info[PIX_FMT_RGB565].color_type = FF_COLOR_RGB; + pix_fmt_info[PIX_FMT_RGB565].pixel_type = FF_PIXEL_PACKED; + pix_fmt_info[PIX_FMT_RGB565].depth = 5; + pix_fmt_info[PIX_FMT_RGB565].x_chroma_shift = 0; + pix_fmt_info[PIX_FMT_RGB565].y_chroma_shift = 0; + + pix_fmt_info[PIX_FMT_RGB555].name = g_strdup ("rgb555"); + pix_fmt_info[PIX_FMT_RGB555].nb_channels = 4; + pix_fmt_info[PIX_FMT_RGB555].is_alpha = 1; + pix_fmt_info[PIX_FMT_RGB555].color_type = FF_COLOR_RGB; + pix_fmt_info[PIX_FMT_RGB555].pixel_type = FF_PIXEL_PACKED; + pix_fmt_info[PIX_FMT_RGB555].depth = 5; + pix_fmt_info[PIX_FMT_RGB555].x_chroma_shift = 0; + pix_fmt_info[PIX_FMT_RGB555].y_chroma_shift = 0; + + /* gray / mono formats */ + pix_fmt_info[PIX_FMT_GRAY8].name = g_strdup ("gray"); + pix_fmt_info[PIX_FMT_GRAY8].nb_channels = 1; + pix_fmt_info[PIX_FMT_GRAY8].color_type = FF_COLOR_GRAY; + pix_fmt_info[PIX_FMT_GRAY8].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_GRAY8].depth = 8; + + pix_fmt_info[PIX_FMT_MONOWHITE].name = g_strdup ("monow"); + pix_fmt_info[PIX_FMT_MONOWHITE].nb_channels = 1; + pix_fmt_info[PIX_FMT_MONOWHITE].color_type = FF_COLOR_GRAY; + pix_fmt_info[PIX_FMT_MONOWHITE].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_MONOWHITE].depth = 1; + + pix_fmt_info[PIX_FMT_MONOBLACK].name = g_strdup ("monob"); + pix_fmt_info[PIX_FMT_MONOBLACK].nb_channels = 1; + pix_fmt_info[PIX_FMT_MONOBLACK].color_type = FF_COLOR_GRAY; + pix_fmt_info[PIX_FMT_MONOBLACK].pixel_type = FF_PIXEL_PLANAR; + pix_fmt_info[PIX_FMT_MONOBLACK].depth = 1; + + /* paletted formats */ + pix_fmt_info[PIX_FMT_PAL8].name = g_strdup ("pal8"); + pix_fmt_info[PIX_FMT_PAL8].nb_channels = 4; + pix_fmt_info[PIX_FMT_PAL8].is_alpha = 1; + pix_fmt_info[PIX_FMT_PAL8].color_type = FF_COLOR_RGB; + pix_fmt_info[PIX_FMT_PAL8].pixel_type = FF_PIXEL_PALETTE; + pix_fmt_info[PIX_FMT_PAL8].depth = 8; +}; + +int +gst_ffmpeg_avpicture_get_size (int pix_fmt, int width, int height) +{ + AVPicture dummy_pict; + + return gst_ffmpeg_avpicture_fill (&dummy_pict, NULL, pix_fmt, width, height); +} + +#define GEN_MASK(x) ((1<<(x))-1) +#define ROUND_UP_X(v,x) (((v) + GEN_MASK(x)) & ~GEN_MASK(x)) +#define ROUND_UP_2(x) ROUND_UP_X (x, 1) +#define ROUND_UP_4(x) ROUND_UP_X (x, 2) +#define ROUND_UP_8(x) ROUND_UP_X (x, 3) +#define DIV_ROUND_UP_X(v,x) (((v) + GEN_MASK(x)) >> (x)) + +int +gst_ffmpeg_avpicture_fill (AVPicture * picture, + uint8_t * ptr, enum PixelFormat pix_fmt, int width, int height) +{ + int size, w2, h2, size2; + int stride, stride2; + PixFmtInfo *pinfo; + + pinfo = &pix_fmt_info[pix_fmt]; + + switch (pix_fmt) { + case PIX_FMT_YUV420P: + case PIX_FMT_YUV422P: + case PIX_FMT_YUV444P: + case PIX_FMT_YUV410P: + case PIX_FMT_YUV411P: + case PIX_FMT_YUVJ420P: + case PIX_FMT_YUVJ422P: + case PIX_FMT_YUVJ444P: + stride = ROUND_UP_4 (width); + h2 = ROUND_UP_X (height, pinfo->y_chroma_shift); + size = stride * h2; + w2 = DIV_ROUND_UP_X (width, pinfo->x_chroma_shift); + stride2 = ROUND_UP_4 (w2); + h2 = DIV_ROUND_UP_X (height, pinfo->y_chroma_shift); + size2 = stride2 * h2; + picture->data[0] = ptr; + picture->data[1] = picture->data[0] + size; + picture->data[2] = picture->data[1] + size2; + picture->data[3] = NULL; + picture->linesize[0] = stride; + picture->linesize[1] = stride2; + picture->linesize[2] = stride2; + picture->linesize[3] = 0; + GST_DEBUG ("planes %d %d %d", 0, size, size + size2); + GST_DEBUG ("strides %d %d %d", stride, stride2, stride2); + return size + 2 * size2; + case PIX_FMT_RGB24: + case PIX_FMT_BGR24: + stride = ROUND_UP_4 (width * 3); + size = stride * height; + picture->data[0] = ptr; + picture->data[1] = NULL; + picture->data[2] = NULL; + picture->data[3] = NULL; + picture->linesize[0] = stride; + picture->linesize[1] = 0; + picture->linesize[2] = 0; + picture->linesize[3] = 0; + return size; + /*case PIX_FMT_AYUV4444: + case PIX_FMT_BGR32: + case PIX_FMT_BGRA32: + case PIX_FMT_RGB32: */ + case PIX_FMT_RGB32: + stride = width * 4; + size = stride * height; + picture->data[0] = ptr; + picture->data[1] = NULL; + picture->data[2] = NULL; + picture->data[3] = NULL; + picture->linesize[0] = stride; + picture->linesize[1] = 0; + picture->linesize[2] = 0; + picture->linesize[3] = 0; + return size; + case PIX_FMT_RGB555: + case PIX_FMT_RGB565: + case PIX_FMT_YUYV422: + case PIX_FMT_UYVY422: + stride = ROUND_UP_4 (width * 2); + size = stride * height; + picture->data[0] = ptr; + picture->data[1] = NULL; + picture->data[2] = NULL; + picture->data[3] = NULL; + picture->linesize[0] = stride; + picture->linesize[1] = 0; + picture->linesize[2] = 0; + picture->linesize[3] = 0; + return size; + case PIX_FMT_UYYVYY411: + /* FIXME, probably not the right stride */ + stride = ROUND_UP_4 (width); + size = stride * height; + picture->data[0] = ptr; + picture->data[1] = NULL; + picture->data[2] = NULL; + picture->data[3] = NULL; + picture->linesize[0] = width + width / 2; + picture->linesize[1] = 0; + picture->linesize[2] = 0; + picture->linesize[3] = 0; + return size + size / 2; + case PIX_FMT_GRAY8: + stride = ROUND_UP_4 (width); + size = stride * height; + picture->data[0] = ptr; + picture->data[1] = NULL; + picture->data[2] = NULL; + picture->data[3] = NULL; + picture->linesize[0] = stride; + picture->linesize[1] = 0; + picture->linesize[2] = 0; + picture->linesize[3] = 0; + return size; + case PIX_FMT_MONOWHITE: + case PIX_FMT_MONOBLACK: + stride = ROUND_UP_4 ((width + 7) >> 3); + size = stride * height; + picture->data[0] = ptr; + picture->data[1] = NULL; + picture->data[2] = NULL; + picture->data[3] = NULL; + picture->linesize[0] = stride; + picture->linesize[1] = 0; + picture->linesize[2] = 0; + picture->linesize[3] = 0; + return size; + case PIX_FMT_PAL8: + /* already forced to be with stride, so same result as other function */ + stride = ROUND_UP_4 (width); + size = stride * height; + picture->data[0] = ptr; + picture->data[1] = ptr + size; /* palette is stored here as 256 32 bit words */ + picture->data[2] = NULL; + picture->data[3] = NULL; + picture->linesize[0] = stride; + picture->linesize[1] = 4; + picture->linesize[2] = 0; + picture->linesize[3] = 0; + return size + 256 * 4; + default: + picture->data[0] = NULL; + picture->data[1] = NULL; + picture->data[2] = NULL; + picture->data[3] = NULL; + return -1; + } + + return 0; +} diff --git a/ext/ffmpeg/gstffmpegutils.h b/ext/ffmpeg/gstffmpegutils.h new file mode 100644 index 0000000..96efd24 --- /dev/null +++ b/ext/ffmpeg/gstffmpegutils.h @@ -0,0 +1,90 @@ +/* GStreamer + * Copyright (C) <2009> Edward Hervey <bilboed@bilboed.com> + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Library General Public + * License as published by the Free Software Foundation; either + * version 2 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Library General Public License for more details. + * + * You should have received a copy of the GNU Library General Public + * License along with this library; if not, write to the + * Free Software Foundation, Inc., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + */ + +#ifndef __GST_FFMPEG_UTILS_H__ +#define __GST_FFMPEG_UTILS_H__ + +#ifdef HAVE_FFMPEG_UNINSTALLED +#include <avcodec.h> +#else +#include <libavcodec/avcodec.h> +#endif +#include <gst/gst.h> + +/* + *Get the size of an picture + */ +int +gst_ffmpeg_avpicture_get_size (int pix_fmt, int width, int height); + +/* + * Fill in pointers in an AVPicture, aligned by 4 (required by X). + */ + +int +gst_ffmpeg_avpicture_fill (AVPicture * picture, + uint8_t * ptr, + enum PixelFormat pix_fmt, + int width, + int height); + +/* + * Convert from/to a GStreamer <-> FFMpeg timestamp. + */ +static inline guint64 +gst_ffmpeg_time_ff_to_gst (gint64 pts, AVRational base) +{ + guint64 out; + + if (pts == AV_NOPTS_VALUE){ + out = GST_CLOCK_TIME_NONE; + } else { + AVRational bq = { 1, GST_SECOND }; + out = av_rescale_q (pts, base, bq); + } + + return out; +} + +static inline gint64 +gst_ffmpeg_time_gst_to_ff (guint64 time, AVRational base) +{ + gint64 out; + + if (!GST_CLOCK_TIME_IS_VALID (time) || base.num == 0) { + out = AV_NOPTS_VALUE; + } else { + AVRational bq = { 1, GST_SECOND }; + out = av_rescale_q (time, bq, base); + } + + return out; +} + +void +gst_ffmpeg_init_pix_fmt_info(); + + +G_CONST_RETURN gchar * +gst_ffmpeg_get_codecid_longname (enum CodecID codec_id); + +gint +av_smp_format_depth(enum SampleFormat smp_fmt); + +#endif /* __GST_FFMPEG_UTILS_H__ */ |