ext/cairo/gstcairooverlay.c

   1 /* GStreamer
   2  * Copyright (C) <2011> Jon Nordby <jononor@gmail.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 /**
  21  * SECTION:element-cairooverlay
  22  *
  23  * cairooverlay renders an overlay using a application provided render function.
  24  *
  25  * The full example can be found in tests/examples/cairo/cairo_overlay.c
  26  * <refsect2>
  27  * <title>Example code</title>
  28  * |[
  29  *
  30  * #include &lt;gst/gst.h&gt;
  31  * #include &lt;gst/video/video.h&gt;
  32  *
  33  * ...
  34  *
  35  * typedef struct {
  36  *   gboolean valid;
  37  *   int width;
  38  *   int height;
  39  * } CairoOverlayState;
  40  *
  41  * ...
  42  *
  43  * static void
  44  * prepare_overlay (GstElement * overlay, GstCaps * caps, gpointer user_data)
  45  * {
  46  *   CairoOverlayState *state = (CairoOverlayState *)user_data;
  47  *
  48  *   gst_video_format_parse_caps (caps, NULL, &amp;state-&gt;width, &amp;state-&gt;height);
  49  *   state-&gt;valid = TRUE;
  50  * }
  51  *
  52  * static void
  53  * draw_overlay (GstElement * overlay, cairo_t * cr, guint64 timestamp,
  54  *   guint64 duration, gpointer user_data)
  55  * {
  56  *   CairoOverlayState *s = (CairoOverlayState *)user_data;
  57  *   double scale;
  58  *
  59  *   if (!s-&gt;valid)
  60  *     return;
  61  *
  62  *   scale = 2*(((timestamp/(int)1e7) % 70)+30)/100.0;
  63  *   cairo_translate(cr, s-&gt;width/2, (s-&gt;height/2)-30);
  64  *   cairo_scale (cr, scale, scale);
  65  *
  66  *   cairo_move_to (cr, 0, 0);
  67  *   cairo_curve_to (cr, 0,-30, -50,-30, -50,0);
  68  *   cairo_curve_to (cr, -50,30, 0,35, 0,60 );
  69  *   cairo_curve_to (cr, 0,35, 50,30, 50,0 ); *
  70  *   cairo_curve_to (cr, 50,-30, 0,-30, 0,0 );
  71  *   cairo_set_source_rgba (cr, 0.9, 0.0, 0.1, 0.7);
  72  *   cairo_fill (cr);
  73  * }
  74  *
  75  * ...
  76  *
  77  * cairo_overlay = gst_element_factory_make (&quot;cairooverlay&quot;, &quot;overlay&quot;);
  78  *
  79  * g_signal_connect (cairo_overlay, &quot;draw&quot;, G_CALLBACK (draw_overlay),
  80  *   overlay_state);
  81  * g_signal_connect (cairo_overlay, &quot;caps-changed&quot;,
  82  *   G_CALLBACK (prepare_overlay), overlay_state);
  83  * ...
  84  *
  85  * ]|
  86  * </refsect2>
  87  */
  88
  89 #ifdef HAVE_CONFIG_H
  90 #include "config.h"
  91 #endif
  92
  93 #include "gstcairooverlay.h"
  94
  95 #include <gst/video/video.h>
  96
  97 #include <cairo.h>
  98
  99 #if G_BYTE_ORDER == G_LITTLE_ENDIAN
 100 #define TEMPLATE_CAPS GST_VIDEO_CAPS_MAKE("{ BGRx, BGRA }")
 101 #else
 102 #define TEMPLATE_CAPS GST_VIDEO_CAPS_MAKE("{ xRGB, ARGB }")
 103
 104 #endif
 105
 106 static GstStaticPadTemplate gst_cairo_overlay_src_template =
 107 GST_STATIC_PAD_TEMPLATE ("src",
 108     GST_PAD_SRC,
 109     GST_PAD_ALWAYS,
 110     GST_STATIC_CAPS (TEMPLATE_CAPS)
 111     );
 112
 113 static GstStaticPadTemplate gst_cairo_overlay_sink_template =
 114 GST_STATIC_PAD_TEMPLATE ("sink",
 115     GST_PAD_SINK,
 116     GST_PAD_ALWAYS,
 117     GST_STATIC_CAPS (TEMPLATE_CAPS)
 118     );
 119
 120 G_DEFINE_TYPE (GstCairoOverlay, gst_cairo_overlay, GST_TYPE_VIDEO_FILTER);
 121
 122 enum
 123 {
 124   SIGNAL_DRAW,
 125   SIGNAL_CAPS_CHANGED,
 126   N_SIGNALS
 127 };
 128
 129 static guint gst_cairo_overlay_signals[N_SIGNALS];
 130
 131 static gboolean
 132 gst_cairo_overlay_set_info (GstVideoFilter * vfilter, GstCaps * in_caps,
 133     GstVideoInfo * in_info, GstCaps * out_caps, GstVideoInfo * out_info)
 134 {
 135   GstCairoOverlay *overlay = GST_CAIRO_OVERLAY (vfilter);
 136
 137   g_signal_emit (overlay, gst_cairo_overlay_signals[SIGNAL_CAPS_CHANGED], 0,
 138       in_caps, NULL);
 139
 140   return TRUE;
 141 }
 142
 143 static GstFlowReturn
 144 gst_cairo_overlay_transform_frame_ip (GstVideoFilter * vfilter,
 145     GstVideoFrame * frame)
 146 {
 147   GstCairoOverlay *overlay = GST_CAIRO_OVERLAY (vfilter);
 148   cairo_surface_t *surface;
 149   cairo_t *cr;
 150   cairo_format_t format;
 151
 152   if (GST_VIDEO_FRAME_N_COMPONENTS (frame) == 4)
 153     format = CAIRO_FORMAT_ARGB32;
 154   else
 155     format = CAIRO_FORMAT_RGB24;
 156
 157   surface =
 158       cairo_image_surface_create_for_data (GST_VIDEO_FRAME_PLANE_DATA (frame,
 159           0), format, GST_VIDEO_FRAME_WIDTH (frame),
 160       GST_VIDEO_FRAME_HEIGHT (frame), GST_VIDEO_FRAME_PLANE_STRIDE (frame, 0));
 161
 162   if (G_UNLIKELY (!surface))
 163     return GST_FLOW_ERROR;
 164
 165   cr = cairo_create (surface);
 166   if (G_UNLIKELY (!cr)) {
 167     cairo_surface_destroy (surface);
 168     return GST_FLOW_ERROR;
 169   }
 170
 171   g_signal_emit (overlay, gst_cairo_overlay_signals[SIGNAL_DRAW], 0,
 172       cr, GST_BUFFER_PTS (frame->buffer), GST_BUFFER_DURATION (frame->buffer),
 173       NULL);
 174
 175   cairo_destroy (cr);
 176   cairo_surface_destroy (surface);
 177
 178   return GST_FLOW_OK;
 179 }
 180
 181 static void
 182 gst_cairo_overlay_class_init (GstCairoOverlayClass * klass)
 183 {
 184   GstVideoFilterClass *vfilter_class;
 185   GstElementClass *element_class;
 186
 187   vfilter_class = (GstVideoFilterClass *) klass;
 188   element_class = (GstElementClass *) klass;
 189
 190   vfilter_class->set_info = gst_cairo_overlay_set_info;
 191   vfilter_class->transform_frame_ip = gst_cairo_overlay_transform_frame_ip;
 192
 193   /**
 194    * GstCairoOverlay::draw:
 195    * @overlay: Overlay element emitting the signal.
 196    * @cr: Cairo context to draw to.
 197    * @timestamp: Timestamp (see #GstClockTime) of the current buffer.
 198    * @duration: Duration (see #GstClockTime) of the current buffer.
 199    *
 200    * This signal is emitted when the overlay should be drawn.
 201    */
 202   gst_cairo_overlay_signals[SIGNAL_DRAW] =
 203       g_signal_new ("draw",
 204       G_TYPE_FROM_CLASS (klass),
 205       0,
 206       0,
 207       NULL,
 208       NULL,
 209       g_cclosure_marshal_generic,
 210       G_TYPE_NONE, 3, CAIRO_GOBJECT_TYPE_CONTEXT, G_TYPE_UINT64, G_TYPE_UINT64);
 211
 212   /**
 213    * GstCairoOverlay::caps-changed:
 214    * @overlay: Overlay element emitting the signal.
 215    * @caps: The #GstCaps of the element.
 216    *
 217    * This signal is emitted when the caps of the element has changed.
 218    */
 219   gst_cairo_overlay_signals[SIGNAL_CAPS_CHANGED] =
 220       g_signal_new ("caps-changed",
 221       G_TYPE_FROM_CLASS (klass),
 222       0,
 223       0, NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 1, GST_TYPE_CAPS);
 224
 225   gst_element_class_set_static_metadata (element_class, "Cairo overlay",
 226       "Filter/Editor/Video",
 227       "Render overlay on a video stream using Cairo",
 228       "Jon Nordby <jononor@gmail.com>");
 229
 230   gst_element_class_add_pad_template (element_class,
 231       gst_static_pad_template_get (&gst_cairo_overlay_sink_template));
 232   gst_element_class_add_pad_template (element_class,
 233       gst_static_pad_template_get (&gst_cairo_overlay_src_template));
 234 }
 235
 236 static void
 237 gst_cairo_overlay_init (GstCairoOverlay * overlay)
 238 {
 239   /* nothing to do */
 240 }