d3d9: StateBlock is also created in EndStateBlock. With tests.
[wine] / dlls / d3d9 / stateblock.c
1 /*
2  * IDirect3DStateBlock9 implementation
3  *
4  * Copyright 2002-2003 Raphael Junqueira
5  * Copyright 2002-2003 Jason Edmeades
6  * Copyright 2005 Oliver Stieber
7  *
8  * This library is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * This library is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with this library; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21  */
22
23 #include "config.h"
24 #include "d3d9_private.h"
25
26 WINE_DEFAULT_DEBUG_CHANNEL(d3d9);
27
28 /* IDirect3DStateBlock9 IUnknown parts follow: */
29 HRESULT WINAPI IDirect3DStateBlock9Impl_QueryInterface(LPDIRECT3DSTATEBLOCK9 iface, REFIID riid, LPVOID* ppobj) {
30     IDirect3DStateBlock9Impl *This = (IDirect3DStateBlock9Impl *)iface;
31
32     if (IsEqualGUID(riid, &IID_IUnknown)
33         || IsEqualGUID(riid, &IID_IDirect3DStateBlock9)) {
34         IUnknown_AddRef(iface);
35         *ppobj = This;
36         return D3D_OK;
37     }
38
39     WARN("(%p)->(%s,%p),not found\n", This, debugstr_guid(riid), ppobj);
40     return E_NOINTERFACE;
41 }
42
43 ULONG WINAPI IDirect3DStateBlock9Impl_AddRef(LPDIRECT3DSTATEBLOCK9 iface) {
44     IDirect3DStateBlock9Impl *This = (IDirect3DStateBlock9Impl *)iface;
45     ULONG ref = InterlockedIncrement(&This->ref);
46
47     TRACE("(%p) : AddRef from %ld\n", This, ref - 1);
48
49     return ref;
50 }
51
52 ULONG WINAPI IDirect3DStateBlock9Impl_Release(LPDIRECT3DSTATEBLOCK9 iface) {
53     IDirect3DStateBlock9Impl *This = (IDirect3DStateBlock9Impl *)iface;
54     ULONG ref = InterlockedDecrement(&This->ref);
55
56     TRACE("(%p) : ReleaseRef to %ld\n", This, ref);
57
58     if (ref == 0) {
59         IWineD3DStateBlock_Release(This->wineD3DStateBlock);    
60         IUnknown_Release(This->parentDevice);
61         HeapFree(GetProcessHeap(), 0, This);
62     }
63     return ref;
64 }
65
66 /* IDirect3DStateBlock9 Interface follow: */
67 HRESULT WINAPI IDirect3DStateBlock9Impl_GetDevice(LPDIRECT3DSTATEBLOCK9 iface, IDirect3DDevice9** ppDevice) {
68     IDirect3DStateBlock9Impl *This = (IDirect3DStateBlock9Impl *)iface;
69     TRACE("(%p) Relay\n", This); 
70     return IDirect3DResource9Impl_GetDevice((LPDIRECT3DRESOURCE9) This, ppDevice);
71 }
72
73 HRESULT WINAPI IDirect3DStateBlock9Impl_Capture(LPDIRECT3DSTATEBLOCK9 iface) {
74     IDirect3DStateBlock9Impl *This = (IDirect3DStateBlock9Impl *)iface;
75     TRACE("(%p) Relay\n", This); 
76     return IWineD3DStateBlock_Capture(This->wineD3DStateBlock);
77 }
78
79 HRESULT WINAPI IDirect3DStateBlock9Impl_Apply(LPDIRECT3DSTATEBLOCK9 iface) {
80     IDirect3DStateBlock9Impl *This = (IDirect3DStateBlock9Impl *)iface;
81     TRACE("(%p) Relay\n", This); 
82     return IWineD3DStateBlock_Apply(This->wineD3DStateBlock);
83 }
84
85
86 const IDirect3DStateBlock9Vtbl Direct3DStateBlock9_Vtbl =
87 {
88     /* IUnknown */
89     IDirect3DStateBlock9Impl_QueryInterface,
90     IDirect3DStateBlock9Impl_AddRef,
91     IDirect3DStateBlock9Impl_Release,
92     /* IDirect3DStateBlock9 */
93     IDirect3DStateBlock9Impl_GetDevice,
94     IDirect3DStateBlock9Impl_Capture,
95     IDirect3DStateBlock9Impl_Apply
96 };
97
98
99 /* IDirect3DDevice9 IDirect3DStateBlock9 Methods follow: */
100 HRESULT WINAPI IDirect3DDevice9Impl_CreateStateBlock(LPDIRECT3DDEVICE9 iface, D3DSTATEBLOCKTYPE Type, IDirect3DStateBlock9** ppStateBlock) {
101    IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
102    IDirect3DStateBlock9Impl* object;
103    HRESULT hrc = D3D_OK;
104    
105    TRACE("(%p) Relay\n", This);
106    
107    object  = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirect3DStateBlock9Impl));
108    if (NULL == object) {
109       FIXME("(%p)  Failed to allocate %d bytes\n", This, sizeof(IDirect3DStateBlock9Impl));
110       return E_OUTOFMEMORY;
111    }
112    object->lpVtbl = &Direct3DStateBlock9_Vtbl;
113    object->ref = 1;
114    
115    hrc = IWineD3DDevice_CreateStateBlock(This->WineD3DDevice, (WINED3DSTATEBLOCKTYPE)Type, &object->wineD3DStateBlock, (IUnknown*)object);
116    if(hrc != D3D_OK){
117        FIXME("(%p) Call to IWineD3DDevice_CreateStateBlock failed.\n", This);
118        HeapFree(GetProcessHeap(), 0, object);
119    } else {
120         IUnknown_AddRef(iface);
121         object->parentDevice = iface;
122         *ppStateBlock = (IDirect3DStateBlock9*)object;
123         TRACE("(%p) : Created stateblock %p\n", This, object);
124    }
125    TRACE("(%p) returning token (ptr to stateblock) of %p\n", This, object);
126    return hrc;
127 }
128
129 HRESULT  WINAPI  IDirect3DDevice9Impl_BeginStateBlock(LPDIRECT3DDEVICE9 iface) {
130     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;    
131     TRACE("(%p) Relay\n", This); 
132     return IWineD3DDevice_BeginStateBlock(This->WineD3DDevice);
133 }
134
135 HRESULT  WINAPI  IDirect3DDevice9Impl_EndStateBlock(LPDIRECT3DDEVICE9 iface, IDirect3DStateBlock9** ppSB) {
136     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;   
137     HRESULT hr;
138     IWineD3DStateBlock* wineD3DStateBlock;    
139     IDirect3DStateBlock9Impl* object;
140
141     TRACE("(%p) Relay\n", This); 
142     
143     /* Tell wineD3D to endstatablock before anything else (in case we run out
144      * of memory later and cause locking problems)
145      */
146     hr=IWineD3DDevice_EndStateBlock(This->WineD3DDevice,&wineD3DStateBlock);
147     if(hr!= D3D_OK){
148        FIXME("IWineD3DDevice_EndStateBlock returned an error\n");
149        return hr;
150     }    
151     /* allocate a new IDirectD3DStateBlock */
152     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY ,sizeof(IDirect3DStateBlock9Impl));      
153     if (!object) {
154         FIXME("(%p)  Failed to allocate %d bytes\n", This, sizeof(IDirect3DStateBlock9Impl));
155         return E_OUTOFMEMORY;
156     }
157     object->ref = 1;
158     object->lpVtbl = &Direct3DStateBlock9_Vtbl;
159     object->wineD3DStateBlock = wineD3DStateBlock;
160
161     IUnknown_AddRef(iface);
162     object->parentDevice = iface;
163     *ppSB=(IDirect3DStateBlock9*)object;
164     TRACE("(%p)Returning %p %p\n", This, *ppSB, wineD3DStateBlock);
165     return D3D_OK;
166 }