dlls/amstream/amstream.c

   1 /*
   2  * Implementation of IAMMultiMediaStream Interface
   3  *
   4  * Copyright 2004 Christian Costa
   5  * Copyright 2006 Ivan Leo Puoti
   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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include "wine/debug.h"
  23
  24 #define COBJMACROS
  25
  26 #include "winbase.h"
  27 #include "wingdi.h"
  28
  29 #include "amstream_private.h"
  30 #include "amstream.h"
  31
  32 WINE_DEFAULT_DEBUG_CHANNEL(amstream);
  33
  34 typedef struct {
  35     IAMMultiMediaStream IAMMultiMediaStream_iface;
  36     LONG ref;
  37     IGraphBuilder* pFilterGraph;
  38     IPin* ipin;
  39     IGraphBuilder* GraphBuilder;
  40     ULONG nbStreams;
  41     IMediaStream** pStreams;
  42     STREAM_TYPE StreamType;
  43 } IAMMultiMediaStreamImpl;
  44
  45 static inline IAMMultiMediaStreamImpl *impl_from_IAMMultiMediaStream(IAMMultiMediaStream *iface)
  46 {
  47     return CONTAINING_RECORD(iface, IAMMultiMediaStreamImpl, IAMMultiMediaStream_iface);
  48 }
  49
  50 static const struct IAMMultiMediaStreamVtbl AM_Vtbl;
  51
  52 HRESULT AM_create(IUnknown *pUnkOuter, LPVOID *ppObj)
  53 {
  54     IAMMultiMediaStreamImpl* object;
  55
  56     TRACE("(%p,%p)\n", pUnkOuter, ppObj);
  57
  58     if( pUnkOuter )
  59         return CLASS_E_NOAGGREGATION;
  60
  61     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(IAMMultiMediaStreamImpl));
  62     if (!object)
  63     {
  64         ERR("Out of memory\n");
  65         return E_OUTOFMEMORY;
  66     }
  67
  68     object->IAMMultiMediaStream_iface.lpVtbl = &AM_Vtbl;
  69     object->ref = 1;
  70
  71     *ppObj = object;
  72
  73     return S_OK;
  74 }
  75
  76 /*** IUnknown methods ***/
  77 static HRESULT WINAPI IAMMultiMediaStreamImpl_QueryInterface(IAMMultiMediaStream* iface, REFIID riid, void** ppvObject)
  78 {
  79     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
  80
  81     TRACE("(%p/%p)->(%s,%p)\n", iface, This, debugstr_guid(riid), ppvObject);
  82
  83     if (IsEqualGUID(riid, &IID_IUnknown) ||
  84         IsEqualGUID(riid, &IID_IMultiMediaStream) ||
  85         IsEqualGUID(riid, &IID_IAMMultiMediaStream))
  86     {
  87         IUnknown_AddRef(iface);
  88         *ppvObject = This;
  89         return S_OK;
  90     }
  91
  92     ERR("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppvObject);
  93
  94     return E_NOINTERFACE;
  95 }
  96
  97 static ULONG WINAPI IAMMultiMediaStreamImpl_AddRef(IAMMultiMediaStream* iface)
  98 {
  99     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 100
 101     TRACE("(%p/%p)\n", iface, This);
 102
 103     return InterlockedIncrement(&This->ref);
 104 }
 105
 106 static ULONG WINAPI IAMMultiMediaStreamImpl_Release(IAMMultiMediaStream* iface)
 107 {
 108     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 109     ULONG ref = InterlockedDecrement(&This->ref);
 110
 111     TRACE("(%p/%p)\n", iface, This);
 112
 113     if (!ref)
 114         HeapFree(GetProcessHeap(), 0, This);
 115
 116     return ref;
 117 }
 118
 119 /*** IMultiMediaStream methods ***/
 120 static HRESULT WINAPI IAMMultiMediaStreamImpl_GetInformation(IAMMultiMediaStream* iface, DWORD* pdwFlags, STREAM_TYPE* pStreamType)
 121 {
 122     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 123
 124     FIXME("(%p/%p)->(%p,%p) stub!\n", This, iface, pdwFlags, pStreamType);
 125
 126     return E_NOTIMPL;
 127 }
 128
 129 static HRESULT WINAPI IAMMultiMediaStreamImpl_GetMediaStream(IAMMultiMediaStream* iface, REFMSPID idPurpose, IMediaStream** ppMediaStream)
 130 {
 131     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 132     MSPID PurposeId;
 133     unsigned int i;
 134
 135     TRACE("(%p/%p)->(%s,%p)\n", This, iface, debugstr_guid(idPurpose), ppMediaStream);
 136
 137     for (i = 0; i < This->nbStreams; i++)
 138     {
 139         IMediaStream_GetInformation(This->pStreams[i], &PurposeId, NULL);
 140         if (IsEqualIID(&PurposeId, idPurpose))
 141         {
 142             *ppMediaStream = This->pStreams[i];
 143             IMediaStream_AddRef(*ppMediaStream);
 144             return S_OK;
 145         }
 146     }
 147
 148     return MS_E_NOSTREAM;
 149 }
 150
 151 static HRESULT WINAPI IAMMultiMediaStreamImpl_EnumMediaStreams(IAMMultiMediaStream* iface, LONG Index, IMediaStream** ppMediaStream)
 152 {
 153     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 154
 155     FIXME("(%p/%p)->(%d,%p) stub!\n", This, iface, Index, ppMediaStream);
 156
 157     return E_NOTIMPL;
 158 }
 159
 160 static HRESULT WINAPI IAMMultiMediaStreamImpl_GetState(IAMMultiMediaStream* iface, STREAM_STATE* pCurrentState)
 161 {
 162     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 163
 164     FIXME("(%p/%p)->(%p) stub!\n", This, iface, pCurrentState);
 165
 166     return E_NOTIMPL;
 167 }
 168
 169 static HRESULT WINAPI IAMMultiMediaStreamImpl_SetState(IAMMultiMediaStream* iface, STREAM_STATE NewState)
 170 {
 171     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 172
 173     FIXME("(%p/%p)->() stub!\n", This, iface);
 174
 175     return E_NOTIMPL;
 176 }
 177
 178 static HRESULT WINAPI IAMMultiMediaStreamImpl_GetTime(IAMMultiMediaStream* iface, STREAM_TIME* pCurrentTime)
 179 {
 180     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 181
 182     FIXME("(%p/%p)->(%p) stub!\n", This, iface, pCurrentTime);
 183
 184     return E_NOTIMPL;
 185 }
 186
 187 static HRESULT WINAPI IAMMultiMediaStreamImpl_GetDuration(IAMMultiMediaStream* iface, STREAM_TIME* pDuration)
 188 {
 189     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 190
 191     FIXME("(%p/%p)->(%p) stub!\n", This, iface, pDuration);
 192
 193     return E_NOTIMPL;
 194 }
 195
 196 static HRESULT WINAPI IAMMultiMediaStreamImpl_Seek(IAMMultiMediaStream* iface, STREAM_TIME SeekTime)
 197 {
 198     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 199
 200     FIXME("(%p/%p)->() stub!\n", This, iface);
 201
 202     return E_NOTIMPL;
 203 }
 204
 205 static HRESULT WINAPI IAMMultiMediaStreamImpl_GetEndOfStream(IAMMultiMediaStream* iface, HANDLE* phEOS)
 206 {
 207     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 208
 209     FIXME("(%p/%p)->(%p) stub!\n", This, iface, phEOS);
 210
 211     return E_NOTIMPL;
 212 }
 213
 214 /*** IAMMultiMediaStream methods ***/
 215 static HRESULT WINAPI IAMMultiMediaStreamImpl_Initialize(IAMMultiMediaStream* iface, STREAM_TYPE StreamType, DWORD dwFlags, IGraphBuilder* pFilterGraph)
 216 {
 217     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 218     HRESULT hr = S_OK;
 219
 220     TRACE("(%p/%p)->(%x,%x,%p)\n", This, iface, (DWORD)StreamType, dwFlags, pFilterGraph);
 221
 222     if (pFilterGraph)
 223     {
 224         This->pFilterGraph = pFilterGraph;
 225         IGraphBuilder_AddRef(This->pFilterGraph);
 226     }
 227     else
 228     {
 229         hr = CoCreateInstance(&CLSID_FilterGraph, NULL, CLSCTX_INPROC_SERVER, &IID_IGraphBuilder, (LPVOID*)&This->pFilterGraph);
 230     }
 231
 232     if (SUCCEEDED(hr))
 233     {
 234         This->StreamType = StreamType;
 235     }
 236
 237     return hr;
 238 }
 239
 240 static HRESULT WINAPI IAMMultiMediaStreamImpl_GetFilterGraph(IAMMultiMediaStream* iface, IGraphBuilder** ppGraphBuilder)
 241 {
 242     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 243
 244     TRACE("(%p/%p)->(%p)\n", This, iface, ppGraphBuilder);
 245
 246     if (!ppGraphBuilder)
 247         return E_POINTER;
 248
 249     if (This->pFilterGraph)
 250         return IFilterGraph_QueryInterface(This->pFilterGraph, &IID_IGraphBuilder, (void**)ppGraphBuilder);
 251     else
 252         *ppGraphBuilder = NULL;
 253
 254     return S_OK;
 255 }
 256
 257 static HRESULT WINAPI IAMMultiMediaStreamImpl_GetFilter(IAMMultiMediaStream* iface, IMediaStreamFilter** ppFilter)
 258 {
 259     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 260
 261     FIXME("(%p/%p)->(%p) stub!\n", This, iface, ppFilter);
 262
 263     return E_NOTIMPL;
 264 }
 265
 266 static HRESULT WINAPI IAMMultiMediaStreamImpl_AddMediaStream(IAMMultiMediaStream* iface, IUnknown* pStreamObject, const MSPID* PurposeId,
 267                                           DWORD dwFlags, IMediaStream** ppNewStream)
 268 {
 269     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 270     HRESULT hr;
 271     IMediaStream* pStream;
 272     IMediaStream** pNewStreams;
 273
 274     FIXME("(%p/%p)->(%p,%s,%x,%p) partial stub!\n", This, iface, pStreamObject, debugstr_guid(PurposeId), dwFlags, ppNewStream);
 275
 276     hr = mediastream_create((IMultiMediaStream*)iface, PurposeId, This->StreamType, &pStream);
 277     if (SUCCEEDED(hr))
 278     {
 279         pNewStreams = CoTaskMemRealloc(This->pStreams, (This->nbStreams+1) * sizeof(IMediaStream*));
 280         if (!pNewStreams)
 281         {
 282             IMediaStream_Release(pStream);
 283             return E_OUTOFMEMORY;
 284         }
 285         This->pStreams = pNewStreams;
 286         This->pStreams[This->nbStreams] = pStream;
 287         This->nbStreams++;
 288
 289         if (ppNewStream)
 290             *ppNewStream = pStream;
 291     }
 292
 293     return hr;
 294 }
 295
 296 static HRESULT WINAPI IAMMultiMediaStreamImpl_OpenFile(IAMMultiMediaStream* iface, LPCWSTR pszFileName, DWORD dwFlags)
 297 {
 298     HRESULT ret;
 299     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 300     IFileSourceFilter *SourceFilter;
 301     IBaseFilter *BaseFilter;
 302     IEnumPins *EnumPins;
 303     IPin *ipin;
 304     PIN_DIRECTION pin_direction;
 305
 306     TRACE("(%p/%p)->(%s,%x)\n", This, iface, debugstr_w(pszFileName), dwFlags);
 307
 308     ret = CoCreateInstance(&CLSID_AsyncReader, NULL, CLSCTX_INPROC_SERVER, &IID_IFileSourceFilter, (void**)&SourceFilter);
 309     if(ret != S_OK)
 310         return ret;
 311
 312     ret = IFileSourceFilter_Load(SourceFilter, pszFileName, NULL);
 313     if(ret != S_OK)
 314     {
 315         IFileSourceFilter_Release(SourceFilter);
 316         return ret;
 317     }
 318
 319     ret = IFileSourceFilter_QueryInterface(SourceFilter, &IID_IBaseFilter, (void**)&BaseFilter);
 320     if(ret != S_OK)
 321     {
 322         IFileSourceFilter_Release(SourceFilter);
 323         return ret;
 324     }
 325
 326     ret = IBaseFilter_EnumPins(BaseFilter, &EnumPins);
 327     if(ret != S_OK)
 328     {
 329         goto end;
 330     }
 331
 332     ret = IEnumPins_Next(EnumPins, 1, &ipin, NULL);
 333     if(ret == S_OK)
 334     {
 335         ret = IPin_QueryDirection(ipin, &pin_direction);
 336         IEnumPins_Release(EnumPins);
 337         if(ret == S_OK && pin_direction == PINDIR_OUTPUT)
 338             This->ipin = ipin;
 339         else
 340             goto end;
 341     }
 342     else
 343     {
 344         IEnumPins_Release(EnumPins);
 345         goto end;
 346     }
 347
 348     /* If Initialize was not called before, we do it here */
 349     if (!This->pFilterGraph)
 350     {
 351         ret = IAMMultiMediaStream_Initialize(iface, STREAMTYPE_READ, 0, NULL);
 352         if (FAILED(ret))
 353             goto end;
 354     }
 355
 356     ret = IFilterGraph_QueryInterface(This->pFilterGraph, &IID_IGraphBuilder, (void**)&This->GraphBuilder);
 357     if(ret != S_OK)
 358     {
 359         goto end;
 360     }
 361
 362     ret = IGraphBuilder_AddSourceFilter(This->GraphBuilder, pszFileName, pszFileName, &BaseFilter);
 363
 364 end:
 365     IBaseFilter_Release(BaseFilter);
 366     IFileSourceFilter_Release(SourceFilter);
 367     return ret;
 368 }
 369
 370 static HRESULT WINAPI IAMMultiMediaStreamImpl_OpenMoniker(IAMMultiMediaStream* iface, IBindCtx* pCtx, IMoniker* pMoniker, DWORD dwFlags)
 371 {
 372     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 373
 374     FIXME("(%p/%p)->(%p,%p,%x) stub!\n", This, iface, pCtx, pMoniker, dwFlags);
 375
 376     return E_NOTIMPL;
 377 }
 378
 379 static HRESULT WINAPI IAMMultiMediaStreamImpl_Render(IAMMultiMediaStream* iface, DWORD dwFlags)
 380 {
 381     IAMMultiMediaStreamImpl *This = impl_from_IAMMultiMediaStream(iface);
 382
 383     FIXME("(%p/%p)->(%x) partial stub!\n", This, iface, dwFlags);
 384
 385     if(dwFlags != AMMSF_NOCLOCK)
 386         return E_INVALIDARG;
 387
 388     return IGraphBuilder_Render(This->GraphBuilder, This->ipin);
 389 }
 390
 391 static const IAMMultiMediaStreamVtbl AM_Vtbl =
 392 {
 393     IAMMultiMediaStreamImpl_QueryInterface,
 394     IAMMultiMediaStreamImpl_AddRef,
 395     IAMMultiMediaStreamImpl_Release,
 396     IAMMultiMediaStreamImpl_GetInformation,
 397     IAMMultiMediaStreamImpl_GetMediaStream,
 398     IAMMultiMediaStreamImpl_EnumMediaStreams,
 399     IAMMultiMediaStreamImpl_GetState,
 400     IAMMultiMediaStreamImpl_SetState,
 401     IAMMultiMediaStreamImpl_GetTime,
 402     IAMMultiMediaStreamImpl_GetDuration,
 403     IAMMultiMediaStreamImpl_Seek,
 404     IAMMultiMediaStreamImpl_GetEndOfStream,
 405     IAMMultiMediaStreamImpl_Initialize,
 406     IAMMultiMediaStreamImpl_GetFilterGraph,
 407     IAMMultiMediaStreamImpl_GetFilter,
 408     IAMMultiMediaStreamImpl_AddMediaStream,
 409     IAMMultiMediaStreamImpl_OpenFile,
 410     IAMMultiMediaStreamImpl_OpenMoniker,
 411     IAMMultiMediaStreamImpl_Render
 412 };