gstreamer/gst/videotestsrc/gstvideotestsrc.c

651 lines
16 KiB
C
Raw Normal View History

/* GStreamer
* Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
*
* 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.
*/
/*#define DEBUG_ENABLED */
#include <gstvideotestsrc.h>
#include <string.h>
#include <stdlib.h>
/* elementfactory information */
static GstElementDetails videotestsrc_details = {
"Video test source",
"Source/Video",
"Creates a test video stream",
VERSION,
"David A. Schleef <ds@schleef.org>",
"(C) 2002",
};
/* GstVideotestsrc signals and args */
enum {
/* FILL ME */
LAST_SIGNAL
};
enum {
ARG_0,
ARG_WIDTH,
ARG_HEIGHT,
ARG_METHOD,
/* FILL ME */
};
GST_PAD_TEMPLATE_FACTORY (src_templ,
"src",
GST_PAD_SRC,
GST_PAD_ALWAYS,
GST_CAPS_NEW (
"videotestsrc_caps",
"video/raw",
"format", GST_PROPS_LIST(
GST_PROPS_FOURCC (GST_MAKE_FOURCC ('I','4','2','0')),
GST_PROPS_FOURCC (GST_MAKE_FOURCC ('Y','V','1','2')),
GST_PROPS_FOURCC (GST_MAKE_FOURCC ('R','G','B',' '))
),
"width", GST_PROPS_INT(640),
"height", GST_PROPS_INT(480)
)
)
#define GST_TYPE_VIDEOTESTSRC_METHOD (gst_videotestsrc_method_get_type())
static GType
gst_videotestsrc_method_get_type (void)
{
static GType videotestsrc_method_type = 0;
static GEnumValue videotestsrc_methods[] = {
{ GST_VIDEOTESTSRC_POINT_SAMPLE, "0", "Point Sample" },
{ GST_VIDEOTESTSRC_NEAREST, "1", "Nearest" },
{ GST_VIDEOTESTSRC_BILINEAR, "2", "Bilinear" },
{ GST_VIDEOTESTSRC_BICUBIC, "3", "Bicubic" },
{ 0, NULL, NULL },
};
if (!videotestsrc_method_type) {
videotestsrc_method_type = g_enum_register_static ("GstVideotestsrcMethod", videotestsrc_methods);
}
return videotestsrc_method_type;
}
static void gst_videotestsrc_class_init (GstVideotestsrcClass *klass);
static void gst_videotestsrc_init (GstVideotestsrc *videotestsrc);
static GstElementStateReturn gst_videotestsrc_change_state (GstElement *element);
static void gst_videotestsrc_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec);
static void gst_videotestsrc_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec);
static GstBuffer * gst_videotestsrc_get (GstPad *pad);
static GstElementClass *parent_class = NULL;
void gst_videotestsrc_setup (GstVideotestsrc *v);
static void random_chars(unsigned char *dest, int nbytes);
void gst_videotestsrc_smpte_I420 (GstVideotestsrc *v, unsigned char *dest, int w, int h);
void gst_videotestsrc_smpte_YV12 (GstVideotestsrc *v, unsigned char *dest, int w, int h);
void gst_videotestsrc_smpte_RGB (GstVideotestsrc *v, unsigned char *dest, int w, int h);
GType
gst_videotestsrc_get_type (void)
{
static GType videotestsrc_type = 0;
if (!videotestsrc_type) {
static const GTypeInfo videotestsrc_info = {
sizeof(GstVideotestsrcClass), NULL,
NULL,
(GClassInitFunc)gst_videotestsrc_class_init,
NULL,
NULL,
sizeof(GstVideotestsrc),
0,
(GInstanceInitFunc)gst_videotestsrc_init,
};
videotestsrc_type = g_type_register_static(GST_TYPE_ELEMENT, "GstVideotestsrc", &videotestsrc_info, 0);
}
return videotestsrc_type;
}
static void
gst_videotestsrc_class_init (GstVideotestsrcClass *klass)
{
GObjectClass *gobject_class;
GstElementClass *gstelement_class;
gobject_class = (GObjectClass*)klass;
gstelement_class = (GstElementClass*)klass;
g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_WIDTH,
g_param_spec_int("width","width","width",
G_MININT,G_MAXINT,0,G_PARAM_READWRITE)); /* CHECKME */
g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_HEIGHT,
g_param_spec_int("height","height","height",
G_MININT,G_MAXINT,0,G_PARAM_READWRITE)); /* CHECKME */
g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_METHOD,
g_param_spec_enum("method","method","method",
GST_TYPE_VIDEOTESTSRC_METHOD,0,G_PARAM_READWRITE)); /* CHECKME! */
parent_class = g_type_class_ref(GST_TYPE_ELEMENT);
gobject_class->set_property = gst_videotestsrc_set_property;
gobject_class->get_property = gst_videotestsrc_get_property;
gstelement_class->change_state = gst_videotestsrc_change_state;
}
static GstPadConnectReturn
gst_videotestsrc_srcconnect (GstPad *pad, GstCaps *caps)
{
GstVideotestsrc *videotestsrc;
GST_DEBUG(0,"gst_videotestsrc_srcconnect");
videotestsrc = GST_VIDEOTESTSRC (gst_pad_get_parent (pad));
#if 0
if (!GST_CAPS_IS_FIXED (caps)) {
return GST_PAD_CONNECT_DELAYED;
}
#endif
gst_caps_get_fourcc_int (caps, "format", &videotestsrc->format);
gst_caps_get_int (caps, "width", &videotestsrc->width);
gst_caps_get_int (caps, "height", &videotestsrc->height);
GST_DEBUG (0,"format is 0x%08x\n",videotestsrc->format);
switch(videotestsrc->format){
case GST_MAKE_FOURCC('R','G','B',' '):
videotestsrc->make_image = gst_videotestsrc_smpte_RGB;
videotestsrc->bpp = 16;
break;
case GST_MAKE_FOURCC('I','4','2','0'):
videotestsrc->make_image = gst_videotestsrc_smpte_I420;
videotestsrc->bpp = 12;
break;
case GST_MAKE_FOURCC('Y','U','Y','V'):
//videotestsrc->make_image = gst_videotestsrc_smpte_YUYV;
return GST_PAD_CONNECT_REFUSED;
break;
case GST_MAKE_FOURCC('Y','V','1','2'):
videotestsrc->make_image = gst_videotestsrc_smpte_YV12;
videotestsrc->bpp = 12;
break;
default:
return GST_PAD_CONNECT_REFUSED;
}
GST_DEBUG (0,"size %d x %d",videotestsrc->width, videotestsrc->height);
return GST_PAD_CONNECT_OK;
}
static GstElementStateReturn
gst_videotestsrc_change_state (GstElement *element)
{
GstVideotestsrc *v;
v = GST_VIDEOTESTSRC(element);
switch(GST_STATE_TRANSITION(element)){
case GST_STATE_PAUSED_TO_PLAYING:
v->pool = gst_pad_get_bufferpool(v->srcpad);
break;
case GST_STATE_PLAYING_TO_PAUSED:
v->pool = NULL;
break;
}
parent_class->change_state(element);
return GST_STATE_SUCCESS;
}
static void
gst_videotestsrc_init (GstVideotestsrc *videotestsrc)
{
GST_DEBUG(0,"gst_videotestsrc_init");
/*gst_pad_set_negotiate_function(videotestsrc->sinkpad,videotestsrc_negotiate_sink); */
//gst_element_add_pad(GST_ELEMENT(videotestsrc),videotestsrc->sinkpad);
//gst_pad_set_chain_function(videotestsrc->sinkpad,gst_videotestsrc_chain);
videotestsrc->srcpad = gst_pad_new_from_template (
GST_PAD_TEMPLATE_GET (src_templ), "src");
/*gst_pad_set_negotiate_function(videotestsrc->srcpad,videotestsrc_negotiate_src); */
gst_element_add_pad(GST_ELEMENT(videotestsrc),videotestsrc->srcpad);
gst_pad_set_get_function(videotestsrc->srcpad,gst_videotestsrc_get);
gst_pad_set_connect_function(videotestsrc->srcpad,gst_videotestsrc_srcconnect);
#if 0
"bpp", GST_PROPS_INT(16),
"depth", GST_PROPS_INT(16),
"endianness", GST_PROPS_INT(1234),
"red_mask", GST_PROPS_INT(63488),
"green_mask", GST_PROPS_INT(2016),
"blue_mask", GST_PROPS_INT(31),
#endif
videotestsrc->width = 640;
videotestsrc->height = 480;
videotestsrc->timestamp = 0;
videotestsrc->interval = GST_SECOND/30;
videotestsrc->pool = NULL;
}
static GstBuffer *
gst_videotestsrc_get (GstPad *pad)
{
GstVideotestsrc *videotestsrc;
gulong newsize;
GstBuffer *buf;
GST_DEBUG (0,"gst_videotestsrc_get");
g_return_val_if_fail (pad != NULL, NULL);
g_return_val_if_fail (GST_IS_PAD (pad), NULL);
videotestsrc = GST_VIDEOTESTSRC (gst_pad_get_parent (pad));
#if 0
/* XXX this is wrong for anything but I420 */
newsize = videotestsrc->width*videotestsrc->height +
videotestsrc->width*videotestsrc->height/2;
#endif
/* XXX this is wrong for anything but RGB16 */
newsize = (videotestsrc->width*videotestsrc->height*videotestsrc->bpp)>>3;
GST_DEBUG(0,"size=%ld %dx%d",newsize,
videotestsrc->width, videotestsrc->height);
buf = NULL;
if(videotestsrc->pool){
buf = gst_buffer_new_from_pool(videotestsrc->pool, 0, 0);
}
if(!buf){
buf = gst_buffer_new();
GST_BUFFER_SIZE(buf) = newsize;
GST_BUFFER_DATA(buf) = g_malloc (newsize);
}
g_return_val_if_fail(GST_BUFFER_DATA(buf) != NULL, NULL);
videotestsrc->timestamp += videotestsrc->interval;
GST_BUFFER_TIMESTAMP(buf) = videotestsrc->timestamp;
videotestsrc->make_image(videotestsrc, (void *)GST_BUFFER_DATA(buf),
videotestsrc->width, videotestsrc->height);
return buf;
}
static void
gst_videotestsrc_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec)
{
GstVideotestsrc *src;
/* it's not null if we got it, but it might not be ours */
g_return_if_fail(GST_IS_VIDEOTESTSRC(object));
src = GST_VIDEOTESTSRC(object);
GST_DEBUG(0,"gst_videotestsrc_set_property");
switch (prop_id) {
case ARG_WIDTH:
src->width = g_value_get_int (value);
break;
case ARG_HEIGHT:
src->height = g_value_get_int (value);
break;
case ARG_METHOD:
src->method = g_value_get_enum (value);
break;
default:
break;
}
}
static void
gst_videotestsrc_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
{
GstVideotestsrc *src;
/* it's not null if we got it, but it might not be ours */
g_return_if_fail(GST_IS_VIDEOTESTSRC(object));
src = GST_VIDEOTESTSRC(object);
switch (prop_id) {
case ARG_WIDTH:
g_value_set_int (value, src->width);
break;
case ARG_HEIGHT:
g_value_set_int (value, src->height);
break;
case ARG_METHOD:
g_value_set_enum (value, src->method);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
}
}
static gboolean
plugin_init (GModule *module, GstPlugin *plugin)
{
GstElementFactory *factory;
/* create an elementfactory for the videotestsrc element */
factory = gst_element_factory_new("videotestsrc",GST_TYPE_VIDEOTESTSRC,
&videotestsrc_details);
g_return_val_if_fail(factory != NULL, FALSE);
gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET (src_templ));
gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory));
return TRUE;
}
GstPluginDesc plugin_desc = {
GST_VERSION_MAJOR,
GST_VERSION_MINOR,
"videotestsrc",
plugin_init
};
/* Non-GST specific stuff */
void
gst_videotestsrc_setup (GstVideotestsrc *v)
{
}
static void
random_chars(unsigned char *dest, int nbytes)
{
int i;
static unsigned int state;
for(i=0;i<nbytes;i++){
state *= 1103515245;
state += 12345;
dest[i] = (state>>16);
}
}
static void
paint_rect_random (unsigned char *dest, int stride, int x, int y, int w, int h)
{
unsigned char *d = dest + stride*y + x;
int i;
for(i=0;i<h;i++){
random_chars(d,w);
d += stride;
}
}
static void
paint_rect (unsigned char *dest, int stride, int x, int y, int w, int h, unsigned char color)
{
unsigned char *d = dest + stride*y + x;
int i;
for(i=0;i<h;i++){
memset(d,color,w);
d += stride;
}
}
static void
paint_rect2 (unsigned char *dest, int stride, int x, int y, int w, int h, unsigned char *col)
{
unsigned char *d = dest + stride*y + x*2;
unsigned char *dp;
int i,j;
for(i=0;i<h;i++){
dp = d;
for(j=0;j<w;j++){
*dp++ = col[0];
*dp++ = col[1];
}
d += stride;
}
}
static void
paint_rect3 (unsigned char *dest, int stride, int x, int y, int w, int h, unsigned char *col)
{
unsigned char *d = dest + stride*y + x*3;
unsigned char *dp;
int i,j;
for(i=0;i<h;i++){
dp = d;
for(j=0;j<w;j++){
*dp++ = col[0];
*dp++ = col[1];
*dp++ = col[2];
}
d += stride;
}
}
/* wht yel cya grn mag red blu blk -I Q */
static int y_colors[] = { 255, 226, 179, 150, 105, 76, 29, 16, 16, 0 };
static int u_colors[] = { 128, 0, 170, 46, 212, 85, 255, 128, 0, 128 };
static int v_colors[] = { 128, 155, 0, 21, 235, 255, 107, 128, 128, 255 };
void
gst_videotestsrc_smpte_I420 (GstVideotestsrc *v, unsigned char *dest, int w, int h)
{
unsigned char *yp = dest;
unsigned char *up = dest + w*h;
unsigned char *vp = dest + w*h + w*h/4;
//int h24 = h/24;
//int j,k;
int i;
int y1,y2;
y1 = h/3;
y2 = h*0.375;
/* color bars */
for(i=0;i<7;i++){
int x1 = i*(w/2)/7;
int x2 = (i+1)*(w/2)/7;
paint_rect(yp, w, x1*2, 0, (x2-x1)*2, y1*2, y_colors[i]);
paint_rect(up, w/2, x1, 0, x2-x1, y1, u_colors[i]);
paint_rect(vp, w/2, x1, 0, x2-x1, y1, v_colors[i]);
}
/* inverse blue bars */
for(i=0;i<7;i++){
int x1 = i*(w/2)/7;
int x2 = (i+1)*(w/2)/7;
int k;
if(i&1){
k = 7;
}else{
k = 6-i;
}
paint_rect(yp, w, x1*2, y1*2, (x2-x1)*2, (y2-y1)*2, y_colors[k]);
paint_rect(up, w/2, x1, y1, x2-x1, y2-y1, u_colors[k]);
paint_rect(vp, w/2, x1, y1, x2-x1, y2-y1, v_colors[k]);
}
/* -I, white, Q regions */
for(i=0;i<3;i++){
int x1 = i*(w/2)/6;
int x2 = (i+1)*(w/2)/6;
int k;
if(i==0){ k = 8; }
else if(i==1){ k = 0; }
else k = 9;
paint_rect(yp, w, x1*2, y2*2, (x2-x1)*2, h-y2*2, y_colors[k]);
paint_rect(up, w/2, x1, y2, x2-x1, h/2-y2, u_colors[k]);
paint_rect(vp, w/2, x1, y2, x2-x1, h/2-y2, v_colors[k]);
}
{
int x1 = 3*(w/2)/6;
int x2 = w/2;
paint_rect_random(yp, w, x1*2, y2*2, (x2-x1)*2, h-y2*2);
paint_rect(up, w/2, x1, y2, x2-x1, h/2-y2, u_colors[0]);
paint_rect(vp, w/2, x1, y2, x2-x1, h/2-y2, v_colors[0]);
}
}
/* same as I420, but with U and V swapped */
void
gst_videotestsrc_smpte_YV12 (GstVideotestsrc *v, unsigned char *dest, int w, int h)
{
unsigned char *yp = dest;
unsigned char *up = dest + w*h + w*h/4;
unsigned char *vp = dest + w*h;
//int h24 = h/24;
//int j,k;
int i;
int y1,y2;
y1 = h/3;
y2 = h*0.375;
/* color bars */
for(i=0;i<7;i++){
int x1 = i*(w/2)/7;
int x2 = (i+1)*(w/2)/7;
paint_rect(yp, w, x1*2, 0, (x2-x1)*2, y1*2, y_colors[i]);
paint_rect(up, w/2, x1, 0, x2-x1, y1, u_colors[i]);
paint_rect(vp, w/2, x1, 0, x2-x1, y1, v_colors[i]);
}
/* inverse blue bars */
for(i=0;i<7;i++){
int x1 = i*(w/2)/7;
int x2 = (i+1)*(w/2)/7;
int k;
if(i&1){
k = 7;
}else{
k = 6-i;
}
paint_rect(yp, w, x1*2, y1*2, (x2-x1)*2, (y2-y1)*2, y_colors[k]);
paint_rect(up, w/2, x1, y1, x2-x1, y2-y1, u_colors[k]);
paint_rect(vp, w/2, x1, y1, x2-x1, y2-y1, v_colors[k]);
}
/* -I, white, Q regions */
for(i=0;i<3;i++){
int x1 = i*(w/2)/6;
int x2 = (i+1)*(w/2)/6;
int k;
if(i==0){ k = 8; }
else if(i==1){ k = 0; }
else k = 9;
paint_rect(yp, w, x1*2, y2*2, (x2-x1)*2, h-y2*2, y_colors[k]);
paint_rect(up, w/2, x1, y2, x2-x1, h/2-y2, u_colors[k]);
paint_rect(vp, w/2, x1, y2, x2-x1, h/2-y2, v_colors[k]);
}
{
int x1 = 3*(w/2)/6;
int x2 = w/2;
paint_rect_random(yp, w, x1*2, y2*2, (x2-x1)*2, h-y2*2);
paint_rect(up, w/2, x1, y2, x2-x1, h/2-y2, u_colors[0]);
paint_rect(vp, w/2, x1, y2, x2-x1, h/2-y2, v_colors[0]);
}
}
/* wht yel cya grn mag red blu blk -I Q */
static int r_colors[] = { 255, 255, 0, 0, 255, 255, 0, 0, 0, 0 };
static int g_colors[] = { 255, 255, 255, 255, 0, 0, 0, 0, 0, 128 };
static int b_colors[] = { 255, 0, 255, 0, 255, 0, 255, 0, 128, 255 };
void
gst_videotestsrc_smpte_RGB (GstVideotestsrc *v, unsigned char *dest, int w, int h)
{
int i;
int y1,y2;
y1 = h*2/3;
y2 = h*0.75;
/* color bars */
for(i=0;i<7;i++){
int x1 = i*w/7;
int x2 = (i+1)*w/7;
unsigned char col[2];
col[0] = (g_colors[i]&0xe0) | (b_colors[i]>>3);
col[1] = (r_colors[i]&0xf8) | (g_colors[i]>>5);
paint_rect2(dest, w*2, x1, 0, x2-x1, y1, col);
}
/* inverse blue bars */
for(i=0;i<7;i++){
int x1 = i*w/7;
int x2 = (i+1)*w/7;
unsigned char col[2];
int k;
if(i&1){
k = 7;
}else{
k = 6-i;
}
col[0] = (g_colors[k]&0xe0) | (b_colors[k]>>3);
col[1] = (r_colors[k]&0xf8) | (g_colors[k]>>5);
paint_rect2(dest, w*2, x1, y1, x2-x1, y2-y1, col);
}
/* -I, white, Q regions */
for(i=0;i<3;i++){
int x1 = i*w/6;
int x2 = (i+1)*w/6;
unsigned char col[2];
int k;
if(i==0){ k = 8; }
else if(i==1){ k = 0; }
else k = 9;
col[0] = (g_colors[k]&0xe0) | (b_colors[k]>>3);
col[1] = (r_colors[k]&0xf8) | (g_colors[k]>>5);
paint_rect2(dest, w*2, x1, y2, x2-x1, h-y2, col);
}
{
int x1 = w/2;
int x2 = w-1;
paint_rect_random(dest, w*2, x1*2, y2, (x2-x1)*2, h-y2);
}
}