1 /* GStreamer
2 * Copyright (C) 2018 Matthew Waters <matthew@centricular.com>
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Library General Public
6 * License as published by the Free Software Foundation; either
7 * version 2 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Library General Public License for more details.
13 *
14 * You should have received a copy of the GNU Library General Public
15 * License along with this library; if not, write to the
16 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
17 * Boston, MA 02110-1301, USA.
18 */
19 /*
20 * This file was modified from the alpha element and converted to OpenGL
21 */
22
23 /**
24 * SECTION:element-glalpha
25 * @title: glalpha
26 *
27 * The glalpha element adds an alpha channel to a video stream. The values
28 * of the alpha channel can be either be set to a constant or can be
29 * dynamically calculated via chroma keying, e.g. blue can be set as
30 * the transparent color.
31 *
32 * Sample pipeline:
33 * |[
34 * gst-launch-1.0 gltestsrc pattern=snow ! mixer.sink_0 \
35 * gltestsrc pattern=smpte ! glalpha method=green ! mixer.sink_1 \
36 * glvideomixer name=mixer sink_0::zorder=0 sink_1::zorder=1 ! \
37 * glimagesink
38 * ]| This pipeline adds a alpha channel to the SMPTE color bars
39 * with green as the transparent color and overlays the output on
40 * top of a snow video stream.
41 *
42 */
43
44 #ifdef HAVE_CONFIG_H
45 #include "config.h"
46 #endif
47
48 #include "gstglalpha.h"
49 #include <string.h>
50 #include <math.h>
51 #include <gst/gl/gstglfuncs.h>
52
53 #ifndef M_PI
54 #define M_PI 3.14159265358979323846
55 #endif
56
57 GST_DEBUG_CATEGORY_STATIC (glalpha_debug);
58 #define GST_CAT_DEFAULT glalpha_debug
59
60 #define GST_TYPE_GL_ALPHA_METHOD (gst_gl_alpha_method_get_type())
61 static GType
gst_gl_alpha_method_get_type(void)62 gst_gl_alpha_method_get_type (void)
63 {
64 static GType alpha_method_type = 0;
65 static const GEnumValue alpha_method[] = {
66 {ALPHA_METHOD_SET, "Set/adjust alpha channel", "set"},
67 {ALPHA_METHOD_GREEN, "Chroma Key on pure green", "green"},
68 {ALPHA_METHOD_BLUE, "Chroma Key on pure blue", "blue"},
69 {ALPHA_METHOD_CUSTOM, "Chroma Key on custom RGB values", "custom"},
70 {0, NULL, NULL},
71 };
72
73 if (!alpha_method_type) {
74 alpha_method_type =
75 g_enum_register_static ("GstGLAlphaMethod", alpha_method);
76 }
77 return alpha_method_type;
78 }
79
80 /* GstGLAlpha properties */
81 #define DEFAULT_METHOD ALPHA_METHOD_SET
82 #define DEFAULT_ALPHA 1.0
83 #define DEFAULT_TARGET_R 0
84 #define DEFAULT_TARGET_G 255
85 #define DEFAULT_TARGET_B 0
86 #define DEFAULT_ANGLE 20.0
87 #define DEFAULT_NOISE_LEVEL 2.0
88 #define DEFAULT_BLACK_SENSITIVITY 100
89 #define DEFAULT_WHITE_SENSITIVITY 100
90
91 enum
92 {
93 PROP_0,
94 PROP_METHOD,
95 PROP_ALPHA,
96 PROP_TARGET_R,
97 PROP_TARGET_G,
98 PROP_TARGET_B,
99 PROP_ANGLE,
100 PROP_NOISE_LEVEL,
101 PROP_BLACK_SENSITIVITY,
102 PROP_WHITE_SENSITIVITY,
103 };
104
105 /* *INDENT-OFF* */
106 static const gchar *alpha_frag =
107 "varying vec2 v_texcoord;\n"
108 "uniform sampler2D tex;\n"
109 "uniform float alpha;\n"
110 "void main () {\n"
111 " vec4 rgba = texture2D (tex, v_texcoord);\n"
112 " rgba.a = alpha;\n"
113 " gl_FragColor = rgba;\n"
114 "}\n";
115
116 static const gchar *chroma_key_frag =
117 "varying vec2 v_texcoord;\n"
118 "uniform sampler2D tex;\n"
119 "uniform float cb;\n"
120 "uniform float cr;\n"
121 "uniform float kg;\n"
122 "uniform float accept_angle_tg;\n"
123 "uniform float accept_angle_ctg;\n"
124 "uniform float one_over_kc;\n"
125 "uniform float kfgy_scale;\n"
126 "uniform float noise_level2;\n"
127 "uniform float smin;\n"
128 "uniform float smax;\n"
129 /* these values are taken from the alpha element and divided by 256 to
130 * get the floating point numbers below.
131 * XXX: They are different from the values produced by videoconvert and used
132 * by glcolorconvert
133 298, 0, 409, -57068,
134 298, -100, -208, 34707,
135 298, 516, 0, -70870,
136 */
137 "#define from_yuv_bt601_offset vec3(-0.0625, -0.5, -0.5)\n"
138 "#define from_yuv_bt601_rcoeff vec3(1.1640625, 0.000, 1.787)\n"
139 "#define from_yuv_bt601_gcoeff vec3(1.1640625,-0.213,-0.531)\n"
140 "#define from_yuv_bt601_bcoeff vec3(1.1640625, 2.112, 0.000)\n"
141 /* these values are taken from the alpha element and divided by 256 to
142 * get the floating point numbers below
143 * XXX: They are different from the values produced by videoconvert and used
144 * by glcolorconvert
145 66, 129, 25, 4096,
146 -38, -74, 112, 32768,
147 112, -94, -18, 32768,
148 */
149 "#define from_rgb_bt601_offset vec3(0.0625, 0.5, 0.5)\n"
150 "#define from_rgb_bt601_ycoeff vec3( 0.2578125, 0.50390625, 0.09765625)\n"
151 "#define from_rgb_bt601_ucoeff vec3(-0.1484375,-0.28906250, 0.43750000)\n"
152 "#define from_rgb_bt601_vcoeff vec3( 0.4375000,-0.36718750,-0.07031250)\n"
153 "#define PI 3.14159265\n"
154 "\n"
155 "vec3 yuv_to_rgb (vec3 val) {\n"
156 " vec3 rgb;\n"
157 " val += from_yuv_bt601_offset;\n"
158 " rgb.r = dot(val, from_yuv_bt601_rcoeff);\n"
159 " rgb.g = dot(val, from_yuv_bt601_gcoeff);\n"
160 " rgb.b = dot(val, from_yuv_bt601_bcoeff);\n"
161 " return rgb;\n"
162 "}\n"
163 "vec3 rgb_to_yuv (vec3 val) {\n"
164 " vec3 yuv;\n"
165 " yuv.r = dot(val.rgb, from_rgb_bt601_ycoeff);\n"
166 " yuv.g = dot(val.rgb, from_rgb_bt601_ucoeff);\n"
167 " yuv.b = dot(val.rgb, from_rgb_bt601_vcoeff);\n"
168 " yuv += from_rgb_bt601_offset;\n"
169 " return yuv;\n"
170 "}\n"
171 "vec4 chroma_keying_yuv (vec4 yuva) {\n"
172 " float y, u, v;\n"
173 " y = yuva.x;\n"
174 " u = yuva.y-0.5;\n"
175 " v = yuva.z-0.5;\n"
176 " if (y < smin || y > smax) {\n"
177 " return yuva;\n"
178 " }\n"
179 " vec4 new;\n"
180 " float tmp, tmp1;\n"
181 " float x = clamp ((u * cb + v * cr) * 2.0, -0.5, 0.5);\n"
182 " float z = clamp ((v * cb - u * cr) * 2.0, -0.5, 0.5);\n"
183 " if (abs(z) > min (0.5, x * accept_angle_tg * 0.0625)) {\n"
184 " return yuva;\n"
185 " }\n"
186 " float x1 = min (0.5, abs(z * accept_angle_ctg * 0.0625));\n"
187 " float y1 = z;\n"
188 " tmp1 = max (0.0, x-x1);\n"
189 " float b_alpha = yuva.a * (1.0 - clamp (tmp1 * one_over_kc, 0.0, 1.0));\n"
190 " tmp = min (1.0, tmp1 * kfgy_scale * 0.0625);\n"
191 " new.x = max (y-tmp, 0.0);//y < tmp ? 0.0 : y - tmp;\n"
192 " new.y = clamp ((x1 * cb - y1 * cr) * 2.0, -0.5, 0.5)+0.5;\n"
193 " new.z = clamp ((y1 * cb + x1 * cr) * 2.0, -0.5, 0.5)+0.5;\n"
194 " new.a = z * z + (x - kg) * (x - kg) < noise_level2 ? 0.0 : b_alpha;\n"
195 " return new;\n"
196 "}\n"
197 "void main () {\n"
198 " vec4 yuva;\n"
199 /* operations translated from alpha and tested with glvideomixer
200 * with one pad's paremeters blend-equation-rgb={subtract,reverse-subtract},
201 * blend-function-src-rgb=src-color and blend-function-dst-rgb=dst-color */
202 " vec4 rgba = texture2D (tex, v_texcoord);\n"
203 " yuva.xyz = rgb_to_yuv (rgba.rgb);\n"
204 " yuva.a = rgba.a;\n"
205 " yuva = chroma_keying_yuv (yuva);\n"
206 " rgba.rgb = yuv_to_rgb (yuva.xyz);\n"
207 " rgba.a = yuva.a;\n"
208 " gl_FragColor = rgba;\n"
209 "}\n";
210 /* *INDENT-ON* */
211
212 static const gfloat cog_rgb_to_ycbcr_matrix_8bit_sdtv[] = {
213 0.2578125, 0.50390625, 0.09765625, 0.0625f,
214 -0.1484375, -0.28906250, 0.43750000, 0.5f,
215 0.4375000, -0.36718750, -0.07031250, 0.5f
216 };
217
218 static void gst_gl_alpha_set_property (GObject * object, guint prop_id,
219 const GValue * value, GParamSpec * pspec);
220 static void gst_gl_alpha_get_property (GObject * object, guint prop_id,
221 GValue * value, GParamSpec * pspec);
222
223 #define gst_gl_alpha_parent_class parent_class
224 G_DEFINE_TYPE (GstGLAlpha, gst_gl_alpha, GST_TYPE_GL_FILTER);
225
226 static gboolean
gst_gl_alpha_is_passthrough(GstGLAlpha * glalpha)227 gst_gl_alpha_is_passthrough (GstGLAlpha * glalpha)
228 {
229 return glalpha->method == ALPHA_METHOD_SET && glalpha->alpha == 1.0;
230 }
231
232 /* move v to be inside [from, to] by some multiple of (to-from) */
233 static float
wrap(float v,float from,float to)234 wrap (float v, float from, float to)
235 {
236 float diff;
237
238 if (from > to) {
239 float t = to;
240 to = from;
241 from = t;
242 }
243
244 if (!isfinite (to) || !isfinite (from) || !isfinite (v))
245 return v;
246
247 diff = to - from;
248
249 while (v < from)
250 v += diff;
251 while (v > to)
252 v -= diff;
253
254 return v;
255 }
256
257 static void
gst_gl_alpha_update_properties(GstGLAlpha * glalpha)258 gst_gl_alpha_update_properties (GstGLAlpha * glalpha)
259 {
260 GstBaseTransform *base = GST_BASE_TRANSFORM (glalpha);
261 gboolean current_passthrough, passthrough;
262 gfloat kgl;
263 gfloat tmp;
264 gfloat target_r, target_g, target_b;
265 gfloat target_y, target_u, target_v;
266 const float *matrix = cog_rgb_to_ycbcr_matrix_8bit_sdtv;
267
268 GST_OBJECT_LOCK (glalpha);
269 switch (glalpha->method) {
270 case ALPHA_METHOD_GREEN:
271 target_r = 0.0;
272 target_g = 1.0;
273 target_b = 0.0;
274 break;
275 case ALPHA_METHOD_BLUE:
276 target_r = 0.0;
277 target_g = 0.0;
278 target_b = 1.0;
279 break;
280 default:
281 target_r = (gfloat) glalpha->target_r / 255.0;
282 target_g = (gfloat) glalpha->target_g / 255.0;
283 target_b = (gfloat) glalpha->target_b / 255.0;
284 break;
285 }
286
287 target_y =
288 matrix[0] * target_r + matrix[1] * target_g + matrix[2] * target_b +
289 matrix[3];
290 /* Cb,Cr without offset here because the chroma keying
291 * works with them being in range [-128,127]
292 */
293 target_u = matrix[4] * target_r + matrix[5] * target_g + matrix[6] * target_b;
294 target_v =
295 matrix[8] * target_r + matrix[9] * target_g + matrix[10] * target_b;
296
297 tmp = target_u * target_u + target_v * target_v;
298 kgl = sqrt (tmp);
299 glalpha->cb = target_u / kgl * 0.5;
300 glalpha->cr = target_v / kgl * 0.5;
301
302 tmp = 15 * tan (M_PI * glalpha->angle / 180);
303 tmp = MIN (tmp, 255);
304 glalpha->accept_angle_tg = tmp;
305 tmp = 15 / tan (M_PI * glalpha->angle / 180);
306 tmp = MIN (tmp, 255);
307 glalpha->accept_angle_ctg = tmp;
308 glalpha->one_over_kc = wrap (2 / kgl - 255, 0, 256);
309 tmp = 15 * target_y / kgl;
310 tmp = MIN (tmp, 255);
311 glalpha->kfgy_scale = tmp;
312 glalpha->kg = MIN (kgl, 0.5);
313
314 glalpha->noise_level2 =
315 glalpha->noise_level / 256.0 * glalpha->noise_level / 256.0;
316
317 GST_INFO_OBJECT (glalpha, "target yuv: %f, %f, %f, "
318 "kgl: %f, cb: %f, cr: %f, accept_angle_tg: %f, accept_angle_ctg: %f, "
319 "one_over_kc: %f, kgfy_scale: %f, kg: %f, noise level: %f",
320 (float) target_y, (float) target_u, (float) target_v, (float) kgl,
321 (float) glalpha->cb, (float) glalpha->cr,
322 (float) glalpha->accept_angle_tg, (float) glalpha->accept_angle_ctg,
323 (float) glalpha->one_over_kc, (float) glalpha->kfgy_scale,
324 (float) glalpha->kg, (float) glalpha->noise_level2);
325
326 passthrough = gst_gl_alpha_is_passthrough (glalpha);
327 GST_OBJECT_UNLOCK (glalpha);
328 current_passthrough = gst_base_transform_is_passthrough (base);
329
330 gst_base_transform_set_passthrough (base, passthrough);
331 if (current_passthrough != passthrough)
332 gst_base_transform_reconfigure_src (base);
333 }
334
335 static gboolean
_create_shader(GstGLAlpha * alpha)336 _create_shader (GstGLAlpha * alpha)
337 {
338 GstGLBaseFilter *base_filter = GST_GL_BASE_FILTER (alpha);
339 GstGLFilter *filter = GST_GL_FILTER (alpha);
340 GError *error = NULL;
341 const gchar *frags[2];
342
343 if (alpha->alpha_shader)
344 gst_object_unref (alpha->alpha_shader);
345
346 frags[0] =
347 gst_gl_shader_string_get_highest_precision (base_filter->context,
348 GST_GLSL_VERSION_NONE,
349 GST_GLSL_PROFILE_ES | GST_GLSL_PROFILE_COMPATIBILITY);
350 frags[1] = alpha_frag;
351
352 if (!(alpha->alpha_shader =
353 gst_gl_shader_new_link_with_stages (base_filter->context, &error,
354 gst_glsl_stage_new_default_vertex (base_filter->context),
355 gst_glsl_stage_new_with_strings (base_filter->context,
356 GL_FRAGMENT_SHADER, GST_GLSL_VERSION_NONE,
357 GST_GLSL_PROFILE_ES | GST_GLSL_PROFILE_COMPATIBILITY, 2,
358 frags), NULL))) {
359 GST_ELEMENT_ERROR (alpha, RESOURCE, NOT_FOUND, ("%s",
360 "Failed to initialize alpha shader"), ("%s",
361 error ? error->message : "Unknown error"));
362 return FALSE;
363 }
364
365 if (alpha->chroma_key_shader)
366 gst_object_unref (alpha->chroma_key_shader);
367
368 frags[1] = chroma_key_frag;
369
370 if (!(alpha->chroma_key_shader =
371 gst_gl_shader_new_link_with_stages (base_filter->context, &error,
372 gst_glsl_stage_new_default_vertex (base_filter->context),
373 gst_glsl_stage_new_with_strings (base_filter->context,
374 GL_FRAGMENT_SHADER, GST_GLSL_VERSION_NONE,
375 GST_GLSL_PROFILE_ES | GST_GLSL_PROFILE_COMPATIBILITY, 2,
376 frags), NULL))) {
377 GST_ELEMENT_ERROR (alpha, RESOURCE, NOT_FOUND, ("%s",
378 "Failed to initialize chroma key shader"), ("%s",
379 error ? error->message : "Unknown error"));
380 return FALSE;
381 }
382
383 filter->draw_attr_position_loc =
384 gst_gl_shader_get_attribute_location (alpha->alpha_shader, "a_position");
385 filter->draw_attr_texture_loc =
386 gst_gl_shader_get_attribute_location (alpha->alpha_shader, "a_texcoord");
387
388 return TRUE;
389 }
390
391 static gboolean
gst_gl_alpha_gl_start(GstGLBaseFilter * base_filter)392 gst_gl_alpha_gl_start (GstGLBaseFilter * base_filter)
393 {
394 GstGLAlpha *alpha = GST_GL_ALPHA (base_filter);
395
396 if (!_create_shader (alpha))
397 return FALSE;
398
399 return GST_GL_BASE_FILTER_CLASS (parent_class)->gl_start (base_filter);
400 }
401
402 static void
gst_gl_alpha_gl_stop(GstGLBaseFilter * base_filter)403 gst_gl_alpha_gl_stop (GstGLBaseFilter * base_filter)
404 {
405 GstGLAlpha *alpha = GST_GL_ALPHA (base_filter);
406
407 if (alpha->alpha_shader)
408 gst_object_unref (alpha->alpha_shader);
409 alpha->alpha_shader = NULL;
410
411 if (alpha->chroma_key_shader)
412 gst_object_unref (alpha->chroma_key_shader);
413 alpha->chroma_key_shader = NULL;
414
415 GST_GL_BASE_FILTER_CLASS (parent_class)->gl_stop (base_filter);
416 }
417
418 static void
gst_gl_alpha_before_transform(GstBaseTransform * base,GstBuffer * buf)419 gst_gl_alpha_before_transform (GstBaseTransform * base, GstBuffer * buf)
420 {
421 GstGLAlpha *alpha = GST_GL_ALPHA (base);
422 GstClockTime timestamp, stream_time;
423
424 timestamp = GST_BUFFER_TIMESTAMP (buf);
425 stream_time =
426 gst_segment_to_stream_time (&base->segment, GST_FORMAT_TIME, timestamp);
427
428 GST_DEBUG_OBJECT (alpha, "sync to %" GST_TIME_FORMAT,
429 GST_TIME_ARGS (timestamp));
430
431 if (GST_CLOCK_TIME_IS_VALID (stream_time))
432 gst_object_sync_values (GST_OBJECT (alpha), stream_time);
433 }
434
435 static gboolean
gst_gl_alpha_filter_texture(GstGLFilter * filter,GstGLMemory * in_tex,GstGLMemory * out_tex)436 gst_gl_alpha_filter_texture (GstGLFilter * filter, GstGLMemory * in_tex,
437 GstGLMemory * out_tex)
438 {
439 GstGLAlpha *alpha = GST_GL_ALPHA (filter);
440 GstGLShader *shader;
441
442 if (!alpha->alpha_shader)
443 _create_shader (alpha);
444
445 GST_OBJECT_LOCK (alpha);
446 if (alpha->method == ALPHA_METHOD_SET) {
447 shader = alpha->alpha_shader;
448 gst_gl_shader_use (shader);
449 gst_gl_shader_set_uniform_1f (shader, "alpha", alpha->alpha);
450 } else {
451 shader = alpha->chroma_key_shader;
452 gst_gl_shader_use (shader);
453
454 gst_gl_shader_set_uniform_1f (shader, "cb", alpha->cb);
455 gst_gl_shader_set_uniform_1f (shader, "cr", alpha->cr);
456 gst_gl_shader_set_uniform_1f (shader, "kg", alpha->kg);
457 gst_gl_shader_set_uniform_1f (shader, "accept_angle_tg",
458 alpha->accept_angle_tg);
459 gst_gl_shader_set_uniform_1f (shader, "accept_angle_ctg",
460 alpha->accept_angle_ctg);
461 gst_gl_shader_set_uniform_1f (shader, "one_over_kc", alpha->one_over_kc);
462 gst_gl_shader_set_uniform_1f (shader, "kfgy_scale", alpha->kfgy_scale);
463 gst_gl_shader_set_uniform_1f (shader, "noise_level2", alpha->noise_level2);
464 gst_gl_shader_set_uniform_1f (shader, "smin",
465 0.5 - alpha->black_sensitivity / 255.0);
466 gst_gl_shader_set_uniform_1f (shader, "smax",
467 0.5 + alpha->white_sensitivity / 255.0);
468 }
469 GST_OBJECT_UNLOCK (alpha);
470
471 gst_gl_filter_render_to_target_with_shader (filter, in_tex, out_tex, shader);
472
473 return TRUE;
474 }
475
476 static void
gst_gl_alpha_class_init(GstGLAlphaClass * klass)477 gst_gl_alpha_class_init (GstGLAlphaClass * klass)
478 {
479 GObjectClass *gobject_class = (GObjectClass *) klass;
480 GstElementClass *gstelement_class = (GstElementClass *) klass;
481 GstBaseTransformClass *trans_class = (GstBaseTransformClass *) klass;
482 GstGLBaseFilterClass *base_filter_class = (GstGLBaseFilterClass *) klass;
483 GstGLFilterClass *filter_class = (GstGLFilterClass *) klass;
484
485 GST_DEBUG_CATEGORY_INIT (glalpha_debug, "glalpha", 0, "glalpha");
486
487 gst_gl_filter_add_rgba_pad_templates (GST_GL_FILTER_CLASS (klass));
488
489 gobject_class->set_property = gst_gl_alpha_set_property;
490 gobject_class->get_property = gst_gl_alpha_get_property;
491
492 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_METHOD,
493 g_param_spec_enum ("method", "Method",
494 "How the alpha channels should be created", GST_TYPE_GL_ALPHA_METHOD,
495 DEFAULT_METHOD, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
496 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_ALPHA,
497 g_param_spec_double ("alpha", "Alpha", "The value for the alpha channel",
498 0.0, 1.0, DEFAULT_ALPHA,
499 G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
500 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_TARGET_R,
501 g_param_spec_uint ("target-r", "Target Red",
502 "The red color value for custom RGB chroma keying", 0, 255,
503 DEFAULT_TARGET_R,
504 G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
505 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_TARGET_G,
506 g_param_spec_uint ("target-g", "Target Green",
507 "The green color value for custom RGB chroma keying", 0, 255,
508 DEFAULT_TARGET_G,
509 G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
510 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_TARGET_B,
511 g_param_spec_uint ("target-b", "Target Blue",
512 "The blue color value for custom RGB chroma keying", 0, 255,
513 DEFAULT_TARGET_B,
514 G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
515 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_ANGLE,
516 g_param_spec_float ("angle", "Angle", "Size of the colorcube to change",
517 0.0, 90.0, DEFAULT_ANGLE,
518 G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
519 g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_NOISE_LEVEL,
520 g_param_spec_float ("noise-level", "Noise Level", "Size of noise radius",
521 0.0, 64.0, DEFAULT_NOISE_LEVEL,
522 G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
523 g_object_class_install_property (G_OBJECT_CLASS (klass),
524 PROP_BLACK_SENSITIVITY, g_param_spec_uint ("black-sensitivity",
525 "Black Sensitivity", "Sensitivity to dark colors", 0, 128,
526 DEFAULT_BLACK_SENSITIVITY,
527 G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
528 g_object_class_install_property (G_OBJECT_CLASS (klass),
529 PROP_WHITE_SENSITIVITY, g_param_spec_uint ("white-sensitivity",
530 "White Sensitivity", "Sensitivity to bright colors", 0, 128,
531 DEFAULT_WHITE_SENSITIVITY,
532 G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE | G_PARAM_STATIC_STRINGS));
533
534 gst_element_class_set_static_metadata (gstelement_class,
535 "OpenGL Alpha Filter", "Filter/Effect/Video",
536 "Adds an alpha channel to video using OpenGL - uniform or chroma-keying",
537 "Matthew Waters <matthew@centricular.com>");
538
539 trans_class->before_transform =
540 GST_DEBUG_FUNCPTR (gst_gl_alpha_before_transform);
541 trans_class->transform_ip_on_passthrough = FALSE;
542
543 base_filter_class->gl_start = GST_DEBUG_FUNCPTR (gst_gl_alpha_gl_start);
544 base_filter_class->gl_stop = GST_DEBUG_FUNCPTR (gst_gl_alpha_gl_stop);
545
546 filter_class->filter_texture =
547 GST_DEBUG_FUNCPTR (gst_gl_alpha_filter_texture);
548 }
549
550 static void
gst_gl_alpha_init(GstGLAlpha * alpha)551 gst_gl_alpha_init (GstGLAlpha * alpha)
552 {
553 alpha->alpha = DEFAULT_ALPHA;
554 alpha->method = DEFAULT_METHOD;
555 alpha->target_r = DEFAULT_TARGET_R;
556 alpha->target_g = DEFAULT_TARGET_G;
557 alpha->target_b = DEFAULT_TARGET_B;
558 alpha->angle = DEFAULT_ANGLE;
559 alpha->noise_level = DEFAULT_NOISE_LEVEL;
560 alpha->black_sensitivity = DEFAULT_BLACK_SENSITIVITY;
561 alpha->white_sensitivity = DEFAULT_WHITE_SENSITIVITY;
562
563 gst_gl_alpha_update_properties (alpha);
564 }
565
566 static void
gst_gl_alpha_set_property(GObject * object,guint prop_id,const GValue * value,GParamSpec * pspec)567 gst_gl_alpha_set_property (GObject * object, guint prop_id,
568 const GValue * value, GParamSpec * pspec)
569 {
570 GstGLAlpha *alpha = GST_GL_ALPHA (object);
571
572 GST_OBJECT_LOCK (alpha);
573 switch (prop_id) {
574 case PROP_METHOD:{
575 alpha->method = g_value_get_enum (value);
576 break;
577 }
578 case PROP_ALPHA:{
579 alpha->alpha = g_value_get_double (value);
580 break;
581 }
582 case PROP_TARGET_R:
583 alpha->target_r = g_value_get_uint (value);
584 break;
585 case PROP_TARGET_G:
586 alpha->target_g = g_value_get_uint (value);
587 break;
588 case PROP_TARGET_B:
589 alpha->target_b = g_value_get_uint (value);
590 break;
591 case PROP_ANGLE:
592 alpha->angle = g_value_get_float (value);
593 break;
594 case PROP_NOISE_LEVEL:
595 alpha->noise_level = g_value_get_float (value);
596 break;
597 case PROP_BLACK_SENSITIVITY:
598 alpha->black_sensitivity = g_value_get_uint (value);
599 break;
600 case PROP_WHITE_SENSITIVITY:
601 alpha->white_sensitivity = g_value_get_uint (value);
602 break;
603 default:
604 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
605 break;
606 }
607 GST_OBJECT_UNLOCK (alpha);
608 gst_gl_alpha_update_properties (alpha);
609 }
610
611 static void
gst_gl_alpha_get_property(GObject * object,guint prop_id,GValue * value,GParamSpec * pspec)612 gst_gl_alpha_get_property (GObject * object, guint prop_id,
613 GValue * value, GParamSpec * pspec)
614 {
615 GstGLAlpha *alpha = GST_GL_ALPHA (object);
616
617 switch (prop_id) {
618 case PROP_METHOD:
619 g_value_set_enum (value, alpha->method);
620 break;
621 case PROP_ALPHA:
622 g_value_set_double (value, alpha->alpha);
623 break;
624 case PROP_TARGET_R:
625 g_value_set_uint (value, alpha->target_r);
626 break;
627 case PROP_TARGET_G:
628 g_value_set_uint (value, alpha->target_g);
629 break;
630 case PROP_TARGET_B:
631 g_value_set_uint (value, alpha->target_b);
632 break;
633 case PROP_ANGLE:
634 g_value_set_float (value, alpha->angle);
635 break;
636 case PROP_NOISE_LEVEL:
637 g_value_set_float (value, alpha->noise_level);
638 break;
639 case PROP_BLACK_SENSITIVITY:
640 g_value_set_uint (value, alpha->black_sensitivity);
641 break;
642 case PROP_WHITE_SENSITIVITY:
643 g_value_set_uint (value, alpha->white_sensitivity);
644 break;
645 default:
646 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
647 break;
648 }
649 }
650