wined3d: Consistently use WINED3DRESOURCETYPE.
[wine] / dlls / d3d9 / pixelshader.c
1 /*
2  * IDirect3DPixelShader9 implementation
3  *
4  * Copyright 2002-2003 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20  */
21
22 #include "config.h"
23 #include "d3d9_private.h"
24
25 WINE_DEFAULT_DEBUG_CHANNEL(d3d9);
26
27 /* IDirect3DPixelShader9 IUnknown parts follow: */
28 HRESULT WINAPI IDirect3DPixelShader9Impl_QueryInterface(LPDIRECT3DPIXELSHADER9 iface, REFIID riid, LPVOID* ppobj) {
29     IDirect3DPixelShader9Impl *This = (IDirect3DPixelShader9Impl *)iface;
30
31     if (IsEqualGUID(riid, &IID_IUnknown)
32         || IsEqualGUID(riid, &IID_IDirect3DPixelShader9)) {
33         IUnknown_AddRef(iface);
34         *ppobj = This;
35         return D3D_OK;
36     }
37
38     WARN("(%p)->(%s,%p),not found\n", This, debugstr_guid(riid), ppobj);
39     return E_NOINTERFACE;
40 }
41
42 ULONG WINAPI IDirect3DPixelShader9Impl_AddRef(LPDIRECT3DPIXELSHADER9 iface) {
43     IDirect3DPixelShader9Impl *This = (IDirect3DPixelShader9Impl *)iface;
44     ULONG ref = InterlockedIncrement(&This->ref);
45
46     TRACE("(%p) : AddRef from %ld\n", This, ref - 1);
47
48     return ref;
49 }
50
51 ULONG WINAPI IDirect3DPixelShader9Impl_Release(LPDIRECT3DPIXELSHADER9 iface) {
52     IDirect3DPixelShader9Impl *This = (IDirect3DPixelShader9Impl *)iface;
53     ULONG ref = InterlockedDecrement(&This->ref);
54
55     TRACE("(%p) : ReleaseRef to %ld\n", This, ref);
56
57     if (ref == 0) {
58         IWineD3DPixelShader_Release(This->wineD3DPixelShader);
59         HeapFree(GetProcessHeap(), 0, This);
60     }
61     return ref;
62 }
63
64 /* IDirect3DPixelShader9 Interface follow: */
65 HRESULT WINAPI IDirect3DPixelShader9Impl_GetDevice(LPDIRECT3DPIXELSHADER9 iface, IDirect3DDevice9** ppDevice) {
66     IDirect3DPixelShader9Impl *This = (IDirect3DPixelShader9Impl *)iface;
67     IWineD3DDevice *myDevice = NULL;
68
69     TRACE("(%p) : Relay\n", This);
70
71     IWineD3DPixelShader_GetDevice(This->wineD3DPixelShader, &myDevice);
72     IWineD3DDevice_GetParent(myDevice, (IUnknown **)ppDevice);
73     IWineD3DDevice_Release(myDevice);
74     TRACE("(%p) returing (%p)", This, *ppDevice);
75     return D3D_OK;
76 }
77
78 HRESULT WINAPI IDirect3DPixelShader9Impl_GetFunction(LPDIRECT3DPIXELSHADER9 iface, VOID* pData, UINT* pSizeOfData) {
79     IDirect3DPixelShader9Impl *This = (IDirect3DPixelShader9Impl *)iface;
80     TRACE("(%p) Relay\n", This);
81     return IWineD3DPixelShader_GetFunction(This->wineD3DPixelShader, pData, pSizeOfData);
82 }
83
84
85 const IDirect3DPixelShader9Vtbl Direct3DPixelShader9_Vtbl =
86 {
87     /* IUnknown */
88     IDirect3DPixelShader9Impl_QueryInterface,
89     IDirect3DPixelShader9Impl_AddRef,
90     IDirect3DPixelShader9Impl_Release,
91     /* IDirect3DPixelShader9 */
92     IDirect3DPixelShader9Impl_GetDevice,
93     IDirect3DPixelShader9Impl_GetFunction
94 };
95
96
97 /* IDirect3DDevice9 IDirect3DPixelShader9 Methods follow:  */
98 HRESULT WINAPI IDirect3DDevice9Impl_CreatePixelShader(LPDIRECT3DDEVICE9 iface, CONST DWORD* pFunction, IDirect3DPixelShader9** ppShader) {
99     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
100     IDirect3DPixelShader9Impl *object;
101     HRESULT hrc = D3D_OK;
102
103     FIXME("(%p) Relay (disabled)\n", This);
104
105     if (ppShader == NULL) {
106         TRACE("(%p) Invalid call\n", This);
107         return D3DERR_INVALIDCALL;
108     }
109
110     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*object));
111
112     if (NULL == object) {
113         FIXME("Allocation of memory failed, returning D3DERR_OUTOFVIDEOMEMORY\n");
114         return E_OUTOFMEMORY;
115     }
116
117     object->ref    = 1;
118     object->lpVtbl = &Direct3DPixelShader9_Vtbl;
119     hrc = IWineD3DDevice_CreatePixelShader(This->WineD3DDevice, pFunction, &object->wineD3DPixelShader , (IUnknown *)object);
120     if (hrc != D3D_OK) {
121
122         /* free up object */
123         FIXME("(%p) call to IWineD3DDevice_CreatePixelShader failed\n", This);
124         HeapFree(GetProcessHeap(), 0 , object);
125     } else {
126         *ppShader = (IDirect3DPixelShader9*) object;
127         TRACE("(%p) : Created pixel shader %p\n", This, object);
128     }
129
130     TRACE("(%p) : returning %p\n", This, *ppShader);
131     return hrc;
132 }
133
134 HRESULT WINAPI IDirect3DDevice9Impl_SetPixelShader(LPDIRECT3DDEVICE9 iface, IDirect3DPixelShader9* pShader) {
135     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
136     IDirect3DPixelShader9Impl *shader = (IDirect3DPixelShader9Impl *)pShader;
137     TRACE("(%p) Relay\n", This);
138     IWineD3DDevice_SetPixelShader(This->WineD3DDevice, shader == NULL ? NULL :shader->wineD3DPixelShader);
139     return D3D_OK;
140 }
141
142 HRESULT WINAPI IDirect3DDevice9Impl_GetPixelShader(LPDIRECT3DDEVICE9 iface, IDirect3DPixelShader9** ppShader) {
143     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
144     IWineD3DPixelShader *object;
145
146     HRESULT hrc = D3D_OK;
147     TRACE("(%p) Relay\n", This);
148     if (ppShader == NULL) {
149         TRACE("(%p) Invalid call\n", This);
150         return D3DERR_INVALIDCALL;
151     }
152
153     hrc = IWineD3DDevice_GetPixelShader(This->WineD3DDevice, &object);
154     if (hrc == D3D_OK && object != NULL) {
155        hrc = IWineD3DPixelShader_GetParent(object, (IUnknown **)ppShader);
156        IWineD3DPixelShader_Release(object);
157     } else {
158         *ppShader = NULL;
159     }
160
161     TRACE("(%p) : returning %p\n", This, *ppShader);
162     return hrc;
163 }
164
165 HRESULT WINAPI IDirect3DDevice9Impl_SetPixelShaderConstantF(LPDIRECT3DDEVICE9 iface, UINT Register, CONST float* pConstantData, UINT Vector4fCount) {
166    IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
167     TRACE("(%p) Relay\n", This);   
168     return IWineD3DDevice_SetPixelShaderConstantF(This->WineD3DDevice, Register, pConstantData, Vector4fCount);
169 }
170
171 HRESULT WINAPI IDirect3DDevice9Impl_GetPixelShaderConstantF(LPDIRECT3DDEVICE9 iface, UINT Register, float* pConstantData, UINT Vector4fCount) {
172     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
173     TRACE("(%p) Relay\n", This);
174     return IWineD3DDevice_GetPixelShaderConstantF(This->WineD3DDevice, Register, pConstantData, Vector4fCount);
175 }
176
177 HRESULT WINAPI IDirect3DDevice9Impl_SetPixelShaderConstantI(LPDIRECT3DDEVICE9 iface, UINT Register, CONST int* pConstantData, UINT Vector4iCount) {
178     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
179     TRACE("(%p) Relay\n", This);
180     return IWineD3DDevice_SetPixelShaderConstantI(This->WineD3DDevice, Register, pConstantData, Vector4iCount);
181 }
182
183 HRESULT WINAPI IDirect3DDevice9Impl_GetPixelShaderConstantI(LPDIRECT3DDEVICE9 iface, UINT Register, int* pConstantData, UINT Vector4iCount) {
184     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
185     TRACE("(%p) Relay\n", This);
186     return IWineD3DDevice_GetPixelShaderConstantI(This->WineD3DDevice, Register, pConstantData, Vector4iCount);
187 }
188
189 HRESULT WINAPI IDirect3DDevice9Impl_SetPixelShaderConstantB(LPDIRECT3DDEVICE9 iface, UINT Register, CONST BOOL* pConstantData, UINT BoolCount) {
190     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
191     TRACE("(%p) Relay\n", This);
192     return  IWineD3DDevice_SetPixelShaderConstantB(This->WineD3DDevice, Register, pConstantData, BoolCount);
193 }
194
195 HRESULT WINAPI IDirect3DDevice9Impl_GetPixelShaderConstantB(LPDIRECT3DDEVICE9 iface, UINT Register, BOOL* pConstantData, UINT BoolCount) {
196     IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
197     TRACE("(%p) Relay\n", This);
198     return IWineD3DDevice_GetPixelShaderConstantB(This->WineD3DDevice, Register, pConstantData, BoolCount);
199 }