dll/directx/amstream/amstream.c

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