+2008-08-24 Ole André Vadla Ravnås <ole.andre.ravnas@tandberg.com>
+
+ patch by: Haakon Sporsheim <hakon.sporsheim@tandberg.com>
+
+ * configure.ac:
+ * sys/Makefile.am:
+ * sys/winscreencap/Makefile.am:
+ * sys/winscreencap/gstdx9screencapsrc.c:
+ * sys/winscreencap/gstdx9screencapsrc.h:
+ * sys/winscreencap/gstgdiscreencapsrc.c:
+ * sys/winscreencap/gstgdiscreencapsrc.h:
+ * sys/winscreencap/gstwinscreencap.c:
+ * sys/winscreencap/gstwinscreencap.h:
+ New plugin: winscreencap (#463941).
+
2008-08-22 Michael Smith <msmith@songbirdnest.com>
* sys/dshowdecwrapper/gstdshowaudiodec.c:
sys/qtwrapper/Makefile
sys/vcd/Makefile
sys/wininet/Makefile
+sys/winscreencap/Makefile
examples/Makefile
examples/app/Makefile
examples/directfb/Makefile
SUBDIRS = $(ACM_DIR) $(DVB_DIR) $(FBDEV_DIR) $(OSS4_DIR) $(QT_DIR) $(VCD_DIR) $(WININET_DIR)
DIST_SUBDIRS = acmenc dvb fbdev dshowdecwrapper dshowsrcwrapper dshowvideosink \
- oss4 qtwrapper vcd wininet
+ oss4 qtwrapper vcd wininet winscreencap
--- /dev/null
+# This plugin isn't buildable with autotools at this point in time, so just
+# ensure everything's listed in EXTRA_DIST
+
+EXTRA_DIST = \
+ gstdx9screencapsrc.c gstdx9screencapsrc.h \
+ gstgdiscreencapsrc.c gstgdiscreencapsrc.h \
+ gstwinscreencap.c gstwinscreencap.h
+
--- /dev/null
+/* GStreamer
+ * Copyright (C) 2007 Haakon Sporsheim <hakon.sporsheim@tandberg.com>
+ *
+ * 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
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include "gstdx9screencapsrc.h"
+#include <gst/video/video.h>
+
+GST_DEBUG_CATEGORY_STATIC (dx9screencapsrc_debug);
+
+static const GstElementDetails gst_dx9screencapsrc_details = {
+ "DirectX 9 screen capture source",
+ "Source/Video",
+ "Captures screen",
+ "Haakon Sporsheim <hakon.sporsheim@tandberg.com>"
+};
+
+#define GST_VIDEO_ALPHA_MASK_15_INT 0x8000
+
+static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("src",
+ GST_PAD_SRC,
+ GST_PAD_ALWAYS,
+ GST_STATIC_CAPS ("video/x-raw-rgb,"
+ "width = " GST_VIDEO_SIZE_RANGE ","
+ "height = " GST_VIDEO_SIZE_RANGE "," "framerate = " GST_VIDEO_FPS_RANGE)
+ );
+
+GST_BOILERPLATE (GstDX9ScreenCapSrc, gst_dx9screencapsrc,
+ GstPushSrc, GST_TYPE_PUSH_SRC);
+
+enum
+{
+ PROP_0,
+ PROP_MONITOR,
+ PROP_X_POS,
+ PROP_Y_POS,
+ PROP_WIDTH,
+ PROP_HEIGHT
+};
+
+static IDirect3D9 *g_d3d9 = NULL;
+
+/* Fwd. decl. */
+static GstCaps *gst_dx9screencapsrc_create_caps_from_format (D3DFORMAT fmt,
+ gint width, gint height);
+
+static void gst_dx9screencapsrc_dispose (GObject * object);
+static void gst_dx9screencapsrc_set_property (GObject * object,
+ guint prop_id, const GValue * value, GParamSpec * pspec);
+static void gst_dx9screencapsrc_get_property (GObject * object,
+ guint prop_id, GValue * value, GParamSpec * pspec);
+
+static void gst_dx9screencapsrc_fixate (GstPad * pad, GstCaps * caps);
+static gboolean gst_dx9screencapsrc_set_caps (GstBaseSrc * bsrc,
+ GstCaps * caps);
+static GstCaps *gst_dx9screencapsrc_get_caps (GstBaseSrc * bsrc);
+static gboolean gst_dx9screencapsrc_start (GstBaseSrc * bsrc);
+static gboolean gst_dx9screencapsrc_stop (GstBaseSrc * bsrc);
+
+static void gst_dx9screencapsrc_get_times (GstBaseSrc * basesrc,
+ GstBuffer * buffer, GstClockTime * start, GstClockTime * end);
+static GstFlowReturn gst_dx9screencapsrc_create (GstPushSrc * src,
+ GstBuffer ** buf);
+
+/* Implementation. */
+static void
+gst_dx9screencapsrc_base_init (gpointer klass)
+{
+ GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
+
+ gst_element_class_add_pad_template (element_class,
+ gst_static_pad_template_get (&src_template));
+
+ gst_element_class_set_details (element_class, &gst_dx9screencapsrc_details);
+}
+
+static void
+gst_dx9screencapsrc_class_init (GstDX9ScreenCapSrcClass * klass)
+{
+ GObjectClass *go_class;
+ GstBaseSrcClass *bs_class;
+ GstPushSrcClass *ps_class;
+
+ go_class = (GObjectClass *) klass;
+ bs_class = (GstBaseSrcClass *) klass;
+ ps_class = (GstPushSrcClass *) klass;
+
+ go_class->dispose = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_dispose);
+ go_class->set_property = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_set_property);
+ go_class->get_property = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_get_property);
+
+ bs_class->get_times = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_get_times);
+ bs_class->get_caps = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_get_caps);
+ bs_class->set_caps = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_set_caps);
+ bs_class->start = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_start);
+ bs_class->stop = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_stop);
+
+ ps_class->create = GST_DEBUG_FUNCPTR (gst_dx9screencapsrc_create);
+
+ g_object_class_install_property (go_class, PROP_MONITOR,
+ g_param_spec_int ("monitor", "Monitor",
+ "Which monitor to use (0 = 1st monitor and default)",
+ 0, G_MAXINT, 0, G_PARAM_READWRITE));
+
+ g_object_class_install_property (go_class, PROP_X_POS,
+ g_param_spec_int ("x", "X",
+ "Horizontal coordinate of top left corner for the screen capture "
+ "area", 0, G_MAXINT, 0, G_PARAM_READWRITE));
+ g_object_class_install_property (go_class, PROP_Y_POS,
+ g_param_spec_int ("y", "Y",
+ "Vertical coordinate of top left corner for the screen capture "
+ "area", 0, G_MAXINT, 0, G_PARAM_READWRITE));
+
+ g_object_class_install_property (go_class, PROP_WIDTH,
+ g_param_spec_int ("width", "Width",
+ "Width of screen capture area (0 = maximum)",
+ 0, G_MAXINT, 0, G_PARAM_READWRITE));
+ g_object_class_install_property (go_class, PROP_HEIGHT,
+ g_param_spec_int ("height", "Height",
+ "Height of screen capture area (0 = maximum)",
+ 0, G_MAXINT, 0, G_PARAM_READWRITE));
+
+ GST_DEBUG_CATEGORY_INIT (dx9screencapsrc_debug, "dx9screencapsrc", 0,
+ "DirectX 9 screen capture source");
+}
+
+static void
+gst_dx9screencapsrc_init (GstDX9ScreenCapSrc * src,
+ GstDX9ScreenCapSrcClass * klass)
+{
+ /* Set src element inital values... */
+ GstPad *src_pad = GST_BASE_SRC_PAD (src);
+
+ gst_pad_set_fixatecaps_function (src_pad, gst_dx9screencapsrc_fixate);
+
+ src->frames = 0;
+ src->surface = NULL;
+ src->d3d9_device = NULL;
+ src->capture_x = 0;
+ src->capture_y = 0;
+ src->capture_w = 0;
+ src->capture_h = 0;
+
+ src->monitor = 0;
+
+ gst_base_src_set_format (GST_BASE_SRC (src), GST_FORMAT_TIME);
+ gst_base_src_set_live (GST_BASE_SRC (src), TRUE);
+
+ if (!g_d3d9)
+ g_d3d9 = Direct3DCreate9 (D3D_SDK_VERSION);
+ else
+ IDirect3D9_AddRef (g_d3d9);
+}
+
+static void
+gst_dx9screencapsrc_dispose (GObject * object)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (object);
+
+ if (src->surface) {
+ GST_ERROR_OBJECT (object,
+ "DX9 surface was not freed in _stop, freeing in _dispose!");
+ IDirect3DSurface9_Release (src->surface);
+ src->surface = NULL;
+ }
+
+ if (src->d3d9_device) {
+ IDirect3DDevice9_Release (src->d3d9_device);
+ src->d3d9_device = NULL;
+ }
+
+ if (!IDirect3D9_Release (g_d3d9))
+ g_d3d9 = NULL;
+}
+
+static void
+gst_dx9screencapsrc_set_property (GObject * object,
+ guint prop_id, const GValue * value, GParamSpec * pspec)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (object);
+
+ switch (prop_id) {
+ case PROP_MONITOR:
+ if (g_value_get_int (value) >= GetSystemMetrics (SM_CMONITORS)) {
+ G_OBJECT_WARN_INVALID_PSPEC (object, "Monitor", prop_id, pspec);
+ break;
+ }
+ src->monitor = g_value_get_int (value);
+ break;
+ case PROP_X_POS:
+ src->capture_x = g_value_get_int (value);
+ break;
+ case PROP_Y_POS:
+ src->capture_y = g_value_get_int (value);
+ break;
+ case PROP_WIDTH:
+ src->capture_w = g_value_get_int (value);
+ break;
+ case PROP_HEIGHT:
+ src->capture_h = g_value_get_int (value);
+ break;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ };
+}
+
+static void
+gst_dx9screencapsrc_get_property (GObject * object, guint prop_id,
+ GValue * value, GParamSpec * pspec)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (object);
+
+ switch (prop_id) {
+ case PROP_MONITOR:
+ g_value_set_int (value, src->monitor);
+ break;
+ case PROP_X_POS:
+ g_value_set_int (value, src->capture_x);
+ break;
+ case PROP_Y_POS:
+ g_value_set_int (value, src->capture_y);
+ break;
+ case PROP_WIDTH:
+ g_value_set_int (value, src->capture_w);
+ break;
+ case PROP_HEIGHT:
+ g_value_set_int (value, src->capture_h);
+ break;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ };
+}
+
+static void
+gst_dx9screencapsrc_fixate (GstPad * pad, GstCaps * caps)
+{
+ GstStructure *structure;
+
+ structure = gst_caps_get_structure (caps, 0);
+
+ gst_structure_fixate_field_nearest_int (structure, "width", 640);
+ gst_structure_fixate_field_nearest_int (structure, "height", 480);
+ gst_structure_fixate_field_nearest_fraction (structure, "framerate", 30, 1);
+}
+
+static gboolean
+gst_dx9screencapsrc_set_caps (GstBaseSrc * bsrc, GstCaps * caps)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (bsrc);
+ GstStructure *structure;
+ const GValue *framerate;
+
+ structure = gst_caps_get_structure (caps, 0);
+
+ src->src_rect = src->screen_rect;
+ if (src->capture_w && src->capture_h) {
+ src->src_rect.left += src->capture_x;
+ src->src_rect.top += src->capture_y;
+ src->src_rect.right = src->src_rect.left + src->capture_w;
+ src->src_rect.bottom = src->src_rect.top + src->capture_h;
+ }
+
+ if (framerate = gst_structure_get_value (structure, "framerate")) {
+ src->rate_numerator = gst_value_get_fraction_numerator (framerate);
+ src->rate_denominator = gst_value_get_fraction_denominator (framerate);
+ }
+
+ GST_DEBUG_OBJECT (src, "size %dx%d, %d/%d fps",
+ src->src_rect.right - src->src_rect.left,
+ src->src_rect.bottom - src->src_rect.top,
+ src->rate_numerator, src->rate_denominator);
+
+ return TRUE;
+}
+
+static GstCaps *
+gst_dx9screencapsrc_get_caps (GstBaseSrc * bsrc)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (bsrc);
+ RECT rect_dst;
+
+ if (src->monitor >= IDirect3D9_GetAdapterCount (g_d3d9) ||
+ FAILED (IDirect3D9_GetAdapterDisplayMode (g_d3d9, src->monitor,
+ &src->disp_mode))) {
+ return NULL;
+ }
+
+ SetRect (&rect_dst, 0, 0, src->disp_mode.Width, src->disp_mode.Height);
+ src->screen_rect = rect_dst;
+
+ if (src->capture_w && src->capture_h &&
+ src->capture_x + src->capture_w < rect_dst.right - rect_dst.left &&
+ src->capture_y + src->capture_h < rect_dst.bottom - rect_dst.top) {
+ rect_dst.left = src->capture_x;
+ rect_dst.top = src->capture_y;
+ rect_dst.right = src->capture_x + src->capture_w;
+ rect_dst.bottom = src->capture_y + src->capture_h;
+ } else {
+ /* Default values */
+ src->capture_x = src->capture_y = 0;
+ src->capture_w = src->capture_h = 0;
+ }
+
+ /* Note:
+ * Expose as xRGB even though the Surface is allocated as ARGB!
+ * This is due to IDirect3DDevice9_GetFrontBufferData which only takes
+ * ARGB surface, but the A channel is in reality never used.
+ * I should add that I had problems specifying ARGB. It might be a bug
+ * in ffmpegcolorspace which I used for testing.
+ * Another interesting thing is that directdrawsink did not support ARGB,
+ * but only xRGB. (On my system, using 32b color depth) And according to
+ * the DirectX documentation ARGB is NOT a valid display buffer format,
+ * but xRGB is.
+ */
+ return gst_dx9screencapsrc_create_caps_from_format (D3DFMT_X8R8G8B8,
+ rect_dst.right - rect_dst.left, rect_dst.bottom - rect_dst.top);
+}
+
+static GstCaps *
+gst_dx9screencapsrc_create_caps_from_format (D3DFORMAT fmt,
+ gint width, gint height)
+{
+ gint bpp, depth, endianness;
+ gint red, green, blue, alpha;
+
+ switch (fmt) {
+ case D3DFMT_A8R8G8B8:
+ bpp = depth = 32;
+ endianness = G_BIG_ENDIAN;
+ alpha = GST_VIDEO_BYTE4_MASK_32_INT;
+ red = GST_VIDEO_BYTE3_MASK_32_INT;
+ green = GST_VIDEO_BYTE2_MASK_32_INT;
+ blue = GST_VIDEO_BYTE1_MASK_32_INT;
+ break;
+ case D3DFMT_X8R8G8B8:
+ bpp = 32;
+ depth = 24;
+ endianness = G_BIG_ENDIAN;
+ alpha = 0;
+ red = GST_VIDEO_BYTE3_MASK_32_INT;
+ green = GST_VIDEO_BYTE2_MASK_32_INT;
+ blue = GST_VIDEO_BYTE1_MASK_32_INT;
+ break;
+ case D3DFMT_R8G8B8:
+ endianness = G_BIG_ENDIAN;
+ bpp = depth = 24;
+ alpha = 0;
+ red = GST_VIDEO_BYTE1_MASK_24_INT;
+ green = GST_VIDEO_BYTE2_MASK_24_INT;
+ blue = GST_VIDEO_BYTE3_MASK_24_INT;
+ break;
+ case D3DFMT_A1R5G5B5:
+ bpp = 16;
+ depth = 15;
+ endianness = G_BYTE_ORDER;
+ alpha = GST_VIDEO_ALPHA_MASK_15_INT;
+ red = GST_VIDEO_RED_MASK_15_INT;
+ green = GST_VIDEO_GREEN_MASK_15_INT;
+ blue = GST_VIDEO_BLUE_MASK_15_INT;
+ break;
+ case D3DFMT_X1R5G5B5:
+ bpp = 16;
+ depth = 15;
+ endianness = G_BYTE_ORDER;
+ alpha = 0;
+ red = GST_VIDEO_RED_MASK_15_INT;
+ green = GST_VIDEO_GREEN_MASK_15_INT;
+ blue = GST_VIDEO_BLUE_MASK_15_INT;
+ break;
+ case D3DFMT_R5G6B5:
+ bpp = depth = 16;
+ endianness = G_BYTE_ORDER;
+ alpha = 0;
+ red = GST_VIDEO_RED_MASK_16_INT;
+ green = GST_VIDEO_GREEN_MASK_16_INT;
+ blue = GST_VIDEO_BLUE_MASK_16_INT;
+ break;
+ default:
+ return NULL;
+ }
+
+ return gst_caps_new_simple ("video/x-raw-rgb",
+ "width", G_TYPE_INT, width,
+ "height", G_TYPE_INT, height,
+ "framerate", GST_TYPE_FRACTION_RANGE, 1, 1, G_MAXINT, 1,
+ "bpp", G_TYPE_INT, bpp,
+ "depth", G_TYPE_INT, depth,
+ "endianness", G_TYPE_INT, endianness,
+ "alpha_mask", G_TYPE_INT, alpha,
+ "red_mask", G_TYPE_INT, red,
+ "green_mask", G_TYPE_INT, green, "blue_mask", G_TYPE_INT, blue, NULL);
+}
+
+static gboolean
+gst_dx9screencapsrc_start (GstBaseSrc * bsrc)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (bsrc);
+ D3DPRESENT_PARAMETERS d3dpp;
+ HRESULT res;
+
+ ZeroMemory (&d3dpp, sizeof (D3DPRESENT_PARAMETERS));
+ d3dpp.Windowed = TRUE;
+ d3dpp.Flags = D3DPRESENTFLAG_LOCKABLE_BACKBUFFER;
+ d3dpp.BackBufferFormat = src->disp_mode.Format;
+ d3dpp.BackBufferHeight = src->disp_mode.Height;
+ d3dpp.BackBufferWidth = src->disp_mode.Width;
+ d3dpp.BackBufferCount = 1;
+ d3dpp.MultiSampleType = D3DMULTISAMPLE_NONE;
+ d3dpp.SwapEffect = D3DSWAPEFFECT_DISCARD;
+ d3dpp.hDeviceWindow = GetDesktopWindow ();
+ d3dpp.PresentationInterval = D3DPRESENT_INTERVAL_DEFAULT;
+ d3dpp.FullScreen_RefreshRateInHz = D3DPRESENT_RATE_DEFAULT;
+
+ src->frames = 0;
+
+ res = IDirect3D9_CreateDevice (g_d3d9, src->monitor, D3DDEVTYPE_HAL,
+ GetDesktopWindow (), D3DCREATE_SOFTWARE_VERTEXPROCESSING,
+ &d3dpp, &src->d3d9_device);
+ if (FAILED (res))
+ return FALSE;
+
+ return SUCCEEDED (IDirect3DDevice9_CreateOffscreenPlainSurface (src->
+ d3d9_device, src->disp_mode.Width, src->disp_mode.Height,
+ D3DFMT_A8R8G8B8, D3DPOOL_SYSTEMMEM, &src->surface, NULL));
+}
+
+static gboolean
+gst_dx9screencapsrc_stop (GstBaseSrc * bsrc)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (bsrc);
+
+ if (src->surface) {
+ IDirect3DSurface9_Release (src->surface);
+ src->surface = NULL;
+ }
+
+ return TRUE;
+}
+
+static void
+gst_dx9screencapsrc_get_times (GstBaseSrc * basesrc,
+ GstBuffer * buffer, GstClockTime * start, GstClockTime * end)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (basesrc);
+ GstClockTime timestamp;
+
+ timestamp = GST_BUFFER_TIMESTAMP (buffer);
+
+ if (GST_CLOCK_TIME_IS_VALID (timestamp)) {
+ GstClockTime duration = GST_BUFFER_DURATION (buffer);
+
+ if (GST_CLOCK_TIME_IS_VALID (duration))
+ *end = timestamp + duration;
+ *start = timestamp;
+ }
+}
+
+static GstFlowReturn
+gst_dx9screencapsrc_create (GstPushSrc * push_src, GstBuffer ** buf)
+{
+ GstDX9ScreenCapSrc *src = GST_DX9SCREENCAPSRC (push_src);
+ GstBuffer *new_buf;
+ GstFlowReturn res;
+ gint new_buf_size, i;
+ gint width, height, stride;
+ GstClock *clock;
+ GstClockTime time, buf_time;
+ D3DLOCKED_RECT locked_rect;
+ LPBYTE p_dst, p_src;
+ HRESULT hres;
+
+ if (G_UNLIKELY (!src->d3d9_device)) {
+ GST_ELEMENT_ERROR (src, CORE, NEGOTIATION, (NULL),
+ ("format wasn't negotiated before create function"));
+ return GST_FLOW_NOT_NEGOTIATED;
+ }
+
+ clock = gst_element_get_clock (GST_ELEMENT (src));
+ if (clock != NULL) {
+ /* Calculate sync time. */
+ GstClockTime base_time;
+ GstClockTime frame_time =
+ gst_util_uint64_scale_int (src->frames * GST_SECOND,
+ src->rate_denominator, src->rate_numerator);
+
+ time = gst_clock_get_time (clock);
+ base_time = gst_element_get_base_time (GST_ELEMENT (src));
+ buf_time = MAX (time - base_time, frame_time);
+ } else {
+ buf_time = GST_CLOCK_TIME_NONE;
+ }
+
+ height = (src->src_rect.bottom - src->src_rect.top);
+ width = (src->src_rect.right - src->src_rect.left);
+ new_buf_size = width * 4 * height;
+ if (G_UNLIKELY (src->rate_numerator == 0 && src->frames == 1)) {
+ GST_DEBUG_OBJECT (src, "eos: 0 framerate, frame %d", (gint) src->frames);
+ return GST_FLOW_UNEXPECTED;
+ }
+
+ GST_LOG_OBJECT (src,
+ "creating buffer of %lu bytes with %dx%d image for frame %d",
+ new_buf_size, width, height, (gint) src->frames);
+
+ res = gst_pad_alloc_buffer_and_set_caps (GST_BASE_SRC_PAD (src),
+ GST_BUFFER_OFFSET_NONE, new_buf_size,
+ GST_PAD_CAPS (GST_BASE_SRC_PAD (push_src)), &new_buf);
+ if (res != GST_FLOW_OK) {
+ GST_DEBUG_OBJECT (src, "could not allocate buffer, reason %s",
+ gst_flow_get_name (res));
+ return res;
+ }
+
+ /* Do screen capture and put it into buffer...
+ * Aquire front buffer, and lock it
+ */
+ hres =
+ IDirect3DDevice9_GetFrontBufferData (src->d3d9_device, 0, src->surface);
+ if (FAILED (hres)) {
+ GST_DEBUG_OBJECT (src, "DirectX::GetBackBuffer failed.");
+ return GST_FLOW_UNEXPECTED;
+ }
+
+ hres =
+ IDirect3DSurface9_LockRect (src->surface, &locked_rect, &(src->src_rect),
+ D3DLOCK_NO_DIRTY_UPDATE | D3DLOCK_NOSYSLOCK | D3DLOCK_READONLY);
+ if (FAILED (hres)) {
+ GST_DEBUG_OBJECT (src, "DirectX::LockRect failed.");
+ return GST_FLOW_UNEXPECTED;
+ }
+
+ p_dst = (LPBYTE) GST_BUFFER_DATA (new_buf);
+ p_src = (LPBYTE) locked_rect.pBits;
+ stride = width * 4;
+ for (i = 0; i < height; ++i) {
+ memcpy (p_dst, p_src, stride);
+ p_dst += stride;
+ p_src += locked_rect.Pitch;
+ }
+
+ /* Unlock copy of front buffer */
+ IDirect3DSurface9_UnlockRect (src->surface);
+
+ GST_BUFFER_TIMESTAMP (new_buf) = buf_time;
+ if (src->rate_numerator) {
+ GST_BUFFER_DURATION (new_buf) =
+ gst_util_uint64_scale_int (GST_SECOND,
+ src->rate_denominator, src->rate_numerator);
+
+ if (clock) {
+ GST_BUFFER_DURATION (new_buf) = MAX (GST_BUFFER_DURATION (new_buf),
+ gst_clock_get_time (clock) - time);
+ }
+ } else {
+ GST_BUFFER_DURATION (new_buf) = GST_CLOCK_TIME_NONE;
+ }
+
+ GST_BUFFER_OFFSET (new_buf) = src->frames;
+ src->frames++;
+ GST_BUFFER_OFFSET_END (new_buf) = src->frames;
+
+ if (clock != NULL)
+ gst_object_unref (clock);
+
+ *buf = new_buf;
+ return GST_FLOW_OK;
+}
--- /dev/null
+/* GStreamer
+ * Copyright (C) 2007 Haakon Sporsheim <hakon.sporsheim@tandberg.com>
+ *
+ * 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
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#ifndef __GST_DX9SCREENCAPSRC_H__
+#define __GST_DX9SCREENCAPSRC_H__
+
+#include <gst/gst.h>
+#include <gst/base/gstpushsrc.h>
+#include <d3d9.h>
+
+#include "gstwinscreencap.h"
+
+G_BEGIN_DECLS
+
+#define GST_TYPE_DX9SCREENCAPSRC (gst_dx9screencapsrc_get_type())
+#define GST_DX9SCREENCAPSRC(obj) \
+ (G_TYPE_CHECK_INSTANCE_CAST((obj), \
+ GST_TYPE_DX9SCREENCAPSRC,GstDX9ScreenCapSrc))
+#define GST_DX9SCREENCAPSRC_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_CAST((klass), \
+ GST_TYPE_DX9SCREENCAPSRC,GstDX9ScreenCapSrcClass))
+#define GST_IS_DX9SCREENCAPSRC(obj) \
+ (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_DX9SCREENCAPSRC))
+#define GST_IS_DX9SCREENCAPSRC_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_DX9SCREENCAPSRC))
+
+typedef struct _GstDX9ScreenCapSrc GstDX9ScreenCapSrc;
+typedef struct _GstDX9ScreenCapSrcClass GstDX9ScreenCapSrcClass;
+
+struct _GstDX9ScreenCapSrc
+{
+ /* Parent */
+ GstPushSrc src;
+
+ /* Properties */
+ gint capture_x;
+ gint capture_y;
+ gint capture_w;
+ gint capture_h;
+ guint monitor;
+
+ /* Source pad frame rate */
+ gint rate_numerator;
+ gint rate_denominator;
+
+ /* Runtime variables */
+ RECT screen_rect;
+ RECT src_rect;
+ gint64 frames;
+
+ D3DDISPLAYMODE disp_mode;
+ IDirect3DSurface9 *surface;
+ IDirect3DDevice9 *d3d9_device;
+};
+
+struct _GstDX9ScreenCapSrcClass
+{
+ GstPushSrcClass parent_class;
+};
+
+GType gst_dx9screencapsrc_get_type (void);
+
+G_END_DECLS
+
+#endif /* __GST_DX9SCREENCAPSRC_H__ */
--- /dev/null
+/* GStreamer
+ * Copyright (C) 2007 Haakon Sporsheim <hakon.sporsheim@tandberg.com>
+ *
+ * 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
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include "gstgdiscreencapsrc.h"
+#include <gst/video/video.h>
+
+GST_DEBUG_CATEGORY_STATIC (gdiscreencapsrc_debug);
+
+static const GstElementDetails gst_gdiscreencapsrc_details = {
+ "GDI screen capture source",
+ "Source/Video",
+ "Captures screen",
+ "Haakon Sporsheim <hakon.sporsheim@tandberg.com>"
+};
+
+static GstStaticPadTemplate src_template =
+GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS,
+ GST_STATIC_CAPS (GST_VIDEO_CAPS_BGR));
+
+GST_BOILERPLATE (GstGDIScreenCapSrc, gst_gdiscreencapsrc,
+ GstPushSrc, GST_TYPE_PUSH_SRC);
+
+enum
+{
+ PROP_0,
+ PROP_MONITOR,
+ PROP_SHOW_CURSOR,
+ PROP_X_POS,
+ PROP_Y_POS,
+ PROP_WIDTH,
+ PROP_HEIGHT
+};
+
+/* Fwd. decl. */
+static void gst_gdiscreencapsrc_dispose (GObject * object);
+static void gst_gdiscreencapsrc_set_property (GObject * object, guint prop_id,
+ const GValue * value, GParamSpec * pspec);
+static void gst_gdiscreencapsrc_get_property (GObject * object, guint prop_id,
+ GValue * value, GParamSpec * pspec);
+
+static void gst_gdiscreencapsrc_fixate (GstPad * pad, GstCaps * caps);
+static gboolean gst_gdiscreencapsrc_set_caps (GstBaseSrc * bsrc,
+ GstCaps * caps);
+static GstCaps *gst_gdiscreencapsrc_get_caps (GstBaseSrc * bsrc);
+static gboolean gst_gdiscreencapsrc_start (GstBaseSrc * bsrc);
+static gboolean gst_gdiscreencapsrc_stop (GstBaseSrc * bsrc);
+
+static void gst_gdiscreencapsrc_get_times (GstBaseSrc * basesrc,
+ GstBuffer * buffer, GstClockTime * start, GstClockTime * end);
+static GstFlowReturn gst_gdiscreencapsrc_create (GstPushSrc * src,
+ GstBuffer ** buf);
+
+static gboolean gst_gdiscreencapsrc_screen_capture (GstGDIScreenCapSrc * src,
+ GstBuffer * buf);
+
+/* Implementation. */
+static void
+gst_gdiscreencapsrc_base_init (gpointer klass)
+{
+ GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
+
+ gst_element_class_add_pad_template (element_class,
+ gst_static_pad_template_get (&src_template));
+ gst_element_class_set_details (element_class, &gst_gdiscreencapsrc_details);
+}
+
+static void
+gst_gdiscreencapsrc_class_init (GstGDIScreenCapSrcClass * klass)
+{
+ GObjectClass *go_class;
+ GstBaseSrcClass *bs_class;
+ GstPushSrcClass *ps_class;
+
+ go_class = (GObjectClass *) klass;
+ bs_class = (GstBaseSrcClass *) klass;
+ ps_class = (GstPushSrcClass *) klass;
+
+ go_class->dispose = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_dispose);
+ go_class->set_property = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_set_property);
+ go_class->get_property = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_get_property);
+
+ bs_class->get_times = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_get_times);
+ bs_class->get_caps = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_get_caps);
+ bs_class->set_caps = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_set_caps);
+ bs_class->start = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_start);
+ bs_class->stop = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_stop);
+
+ ps_class->create = GST_DEBUG_FUNCPTR (gst_gdiscreencapsrc_create);
+
+ g_object_class_install_property (go_class, PROP_MONITOR,
+ g_param_spec_int ("monitor",
+ "Monitor", "Which monitor to use (0 = 1st monitor and default)",
+ 0, G_MAXINT, 0, G_PARAM_READWRITE));
+
+ g_object_class_install_property (go_class, PROP_SHOW_CURSOR,
+ g_param_spec_boolean ("cursor", "Show mouse cursor",
+ "Whether to show mouse cursor (default off)",
+ FALSE, G_PARAM_READWRITE));
+
+ g_object_class_install_property (go_class, PROP_X_POS,
+ g_param_spec_int ("x", "X",
+ "Horizontal coordinate of top left corner for the screen capture "
+ "area", 0, G_MAXINT, 0, G_PARAM_READWRITE));
+ g_object_class_install_property (go_class, PROP_Y_POS,
+ g_param_spec_int ("y", "Y",
+ "Vertical coordinate of top left corner for the screen capture "
+ "area", 0, G_MAXINT, 0, G_PARAM_READWRITE));
+
+ g_object_class_install_property (go_class, PROP_WIDTH,
+ g_param_spec_int ("width", "Width",
+ "Width of screen capture area (0 = maximum)",
+ 0, G_MAXINT, 0, G_PARAM_READWRITE));
+ g_object_class_install_property (go_class, PROP_HEIGHT,
+ g_param_spec_int ("height", "Height",
+ "Height of screen capture area (0 = maximum)",
+ 0, G_MAXINT, 0, G_PARAM_READWRITE));
+
+ GST_DEBUG_CATEGORY_INIT (gdiscreencapsrc_debug,
+ "gdiscreencapsrc", 0, "GDI screen capture source");
+}
+
+static void
+gst_gdiscreencapsrc_init (GstGDIScreenCapSrc * src,
+ GstGDIScreenCapSrcClass * klass)
+{
+ /* Set src element inital values... */
+ GstPad *src_pad = GST_BASE_SRC_PAD (src);
+ gst_pad_set_fixatecaps_function (src_pad, gst_gdiscreencapsrc_fixate);
+
+ src->frames = 0;
+ src->dibMem = NULL;
+ src->hBitmap = (HBITMAP) INVALID_HANDLE_VALUE;
+ src->memDC = (HDC) INVALID_HANDLE_VALUE;
+ src->capture_x = 0;
+ src->capture_y = 0;
+ src->capture_w = 0;
+ src->capture_h = 0;
+
+ src->monitor = 0;
+ src->show_cursor = FALSE;
+
+ gst_base_src_set_format (GST_BASE_SRC (src), GST_FORMAT_TIME);
+ gst_base_src_set_live (GST_BASE_SRC (src), TRUE);
+}
+
+static void
+gst_gdiscreencapsrc_dispose (GObject * object)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (object);
+
+ if (src->hBitmap != INVALID_HANDLE_VALUE)
+ DeleteObject (src->hBitmap);
+
+ if (src->memDC != INVALID_HANDLE_VALUE)
+ DeleteDC (src->memDC);
+
+ src->hBitmap = (HBITMAP) INVALID_HANDLE_VALUE;
+ src->memDC = (HDC) INVALID_HANDLE_VALUE;
+ src->dibMem = NULL;
+}
+
+static void
+gst_gdiscreencapsrc_set_property (GObject * object, guint prop_id,
+ const GValue * value, GParamSpec * pspec)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (object);
+
+ switch (prop_id) {
+ case PROP_MONITOR:
+ if (g_value_get_int (value) >= GetSystemMetrics (SM_CMONITORS)) {
+ G_OBJECT_WARN_INVALID_PSPEC (object, "Monitor", prop_id, pspec);
+ break;
+ }
+ src->monitor = g_value_get_int (value);
+ break;
+ case PROP_SHOW_CURSOR:
+ src->show_cursor = g_value_get_boolean (value);
+ break;
+ case PROP_X_POS:
+ src->capture_x = g_value_get_int (value);
+ break;
+ case PROP_Y_POS:
+ src->capture_y = g_value_get_int (value);
+ break;
+ case PROP_WIDTH:
+ src->capture_w = g_value_get_int (value);
+ break;
+ case PROP_HEIGHT:
+ src->capture_h = g_value_get_int (value);
+ break;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ };
+}
+
+static void
+gst_gdiscreencapsrc_get_property (GObject * object, guint prop_id,
+ GValue * value, GParamSpec * pspec)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (object);
+
+ switch (prop_id) {
+ case PROP_MONITOR:
+ g_value_set_int (value, src->monitor);
+ break;
+ case PROP_SHOW_CURSOR:
+ g_value_set_boolean (value, src->show_cursor);
+ break;
+ case PROP_X_POS:
+ g_value_set_int (value, src->capture_x);
+ break;
+ case PROP_Y_POS:
+ g_value_set_int (value, src->capture_y);
+ break;
+ case PROP_WIDTH:
+ g_value_set_int (value, src->capture_w);
+ break;
+ case PROP_HEIGHT:
+ g_value_set_int (value, src->capture_h);
+ break;
+ default:
+ G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
+ break;
+ };
+}
+
+static void
+gst_gdiscreencapsrc_fixate (GstPad * pad, GstCaps * caps)
+{
+ GstStructure *structure;
+
+ structure = gst_caps_get_structure (caps, 0);
+
+ gst_structure_fixate_field_nearest_int (structure, "width", 640);
+ gst_structure_fixate_field_nearest_int (structure, "height", 480);
+ gst_structure_fixate_field_nearest_fraction (structure, "framerate", 30, 1);
+}
+
+static gboolean
+gst_gdiscreencapsrc_set_caps (GstBaseSrc * bsrc, GstCaps * caps)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (bsrc);
+ HWND capture;
+ HDC device;
+ GstStructure *structure;
+ const GValue *framerate;
+ gint red_mask, green_mask, blue_mask;
+ gint bpp;
+
+ structure = gst_caps_get_structure (caps, 0);
+
+ gst_structure_get_int (structure, "red_mask", &red_mask);
+ gst_structure_get_int (structure, "green_mask", &green_mask);
+ gst_structure_get_int (structure, "blue_mask", &blue_mask);
+ if (blue_mask != GST_VIDEO_BYTE1_MASK_24_INT ||
+ green_mask != GST_VIDEO_BYTE2_MASK_24_INT ||
+ red_mask != GST_VIDEO_BYTE3_MASK_24_INT) {
+ GST_ERROR ("Wrong red_,green_,blue_ mask provided. "
+ "We only support RGB and BGR");
+
+ return FALSE;
+ }
+
+ gst_structure_get_int (structure, "bpp", &bpp);
+ if (bpp != 24) {
+ GST_ERROR ("Wrong bpp provided %d. We only support 24 bpp", bpp);
+ return FALSE;
+ }
+
+ src->src_rect = src->screen_rect;
+ if (src->capture_w && src->capture_h) {
+ src->src_rect.left += src->capture_x;
+ src->src_rect.top += src->capture_y;
+ src->src_rect.right = src->src_rect.left + src->capture_w;
+ src->src_rect.bottom = src->src_rect.top + src->capture_h;
+ }
+
+ if (framerate = gst_structure_get_value (structure, "framerate")) {
+ src->rate_numerator = gst_value_get_fraction_numerator (framerate);
+ src->rate_denominator = gst_value_get_fraction_denominator (framerate);
+ }
+
+ src->info.bmiHeader.biSize = sizeof (BITMAPINFOHEADER);
+ src->info.bmiHeader.biWidth = src->src_rect.right - src->src_rect.left;
+ src->info.bmiHeader.biHeight = src->src_rect.top - src->src_rect.bottom;
+ src->info.bmiHeader.biPlanes = 1;
+ src->info.bmiHeader.biBitCount = 24;
+ src->info.bmiHeader.biCompression = BI_RGB;
+ src->info.bmiHeader.biSizeImage = 0;
+ src->info.bmiHeader.biXPelsPerMeter = 0;
+ src->info.bmiHeader.biYPelsPerMeter = 0;
+ src->info.bmiHeader.biClrUsed = 0;
+ src->info.bmiHeader.biClrImportant = 0;
+
+ /* Cleanup first */
+ if (src->hBitmap != INVALID_HANDLE_VALUE)
+ DeleteObject (src->hBitmap);
+
+ if (src->memDC != INVALID_HANDLE_VALUE)
+ DeleteDC (src->memDC);
+
+ /* Allocate */
+ capture = GetDesktopWindow ();
+ device = GetDC (capture);
+ src->hBitmap = CreateDIBSection (device, &(src->info), DIB_RGB_COLORS,
+ (void **) &(src->dibMem), 0, 0);
+ src->memDC = CreateCompatibleDC (device);
+ SelectObject (src->memDC, src->hBitmap);
+ ReleaseDC (capture, device);
+
+ GST_DEBUG_OBJECT (src, "size %dx%d, %d/%d fps",
+ src->info.bmiHeader.biWidth,
+ -src->info.bmiHeader.biHeight,
+ src->rate_numerator, src->rate_denominator);
+
+ return TRUE;
+}
+
+static GstCaps *
+gst_gdiscreencapsrc_get_caps (GstBaseSrc * bsrc)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (bsrc);
+ RECT rect_dst;
+
+ src->screen_rect = rect_dst = gst_win32_get_monitor_rect (src->monitor);
+
+ if (src->capture_w && src->capture_h &&
+ src->capture_x + src->capture_w < rect_dst.right - rect_dst.left &&
+ src->capture_y + src->capture_h < rect_dst.bottom - rect_dst.top) {
+ rect_dst.left = src->capture_x;
+ rect_dst.top = src->capture_y;
+ rect_dst.right = src->capture_x + src->capture_w;
+ rect_dst.bottom = src->capture_y + src->capture_h;
+ } else {
+ /* Default values. */
+ src->capture_x = src->capture_y = 0;
+ src->capture_w = src->capture_h = 0;
+ }
+
+ GST_DEBUG ("width = %d, height=%d",
+ rect_dst.right - rect_dst.left, rect_dst.bottom - rect_dst.top);
+
+ return gst_caps_new_simple ("video/x-raw-rgb",
+ "bpp", G_TYPE_INT, 24,
+ "depth", G_TYPE_INT, 24,
+ "endianness", G_TYPE_INT, G_BIG_ENDIAN,
+ "red_mask", G_TYPE_INT, GST_VIDEO_BYTE3_MASK_24_INT,
+ "green_mask", G_TYPE_INT, GST_VIDEO_BYTE2_MASK_24_INT,
+ "blue_mask", G_TYPE_INT, GST_VIDEO_BYTE1_MASK_24_INT,
+ "width", G_TYPE_INT, rect_dst.right - rect_dst.left,
+ "height", G_TYPE_INT, rect_dst.bottom - rect_dst.top,
+ "framerate", GST_TYPE_FRACTION_RANGE, 1, G_MAXINT, G_MAXINT, 1, NULL);
+}
+
+static gboolean
+gst_gdiscreencapsrc_start (GstBaseSrc * bsrc)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (bsrc);
+
+ src->frames = 0;
+
+ return TRUE;
+}
+
+static gboolean
+gst_gdiscreencapsrc_stop (GstBaseSrc * bsrc)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (bsrc);
+
+ src->frames = 0;
+
+ return TRUE;
+}
+
+static void
+gst_gdiscreencapsrc_get_times (GstBaseSrc * basesrc, GstBuffer * buffer,
+ GstClockTime * start, GstClockTime * end)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (basesrc);
+ GstClockTime timestamp;
+
+ timestamp = GST_BUFFER_TIMESTAMP (buffer);
+
+ if (GST_CLOCK_TIME_IS_VALID (timestamp)) {
+ GstClockTime duration = GST_BUFFER_DURATION (buffer);
+
+ if (GST_CLOCK_TIME_IS_VALID (duration))
+ *end = timestamp + duration;
+ *start = timestamp;
+ }
+}
+
+static GstFlowReturn
+gst_gdiscreencapsrc_create (GstPushSrc * push_src, GstBuffer ** buf)
+{
+ GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (push_src);
+ GstBuffer *new_buf;
+ GstFlowReturn res;
+ gint new_buf_size;
+ GstClock *clock;
+ GstClockTime time;
+ GstClockTime base_time;
+
+ if (G_UNLIKELY (!src->info.bmiHeader.biWidth ||
+ !src->info.bmiHeader.biHeight)) {
+ GST_ELEMENT_ERROR (src, CORE, NEGOTIATION, (NULL),
+ ("format wasn't negotiated before create function"));
+ return GST_FLOW_NOT_NEGOTIATED;
+ } else if (G_UNLIKELY (src->rate_numerator == 0 && src->frames == 1)) {
+ GST_DEBUG_OBJECT (src, "eos: 0 framerate, frame %d", (gint) src->frames);
+ return GST_FLOW_UNEXPECTED;
+ }
+
+ new_buf_size = GST_ROUND_UP_4 (src->info.bmiHeader.biWidth * 3) *
+ (-src->info.bmiHeader.biHeight);
+
+ GST_LOG_OBJECT (src,
+ "creating buffer of %lu bytes with %dx%d image for frame %d",
+ new_buf_size, src->info.bmiHeader.biWidth,
+ -src->info.bmiHeader.biHeight, (gint) src->frames);
+
+ res =
+ gst_pad_alloc_buffer_and_set_caps (GST_BASE_SRC_PAD (src),
+ GST_BUFFER_OFFSET_NONE, new_buf_size,
+ GST_PAD_CAPS (GST_BASE_SRC_PAD (push_src)), &new_buf);
+ if (res != GST_FLOW_OK) {
+ GST_DEBUG_OBJECT (src, "could not allocate buffer, reason %s",
+ gst_flow_get_name (res));
+ return res;
+ }
+
+ clock = gst_element_get_clock (GST_ELEMENT (src));
+ if (clock) {
+ /* Calculate sync time. */
+ GstClockTime frame_time =
+ gst_util_uint64_scale_int (src->frames * GST_SECOND,
+ src->rate_denominator, src->rate_numerator);
+
+ time = gst_clock_get_time (clock);
+ base_time = gst_element_get_base_time (GST_ELEMENT (src));
+ GST_BUFFER_TIMESTAMP (new_buf) = MAX (time - base_time, frame_time);
+ } else {
+ GST_BUFFER_TIMESTAMP (new_buf) = GST_CLOCK_TIME_NONE;
+ }
+
+ /* Do screen capture and put it into buffer... */
+ gst_gdiscreencapsrc_screen_capture (src, new_buf);
+
+ if (src->rate_numerator) {
+ GST_BUFFER_DURATION (new_buf) =
+ gst_util_uint64_scale_int (GST_SECOND,
+ src->rate_denominator, src->rate_numerator);
+ if (clock) {
+ GST_BUFFER_DURATION (new_buf) =
+ MAX (GST_BUFFER_DURATION (new_buf),
+ gst_clock_get_time (clock) - time);
+ }
+ } else {
+ /* NONE means forever */
+ GST_BUFFER_DURATION (new_buf) = GST_CLOCK_TIME_NONE;
+ }
+
+ GST_BUFFER_OFFSET (new_buf) = src->frames;
+ src->frames++;
+ GST_BUFFER_OFFSET_END (new_buf) = src->frames;
+
+ gst_object_unref (clock);
+
+ *buf = new_buf;
+ return GST_FLOW_OK;
+}
+
+static gboolean
+gst_gdiscreencapsrc_screen_capture (GstGDIScreenCapSrc * src, GstBuffer * buf)
+{
+ HWND capture;
+ HDC winDC;
+ gint height, width;
+
+ if (G_UNLIKELY (!src->hBitmap || !src->dibMem))
+ return FALSE;
+
+ width = src->info.bmiHeader.biWidth;
+ height = -src->info.bmiHeader.biHeight;
+
+ /* Capture screen */
+ capture = GetDesktopWindow ();
+ winDC = GetWindowDC (capture);
+
+ BitBlt (src->memDC, 0, 0, width, height,
+ winDC, src->src_rect.left, src->src_rect.top, SRCCOPY);
+
+ ReleaseDC (capture, winDC);
+
+ /* Capture mouse cursor */
+ if (src->show_cursor) {
+ CURSORINFO ci;
+
+ ci.cbSize = sizeof (CURSORINFO);
+ GetCursorInfo (&ci);
+ if (ci.flags & CURSOR_SHOWING) {
+ ICONINFO ii;
+ GetIconInfo (ci.hCursor, &ii);
+
+ DrawIconEx (src->memDC,
+ ci.ptScreenPos.x - src->src_rect.left - ii.xHotspot,
+ ci.ptScreenPos.y - src->src_rect.top - ii.yHotspot, ci.hCursor, 0, 0,
+ 0, NULL, DI_DEFAULTSIZE | DI_NORMAL | DI_COMPAT);
+ }
+ }
+
+ /* Copy DC bits to GST buffer */
+ memcpy (GST_BUFFER_DATA (buf), src->dibMem, GST_BUFFER_SIZE (buf));
+
+ return TRUE;
+}
--- /dev/null
+/* GStreamer
+ * Copyright (C) 2007 Haakon Sporsheim <hakon.sporsheim@tandberg.com>
+ *
+ * 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
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#ifndef __GST_GDISCREENCAPSRC_H__
+#define __GST_GDISCREENCAPSRC_H__
+
+#include <gst/gst.h>
+#include <gst/base/gstpushsrc.h>
+#include <gst/interfaces/propertyprobe.h>
+
+#include "gstwinscreencap.h"
+
+G_BEGIN_DECLS
+
+#define GST_TYPE_GDISCREENCAPSRC (gst_gdiscreencapsrc_get_type())
+#define GST_GDISCREENCAPSRC(obj) \
+ (G_TYPE_CHECK_INSTANCE_CAST ((obj), \
+ GST_TYPE_GDISCREENCAPSRC,GstGDIScreenCapSrc))
+#define GST_GDISCREENCAPSRC_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_CAST ((klass), \
+ GST_TYPE_GDISCREENCAPSRC,GstGDIScreenCapSrcClass))
+#define GST_IS_GDISCREENCAPSRC(obj) \
+ (G_TYPE_CHECK_INSTANCE_TYPE ((obj),GST_TYPE_GDISCREENCAPSRC))
+#define GST_IS_GDISCREENCAPSRC_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_TYPE ((klass),GST_TYPE_GDISCREENCAPSRC))
+
+typedef struct _GstGDIScreenCapSrc GstGDIScreenCapSrc;
+typedef struct _GstGDIScreenCapSrcClass GstGDIScreenCapSrcClass;
+
+struct _GstGDIScreenCapSrc
+{
+ /* Parent */
+ GstPushSrc src;
+
+ /* Properties */
+ gint capture_x;
+ gint capture_y;
+ gint capture_w;
+ gint capture_h;
+ gint monitor;
+ gboolean show_cursor;
+
+ /* Source pad frame rate */
+ gint rate_numerator;
+ gint rate_denominator;
+
+ /* Runtime variables */
+ RECT screen_rect;
+ RECT src_rect;
+ gint64 frames;
+ BITMAPINFO info;
+ BYTE *dibMem;
+ HBITMAP hBitmap;
+ HDC memDC;
+};
+
+struct _GstGDIScreenCapSrcClass
+{
+ GstPushSrcClass parent_class;
+};
+
+GType gst_gdiscreencapsrc_get_type (void);
+
+G_END_DECLS
+
+#endif /* __GST_GDISCREENCAPSRC_H__ */
--- /dev/null
+/* GStreamer
+ * Copyright (C) 2007 Haakon Sporsheim <hakon.sporsheim@tandberg.com>
+ *
+ * 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
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include "gstwinscreencap.h"
+#include "gstgdiscreencapsrc.h"
+#include "gstdx9screencapsrc.h"
+
+static BOOL CALLBACK
+_diplay_monitor_enum (HMONITOR hMon, HDC hdc, LPRECT rect, LPARAM param)
+{
+ LPRECT *pp_rect = (LPRECT *) param;
+ CopyRect (*pp_rect, rect);
+ (*pp_rect)++;
+ return TRUE;
+}
+
+RECT
+gst_win32_get_monitor_rect (UINT index)
+{
+ RECT ret_rect;
+ LPRECT data;
+
+ data = (LPRECT) malloc (sizeof (RECT) * GetSystemMetrics (SM_CMONITORS));
+ if (data) {
+ LPRECT tmp = data;
+ EnumDisplayMonitors (NULL, NULL, _diplay_monitor_enum, (LPARAM) & tmp);
+
+ ret_rect = data[index];
+ free (data);
+ } else {
+ ZeroMemory (&ret_rect, sizeof (RECT));
+ }
+
+ return ret_rect;
+}
+
+static gboolean
+plugin_init (GstPlugin * plugin)
+{
+ if (!gst_element_register (plugin, "gdiscreencapsrc",
+ GST_RANK_NONE, GST_TYPE_GDISCREENCAPSRC)) {
+ return FALSE;
+ }
+
+ if (!gst_element_register (plugin, "dx9screencapsrc",
+ GST_RANK_NONE, GST_TYPE_DX9SCREENCAPSRC)) {
+ return FALSE;
+ }
+
+ return TRUE;
+}
+
+GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
+ GST_VERSION_MINOR,
+ "winscreencap",
+ "Screen capture plugin for Windows",
+ plugin_init, VERSION, GST_LICENSE, GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)
--- /dev/null
+/* GStreamer
+ * Copyright (C) 2007 Haakon Sporsheim <hakon.sporsheim@tandberg.com>
+ *
+ * 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
+ * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
+ */
+
+#ifndef __GST_WINSCREENCAP_H__
+#define __GST_WINSCREENCAP_H__
+
+#include <glib.h>
+#include <windows.h>
+
+G_BEGIN_DECLS
+
+RECT gst_win32_get_monitor_rect (UINT index);
+
+G_END_DECLS
+
+#endif /* __GST_WINSCREENCAP_H__ */
projects / platform / upstream / gstreamer.git / commitdiff