git.oblomov.eu Git - wine/blob - dlls/amstream/tests/amstream.c

   1 /*
   2  * Unit tests for MultiMedia Stream functions
   3  *
   4  * Copyright (C) 2009, 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 #define COBJMACROS
  22
  23 #include "wine/test.h"
  24 #include "initguid.h"
  25 #include "amstream.h"
  26 #include "vfwmsgs.h"
  27
  28 static const WCHAR filenameW[] = {'t','e','s','t','.','a','v','i',0};
  29
  30 static IAMMultiMediaStream* pams;
  31 static IDirectDraw7* pdd7;
  32 static IDirectDrawSurface7* pdds7;
  33
  34 static int create_ammultimediastream(void)
  35 {
  36     return S_OK == CoCreateInstance(
  37         &CLSID_AMMultiMediaStream, NULL, CLSCTX_INPROC_SERVER, &IID_IAMMultiMediaStream, (LPVOID*)&pams);
  38 }
  39
  40 static void release_ammultimediastream(void)
  41 {
  42     IAMMultiMediaStream_Release(pams);
  43 }
  44
  45 static int create_directdraw(void)
  46 {
  47     HRESULT hr;
  48     IDirectDraw* pdd = NULL;
  49     DDSURFACEDESC2 ddsd;
  50
  51     hr = DirectDrawCreate(NULL, &pdd, NULL);
  52     ok(hr==DD_OK, "DirectDrawCreate returned: %x\n", hr);
  53     if (hr != DD_OK)
  54        goto error;
  55
  56     hr = IDirectDraw_QueryInterface(pdd, &IID_IDirectDraw7, (LPVOID*)&pdd7);
  57     ok(hr==DD_OK, "QueryInterface returned: %x\n", hr);
  58     if (hr != DD_OK) goto error;
  59
  60     hr = IDirectDraw7_SetCooperativeLevel(pdd7, GetDesktopWindow(), DDSCL_NORMAL);
  61     ok(hr==DD_OK, "SetCooperativeLevel returned: %x\n", hr);
  62
  63     ZeroMemory(&ddsd, sizeof(ddsd));
  64     ddsd.dwSize = sizeof(ddsd);
  65     ddsd.dwFlags = DDSD_CAPS;
  66     ddsd.ddsCaps.dwCaps = DDSCAPS_PRIMARYSURFACE;
  67     hr = IDirectDraw7_CreateSurface(pdd7, &ddsd, &pdds7, NULL);
  68     ok(hr==DD_OK, "CreateSurface returned: %x\n", hr);
  69
  70     return TRUE;
  71
  72 error:
  73     if (pdds7)
  74         IDirectDrawSurface7_Release(pdds7);
  75     if (pdd7)
  76         IDirectDraw7_Release(pdd7);
  77     if (pdd)
  78         IDirectDraw_Release(pdd);
  79
  80     return FALSE;
  81 }
  82
  83 static void release_directdraw(void)
  84 {
  85     IDirectDrawSurface7_Release(pdds7);
  86     IDirectDraw7_Release(pdd7);
  87 }
  88
  89 static void test_openfile(void)
  90 {
  91     HRESULT hr;
  92     IGraphBuilder* pgraph;
  93
  94     if (!create_ammultimediastream())
  95         return;
  96
  97     hr = IAMMultiMediaStream_GetFilterGraph(pams, &pgraph);
  98     ok(hr==S_OK, "IAMMultiMediaStream_GetFilterGraph returned: %x\n", hr);
  99     ok(pgraph==NULL, "Filtergraph should not be created yet\n");
 100
 101     if (pgraph)
 102         IGraphBuilder_Release(pgraph);
 103
 104     hr = IAMMultiMediaStream_OpenFile(pams, filenameW, 0);
 105     ok(hr==S_OK, "IAMMultiMediaStream_OpenFile returned: %x\n", hr);
 106
 107     hr = IAMMultiMediaStream_GetFilterGraph(pams, &pgraph);
 108     ok(hr==S_OK, "IAMMultiMediaStream_GetFilterGraph returned: %x\n", hr);
 109     ok(pgraph!=NULL, "Filtergraph should be created\n");
 110
 111     if (pgraph)
 112         IGraphBuilder_Release(pgraph);
 113
 114     release_ammultimediastream();
 115 }
 116
 117 static void test_renderfile(void)
 118 {
 119     HRESULT hr;
 120     IMediaStream *pvidstream = NULL;
 121     IDirectDrawMediaStream *pddstream = NULL;
 122     IDirectDrawStreamSample *pddsample = NULL;
 123
 124     if (!create_ammultimediastream())
 125         return;
 126     if (!create_directdraw())
 127     {
 128         release_ammultimediastream();
 129         return;
 130     }
 131
 132     hr = IAMMultiMediaStream_Initialize(pams, STREAMTYPE_READ, 0, NULL);
 133     ok(hr==S_OK, "IAMMultiMediaStream_Initialize returned: %x\n", hr);
 134
 135     hr = IAMMultiMediaStream_AddMediaStream(pams, (IUnknown*)pdd7, &MSPID_PrimaryVideo, 0, NULL);
 136     ok(hr==S_OK, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 137
 138     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryAudio, AMMSF_ADDDEFAULTRENDERER, NULL);
 139     ok(hr==S_OK, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 140
 141     hr = IAMMultiMediaStream_OpenFile(pams, filenameW, 0);
 142     ok(hr==S_OK, "IAMMultiMediaStream_OpenFile returned: %x\n", hr);
 143
 144     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &pvidstream);
 145     ok(hr==S_OK, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 146     if (FAILED(hr)) goto error;
 147
 148     hr = IMediaStream_QueryInterface(pvidstream, &IID_IDirectDrawMediaStream, (LPVOID*)&pddstream);
 149     ok(hr==S_OK, "IMediaStream_QueryInterface returned: %x\n", hr);
 150     if (FAILED(hr)) goto error;
 151
 152     hr = IDirectDrawMediaStream_CreateSample(pddstream, NULL, NULL, 0, &pddsample);
 153     todo_wine ok(hr==S_OK, "IDirectDrawMediaStream_CreateSample returned: %x\n", hr);
 154
 155 error:
 156     if (pddsample)
 157         IDirectDrawMediaSample_Release(pddsample);
 158     if (pddstream)
 159         IDirectDrawMediaStream_Release(pddstream);
 160     if (pvidstream)
 161         IMediaStream_Release(pvidstream);
 162
 163     release_directdraw();
 164     release_ammultimediastream();
 165 }
 166
 167 static void test_media_streams(void)
 168 {
 169     HRESULT hr;
 170     IMediaStream *video_stream = NULL;
 171     IMediaStream *audio_stream = NULL;
 172     IMediaStream *dummy_stream;
 173     IDirectDrawMediaStream *ddraw_stream = NULL;
 174     IDirectDrawStreamSample *ddraw_sample = NULL;
 175     IMediaStreamFilter* media_stream_filter = NULL;
 176
 177     if (!create_ammultimediastream())
 178         return;
 179     if (!create_directdraw())
 180     {
 181         release_ammultimediastream();
 182         return;
 183     }
 184
 185     hr = IAMMultiMediaStream_Initialize(pams, STREAMTYPE_READ, 0, NULL);
 186     ok(hr == S_OK, "IAMMultiMediaStream_Initialize returned: %x\n", hr);
 187
 188     /* Retreive media stream filter */
 189     hr = IAMMultiMediaStream_GetFilter(pams, NULL);
 190     ok(hr == E_POINTER, "IAMMultiMediaStream_GetFilter returned: %x\n", hr);
 191     hr = IAMMultiMediaStream_GetFilter(pams, &media_stream_filter);
 192     ok(hr == S_OK, "IAMMultiMediaStream_GetFilter returned: %x\n", hr);
 193
 194     /* Verify there is no video media stream */
 195     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &video_stream);
 196     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 197
 198     /* Verify there is no default renderer for video stream */
 199     hr = IAMMultiMediaStream_AddMediaStream(pams, (IUnknown*)pdd7, &MSPID_PrimaryVideo, AMMSF_ADDDEFAULTRENDERER, NULL);
 200     ok(hr == MS_E_PURPOSEID, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 201     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &video_stream);
 202     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 203     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryVideo, AMMSF_ADDDEFAULTRENDERER, NULL);
 204     ok(hr == MS_E_PURPOSEID, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 205     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &video_stream);
 206     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 207
 208     /* Verify normal case for video stream */
 209     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryVideo, 0, NULL);
 210     ok(hr == S_OK, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 211     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryVideo, &video_stream);
 212     ok(hr == S_OK, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 213
 214     /* Verify the video stream has been added to the media stream filter */
 215     if (media_stream_filter)
 216     {
 217         hr = IMediaStreamFilter_GetMediaStream(media_stream_filter, &MSPID_PrimaryVideo, &dummy_stream);
 218         todo_wine ok(hr == S_OK, "IMediaStreamFilter_GetMediaStream returned: %x\n", hr);
 219         todo_wine ok(dummy_stream == video_stream, "Got wrong returned pointer %p, expected %p\n", dummy_stream, video_stream);
 220         if (SUCCEEDED(hr))
 221             IMediaStream_Release(dummy_stream);
 222     }
 223
 224     /* Verify there is no audio media stream */
 225     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryAudio, &audio_stream);
 226     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 227
 228     /* Verify no stream is created when using the default renderer for audio stream */
 229     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryAudio, AMMSF_ADDDEFAULTRENDERER, NULL);
 230     ok((hr == S_OK) || (hr == VFW_E_NO_AUDIO_HARDWARE), "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 231     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryAudio, &audio_stream);
 232     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 233
 234     /* Verify a stream is created when no default renderer is used */
 235     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &MSPID_PrimaryAudio, 0, NULL);
 236     ok(hr == S_OK, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 237     hr = IAMMultiMediaStream_GetMediaStream(pams, &MSPID_PrimaryAudio, &audio_stream);
 238     ok(hr == S_OK, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 239
 240     /* verify the audio stream has been added to the media stream filter */
 241     if (media_stream_filter)
 242     {
 243         hr = IMediaStreamFilter_GetMediaStream(media_stream_filter, &MSPID_PrimaryAudio, &dummy_stream);
 244         todo_wine ok(hr == S_OK, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 245         todo_wine ok(dummy_stream == audio_stream, "Got wrong returned pointer %p, expected %p\n", dummy_stream, audio_stream);
 246         if (SUCCEEDED(hr))
 247             IMediaStream_Release(dummy_stream);
 248     }
 249
 250     /* Verify behaviour with invalid purpose id */
 251     hr = IAMMultiMediaStream_GetMediaStream(pams, &IID_IUnknown, &dummy_stream);
 252     ok(hr == MS_E_NOSTREAM, "IAMMultiMediaStream_GetMediaStream returned: %x\n", hr);
 253     hr = IAMMultiMediaStream_AddMediaStream(pams, NULL, &IID_IUnknown, 0, NULL);
 254     ok(hr == MS_E_PURPOSEID, "IAMMultiMediaStream_AddMediaStream returned: %x\n", hr);
 255
 256     if (video_stream)
 257     {
 258         hr = IMediaStream_QueryInterface(video_stream, &IID_IDirectDrawMediaStream, (LPVOID*)&ddraw_stream);
 259         ok(hr == S_OK, "IMediaStream_QueryInterface returned: %x\n", hr);
 260
 261         if (SUCCEEDED(hr))
 262         {
 263             hr = IDirectDrawMediaStream_CreateSample(ddraw_stream, NULL, NULL, 0, &ddraw_sample);
 264             todo_wine ok(hr==S_OK, "IDirectDrawMediaStream_CreateSample returned: %x\n", hr);
 265         }
 266     }
 267
 268     if (ddraw_sample)
 269         IDirectDrawMediaSample_Release(ddraw_sample);
 270     if (ddraw_stream)
 271         IDirectDrawMediaStream_Release(ddraw_stream);
 272     if (video_stream)
 273         IMediaStream_Release(video_stream);
 274     if (audio_stream)
 275         IMediaStream_Release(audio_stream);
 276     if (media_stream_filter)
 277         IMediaStreamFilter_Release(media_stream_filter);
 278
 279     release_directdraw();
 280     release_ammultimediastream();
 281 }
 282
 283 START_TEST(amstream)
 284 {
 285     HANDLE file;
 286
 287     CoInitializeEx(NULL, COINIT_MULTITHREADED);
 288
 289     test_media_streams();
 290
 291     file = CreateFileW(filenameW, 0, 0, NULL, OPEN_EXISTING, 0, NULL);
 292     if (file != INVALID_HANDLE_VALUE)
 293     {
 294         CloseHandle(file);
 295
 296         test_openfile();
 297         test_renderfile();
 298     }
 299
 300     CoUninitialize();
 301 }