ffmpegcolorspace: Add support for Y800 and Y16

Fixes bug #620441.
This commit is contained in:
Martin Bisson 2010-06-07 08:17:13 +02:00 committed by Sebastian Dröge
parent b0d15133d7
commit 4c0b39b680
4 changed files with 257 additions and 4 deletions

View file

@ -72,6 +72,8 @@ enum PixelFormat {
PIX_FMT_YUV410P, ///< Planar YUV 4:1:0 (1 Cr & Cb sample per 4x4 Y samples)
PIX_FMT_YVU410P, ///< Planar YVU 4:1:0 (1 Cr & Cb sample per 4x4 Y samples)
PIX_FMT_YUV411P, ///< Planar YUV 4:1:1 (1 Cr & Cb sample per 4x1 Y samples)
PIX_FMT_Y800, ///< 8 bit Y plane (range [16-235])
PIX_FMT_Y16, ///< 16 bit Y plane (little endian)
PIX_FMT_RGB565, ///< always stored in cpu endianness
PIX_FMT_RGB555, ///< always stored in cpu endianness, most significant bit to 1
PIX_FMT_GRAY8,

View file

@ -340,6 +340,12 @@ gst_ffmpeg_pixfmt_to_caps (enum PixelFormat pix_fmt, AVCodecContext * context)
case PIX_FMT_YUV411P:
fmt = GST_MAKE_FOURCC ('Y', '4', '1', 'B');
break;
case PIX_FMT_Y800:
fmt = GST_MAKE_FOURCC ('Y', '8', '0', '0');
break;
case PIX_FMT_Y16:
fmt = GST_MAKE_FOURCC ('Y', '1', '6', ' ');
break;
case PIX_FMT_RGB565:
bpp = depth = 16;
endianness = G_BYTE_ORDER;
@ -655,7 +661,12 @@ gst_ffmpeg_caps_to_pixfmt (const GstCaps * caps,
context->pix_fmt = PIX_FMT_YUV444P;
break;
case GST_MAKE_FOURCC ('Y', '8', '0', '0'):
context->pix_fmt = PIX_FMT_GRAY8;
case GST_MAKE_FOURCC ('Y', '8', ' ', ' '):
case GST_MAKE_FOURCC ('G', 'R', 'E', 'Y'):
context->pix_fmt = PIX_FMT_Y800;
break;
case GST_MAKE_FOURCC ('Y', '1', '6', ' '):
context->pix_fmt = PIX_FMT_Y16;
break;
}
}
@ -926,6 +937,7 @@ gst_ffmpegcsp_avpicture_fill (AVPicture * picture,
picture->data[2] = NULL;
picture->linesize[0] = width + width / 2;
return size + size / 2;
case PIX_FMT_Y800:
case PIX_FMT_GRAY8:
stride = GST_ROUND_UP_4 (width);
size = stride * height;
@ -934,6 +946,7 @@ gst_ffmpegcsp_avpicture_fill (AVPicture * picture,
picture->data[2] = NULL;
picture->linesize[0] = stride;
return size;
case PIX_FMT_Y16:
case PIX_FMT_GRAY16_L:
case PIX_FMT_GRAY16_B:
stride = GST_ROUND_UP_4 (width * 2);

View file

@ -187,6 +187,28 @@ static PixFmtInfo pix_fmt_info[PIX_FMT_NB] = {
/* .y_chroma_shift = */ 0,
/* .depth = */ 8,
},
/* [PIX_FMT_Y800] = */ {
/* .format = */ PIX_FMT_Y800,
/* .name = */ "y800",
/* .nb_channels = */ 1,
/* .color_type = */ FF_COLOR_YUV,
/* .pixel_type = */ FF_PIXEL_PLANAR,
/* .is_alpha = */ 0,
/* .x_chroma_shift = */ 0,
/* .y_chroma_shift = */ 0,
/* .depth = */ 8,
},
/* [PIX_FMT_Y16] = */ {
/* .format = */ PIX_FMT_Y16,
/* .name = */ "y16",
/* .nb_channels = */ 1,
/* .color_type = */ FF_COLOR_YUV,
/* .pixel_type = */ FF_PIXEL_PLANAR,
/* .is_alpha = */ 0,
/* .x_chroma_shift = */ 0,
/* .y_chroma_shift = */ 0,
/* .depth = */ 16,
},
/* JPEG YUV */
/* [PIX_FMT_YUVJ420P] = */ {
@ -2586,6 +2608,56 @@ gray_to_monoblack (AVPicture * dst, const AVPicture * src,
gray_to_mono (dst, src, width, height, 0x00);
}
static void
y800_to_y16 (AVPicture * dst, const AVPicture * src, int width, int height)
{
const unsigned char *p;
unsigned char *q;
int dst_wrap, src_wrap;
int x, y;
p = src->data[0];
src_wrap = src->linesize[0] - width;
q = dst->data[0];
dst_wrap = dst->linesize[0] - 2 * width;
for (y = 0; y < height; y++) {
for (x = 0; x < width; x++) {
GST_WRITE_UINT16_LE (q, (*p << 8));
q += 2;
p++;
}
p += src_wrap;
q += dst_wrap;
}
}
static void
y16_to_y800 (AVPicture * dst, const AVPicture * src, int width, int height)
{
const unsigned char *p;
unsigned char *q;
int dst_wrap, src_wrap;
int x, y;
p = src->data[0];
src_wrap = src->linesize[0] - 2 * width;
q = dst->data[0];
dst_wrap = dst->linesize[0] - width;
for (y = 0; y < height; y++) {
for (x = 0; x < width; x++) {
q[0] = GST_READ_UINT16_LE (p) >> 8;
q++;
p += 2;
}
p += src_wrap;
q += dst_wrap;
}
}
typedef struct ConvertEntry
{
enum PixelFormat src;
@ -2714,6 +2786,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_RGB24, PIX_FMT_BGRA32, rgb24_to_bgra32},
{PIX_FMT_RGB24, PIX_FMT_ARGB32, rgb24_to_argb32},
{PIX_FMT_RGB24, PIX_FMT_ABGR32, rgb24_to_abgr32},
{PIX_FMT_RGB24, PIX_FMT_Y800, rgb24_to_y800},
{PIX_FMT_RGB24, PIX_FMT_Y16, rgb24_to_y16},
{PIX_FMT_RGB24, PIX_FMT_GRAY8, rgb24_to_gray},
{PIX_FMT_RGB24, PIX_FMT_GRAY16_L, rgb24_to_gray16_l},
{PIX_FMT_RGB24, PIX_FMT_GRAY16_B, rgb24_to_gray16_b},
@ -2730,6 +2804,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_RGB32, PIX_FMT_YUV420P, rgb32_to_yuv420p},
{PIX_FMT_RGB32, PIX_FMT_NV12, rgb32_to_nv12},
{PIX_FMT_RGB32, PIX_FMT_NV21, rgb32_to_nv21},
{PIX_FMT_RGB32, PIX_FMT_Y800, rgb32_to_y800},
{PIX_FMT_RGB32, PIX_FMT_Y16, rgb32_to_y16},
{PIX_FMT_RGB32, PIX_FMT_GRAY8, rgb32_to_gray},
{PIX_FMT_RGB32, PIX_FMT_GRAY16_L, rgb32_to_gray16_l},
{PIX_FMT_RGB32, PIX_FMT_GRAY16_B, rgb32_to_gray16_b},
@ -2739,6 +2815,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_xRGB32, PIX_FMT_YUV420P, xrgb32_to_yuv420p},
{PIX_FMT_xRGB32, PIX_FMT_NV12, xrgb32_to_nv12},
{PIX_FMT_xRGB32, PIX_FMT_NV21, xrgb32_to_nv21},
{PIX_FMT_xRGB32, PIX_FMT_Y800, xrgb32_to_y800},
{PIX_FMT_xRGB32, PIX_FMT_Y16, xrgb32_to_y16},
{PIX_FMT_xRGB32, PIX_FMT_GRAY8, xrgb32_to_gray},
{PIX_FMT_xRGB32, PIX_FMT_GRAY16_L, xrgb32_to_gray16_l},
{PIX_FMT_xRGB32, PIX_FMT_GRAY16_B, xrgb32_to_gray16_b},
@ -2755,6 +2833,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_RGBA32, PIX_FMT_YUV420P, rgba32_to_yuv420p},
{PIX_FMT_RGBA32, PIX_FMT_NV12, rgba32_to_nv12},
{PIX_FMT_RGBA32, PIX_FMT_NV21, rgba32_to_nv21},
{PIX_FMT_RGBA32, PIX_FMT_Y800, rgba32_to_y800},
{PIX_FMT_RGBA32, PIX_FMT_Y16, rgba32_to_y16},
{PIX_FMT_RGBA32, PIX_FMT_GRAY8, rgba32_to_gray},
{PIX_FMT_RGBA32, PIX_FMT_GRAY16_L, rgba32_to_gray16_l},
{PIX_FMT_RGBA32, PIX_FMT_GRAY16_B, rgba32_to_gray16_b},
@ -2764,6 +2844,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_BGR24, PIX_FMT_YUV420P, bgr24_to_yuv420p},
{PIX_FMT_BGR24, PIX_FMT_NV12, bgr24_to_nv12},
{PIX_FMT_BGR24, PIX_FMT_NV21, bgr24_to_nv21},
{PIX_FMT_BGR24, PIX_FMT_Y800, bgr24_to_y800},
{PIX_FMT_BGR24, PIX_FMT_Y16, bgr24_to_y16},
{PIX_FMT_BGR24, PIX_FMT_GRAY8, bgr24_to_gray},
{PIX_FMT_BGR24, PIX_FMT_GRAY16_L, bgr24_to_gray16_l},
{PIX_FMT_BGR24, PIX_FMT_GRAY16_B, bgr24_to_gray16_b},
@ -2773,6 +2855,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_BGR32, PIX_FMT_YUV420P, bgr32_to_yuv420p},
{PIX_FMT_BGR32, PIX_FMT_NV12, bgr32_to_nv12},
{PIX_FMT_BGR32, PIX_FMT_NV21, bgr32_to_nv21},
{PIX_FMT_BGR32, PIX_FMT_Y800, bgr32_to_y800},
{PIX_FMT_BGR32, PIX_FMT_Y16, bgr32_to_y16},
{PIX_FMT_BGR32, PIX_FMT_GRAY8, bgr32_to_gray},
{PIX_FMT_BGR32, PIX_FMT_GRAY16_L, bgr32_to_gray16_l},
{PIX_FMT_BGR32, PIX_FMT_GRAY16_B, bgr32_to_gray16_b},
@ -2782,6 +2866,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_BGRx32, PIX_FMT_YUV420P, bgrx32_to_yuv420p},
{PIX_FMT_BGRx32, PIX_FMT_NV12, bgrx32_to_nv12},
{PIX_FMT_BGRx32, PIX_FMT_NV21, bgrx32_to_nv21},
{PIX_FMT_BGRx32, PIX_FMT_Y800, bgrx32_to_y800},
{PIX_FMT_BGRx32, PIX_FMT_Y16, bgrx32_to_y16},
{PIX_FMT_BGRx32, PIX_FMT_GRAY8, bgrx32_to_gray},
{PIX_FMT_BGRx32, PIX_FMT_GRAY16_L, bgrx32_to_gray16_l},
{PIX_FMT_BGRx32, PIX_FMT_GRAY16_B, bgrx32_to_gray16_b},
@ -2791,6 +2877,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_BGRA32, PIX_FMT_YUV420P, bgra32_to_yuv420p},
{PIX_FMT_BGRA32, PIX_FMT_NV12, bgra32_to_nv12},
{PIX_FMT_BGRA32, PIX_FMT_NV21, bgra32_to_nv21},
{PIX_FMT_BGRA32, PIX_FMT_Y800, bgra32_to_y800},
{PIX_FMT_BGRA32, PIX_FMT_Y16, bgra32_to_y16},
{PIX_FMT_BGRA32, PIX_FMT_GRAY8, bgra32_to_gray},
{PIX_FMT_BGRA32, PIX_FMT_GRAY16_L, bgra32_to_gray16_l},
{PIX_FMT_BGRA32, PIX_FMT_GRAY16_B, bgra32_to_gray16_b},
@ -2801,6 +2889,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_ABGR32, PIX_FMT_YUV420P, abgr32_to_yuv420p},
{PIX_FMT_ABGR32, PIX_FMT_NV12, abgr32_to_nv12},
{PIX_FMT_ABGR32, PIX_FMT_NV21, abgr32_to_nv21},
{PIX_FMT_ABGR32, PIX_FMT_Y800, abgr32_to_y800},
{PIX_FMT_ABGR32, PIX_FMT_Y16, abgr32_to_y16},
{PIX_FMT_ABGR32, PIX_FMT_GRAY8, abgr32_to_gray},
{PIX_FMT_ABGR32, PIX_FMT_GRAY16_L, abgr32_to_gray16_l},
{PIX_FMT_ABGR32, PIX_FMT_GRAY16_B, abgr32_to_gray16_b},
@ -2811,6 +2901,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_ARGB32, PIX_FMT_YUV420P, argb32_to_yuv420p},
{PIX_FMT_ARGB32, PIX_FMT_NV12, argb32_to_nv12},
{PIX_FMT_ARGB32, PIX_FMT_NV21, argb32_to_nv21},
{PIX_FMT_ARGB32, PIX_FMT_Y800, argb32_to_y800},
{PIX_FMT_ARGB32, PIX_FMT_Y16, argb32_to_y16},
{PIX_FMT_ARGB32, PIX_FMT_GRAY8, argb32_to_gray},
{PIX_FMT_ARGB32, PIX_FMT_GRAY16_L, argb32_to_gray16_l},
{PIX_FMT_ARGB32, PIX_FMT_GRAY16_B, argb32_to_gray16_b},
@ -2822,6 +2914,8 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_RGB555, PIX_FMT_YUV420P, rgb555_to_yuv420p},
{PIX_FMT_RGB555, PIX_FMT_NV12, rgb555_to_nv12},
{PIX_FMT_RGB555, PIX_FMT_NV21, rgb555_to_nv21},
{PIX_FMT_RGB555, PIX_FMT_Y800, rgb555_to_y800},
{PIX_FMT_RGB555, PIX_FMT_Y16, rgb555_to_y16},
{PIX_FMT_RGB555, PIX_FMT_GRAY8, rgb555_to_gray},
{PIX_FMT_RGB555, PIX_FMT_GRAY16_L, rgb555_to_gray16_l},
{PIX_FMT_RGB555, PIX_FMT_GRAY16_B, rgb555_to_gray16_b},
@ -2830,10 +2924,40 @@ static ConvertEntry convert_table[] = {
{PIX_FMT_RGB565, PIX_FMT_YUV420P, rgb565_to_yuv420p},
{PIX_FMT_RGB565, PIX_FMT_NV12, rgb565_to_nv12},
{PIX_FMT_RGB565, PIX_FMT_NV21, rgb565_to_nv21},
{PIX_FMT_RGB565, PIX_FMT_Y800, rgb565_to_y800},
{PIX_FMT_RGB565, PIX_FMT_Y16, rgb565_to_y16},
{PIX_FMT_RGB565, PIX_FMT_GRAY8, rgb565_to_gray},
{PIX_FMT_RGB565, PIX_FMT_GRAY16_L, rgb565_to_gray16_l},
{PIX_FMT_RGB565, PIX_FMT_GRAY16_B, rgb565_to_gray16_b},
{PIX_FMT_Y800, PIX_FMT_RGB555, y800_to_rgb555},
{PIX_FMT_Y800, PIX_FMT_RGB565, y800_to_rgb565},
{PIX_FMT_Y800, PIX_FMT_BGR24, y800_to_bgr24},
{PIX_FMT_Y800, PIX_FMT_RGB24, y800_to_rgb24},
{PIX_FMT_Y800, PIX_FMT_RGB32, y800_to_rgb32},
{PIX_FMT_Y800, PIX_FMT_BGR32, y800_to_bgr32},
{PIX_FMT_Y800, PIX_FMT_RGB32, y800_to_xrgb32},
{PIX_FMT_Y800, PIX_FMT_BGR32, y800_to_bgrx32},
{PIX_FMT_Y800, PIX_FMT_RGBA32, y800_to_rgba32},
{PIX_FMT_Y800, PIX_FMT_BGRA32, y800_to_bgra32},
{PIX_FMT_Y800, PIX_FMT_ARGB32, y800_to_argb32},
{PIX_FMT_Y800, PIX_FMT_ABGR32, y800_to_abgr32},
{PIX_FMT_Y800, PIX_FMT_Y16, y800_to_y16},
{PIX_FMT_Y16, PIX_FMT_RGB555, y16_to_rgb555},
{PIX_FMT_Y16, PIX_FMT_RGB565, y16_to_rgb565},
{PIX_FMT_Y16, PIX_FMT_BGR24, y16_to_bgr24},
{PIX_FMT_Y16, PIX_FMT_RGB24, y16_to_rgb24},
{PIX_FMT_Y16, PIX_FMT_RGB32, y16_to_rgb32},
{PIX_FMT_Y16, PIX_FMT_BGR32, y16_to_bgr32},
{PIX_FMT_Y16, PIX_FMT_RGB32, y16_to_xrgb32},
{PIX_FMT_Y16, PIX_FMT_BGR32, y16_to_bgrx32},
{PIX_FMT_Y16, PIX_FMT_RGBA32, y16_to_rgba32},
{PIX_FMT_Y16, PIX_FMT_BGRA32, y16_to_bgra32},
{PIX_FMT_Y16, PIX_FMT_ARGB32, y16_to_argb32},
{PIX_FMT_Y16, PIX_FMT_ABGR32, y16_to_abgr32},
{PIX_FMT_Y16, PIX_FMT_Y800, y16_to_y800},
{PIX_FMT_GRAY8, PIX_FMT_RGB555, gray_to_rgb555},
{PIX_FMT_GRAY8, PIX_FMT_RGB565, gray_to_rgb565},
{PIX_FMT_GRAY8, PIX_FMT_RGB24, gray_to_rgb24},
@ -3036,7 +3160,8 @@ img_convert (AVPicture * dst, int dst_pix_fmt,
}
/* YUV to YUV planar */
if (is_yuv_planar (dst_pix) && is_yuv_planar (src_pix)) {
if (is_yuv_planar (dst_pix) && is_yuv_planar (src_pix) &&
dst_pix->depth == src_pix->depth) {
int x_shift, y_shift, xy_shift;
void (*resize_func) (uint8_t * dst, int dst_wrap,
const uint8_t * src, int src_wrap, int width, int height);
@ -3098,8 +3223,12 @@ img_convert (AVPicture * dst, int dst_pix_fmt,
w = DIV_ROUND_UP_X (dst_width, dst_pix->x_chroma_shift);
h = DIV_ROUND_UP_X (dst_height, dst_pix->y_chroma_shift);
resize_func (dst->data[i], dst->linesize[i],
src->data[i], src->linesize[i], w, h);
if (src->data[i] != NULL && dst->data[i] != NULL) {
resize_func (dst->data[i], dst->linesize[i],
src->data[i], src->linesize[i], w, h);
} else if (dst->data[i] != NULL) {
memset (dst->data[i], 128, dst->linesize[i] * h);
}
}
/* if yuv color space conversion is needed, we do it here on
the destination image */
@ -3145,6 +3274,9 @@ no_chroma_filter:
(dst_pix->color_type == FF_COLOR_GRAY && dst_pix_fmt != PIX_FMT_GRAY8)) {
/* gray8 is the normalized format */
int_pix_fmt = PIX_FMT_GRAY8;
} else if (src_pix_fmt == PIX_FMT_Y16 || dst_pix_fmt == PIX_FMT_Y16) {
/* y800 is the normalized format */
int_pix_fmt = PIX_FMT_Y800;
} else if ((is_yuv_planar (src_pix) &&
src_pix_fmt != PIX_FMT_YUV444P && src_pix_fmt != PIX_FMT_YUVJ444P)) {
/* yuv444 is the normalized format */

View file

@ -416,6 +416,60 @@ static void glue (yuvj420p_to_, RGB_NAME) (AVPicture * dst,
}
}
static void glue (y800_to_, RGB_NAME) (AVPicture * dst, const AVPicture * src,
int width, int height)
{
const unsigned char *p;
unsigned char *q;
uint8_t *cm = cropTbl + MAX_NEG_CROP;
int r, dst_wrap, src_wrap;
int x, y;
p = src->data[0];
src_wrap = src->linesize[0] - width;
q = dst->data[0];
dst_wrap = dst->linesize[0] - BPP * width;
for (y = 0; y < height; y++) {
for (x = 0; x < width; x++) {
r = Y_CCIR_TO_JPEG (p[0]);
RGB_OUT (q, r, r, r);
q += BPP;
p++;
}
p += src_wrap;
q += dst_wrap;
}
}
static void glue (y16_to_, RGB_NAME) (AVPicture * dst,
const AVPicture * src, int width, int height)
{
const unsigned char *p;
unsigned char *q;
uint8_t *cm = cropTbl + MAX_NEG_CROP;
int r, dst_wrap, src_wrap;
int x, y;
p = src->data[0];
src_wrap = src->linesize[0] - 2 * width;
q = dst->data[0];
dst_wrap = dst->linesize[0] - BPP * width;
for (y = 0; y < height; y++) {
for (x = 0; x < width; x++) {
r = Y_CCIR_TO_JPEG (GST_READ_UINT16_LE (p) >> 8);
RGB_OUT (q, r, r, r);
q += BPP;
p += 2;
}
p += src_wrap;
q += dst_wrap;
}
}
static void glue (RGB_NAME, _to_yuv420p) (AVPicture * dst,
const AVPicture * src, int width, int height)
{
@ -756,6 +810,58 @@ static void glue (RGB_NAME, _to_gray) (AVPicture * dst, const AVPicture * src,
}
}
static void glue (RGB_NAME, _to_y800) (AVPicture * dst, const AVPicture * src,
int width, int height)
{
const unsigned char *p;
unsigned char *q;
int r, g, b, dst_wrap, src_wrap;
int x, y;
p = src->data[0];
src_wrap = src->linesize[0] - BPP * width;
q = dst->data[0];
dst_wrap = dst->linesize[0] - width;
for (y = 0; y < height; y++) {
for (x = 0; x < width; x++) {
RGB_IN (r, g, b, p);
q[0] = RGB_TO_Y_CCIR (r, g, b);
q++;
p += BPP;
}
p += src_wrap;
q += dst_wrap;
}
}
static void glue (RGB_NAME, _to_y16) (AVPicture * dst,
const AVPicture * src, int width, int height)
{
const unsigned char *p;
unsigned char *q;
int r, g, b, dst_wrap, src_wrap;
int x, y;
p = src->data[0];
src_wrap = src->linesize[0] - BPP * width;
q = dst->data[0];
dst_wrap = dst->linesize[0] - 2 * width;
for (y = 0; y < height; y++) {
for (x = 0; x < width; x++) {
RGB_IN (r, g, b, p);
GST_WRITE_UINT16_LE (q, RGB_TO_Y_CCIR (r, g, b) << 8);
q += 2;
p += BPP;
}
p += src_wrap;
q += dst_wrap;
}
}
static void glue (gray_to_, RGB_NAME) (AVPicture * dst, const AVPicture * src,
int width, int height)
{