2 * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
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.
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.
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., 59 Temple Place - Suite 330,
17 * Boston, MA 02111-1307, USA.
25 #include <gst/video/video.h>
28 /* elementfactory information */
29 static GstElementDetails smpte_details = {
31 "Filter/Editor/Video",
32 "Apply the standard SMPTE transitions on video images",
33 "Wim Taymans <wim.taymans@chello.be>"
36 static GstStaticPadTemplate gst_smpte_src_template =
37 GST_STATIC_PAD_TEMPLATE ("src",
40 GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420")
44 static GstStaticPadTemplate gst_smpte_sink1_template =
45 GST_STATIC_PAD_TEMPLATE ("sink1",
48 GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420")
52 static GstStaticPadTemplate gst_smpte_sink2_template =
53 GST_STATIC_PAD_TEMPLATE ("sink2",
56 GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("I420")
61 /* SMPTE signals and args */
77 #define GST_TYPE_SMPTE_TRANSITION_TYPE (gst_smpte_transition_type_get_type())
79 gst_smpte_transition_type_get_type (void)
81 static GType smpte_transition_type = 0;
82 GEnumValue *smpte_transitions;
84 if (!smpte_transition_type) {
85 const GList *definitions;
88 definitions = gst_mask_get_definitions ();
90 g_new0 (GEnumValue, g_list_length ((GList *) definitions) + 1);
93 GstMaskDefinition *definition = (GstMaskDefinition *) definitions->data;
95 definitions = g_list_next (definitions);
97 smpte_transitions[i].value = definition->type;
98 smpte_transitions[i].value_name = definition->short_name;
99 smpte_transitions[i].value_nick = definition->long_name;
104 smpte_transition_type =
105 g_enum_register_static ("GstSMPTETransitionType", smpte_transitions);
107 return smpte_transition_type;
111 static void gst_smpte_class_init (GstSMPTEClass * klass);
112 static void gst_smpte_base_init (GstSMPTEClass * klass);
113 static void gst_smpte_init (GstSMPTE * smpte);
115 static GstFlowReturn gst_smpte_collected (GstCollectPads * pads,
118 static void gst_smpte_set_property (GObject * object, guint prop_id,
119 const GValue * value, GParamSpec * pspec);
120 static void gst_smpte_get_property (GObject * object, guint prop_id,
121 GValue * value, GParamSpec * pspec);
123 static GstElementClass *parent_class = NULL;
125 /*static guint gst_smpte_signals[LAST_SIGNAL] = { 0 }; */
128 gst_smpte_get_type (void)
130 static GType smpte_type = 0;
133 static const GTypeInfo smpte_info = {
134 sizeof (GstSMPTEClass),
135 (GBaseInitFunc) gst_smpte_base_init,
137 (GClassInitFunc) gst_smpte_class_init,
142 (GInstanceInitFunc) gst_smpte_init,
146 g_type_register_static (GST_TYPE_ELEMENT, "GstSMPTE", &smpte_info, 0);
152 gst_smpte_base_init (GstSMPTEClass * klass)
154 GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
156 gst_element_class_add_pad_template (element_class,
157 gst_static_pad_template_get (&gst_smpte_sink1_template));
158 gst_element_class_add_pad_template (element_class,
159 gst_static_pad_template_get (&gst_smpte_sink2_template));
160 gst_element_class_add_pad_template (element_class,
161 gst_static_pad_template_get (&gst_smpte_src_template));
162 gst_element_class_set_details (element_class, &smpte_details);
166 gst_smpte_class_init (GstSMPTEClass * klass)
168 GObjectClass *gobject_class;
169 GstElementClass *gstelement_class;
171 gobject_class = (GObjectClass *) klass;
172 gstelement_class = (GstElementClass *) klass;
174 parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
176 gobject_class->set_property = gst_smpte_set_property;
177 gobject_class->get_property = gst_smpte_get_property;
181 g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_TYPE,
182 g_param_spec_enum ("type", "Type", "The type of transition to use",
183 GST_TYPE_SMPTE_TRANSITION_TYPE, 1, G_PARAM_READWRITE));
184 g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_FPS,
185 g_param_spec_float ("fps", "FPS",
186 "Frames per second if no input files are given", 0., G_MAXFLOAT, 25.,
188 g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_BORDER,
189 g_param_spec_int ("border", "Border",
190 "The border width of the transition", 0, G_MAXINT, 0,
192 g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_DEPTH,
193 g_param_spec_int ("depth", "Depth", "Depth of the mask in bits", 1, 24,
194 16, G_PARAM_READWRITE));
197 /* wht yel cya grn mag red blu blk -I Q */
198 static int y_colors[] = { 255, 226, 179, 150, 105, 76, 29, 16, 16, 0 };
199 static int u_colors[] = { 128, 0, 170, 46, 212, 85, 255, 128, 0, 128 };
200 static int v_colors[] = { 128, 155, 0, 21, 235, 255, 107, 128, 128, 255 };
203 fill_i420 (guint8 * data, gint width, gint height, gint color)
205 gint size = width * height, size4 = size >> 2;
207 guint8 *up = data + size;
208 guint8 *vp = data + size + size4;
210 memset (yp, y_colors[color], size);
211 memset (up, u_colors[color], size4);
212 memset (vp, v_colors[color], size4);
216 gst_smpte_update_mask (GstSMPTE * smpte, gint type, gint depth, gint width,
221 newmask = gst_mask_factory_new (type, depth, width, height);
224 gst_mask_destroy (smpte->mask);
226 smpte->mask = newmask;
228 smpte->depth = depth;
229 smpte->width = width;
230 smpte->height = height;
238 gst_smpte_setcaps (GstPad * pad, GstCaps * caps)
241 GstStructure *structure;
244 smpte = GST_SMPTE (GST_PAD_PARENT (pad));
246 structure = gst_caps_get_structure (caps, 0);
248 ret = gst_structure_get_int (structure, "width", &smpte->width);
249 ret &= gst_structure_get_int (structure, "height", &smpte->height);
250 ret &= gst_structure_get_double (structure, "framerate", &smpte->fps);
254 gst_smpte_update_mask (smpte, smpte->type, smpte->depth, smpte->width,
257 /* forward to the next plugin */
262 gst_smpte_init (GstSMPTE * smpte)
265 gst_pad_new_from_template (gst_static_pad_template_get
266 (&gst_smpte_sink1_template), "sink1");
267 gst_pad_set_setcaps_function (smpte->sinkpad1, gst_smpte_setcaps);
268 gst_element_add_pad (GST_ELEMENT (smpte), smpte->sinkpad1);
271 gst_pad_new_from_template (gst_static_pad_template_get
272 (&gst_smpte_sink2_template), "sink2");
273 gst_pad_set_setcaps_function (smpte->sinkpad2, gst_smpte_setcaps);
274 gst_element_add_pad (GST_ELEMENT (smpte), smpte->sinkpad2);
277 gst_pad_new_from_template (gst_static_pad_template_get
278 (&gst_smpte_src_template), "src");
279 gst_element_add_pad (GST_ELEMENT (smpte), smpte->srcpad);
281 smpte->collect = gst_collectpads_new ();
282 gst_collectpads_set_function (smpte->collect,
283 (GstCollectPadsFunction) gst_smpte_collected, smpte);
284 gst_collectpads_start (smpte->collect);
286 gst_collectpads_add_pad (smpte->collect, smpte->sinkpad1,
287 sizeof (GstCollectData));
288 gst_collectpads_add_pad (smpte->collect, smpte->sinkpad2,
289 sizeof (GstCollectData));
294 smpte->duration = 512;
299 gst_smpte_update_mask (smpte, smpte->type, smpte->depth, smpte->width,
304 gst_smpte_blend_i420 (guint8 * in1, guint8 * in2, guint8 * out, GstMask * mask,
305 gint width, gint height, gint border, gint pos)
311 guint8 *in1u, *in1v, *in2u, *in2v, *outu, *outv;
312 gint lumsize = width * height;
313 gint chromsize = lumsize >> 2;
321 in1u = in1 + lumsize;
322 in1v = in1u + chromsize;
323 in2u = in2 + lumsize;
324 in2v = in2u + chromsize;
325 outu = out + lumsize;
326 outv = outu + chromsize;
330 for (i = 0; i < height; i++) {
331 for (j = 0; j < width; j++) {
333 value = ((CLAMP (value, min, max) - min) << 8) / border;
335 *out++ = ((*in1++ * value) + (*in2++ * (256 - value))) >> 8;
336 if (!(i & 1) && !(j & 1)) {
337 *outu++ = ((*in1u++ * value) + (*in2u++ * (256 - value))) >> 8;
338 *outv++ = ((*in1v++ * value) + (*in2v++ * (256 - value))) >> 8;
345 gst_smpte_collected (GstCollectPads * pads, GstSMPTE * smpte)
349 GstBuffer *in1 = NULL, *in2 = NULL;
352 ts = smpte->position * GST_SECOND / smpte->fps;
354 for (collected = pads->data; collected; collected = g_slist_next (collected)) {
355 GstCollectData *data;
357 data = (GstCollectData *) collected->data;
359 if (data->pad == smpte->sinkpad1)
360 in1 = gst_collectpads_pop (pads, data);
361 else if (data->pad == smpte->sinkpad2)
362 in2 = gst_collectpads_pop (pads, data);
366 in1 = gst_buffer_new_and_alloc (smpte->width * smpte->height * 3);
367 fill_i420 (GST_BUFFER_DATA (in1), smpte->width, smpte->height, 7);
370 in2 = gst_buffer_new_and_alloc (smpte->width * smpte->height * 3);
371 fill_i420 (GST_BUFFER_DATA (in2), smpte->width, smpte->height, 0);
374 if (smpte->position < smpte->duration) {
375 outbuf = gst_buffer_new_and_alloc (smpte->width * smpte->height * 3);
377 /* set caps if not done yet */
378 if (!GST_PAD_CAPS (smpte->srcpad)) {
382 gst_caps_copy (gst_static_caps_get (&gst_smpte_src_template.
384 gst_caps_set_simple (caps, "width", G_TYPE_INT, smpte->width, "height",
385 G_TYPE_INT, smpte->height, "framerate", G_TYPE_DOUBLE, smpte->fps,
388 gst_pad_set_caps (smpte->srcpad, caps);
390 gst_buffer_set_caps (outbuf, GST_PAD_CAPS (smpte->srcpad));
392 gst_smpte_blend_i420 (GST_BUFFER_DATA (in1),
393 GST_BUFFER_DATA (in2),
394 GST_BUFFER_DATA (outbuf),
395 smpte->mask, smpte->width, smpte->height,
397 ((1 << smpte->depth) + smpte->border) *
398 smpte->position / smpte->duration);
402 gst_buffer_ref (in2);
408 gst_buffer_unref (in1);
410 gst_buffer_unref (in2);
412 GST_BUFFER_TIMESTAMP (outbuf) = ts;
414 return gst_pad_push (smpte->srcpad, outbuf);
418 gst_smpte_set_property (GObject * object, guint prop_id,
419 const GValue * value, GParamSpec * pspec)
423 smpte = GST_SMPTE (object);
428 gint type = g_value_get_enum (value);
430 gst_smpte_update_mask (smpte, type, smpte->depth,
431 smpte->width, smpte->height);
435 smpte->border = g_value_get_int (value);
438 smpte->fps = g_value_get_float (value);
442 gint depth = g_value_get_int (value);
444 gst_smpte_update_mask (smpte, smpte->type, depth,
445 smpte->width, smpte->height);
449 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
455 gst_smpte_get_property (GObject * object, guint prop_id,
456 GValue * value, GParamSpec * pspec)
460 smpte = GST_SMPTE (object);
465 g_value_set_enum (value, smpte->mask->type);
469 g_value_set_float (value, smpte->fps);
472 g_value_set_int (value, smpte->border);
475 g_value_set_int (value, smpte->depth);
478 G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
485 plugin_init (GstPlugin * plugin)
487 return gst_element_register (plugin, "smpte", GST_RANK_NONE, GST_TYPE_SMPTE);
490 GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
493 "Apply the standard SMPTE transitions on video images",
494 plugin_init, VERSION, "LGPL", GST_PACKAGE, GST_ORIGIN)