dlls/amstream/mediastreamfilter.c

   1 /*
   2  * Implementation of MediaStream Filter
   3  *
   4  * Copyright 2008, 2012 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 "wine/debug.h"
  22
  23 #define COBJMACROS
  24
  25 #include "winbase.h"
  26 #include "wingdi.h"
  27 #include "dshow.h"
  28
  29 #include "wine/strmbase.h"
  30
  31 #include "amstream_private.h"
  32
  33 #include "ddstream.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(amstream);
  36
  37 typedef struct {
  38     BaseFilter filter;
  39     ULONG nb_streams;
  40     IAMMediaStream** streams;
  41 } IMediaStreamFilterImpl;
  42
  43 static inline IMediaStreamFilterImpl *impl_from_IMediaStreamFilter(IMediaStreamFilter *iface)
  44 {
  45     return CONTAINING_RECORD(iface, IMediaStreamFilterImpl, filter);
  46 }
  47
  48 /*** IUnknown methods ***/
  49
  50 static HRESULT WINAPI MediaStreamFilterImpl_QueryInterface(IMediaStreamFilter *iface, REFIID riid, void **ret_iface)
  51 {
  52     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), ret_iface);
  53
  54     *ret_iface = NULL;
  55
  56     if (IsEqualIID(riid, &IID_IUnknown) ||
  57         IsEqualIID(riid, &IID_IPersist) ||
  58         IsEqualIID(riid, &IID_IMediaFilter) ||
  59         IsEqualIID(riid, &IID_IBaseFilter) ||
  60         IsEqualIID(riid, &IID_IMediaStreamFilter))
  61         *ret_iface = iface;
  62
  63     if (*ret_iface)
  64     {
  65         IMediaStreamFilter_AddRef(*ret_iface);
  66         return S_OK;
  67     }
  68
  69     return E_NOINTERFACE;
  70 }
  71
  72 static ULONG WINAPI MediaStreamFilterImpl_AddRef(IMediaStreamFilter *iface)
  73 {
  74     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
  75     ULONG ref = BaseFilterImpl_AddRef(&This->filter.IBaseFilter_iface);
  76
  77     TRACE("(%p)->(): new ref = %u\n", iface, ref);
  78
  79     return ref;
  80 }
  81
  82 static ULONG WINAPI MediaStreamFilterImpl_Release(IMediaStreamFilter *iface)
  83 {
  84     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
  85     ULONG ref = InterlockedDecrement(&This->filter.refCount);
  86
  87     TRACE("(%p)->(): new ref = %u\n", iface, ref);
  88
  89     if (!ref)
  90     {
  91         ULONG i;
  92         for (i = 0; i < This->nb_streams; i++)
  93         {
  94             IAMMediaStream_JoinFilter(This->streams[i], NULL);
  95             IAMMediaStream_Release(This->streams[i]);
  96         }
  97         CoTaskMemFree(This->streams);
  98         BaseFilter_Destroy(&This->filter);
  99         HeapFree(GetProcessHeap(), 0, This);
 100     }
 101
 102     return ref;
 103 }
 104
 105 /*** IPersist methods ***/
 106
 107 static HRESULT WINAPI MediaStreamFilterImpl_GetClassID(IMediaStreamFilter *iface, CLSID *clsid)
 108 {
 109     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 110     return BaseFilterImpl_GetClassID(&This->filter.IBaseFilter_iface, clsid);
 111 }
 112
 113 /*** IBaseFilter methods ***/
 114
 115 static HRESULT WINAPI MediaStreamFilterImpl_Stop(IMediaStreamFilter *iface)
 116 {
 117     FIXME("(%p)->(): Stub!\n", iface);
 118
 119     return E_NOTIMPL;
 120 }
 121
 122 static HRESULT WINAPI MediaStreamFilterImpl_Pause(IMediaStreamFilter *iface)
 123 {
 124     FIXME("(%p)->(): Stub!\n", iface);
 125
 126     return E_NOTIMPL;
 127 }
 128
 129 static HRESULT WINAPI MediaStreamFilterImpl_Run(IMediaStreamFilter *iface, REFERENCE_TIME start)
 130 {
 131     FIXME("(%p)->(%s): Stub!\n", iface, wine_dbgstr_longlong(start));
 132
 133     return E_NOTIMPL;
 134 }
 135
 136 static HRESULT WINAPI MediaStreamFilterImpl_GetState(IMediaStreamFilter *iface, DWORD ms_timeout, FILTER_STATE *state)
 137 {
 138     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 139     return BaseFilterImpl_GetState(&This->filter.IBaseFilter_iface, ms_timeout, state);
 140 }
 141
 142 static HRESULT WINAPI MediaStreamFilterImpl_SetSyncSource(IMediaStreamFilter *iface, IReferenceClock *clock)
 143 {
 144     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 145     return BaseFilterImpl_SetSyncSource(&This->filter.IBaseFilter_iface, clock);
 146 }
 147
 148 static HRESULT WINAPI MediaStreamFilterImpl_GetSyncSource(IMediaStreamFilter *iface, IReferenceClock **clock)
 149 {
 150     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 151     return BaseFilterImpl_GetSyncSource(&This->filter.IBaseFilter_iface, clock);
 152 }
 153
 154 static HRESULT WINAPI MediaStreamFilterImpl_EnumPins(IMediaStreamFilter *iface, IEnumPins **enum_pins)
 155 {
 156     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 157     return BaseFilterImpl_EnumPins(&This->filter.IBaseFilter_iface, enum_pins);
 158 }
 159
 160 static HRESULT WINAPI MediaStreamFilterImpl_FindPin(IMediaStreamFilter *iface, LPCWSTR id, IPin **pin)
 161 {
 162     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 163     return BaseFilterImpl_FindPin(&This->filter.IBaseFilter_iface, id, pin);
 164 }
 165
 166 static HRESULT WINAPI MediaStreamFilterImpl_QueryFilterInfo(IMediaStreamFilter *iface, FILTER_INFO *info)
 167 {
 168     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 169     return BaseFilterImpl_QueryFilterInfo(&This->filter.IBaseFilter_iface, info);
 170 }
 171
 172 static HRESULT WINAPI MediaStreamFilterImpl_JoinFilterGraph(IMediaStreamFilter *iface, IFilterGraph *graph, LPCWSTR name)
 173 {
 174     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 175     return BaseFilterImpl_JoinFilterGraph(&This->filter.IBaseFilter_iface, graph, name);
 176 }
 177
 178 static HRESULT WINAPI MediaStreamFilterImpl_QueryVendorInfo(IMediaStreamFilter *iface, LPWSTR *vendor_info)
 179 {
 180     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 181     return BaseFilterImpl_QueryVendorInfo(&This->filter.IBaseFilter_iface, vendor_info);
 182 }
 183
 184 /*** IMediaStreamFilter methods ***/
 185
 186 static HRESULT WINAPI MediaStreamFilterImpl_AddMediaStream(IMediaStreamFilter* iface, IAMMediaStream *pAMMediaStream)
 187 {
 188     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 189     IAMMediaStream** streams;
 190     HRESULT hr;
 191
 192     TRACE("(%p)->(%p)\n", iface, pAMMediaStream);
 193
 194     streams = CoTaskMemRealloc(This->streams, (This->nb_streams + 1) * sizeof(IAMMediaStream*));
 195     if (!streams)
 196         return E_OUTOFMEMORY;
 197     This->streams = streams;
 198
 199     hr = IAMMediaStream_JoinFilter(pAMMediaStream, iface);
 200     if (FAILED(hr))
 201         return hr;
 202
 203     This->streams[This->nb_streams] = pAMMediaStream;
 204     This->nb_streams++;
 205
 206     IAMMediaStream_AddRef(pAMMediaStream);
 207
 208     return S_OK;
 209 }
 210
 211 static HRESULT WINAPI MediaStreamFilterImpl_GetMediaStream(IMediaStreamFilter* iface, REFMSPID idPurpose, IMediaStream **ppMediaStream)
 212 {
 213     IMediaStreamFilterImpl *This = impl_from_IMediaStreamFilter(iface);
 214     MSPID purpose_id;
 215     unsigned int i;
 216
 217     TRACE("(%p)->(%s,%p)\n", iface, debugstr_guid(idPurpose), ppMediaStream);
 218
 219     for (i = 0; i < This->nb_streams; i++)
 220     {
 221         IAMMediaStream_GetInformation(This->streams[i], &purpose_id, NULL);
 222         if (IsEqualIID(&purpose_id, idPurpose))
 223         {
 224             *ppMediaStream = (IMediaStream *)This->streams[i];
 225             IMediaStream_AddRef(*ppMediaStream);
 226             return S_OK;
 227         }
 228     }
 229
 230     return MS_E_NOSTREAM;
 231 }
 232
 233 static HRESULT WINAPI MediaStreamFilterImpl_EnumMediaStreams(IMediaStreamFilter* iface, LONG Index, IMediaStream **ppMediaStream)
 234 {
 235     FIXME("(%p)->(%d,%p): Stub!\n", iface, Index, ppMediaStream);
 236
 237     return E_NOTIMPL;
 238 }
 239
 240 static HRESULT WINAPI MediaStreamFilterImpl_SupportSeeking(IMediaStreamFilter* iface, BOOL bRenderer)
 241 {
 242     FIXME("(%p)->(%d): Stub!\n", iface, bRenderer);
 243
 244     return E_NOTIMPL;
 245 }
 246
 247 static HRESULT WINAPI MediaStreamFilterImpl_ReferenceTimeToStreamTime(IMediaStreamFilter* iface, REFERENCE_TIME *pTime)
 248 {
 249     FIXME("(%p)->(%p): Stub!\n", iface, pTime);
 250
 251     return E_NOTIMPL;
 252 }
 253
 254 static HRESULT WINAPI MediaStreamFilterImpl_GetCurrentStreamTime(IMediaStreamFilter* iface, REFERENCE_TIME *pCurrentStreamTime)
 255 {
 256     FIXME("(%p)->(%p): Stub!\n", iface, pCurrentStreamTime);
 257
 258     return E_NOTIMPL;
 259 }
 260
 261 static HRESULT WINAPI MediaStreamFilterImpl_WaitUntil(IMediaStreamFilter* iface, REFERENCE_TIME WaitStreamTime)
 262 {
 263     FIXME("(%p)->(%s): Stub!\n", iface, wine_dbgstr_longlong(WaitStreamTime));
 264
 265     return E_NOTIMPL;
 266 }
 267
 268 static HRESULT WINAPI MediaStreamFilterImpl_Flush(IMediaStreamFilter* iface, BOOL bCancelEOS)
 269 {
 270     FIXME("(%p)->(%d): Stub!\n", iface, bCancelEOS);
 271
 272     return E_NOTIMPL;
 273 }
 274
 275 static HRESULT WINAPI MediaStreamFilterImpl_EndOfStream(IMediaStreamFilter* iface)
 276 {
 277     FIXME("(%p)->(): Stub!\n",  iface);
 278
 279     return E_NOTIMPL;
 280 }
 281
 282 static const IMediaStreamFilterVtbl MediaStreamFilter_Vtbl =
 283 {
 284     MediaStreamFilterImpl_QueryInterface,
 285     MediaStreamFilterImpl_AddRef,
 286     MediaStreamFilterImpl_Release,
 287     MediaStreamFilterImpl_GetClassID,
 288     MediaStreamFilterImpl_Stop,
 289     MediaStreamFilterImpl_Pause,
 290     MediaStreamFilterImpl_Run,
 291     MediaStreamFilterImpl_GetState,
 292     MediaStreamFilterImpl_SetSyncSource,
 293     MediaStreamFilterImpl_GetSyncSource,
 294     MediaStreamFilterImpl_EnumPins,
 295     MediaStreamFilterImpl_FindPin,
 296     MediaStreamFilterImpl_QueryFilterInfo,
 297     MediaStreamFilterImpl_JoinFilterGraph,
 298     MediaStreamFilterImpl_QueryVendorInfo,
 299     MediaStreamFilterImpl_AddMediaStream,
 300     MediaStreamFilterImpl_GetMediaStream,
 301     MediaStreamFilterImpl_EnumMediaStreams,
 302     MediaStreamFilterImpl_SupportSeeking,
 303     MediaStreamFilterImpl_ReferenceTimeToStreamTime,
 304     MediaStreamFilterImpl_GetCurrentStreamTime,
 305     MediaStreamFilterImpl_WaitUntil,
 306     MediaStreamFilterImpl_Flush,
 307     MediaStreamFilterImpl_EndOfStream
 308 };
 309
 310 static IPin* WINAPI MediaStreamFilterImpl_GetPin(BaseFilter *iface, int pos)
 311 {
 312     IMediaStreamFilterImpl* This = (IMediaStreamFilterImpl*)iface;
 313
 314     if (pos < This->nb_streams)
 315     {
 316         IPin *pin = NULL;
 317         IAMMediaStream_QueryInterface(This->streams[pos], &IID_IPin, (void **)&pin);
 318         return pin;
 319     }
 320
 321     return NULL;
 322 }
 323
 324 static LONG WINAPI MediaStreamFilterImpl_GetPinCount(BaseFilter *iface)
 325 {
 326     IMediaStreamFilterImpl* This = (IMediaStreamFilterImpl*)iface;
 327
 328     return This->nb_streams;
 329 }
 330
 331 static const BaseFilterFuncTable BaseFuncTable = {
 332     MediaStreamFilterImpl_GetPin,
 333     MediaStreamFilterImpl_GetPinCount
 334 };
 335
 336 HRESULT MediaStreamFilter_create(IUnknown *pUnkOuter, void **ppObj)
 337 {
 338     IMediaStreamFilterImpl* object;
 339
 340     TRACE("(%p,%p)\n", pUnkOuter, ppObj);
 341
 342     if( pUnkOuter )
 343         return CLASS_E_NOAGGREGATION;
 344
 345     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IMediaStreamFilterImpl));
 346     if (!object)
 347         return E_OUTOFMEMORY;
 348
 349     BaseFilter_Init(&object->filter, (IBaseFilterVtbl*)&MediaStreamFilter_Vtbl, &CLSID_MediaStreamFilter, (DWORD_PTR)(__FILE__ ": MediaStreamFilterImpl.csFilter"), &BaseFuncTable);
 350
 351     *ppObj = &object->filter.IBaseFilter_iface;
 352
 353     return S_OK;
 354 }