← Back to branch summary

~ubuntu-branches/ubuntu/natty/gst-entrans/natty

« back to all changes in this revision

Viewing changes to gst/mencoder/gstdelogo.c

  • Committer: Bazaar Package Importer
  • Author(s): Jonas Smedegaard
  • Date: 2010-09-13 19:49:29 UTC
  • Revision ID: james.westby@ubuntu.com-20100913194929-qz90a14xyxln9yfz
Tags: upstream-0.10.2
Import upstream version 0.10.2

Show diffs side-by-side

added added

removed removed

Lines of Context:
gst/mencoder/gstdelogo.c
 
1
/* GStreamer Element
 
2
 * Copyright (C) 2006 Mark Nauwelaerts <mnauw@users.sourceforge.net>
 
3
 *
 
4
 * mplayer filter:
 
5
 * Copyright (C) 2002 Jindrich Makovicka <makovick@gmail.com>
 
6
 *
 
7
 * This program is free software; you can redistribute it and/or modify
 
8
 * it under the terms of the GNU General Public License as published by
 
9
 * the Free Software Foundation; either version 2 of the License, or
 
10
 * (at your option) any later version.
 
11
 *
 
12
 * This program is distributed in the hope that it will be useful,
 
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
15
 * GNU General Public License for more details.
 
16
 *
 
17
 * You should have received a copy of the GNU General Public License
 
18
 * along with this program; if not, write to the Free Software
 
19
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02110-1307  USA
 
20
 */
 
21
 
 
22
/**
 
23
 * SECTION:element-delogo
 
24
 *
 
25
 * <refsect2>
 
26
 * <para>
 
27
 * Suppresses  a TV station logo by a simple interpolation of the surrounding
 
28
 * pixels.  Just set a rectangle covering the logo and watch it disappear
 
29
 * (and sometimes something even uglier appear - your mileage may vary).
 
30
 * </para>
 
31
 * <para>
 
32
 * If <link linkend="GstDelogo--border">border</link> is set to a negative
 
33
 * value, a green border is drawn to simplify finding the right location
 
34
 * parameters.
 
35
 * </para>
 
36
 * <title>History</title>
 
37
 * <para>
 
38
 * <itemizedlist>
 
39
 * <listitem>
 
40
 * mplayer delogo filter [Jindrich Makovicka]
 
41
 * </listitem>
 
42
 * <listitem>
 
43
 * Also available in avidemux (MPlayer delogo)
 
44
 * </listitem>
 
45
 * </itemizedlist>
 
46
 * </para>
 
47
 * </refsect2>
 
48
 *
 
49
 */
 
50
 
 
51
 
 
52
#ifdef HAVE_CONFIG_H
 
53
#include "config.h"
 
54
#endif
 
55
 
 
56
#include "plugin-mencoder.h"
 
57
 
 
58
#include <string.h>
 
59
 
 
60
 
 
61
#define GST_TYPE_DELOGO \
 
62
  (gst_delogo_get_type())
 
63
#define GST_DELOGO(obj) \
 
64
  (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_DELOGO,GstDelogo))
 
65
#define GST_DELOGO_CLASS(klass) \
 
66
  (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_DELOGO,GstDelogoClass))
 
67
#define GST_IS_DELOGO(obj) \
 
68
  (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_DELOGO))
 
69
#define GST_IS_DELOGO_CLASS(klass) \
 
70
  (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_DELOGO))
 
71
 
 
72
 
 
73
typedef struct _GstDelogo GstDelogo;
 
74
typedef struct _GstDelogoClass GstDelogoClass;
 
75
 
 
76
struct _GstDelogo
 
77
{
 
78
  GstVideoFilter videofilter;
 
79
 
 
80
  gint width, height;
 
81
 
 
82
  /* properties; external view */
 
83
  guint logo_x, logo_y, logo_w, logo_h;
 
84
  gint border;
 
85
 
 
86
  /* properties; processed */
 
87
  guint xoff, yoff, lw, lh;
 
88
  gint band, show;
 
89
};
 
90
 
 
91
 
 
92
struct _GstDelogoClass
 
93
{
 
94
  GstVideoFilterClass parent_class;
 
95
};
 
96
 
 
97
GST_DEBUG_CATEGORY_STATIC (delogo_debug);
 
98
#define GST_CAT_DEFAULT delogo_debug
 
99
 
 
100
 
 
101
/* signals and args */
 
102
enum
 
103
{
 
104
  /* FILL ME */
 
105
  LAST_SIGNAL
 
106
};
 
107
 
 
108
enum
 
109
{
 
110
  PROP_0,
 
111
  PROP_LOGO_X,
 
112
  PROP_LOGO_Y,
 
113
  PROP_LOGO_W,
 
114
  PROP_LOGO_H,
 
115
  PROP_BORDER
 
116
      /* FILL ME */
 
117
};
 
118
 
 
119
#define DEFAULT_LOGO_X       0
 
120
#define DEFAULT_LOGO_Y       0
 
121
#define DEFAULT_LOGO_W      10
 
122
#define DEFAULT_LOGO_H      10
 
123
#define DEFAULT_BORDER       0
 
124
 
 
125
 
 
126
static GstElementDetails delogo_details =
 
127
GST_ELEMENT_DETAILS ("Delogo",
 
128
    "Filter/Effect/Video",
 
129
    "Simple tv station logo remover",
 
130
    "Mark Nauwelaerts <mnauw@users.sourceforge.net>,\n"
 
131
    "Jindrich Makovicka");
 
132
 
 
133
static GstStaticPadTemplate gst_delogo_src_template =
 
134
GST_STATIC_PAD_TEMPLATE (GST_BASE_TRANSFORM_SRC_NAME,
 
135
    GST_PAD_SRC,
 
136
    GST_PAD_ALWAYS,
 
137
    GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("{ IYUV, I420, YV12 }"))
 
138
    );
 
139
 
 
140
static GstStaticPadTemplate gst_delogo_sink_template =
 
141
GST_STATIC_PAD_TEMPLATE (GST_BASE_TRANSFORM_SINK_NAME,
 
142
    GST_PAD_SINK,
 
143
    GST_PAD_ALWAYS,
 
144
    GST_STATIC_CAPS (GST_VIDEO_CAPS_YUV ("{ IYUV, I420, YV12 }"))
 
145
    );
 
146
 
 
147
static gboolean gst_delogo_set_caps (GstBaseTransform * btrans,
 
148
    GstCaps * incaps, GstCaps * outcaps);
 
149
static gboolean gst_delogo_get_unit_size (GstBaseTransform * btrans,
 
150
    GstCaps * caps, guint * size);
 
151
static GstFlowReturn gst_delogo_transform (GstBaseTransform * btrans,
 
152
    GstBuffer * in, GstBuffer * out);
 
153
static gboolean gst_delogo_start (GstBaseTransform * btrans);
 
154
static gboolean gst_delogo_stop (GstBaseTransform * btrans);
 
155
 
 
156
static void gst_delogo_set_property (GObject * object, guint prop_id,
 
157
    const GValue * value, GParamSpec * pspec);
 
158
static void gst_delogo_get_property (GObject * object, guint prop_id,
 
159
    GValue * value, GParamSpec * pspec);
 
160
 
 
161
GST_BOILERPLATE (GstDelogo, gst_delogo, GstVideoFilter, GST_TYPE_VIDEO_FILTER);
 
162
 
 
163
GST_VIDEO_FILTER_SET_CAPS_BOILERPLATE (GstDelogo, gst_delogo);
 
164
 
 
165
GST_VIDEO_FILTER_GET_UNIT_SIZE_BOILERPLATE (gst_delogo);
 
166
 
 
167
static void
 
168
gst_delogo_base_init (gpointer g_class)
 
169
{
 
170
  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
 
171
 
 
172
  gst_element_class_set_details (element_class, &delogo_details);
 
173
 
 
174
  gst_element_class_add_pad_template (element_class,
 
175
      gst_static_pad_template_get (&gst_delogo_sink_template));
 
176
  gst_element_class_add_pad_template (element_class,
 
177
      gst_static_pad_template_get (&gst_delogo_src_template));
 
178
}
 
179
 
 
180
static void
 
181
gst_delogo_class_init (GstDelogoClass * g_class)
 
182
{
 
183
  GObjectClass *gobject_class;
 
184
  GstBaseTransformClass *trans_class;
 
185
 
 
186
  gobject_class = G_OBJECT_CLASS (g_class);
 
187
  trans_class = GST_BASE_TRANSFORM_CLASS (g_class);
 
188
 
 
189
  GST_DEBUG_CATEGORY_INIT (delogo_debug, "delogo", 0, "delogo");
 
190
 
 
191
  gobject_class->set_property = gst_delogo_set_property;
 
192
  gobject_class->get_property = gst_delogo_get_property;
 
193
 
 
194
  g_object_class_install_property (gobject_class, PROP_LOGO_X,
 
195
      g_param_spec_uint ("logo-x", "Logo X",
 
196
          "X-coordinate of logo top left corner",
 
197
          0, G_MAXUINT, DEFAULT_LOGO_X,
 
198
          G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE));
 
199
 
 
200
  g_object_class_install_property (gobject_class, PROP_LOGO_Y,
 
201
      g_param_spec_uint ("logo-y", "Logo Y",
 
202
          "Y-coordinate of logo top left corner",
 
203
          0, G_MAXUINT, DEFAULT_LOGO_Y,
 
204
          G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE));
 
205
 
 
206
  g_object_class_install_property (gobject_class, PROP_LOGO_W,
 
207
      g_param_spec_uint ("logo-w", "Logo Width",
 
208
          "Width of logo",
 
209
          0, G_MAXUINT, DEFAULT_LOGO_W,
 
210
          G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE));
 
211
 
 
212
  g_object_class_install_property (gobject_class, PROP_LOGO_H,
 
213
      g_param_spec_uint ("logo-h", "Logo Width",
 
214
          "Height of logo",
 
215
          0, G_MAXUINT, DEFAULT_LOGO_H,
 
216
          G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE));
 
217
 
 
218
  g_object_class_install_property (gobject_class, PROP_BORDER,
 
219
      g_param_spec_int ("border", "Border",
 
220
          "Thickness of fuzzy edge of rectangle (<0 = draw border)",
 
221
          -1, G_MAXINT, DEFAULT_BORDER,
 
222
          G_PARAM_READWRITE | GST_PARAM_CONTROLLABLE));
 
223
 
 
224
  trans_class->set_caps = GST_DEBUG_FUNCPTR (gst_delogo_set_caps);
 
225
  trans_class->get_unit_size = GST_DEBUG_FUNCPTR (gst_delogo_get_unit_size);
 
226
  trans_class->transform = GST_DEBUG_FUNCPTR (gst_delogo_transform);
 
227
  trans_class->start = GST_DEBUG_FUNCPTR (gst_delogo_start);
 
228
  trans_class->stop = GST_DEBUG_FUNCPTR (gst_delogo_stop);
 
229
}
 
230
 
 
231
static void
 
232
gst_delogo_init (GstDelogo * filter, GstDelogoClass * g_class)
 
233
{
 
234
  filter->logo_x = DEFAULT_LOGO_X;
 
235
  filter->logo_y = DEFAULT_LOGO_Y;
 
236
  filter->logo_w = DEFAULT_LOGO_W;
 
237
  filter->logo_h = DEFAULT_LOGO_H;
 
238
  filter->border = DEFAULT_BORDER;
 
239
}
 
240
 
 
241
static void
 
242
gst_delogo_delogo(guint8 * dst, guint8 * src, gint dstStride, gint srcStride,
 
243
    gint width, gint height, gint logo_x, gint logo_y, gint logo_w, gint logo_h,
 
244
    gint band, gint show, gint direct)
 
245
{
 
246
  gint y, x;
 
247
  gint interp, dist;
 
248
  guint8 *xdst, *xsrc;
 
249
 
 
250
  guint8 *topleft, *botleft, *topright;
 
251
  gint xclipl, xclipr, yclipt, yclipb;
 
252
  gint logo_x1, logo_x2, logo_y1, logo_y2;
 
253
 
 
254
  xclipl = MAX (-logo_x, 0);
 
255
  xclipr = MAX (logo_x + logo_w - width, 0);
 
256
  yclipt = MAX (-logo_y, 0);
 
257
  yclipb = MAX (logo_y + logo_h - height, 0);
 
258
 
 
259
  logo_x1 = logo_x + xclipl;
 
260
  logo_x2 = logo_x + logo_w - xclipr;
 
261
  logo_y1 = logo_y + yclipt;
 
262
  logo_y2 = logo_y + logo_h - yclipb;
 
263
 
 
264
  topleft = src + logo_y1 * srcStride + logo_x1;
 
265
  topright = src + logo_y1 * srcStride + logo_x2 - 1;
 
266
  botleft = src + (logo_y2 - 1) * srcStride + logo_x1;
 
267
 
 
268
  if (!direct)
 
269
    oil_memcpy (dst, src, width * height);
 
270
 
 
271
  dst += (logo_y1 + 1) * dstStride;
 
272
  src += (logo_y1 + 1) * srcStride;
 
273
 
 
274
  for (y = logo_y1 + 1; y < logo_y2 - 1; y++) {
 
275
    for (x = logo_x1 + 1, xdst = dst + logo_x1 + 1, xsrc = src + logo_x1 + 1;
 
276
         x < logo_x2 - 1; x++, xdst++, xsrc++) {
 
277
      interp = ((topleft[srcStride * (y - logo_y - yclipt)]
 
278
          + topleft[srcStride *(y - logo_y-1 - yclipt)]
 
279
          + topleft[srcStride *(y - logo_y+1 - yclipt)])
 
280
            * (logo_w - (x - logo_x)) / logo_w
 
281
          + (topright[srcStride *(y - logo_y - yclipt)]
 
282
          + topright[srcStride *(y - logo_y -1 - yclipt)]
 
283
          + topright[srcStride *(y - logo_y +1 - yclipt)])
 
284
            * (x - logo_x) / logo_w
 
285
          + (topleft[x -logo_x - xclipl]
 
286
          + topleft[x - logo_x - 1 - xclipl]
 
287
          + topleft[x - logo_x + 1 - xclipl])
 
288
            * (logo_h - (y - logo_y)) / logo_h
 
289
          + (botleft[x - logo_x -xclipl]
 
290
          + botleft[x - logo_x - 1 - xclipl]
 
291
          + botleft[x - logo_x + 1- xclipl])
 
292
            * (y - logo_y) / logo_h
 
293
               )/6;
 
294
/*              interp = (topleft[srcStride*(y-logo_y)]*(logo_w-(x-logo_x))/logo_w
 
295
      + topright[srcStride*(y-logo_y)]*(x-logo_x)/logo_w
 
296
      + topleft[x-logo_x]*(logo_h-(y-logo_y))/logo_h
 
297
      + botleft[x-logo_x]*(y-logo_y)/logo_h
 
298
                          )/2;*/
 
299
      if (y >= logo_y + band && y < logo_y + logo_h - band &&
 
300
          x >= logo_x+band && x < logo_x + logo_w - band) {
 
301
        *xdst = interp;
 
302
      } else {
 
303
        dist = 0;
 
304
        if (x < logo_x + band)
 
305
          dist = MAX (dist, logo_x - x + band);
 
306
        else if (x >= logo_x + logo_w-band)
 
307
          dist = MAX (dist, x -(logo_x + logo_w - 1 - band));
 
308
        if (y < logo_y + band)
 
309
          dist = MAX (dist, logo_y - y + band);
 
310
        else if (y >= logo_y + logo_h - band)
 
311
          dist = MAX (dist, y - (logo_y + logo_h - 1 - band));
 
312
        *xdst = (*xsrc * dist + interp * (band - dist)) / band;
 
313
 
 
314
        if (show && (dist == band - 1))
 
315
          *xdst = 0;
 
316
      }
 
317
    }
 
318
 
 
319
    dst+= dstStride;
 
320
    src+= srcStride;
 
321
  }
 
322
}
 
323
 
 
324
static void
 
325
gst_delogo_set_props (GstDelogo * filter)
 
326
{
 
327
  filter->show = 0;
 
328
  filter->band = filter->border;
 
329
 
 
330
  if (filter->band < 0) {
 
331
    filter->band = 4;
 
332
    filter->show = 1;
 
333
  }
 
334
 
 
335
  filter->lw = filter->logo_w + filter->band * 2;
 
336
  filter->lh = filter->logo_h + filter->band * 2;
 
337
  filter->xoff = filter->logo_x - filter->band;
 
338
  filter->yoff = filter->logo_y - filter->band;
 
339
}
 
340
 
 
341
static GstFlowReturn
 
342
gst_delogo_transform (GstBaseTransform * btrans, GstBuffer * in,
 
343
    GstBuffer * out)
 
344
{
 
345
  GstDelogo *filter;
 
346
  guint8 *src, *dest;
 
347
  guint width, height, stride;
 
348
 
 
349
  gst_object_sync_values (G_OBJECT (btrans), GST_BUFFER_TIMESTAMP (in));
 
350
 
 
351
  filter = GST_DELOGO (btrans);
 
352
 
 
353
  width = filter->width;
 
354
  height = filter->height;
 
355
 
 
356
  /* luma */
 
357
  src = (guint8 *) GST_BUFFER_DATA (in);
 
358
  dest = (guint8 *) GST_BUFFER_DATA (out);
 
359
  stride = GST_VIDEO_I420_Y_ROWSTRIDE (width);
 
360
 
 
361
  gst_delogo_delogo (dest, src, stride, stride, stride, height,
 
362
      filter->xoff, filter->yoff, filter->lw, filter->lh,
 
363
      filter->band, filter->show, FALSE);
 
364
 
 
365
  /* chroma */
 
366
  src += GST_VIDEO_I420_U_OFFSET (width, height);
 
367
  dest += GST_VIDEO_I420_U_OFFSET (width, height);
 
368
  stride = GST_VIDEO_I420_U_ROWSTRIDE (width);
 
369
 
 
370
  gst_delogo_delogo (dest, src, stride, stride, stride, height / 2,
 
371
      filter->xoff / 2, filter->yoff / 2, filter->lw / 2, filter->lh / 2,
 
372
      filter->band / 2, filter->show, FALSE);
 
373
 
 
374
  src = (guint8 *) GST_BUFFER_DATA (in) + GST_VIDEO_I420_V_OFFSET (width, height);
 
375
  dest = (guint8 *) GST_BUFFER_DATA (out) + GST_VIDEO_I420_V_OFFSET (width, height);
 
376
 
 
377
  gst_delogo_delogo (dest, src, stride, stride, stride, height / 2,
 
378
      filter->xoff / 2, filter->yoff / 2, filter->lw / 2, filter->lh / 2,
 
379
      filter->band / 2, filter->show, FALSE);
 
380
 
 
381
  return GST_FLOW_OK;
 
382
}
 
383
 
 
384
 
 
385
static gboolean
 
386
gst_delogo_start (GstBaseTransform * btrans)
 
387
{
 
388
  return TRUE;
 
389
}
 
390
 
 
391
static gboolean
 
392
gst_delogo_stop (GstBaseTransform * btrans)
 
393
{
 
394
  return TRUE;
 
395
}
 
396
 
 
397
static void
 
398
gst_delogo_set_property (GObject * object, guint prop_id,
 
399
    const GValue * value, GParamSpec * pspec)
 
400
{
 
401
  GstDelogo *src;
 
402
 
 
403
  g_return_if_fail (GST_IS_DELOGO (object));
 
404
  src = GST_DELOGO (object);
 
405
 
 
406
  switch (prop_id) {
 
407
    case PROP_LOGO_X:
 
408
      src->logo_x = g_value_get_uint (value);
 
409
      break;
 
410
    case PROP_LOGO_Y:
 
411
      src->logo_y = g_value_get_uint (value);
 
412
      break;
 
413
    case PROP_LOGO_W:
 
414
      src->logo_w = g_value_get_uint (value);
 
415
      break;
 
416
    case PROP_LOGO_H:
 
417
      src->logo_h = g_value_get_uint (value);
 
418
      break;
 
419
    case PROP_BORDER:
 
420
      src->border = g_value_get_int (value);
 
421
      break;
 
422
    default:
 
423
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 
424
      break;
 
425
  }
 
426
 
 
427
  gst_delogo_set_props (src);
 
428
}
 
429
 
 
430
static void
 
431
gst_delogo_get_property (GObject * object, guint prop_id, GValue * value,
 
432
    GParamSpec * pspec)
 
433
{
 
434
  GstDelogo *src;
 
435
 
 
436
  g_return_if_fail (GST_IS_DELOGO (object));
 
437
  src = GST_DELOGO (object);
 
438
 
 
439
  switch (prop_id) {
 
440
    case PROP_LOGO_X:
 
441
      g_value_set_uint (value, src->logo_x);
 
442
      break;
 
443
    case PROP_LOGO_Y:
 
444
      g_value_set_uint (value, src->logo_y);
 
445
      break;
 
446
    case PROP_LOGO_W:
 
447
      g_value_set_uint (value, src->logo_w);
 
448
      break;
 
449
    case PROP_LOGO_H:
 
450
      g_value_set_uint (value, src->logo_h);
 
451
      break;
 
452
    case PROP_BORDER:
 
453
      g_value_set_int (value, src->border);
 
454
      break;
 
455
    default:
 
456
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 
457
      break;
 
458
  }
 
459
}