dlls/d3d9/surface.c

   1 /*
   2  * IDirect3DSurface9 implementation
   3  *
   4  * Copyright 2002-2005 Jason Edmeades
   5  *                     Raphael Junqueira
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include "config.h"
  23 #include "d3d9_private.h"
  24
  25 WINE_DEFAULT_DEBUG_CHANNEL(d3d9);
  26
  27 static inline struct d3d9_surface *impl_from_IDirect3DSurface9(IDirect3DSurface9 *iface)
  28 {
  29     return CONTAINING_RECORD(iface, struct d3d9_surface, IDirect3DSurface9_iface);
  30 }
  31
  32 static HRESULT WINAPI d3d9_surface_QueryInterface(IDirect3DSurface9 *iface, REFIID riid, void **out)
  33 {
  34     TRACE("iface %p, riid %s, out %p.\n", iface, debugstr_guid(riid), out);
  35
  36     if (IsEqualGUID(riid, &IID_IDirect3DSurface9)
  37             || IsEqualGUID(riid, &IID_IDirect3DResource9)
  38             || IsEqualGUID(riid, &IID_IUnknown))
  39     {
  40         IDirect3DSurface9_AddRef(iface);
  41         *out = iface;
  42         return S_OK;
  43     }
  44
  45     WARN("%s not implemented, returning E_NOINTERFACE.\n", debugstr_guid(riid));
  46
  47     *out = NULL;
  48     return E_NOINTERFACE;
  49 }
  50
  51 static ULONG WINAPI d3d9_surface_AddRef(IDirect3DSurface9 *iface)
  52 {
  53     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
  54     ULONG refcount;
  55
  56     TRACE("iface %p.\n", iface);
  57
  58     if (surface->texture)
  59     {
  60         TRACE("Forwarding to %p.\n", surface->texture);
  61         return IDirect3DBaseTexture9_AddRef(&surface->texture->IDirect3DBaseTexture9_iface);
  62     }
  63
  64     refcount = InterlockedIncrement(&surface->resource.refcount);
  65     TRACE("%p increasing refcount to %u.\n", iface, refcount);
  66
  67     if (refcount == 1)
  68     {
  69         if (surface->parent_device)
  70             IDirect3DDevice9Ex_AddRef(surface->parent_device);
  71         wined3d_mutex_lock();
  72         if (surface->wined3d_rtv)
  73             wined3d_rendertarget_view_incref(surface->wined3d_rtv);
  74         wined3d_surface_incref(surface->wined3d_surface);
  75         wined3d_mutex_unlock();
  76     }
  77
  78     return refcount;
  79 }
  80
  81 static ULONG WINAPI d3d9_surface_Release(IDirect3DSurface9 *iface)
  82 {
  83     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
  84     ULONG refcount;
  85
  86     TRACE("iface %p.\n", iface);
  87
  88     if (surface->texture)
  89     {
  90         TRACE("Forwarding to %p.\n", surface->texture);
  91         return IDirect3DBaseTexture9_Release(&surface->texture->IDirect3DBaseTexture9_iface);
  92     }
  93
  94     refcount = InterlockedDecrement(&surface->resource.refcount);
  95     TRACE("%p decreasing refcount to %u.\n", iface, refcount);
  96
  97     if (!refcount)
  98     {
  99         IDirect3DDevice9Ex *parent_device = surface->parent_device;
 100
 101         wined3d_mutex_lock();
 102         if (surface->wined3d_rtv)
 103             wined3d_rendertarget_view_decref(surface->wined3d_rtv);
 104         wined3d_surface_decref(surface->wined3d_surface);
 105         wined3d_mutex_unlock();
 106
 107         /* Release the device last, as it may cause the device to be destroyed. */
 108         if (parent_device)
 109             IDirect3DDevice9Ex_Release(parent_device);
 110     }
 111
 112     return refcount;
 113 }
 114
 115 static HRESULT WINAPI d3d9_surface_GetDevice(IDirect3DSurface9 *iface, IDirect3DDevice9 **device)
 116 {
 117     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 118
 119     TRACE("iface %p, device %p.\n", iface, device);
 120
 121     if (surface->texture)
 122         return IDirect3DBaseTexture9_GetDevice(&surface->texture->IDirect3DBaseTexture9_iface, device);
 123
 124     *device = (IDirect3DDevice9 *)surface->parent_device;
 125     IDirect3DDevice9_AddRef(*device);
 126
 127     TRACE("Returning device %p.\n", *device);
 128
 129     return D3D_OK;
 130 }
 131
 132 static HRESULT WINAPI d3d9_surface_SetPrivateData(IDirect3DSurface9 *iface, REFGUID guid,
 133         const void *data, DWORD data_size, DWORD flags)
 134 {
 135     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 136     TRACE("iface %p, guid %s, data %p, data_size %u, flags %#x.\n",
 137             iface, debugstr_guid(guid), data, data_size, flags);
 138
 139     return d3d9_resource_set_private_data(&surface->resource, guid, data, data_size, flags);
 140 }
 141
 142 static HRESULT WINAPI d3d9_surface_GetPrivateData(IDirect3DSurface9 *iface, REFGUID guid,
 143         void *data, DWORD *data_size)
 144 {
 145     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 146     TRACE("iface %p, guid %s, data %p, data_size %p.\n",
 147             iface, debugstr_guid(guid), data, data_size);
 148
 149     return d3d9_resource_get_private_data(&surface->resource, guid, data, data_size);
 150 }
 151
 152 static HRESULT WINAPI d3d9_surface_FreePrivateData(IDirect3DSurface9 *iface, REFGUID guid)
 153 {
 154     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 155     TRACE("iface %p, guid %s.\n", iface, debugstr_guid(guid));
 156
 157     return d3d9_resource_free_private_data(&surface->resource, guid);
 158 }
 159
 160 static DWORD WINAPI d3d9_surface_SetPriority(IDirect3DSurface9 *iface, DWORD priority)
 161 {
 162     TRACE("iface %p, priority %u. Ignored on surfaces.\n", iface, priority);
 163     return 0;
 164 }
 165
 166 static DWORD WINAPI d3d9_surface_GetPriority(IDirect3DSurface9 *iface)
 167 {
 168     TRACE("iface %p. Ignored on surfaces.\n", iface);
 169     return 0;
 170 }
 171
 172 static void WINAPI d3d9_surface_PreLoad(IDirect3DSurface9 *iface)
 173 {
 174     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 175
 176     TRACE("iface %p.\n", iface);
 177
 178     wined3d_mutex_lock();
 179     wined3d_texture_preload(surface->wined3d_texture);
 180     wined3d_mutex_unlock();
 181 }
 182
 183 static D3DRESOURCETYPE WINAPI d3d9_surface_GetType(IDirect3DSurface9 *iface)
 184 {
 185     TRACE("iface %p.\n", iface);
 186
 187     return D3DRTYPE_SURFACE;
 188 }
 189
 190 static HRESULT WINAPI d3d9_surface_GetContainer(IDirect3DSurface9 *iface, REFIID riid, void **container)
 191 {
 192     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 193     HRESULT hr;
 194
 195     TRACE("iface %p, riid %s, container %p.\n", iface, debugstr_guid(riid), container);
 196
 197     if (!surface->container)
 198         return E_NOINTERFACE;
 199
 200     hr = IUnknown_QueryInterface(surface->container, riid, container);
 201
 202     TRACE("Returning %p.\n", *container);
 203
 204     return hr;
 205 }
 206
 207 static HRESULT WINAPI d3d9_surface_GetDesc(IDirect3DSurface9 *iface, D3DSURFACE_DESC *desc)
 208 {
 209     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 210     struct wined3d_resource_desc wined3d_desc;
 211     struct wined3d_resource *sub_resource;
 212
 213     TRACE("iface %p, desc %p.\n", iface, desc);
 214
 215     wined3d_mutex_lock();
 216     sub_resource = wined3d_texture_get_sub_resource(surface->wined3d_texture, surface->sub_resource_idx);
 217     wined3d_resource_get_desc(sub_resource, &wined3d_desc);
 218     wined3d_mutex_unlock();
 219
 220     desc->Format = d3dformat_from_wined3dformat(wined3d_desc.format);
 221     desc->Type = wined3d_desc.resource_type;
 222     desc->Usage = wined3d_desc.usage & WINED3DUSAGE_MASK;
 223     desc->Pool = wined3d_desc.pool;
 224     desc->MultiSampleType = wined3d_desc.multisample_type;
 225     desc->MultiSampleQuality = wined3d_desc.multisample_quality;
 226     desc->Width = wined3d_desc.width;
 227     desc->Height = wined3d_desc.height;
 228
 229     return D3D_OK;
 230 }
 231
 232 static HRESULT WINAPI d3d9_surface_LockRect(IDirect3DSurface9 *iface,
 233         D3DLOCKED_RECT *locked_rect, const RECT *rect, DWORD flags)
 234 {
 235     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 236     struct wined3d_box box;
 237     struct wined3d_map_desc map_desc;
 238     HRESULT hr;
 239
 240     TRACE("iface %p, locked_rect %p, rect %s, flags %#x.\n",
 241             iface, locked_rect, wine_dbgstr_rect(rect), flags);
 242
 243     if (rect)
 244     {
 245         box.left = rect->left;
 246         box.top = rect->top;
 247         box.right = rect->right;
 248         box.bottom = rect->bottom;
 249         box.front = 0;
 250         box.back = 1;
 251     }
 252
 253     wined3d_mutex_lock();
 254     hr = wined3d_resource_map(wined3d_texture_get_resource(surface->wined3d_texture), surface->sub_resource_idx,
 255             &map_desc, rect ? &box : NULL, flags);
 256     wined3d_mutex_unlock();
 257
 258     if (SUCCEEDED(hr))
 259     {
 260         locked_rect->Pitch = map_desc.row_pitch;
 261         locked_rect->pBits = map_desc.data;
 262     }
 263
 264     return hr;
 265 }
 266
 267 static HRESULT WINAPI d3d9_surface_UnlockRect(IDirect3DSurface9 *iface)
 268 {
 269     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 270     HRESULT hr;
 271
 272     TRACE("iface %p.\n", iface);
 273
 274     wined3d_mutex_lock();
 275     hr = wined3d_resource_unmap(wined3d_texture_get_resource(surface->wined3d_texture), surface->sub_resource_idx);
 276     wined3d_mutex_unlock();
 277
 278     switch(hr)
 279     {
 280         case WINEDDERR_NOTLOCKED:       return D3DERR_INVALIDCALL;
 281         default:                        return hr;
 282     }
 283 }
 284
 285 static HRESULT WINAPI d3d9_surface_GetDC(IDirect3DSurface9 *iface, HDC *dc)
 286 {
 287     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 288     HRESULT hr;
 289
 290     TRACE("iface %p, dc %p.\n", iface, dc);
 291
 292     if (!surface->getdc_supported)
 293     {
 294         WARN("Surface does not support GetDC, returning D3DERR_INVALIDCALL\n");
 295         /* Don't touch the DC */
 296         return D3DERR_INVALIDCALL;
 297     }
 298
 299     wined3d_mutex_lock();
 300     hr = wined3d_texture_get_dc(surface->wined3d_texture, surface->sub_resource_idx, dc);
 301     wined3d_mutex_unlock();
 302
 303     return hr;
 304 }
 305
 306 static HRESULT WINAPI d3d9_surface_ReleaseDC(IDirect3DSurface9 *iface, HDC dc)
 307 {
 308     struct d3d9_surface *surface = impl_from_IDirect3DSurface9(iface);
 309     HRESULT hr;
 310
 311     TRACE("iface %p, dc %p.\n", iface, dc);
 312
 313     wined3d_mutex_lock();
 314     hr = wined3d_texture_release_dc(surface->wined3d_texture, surface->sub_resource_idx, dc);
 315     wined3d_mutex_unlock();
 316
 317     switch (hr)
 318     {
 319         case WINEDDERR_NODC:    return D3DERR_INVALIDCALL;
 320         default:                return hr;
 321     }
 322 }
 323
 324 static const struct IDirect3DSurface9Vtbl d3d9_surface_vtbl =
 325 {
 326     /* IUnknown */
 327     d3d9_surface_QueryInterface,
 328     d3d9_surface_AddRef,
 329     d3d9_surface_Release,
 330     /* IDirect3DResource9 */
 331     d3d9_surface_GetDevice,
 332     d3d9_surface_SetPrivateData,
 333     d3d9_surface_GetPrivateData,
 334     d3d9_surface_FreePrivateData,
 335     d3d9_surface_SetPriority,
 336     d3d9_surface_GetPriority,
 337     d3d9_surface_PreLoad,
 338     d3d9_surface_GetType,
 339     /* IDirect3DSurface9 */
 340     d3d9_surface_GetContainer,
 341     d3d9_surface_GetDesc,
 342     d3d9_surface_LockRect,
 343     d3d9_surface_UnlockRect,
 344     d3d9_surface_GetDC,
 345     d3d9_surface_ReleaseDC,
 346 };
 347
 348 static void STDMETHODCALLTYPE surface_wined3d_object_destroyed(void *parent)
 349 {
 350     struct d3d9_surface *surface = parent;
 351     d3d9_resource_cleanup(&surface->resource);
 352     HeapFree(GetProcessHeap(), 0, surface);
 353 }
 354
 355 static const struct wined3d_parent_ops d3d9_surface_wined3d_parent_ops =
 356 {
 357     surface_wined3d_object_destroyed,
 358 };
 359
 360 void surface_init(struct d3d9_surface *surface, struct wined3d_texture *wined3d_texture, unsigned int sub_resource_idx,
 361         struct wined3d_surface *wined3d_surface, const struct wined3d_parent_ops **parent_ops)
 362 {
 363     struct wined3d_resource_desc desc;
 364     IDirect3DBaseTexture9 *texture;
 365
 366     surface->IDirect3DSurface9_iface.lpVtbl = &d3d9_surface_vtbl;
 367     d3d9_resource_init(&surface->resource);
 368     surface->resource.refcount = 0;
 369     surface->wined3d_surface = wined3d_surface;
 370     list_init(&surface->rtv_entry);
 371     surface->container = wined3d_texture_get_parent(wined3d_texture);
 372     surface->wined3d_texture = wined3d_texture;
 373     surface->sub_resource_idx = sub_resource_idx;
 374
 375     if (surface->container && SUCCEEDED(IUnknown_QueryInterface(surface->container,
 376             &IID_IDirect3DBaseTexture9, (void **)&texture)))
 377     {
 378         surface->texture = unsafe_impl_from_IDirect3DBaseTexture9(texture);
 379         IDirect3DBaseTexture9_Release(texture);
 380     }
 381
 382     wined3d_resource_get_desc(wined3d_texture_get_resource(wined3d_texture), &desc);
 383     switch (d3dformat_from_wined3dformat(desc.format))
 384     {
 385         case D3DFMT_A8R8G8B8:
 386         case D3DFMT_X8R8G8B8:
 387         case D3DFMT_R5G6B5:
 388         case D3DFMT_X1R5G5B5:
 389         case D3DFMT_A1R5G5B5:
 390         case D3DFMT_R8G8B8:
 391             surface->getdc_supported = TRUE;
 392             break;
 393
 394         default:
 395             surface->getdc_supported = FALSE;
 396             break;
 397     }
 398
 399     *parent_ops = &d3d9_surface_wined3d_parent_ops;
 400 }
 401
 402 static void STDMETHODCALLTYPE view_wined3d_object_destroyed(void *parent)
 403 {
 404     struct d3d9_surface *surface = parent;
 405
 406     /* If the surface reference count drops to zero, we release our reference
 407      * to the view, but don't clear the pointer yet, in case e.g. a
 408      * GetRenderTarget() call brings the surface back before the view is
 409      * actually destroyed. When the view is destroyed, we need to clear the
 410      * pointer, or a subsequent surface AddRef() would reference it again.
 411      *
 412      * This is safe because as long as the view still has a reference to the
 413      * texture, the surface is also still alive, and we're called before the
 414      * view releases that reference. */
 415     surface->wined3d_rtv = NULL;
 416     list_remove(&surface->rtv_entry);
 417 }
 418
 419 static const struct wined3d_parent_ops d3d9_view_wined3d_parent_ops =
 420 {
 421     view_wined3d_object_destroyed,
 422 };
 423
 424 struct wined3d_rendertarget_view *d3d9_surface_get_rendertarget_view(struct d3d9_surface *surface)
 425 {
 426     HRESULT hr;
 427
 428     if (surface->wined3d_rtv)
 429         return surface->wined3d_rtv;
 430
 431     if (FAILED(hr = wined3d_rendertarget_view_create_from_sub_resource(surface->wined3d_texture,
 432             surface->sub_resource_idx, surface, &d3d9_view_wined3d_parent_ops, &surface->wined3d_rtv)))
 433     {
 434         ERR("Failed to create rendertarget view, hr %#x.\n", hr);
 435         return NULL;
 436     }
 437
 438     if (surface->texture)
 439         list_add_head(&surface->texture->rtv_list, &surface->rtv_entry);
 440
 441     return surface->wined3d_rtv;
 442 }
 443
 444 struct d3d9_surface *unsafe_impl_from_IDirect3DSurface9(IDirect3DSurface9 *iface)
 445 {
 446     if (!iface)
 447         return NULL;
 448     assert(iface->lpVtbl == &d3d9_surface_vtbl);
 449
 450     return impl_from_IDirect3DSurface9(iface);
 451 }