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