dlls/amstream/amstream.c

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