git.sesse.net Git - vlc/blob - modules/video_output/msw/direct2d.c

   1 /*****************************************************************************
   2  * direct2d.c : Direct2D video output plugin for vlc (Win7/Vista SP2 PF Update)
   3  *****************************************************************************
   4  * Copyright (C) 2010 VideoLAN and AUTHORS
   5  * $Id$
   6  *
   7  * Author: David Kaplan <david@2of1.org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27
  28 #ifdef HAVE_CONFIG_H
  29 # include "config.h"
  30 #endif
  31 #include <assert.h>
  32
  33 #include <vlc_common.h>
  34 #include <vlc_plugin.h>
  35 #include <vlc_playlist.h>
  36 #include <vlc_vout_display.h>
  37
  38 #include <windows.h>
  39 #include <commctrl.h>
  40
  41 #include <d2d1.h>
  42
  43 #include "common.h"
  44
  45 #include <initguid.h>
  46 #undef GUID_EXT
  47 #define GUID_EXT
  48 DEFINE_GUID(IID_ID2D1Factory, 0x6152247, 0x6f50, 0x465a, 0x92, 0x45, 0x11, 0x8b, 0xfd, 0x3b, 0x60, 0x7);
  49
  50 /*****************************************************************************
  51  * Module descriptor
  52  *****************************************************************************/
  53 static int  Open (vlc_object_t *);
  54 static void Close(vlc_object_t *);
  55
  56 #define D2D_HELP N_("Video output for Windows 7/Windows Vista with Platform update")
  57
  58 vlc_module_begin ()
  59     set_category(CAT_VIDEO)
  60     set_subcategory(SUBCAT_VIDEO_VOUT)
  61     set_help(D2D_HELP)
  62     set_shortname("Direct2D")
  63     set_description(N_("Direct2D video output"))
  64     set_capability("vout display", 65)
  65     add_shortcut("direct2d")
  66     set_callbacks(Open, Close)
  67 vlc_module_end ()
  68
  69
  70 /*****************************************************************************
  71  * Local prototypes
  72  *****************************************************************************/
  73 static picture_pool_t *Pool  (vout_display_t *, unsigned);
  74 static void           Prepare(vout_display_t *, picture_t *);
  75 static void           Display(vout_display_t *, picture_t *);
  76 static int            Control(vout_display_t *, int, va_list);
  77 static void           Manage (vout_display_t *);
  78
  79 static int      D2D_CreateRenderTarget(vout_display_t *vd);
  80 static void     D2D_ResizeRenderTarget(vout_display_t *vd);
  81 static void     D2D_DestroyRenderTarget(vout_display_t *vd);
  82
  83 /**
  84  * Initialises Direct2D vout module
  85  */
  86 static int Open(vlc_object_t *object)
  87 {
  88     vout_display_t *vd = (vout_display_t *)object;
  89     vout_display_sys_t *sys;
  90
  91     vd->sys = sys = calloc(1, sizeof(*sys));
  92     if (!sys)
  93         return VLC_ENOMEM;
  94
  95     sys->d2_render_target = NULL;
  96
  97     sys->d2_dll = LoadLibrary(TEXT("D2D1.DLL"));
  98     if (!sys->d2_dll) {
  99         if (object->b_force)
 100             msg_Err(vd, "Cannot load D2D1.DLL, aborting");
 101         goto error;
 102     }
 103
 104     D2D1_FACTORY_OPTIONS fo = {
 105         D2D1_DEBUG_LEVEL_NONE
 106     };
 107
 108     HRESULT (WINAPI *D2D1CreateFactory)(D2D1_FACTORY_TYPE, REFIID,
 109                                         const D2D1_FACTORY_OPTIONS *,
 110                                         void **);
 111
 112     D2D1CreateFactory = (void *)GetProcAddress(sys->d2_dll,
 113                                                TEXT("D2D1CreateFactory"));
 114     if (!D2D1CreateFactory) {
 115         msg_Err(vd,
 116                 "Cannot locate reference to a D2D1CreateFactory ABI in D2D1.DLL");
 117         goto error;
 118     }
 119
 120 #ifndef NDEBUG
 121     msg_Dbg(vd, "D2D1.DLL loaded");
 122 #endif
 123
 124     HRESULT hr = D2D1CreateFactory(D2D1_FACTORY_TYPE_SINGLE_THREADED,
 125                                    (REFIID)&IID_ID2D1Factory,
 126                                    &fo,
 127                                    (void **)&sys->d2_factory);
 128     if (hr != S_OK) {
 129         msg_Err(vd, "Cannot create Direct2D factory (hr = 0x%x)!",
 130                 (unsigned)hr);
 131         goto error;
 132     }
 133
 134     if (CommonInit(vd))
 135         goto error;
 136
 137     if (D2D_CreateRenderTarget(vd) != VLC_SUCCESS)
 138         goto error;
 139
 140     vout_display_info_t info = vd->info;
 141     info.is_slow              = false;
 142     info.has_double_click     = true;
 143     info.has_hide_mouse       = false;
 144     info.has_pictures_invalid = false;
 145     vd->info = info;
 146
 147     vd->fmt.i_chroma = VLC_CODEC_RGB32; /* masks change this to BGR32 for ID2D1Bitmap */
 148     vd->fmt.i_rmask  = 0x0000ff00;
 149     vd->fmt.i_gmask  = 0x00ff0000;
 150     vd->fmt.i_bmask  = 0xff000000;
 151
 152     vd->pool    = Pool;
 153     vd->prepare = Prepare;
 154     vd->display = Display;
 155     vd->manage  = Manage;
 156     vd->control = Control;
 157
 158     EventThreadUpdateTitle(sys->event, VOUT_TITLE " (Direct2D output)");
 159
 160 #ifndef NDEBUG
 161     msg_Dbg(vd, "Ready");
 162 #endif
 163
 164     return VLC_SUCCESS;
 165
 166 error:
 167     Close(VLC_OBJECT(vd));
 168     return VLC_EGENERIC;
 169 }
 170
 171 /**
 172  * Close Direct2D vout
 173  */
 174 static void Close(vlc_object_t *object)
 175 {
 176     vout_display_t *vd = (vout_display_t *)object;
 177
 178     D2D_DestroyRenderTarget(vd);
 179
 180     if (vd->sys->pool)
 181         picture_pool_Delete(vd->sys->pool);
 182
 183     CommonClean(vd);
 184
 185     free(vd->sys);
 186 }
 187
 188 /**
 189  * Handles pool allocations for bitmaps
 190  */
 191 static picture_pool_t *Pool(vout_display_t *vd, unsigned count)
 192 {
 193     vout_display_sys_t *sys = vd->sys;
 194
 195     if (!sys->pool) {
 196         sys->pool = picture_pool_NewFromFormat(&vd->fmt, count);
 197 #ifndef NDEBUG
 198         msg_Dbg(vd, "New picture pool created");
 199 #endif
 200     }
 201
 202     return sys->pool;
 203 }
 204
 205 /**
 206  * Performs set up of ID2D1Bitmap memory ready for blitting
 207  */
 208 static void Prepare(vout_display_t *vd, picture_t *picture)
 209 {
 210     vout_display_sys_t *sys = vd->sys;
 211
 212     if (sys->d2_render_target && sys->d2_bitmap) {
 213
 214         HRESULT hr = ID2D1Bitmap_CopyFromMemory(sys->d2_bitmap,
 215                                                 NULL /*&r_src*/,
 216                                                 picture->p[0].p_pixels,
 217                                                 picture->p[0].i_pitch);
 218         if (hr != S_OK)
 219             msg_Err(vd, "Failed to copy bitmap memory (hr = 0x%x)!",
 220                     (unsigned)hr);
 221
 222 #ifndef NDEBUG
 223         /*msg_Dbg(vd, "Bitmap dbg: target = %p, pitch = %d, bitmap = %p",
 224                 sys->d2_render_target, pitch, sys->d2_bitmap);*/
 225 #endif
 226     }
 227 }
 228
 229 /**
 230  * Blits a scaled picture_t to the render target
 231  */
 232 static void Display(vout_display_t *vd, picture_t *picture)
 233 {
 234     vout_display_sys_t *sys = vd->sys;
 235
 236     D2D1_RECT_F r_dest = {
 237         sys->rect_dest.left,
 238         sys->rect_dest.top,
 239         sys->rect_dest.right,
 240         sys->rect_dest.bottom
 241     };
 242
 243     if (sys->d2_render_target && sys->d2_bitmap) {
 244         ID2D1HwndRenderTarget_BeginDraw(sys->d2_render_target);
 245
 246         ID2D1HwndRenderTarget_DrawBitmap(sys->d2_render_target,
 247                                          sys->d2_bitmap,
 248                                          &r_dest,
 249                                          1.0f,
 250                                          D2D1_BITMAP_INTERPOLATION_MODE_LINEAR,
 251                                          NULL);
 252
 253         HRESULT hr = ID2D1HwndRenderTarget_EndDraw(sys->d2_render_target,
 254                                                    NULL,
 255                                                    NULL);
 256         if (hr ==  D2DERR_RECREATE_TARGET) {
 257             D2D_DestroyRenderTarget(vd);
 258             D2D_CreateRenderTarget(vd);
 259         }
 260     }
 261
 262     picture_Release(picture);
 263
 264     CommonDisplay(vd);
 265 }
 266
 267  /**
 268   * Control event handler
 269   */
 270 static int Control(vout_display_t *vd, int query, va_list args)
 271 {
 272     return CommonControl(vd, query, args);
 273 }
 274
 275 /**
 276  * Handles surface management
 277  * ID2D1RenderTargets cannot be resized and must be recreated
 278  */
 279 static void Manage(vout_display_t *vd)
 280 {
 281     vout_display_sys_t *sys = vd->sys;
 282
 283     CommonManage(vd);
 284
 285     if (sys->changes & DX_POSITION_CHANGE) {
 286         D2D_ResizeRenderTarget(vd);
 287         sys->changes &= ~DX_POSITION_CHANGE;
 288     }
 289 }
 290
 291 /**
 292  * Creates a ID2D1HwndRenderTarget and associated ID2D1Bitmap
 293  */
 294 static int D2D_CreateRenderTarget(vout_display_t *vd)
 295 {
 296     vout_display_sys_t *sys = vd->sys;
 297
 298     sys->d2_render_target = NULL;
 299
 300     D2D1_PIXEL_FORMAT pf = {
 301         DXGI_FORMAT_B8G8R8A8_UNORM,
 302         D2D1_ALPHA_MODE_IGNORE
 303     };
 304
 305     D2D1_RENDER_TARGET_PROPERTIES rtp = {
 306         D2D1_RENDER_TARGET_TYPE_DEFAULT,
 307         pf,
 308         0,
 309         0,
 310         D2D1_RENDER_TARGET_USAGE_NONE,
 311         D2D1_FEATURE_LEVEL_DEFAULT
 312     };
 313
 314     D2D1_SIZE_U size = {
 315         sys->rect_dest.right - sys->rect_dest.left,
 316         sys->rect_dest.bottom - sys->rect_dest.top
 317     };
 318
 319     D2D1_HWND_RENDER_TARGET_PROPERTIES hrtp = {
 320         sys->hvideownd,
 321         size,
 322         D2D1_PRESENT_OPTIONS_IMMEDIATELY /* this might need fiddling */
 323     };
 324
 325     HRESULT hr  = ID2D1Factory_CreateHwndRenderTarget(sys->d2_factory,
 326                                                       &rtp,
 327                                                       &hrtp,
 328                                                       &sys->d2_render_target);
 329     if (hr != S_OK) {
 330         msg_Err(vd, "Cannot create render target (hvideownd = 0x%x, width = %d, height = %d, pf.format = %d, hr = 0x%x)!",
 331                 (unsigned)hrtp.hwnd, hrtp.pixelSize.width,
 332                 hrtp.pixelSize.height, pf.format, (unsigned)hr);
 333
 334         sys->d2_render_target = NULL;
 335
 336         return VLC_EGENERIC;
 337     }
 338
 339     FLOAT dpi_x, dpi_y;
 340
 341     ID2D1Factory_GetDesktopDpi(sys->d2_factory,
 342                                &dpi_x,
 343                                &dpi_y);
 344
 345     D2D1_BITMAP_PROPERTIES bp = {
 346         pf,
 347         dpi_x,
 348         dpi_y
 349     };
 350
 351     D2D1_SIZE_U bitmap_size = {
 352         vd->fmt.i_width,
 353         vd->fmt.i_height
 354     };
 355
 356     hr = ID2D1HwndRenderTarget_CreateBitmap(sys->d2_render_target,
 357                                             bitmap_size,
 358                                             NULL,
 359                                             0,
 360                                             &bp,
 361                                             &sys->d2_bitmap);
 362     if (hr != S_OK) {
 363         msg_Err(vd, "Failed to create bitmap (hr = 0x%x)!", (unsigned)hr);
 364
 365         sys->d2_bitmap = NULL;
 366         D2D_DestroyRenderTarget(vd);
 367
 368         return VLC_EGENERIC;
 369     }
 370
 371 #ifndef NDEBUG
 372     msg_Dbg(vd, "Render trgt dbg: dpi = %f, render_target = %p, bitmap = %p",
 373             dpi_x, sys->d2_render_target, sys->d2_bitmap);
 374 #endif
 375
 376     return VLC_SUCCESS;
 377 }
 378
 379 /**
 380  * Resizes a ID2D1HWndRenderTarget
 381  */
 382 static void D2D_ResizeRenderTarget(vout_display_t *vd)
 383 {
 384     vout_display_sys_t *sys = vd->sys;
 385
 386     D2D1_SIZE_U size = {
 387         sys->rect_dest.right - sys->rect_dest.left,
 388         sys->rect_dest.bottom - sys->rect_dest.top
 389     };
 390
 391     HRESULT hr  = ID2D1HwndRenderTarget_Resize(sys->d2_render_target, &size);
 392     if (hr != S_OK)
 393         msg_Err(vd, "Cannot resize render target (width = %d, height = %d, hr = 0x%x)!",
 394                 size.width, size.height, (unsigned)hr);
 395 }
 396
 397 /**
 398  * Cleans up ID2D1HwndRenderTarget and ID2D1Bitmap
 399  */
 400 static void D2D_DestroyRenderTarget(vout_display_t *vd)
 401 {
 402     vout_display_sys_t *sys = vd->sys;
 403
 404     if (sys->d2_render_target) {
 405         ID2D1HwndRenderTarget_Release(sys->d2_render_target);
 406         sys->d2_render_target = NULL;
 407     }
 408
 409     if (sys->d2_bitmap) {
 410         ID2D1Bitmap_Release(sys->d2_bitmap);
 411         sys->d2_bitmap = NULL;
 412     }
 413
 414 #ifndef NDEBUG
 415     msg_Dbg(vd, "Destroyed");
 416 #endif
 417 }