[QUARTZ][STRMBASE]
[reactos.git] / reactos / dll / directx / wine / quartz / enumfilters.c
1 /*
2 * Implementation of IEnumFilters Interface
3 *
4 * Copyright 2004 Christian Costa
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 typedef struct IEnumFiltersImpl
24 {
25 IEnumFilters IEnumFilters_iface;
26 LONG refCount;
27 IGraphVersion * pVersionSource;
28 LONG Version;
29 IBaseFilter *** pppFilters;
30 ULONG * pNumFilters;
31 ULONG uIndex;
32 } IEnumFiltersImpl;
33
34 static const struct IEnumFiltersVtbl IEnumFiltersImpl_Vtbl;
35
36 static inline IEnumFiltersImpl *impl_from_IEnumFilters(IEnumFilters *iface)
37 {
38 return CONTAINING_RECORD(iface, IEnumFiltersImpl, IEnumFilters_iface);
39 }
40
41 HRESULT IEnumFiltersImpl_Construct(IGraphVersion * pVersionSource, IBaseFilter *** pppFilters, ULONG * pNumFilters, IEnumFilters ** ppEnum)
42 {
43 /* Note: The incoming IBaseFilter interfaces are not AddRef'd here as in Windows,
44 * they should have been previously AddRef'd. */
45 IEnumFiltersImpl * pEnumFilters = CoTaskMemAlloc(sizeof(IEnumFiltersImpl));
46 HRESULT hr;
47 LONG currentVersion;
48
49 TRACE("(%p, %p, %p)\n", pppFilters, pNumFilters, ppEnum);
50
51 *ppEnum = NULL;
52
53 if (!pEnumFilters)
54 {
55 return E_OUTOFMEMORY;
56 }
57
58 pEnumFilters->IEnumFilters_iface.lpVtbl = &IEnumFiltersImpl_Vtbl;
59 pEnumFilters->refCount = 1;
60 pEnumFilters->uIndex = 0;
61 pEnumFilters->pNumFilters = pNumFilters;
62 pEnumFilters->pppFilters = pppFilters;
63 IGraphVersion_AddRef(pVersionSource);
64 pEnumFilters->pVersionSource = pVersionSource;
65
66 /* Store the current version of the graph */
67 hr = IGraphVersion_QueryVersion(pVersionSource, &currentVersion);
68 pEnumFilters->Version = (hr==S_OK) ? currentVersion : 0;
69
70 *ppEnum = &pEnumFilters->IEnumFilters_iface;
71 return S_OK;
72 }
73
74 static HRESULT WINAPI IEnumFiltersImpl_QueryInterface(IEnumFilters * iface, REFIID riid, LPVOID * ppv)
75 {
76 TRACE("(%p)->(%s, %p)\n", iface, qzdebugstr_guid(riid), ppv);
77
78 *ppv = NULL;
79
80 if (IsEqualIID(riid, &IID_IUnknown))
81 *ppv = iface;
82 else if (IsEqualIID(riid, &IID_IEnumFilters))
83 *ppv = iface;
84
85 if (*ppv)
86 {
87 IUnknown_AddRef((IUnknown *)(*ppv));
88 return S_OK;
89 }
90
91 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
92
93 return E_NOINTERFACE;
94 }
95
96 static ULONG WINAPI IEnumFiltersImpl_AddRef(IEnumFilters * iface)
97 {
98 IEnumFiltersImpl *This = impl_from_IEnumFilters(iface);
99 ULONG refCount = InterlockedIncrement(&This->refCount);
100
101 TRACE("(%p)->()\n", iface);
102
103 return refCount;
104 }
105
106 static ULONG WINAPI IEnumFiltersImpl_Release(IEnumFilters * iface)
107 {
108 IEnumFiltersImpl *This = impl_from_IEnumFilters(iface);
109 ULONG refCount = InterlockedDecrement(&This->refCount);
110
111 TRACE("(%p)->()\n", iface);
112
113 if (!refCount)
114 {
115 IGraphVersion_Release(This->pVersionSource);
116 CoTaskMemFree(This);
117 return 0;
118 }
119 else
120 return refCount;
121 }
122
123 static HRESULT WINAPI IEnumFiltersImpl_Next(IEnumFilters * iface, ULONG cFilters, IBaseFilter ** ppFilters, ULONG * pcFetched)
124 {
125 ULONG cFetched;
126 ULONG i;
127 LONG currentVersion;
128 IEnumFiltersImpl *This = impl_from_IEnumFilters(iface);
129 HRESULT hr;
130
131 cFetched = min(*This->pNumFilters, This->uIndex + cFilters) - This->uIndex;
132
133 TRACE("(%p)->(%u, %p, %p)\n", iface, cFilters, ppFilters, pcFetched);
134
135 /* First of all check if the graph has changed */
136 hr = IGraphVersion_QueryVersion(This->pVersionSource, &currentVersion);
137 if (hr==S_OK && This->Version != currentVersion)
138 return VFW_E_ENUM_OUT_OF_SYNC;
139
140
141 if (!ppFilters)
142 return E_POINTER;
143
144 for (i = 0; i < cFetched; i++)
145 {
146 ppFilters[i] = (*This->pppFilters)[This->uIndex + i];
147 IBaseFilter_AddRef(ppFilters[i]);
148 }
149
150 This->uIndex += cFetched;
151
152 if (pcFetched)
153 *pcFetched = cFetched;
154
155 if (cFetched != cFilters)
156 return S_FALSE;
157 return S_OK;
158 }
159
160 static HRESULT WINAPI IEnumFiltersImpl_Skip(IEnumFilters * iface, ULONG cFilters)
161 {
162 IEnumFiltersImpl *This = impl_from_IEnumFilters(iface);
163
164 TRACE("(%p)->(%u)\n", iface, cFilters);
165
166 if (This->uIndex + cFilters < *This->pNumFilters)
167 {
168 This->uIndex += cFilters;
169 return S_OK;
170 }
171 return S_FALSE;
172 }
173
174 static HRESULT WINAPI IEnumFiltersImpl_Reset(IEnumFilters * iface)
175 {
176 IEnumFiltersImpl *This = impl_from_IEnumFilters(iface);
177 HRESULT hr;
178 LONG currentVersion;
179
180 TRACE("(%p)->()\n", iface);
181
182 This->uIndex = 0;
183 hr = IGraphVersion_QueryVersion(This->pVersionSource, &currentVersion);
184 if (!hr)
185 This->Version = currentVersion;
186 return S_OK;
187 }
188
189 static HRESULT WINAPI IEnumFiltersImpl_Clone(IEnumFilters * iface, IEnumFilters ** ppEnum)
190 {
191 HRESULT hr;
192 IEnumFiltersImpl *This = impl_from_IEnumFilters(iface);
193
194 TRACE("(%p)->(%p)\n", iface, ppEnum);
195
196 hr = IEnumFiltersImpl_Construct(This->pVersionSource, This->pppFilters, This->pNumFilters, ppEnum);
197 if (FAILED(hr))
198 return hr;
199 return IEnumFilters_Skip(*ppEnum, This->uIndex);
200 }
201
202 static const IEnumFiltersVtbl IEnumFiltersImpl_Vtbl =
203 {
204 IEnumFiltersImpl_QueryInterface,
205 IEnumFiltersImpl_AddRef,
206 IEnumFiltersImpl_Release,
207 IEnumFiltersImpl_Next,
208 IEnumFiltersImpl_Skip,
209 IEnumFiltersImpl_Reset,
210 IEnumFiltersImpl_Clone
211 };