quartz: Make the EnumPins interface dynamic.
[wine] / dlls / quartz / enumpins.c
1 /*
2  * Implementation of IEnumPins Interface
3  *
4  * Copyright 2003 Robert Shearman
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include "quartz_private.h"
22
23 #include "wine/debug.h"
24
25 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
26
27 typedef struct IEnumPinsImpl
28 {
29     const IEnumPinsVtbl * lpVtbl;
30     LONG refCount;
31     ULONG uIndex;
32     IBaseFilter *base;
33     FNOBTAINPIN receive_pin;
34     DWORD synctime;
35 } IEnumPinsImpl;
36
37 static const struct IEnumPinsVtbl IEnumPinsImpl_Vtbl;
38
39 HRESULT IEnumPinsImpl_Construct(IEnumPins ** ppEnum, FNOBTAINPIN receive_pin, IBaseFilter *base)
40 {
41     IEnumPinsImpl * pEnumPins;
42
43     if (!ppEnum)
44         return E_POINTER;
45
46     pEnumPins = CoTaskMemAlloc(sizeof(IEnumPinsImpl));
47     if (!pEnumPins)
48     {
49         *ppEnum = NULL;
50         return E_OUTOFMEMORY;
51     }
52     pEnumPins->lpVtbl = &IEnumPinsImpl_Vtbl;
53     pEnumPins->refCount = 1;
54     pEnumPins->uIndex = 0;
55     pEnumPins->receive_pin = receive_pin;
56     pEnumPins->base = base;
57     IBaseFilter_AddRef(base);
58     *ppEnum = (IEnumPins *)(&pEnumPins->lpVtbl);
59
60     receive_pin(base, ~0, NULL, &pEnumPins->synctime);
61
62     TRACE("Created new enumerator (%p)\n", *ppEnum);
63     return S_OK;
64 }
65
66 static HRESULT WINAPI IEnumPinsImpl_QueryInterface(IEnumPins * iface, REFIID riid, LPVOID * ppv)
67 {
68     TRACE("(%s, %p)\n", qzdebugstr_guid(riid), ppv);
69
70     *ppv = NULL;
71
72     if (IsEqualIID(riid, &IID_IUnknown))
73         *ppv = (LPVOID)iface;
74     else if (IsEqualIID(riid, &IID_IEnumPins))
75         *ppv = (LPVOID)iface;
76
77     if (*ppv)
78     {
79         IUnknown_AddRef((IUnknown *)(*ppv));
80         return S_OK;
81     }
82
83     FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
84
85     return E_NOINTERFACE;
86 }
87
88 static ULONG WINAPI IEnumPinsImpl_AddRef(IEnumPins * iface)
89 {
90     IEnumPinsImpl *This = (IEnumPinsImpl *)iface;
91     ULONG refCount = InterlockedIncrement(&This->refCount);
92
93     TRACE("(%p)->() AddRef from %d\n", This, refCount - 1);
94
95     return refCount;
96 }
97
98 static ULONG WINAPI IEnumPinsImpl_Release(IEnumPins * iface)
99 {
100     IEnumPinsImpl *This = (IEnumPinsImpl *)iface;
101     ULONG refCount = InterlockedDecrement(&This->refCount);
102
103     TRACE("(%p)->() Release from %d\n", This, refCount + 1);
104
105     if (!refCount)
106     {
107         IBaseFilter_Release(This->base);
108         CoTaskMemFree(This);
109         return 0;
110     }
111     else
112         return refCount;
113 }
114
115 static HRESULT WINAPI IEnumPinsImpl_Next(IEnumPins * iface, ULONG cPins, IPin ** ppPins, ULONG * pcFetched)
116 {
117     IEnumPinsImpl *This = (IEnumPinsImpl *)iface;
118     DWORD synctime = This->synctime;
119     HRESULT hr = S_OK;
120     ULONG i = 0;
121
122     TRACE("(%u, %p, %p)\n", cPins, ppPins, pcFetched);
123
124     if (!ppPins)
125         return E_POINTER;
126
127     if (cPins > 1 && !pcFetched)
128         return E_INVALIDARG;
129
130     if (pcFetched)
131         *pcFetched = 0;
132
133     while (i < cPins && hr == S_OK)
134     {
135         hr = This->receive_pin(This->base, This->uIndex + i, &ppPins[i], &synctime);
136
137         if (hr == S_OK)
138             ++i;
139
140         if (synctime != This->synctime)
141             break;
142     }
143
144     if (!i && synctime != This->synctime)
145         return VFW_E_ENUM_OUT_OF_SYNC;
146
147     if (pcFetched)
148         *pcFetched = i;
149     This->uIndex += i;
150
151     if (i < cPins)
152         return S_FALSE;
153     return S_OK;
154 }
155
156 static HRESULT WINAPI IEnumPinsImpl_Skip(IEnumPins * iface, ULONG cPins)
157 {
158     IEnumPinsImpl *This = (IEnumPinsImpl *)iface;
159     DWORD synctime = This->synctime;
160     HRESULT hr;
161     IPin *pin = NULL;
162
163     TRACE("(%u)\n", cPins);
164
165     hr = This->receive_pin(This->base, This->uIndex + cPins, &pin, &synctime);
166     if (pin)
167         IPin_Release(pin);
168
169     if (synctime != This->synctime)
170         return VFW_E_ENUM_OUT_OF_SYNC;
171
172     if (hr == S_OK)
173         This->uIndex += cPins;
174
175     return hr;
176 }
177
178 static HRESULT WINAPI IEnumPinsImpl_Reset(IEnumPins * iface)
179 {
180     IEnumPinsImpl *This = (IEnumPinsImpl *)iface;
181
182     TRACE("IEnumPinsImpl::Reset()\n");
183     This->receive_pin(This->base, ~0, NULL, &This->synctime);
184
185     This->uIndex = 0;
186     return S_OK;
187 }
188
189 static HRESULT WINAPI IEnumPinsImpl_Clone(IEnumPins * iface, IEnumPins ** ppEnum)
190 {
191     HRESULT hr;
192     IEnumPinsImpl *This = (IEnumPinsImpl *)iface;
193
194     TRACE("(%p)\n", ppEnum);
195
196     hr = IEnumPinsImpl_Construct(ppEnum, This->receive_pin, This->base);
197     if (FAILED(hr))
198         return hr;
199     return IEnumPins_Skip(*ppEnum, This->uIndex);
200 }
201
202 static const IEnumPinsVtbl IEnumPinsImpl_Vtbl =
203 {
204     IEnumPinsImpl_QueryInterface,
205     IEnumPinsImpl_AddRef,
206     IEnumPinsImpl_Release,
207     IEnumPinsImpl_Next,
208     IEnumPinsImpl_Skip,
209     IEnumPinsImpl_Reset,
210     IEnumPinsImpl_Clone
211 };