windows-nt/Source/XPSP1/NT/multimedia/dshow/streams/raltest/main.cpp
2020-09-26 16:20:57 +08:00

126 lines
3.3 KiB
C++

// Copyright (c) 1997 Microsoft Corporation. All Rights Reserved.
#define INIT_GUID
#define INTTGUIDS
#define INITGUID
#define INIT_GUIDS
#include <windows.h>
#include <stdio.h>
#include "ddraw.h"
#include "mmstream.h"
#include "amstream.h"
#include "ddstream.h"
#define RELEASE(x) if (x) (x)->Release();
#define CHECK_ERROR(x) if (FAILED(hr = (x))) goto Exit;
HRESULT RenderFileToMMStream(const char * pszFileName, IMultiMediaStream **ppMMStream, IDirectDraw *pDD)
{
*ppMMStream = NULL;
IAMMultiMediaStream *pAMStream;
HRESULT hr;
CHECK_ERROR(CoCreateInstance(CLSID_AMMultiMediaStream, NULL, CLSCTX_INPROC_SERVER,
IID_IAMMultiMediaStream, (void **)&pAMStream));
CHECK_ERROR(pAMStream->Initialize(STREAMTYPE_READ, NULL));
CHECK_ERROR(pAMStream->AddMediaStream(pDD, MSPID_PrimaryVideo, 0, NULL));
CHECK_ERROR(pAMStream->AddMediaStream(NULL, MSPID_PrimaryAudio, AMMSF_ADDDEFAULTRENDERER, NULL));
WCHAR wPath[MAX_PATH];
MultiByteToWideChar(CP_ACP, 0, pszFileName, -1, wPath, sizeof(wPath)/sizeof(wPath[0]));
CHECK_ERROR(pAMStream->OpenFile(wPath, 0));
// CHECK_ERROR(pAMStream->SetState(STREAMSTATE_RUN));
*ppMMStream = pAMStream;
pAMStream->AddRef();
Exit:
RELEASE(pAMStream);
return hr;
}
HRESULT RenderStreamToSurface(IDirectDraw *pDD, IDirectDrawSurface *pPrimary, IMultiMediaStream *pMMStream)
{
HRESULT hr;
IMediaStream *pPrimaryVidStream = NULL;
IDirectDrawMediaStream *pDDStream = NULL;
IDirectDrawSurface *pSurface = NULL;
IDirectDrawStreamSample *pSample = NULL;
CHECK_ERROR(pMMStream->GetMediaStream(MSPID_PrimaryVideo, &pPrimaryVidStream));
CHECK_ERROR(pPrimaryVidStream->QueryInterface(IID_IDirectDrawMediaStream, (void **)&pDDStream));
DDSURFACEDESC ddsd;
ddsd.dwSize = sizeof(ddsd);
CHECK_ERROR(pDDStream->GetFormat(&ddsd, NULL, NULL));
RECT rect;
rect.top = rect.left = 0;
rect.bottom = ddsd.dwHeight;
rect.right = ddsd.dwWidth;
CHECK_ERROR(pDD->CreateSurface(&ddsd, &pSurface, NULL));
CHECK_ERROR(pDDStream->CreateSample(pSurface, NULL, 0, &pSample));
while (true) {
if (pSample->Update(0, NULL, NULL, 0) != S_OK) {
break;
}
pPrimary->Blt(&rect, pSurface, &rect, DDBLT_WAIT, NULL);
}
Exit:
RELEASE(pPrimaryVidStream);
RELEASE(pDDStream);
RELEASE(pSample);
RELEASE(pSurface);
return hr;
}
int _CRTAPI1
main(
int argc,
char *argv[]
)
{
if (argc < 2) {
printf("Usage : mmtest foo.bar\n");
exit(0);
}
CoInitialize(NULL);
IDirectDraw *pDD;
HRESULT hr = DirectDrawCreate(NULL, &pDD, NULL);
if (SUCCEEDED(hr)) {
DDSURFACEDESC ddsd;
IDirectDrawSurface *pPrimarySurface;
pDD->SetCooperativeLevel(GetDesktopWindow(), DDSCL_NORMAL);
ddsd.dwSize = sizeof(ddsd);
ddsd.dwFlags = DDSD_CAPS;
ddsd.ddsCaps.dwCaps = DDSCAPS_PRIMARYSURFACE;
hr = pDD->CreateSurface(&ddsd, &pPrimarySurface, NULL);
if (SUCCEEDED(hr)) {
IMultiMediaStream *pMMStream;
hr = RenderFileToMMStream(argv[1], &pMMStream, pDD);
if (SUCCEEDED(hr)) {
RenderStreamToSurface(pDD, pPrimarySurface, pMMStream);
pMMStream->Release();
}
pPrimarySurface->Release();
}
pDD->Release();
}
CoUninitialize();
return 0;
}