gsttextrender.c 22.3 KB
Newer Older
1 2 3
/* GStreamer
 * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
 * Copyright (C) <2003> David Schleef <ds@schleef.org>
4
 * Copyright (C) <2009> Young-Ho Cha <ganadist@gmail.com>
5 6 7 8 9 10 11 12 13 14 15 16 17
 *
 * 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
Tim-Philipp Müller's avatar
Tim-Philipp Müller committed
18 19
 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
 * Boston, MA 02110-1301, USA.
20 21
 */

22 23 24

/**
 * SECTION:element-textrender
25
 * @title: textrender
26 27 28 29 30
 * @see_also: #GstTextOverlay
 *
 * This plugin renders text received on the text sink pad to a video
 * buffer (retaining the alpha channel), so it can later be overlayed
 * on top of video streams using other elements.
31
 *
32
 * The text can contain newline characters. (FIXME: What about text
33
 * wrapping? It does not make sense in this context)
34
 *
35
 * ## Example launch lines
36
 * |[
37
 * gst-launch-1.0 -v filesrc location=subtitles.srt ! subparse ! textrender ! videoconvert ! autovideosink
38
 * ]|
39
 *
40 41
 */

42 43 44 45 46 47 48 49
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <gst/gst.h>
#include <gst/video/video.h>

#include "gsttextrender.h"
50
#include <string.h>
51

52 53 54 55 56 57 58 59 60 61 62 63
#if G_BYTE_ORDER == G_LITTLE_ENDIAN
# define CAIRO_ARGB_A 3
# define CAIRO_ARGB_R 2
# define CAIRO_ARGB_G 1
# define CAIRO_ARGB_B 0
#else
# define CAIRO_ARGB_A 0
# define CAIRO_ARGB_R 1
# define CAIRO_ARGB_G 2
# define CAIRO_ARGB_B 3
#endif

64 65 66
GST_DEBUG_CATEGORY_EXTERN (pango_debug);
#define GST_CAT_DEFAULT pango_debug

67 68
#define MINIMUM_OUTLINE_OFFSET 1.0

69 70 71 72 73 74 75 76 77
#define DEFAULT_PROP_VALIGNMENT GST_TEXT_RENDER_VALIGN_BASELINE
#define DEFAULT_PROP_HALIGNMENT GST_TEXT_RENDER_HALIGN_CENTER
#define DEFAULT_PROP_LINE_ALIGNMENT GST_TEXT_RENDER_LINE_ALIGN_CENTER
#define DEFAULT_PROP_XPAD       25
#define DEFAULT_PROP_YPAD       25

#define DEFAULT_RENDER_WIDTH 720
#define DEFAULT_RENDER_HEIGHT 576

78 79
enum
{
80
  PROP_0,
81 82 83 84 85
  PROP_HALIGNMENT,
  PROP_VALIGNMENT,
  PROP_LINE_ALIGNMENT,
  PROP_XPAD,
  PROP_YPAD,
86
  PROP_FONT_DESC
87 88
};

89
#define VIDEO_FORMATS "{ AYUV, ARGB } "
90 91

static GstStaticPadTemplate src_template_factory =
92
GST_STATIC_PAD_TEMPLATE ("src",
93 94
    GST_PAD_SRC,
    GST_PAD_ALWAYS,
95
    GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE (VIDEO_FORMATS))
96 97 98
    );

static GstStaticPadTemplate sink_template_factory =
99
GST_STATIC_PAD_TEMPLATE ("sink",
100 101
    GST_PAD_SINK,
    GST_PAD_ALWAYS,
102
    GST_STATIC_CAPS ("text/x-raw, format = { pango-markup, utf8 }")
103 104
    );

105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
#define GST_TYPE_TEXT_RENDER_VALIGN (gst_text_render_valign_get_type())
static GType
gst_text_render_valign_get_type (void)
{
  static GType text_render_valign_type = 0;
  static const GEnumValue text_render_valign[] = {
    {GST_TEXT_RENDER_VALIGN_BASELINE, "baseline", "baseline"},
    {GST_TEXT_RENDER_VALIGN_BOTTOM, "bottom", "bottom"},
    {GST_TEXT_RENDER_VALIGN_TOP, "top", "top"},
    {0, NULL, NULL},
  };

  if (!text_render_valign_type) {
    text_render_valign_type =
        g_enum_register_static ("GstTextRenderVAlign", text_render_valign);
  }
  return text_render_valign_type;
}

#define GST_TYPE_TEXT_RENDER_HALIGN (gst_text_render_halign_get_type())
static GType
gst_text_render_halign_get_type (void)
{
  static GType text_render_halign_type = 0;
  static const GEnumValue text_render_halign[] = {
    {GST_TEXT_RENDER_HALIGN_LEFT, "left", "left"},
    {GST_TEXT_RENDER_HALIGN_CENTER, "center", "center"},
    {GST_TEXT_RENDER_HALIGN_RIGHT, "right", "right"},
    {0, NULL, NULL},
  };

  if (!text_render_halign_type) {
    text_render_halign_type =
        g_enum_register_static ("GstTextRenderHAlign", text_render_halign);
  }
  return text_render_halign_type;
}

#define GST_TYPE_TEXT_RENDER_LINE_ALIGN (gst_text_render_line_align_get_type())
static GType
gst_text_render_line_align_get_type (void)
{
  static GType text_render_line_align_type = 0;
  static const GEnumValue text_render_line_align[] = {
    {GST_TEXT_RENDER_LINE_ALIGN_LEFT, "left", "left"},
    {GST_TEXT_RENDER_LINE_ALIGN_CENTER, "center", "center"},
    {GST_TEXT_RENDER_LINE_ALIGN_RIGHT, "right", "right"},
    {0, NULL, NULL}
  };

  if (!text_render_line_align_type) {
    text_render_line_align_type =
        g_enum_register_static ("GstTextRenderLineAlign",
        text_render_line_align);
  }
  return text_render_line_align_type;
}

163 164 165
static void gst_text_render_adjust_values_with_fontdesc (GstTextRender *
    render, PangoFontDescription * desc);

166 167
#define gst_text_render_parent_class parent_class
G_DEFINE_TYPE (GstTextRender, gst_text_render, GST_TYPE_ELEMENT);
168

169 170
static void gst_text_render_finalize (GObject * object);
static void gst_text_render_set_property (GObject * object,
171
    guint prop_id, const GValue * value, GParamSpec * pspec);
172 173
static void gst_text_render_get_property (GObject * object,
    guint prop_id, GValue * value, GParamSpec * pspec);
174 175 176 177 178

static void
gst_text_render_class_init (GstTextRenderClass * klass)
{
  GObjectClass *gobject_class;
179
  GstElementClass *gstelement_class;
180
  PangoFontMap *fontmap;
181 182

  gobject_class = (GObjectClass *) klass;
183
  gstelement_class = (GstElementClass *) klass;
184 185 186 187 188

  parent_class = g_type_class_peek_parent (klass);

  gobject_class->finalize = gst_text_render_finalize;
  gobject_class->set_property = gst_text_render_set_property;
189
  gobject_class->get_property = gst_text_render_get_property;
190

191 192 193 194
  gst_element_class_add_static_pad_template (gstelement_class,
      &src_template_factory);
  gst_element_class_add_static_pad_template (gstelement_class,
      &sink_template_factory);
195

196
  gst_element_class_set_static_metadata (gstelement_class, "Text renderer",
197 198 199
      "Filter/Editor/Video",
      "Renders a text string to an image bitmap",
      "David Schleef <ds@schleef.org>, "
200
      "GStreamer maintainers <gstreamer-devel@lists.freedesktop.org>");
201

202 203
  fontmap = pango_cairo_font_map_get_default ();
  klass->pango_context =
204
      pango_font_map_create_context (PANGO_FONT_MAP (fontmap));
205
  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_FONT_DESC,
206 207 208 209 210
      g_param_spec_string ("font-desc", "font description",
          "Pango font description of font "
          "to be used for rendering. "
          "See documentation of "
          "pango_font_description_from_string"
211
          " for syntax.", "", G_PARAM_WRITABLE | G_PARAM_STATIC_STRINGS));
212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232
  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_VALIGNMENT,
      g_param_spec_enum ("valignment", "vertical alignment",
          "Vertical alignment of the text", GST_TYPE_TEXT_RENDER_VALIGN,
          DEFAULT_PROP_VALIGNMENT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_HALIGNMENT,
      g_param_spec_enum ("halignment", "horizontal alignment",
          "Horizontal alignment of the text", GST_TYPE_TEXT_RENDER_HALIGN,
          DEFAULT_PROP_HALIGNMENT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_XPAD,
      g_param_spec_int ("xpad", "horizontal paddding",
          "Horizontal paddding when using left/right alignment", 0, G_MAXINT,
          DEFAULT_PROP_XPAD, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_YPAD,
      g_param_spec_int ("ypad", "vertical padding",
          "Vertical padding when using top/bottom alignment", 0, G_MAXINT,
          DEFAULT_PROP_YPAD, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
  g_object_class_install_property (G_OBJECT_CLASS (klass), PROP_LINE_ALIGNMENT,
      g_param_spec_enum ("line-alignment", "line alignment",
          "Alignment of text lines relative to each other.",
          GST_TYPE_TEXT_RENDER_LINE_ALIGN, DEFAULT_PROP_LINE_ALIGNMENT,
          G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
233 234 235
}

static void
236 237
gst_text_render_adjust_values_with_fontdesc (GstTextRender * render,
    PangoFontDescription * desc)
238
{
239 240 241 242 243 244
  gint font_size = pango_font_description_get_size (desc) / PANGO_SCALE;

  render->shadow_offset = (double) (font_size) / 13.0;
  render->outline_offset = (double) (font_size) / 15.0;
  if (render->outline_offset < MINIMUM_OUTLINE_OFFSET)
    render->outline_offset = MINIMUM_OUTLINE_OFFSET;
245 246 247
}

static void
248
gst_text_render_render_pangocairo (GstTextRender * render)
249
{
250 251 252 253
  cairo_t *cr;
  cairo_surface_t *surface;
  cairo_t *cr_shadow;
  cairo_surface_t *surface_shadow;
254
  PangoRectangle ink_rect, logical_rect;
255
  gint width, height;
256 257

  pango_layout_get_pixel_extents (render->layout, &ink_rect, &logical_rect);
258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278

  width = logical_rect.width + render->shadow_offset;
  height = logical_rect.height + logical_rect.y + render->shadow_offset;

  surface_shadow = cairo_image_surface_create (CAIRO_FORMAT_A8, width, height);
  cr_shadow = cairo_create (surface_shadow);

  /* clear shadow surface */
  cairo_set_operator (cr_shadow, CAIRO_OPERATOR_CLEAR);
  cairo_paint (cr_shadow);
  cairo_set_operator (cr_shadow, CAIRO_OPERATOR_OVER);

  /* draw shadow text */
  cairo_save (cr_shadow);
  cairo_set_source_rgba (cr_shadow, 0.0, 0.0, 0.0, 0.5);
  cairo_translate (cr_shadow, render->shadow_offset, render->shadow_offset);
  pango_cairo_show_layout (cr_shadow, render->layout);
  cairo_restore (cr_shadow);

  /* draw outline text */
  cairo_save (cr_shadow);
279
  cairo_set_source_rgb (cr_shadow, 0.0, 0.0, 0.0);
280 281 282 283 284 285 286 287 288 289 290 291
  cairo_set_line_width (cr_shadow, render->outline_offset);
  pango_cairo_layout_path (cr_shadow, render->layout);
  cairo_stroke (cr_shadow);
  cairo_restore (cr_shadow);

  cairo_destroy (cr_shadow);

  render->text_image = g_realloc (render->text_image, 4 * width * height);

  surface = cairo_image_surface_create_for_data (render->text_image,
      CAIRO_FORMAT_ARGB32, width, height, width * 4);
  cr = cairo_create (surface);
292 293 294
  cairo_set_operator (cr, CAIRO_OPERATOR_CLEAR);
  cairo_paint (cr);
  cairo_set_operator (cr, CAIRO_OPERATOR_OVER);
295 296 297 298

  /* set default color */
  cairo_set_source_rgb (cr, 1.0, 1.0, 1.0);

299
  cairo_save (cr);
300 301
  /* draw text */
  pango_cairo_show_layout (cr, render->layout);
302
  cairo_restore (cr);
303 304 305 306 307 308 309 310 311 312 313

  /* composite shadow with offset */
  cairo_set_operator (cr, CAIRO_OPERATOR_DEST_OVER);
  cairo_set_source_surface (cr, surface_shadow, 0.0, 0.0);
  cairo_paint (cr);

  cairo_destroy (cr);
  cairo_surface_destroy (surface_shadow);
  cairo_surface_destroy (surface);
  render->image_width = width;
  render->image_height = height;
314 315
}

316 317 318 319 320 321 322 323 324 325 326 327
static void
gst_text_render_check_argb (GstTextRender * render)
{
  GstCaps *peer_caps;
  peer_caps = gst_pad_get_allowed_caps (render->srcpad);
  if (G_LIKELY (peer_caps)) {
    guint i = 0, n = 0;

    n = gst_caps_get_size (peer_caps);
    GST_DEBUG_OBJECT (render, "peer allowed caps (%u structure(s)) are %"
        GST_PTR_FORMAT, n, peer_caps);

328
    /* Check if AYUV or ARGB is first */
329
    for (i = 0; i < n; i++) {
330 331
      GstStructure *s;
      GstVideoFormat vformat;
332
      const GstVideoFormatInfo *info;
333 334 335 336 337 338 339 340 341 342 343
      const gchar *fmt;

      s = gst_caps_get_structure (peer_caps, i);
      if (!gst_structure_has_name (s, "video/x-raw"))
        continue;

      fmt = gst_structure_get_string (s, "format");
      if (fmt == NULL)
        continue;

      vformat = gst_video_format_from_string (fmt);
344 345 346 347 348
      info = gst_video_format_get_info (vformat);
      if (info == NULL)
        continue;

      render->use_ARGB = GST_VIDEO_FORMAT_INFO_HAS_ALPHA (info);
349 350 351 352 353
    }
    gst_caps_unref (peer_caps);
  }
}

354
static gboolean
Wim Taymans's avatar
Wim Taymans committed
355
gst_text_render_src_setcaps (GstTextRender * render, GstCaps * caps)
356 357
{
  GstStructure *structure;
Wim Taymans's avatar
Wim Taymans committed
358
  gboolean ret;
359 360 361 362 363 364
  gint width = 0, height = 0;

  structure = gst_caps_get_structure (caps, 0);
  gst_structure_get_int (structure, "width", &width);
  gst_structure_get_int (structure, "height", &height);

Wim Taymans's avatar
Wim Taymans committed
365
  GST_DEBUG_OBJECT (render, "Got caps %" GST_PTR_FORMAT, caps);
366

367
  if (width >= render->image_width && height >= render->image_height) {
368 369 370 371
    render->width = width;
    render->height = height;
  }

372 373
  gst_text_render_check_argb (render);

374
  ret = gst_pad_set_caps (render->srcpad, caps);
Wim Taymans's avatar
Wim Taymans committed
375

376 377 378
  return ret;
}

Wim Taymans's avatar
Wim Taymans committed
379
static GstCaps *
380
gst_text_render_fixate_caps (GstTextRender * render, GstCaps * caps)
381
{
382 383
  GstStructure *s;

Wim Taymans's avatar
Wim Taymans committed
384 385 386
  caps = gst_caps_truncate (caps);

  caps = gst_caps_make_writable (caps);
387
  s = gst_caps_get_structure (caps, 0);
388 389

  GST_DEBUG ("Fixating caps %" GST_PTR_FORMAT, caps);
390 391 392 393
  gst_structure_fixate_field_nearest_int (s, "width", MAX (render->image_width,
          DEFAULT_RENDER_WIDTH));
  gst_structure_fixate_field_nearest_int (s, "height",
      MAX (render->image_height + render->ypad, DEFAULT_RENDER_HEIGHT));
394
  caps = gst_caps_fixate (caps);
395
  GST_DEBUG ("Fixated to    %" GST_PTR_FORMAT, caps);
Wim Taymans's avatar
Wim Taymans committed
396 397

  return caps;
398 399
}

400 401 402 403 404 405
#define CAIRO_UNPREMULTIPLY(a,r,g,b) G_STMT_START { \
  b = (a > 0) ? MIN ((b * 255 + a / 2) / a, 255) : 0; \
  g = (a > 0) ? MIN ((g * 255 + a / 2) / a, 255) : 0; \
  r = (a > 0) ? MIN ((r * 255 + a / 2) / a, 255) : 0; \
} G_STMT_END

406
static void
407 408
gst_text_renderer_image_to_ayuv (GstTextRender * render, guchar * pixbuf,
    int xpos, int ypos, int stride)
409 410 411
{
  int y;                        /* text bitmap coordinates */
  guchar *p, *bitp;
412 413
  guchar a, r, g, b;
  int width, height;
414

415 416
  width = render->image_width;
  height = render->image_height;
417

418
  for (y = 0; y < height && ypos + y < render->height; y++) {
419
    int n;
420
    p = pixbuf + (ypos + y) * stride + xpos * 4;
421 422
    bitp = render->text_image + y * width * 4;
    for (n = 0; n < width && n < render->width; n++) {
423 424 425 426 427
      b = bitp[CAIRO_ARGB_B];
      g = bitp[CAIRO_ARGB_G];
      r = bitp[CAIRO_ARGB_R];
      a = bitp[CAIRO_ARGB_A];
      bitp += 4;
428

429 430 431
      /* Cairo uses pre-multiplied ARGB, unpremultiply it */
      CAIRO_UNPREMULTIPLY (a, r, g, b);

432 433 434 435 436 437 438
      *p++ = a;
      *p++ = CLAMP ((int) (((19595 * r) >> 16) + ((38470 * g) >> 16) +
              ((7471 * b) >> 16)), 0, 255);
      *p++ = CLAMP ((int) (-((11059 * r) >> 16) - ((21709 * g) >> 16) +
              ((32768 * b) >> 16) + 128), 0, 255);
      *p++ = CLAMP ((int) (((32768 * r) >> 16) - ((27439 * g) >> 16) -
              ((5329 * b) >> 16) + 128), 0, 255);
439 440 441 442
    }
  }
}

443
static void
444 445
gst_text_renderer_image_to_argb (GstTextRender * render, guchar * pixbuf,
    int xpos, int ypos, int stride)
446
{
447
  int i, j;
448
  guchar *p, *bitp;
449
  int width, height;
450

451 452
  width = render->image_width;
  height = render->image_height;
453

454
  for (i = 0; i < height && ypos + i < render->height; i++) {
455
    p = pixbuf + (ypos + i) * stride + xpos * 4;
456 457
    bitp = render->text_image + i * width * 4;
    for (j = 0; j < width && j < render->width; j++) {
458 459 460 461
      p[0] = bitp[CAIRO_ARGB_A];
      p[1] = bitp[CAIRO_ARGB_R];
      p[2] = bitp[CAIRO_ARGB_G];
      p[3] = bitp[CAIRO_ARGB_B];
462

463 464 465
      /* Cairo uses pre-multiplied ARGB, unpremultiply it */
      CAIRO_UNPREMULTIPLY (p[0], p[1], p[2], p[3]);

466 467 468
      bitp += 4;
      p += 4;
    }
469 470
  }
}
471 472

static GstFlowReturn
Wim Taymans's avatar
Wim Taymans committed
473
gst_text_render_chain (GstPad * pad, GstObject * parent, GstBuffer * inbuf)
474 475 476 477
{
  GstTextRender *render;
  GstFlowReturn ret;
  GstBuffer *outbuf;
478
  GstCaps *caps = NULL, *padcaps;
Wim Taymans's avatar
Wim Taymans committed
479
  GstMapInfo map;
480 481
  guint8 *data;
  gsize size;
482
  gint n;
483
  gint xpos, ypos;
484

Wim Taymans's avatar
Wim Taymans committed
485
  render = GST_TEXT_RENDER (parent);
486

Wim Taymans's avatar
Wim Taymans committed
487 488 489
  gst_buffer_map (inbuf, &map, GST_MAP_READ);
  data = map.data;
  size = map.size;
490

491 492 493 494 495 496 497 498
  /* somehow pango barfs over "\0" buffers... */
  while (size > 0 &&
      (data[size - 1] == '\r' ||
          data[size - 1] == '\n' || data[size - 1] == '\0')) {
    size--;
  }

  /* render text */
Josep Torra's avatar
Josep Torra committed
499
  GST_DEBUG ("rendering '%*s'", (gint) size, data);
500
  pango_layout_set_markup (render->layout, (gchar *) data, size);
501
  gst_text_render_render_pangocairo (render);
Wim Taymans's avatar
Wim Taymans committed
502
  gst_buffer_unmap (inbuf, &map);
503

504
  gst_text_render_check_argb (render);
505

506
  padcaps = gst_pad_query_caps (render->srcpad, NULL);
507
  caps = gst_pad_peer_query_caps (render->srcpad, padcaps);
508 509 510 511 512 513
  gst_caps_unref (padcaps);

  if (!caps || gst_caps_is_empty (caps)) {
    GST_ELEMENT_ERROR (render, CORE, NEGOTIATION, (NULL), (NULL));
    ret = GST_FLOW_ERROR;
    goto done;
514
  }
515

Wim Taymans's avatar
Wim Taymans committed
516
  caps = gst_text_render_fixate_caps (render, caps);
517

Wim Taymans's avatar
Wim Taymans committed
518
  if (!gst_text_render_src_setcaps (render, caps)) {
519 520 521 522 523
    GST_ELEMENT_ERROR (render, CORE, NEGOTIATION, (NULL), (NULL));
    ret = GST_FLOW_ERROR;
    goto done;
  }

524 525 526 527 528
  if (render->segment_event) {
    gst_pad_push_event (render->srcpad, render->segment_event);
    render->segment_event = NULL;
  }

529
  GST_DEBUG ("Allocating buffer WxH = %dx%d", render->width, render->height);
Wim Taymans's avatar
Wim Taymans committed
530
  outbuf = gst_buffer_new_and_alloc (render->width * render->height * 4);
531

532
  gst_buffer_copy_into (outbuf, inbuf, GST_BUFFER_COPY_TIMESTAMPS, 0, -1);
Wim Taymans's avatar
Wim Taymans committed
533 534 535 536

  gst_buffer_map (outbuf, &map, GST_MAP_WRITE);
  data = map.data;
  size = map.size;
537

538
  if (render->use_ARGB) {
539
    memset (data, 0, render->width * render->height * 4);
540 541 542 543 544
  } else {
    for (n = 0; n < render->width * render->height; n++) {
      data[n * 4] = data[n * 4 + 1] = 0;
      data[n * 4 + 2] = data[n * 4 + 3] = 128;
    }
545 546
  }

547 548 549 550 551
  switch (render->halign) {
    case GST_TEXT_RENDER_HALIGN_LEFT:
      xpos = render->xpad;
      break;
    case GST_TEXT_RENDER_HALIGN_CENTER:
552
      xpos = (render->width - render->image_width) / 2;
553 554
      break;
    case GST_TEXT_RENDER_HALIGN_RIGHT:
555
      xpos = render->width - render->image_width - render->xpad;
556 557 558 559 560 561 562
      break;
    default:
      xpos = 0;
  }

  switch (render->valign) {
    case GST_TEXT_RENDER_VALIGN_BOTTOM:
563
      ypos = render->height - render->image_height - render->ypad;
564 565
      break;
    case GST_TEXT_RENDER_VALIGN_BASELINE:
566
      ypos = render->height - (render->image_height + render->ypad);
567 568 569 570 571 572 573 574 575
      break;
    case GST_TEXT_RENDER_VALIGN_TOP:
      ypos = render->ypad;
      break;
    default:
      ypos = render->ypad;
      break;
  }

576
  if (render->text_image) {
577
    if (render->use_ARGB) {
578 579
      gst_text_renderer_image_to_argb (render, data, xpos, ypos,
          render->width * 4);
580
    } else {
581 582
      gst_text_renderer_image_to_ayuv (render, data, xpos, ypos,
          render->width * 4);
583
    }
584
  }
Wim Taymans's avatar
Wim Taymans committed
585
  gst_buffer_unmap (outbuf, &map);
586 587 588 589 590 591 592

  ret = gst_pad_push (render->srcpad, outbuf);

done:
  if (caps)
    gst_caps_unref (caps);
  gst_buffer_unref (inbuf);
Wim Taymans's avatar
Wim Taymans committed
593

594 595 596
  return ret;
}

597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621
static gboolean
gst_text_render_event (GstPad * pad, GstObject * parent, GstEvent * event)
{
  GstTextRender *render = GST_TEXT_RENDER (parent);
  gboolean ret = TRUE;

  switch (GST_EVENT_TYPE (event)) {
    case GST_EVENT_SEGMENT:
    {
      if (gst_pad_has_current_caps (render->srcpad)) {
        ret = gst_pad_push_event (render->srcpad, event);
      } else {
        gst_event_replace (&render->segment_event, event);
        gst_event_unref (event);
      }
      break;
    }
    default:
      ret = gst_pad_push_event (render->srcpad, event);
      break;
  }

  return ret;
}

622 623 624 625 626
static void
gst_text_render_finalize (GObject * object)
{
  GstTextRender *render = GST_TEXT_RENDER (object);

627 628
  gst_event_replace (&render->segment_event, NULL);

629
  g_free (render->text_image);
630 631 632 633 634 635 636 637

  if (render->layout)
    g_object_unref (render->layout);

  G_OBJECT_CLASS (parent_class)->finalize (object);
}

static void
638
gst_text_render_init (GstTextRender * render)
639
{
Christophe Fergeau's avatar
Christophe Fergeau committed
640 641
  GstPadTemplate *template;

642
  /* sink */
Christophe Fergeau's avatar
Christophe Fergeau committed
643 644 645
  template = gst_static_pad_template_get (&sink_template_factory);
  render->sinkpad = gst_pad_new_from_template (template, "sink");
  gst_object_unref (template);
646 647
  gst_pad_set_chain_function (render->sinkpad,
      GST_DEBUG_FUNCPTR (gst_text_render_chain));
648 649 650
  gst_pad_set_event_function (render->sinkpad,
      GST_DEBUG_FUNCPTR (gst_text_render_event));

651 652 653
  gst_element_add_pad (GST_ELEMENT (render), render->sinkpad);

  /* source */
Christophe Fergeau's avatar
Christophe Fergeau committed
654 655 656
  template = gst_static_pad_template_get (&src_template_factory);
  render->srcpad = gst_pad_new_from_template (template, "src");
  gst_object_unref (template);
657

658 659
  gst_element_add_pad (GST_ELEMENT (render), render->srcpad);

660
  render->line_align = DEFAULT_PROP_LINE_ALIGNMENT;
661 662
  render->layout =
      pango_layout_new (GST_TEXT_RENDER_GET_CLASS (render)->pango_context);
663 664 665 666 667 668 669 670 671 672 673 674
  pango_layout_set_alignment (render->layout,
      (PangoAlignment) render->line_align);

  render->halign = DEFAULT_PROP_HALIGNMENT;
  render->valign = DEFAULT_PROP_VALIGNMENT;
  render->xpad = DEFAULT_PROP_XPAD;
  render->ypad = DEFAULT_PROP_YPAD;

  render->width = DEFAULT_RENDER_WIDTH;
  render->height = DEFAULT_RENDER_HEIGHT;

  render->use_ARGB = FALSE;
675 676 677 678 679 680 681 682 683
}

static void
gst_text_render_set_property (GObject * object, guint prop_id,
    const GValue * value, GParamSpec * pspec)
{
  GstTextRender *render = GST_TEXT_RENDER (object);

  switch (prop_id) {
684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700
    case PROP_VALIGNMENT:
      render->valign = g_value_get_enum (value);
      break;
    case PROP_HALIGNMENT:
      render->halign = g_value_get_enum (value);
      break;
    case PROP_LINE_ALIGNMENT:
      render->line_align = g_value_get_enum (value);
      pango_layout_set_alignment (render->layout,
          (PangoAlignment) render->line_align);
      break;
    case PROP_XPAD:
      render->xpad = g_value_get_int (value);
      break;
    case PROP_YPAD:
      render->ypad = g_value_get_int (value);
      break;
701
    case PROP_FONT_DESC:
702 703 704 705 706 707 708 709
    {
      PangoFontDescription *desc;

      desc = pango_font_description_from_string (g_value_get_string (value));
      if (desc) {
        GST_LOG ("font description set: %s", g_value_get_string (value));
        GST_OBJECT_LOCK (render);
        pango_layout_set_font_description (render->layout, desc);
710
        gst_text_render_adjust_values_with_fontdesc (render, desc);
711
        pango_font_description_free (desc);
712
        gst_text_render_render_pangocairo (render);
713 714 715 716 717 718 719 720 721 722 723 724
        GST_OBJECT_UNLOCK (render);
      } else {
        GST_WARNING ("font description parse failed: %s",
            g_value_get_string (value));
      }
      break;
    }

    default:
      break;
  }
}
725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752

static void
gst_text_render_get_property (GObject * object, guint prop_id,
    GValue * value, GParamSpec * pspec)
{
  GstTextRender *render = GST_TEXT_RENDER (object);

  switch (prop_id) {
    case PROP_VALIGNMENT:
      g_value_set_enum (value, render->valign);
      break;
    case PROP_HALIGNMENT:
      g_value_set_enum (value, render->halign);
      break;
    case PROP_LINE_ALIGNMENT:
      g_value_set_enum (value, render->line_align);
      break;
    case PROP_XPAD:
      g_value_set_int (value, render->xpad);
      break;
    case PROP_YPAD:
      g_value_set_int (value, render->ypad);
      break;
    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
      break;
  }
}