Fix ref-counting rules to match native DCOM Dlls.
[wine] / dlls / d3d9 / query.c
1 /*
2  * IDirect3DQuery9 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21  */
22
23 #include "config.h"
24 #include "d3d9_private.h"
25
26 WINE_DEFAULT_DEBUG_CHANNEL(d3d9);
27
28 /* IDirect3DQuery9 IUnknown parts follow: */
29 HRESULT WINAPI IDirect3DQuery9Impl_QueryInterface(LPDIRECT3DQUERY9 iface, REFIID riid, LPVOID* ppobj) {
30     IDirect3DQuery9Impl *This = (IDirect3DQuery9Impl *)iface;
31     TRACE("(%p) Relay\n", This);
32
33     if (IsEqualGUID(riid, &IID_IUnknown)
34         || IsEqualGUID(riid, &IID_IDirect3DQuery9)) {
35         IUnknown_AddRef(iface);
36         *ppobj = This;
37         return D3D_OK;
38     }
39
40     WARN("(%p)->(%s,%p),not found\n", This, debugstr_guid(riid), ppobj);
41     return E_NOINTERFACE;
42 }
43
44 ULONG WINAPI IDirect3DQuery9Impl_AddRef(LPDIRECT3DQUERY9 iface) {
45     IDirect3DQuery9Impl *This = (IDirect3DQuery9Impl *)iface;
46     ULONG ref = InterlockedIncrement(&This->ref);
47
48     TRACE("(%p) : AddRef from %ld\n", This, ref - 1);
49     return ref;
50 }
51
52 ULONG WINAPI IDirect3DQuery9Impl_Release(LPDIRECT3DQUERY9 iface) {
53     IDirect3DQuery9Impl *This = (IDirect3DQuery9Impl *)iface;
54     ULONG ref = InterlockedDecrement(&This->ref);
55
56     TRACE("(%p) : ReleaseRef to %ld\n", This, ref);
57
58     if (ref == 0) {
59         HeapFree(GetProcessHeap(), 0, This);
60     }
61     return ref;
62 }
63
64 /* IDirect3DQuery9 Interface follow: */
65 HRESULT WINAPI IDirect3DQuery9Impl_GetDevice(LPDIRECT3DQUERY9 iface, IDirect3DDevice9** ppDevice) {
66     IDirect3DQuery9Impl *This = (IDirect3DQuery9Impl *)iface;
67     IWineD3DDevice* pDevice;
68     HRESULT hr;
69
70     TRACE("(%p) Relay\n", This);
71
72     hr = IWineD3DQuery_GetDevice(This->wineD3DQuery, &pDevice);
73     if(hr != D3D_OK){
74         *ppDevice = NULL;
75     }else{
76         hr = IWineD3DDevice_GetParent(pDevice, (IUnknown **)ppDevice);
77         IWineD3DDevice_Release(pDevice);
78     }
79     return hr;
80 }
81
82 D3DQUERYTYPE WINAPI IDirect3DQuery9Impl_GetType(LPDIRECT3DQUERY9 iface) {
83     IDirect3DQuery9Impl *This = (IDirect3DQuery9Impl *)iface;
84     TRACE("(%p) Relay\n", This);
85     return IWineD3DQuery_GetType(This->wineD3DQuery);
86 }
87
88 DWORD WINAPI IDirect3DQuery9Impl_GetDataSize(LPDIRECT3DQUERY9 iface) {
89     IDirect3DQuery9Impl *This = (IDirect3DQuery9Impl *)iface;
90     TRACE("(%p) Relay\n", This);
91     return IWineD3DQuery_GetDataSize(This->wineD3DQuery);
92 }
93
94 HRESULT WINAPI IDirect3DQuery9Impl_Issue(LPDIRECT3DQUERY9 iface, DWORD dwIssueFlags) {
95     IDirect3DQuery9Impl *This = (IDirect3DQuery9Impl *)iface;
96     TRACE("(%p) Relay\n", This);
97     return IWineD3DQuery_Issue(This->wineD3DQuery, dwIssueFlags);
98 }
99
100 HRESULT WINAPI IDirect3DQuery9Impl_GetData(LPDIRECT3DQUERY9 iface, void* pData, DWORD dwSize, DWORD dwGetDataFlags) {
101     IDirect3DQuery9Impl *This = (IDirect3DQuery9Impl *)iface;
102     TRACE("(%p) Relay\n", This);
103     return IWineD3DQuery_GetData(This->wineD3DQuery, pData, dwSize, dwGetDataFlags);
104 }
105
106
107 const IDirect3DQuery9Vtbl Direct3DQuery9_Vtbl =
108 {
109     IDirect3DQuery9Impl_QueryInterface,
110     IDirect3DQuery9Impl_AddRef,
111     IDirect3DQuery9Impl_Release,
112     IDirect3DQuery9Impl_GetDevice,
113     IDirect3DQuery9Impl_GetType,
114     IDirect3DQuery9Impl_GetDataSize,
115     IDirect3DQuery9Impl_Issue,
116     IDirect3DQuery9Impl_GetData
117 };
118
119
120 /* IDirect3DDevice9 IDirect3DQuery9 Methods follow: */
121 HRESULT WINAPI IDirect3DDevice9Impl_CreateQuery(LPDIRECT3DDEVICE9 iface, D3DQUERYTYPE Type, IDirect3DQuery9** ppQuery) {
122   IDirect3DDevice9Impl *This = (IDirect3DDevice9Impl *)iface;
123   IDirect3DQuery9Impl *object = NULL;
124   HRESULT hr = D3D_OK;
125
126   TRACE("(%p) Relay\n", This);
127   if (NULL == ppQuery) {
128     return D3DERR_INVALIDCALL;
129   }
130   /* Allocate the storage for the device */
131   object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IDirect3DQuery9Impl));
132   if (NULL == object) {
133     FIXME("Allocation of memory failed\n");
134     *ppQuery = NULL;
135     return D3DERR_OUTOFVIDEOMEMORY;
136   }
137   
138   object->lpVtbl = &Direct3DQuery9_Vtbl;
139   object->ref = 1;
140   hr = IWineD3DDevice_CreateQuery(This->WineD3DDevice, Type, &(object->wineD3DQuery), (IUnknown*)object);
141   
142   if (FAILED(hr)) {
143     /* free up object */ 
144     FIXME("(%p) call to IWineD3DDevice_CreateQuery failed\n", This);
145     HeapFree(GetProcessHeap(), 0, object);
146     *ppQuery = NULL;
147   } else {
148     *ppQuery = (LPDIRECT3DQUERY9) object;
149   }
150     TRACE("(%p) : returning %p \n", This, *ppQuery);
151   return hr;
152 }