mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-08 21:28:45 +00:00
Allow the full range of 32 bit masks for RGBA/RGB data by making sure the mask properties are unsigned.
421 lines
14 KiB
C
421 lines
14 KiB
C
/* GStreamer
|
|
* Copyright (C) 2006 David A. Schleef <ds@schleef.org>
|
|
* Copyright (C) 2007 Sebastian Dröge <slomo@circular-chaos.org>
|
|
*
|
|
* gstvideoparse.c:
|
|
*
|
|
* 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.
|
|
*/
|
|
/**
|
|
* SECTION:element-videoparse
|
|
*
|
|
* Converts a byte stream into video frames.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
|
|
#include "gstvideoparse.h"
|
|
|
|
typedef enum
|
|
{
|
|
GST_VIDEO_PARSE_FORMAT_I420,
|
|
GST_VIDEO_PARSE_FORMAT_YV12,
|
|
GST_VIDEO_PARSE_FORMAT_YUY2,
|
|
GST_VIDEO_PARSE_FORMAT_UYVY,
|
|
GST_VIDEO_PARSE_FORMAT_RGB = 10,
|
|
GST_VIDEO_PARSE_FORMAT_GRAY
|
|
} GstVideoParseFormat;
|
|
|
|
typedef enum
|
|
{
|
|
GST_VIDEO_PARSE_ENDIANNESS_LITTLE = 1234,
|
|
GST_VIDEO_PARSE_ENDIANNESS_BIG = 4321
|
|
} GstVideoParseEndianness;
|
|
|
|
static void gst_video_parse_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec);
|
|
static void gst_video_parse_get_property (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec);
|
|
|
|
static GstCaps *gst_video_parse_get_caps (GstRawParse * rp);
|
|
|
|
static void gst_video_parse_update_frame_size (GstVideoParse * vp);
|
|
|
|
GST_DEBUG_CATEGORY_STATIC (gst_video_parse_debug);
|
|
#define GST_CAT_DEFAULT gst_video_parse_debug
|
|
|
|
static const GstElementDetails gst_video_parse_details =
|
|
GST_ELEMENT_DETAILS ("Video Parse",
|
|
"Filter/Video",
|
|
"Converts stream into video frames",
|
|
"David Schleef <ds@schleef.org>, "
|
|
"Sebastian Dröge <slomo@circular-chaos.org>");
|
|
|
|
enum
|
|
{
|
|
ARG_0,
|
|
ARG_WIDTH,
|
|
ARG_HEIGHT,
|
|
ARG_FORMAT,
|
|
ARG_PAR,
|
|
ARG_FRAMERATE,
|
|
ARG_BPP,
|
|
ARG_DEPTH,
|
|
ARG_ENDIANNESS,
|
|
ARG_RED_MASK,
|
|
ARG_GREEN_MASK,
|
|
ARG_BLUE_MASK,
|
|
ARG_ALPHA_MASK
|
|
};
|
|
|
|
|
|
#define GST_VIDEO_PARSE_FORMAT (gst_video_parse_format_get_type ())
|
|
static GType
|
|
gst_video_parse_format_get_type (void)
|
|
{
|
|
static GType video_parse_format_type = 0;
|
|
static const GEnumValue format_types[] = {
|
|
{GST_VIDEO_PARSE_FORMAT_I420, "I420", "I420"},
|
|
{GST_VIDEO_PARSE_FORMAT_YV12, "YV12", "YV12"},
|
|
{GST_VIDEO_PARSE_FORMAT_YUY2, "YUY2", "YUY2"},
|
|
{GST_VIDEO_PARSE_FORMAT_UYVY, "UYVY", "UYVY"},
|
|
{GST_VIDEO_PARSE_FORMAT_RGB, "RGB", "RGB"},
|
|
{GST_VIDEO_PARSE_FORMAT_GRAY, "GRAY", "GRAY"},
|
|
{0, NULL, NULL}
|
|
};
|
|
|
|
if (!video_parse_format_type) {
|
|
video_parse_format_type =
|
|
g_enum_register_static ("GstVideoParseFormat", format_types);
|
|
}
|
|
|
|
return video_parse_format_type;
|
|
}
|
|
|
|
#define GST_VIDEO_PARSE_ENDIANNESS (gst_video_parse_endianness_get_type ())
|
|
static GType
|
|
gst_video_parse_endianness_get_type (void)
|
|
{
|
|
static GType video_parse_endianness_type = 0;
|
|
static const GEnumValue endian_types[] = {
|
|
{GST_VIDEO_PARSE_ENDIANNESS_LITTLE, "Little Endian", "little"},
|
|
{GST_VIDEO_PARSE_ENDIANNESS_BIG, "Big Endian", "big"},
|
|
{0, NULL, NULL}
|
|
};
|
|
|
|
if (!video_parse_endianness_type) {
|
|
video_parse_endianness_type =
|
|
g_enum_register_static ("GstVideoParseEndianness", endian_types);
|
|
}
|
|
|
|
return video_parse_endianness_type;
|
|
}
|
|
|
|
GST_BOILERPLATE (GstVideoParse, gst_video_parse, GstRawParse,
|
|
GST_TYPE_RAW_PARSE);
|
|
|
|
static void
|
|
gst_video_parse_base_init (gpointer g_class)
|
|
{
|
|
GstRawParseClass *rp_class = GST_RAW_PARSE_CLASS (g_class);
|
|
GstElementClass *gstelement_class = GST_ELEMENT_CLASS (g_class);
|
|
GstCaps *caps;
|
|
|
|
GST_DEBUG_CATEGORY_INIT (gst_video_parse_debug, "videoparse", 0,
|
|
"videoparse element");
|
|
|
|
gst_element_class_set_details (gstelement_class, &gst_video_parse_details);
|
|
|
|
caps =
|
|
gst_caps_from_string (GST_VIDEO_CAPS_YUV
|
|
("{ I420, YV12, YUY2, UYVY }") ";" "video/x-raw-rgb; video/x-raw-gray");
|
|
|
|
gst_raw_parse_class_set_src_pad_template (rp_class, caps);
|
|
gst_raw_parse_class_set_multiple_frames_per_buffer (rp_class, FALSE);
|
|
gst_caps_unref (caps);
|
|
}
|
|
|
|
static void
|
|
gst_video_parse_class_init (GstVideoParseClass * klass)
|
|
{
|
|
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
|
|
GstRawParseClass *rp_class = GST_RAW_PARSE_CLASS (klass);
|
|
|
|
gobject_class->set_property = gst_video_parse_set_property;
|
|
gobject_class->get_property = gst_video_parse_get_property;
|
|
|
|
rp_class->get_caps = gst_video_parse_get_caps;
|
|
|
|
g_object_class_install_property (gobject_class, ARG_WIDTH,
|
|
g_param_spec_int ("width", "Width", "Width of images in raw stream",
|
|
0, INT_MAX, 320, G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_HEIGHT,
|
|
g_param_spec_int ("height", "Height", "Height of images in raw stream",
|
|
0, INT_MAX, 240, G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_FORMAT,
|
|
g_param_spec_enum ("format", "Format", "Format of images in raw stream",
|
|
GST_VIDEO_PARSE_FORMAT, GST_VIDEO_PARSE_FORMAT_I420,
|
|
G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_FRAMERATE,
|
|
gst_param_spec_fraction ("framerate", "Frame Rate",
|
|
"Frame rate of images in raw stream", 0, 1, 100, 1, 25, 1,
|
|
G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_PAR,
|
|
gst_param_spec_fraction ("pixel_aspect_ratio", "Pixel Aspect Ratio",
|
|
"Pixel aspect ratio of images in raw stream", 1, 100, 100, 1, 1, 1,
|
|
G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_BPP,
|
|
g_param_spec_int ("bpp", "Bpp", "Bits per pixel of images in raw stream",
|
|
0, INT_MAX, 24, G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_DEPTH,
|
|
g_param_spec_int ("depth", "Depth", "Depth of images in raw stream",
|
|
0, INT_MAX, 24, G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_ENDIANNESS,
|
|
g_param_spec_enum ("endianness", "Endianness",
|
|
"Endianness of images in raw stream", GST_VIDEO_PARSE_ENDIANNESS,
|
|
G_BYTE_ORDER, G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_RED_MASK,
|
|
g_param_spec_int ("red-mask", "Red mask",
|
|
"Red mask of images in raw stream", INT_MIN, INT_MAX,
|
|
GST_VIDEO_BYTE1_MASK_24_INT, G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_GREEN_MASK,
|
|
g_param_spec_int ("green-mask", "Green mask",
|
|
"Green mask of images in raw stream", INT_MIN, INT_MAX,
|
|
GST_VIDEO_BYTE2_MASK_24_INT, G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_BLUE_MASK,
|
|
g_param_spec_int ("blue-mask", "Blue mask",
|
|
"Blue mask of images in raw stream", INT_MIN, INT_MAX,
|
|
GST_VIDEO_BYTE3_MASK_24_INT, G_PARAM_READWRITE));
|
|
g_object_class_install_property (gobject_class, ARG_ALPHA_MASK,
|
|
g_param_spec_int ("alpha-mask", "Alpha mask",
|
|
"Alpha mask of images in raw stream", INT_MIN, INT_MAX, 0,
|
|
G_PARAM_READWRITE));
|
|
}
|
|
|
|
static void
|
|
gst_video_parse_init (GstVideoParse * vp, GstVideoParseClass * g_class)
|
|
{
|
|
vp->width = 320;
|
|
vp->height = 240;
|
|
vp->format = GST_VIDEO_PARSE_FORMAT_I420;
|
|
vp->par_n = 1;
|
|
vp->par_d = 1;
|
|
vp->bpp = 24;
|
|
vp->depth = 24;
|
|
vp->endianness = G_BYTE_ORDER;
|
|
vp->red_mask = GST_VIDEO_BYTE1_MASK_24_INT;
|
|
vp->green_mask = GST_VIDEO_BYTE2_MASK_24_INT;
|
|
vp->blue_mask = GST_VIDEO_BYTE3_MASK_24_INT;
|
|
vp->alpha_mask = 0;
|
|
|
|
gst_video_parse_update_frame_size (vp);
|
|
gst_raw_parse_set_fps (GST_RAW_PARSE (vp), 25, 1);
|
|
}
|
|
|
|
static void
|
|
gst_video_parse_set_property (GObject * object, guint prop_id,
|
|
const GValue * value, GParamSpec * pspec)
|
|
{
|
|
GstVideoParse *vp = GST_VIDEO_PARSE (object);
|
|
|
|
g_return_if_fail (!gst_raw_parse_is_negotiated (GST_RAW_PARSE (vp)));
|
|
|
|
switch (prop_id) {
|
|
case ARG_WIDTH:
|
|
vp->width = g_value_get_int (value);
|
|
break;
|
|
case ARG_HEIGHT:
|
|
vp->height = g_value_get_int (value);
|
|
break;
|
|
case ARG_FORMAT:
|
|
vp->format = g_value_get_enum (value);
|
|
break;
|
|
case ARG_FRAMERATE:
|
|
gst_raw_parse_set_fps (GST_RAW_PARSE (vp),
|
|
gst_value_get_fraction_numerator (value),
|
|
gst_value_get_fraction_denominator (value));
|
|
break;
|
|
case ARG_PAR:
|
|
vp->par_n = gst_value_get_fraction_numerator (value);
|
|
vp->par_d = gst_value_get_fraction_denominator (value);
|
|
break;
|
|
case ARG_BPP:
|
|
vp->bpp = g_value_get_int (value);
|
|
break;
|
|
case ARG_DEPTH:
|
|
vp->depth = g_value_get_int (value);
|
|
break;
|
|
case ARG_ENDIANNESS:
|
|
vp->endianness = g_value_get_enum (value);
|
|
break;
|
|
case ARG_RED_MASK:
|
|
vp->red_mask = g_value_get_int (value);
|
|
break;
|
|
case ARG_GREEN_MASK:
|
|
vp->green_mask = g_value_get_int (value);
|
|
break;
|
|
case ARG_BLUE_MASK:
|
|
vp->blue_mask = g_value_get_int (value);
|
|
break;
|
|
case ARG_ALPHA_MASK:
|
|
vp->alpha_mask = g_value_get_int (value);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
|
|
gst_video_parse_update_frame_size (vp);
|
|
}
|
|
|
|
static void
|
|
gst_video_parse_get_property (GObject * object, guint prop_id, GValue * value,
|
|
GParamSpec * pspec)
|
|
{
|
|
GstVideoParse *vp = GST_VIDEO_PARSE (object);
|
|
|
|
switch (prop_id) {
|
|
case ARG_WIDTH:
|
|
g_value_set_int (value, vp->width);
|
|
break;
|
|
case ARG_HEIGHT:
|
|
g_value_set_int (value, vp->height);
|
|
break;
|
|
case ARG_FORMAT:
|
|
g_value_set_enum (value, vp->format);
|
|
break;
|
|
case ARG_FRAMERATE:{
|
|
gint fps_n, fps_d;
|
|
|
|
gst_raw_parse_get_fps (GST_RAW_PARSE (vp), &fps_n, &fps_d);
|
|
gst_value_set_fraction (value, fps_n, fps_d);
|
|
break;
|
|
}
|
|
case ARG_PAR:
|
|
gst_value_set_fraction (value, vp->par_n, vp->par_d);
|
|
break;
|
|
case ARG_BPP:
|
|
g_value_set_int (value, vp->bpp);
|
|
break;
|
|
case ARG_DEPTH:
|
|
g_value_set_int (value, vp->depth);
|
|
break;
|
|
case ARG_ENDIANNESS:
|
|
g_value_set_enum (value, vp->endianness);
|
|
break;
|
|
case ARG_RED_MASK:
|
|
g_value_set_int (value, vp->red_mask);
|
|
break;
|
|
case ARG_GREEN_MASK:
|
|
g_value_set_int (value, vp->green_mask);
|
|
break;
|
|
case ARG_BLUE_MASK:
|
|
g_value_set_int (value, vp->blue_mask);
|
|
break;
|
|
case ARG_ALPHA_MASK:
|
|
g_value_set_int (value, vp->alpha_mask);
|
|
break;
|
|
default:
|
|
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
|
|
break;
|
|
}
|
|
}
|
|
|
|
static guint32
|
|
gst_video_parse_format_to_fourcc (GstVideoParseFormat format)
|
|
{
|
|
switch (format) {
|
|
case GST_VIDEO_PARSE_FORMAT_I420:
|
|
return GST_MAKE_FOURCC ('I', '4', '2', '0');
|
|
case GST_VIDEO_PARSE_FORMAT_YV12:
|
|
return GST_MAKE_FOURCC ('Y', 'V', '1', '2');
|
|
case GST_VIDEO_PARSE_FORMAT_YUY2:
|
|
return GST_MAKE_FOURCC ('Y', 'U', 'Y', '2');
|
|
case GST_VIDEO_PARSE_FORMAT_UYVY:
|
|
return GST_MAKE_FOURCC ('U', 'Y', 'V', 'Y');
|
|
default:
|
|
g_assert_not_reached ();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
gst_video_parse_update_frame_size (GstVideoParse * vp)
|
|
{
|
|
gint framesize;
|
|
|
|
if (vp->format == GST_VIDEO_PARSE_FORMAT_I420 ||
|
|
vp->format == GST_VIDEO_PARSE_FORMAT_YV12) {
|
|
framesize = GST_ROUND_UP_4 (vp->width) * GST_ROUND_UP_2 (vp->height)
|
|
+
|
|
2 * (GST_ROUND_UP_8 (vp->width) / 2) * (GST_ROUND_UP_2 (vp->height) /
|
|
2);
|
|
} else if (vp->format == GST_VIDEO_PARSE_FORMAT_YUY2
|
|
|| vp->format == GST_VIDEO_PARSE_FORMAT_UYVY) {
|
|
framesize = GST_ROUND_UP_4 (vp->width * 2) * vp->height;
|
|
} else if (vp->format == GST_VIDEO_PARSE_FORMAT_RGB) {
|
|
framesize = GST_ROUND_UP_4 (vp->width * vp->bpp / 8) * vp->height;
|
|
} else {
|
|
framesize = GST_ROUND_UP_4 (vp->width * vp->bpp / 8) * vp->height;
|
|
}
|
|
|
|
gst_raw_parse_set_framesize (GST_RAW_PARSE (vp), framesize);
|
|
}
|
|
|
|
static GstCaps *
|
|
gst_video_parse_get_caps (GstRawParse * rp)
|
|
{
|
|
GstVideoParse *vp = GST_VIDEO_PARSE (rp);
|
|
GstCaps *caps;
|
|
|
|
gint fps_n, fps_d;
|
|
|
|
gst_raw_parse_get_fps (rp, &fps_n, &fps_d);
|
|
|
|
if (vp->format < GST_VIDEO_PARSE_FORMAT_RGB) {
|
|
caps = gst_caps_new_simple ("video/x-raw-yuv",
|
|
"width", G_TYPE_INT, vp->width,
|
|
"height", G_TYPE_INT, vp->height,
|
|
"format", GST_TYPE_FOURCC,
|
|
gst_video_parse_format_to_fourcc (vp->format), "framerate",
|
|
GST_TYPE_FRACTION, fps_n, fps_d, "pixel_aspect_ratio",
|
|
GST_TYPE_FRACTION, vp->par_n, vp->par_d, NULL);
|
|
} else if (vp->format == GST_VIDEO_PARSE_FORMAT_RGB) {
|
|
caps = gst_caps_new_simple ("video/x-raw-rgb",
|
|
"width", G_TYPE_INT, vp->width,
|
|
"height", G_TYPE_INT, vp->height,
|
|
"bpp", G_TYPE_INT, vp->bpp,
|
|
"depth", G_TYPE_INT, vp->depth,
|
|
"framerate", GST_TYPE_FRACTION, fps_n, fps_d,
|
|
"pixel_aspect_ratio", GST_TYPE_FRACTION, vp->par_n, vp->par_d,
|
|
"red_mask", G_TYPE_INT, vp->red_mask,
|
|
"green_mask", G_TYPE_INT, vp->green_mask,
|
|
"blue_mask", G_TYPE_INT, vp->blue_mask,
|
|
"alpha_mask", G_TYPE_INT, vp->alpha_mask,
|
|
"endianness", G_TYPE_INT, vp->endianness, NULL);
|
|
} else {
|
|
caps = gst_caps_new_simple ("video/x-raw-gray",
|
|
"width", G_TYPE_INT, vp->width,
|
|
"height", G_TYPE_INT, vp->height,
|
|
"bpp", G_TYPE_INT, vp->bpp,
|
|
"depth", G_TYPE_INT, vp->depth,
|
|
"framerate", GST_TYPE_FRACTION, fps_n, fps_d,
|
|
"pixel_aspect_ratio", GST_TYPE_FRACTION, vp->par_n, vp->par_d, NULL);
|
|
}
|
|
return caps;
|
|
}
|