/* GStreamer * Copyright (C) 2012 Roland Krikava * Copyright (C) 2010-2011 David Hoyt * Copyright (C) 2010 Andoni Morales * * 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 "d3dvideosink.h" #include "d3dhelpers.h" #include /** FWD DECLS **/ static gboolean d3d_hidden_window_thread (GstD3DVideoSinkClass * class); static gboolean d3d_window_wndproc_set (GstD3DVideoSink * sink); static void d3d_window_wndproc_unset (GstD3DVideoSink * sink); static gboolean d3d_init_swap_chain (GstD3DVideoSink * sink, HWND hWnd); static gboolean d3d_release_swap_chain (GstD3DVideoSink * sink); static gboolean d3d_resize_swap_chain (GstD3DVideoSink * sink); static gboolean d3d_present_swap_chain (GstD3DVideoSink * sink); static gboolean d3d_copy_buffer_to_surface (GstD3DVideoSink * sink, GstBuffer * buffer); static gboolean d3d_stretch_and_copy (GstD3DVideoSink * sink, LPDIRECT3DSURFACE9 back_buffer); static HWND d3d_create_internal_window (GstD3DVideoSink * sink); static void d3d_class_notify_device_lost (GstD3DVideoSink * sink); static LRESULT APIENTRY d3d_wnd_proc_internal (HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam); static LRESULT APIENTRY d3d_wnd_proc (HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam); GST_DEBUG_CATEGORY_EXTERN (gst_d3dvideosink_debug); #define GST_CAT_DEFAULT gst_d3dvideosink_debug static gint WM_D3DVIDEO_NOTIFY_DEVICE_LOST = 0; #define IDT_DEVICE_RESET_TIMER 0 /** Helpers **/ #define ERROR_CHECK_HR(hr) \ if(hr != S_OK) { \ const gchar * str_err=NULL, *t1=NULL; \ gchar tmp[128]=""; \ switch(hr) #define CASE_HR_ERR(hr_err) \ case hr_err: str_err = #hr_err; break; #define CASE_HR_DBG_ERR_END(sink, gst_err_msg, level) \ default: \ t1=gst_err_msg; \ sprintf(tmp, "HR-SEV:%u HR-FAC:%u HR-CODE:%u", (guint)HRESULT_SEVERITY(hr), (guint)HRESULT_FACILITY(hr), (guint)HRESULT_CODE(hr)); \ str_err = tmp; \ } /* end switch */ \ GST_CAT_LEVEL_LOG(GST_CAT_DEFAULT, level, sink, "%s HRESULT: %s", t1?t1:"", str_err); #define CASE_HR_ERR_END(sink, gst_err_msg) \ CASE_HR_DBG_ERR_END(sink, gst_err_msg, GST_LEVEL_ERROR) #define CASE_HR_DBG_END(sink, gst_err_msg) \ CASE_HR_DBG_ERR_END(sink, gst_err_msg, GST_LEVEL_DEBUG) #define CHECK_D3D_DEVICE(class, sink, goto_label) \ if(!class->d3d.d3d || !class->d3d.device.d3d_device) { \ GST_ERROR_OBJECT(sink, "Direct3D device or object does not exist"); \ goto goto_label; \ } #define CHECK_D3D_SWAPCHAIN(sink, goto_label) \ if(!sink->d3d.swapchain) { \ GST_ERROR_OBJECT(sink, "Direct3D swap chain does not exist"); \ goto goto_label; \ } #define CHECK_D3D_SURFACE(sink, goto_label) \ if(!sink->d3d.surface) { \ GST_ERROR_OBJECT(sink, "NULL D3D offscreen surface"); \ goto goto_label; \ } #define CHECK_WINDOW_HANDLE(sink, goto_label, is_error) \ if(!sink->d3d.window_handle) { \ GST_CAT_LEVEL_LOG(GST_CAT_DEFAULT, \ (is_error?GST_LEVEL_ERROR:GST_LEVEL_DEBUG), \ sink, "No window handle is set"); \ goto goto_label; \ } #ifndef D3DFMT_YV12 #define D3DFMT_YV12 MAKEFOURCC ('Y', 'V', '1', '2') #endif #ifndef D3DFMT_NV12 #define D3DFMT_NV12 MAKEFOURCC ('N', 'V', '1', '2') #endif #define BIT_MAX(bits) ((1<d3d.d3d, class->d3d.device.adapter, D3DDEVTYPE_HAL, class->d3d.device.format, 0, D3DRTYPE_SURFACE, fmt); if (hr == D3D_OK) { /* test whether device can perform color-conversion ** from that format to target format */ hr = IDirect3D9_CheckDeviceFormatConversion (class->d3d.d3d, class->d3d.device.adapter, D3DDEVTYPE_HAL, fmt, class->d3d.device.format); if (hr == D3D_OK) ret = TRUE; } //GST_DEBUG_OBJECT(sink, "Checking: %s - %s", d3d_format2string(fmt), ret?"TRUE":"FALSE"); return ret; } static D3DFORMAT gst_video_query_d3d_format (GstD3DVideoSink * sink, GstVideoFormat format, gboolean * exact) { D3DFORMAT ret = D3DFMT_UNKNOWN; const GstVideoFormatInfo *info; g_return_val_if_fail (exact != NULL, D3DFMT_UNKNOWN); *exact = FALSE; if (!(info = gst_video_format_get_info (format))) { GST_ERROR_OBJECT (sink, "Failed to get GstVideoFormatInfo for format: %u", format); return D3DFMT_UNKNOWN; } if (GST_VIDEO_FORMAT_INFO_IS_RGB (info)) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); D3DFORMAT try1 = D3DFMT_UNKNOWN, try2 = D3DFMT_UNKNOWN; if (gst_video_can_handle_gst_rgb (format)) { try1 = gst_rgb_video_format_exact_d3d (format, &try2); if (try1 != D3DFMT_UNKNOWN && gst_video_d3d_format_check (sink, try1)) { ret = try1; *exact = TRUE; } else if (try2 != D3DFMT_UNKNOWN && gst_video_d3d_format_check (sink, try2)) { ret = try2; } /* Fall back to display adapter format */ if (ret == D3DFMT_UNKNOWN) { if (gst_video_can_handle_d3d_rgb (class->d3d.device.format)) { ret = class->d3d.device.format; } } } } else if (GST_VIDEO_FORMAT_INFO_IS_YUV (info)) { switch (format) { case GST_VIDEO_FORMAT_YV12: case GST_VIDEO_FORMAT_I420: if (gst_video_d3d_format_check (sink, D3DFMT_YV12)) ret = D3DFMT_YV12; break; case GST_VIDEO_FORMAT_YUY2: if (gst_video_d3d_format_check (sink, D3DFMT_YUY2)) { ret = D3DFMT_YUY2; } break; case GST_VIDEO_FORMAT_UYVY: if (gst_video_d3d_format_check (sink, D3DFMT_UYVY)) { ret = D3DFMT_UYVY; } break; case GST_VIDEO_FORMAT_NV12: if (gst_video_d3d_format_check (sink, D3DFMT_NV12)) { ret = D3DFMT_NV12; } break; default:; } } return ret; } typedef struct { GstVideoFormat fmt; D3DFORMAT d3d_fmt; gboolean exact; gboolean display; } GstFormatComp; static gint format_points (GstFormatComp * dat) { gint points = 0; const GstVideoFormatInfo *info; if (!(info = gst_video_format_get_info (dat->fmt))) return 0; if (dat->display) points += 1; else if (dat->exact) points += 2; if (dat->exact && dat->display) points += 10; if (GST_VIDEO_FORMAT_INFO_IS_YUV (info)) points += 5; else if (GST_VIDEO_FORMAT_INFO_IS_RGB (info)) { guint i, bit_depth = 0; for (i = 0; i < GST_VIDEO_FORMAT_INFO_N_COMPONENTS (info); i++) bit_depth += GST_VIDEO_FORMAT_INFO_DEPTH (info, i); if (bit_depth >= 24) points += 1; } return points; } static gint format_compare_cb (gconstpointer a, gconstpointer b) { gint ptsa = 0, ptsb = 0; ptsa = format_points ((GstFormatComp *) a); ptsb = format_points ((GstFormatComp *) b); if (ptsa < ptsb) return -1; else if (ptsa == ptsb) return 0; else return 1; } static GstCaps * d3dvideosink_format_new_template_caps (GstVideoFormat fmt) { gchar *tmp = g_strdup_printf ("video/x-raw, " "format=(string)%s, " "width=(int)[ 1, 2147483647 ], " "height=(int)[ 1, 2147483647 ], " "framerate=(fraction)[ 0/1, 2147483647/1 ]", gst_video_format_to_string (fmt)); //GST_DEBUG("CAPS: %s", tmp); return gst_caps_from_string (tmp); } GstCaps * d3d_supported_caps (GstD3DVideoSink * sink) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); int i; gboolean exact; GList *lst = NULL, *lsto = NULL; GstCaps *caps = NULL; D3DFORMAT d3d_format; gchar *tmp = NULL; LOCK_SINK (sink); if (sink->supported_caps) { caps = gst_caps_ref (sink->supported_caps); goto unlock; } for (i = 0; i <= GST_VIDEO_FORMAT_GBR_10LE; i++) { GstFormatComp *dat; d3d_format = gst_video_query_d3d_format (sink, (GstVideoFormat) i, &exact); if (d3d_format == D3DFMT_UNKNOWN) continue; dat = g_new0 (GstFormatComp, 1); dat->fmt = (GstVideoFormat) i; dat->d3d_fmt = d3d_format; dat->exact = exact; dat->display = (d3d_format == class->d3d.device.format); lst = g_list_insert_sorted (lst, dat, format_compare_cb); } GST_DEBUG_OBJECT (sink, "Supported Caps:"); lsto = lst; for (lst = g_list_last (lst); lst != NULL; lst = g_list_previous (lst)) { GstFormatComp *dat = (GstFormatComp *) lst->data; GST_DEBUG_OBJECT (sink, "%s -> %s %s%s", gst_video_format2string (dat->fmt), d3d_format2string (dat->d3d_fmt), dat->exact ? "[exact]" : "", dat->display ? "[display]" : ""); if (!caps) caps = d3dvideosink_format_new_template_caps (dat->fmt); else gst_caps_append (caps, d3dvideosink_format_new_template_caps (dat->fmt)); } g_list_foreach (lsto, (GFunc) g_free, NULL); g_list_free (lsto); sink->supported_caps = gst_caps_ref (caps); GST_DEBUG_OBJECT (sink, "Supported caps: %s", (tmp = gst_caps_to_string (caps))); g_free (tmp); unlock: UNLOCK_SINK (sink); return caps; } gboolean d3d_set_render_format (GstD3DVideoSink * sink) { D3DFORMAT fmt; gboolean exact = FALSE; gboolean ret = FALSE; LOCK_SINK (sink); fmt = gst_video_query_d3d_format (sink, sink->format, &exact); if (fmt == D3DFMT_UNKNOWN) { GST_ERROR_OBJECT (sink, "Failed to query a D3D render format for %s", gst_video_format2string (sink->format)); goto end; } GST_DEBUG_OBJECT (sink, "Selected %s -> %s %s", gst_video_format2string (sink->format), d3d_format2string (fmt), exact ? "(exact)" : ""); sink->d3d.format = fmt; sink->d3d.exact_copy = exact; ret = TRUE; end: UNLOCK_SINK (sink); return ret; } static gboolean d3d_get_hwnd_window_size (HWND hwnd, gint * width, gint * height) { RECT sz; g_return_val_if_fail (width != NULL, FALSE); g_return_val_if_fail (height != NULL, FALSE); *width = 0; *height = 0; if (!hwnd) return FALSE; GetClientRect (hwnd, &sz); *width = MAX (1, ABS (sz.right - sz.left)); *height = MAX (1, ABS (sz.bottom - sz.top)); return TRUE; } static gboolean d3d_get_render_rects (GstVideoRectangle * rr, RECT * dst, RECT * src) { if (!rr) return FALSE; /* Rect on target */ if (dst) { dst->left = rr->x; dst->top = rr->y; dst->right = rr->x + rr->w; dst->bottom = rr->y + rr->h; } /* Rect on source */ if (src) { src->left = 0; src->top = 0; src->right = rr->w; src->bottom = rr->h; } return TRUE; } static gboolean d3d_get_render_coordinates (GstD3DVideoSink * sink, gint in_x, gint in_y, gdouble * out_x, gdouble * out_y) { GstVideoRectangle r_area; gdouble tmp; gboolean ret = FALSE; g_return_val_if_fail (out_x != NULL, FALSE); g_return_val_if_fail (out_y != NULL, FALSE); LOCK_SINK (sink); CHECK_WINDOW_HANDLE (sink, end, FALSE); /* Get renderable area of the window */ if (sink->d3d.render_rect) { memcpy (&r_area, sink->d3d.render_rect, sizeof (r_area)); } else { memset (&r_area, 0, sizeof (r_area)); d3d_get_hwnd_window_size (sink->d3d.window_handle, &r_area.w, &r_area.h); } /* If window coords outside render area.. return */ if (in_x < r_area.x || in_x > r_area.x + r_area.w || in_y < r_area.y || in_y > r_area.y + r_area.h) goto end; /* Convert window coordinates to source frame pixel coordinates */ if (sink->force_aspect_ratio) { GstVideoRectangle tmp = { 0, 0, 0, 0 } , dst = { 0, 0, 0, 0}; tmp.w = GST_VIDEO_SINK_WIDTH (sink); tmp.h = GST_VIDEO_SINK_HEIGHT (sink); gst_video_sink_center_rect (tmp, r_area, &dst, TRUE); r_area.x = r_area.x + dst.x; r_area.y = r_area.y + dst.y; r_area.w = dst.w; r_area.h = dst.h; /* If window coords outside render area.. return */ if (in_x < r_area.x || in_x > (r_area.x + r_area.w) || in_y < r_area.y || in_y > (r_area.y + r_area.h)) goto end; } tmp = in_x - r_area.x; if (r_area.w == GST_VIDEO_SINK_WIDTH (sink)) *out_x = tmp; else if (r_area.w > GST_VIDEO_SINK_WIDTH (sink)) *out_x = ((gdouble) tmp / ((gdouble) r_area.w / (gdouble) GST_VIDEO_SINK_WIDTH (sink))); else *out_x = ((gdouble) GST_VIDEO_SINK_WIDTH (sink) / (gdouble) r_area.w) * (gdouble) tmp; tmp = in_y - r_area.y; if (r_area.h == GST_VIDEO_SINK_HEIGHT (sink)) *out_y = tmp; else if (r_area.h > GST_VIDEO_SINK_HEIGHT (sink)) *out_y = ((gdouble) tmp / ((gdouble) r_area.h / (gdouble) GST_VIDEO_SINK_HEIGHT (sink))); else *out_y = ((gdouble) GST_VIDEO_SINK_HEIGHT (sink) / (gdouble) r_area.h) * (gdouble) tmp; ret = TRUE; end: UNLOCK_SINK (sink); return ret; } /** Windows for rendering (User Set or Internal) **/ static void d3d_window_wndproc_unset (GstD3DVideoSink * sink) { WNDPROC cur_wnd_proc = NULL; LOCK_SINK (sink); GST_DEBUG_OBJECT (sink, " "); if (sink->d3d.window_handle == NULL) { GST_WARNING_OBJECT (sink, "D3D window_handle is NULL"); goto end; } cur_wnd_proc = (WNDPROC) GetWindowLongPtr (sink->d3d.window_handle, GWLP_WNDPROC); if (cur_wnd_proc != d3d_wnd_proc) { GST_WARNING_OBJECT (sink, "D3D window proc is not set on current window"); goto end; } if (sink->d3d.orig_wnd_proc == NULL) { GST_WARNING_OBJECT (sink, "D3D orig window proc is NULL, can not restore"); goto end; } /* Restore orignal WndProc for window_handle */ if (!SetWindowLongPtr (sink->d3d.window_handle, GWLP_WNDPROC, (LONG_PTR) sink->d3d.orig_wnd_proc)) { GST_WARNING_OBJECT (sink, "D3D failed to set original WndProc"); goto end; } end: sink->d3d.orig_wnd_proc = NULL; sink->d3d.window_handle = NULL; UNLOCK_SINK (sink); } static gboolean d3d_window_wndproc_set (GstD3DVideoSink * sink) { WNDPROC cur_wnd_proc; gboolean ret = FALSE; LOCK_SINK (sink); cur_wnd_proc = (WNDPROC) GetWindowLongPtr (sink->d3d.window_handle, GWLP_WNDPROC); if (cur_wnd_proc != NULL && cur_wnd_proc == d3d_wnd_proc) { GST_DEBUG_OBJECT (sink, "D3D window proc func is already set on the current window"); ret = TRUE; goto end; } /* Store the original window proc function */ sink->d3d.orig_wnd_proc = (WNDPROC) SetWindowLongPtr (sink->d3d.window_handle, GWLP_WNDPROC, (LONG_PTR) d3d_wnd_proc); /* Note: If the window belongs to another process this will fail */ if (sink->d3d.orig_wnd_proc == NULL) { GST_ERROR_OBJECT (sink, "Failed to set WndProc function on window. Error: %d", (gint) GetLastError ()); goto end; } /* Make sink accessible to d3d_wnd_proc */ SetProp (sink->d3d.window_handle, TEXT ("GstD3DVideoSink"), sink); ret = TRUE; end: UNLOCK_SINK (sink); return ret; } static void d3d_prepare_render_window (GstD3DVideoSink * sink) { LOCK_SINK (sink); if (sink->d3d.window_handle == NULL) { GST_DEBUG_OBJECT (sink, "No window handle has been set.."); goto end; } if (d3d_init_swap_chain (sink, sink->d3d.window_handle)) { d3d_window_wndproc_set (sink); sink->d3d.renderable = TRUE; GST_DEBUG_OBJECT (sink, "Prepared window for render [HWND:%p]", sink->d3d.window_handle); } else { GST_ERROR_OBJECT (sink, "Failed preparing window for render [HWND:%p]", sink->d3d.window_handle); } end: UNLOCK_SINK (sink); } void d3d_set_window_handle (GstD3DVideoSink * sink, guintptr window_id, gboolean is_internal) { LOCK_SINK (sink); if (sink->d3d.window_handle == (HWND) window_id) { GST_WARNING_OBJECT (sink, "Window HWND already set to: %" G_GUINTPTR_FORMAT, window_id); goto end; } /* Unset current window */ if (sink->d3d.window_handle != NULL) { GST_DEBUG_OBJECT (sink, "Unsetting window [HWND:%p]", sink->d3d.window_handle); d3d_window_wndproc_unset (sink); d3d_release_swap_chain (sink); sink->d3d.window_handle = NULL; sink->d3d.window_is_internal = FALSE; sink->d3d.renderable = FALSE; } /* Set new one */ if (window_id) { sink->d3d.window_handle = (HWND) window_id; sink->d3d.window_is_internal = is_internal; /* If caps have been set.. prepare window */ if (sink->format != 0) d3d_prepare_render_window (sink); } end: UNLOCK_SINK (sink); } void d3d_set_render_rectangle (GstD3DVideoSink * sink) { LOCK_SINK (sink); /* Setting the pointer lets us know render rect is set */ sink->d3d.render_rect = &sink->render_rect; d3d_resize_swap_chain (sink); d3d_present_swap_chain (sink); UNLOCK_SINK (sink); } void d3d_expose_window (GstD3DVideoSink * sink) { GST_DEBUG_OBJECT (sink, "EXPOSE"); d3d_present_swap_chain (sink); } gboolean d3d_prepare_window (GstD3DVideoSink * sink) { HWND hWnd; gboolean ret = FALSE; LOCK_SINK (sink); /* Give the app a last chance to set a window id */ if (!sink->d3d.window_handle) gst_video_overlay_prepare_window_handle (GST_VIDEO_OVERLAY (sink)); /* If the user did not set a window id .. check if we should create one */ if (!sink->d3d.window_handle) { if (sink->create_internal_window) { if ((hWnd = d3d_create_internal_window (sink))) { GST_DEBUG_OBJECT (sink, "No window id was set.. creating internal window"); d3d_set_window_handle (sink, (guintptr) hWnd, TRUE); } else { GST_ERROR_OBJECT (sink, "Failed to create internal window"); goto end; } } else { GST_DEBUG_OBJECT (sink, "No window id is set.."); goto end; } } else { d3d_prepare_render_window (sink); } ret = TRUE; end: UNLOCK_SINK (sink); return ret; } gboolean d3d_stop (GstD3DVideoSink * sink) { /* Release D3D resources */ d3d_set_window_handle (sink, 0, FALSE); return TRUE; } /** D3D Lost and Reset Device **/ static void d3d_notify_device_lost (GstD3DVideoSink * sink) { gboolean notify = FALSE; g_return_if_fail (GST_IS_D3DVIDEOSINK (sink)); LOCK_SINK (sink); if (!sink->d3d.device_lost) { GST_WARNING_OBJECT (sink, "D3D Device has been lost. Cleanup up resources.."); /* Stream will continue with GST_FLOW_OK, until device has been reset */ sink->d3d.device_lost = TRUE; /* First we clean up all resources in this d3dvideo instance */ d3d_release_swap_chain (sink); /* Notify our hidden thread */ notify = TRUE; } UNLOCK_SINK (sink); if (notify) d3d_class_notify_device_lost (sink); } static void d3d_notify_device_reset (GstD3DVideoSink * sink) { LOCK_SINK (sink); if (sink->d3d.device_lost) { GST_DEBUG_OBJECT (sink, "D3D Device has been reset. Re-init swap chain if still streaming"); /* If we're still streaming.. reset swap chain */ if (sink->d3d.window_handle != NULL) d3d_init_swap_chain (sink, sink->d3d.window_handle); sink->d3d.device_lost = FALSE; } UNLOCK_SINK (sink); } /** Swap Chains **/ static gboolean d3d_init_swap_chain (GstD3DVideoSink * sink, HWND hWnd) { D3DPRESENT_PARAMETERS present_params; LPDIRECT3DSWAPCHAIN9 d3d_swapchain = NULL; LPDIRECT3DSURFACE9 d3d_surface = NULL; D3DTEXTUREFILTERTYPE d3d_filtertype; HRESULT hr; GstD3DVideoSinkClass *class; gboolean ret = FALSE; g_return_val_if_fail (sink != NULL, FALSE); class = GST_D3DVIDEOSINK_GET_CLASS (sink); g_return_val_if_fail (class != NULL, FALSE); LOCK_SINK (sink); LOCK_CLASS (sink, class); /* We need a display device */ CHECK_D3D_DEVICE (class, sink, error); GST_DEBUG ("Initializing Direct3D swap chain"); GST_DEBUG ("Direct3D back buffer size: %dx%d", GST_VIDEO_SINK_WIDTH (sink), GST_VIDEO_SINK_HEIGHT (sink)); /* When windowed, width and height determined by HWND */ ZeroMemory (&present_params, sizeof (present_params)); present_params.Windowed = TRUE; present_params.SwapEffect = D3DSWAPEFFECT_DISCARD; //D3DSWAPEFFECT_COPY present_params.hDeviceWindow = hWnd; present_params.BackBufferFormat = class->d3d.device.format; //d3d_format; hr = IDirect3DDevice9_CreateAdditionalSwapChain (class->d3d.device.d3d_device, &present_params, &d3d_swapchain); ERROR_CHECK_HR (hr) { CASE_HR_ERR (D3DERR_NOTAVAILABLE); CASE_HR_ERR (D3DERR_DEVICELOST); CASE_HR_ERR (D3DERR_INVALIDCALL); CASE_HR_ERR (D3DERR_OUTOFVIDEOMEMORY); CASE_HR_ERR (E_OUTOFMEMORY); CASE_HR_ERR_END (sink, "Error creating D3D swapchian"); goto error; } hr = IDirect3DDevice9_CreateOffscreenPlainSurface (class->d3d. device.d3d_device, GST_VIDEO_SINK_WIDTH (sink), GST_VIDEO_SINK_HEIGHT (sink), sink->d3d.format, D3DPOOL_DEFAULT, &d3d_surface, NULL); if (hr != D3D_OK) { GST_ERROR_OBJECT (sink, "Failed to create D3D surface"); goto error; } /* Determine texture filtering support. If it's supported for this format, * use the filter type determined when we created the dev and checked the * dev caps. */ hr = IDirect3D9_CheckDeviceFormat (class->d3d.d3d, class->d3d.device.adapter, D3DDEVTYPE_HAL, class->d3d.device.format, D3DUSAGE_QUERY_FILTER, D3DRTYPE_TEXTURE, sink->d3d.format); if (hr == D3D_OK) d3d_filtertype = class->d3d.device.filter_type; else d3d_filtertype = D3DTEXF_NONE; GST_DEBUG ("Direct3D stretch rect texture filter: %d", d3d_filtertype); sink->d3d.filtertype = d3d_filtertype; sink->d3d.swapchain = d3d_swapchain; sink->d3d.surface = d3d_surface; ret = TRUE; error: if (!ret) { if (d3d_swapchain) IDirect3DSwapChain9_Release (d3d_swapchain); if (d3d_surface) IDirect3DSurface9_Release (d3d_surface); } UNLOCK_CLASS (sink, class); UNLOCK_SINK (sink); return ret; } static gboolean d3d_release_swap_chain (GstD3DVideoSink * sink) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); int ref_count; gboolean ret = FALSE; LOCK_SINK (sink); GST_DEBUG_OBJECT (sink, "Releasing Direct3D swap chain"); CHECK_D3D_DEVICE (class, sink, end); if (!sink->d3d.swapchain && !sink->d3d.surface) { ret = TRUE; goto end; } if (sink->d3d.surface) { ref_count = IDirect3DSurface9_Release (sink->d3d.surface); if (ref_count > 0) while (IDirect3DSurface9_Release (sink->d3d.surface) > 0); sink->d3d.surface = NULL; GST_DEBUG_OBJECT (sink, "D3D surface released. Ref count: %d", ref_count); } if (sink->d3d.swapchain) { ref_count = IDirect3DSwapChain9_Release (sink->d3d.swapchain); if (ref_count > 0) while (IDirect3DSwapChain9_Release (sink->d3d.swapchain) > 0); sink->d3d.swapchain = NULL; GST_DEBUG_OBJECT (sink, "D3D swapchain released. Ref count: %d", ref_count); } ret = TRUE; end: UNLOCK_SINK (sink); return ret; } static gboolean d3d_resize_swap_chain (GstD3DVideoSink * sink) { GstD3DVideoSinkClass *class; D3DPRESENT_PARAMETERS d3d_pp; LPDIRECT3DSWAPCHAIN9 swapchain = NULL; gint w = 0, h = 0, ref_count = 0; gboolean ret = FALSE; HRESULT hr; gboolean need_new = FALSE; g_return_val_if_fail (sink != NULL, FALSE); class = GST_D3DVIDEOSINK_GET_CLASS (sink); g_return_val_if_fail (class != NULL, FALSE); LOCK_SINK (sink); if (!sink->d3d.renderable || sink->d3d.device_lost) { UNLOCK_SINK (sink); return FALSE; } LOCK_CLASS (sink, class); CHECK_WINDOW_HANDLE (sink, end, FALSE); CHECK_D3D_DEVICE (class, sink, end); CHECK_D3D_SWAPCHAIN (sink, end); d3d_get_hwnd_window_size (sink->d3d.window_handle, &w, &h); ZeroMemory (&d3d_pp, sizeof (d3d_pp)); /* Get the parameters used to create this swap chain */ hr = IDirect3DSwapChain9_GetPresentParameters (sink->d3d.swapchain, &d3d_pp); if (hr != D3D_OK) { GST_ERROR_OBJECT (sink, "Unable to determine Direct3D present parameters for swap chain"); goto end; } /* Reisze needed? */ if (d3d_pp.BackBufferWidth != w || d3d_pp.BackBufferHeight != h) need_new = TRUE; #if 0 /* Render rect set or unset? */ if ((d3d_pp.SwapEffect != D3DSWAPEFFECT_COPY && sink->d3d.render_rect) || (d3d_pp.SwapEffect != D3DSWAPEFFECT_DISCARD && sink->d3d.render_rect == NULL)) { d3d_pp.SwapEffect = (sink->d3d.render_rect == NULL) ? D3DSWAPEFFECT_DISCARD : D3DSWAPEFFECT_COPY; GST_DEBUG_OBJECT (sink, "Setting SwapEffect: %s", sink->d3d.render_rect ? "COPY" : "DISCARD"); need_new = TRUE; } #endif if (!need_new) { ret = TRUE; goto end; } GST_DEBUG_OBJECT (sink, "Resizing swapchain %dx%d to %dx%d", d3d_pp.BackBufferWidth, d3d_pp.BackBufferHeight, w, h); /* As long as present params windowed == TRUE, width or height * of 0 will force use of HWND's size. */ d3d_pp.BackBufferWidth = 0; d3d_pp.BackBufferHeight = 0; /* Release current swapchain */ if (sink->d3d.swapchain != NULL) { ref_count = IDirect3DSwapChain9_Release (sink->d3d.swapchain); if (ref_count > 0) while (IDirect3DSwapChain9_Release (sink->d3d.swapchain) > 0); if (ref_count > 0) { GST_WARNING_OBJECT (sink, "Release swapchain refcount: %d", ref_count); } sink->d3d.swapchain = NULL; } hr = IDirect3DDevice9_CreateAdditionalSwapChain (class->d3d.device.d3d_device, &d3d_pp, &swapchain); ERROR_CHECK_HR (hr) { CASE_HR_ERR (D3DERR_NOTAVAILABLE); CASE_HR_ERR (D3DERR_DEVICELOST); CASE_HR_ERR (D3DERR_INVALIDCALL); CASE_HR_ERR (D3DERR_OUTOFVIDEOMEMORY); CASE_HR_ERR (E_OUTOFMEMORY); CASE_HR_ERR_END (sink, "Error creating swapchian"); goto end; } sink->d3d.swapchain = swapchain; ret = TRUE; end: UNLOCK_CLASS (sink, class); UNLOCK_SINK (sink); return ret; } static gboolean d3d_copy_buffer_to_surface (GstD3DVideoSink * sink, GstBuffer * buffer) { D3DLOCKED_RECT lr; guint8 *dest, *source; int i; gboolean ret = FALSE; gint unhdl_line = 0; GstMapInfo map; LOCK_SINK (sink); if (!sink->d3d.renderable || sink->d3d.device_lost) goto end; if (!buffer || !gst_buffer_map (buffer, &map, GST_MAP_READ)) { GST_ERROR_OBJECT (sink, "NULL GstBuffer"); goto end; } CHECK_D3D_SURFACE (sink, end); IDirect3DSurface9_LockRect (sink->d3d.surface, &lr, NULL, 0); dest = (guint8 *) lr.pBits; source = map.data; if (!dest) { GST_ERROR_OBJECT (sink, "No D3D surface dest buffer"); goto unlock_surface; } if (sink->d3d.exact_copy) { memcpy (dest, source, gst_buffer_get_size (buffer)); goto done; } if (GST_VIDEO_INFO_IS_YUV (&sink->info)) { switch (sink->format) { case GST_VIDEO_FORMAT_YUY2: case GST_VIDEO_FORMAT_UYVY:{ int srcstride, dststride; dststride = lr.Pitch; srcstride = gst_buffer_get_size (buffer) / GST_VIDEO_SINK_HEIGHT (sink); for (i = 0; i < GST_VIDEO_SINK_HEIGHT (sink); ++i) memcpy (dest + dststride * i, source + srcstride * i, srcstride); break; } case GST_VIDEO_FORMAT_YV12: case GST_VIDEO_FORMAT_I420:{ int srcystride, srcvstride, srcustride; int dstystride, dstvstride, dstustride; guint8 *srcv, *srcu, *dstv, *dstu; int rows; rows = sink->height; /* Source y, u and v strides */ srcystride = GST_ROUND_UP_4 (sink->width); srcustride = GST_ROUND_UP_8 (sink->width) / 2; srcvstride = GST_ROUND_UP_8 (srcystride) / 2; /* Destination y, u and v strides */ dstystride = lr.Pitch; dstustride = dstystride / 2; dstvstride = dstustride; srcu = source + srcystride * GST_ROUND_UP_2 (rows); srcv = srcu + srcustride * GST_ROUND_UP_2 (rows) / 2; if (sink->format == GST_VIDEO_FORMAT_I420) { /* swap u and v planes */ dstv = dest + dstystride * rows; dstu = dstv + dstustride * rows / 2; } else { dstu = dest + dstystride * rows; dstv = dstu + dstustride * rows / 2; } for (i = 0; i < rows; ++i) { /* Copy the y plane */ memcpy (dest + (dstystride * i), source + (srcystride * i), srcystride); } for (i = 0; i < (rows / 2); ++i) { /* Copy the u plane */ memcpy (dstu + (dstustride * i), srcu + (srcustride * i), srcustride); /* Copy the v plane */ memcpy (dstv + (dstvstride * i), srcv + (srcvstride * i), srcvstride); } goto done; } case GST_VIDEO_FORMAT_NV12:{ int srcstride, dststride; guint8 *dst = dest; int component; dststride = lr.Pitch; for (component = 0; component < 2; component++) { const int compHeight = GST_VIDEO_INFO_COMP_HEIGHT (&sink->info, component); guint8 *src = source + GST_VIDEO_INFO_COMP_OFFSET (&sink->info, component); srcstride = GST_VIDEO_INFO_COMP_STRIDE (&sink->info, component); for (i = 0; i < compHeight; i++) { memcpy (dst + dststride * i, src + srcstride * i, srcstride); } dst += dststride * compHeight; } break; } default:; unhdl_line = __LINE__; goto unhandled_format; } /* end switch */ } else if (GST_VIDEO_INFO_IS_RGB (&sink->info)) { for (i = 0; i < sink->height * sink->width; i++) { guint8 p[4] = { 0, 0, 0, 0 }; /* Input */ switch (sink->fmt_details.bpp) { case 32: case 24:{ guint32 *p32 = (guint32 *) (source + (i * sink->fmt_details.pixel_width)); if (sink->fmt_details.a_shift) p[ALPHA] = (*p32 & sink->fmt_details.a_mask) >> sink->fmt_details.a_shift; p[RED] = (*p32 & sink->fmt_details.r_mask) >> sink->fmt_details.r_shift; p[GREEN] = (*p32 & sink->fmt_details.g_mask) >> sink->fmt_details.g_shift; p[BLUE] = (*p32 & sink->fmt_details.b_mask) >> sink->fmt_details.b_shift; break; } case 16:{ guint16 *p16 = (guint16 *) (source + (i * sink->fmt_details.pixel_width)); if (sink->fmt_details.a_shift) p[ALPHA] = (*p16 & sink->fmt_details.a_mask16) >> sink->fmt_details. a_shift; p[RED] = (*p16 & sink->fmt_details.r_mask16) >> sink->fmt_details.r_shift; p[GREEN] = (*p16 & sink->fmt_details.g_mask16) >> sink->fmt_details.g_shift; p[BLUE] = (*p16 & sink->fmt_details.b_mask16) >> sink->fmt_details.b_shift; break; } default:; unhdl_line = __LINE__; goto unhandled_format; } /* Output */ switch (sink->d3d.format) { case D3DFMT_A8R8G8B8: p[RED] = BIT_CONV (p[RED], sink->fmt_details.r_bits, 8); p[GREEN] = BIT_CONV (p[GREEN], sink->fmt_details.g_bits, 8); p[BLUE] = BIT_CONV (p[BLUE], sink->fmt_details.b_bits, 8); if (sink->fmt_details.a_shift) p[ALPHA] = BIT_CONV (p[ALPHA], sink->fmt_details.a_bits, 8); else p[ALPHA] = 255; memcpy ((dest + (i * 4)), &p, 4); break; case D3DFMT_X8R8G8B8: p[RED] = BIT_CONV (p[RED], sink->fmt_details.r_bits, 8); p[GREEN] = BIT_CONV (p[GREEN], sink->fmt_details.g_bits, 8); p[BLUE] = BIT_CONV (p[BLUE], sink->fmt_details.b_bits, 8); memcpy ((dest + (i * 4)), &p, 4); break; case D3DFMT_A8B8G8R8: *(dest + (i * 4) + 0) = p[RED]; *(dest + (i * 4) + 1) = p[GREEN]; *(dest + (i * 4) + 2) = p[BLUE]; if (sink->fmt_details.a_shift) *(dest + (i * 4) + 3) = p[ALPHA]; else *(dest + (i * 4) + 3) = 255; break; case D3DFMT_X8B8G8R8: *(dest + (i * 4) + 0) = p[RED]; *(dest + (i * 4) + 1) = p[GREEN]; *(dest + (i * 4) + 2) = p[BLUE]; break; case D3DFMT_R8G8B8: p[RED] = BIT_CONV (p[RED], sink->fmt_details.r_bits, 8); p[GREEN] = BIT_CONV (p[GREEN], sink->fmt_details.g_bits, 8); p[BLUE] = BIT_CONV (p[BLUE], sink->fmt_details.b_bits, 8); memcpy ((dest + (i * 3)), &p, 3); break; case D3DFMT_R5G6B5:{ p[RED] = BIT_CONV (p[RED], sink->fmt_details.r_bits, 5); p[GREEN] = BIT_CONV (p[GREEN], sink->fmt_details.g_bits, 6); p[BLUE] = BIT_CONV (p[BLUE], sink->fmt_details.b_bits, 5); *((guint16 *) (dest + (i * 2))) = (p[RED] << 11 | p[GREEN] << 5 | p[BLUE]); break; } case D3DFMT_X1R5G5B5: case D3DFMT_A1R5G5B5: case D3DFMT_X4R4G4B4: case D3DFMT_A4R4G4B4: case D3DFMT_A8R3G3B2: case D3DFMT_R3G3B2: //break; default: unhdl_line = __LINE__; goto unhandled_format; } } /* end for */ } /* end if rgb */ goto done; unhandled_format: GST_ERROR_OBJECT (sink, "Unhandled format [LN:%d] '%s' -> '%s' (should not get here)", unhdl_line, gst_video_format2string (sink->format), d3d_format2string (sink->d3d.format)); goto unlock_surface; done: ret = TRUE; unlock_surface: IDirect3DSurface9_UnlockRect (sink->d3d.surface); end: UNLOCK_SINK (sink); return ret; } static gboolean d3d_present_swap_chain (GstD3DVideoSink * sink) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); LPDIRECT3DSURFACE9 back_buffer = NULL; gboolean ret = FALSE; HRESULT hr; RECT dstr, srcr, *pDestRect = NULL, *pSrcRect = NULL; LOCK_SINK (sink); if (!sink->d3d.renderable || sink->d3d.device_lost) { UNLOCK_SINK (sink); return FALSE; } LOCK_CLASS (sink, class); CHECK_WINDOW_HANDLE (sink, end, FALSE); CHECK_D3D_DEVICE (class, sink, end); CHECK_D3D_SWAPCHAIN (sink, end); /* Set the render target to our swap chain */ IDirect3DSwapChain9_GetBackBuffer (sink->d3d.swapchain, 0, D3DBACKBUFFER_TYPE_MONO, &back_buffer); IDirect3DDevice9_SetRenderTarget (class->d3d.device.d3d_device, 0, back_buffer); IDirect3DSurface9_Release (back_buffer); /* Clear the target */ IDirect3DDevice9_Clear (class->d3d.device.d3d_device, 0, NULL, D3DCLEAR_TARGET, D3DCOLOR_XRGB (0, 0, 0), 1.0f, 0); hr = IDirect3DDevice9_BeginScene (class->d3d.device.d3d_device); ERROR_CHECK_HR (hr) { CASE_HR_ERR (D3DERR_INVALIDCALL); CASE_HR_ERR_END (sink, "IDirect3DDevice9_BeginScene"); goto end; } /* Stretch and blit ops, to copy offscreen surface buffer * to Display back buffer. */ d3d_stretch_and_copy (sink, back_buffer); IDirect3DDevice9_EndScene (class->d3d.device.d3d_device); if (d3d_get_render_rects (sink->d3d.render_rect, &dstr, &srcr)) { pDestRect = &dstr; pSrcRect = &srcr; } /* * Swap back and front buffers on video card and present to the user */ hr = IDirect3DSwapChain9_Present (sink->d3d.swapchain, pSrcRect, pDestRect, NULL, NULL, 0); if (hr == D3DERR_DEVICELOST) { d3d_notify_device_lost (sink); ret = TRUE; goto end; } ERROR_CHECK_HR (hr) { CASE_HR_ERR (D3DERR_DEVICELOST); CASE_HR_ERR (D3DERR_DRIVERINTERNALERROR); CASE_HR_ERR (D3DERR_INVALIDCALL); CASE_HR_ERR (D3DERR_OUTOFVIDEOMEMORY); CASE_HR_ERR (E_OUTOFMEMORY); CASE_HR_DBG_END (sink, "IDirect3DSwapChain9_Present failure"); goto end; } ret = TRUE; end: UNLOCK_SINK (sink); UNLOCK_CLASS (sink, class); return ret; } static gboolean d3d_stretch_and_copy (GstD3DVideoSink * sink, LPDIRECT3DSURFACE9 back_buffer) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); GstVideoRectangle *render_rect = NULL; RECT *r_ptr = NULL; RECT r; HRESULT hr; gboolean ret = FALSE; LOCK_SINK (sink); CHECK_WINDOW_HANDLE (sink, end, FALSE); CHECK_D3D_DEVICE (class, sink, end); CHECK_D3D_SURFACE (sink, end); render_rect = sink->d3d.render_rect; if (sink->force_aspect_ratio) { gint window_width; gint window_height; GstVideoRectangle src; GstVideoRectangle dst; GstVideoRectangle result; memset (&dst, 0, sizeof (dst)); memset (&src, 0, sizeof (src)); /* Set via GstXOverlay set_render_rect */ if (render_rect) { memcpy (&dst, render_rect, sizeof (dst)); } else { d3d_get_hwnd_window_size (sink->d3d.window_handle, &window_width, &window_height); dst.w = window_width; dst.h = window_height; } src.w = GST_VIDEO_SINK_WIDTH (sink); src.h = GST_VIDEO_SINK_HEIGHT (sink); gst_video_sink_center_rect (src, dst, &result, TRUE); r.left = result.x; r.top = result.y; r.right = result.x + result.w; r.bottom = result.y + result.h; r_ptr = &r; } else if (render_rect) { r.left = 0; r.top = 0; r.right = render_rect->w; r.bottom = render_rect->h; r_ptr = &r; } /* TODO: StretchRect returns error if the dest rect is outside * the backbuffer area. So we need to calc how much of the src * surface is being scaled / copied to the render rect.. */ hr = IDirect3DDevice9_StretchRect (class->d3d.device.d3d_device, sink->d3d.surface, /* Source Surface */ NULL, /* Source Surface Rect (NULL: Whole) */ back_buffer, /* Dest Surface */ r_ptr, /* Dest Surface Rect (NULL: Whole) */ class->d3d.device.filter_type); if (hr == D3D_OK) { ret = TRUE; } else { GST_ERROR_OBJECT (sink, "Failure calling Direct3DDevice9_StretchRect"); } end: UNLOCK_SINK (sink); return ret; } GstFlowReturn d3d_render_buffer (GstD3DVideoSink * sink, GstBuffer * buf) { GstFlowReturn ret = GST_FLOW_OK; GstMapInfo map; g_return_val_if_fail (gst_buffer_map (buf, &map, GST_MAP_READ) != FALSE, GST_FLOW_ERROR); LOCK_SINK (sink); if (!sink->d3d.window_handle) { if (sink->stream_stop_on_close) { GST_LOG_OBJECT (sink, "Stopping stream. No render window"); ret = GST_FLOW_ERROR; } goto end; } if (sink->d3d.device_lost) { GST_LOG_OBJECT (sink, "Device lost, waiting for reset.."); goto end; } GST_INFO_OBJECT (sink, "%s %" GST_TIME_FORMAT, (sink->d3d.window_handle != NULL) ? "Render" : "No Win", GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf))); /* Reize swapchain if needed */ if (!d3d_resize_swap_chain (sink)) { ret = GST_FLOW_ERROR; goto end; } d3d_copy_buffer_to_surface (sink, buf); if (!d3d_present_swap_chain (sink)) { ret = GST_FLOW_ERROR; goto end; } end: UNLOCK_SINK (sink); return ret; } /** D3D Window Proc Functions **/ static LRESULT APIENTRY d3d_wnd_proc (HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { GstD3DVideoSink *sink = (GstD3DVideoSink *) GetProp (hWnd, TEXT ("GstD3DVideoSink")); WNDPROC proc; LRESULT ret = 0; LOCK_SINK (sink); proc = sink->d3d.orig_wnd_proc; UNLOCK_SINK (sink); switch (message) { case WM_ERASEBKGND: return TRUE; case WM_PAINT:{ if (proc) ret = CallWindowProc (proc, hWnd, message, wParam, lParam); /* Call this afterwards to ensure that our paint happens last */ d3d_present_swap_chain (sink); goto end; } case WM_SIZE:{ if (proc) ret = CallWindowProc (proc, hWnd, message, wParam, lParam); d3d_resize_swap_chain (sink); goto end; } case WM_KEYDOWN: case WM_KEYUP: if (sink->enable_navigation_events) { gunichar2 wcrep[128]; if (GetKeyNameTextW (lParam, (LPWSTR) wcrep, 128)) { gchar *utfrep = g_utf16_to_utf8 (wcrep, 128, NULL, NULL, NULL); if (utfrep) { if (message == WM_KEYDOWN) gst_navigation_send_key_event (GST_NAVIGATION (sink), "key-press", utfrep); else if (message == WM_KEYUP) gst_navigation_send_key_event (GST_NAVIGATION (sink), "key-release", utfrep); g_free (utfrep); } } } break; case WM_LBUTTONDOWN: case WM_LBUTTONUP: case WM_RBUTTONDOWN: case WM_RBUTTONUP: case WM_MBUTTONDOWN: case WM_MBUTTONUP: case WM_MOUSEMOVE:{ gdouble x = 0, y = 0; if (sink->enable_navigation_events && d3d_get_render_coordinates (sink, LOWORD (lParam), HIWORD (lParam), &x, &y)) { gint button; const gchar *action = NULL; switch (message) { case WM_MOUSEMOVE: button = 0; action = "mouse-move"; break; case WM_LBUTTONDOWN: button = 1; action = "mouse-button-press"; break; case WM_LBUTTONUP: button = 1; action = "mouse-button-release"; break; case WM_RBUTTONDOWN: button = 2; action = "mouse-button-press"; break; case WM_RBUTTONUP: button = 2; action = "mouse-button-release"; break; case WM_MBUTTONDOWN: button = 3; action = "mouse-button-press"; break; case WM_MBUTTONUP: button = 3; action = "mouse-button-release"; break; default: break; } if (action) { //GST_DEBUG_OBJECT(sink, "%s: %lfx%lf", action, x, y); gst_navigation_send_mouse_event (GST_NAVIGATION (sink), action, button, x, y); } } break; } case WM_CLOSE: d3d_set_window_handle (sink, 0, FALSE); break; default: break; } if (proc) ret = CallWindowProc (proc, hWnd, message, wParam, lParam); else ret = DefWindowProc (hWnd, message, wParam, lParam); end: return ret; } /** Internal Window **/ static LRESULT APIENTRY d3d_wnd_proc_internal (HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { //GstD3DVideoSink * sink = (GstD3DVideoSink*)GetProp(hWnd, TEXT("GstD3DVideoSink")); switch (message) { case WM_DESTROY: GST_DEBUG ("Internal window: WM_DESTROY"); /* Tell the internal window thread to shut down */ PostQuitMessage (0); GST_DEBUG ("Posted quit.."); break; } return DefWindowProc (hWnd, message, wParam, lParam); } static HWND _d3d_create_internal_window (GstD3DVideoSink * sink) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); int width, height; int offx, offy; DWORD exstyle, style; HWND video_window; RECT rect; int screenwidth; int screenheight; /* * GST_VIDEO_SINK_WIDTH() is the aspect-ratio-corrected size of the video. * GetSystemMetrics() returns the width of the dialog's border (doubled * b/c of left and right borders). */ width = GST_VIDEO_SINK_WIDTH (sink) + GetSystemMetrics (SM_CXSIZEFRAME) * 2; height = GST_VIDEO_SINK_HEIGHT (sink) + GetSystemMetrics (SM_CYCAPTION) + (GetSystemMetrics (SM_CYSIZEFRAME) * 2); SystemParametersInfo (SPI_GETWORKAREA, 0, &rect, 0); screenwidth = rect.right - rect.left; screenheight = rect.bottom - rect.top; offx = rect.left; offy = rect.top; /* Make it fit into the screen without changing the aspect ratio. */ if (width > screenwidth) { double ratio = (double) screenwidth / (double) width; width = screenwidth; height = (int) (height * ratio); } if (height > screenheight) { double ratio = (double) screenheight / (double) height; height = screenheight; width = (int) (width * ratio); } style = WS_OVERLAPPEDWINDOW; /* Normal top-level window */ exstyle = 0; video_window = CreateWindowEx (exstyle, class->d3d.wnd_class.lpszClassName, TEXT ("GStreamer D3D video sink (internal window)"), style, offx, offy, width, height, NULL, NULL, class->d3d.wnd_class.hInstance, sink); if (video_window == NULL) { GST_ERROR_OBJECT (sink, "Failed to create internal window: %lu", GetLastError ()); return NULL; } /* Now show the window, as appropriate */ ShowWindow (video_window, SW_SHOWNORMAL); /* Trigger the initial paint of the window */ UpdateWindow (video_window); return video_window; } typedef struct { GstD3DVideoSink *sink; gboolean running; HWND hWnd; } D3DInternalWindowDat; static gpointer d3d_internal_window_thread (D3DInternalWindowDat * dat) { GstD3DVideoSink *sink; HWND hWnd; MSG msg; g_return_val_if_fail (dat != NULL, NULL); sink = dat->sink; GST_DEBUG_OBJECT (sink, "Entering internal window thread: %p", g_thread_self ()); /* Create internal window */ hWnd = _d3d_create_internal_window (sink); if (!hWnd) { GST_ERROR_OBJECT (sink, "Failed to create internal window"); goto end; } dat->hWnd = hWnd; dat->running = TRUE; /* * Internal window message loop */ while (GetMessage (&msg, NULL, 0, 0)) { TranslateMessage (&msg); DispatchMessage (&msg); } end: GST_DEBUG_OBJECT (sink, "Exiting internal window thread: %p", g_thread_self ()); return NULL; } static HWND d3d_create_internal_window (GstD3DVideoSink * sink) { GThread *thread; D3DInternalWindowDat dat; gulong timeout_interval = 10000; /* 10 ms interval */ gulong intervals = (10000000 / timeout_interval); /* 10 secs */ gulong i; dat.sink = sink; dat.running = FALSE; dat.hWnd = 0; thread = g_thread_new ("d3dvideosink-window-thread", (GThreadFunc) d3d_internal_window_thread, &dat); if (!thread) { GST_ERROR ("Failed to created internal window thread"); return 0; } /* Wait 10 seconds for window proc loop to start up */ for (i = 0; dat.running == FALSE && i < intervals; i++) { g_usleep (timeout_interval); } GST_DEBUG_OBJECT (sink, "Created window: %p (intervals: %lu)", dat.hWnd, i); return dat.hWnd; } /*** D3D Video Class Methdos ***/ gboolean d3d_class_init (GstD3DVideoSink * sink) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); gulong timeout_interval = 10000; /* 10 ms interval */ gulong intervals = (10000000 / timeout_interval); /* 10 secs */ gboolean ret = FALSE; gulong i; g_return_val_if_fail (class != NULL, FALSE); LOCK_CLASS (sink, class); class->d3d.refs += 1; GST_DEBUG ("D3D class init [refs:%u]", class->d3d.refs); class->d3d.sink_list = g_list_append (class->d3d.sink_list, sink); if (class->d3d.refs > 1) goto end; WM_D3DVIDEO_NOTIFY_DEVICE_LOST = RegisterWindowMessage ("WM_D3DVIDEO_NOTIFY_DEVICE_LOST"); class->d3d.d3d = Direct3DCreate9 (D3D_SDK_VERSION); if (!class->d3d.d3d) { GST_ERROR ("Unable to create Direct3D interface"); goto error; } /* Register Window Class for internal Windows */ memset (&class->d3d.wnd_class, 0, sizeof (WNDCLASS)); class->d3d.wnd_class.style = CS_OWNDC | CS_HREDRAW | CS_VREDRAW; class->d3d.wnd_class.hInstance = GetModuleHandle (NULL); class->d3d.wnd_class.lpszClassName = TEXT ("GstD3DVideoSinkInternalWindow"); class->d3d.wnd_class.hbrBackground = (HBRUSH) GetStockObject (BLACK_BRUSH); class->d3d.wnd_class.hCursor = LoadCursor (NULL, IDC_ARROW); class->d3d.wnd_class.hIcon = LoadIcon (NULL, IDI_APPLICATION); class->d3d.wnd_class.cbClsExtra = 0; class->d3d.wnd_class.cbWndExtra = 0; class->d3d.wnd_class.lpfnWndProc = d3d_wnd_proc_internal; if (RegisterClass (&class->d3d.wnd_class) == 0) { GST_ERROR ("Failed to register window class: %lu", GetLastError ()); goto error; } class->d3d.running = FALSE; class->d3d.error_exit = FALSE; UNLOCK_CLASS (sink, class); class->d3d.thread = g_thread_new ("d3dvideosink-window-thread", (GThreadFunc) d3d_hidden_window_thread, class); LOCK_CLASS (sink, class); if (!class->d3d.thread) { GST_ERROR ("Failed to created hidden window thread"); goto error; } UNLOCK_CLASS (sink, class); /* Wait 10 seconds for window proc loop to start up */ for (i = 0; class->d3d.running == FALSE && i < intervals; i++) { g_usleep (timeout_interval); } LOCK_CLASS (sink, class); if (class->d3d.error_exit) goto error; if (!class->d3d.running) { GST_ERROR ("Waited %lu ms, window proc loop has not started", (timeout_interval * intervals) / 1000); goto error; } GST_DEBUG ("Hidden window message loop is running.."); end: ret = TRUE; error: UNLOCK_CLASS (sink, class); if (!ret) d3d_class_destroy (sink); return ret; } void d3d_class_destroy (GstD3DVideoSink * sink) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); g_return_if_fail (class != NULL); LOCK_CLASS (sink, class); class->d3d.refs -= 1; GST_DEBUG ("D3D class destroy [refs:%u]", class->d3d.refs); class->d3d.sink_list = g_list_remove (class->d3d.sink_list, sink); if (class->d3d.refs >= 1) goto end; UNLOCK_CLASS (sink, class); if (class->d3d.running) { GST_DEBUG ("Shutting down window proc thread, waiting to join.."); PostMessage (class->d3d.hidden_window, WM_QUIT, 0, 0); g_thread_join (class->d3d.thread); GST_DEBUG ("Joined.."); } LOCK_CLASS (sink, class); if (class->d3d.d3d) { int ref_count; ref_count = IDirect3D9_Release (class->d3d.d3d); GST_DEBUG ("Direct3D object released. Reference count: %d", ref_count); } UnregisterClass (class->d3d.wnd_class.lpszClassName, class->d3d.wnd_class.hInstance); g_list_free (class->d3d.sink_list); memset (&class->d3d, 0, sizeof (GstD3DDataClass)); end: UNLOCK_CLASS (sink, class); } static gboolean d3d_class_display_device_create (GstD3DVideoSinkClass * class, UINT adapter) { LPDIRECT3D9 d3d; GstD3DDisplayDevice *device; HWND hwnd; D3DCAPS9 caps; D3DDISPLAYMODE disp_mode; DWORD create_mask = 0; HRESULT hr; gboolean ret = FALSE; g_return_val_if_fail (class != NULL, FALSE); GST_DEBUG (" "); LOCK_CLASS (NULL, class); d3d = class->d3d.d3d; device = &class->d3d.device; hwnd = class->d3d.hidden_window; memset (&caps, 0, sizeof (caps)); memset (&disp_mode, 0, sizeof (disp_mode)); memset (&device->present_params, 0, sizeof (device->present_params)); device->adapter = adapter; if (IDirect3D9_GetAdapterDisplayMode (d3d, adapter, &disp_mode) != D3D_OK) { GST_ERROR ("Unable to request adapter[%u] display mode", adapter); goto error; } if (IDirect3D9_GetDeviceCaps (d3d, adapter, D3DDEVTYPE_HAL, &caps) != D3D_OK) { GST_ERROR ("Unable to request adapter[%u] device caps", adapter); goto error; } /* Ask DirectX to please not clobber the FPU state when making DirectX * API calls. This can cause libraries such as cairo to misbehave in * certain scenarios. */ create_mask = 0 | D3DCREATE_FPU_PRESERVE; /* Determine vertex processing capabilities. Some cards have issues * using software vertex processing. Courtesy: * http://www.chadvernon.com/blog/resources/directx9/improved-direct3d-initialization/ */ if ((caps.DevCaps & D3DDEVCAPS_HWTRANSFORMANDLIGHT) == D3DDEVCAPS_HWTRANSFORMANDLIGHT) { create_mask |= D3DCREATE_HARDWARE_VERTEXPROCESSING; /* if ((d3dcaps.DevCaps & D3DDEVCAPS_PUREDEVICE) == D3DDEVCAPS_PUREDEVICE) */ /* d3dcreate |= D3DCREATE_PUREDEVICE; */ } else { create_mask |= D3DCREATE_SOFTWARE_VERTEXPROCESSING; } /* Check the filter type. */ if ((caps.StretchRectFilterCaps & D3DPTFILTERCAPS_MINFLINEAR) == D3DPTFILTERCAPS_MINFLINEAR && (caps.StretchRectFilterCaps & D3DPTFILTERCAPS_MAGFLINEAR) == D3DPTFILTERCAPS_MAGFLINEAR) { device->filter_type = D3DTEXF_LINEAR; } else { device->filter_type = D3DTEXF_NONE; } /* Setup the display mode format. */ device->format = disp_mode.Format; //present_params.Flags = D3DPRESENTFLAG_VIDEO; device->present_params.Windowed = TRUE; device->present_params.SwapEffect = D3DSWAPEFFECT_DISCARD; device->present_params.BackBufferCount = 1; device->present_params.BackBufferFormat = device->format; device->present_params.BackBufferWidth = 1; device->present_params.BackBufferHeight = 1; device->present_params.MultiSampleType = D3DMULTISAMPLE_NONE; device->present_params.PresentationInterval = D3DPRESENT_INTERVAL_DEFAULT; //D3DPRESENT_INTERVAL_IMMEDIATE; GST_DEBUG ("Creating Direct3D device for hidden window %p", NULL); //shared.hidden_window_handle); if ((hr = IDirect3D9_CreateDevice (d3d, adapter, D3DDEVTYPE_HAL, hwnd, create_mask, &device->present_params, &device->d3d_device)) != D3D_OK) { GST_ERROR ("Unable to create Direct3D device. Result: %ld (0x%lx)", hr, hr); goto error; } GST_DEBUG ("Display Device format: %s", d3d_format2string (disp_mode.Format)); ret = TRUE; goto end; error: memset (device, 0, sizeof (GstD3DDisplayDevice)); end: UNLOCK_CLASS (NULL, class); return ret; } static void d3d_class_display_device_destroy (GstD3DVideoSinkClass * class) { g_return_if_fail (class != NULL); LOCK_CLASS (NULL, class); if (class->d3d.device.d3d_device) { int ref_count; ref_count = IDirect3DDevice9_Release (class->d3d.device.d3d_device); GST_DEBUG ("Direct3D device [adapter:%u] released. Reference count: %d", class->d3d.device.adapter, ref_count); } memset (&class->d3d.device, 0, sizeof (GstD3DDisplayDevice)); UNLOCK_CLASS (NULL, class); } static void d3d_class_notify_device_lost (GstD3DVideoSink * sink) { GstD3DVideoSinkClass *class = GST_D3DVIDEOSINK_GET_CLASS (sink); PostMessage (class->d3d.hidden_window, WM_D3DVIDEO_NOTIFY_DEVICE_LOST, 0, 0); } static void d3d_class_notify_device_lost_all (GstD3DVideoSinkClass * class) { g_return_if_fail (class != NULL); LOCK_CLASS (NULL, class); if (!class->d3d.device_lost) { GList *lst, *clst; class->d3d.device_lost = TRUE; GST_DEBUG ("Notifying all instances of device loss"); clst = g_list_copy (class->d3d.sink_list); UNLOCK_CLASS (NULL, class); for (lst = clst; lst != NULL; lst = lst->next) { GstD3DVideoSink *sink = (GstD3DVideoSink *) lst->data; if (!sink) continue; d3d_notify_device_lost (sink); } g_list_free (clst); LOCK_CLASS (NULL, class); /* Set timer to try reset at given interval */ SetTimer (class->d3d.hidden_window, IDT_DEVICE_RESET_TIMER, 500, NULL); } UNLOCK_CLASS (NULL, class); } static void d3d_class_reset_display_device (GstD3DVideoSinkClass * class) { HRESULT hr; g_return_if_fail (class != NULL); LOCK_CLASS (NULL, class); hr = IDirect3DDevice9_Reset (class->d3d.device.d3d_device, &class->d3d.device.present_params); ERROR_CHECK_HR (hr) { CASE_HR_ERR (D3DERR_DEVICELOST); CASE_HR_ERR (D3DERR_DEVICEREMOVED); CASE_HR_ERR (D3DERR_DRIVERINTERNALERROR); CASE_HR_ERR (D3DERR_OUTOFVIDEOMEMORY); CASE_HR_DBG_END (NULL, "Attempt device reset.. failed"); goto end; } GST_INFO ("Attempt device reset.. success"); class->d3d.device_lost = FALSE; KillTimer (class->d3d.hidden_window, IDT_DEVICE_RESET_TIMER); g_list_foreach (class->d3d.sink_list, (GFunc) d3d_notify_device_reset, NULL); end:; UNLOCK_CLASS (NULL, class); } /** Hidden Window Loop Thread **/ static LRESULT APIENTRY D3DHiddenWndProc (HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) { switch (message) { case WM_TIMER: switch (wParam) { case IDT_DEVICE_RESET_TIMER: d3d_class_reset_display_device ((GstD3DVideoSinkClass *) GetWindowLongPtr (hWnd, GWLP_USERDATA)); break; default:; } return 0; case WM_DESTROY: PostQuitMessage (0); return 0; default: /* non constants */ if (message == WM_D3DVIDEO_NOTIFY_DEVICE_LOST) { d3d_class_notify_device_lost_all ((GstD3DVideoSinkClass *) GetWindowLongPtr (hWnd, GWLP_USERDATA)); return 0; } } return DefWindowProc (hWnd, message, wParam, lParam); } static gboolean d3d_hidden_window_thread (GstD3DVideoSinkClass * class) { WNDCLASS WndClass; gboolean reged = FALSE; HWND hWnd = 0; gboolean ret = FALSE; g_return_val_if_fail (class != NULL, FALSE); memset (&WndClass, 0, sizeof (WNDCLASS)); WndClass.hInstance = GetModuleHandle (NULL); WndClass.lpszClassName = TEXT ("gstd3dvideo-hidden-window-class"); WndClass.lpfnWndProc = D3DHiddenWndProc; if (!RegisterClass (&WndClass)) { GST_ERROR ("Unable to register Direct3D hidden window class"); goto error; } reged = TRUE; hWnd = CreateWindowEx (0, WndClass.lpszClassName, TEXT ("GStreamer Direct3D hidden window"), WS_POPUP, 0, 0, 1, 1, HWND_MESSAGE, NULL, WndClass.hInstance, class); if (hWnd == NULL) { GST_ERROR ("Failed to create Direct3D hidden window"); goto error; } GST_DEBUG ("Direct3D hidden window handle: %p", hWnd); class->d3d.hidden_window = hWnd; /* TODO: Multi-monitor setup? */ if (!d3d_class_display_device_create (class, D3DADAPTER_DEFAULT)) { GST_ERROR ("Failed to initiazlize adapter: %u", D3DADAPTER_DEFAULT); goto error; } /* Attach data to window */ SetWindowLongPtr (hWnd, GWLP_USERDATA, (LONG_PTR) class); GST_DEBUG ("Entering Direct3D hidden window message loop"); class->d3d.running = TRUE; /* Hidden Window Message Loop */ while (1) { MSG msg; while (GetMessage (&msg, NULL, 0, 0)) { TranslateMessage (&msg); DispatchMessage (&msg); } if (msg.message == WM_QUIT || msg.message == WM_CLOSE) break; } class->d3d.running = FALSE; GST_DEBUG ("Leaving Direct3D hidden window message loop"); ret = TRUE; error: if (!ret) class->d3d.error_exit = TRUE; if (reged) UnregisterClass (WndClass.lpszClassName, WndClass.hInstance); if (hWnd) { DestroyWindow (hWnd); class->d3d.hidden_window = 0; } d3d_class_display_device_destroy (class); return ret; }