From: <ol...@ke...> - 2008-08-24 13:23:47
|
CVS Root: /cvs/gstreamer Module: gst-plugins-bad Changes by: oleavr Date: Sun Aug 24 2008 13:23:58 UTC Log message: patch by: Haakon Sporsheim <hak...@ta...> * 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). Modified files: . : ChangeLog configure.ac sys : Makefile.am Added files: sys/winscreencap: Makefile.am gstdx9screencapsrc.c gstdx9screencapsrc.h gstgdiscreencapsrc.c gstgdiscreencapsrc.h gstwinscreencap.c gstwinscreencap.h Links: http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/ChangeLog.diff?r1=1.3502&r2=1.3503 http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/configure.ac.diff?r1=1.881&r2=1.882 http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/sys/Makefile.am.diff?r1=1.59&r2=1.60 http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/sys/winscreencap/Makefile.am?rev=1.1&content-type=text/vnd.viewcvs-markup http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/sys/winscreencap/gstdx9screencapsrc.c?rev=1.1&content-type=text/vnd.viewcvs-markup http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/sys/winscreencap/gstdx9screencapsrc.h?rev=1.1&content-type=text/vnd.viewcvs-markup http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/sys/winscreencap/gstgdiscreencapsrc.c?rev=1.1&content-type=text/vnd.viewcvs-markup http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/sys/winscreencap/gstgdiscreencapsrc.h?rev=1.1&content-type=text/vnd.viewcvs-markup http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/sys/winscreencap/gstwinscreencap.c?rev=1.1&content-type=text/vnd.viewcvs-markup http://freedesktop.org/cgi-bin/viewcvs.cgi/gstreamer/gst-plugins-bad/sys/winscreencap/gstwinscreencap.h?rev=1.1&content-type=text/vnd.viewcvs-markup ====Begin Diffs==== Index: ChangeLog =================================================================== RCS file: /cvs/gstreamer/gst-plugins-bad/ChangeLog,v retrieving revision 1.3502 retrieving revision 1.3503 diff -u -d -r1.3502 -r1.3503 --- ChangeLog 23 Aug 2008 00:01:37 -0000 1.3502 +++ ChangeLog 24 Aug 2008 13:23:41 -0000 1.3503 @@ -1,3 +1,18 @@ +2008-08-24 Ole André Vadla Ravnås <ole...@ta...> + + patch by: Haakon Sporsheim <hak...@ta...> + * 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 <ms...@so...> * sys/dshowdecwrapper/gstdshowaudiodec.c: Index: configure.ac RCS file: /cvs/gstreamer/gst-plugins-bad/configure.ac,v retrieving revision 1.881 retrieving revision 1.882 diff -u -d -r1.881 -r1.882 --- configure.ac 21 Aug 2008 13:22:28 -0000 1.881 +++ configure.ac 24 Aug 2008 13:23:42 -0000 1.882 @@ -1296,6 +1296,7 @@ sys/qtwrapper/Makefile sys/vcd/Makefile sys/wininet/Makefile +sys/winscreencap/Makefile examples/Makefile examples/app/Makefile examples/directfb/Makefile Index: Makefile.am RCS file: /cvs/gstreamer/gst-plugins-bad/sys/Makefile.am,v retrieving revision 1.59 retrieving revision 1.60 diff -u -d -r1.59 -r1.60 --- Makefile.am 20 Aug 2008 21:44:22 -0000 1.59 +++ Makefile.am 24 Aug 2008 13:23:42 -0000 1.60 @@ -61,5 +61,5 @@ 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 --- NEW FILE: Makefile.am --- # 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 --- NEW FILE: gstdx9screencapsrc.c --- /* GStreamer * Copyright (C) 2007 Haakon Sporsheim <hak...@ta...> * * 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 <hak...@ta...>" }; #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); } 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 " g_object_class_install_property (go_class, PROP_WIDTH, g_param_spec_int ("width", "Width", "Width of screen capture area (0 = maximum)", g_object_class_install_property (go_class, PROP_HEIGHT, g_param_spec_int ("height", "Height", "Height of screen capture area (0 = maximum)", GST_DEBUG_CATEGORY_INIT (dx9screencapsrc_debug, "dx9screencapsrc", 0, "DirectX 9 screen capture source"); 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); 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; gst_dx9screencapsrc_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) 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); case PROP_Y_POS: src->capture_y = g_value_get_int (value); case PROP_WIDTH: src->capture_w = g_value_get_int (value); case PROP_HEIGHT: src->capture_h = g_value_get_int (value); default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); }; gst_dx9screencapsrc_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec) g_value_set_int (value, src->monitor); g_value_set_int (value, src->capture_x); g_value_set_int (value, src->capture_y); g_value_set_int (value, src->capture_w); g_value_set_int (value, src->capture_h); 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); const GValue *framerate; 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) 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); 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; case D3DFMT_X8R8G8B8: bpp = 32; depth = 24; alpha = 0; case D3DFMT_R8G8B8: bpp = depth = 24; red = GST_VIDEO_BYTE1_MASK_24_INT; green = GST_VIDEO_BYTE2_MASK_24_INT; blue = GST_VIDEO_BYTE3_MASK_24_INT; 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; case D3DFMT_X1R5G5B5: case D3DFMT_R5G6B5: bpp = depth = 16; red = GST_VIDEO_RED_MASK_16_INT; green = GST_VIDEO_GREEN_MASK_16_INT; blue = GST_VIDEO_BLUE_MASK_16_INT; 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); gst_dx9screencapsrc_start (GstBaseSrc * 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; 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)); gst_dx9screencapsrc_stop (GstBaseSrc * bsrc) 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); 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."); IDirect3DSurface9_LockRect (src->surface, &locked_rect, &(src->src_rect), D3DLOCK_NO_DIRTY_UPDATE | D3DLOCK_NOSYSLOCK | D3DLOCK_READONLY); GST_DEBUG_OBJECT (src, "DirectX::LockRect failed."); 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, if (clock) { GST_BUFFER_DURATION (new_buf) = MAX (GST_BUFFER_DURATION (new_buf), gst_clock_get_time (clock) - time); } 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; --- NEW FILE: gstdx9screencapsrc.h --- #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__ */ --- NEW FILE: gstgdiscreencapsrc.c --- #include "gstgdiscreencapsrc.h" GST_DEBUG_CATEGORY_STATIC (gdiscreencapsrc_debug); static const GstElementDetails gst_gdiscreencapsrc_details = { "GDI screen capture source", 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, PROP_SHOW_CURSOR, 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, 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, static GstFlowReturn gst_gdiscreencapsrc_create (GstPushSrc * src, static gboolean gst_gdiscreencapsrc_screen_capture (GstGDIScreenCapSrc * src, GstBuffer * buf); gst_gdiscreencapsrc_base_init (gpointer klass) gst_element_class_set_details (element_class, &gst_gdiscreencapsrc_details); gst_gdiscreencapsrc_class_init (GstGDIScreenCapSrcClass * 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_param_spec_int ("monitor", "Monitor", "Which monitor to use (0 = 1st monitor and default)", 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)); GST_DEBUG_CATEGORY_INIT (gdiscreencapsrc_debug, "gdiscreencapsrc", 0, "GDI screen capture source"); gst_gdiscreencapsrc_init (GstGDIScreenCapSrc * src, GstGDIScreenCapSrcClass * klass) gst_pad_set_fixatecaps_function (src_pad, gst_gdiscreencapsrc_fixate); src->dibMem = NULL; src->hBitmap = (HBITMAP) INVALID_HANDLE_VALUE; src->memDC = (HDC) INVALID_HANDLE_VALUE; src->show_cursor = FALSE; 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); gst_gdiscreencapsrc_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) case PROP_SHOW_CURSOR: src->show_cursor = g_value_get_boolean (value); gst_gdiscreencapsrc_get_property (GObject * object, guint prop_id, g_value_set_boolean (value, src->show_cursor); gst_gdiscreencapsrc_fixate (GstPad * pad, GstCaps * caps) gst_gdiscreencapsrc_set_caps (GstBaseSrc * bsrc, GstCaps * caps) GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (bsrc); HWND capture; HDC device; gint red_mask, green_mask, blue_mask; gint bpp; 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"); gst_structure_get_int (structure, "bpp", &bpp); if (bpp != 24) { GST_ERROR ("Wrong bpp provided %d. We only support 24 bpp", bpp); 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 */ /* 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); src->info.bmiHeader.biWidth, -src->info.bmiHeader.biHeight, gst_gdiscreencapsrc_get_caps (GstBaseSrc * bsrc) src->screen_rect = rect_dst = gst_win32_get_monitor_rect (src->monitor); /* Default values. */ GST_DEBUG ("width = %d, height=%d", "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); gst_gdiscreencapsrc_start (GstBaseSrc * bsrc) gst_gdiscreencapsrc_stop (GstBaseSrc * bsrc) gst_gdiscreencapsrc_get_times (GstBaseSrc * basesrc, GstBuffer * buffer, GstClockTime * start, GstClockTime * end) GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (basesrc); gst_gdiscreencapsrc_create (GstPushSrc * push_src, GstBuffer ** buf) GstGDIScreenCapSrc *src = GST_GDISCREENCAPSRC (push_src); gint new_buf_size; GstClockTime time; GstClockTime base_time; if (G_UNLIKELY (!src->info.bmiHeader.biWidth || !src->info.bmiHeader.biHeight)) { } else if (G_UNLIKELY (src->rate_numerator == 0 && src->frames == 1)) { new_buf_size = GST_ROUND_UP_4 (src->info.bmiHeader.biWidth * 3) * (-src->info.bmiHeader.biHeight); 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), if (clock) { GST_BUFFER_TIMESTAMP (new_buf) = MAX (time - base_time, frame_time); GST_BUFFER_TIMESTAMP (new_buf) = GST_CLOCK_TIME_NONE; /* Do screen capture and put it into buffer... */ gst_gdiscreencapsrc_screen_capture (src, new_buf); GST_BUFFER_DURATION (new_buf) = MAX (GST_BUFFER_DURATION (new_buf), /* NONE means forever */ gst_object_unref (clock); gst_gdiscreencapsrc_screen_capture (GstGDIScreenCapSrc * src, GstBuffer * buf) HDC winDC; gint height, width; if (G_UNLIKELY (!src->hBitmap || !src->dibMem)) width = src->info.bmiHeader.biWidth; height = -src->info.bmiHeader.biHeight; /* Capture screen */ 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)); --- NEW FILE: gstgdiscreencapsrc.h --- #ifndef __GST_GDISCREENCAPSRC_H__ #define __GST_GDISCREENCAPSRC_H__ #include <gst/interfaces/propertyprobe.h> #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 gint monitor; gboolean show_cursor; BITMAPINFO info; BYTE *dibMem; HBITMAP hBitmap; HDC memDC; struct _GstGDIScreenCapSrcClass GType gst_gdiscreencapsrc_get_type (void); #endif /* __GST_GDISCREENCAPSRC_H__ */ --- NEW FILE: gstwinscreencap.c --- 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)++; 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); ZeroMemory (&ret_rect, sizeof (RECT)); return ret_rect; plugin_init (GstPlugin * plugin) if (!gst_element_register (plugin, "gdiscreencapsrc", GST_RANK_NONE, GST_TYPE_GDISCREENCAPSRC)) { if (!gst_element_register (plugin, "dx9screencapsrc", GST_RANK_NONE, GST_TYPE_DX9SCREENCAPSRC)) { 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) --- NEW FILE: gstwinscreencap.h --- #ifndef __GST_WINSCREENCAP_H__ #define __GST_WINSCREENCAP_H__ #include <glib.h> #include <windows.h> RECT gst_win32_get_monitor_rect (UINT index); #endif /* __GST_WINSCREENCAP_H__ */ |