ref: f149843f7adbdd39affd549cc65bf1d4f695567f
dir: /codec/console/dec/src/d3d9_utils.cpp/
/*!
* \copy
* Copyright (c) 2013, Cisco Systems
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in
* the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
* COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*
*/
#include "d3d9_utils.h"
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
void Write2File (FILE* pFp, unsigned char* pData[3], int iStride[2], int iWidth, int iHeight);
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#ifdef ENABLE_DISPLAY_MODULE
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#define IDM_ABOUT 104
#define IDM_EXIT 105
#define IDI_TESTSHARESURFACE 107
#define IDI_SMALL 108
#define IDC_TESTSHARESURFACE 109
#define NV12_FORMAT MAKEFOURCC('N','V','1','2')
typedef struct {
UINT uiWidth;
UINT uiHeight;
D3DFORMAT D3Dformat;
D3DPOOL D3DPool;
} SHandleInfo;
#define SAFE_RELEASE(p) if(p) { (p)->Release(); (p) = NULL; }
#define SAFE_FREE(p) if(p) { free (p); (p) = NULL; }
HRESULT Dump2YUV (void* pDst[3], void* pSurface, int iWidth, int iHeight, int iStride[2]);
HRESULT Dump2Surface (void* pDst[3], void* pSurface, int iWidth, int iHeight, int iStride[2]);
HRESULT InitWindow (HWND* hWnd);
LRESULT CALLBACK WndProc (HWND, UINT, WPARAM, LPARAM);
typedef HRESULT (WINAPI* pFnCreateD3D9Ex) (UINT SDKVersion, IDirect3D9Ex**);
typedef LPDIRECT3D9 (WINAPI* pFnCreateD3D9) (UINT SDKVersion);
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
CD3D9Utils::CD3D9Utils() {
m_hDll = NULL;
m_hWnd = NULL;
m_pDumpYUV = NULL;
m_bInitDone = FALSE;
m_lpD3D9 = NULL;
m_lpD3D9Device = NULL;
m_lpD3D9RawSurfaceShare = NULL;
// coverity scan uninitial
ZeroMemory (&m_d3dpp, sizeof (m_d3dpp));
}
CD3D9Utils::~CD3D9Utils() {
Uninit();
}
HRESULT CD3D9Utils::Init (BOOL bWindowed) {
if (m_bInitDone)
return S_OK;
m_hDll = LoadLibrary (TEXT ("d3d9.dll"));
pFnCreateD3D9 pCreateD3D9 = NULL;
if (m_hDll)
pCreateD3D9 = (pFnCreateD3D9) GetProcAddress (m_hDll, TEXT ("Direct3DCreate9"));
else
return E_FAIL;
m_lpD3D9 = pCreateD3D9 (D3D_SDK_VERSION);
return bWindowed ? InitWindow (&m_hWnd) : S_OK;
}
HRESULT CD3D9Utils::Uninit() {
SAFE_RELEASE (m_lpD3D9RawSurfaceShare);
SAFE_RELEASE (m_lpD3D9Device);
SAFE_RELEASE (m_lpD3D9);
SAFE_FREE (m_pDumpYUV);
if (m_hDll) {
FreeLibrary (m_hDll);
m_hDll = NULL;
}
return S_OK;
}
HRESULT CD3D9Utils::Process (void* pDst[3], SBufferInfo* pInfo, FILE* pFp) {
HRESULT hResult = E_FAIL;
if (pDst == NULL || pInfo == NULL)
return hResult;
BOOL bWindowed = pFp ? FALSE : TRUE;
BOOL bNeedD3D9 = ! (!bWindowed && pInfo->eBufferProperty == BUFFER_HOST);
if (!m_bInitDone)
m_bInitDone = !bNeedD3D9;
if (!m_bInitDone) {
hResult = Init (bWindowed);
if (SUCCEEDED (hResult))
m_bInitDone = TRUE;
}
if (m_bInitDone) {
if (bWindowed) {
hResult = Render (pDst, pInfo);
Sleep (30);
} else if (pFp) {
hResult = Dump (pDst, pInfo, pFp);
Sleep (0);
}
}
return hResult;
}
HRESULT CD3D9Utils::Render (void* pDst[3], SBufferInfo* pInfo) {
HRESULT hResult = E_FAIL;
EBufferProperty eBufferProperty = pInfo->eBufferProperty;
if (eBufferProperty == BUFFER_HOST) {
hResult = InitResource (NULL, pInfo);
if (SUCCEEDED (hResult))
hResult = Dump2Surface (pDst, m_lpD3D9RawSurfaceShare, pInfo->UsrData.sSystemBuffer.iWidth,
pInfo->UsrData.sSystemBuffer.iHeight, pInfo->UsrData.sSystemBuffer.iStride);
}
if (SUCCEEDED (hResult)) {
IDirect3DSurface9* pBackBuffer = NULL;
hResult = m_lpD3D9Device->GetBackBuffer (0, 0, D3DBACKBUFFER_TYPE_MONO, &pBackBuffer);
hResult = m_lpD3D9Device->StretchRect (m_lpD3D9RawSurfaceShare, NULL, pBackBuffer, NULL, D3DTEXF_NONE);
hResult = m_lpD3D9Device->Present (0, 0, NULL, NULL);
}
return hResult;
}
HRESULT CD3D9Utils::Dump (void* pDst[3], SBufferInfo* pInfo, FILE* pFp) {
HRESULT hResult = E_FAIL;
EBufferProperty eBufferProperty = pInfo->eBufferProperty;
int iStride[2];
int iWidth;
int iHeight;
iWidth = pInfo->UsrData.sSystemBuffer.iWidth;
iHeight = pInfo->UsrData.sSystemBuffer.iHeight;
iStride[0] = pInfo->UsrData.sSystemBuffer.iStride[0];
iStride[1] = pInfo->UsrData.sSystemBuffer.iStride[1];
if (pDst[0] && pDst[1] && pDst[2])
Write2File (pFp, (unsigned char**)pDst, iStride, iWidth, iHeight);
return hResult;
}
HRESULT CD3D9Utils::InitResource (void* pSharedHandle, SBufferInfo* pInfo) {
HRESULT hResult = S_OK;
// coverity scan uninitial
int iWidth = 0;
int iHeight = 0;
D3DFORMAT D3Dformat = (D3DFORMAT)D3DFMT_UNKNOWN;
D3DPOOL D3Dpool = (D3DPOOL)D3DPOOL_DEFAULT;
if (pInfo == NULL)
return E_FAIL;
if (m_lpD3D9Device == NULL && m_lpD3D9RawSurfaceShare == NULL) {
HMONITOR hMonitorWnd = MonitorFromWindow (m_hWnd, MONITOR_DEFAULTTONULL);
UINT uiAdapter = D3DADAPTER_DEFAULT;
UINT uiCnt = m_lpD3D9->GetAdapterCount();
for (UINT i = 0; i < uiCnt; i++) {
HMONITOR hMonitor = m_lpD3D9->GetAdapterMonitor (i);
if (hMonitor == hMonitorWnd) {
uiAdapter = i;
break;
}
}
D3DDISPLAYMODE D3DDisplayMode;
hResult = m_lpD3D9->GetAdapterDisplayMode (uiAdapter, &D3DDisplayMode);
D3DDEVTYPE D3DDevType = D3DDEVTYPE_HAL;
DWORD dwBehaviorFlags = D3DCREATE_SOFTWARE_VERTEXPROCESSING | D3DCREATE_MULTITHREADED;
ZeroMemory (&m_d3dpp, sizeof (m_d3dpp));
m_d3dpp.Flags = D3DPRESENTFLAG_VIDEO;
m_d3dpp.SwapEffect = D3DSWAPEFFECT_DISCARD;
m_d3dpp.BackBufferFormat = D3DDisplayMode.Format;
m_d3dpp.Windowed = TRUE;
m_d3dpp.hDeviceWindow = m_hWnd;
m_d3dpp.PresentationInterval = D3DPRESENT_INTERVAL_IMMEDIATE;
hResult = m_lpD3D9->CreateDevice (uiAdapter, D3DDevType, NULL, dwBehaviorFlags, &m_d3dpp, &m_lpD3D9Device);
if (pInfo->eBufferProperty == BUFFER_HOST) {
iWidth = pInfo->UsrData.sSystemBuffer.iWidth;
iHeight = pInfo->UsrData.sSystemBuffer.iHeight;
D3Dformat = (D3DFORMAT)NV12_FORMAT;
D3Dpool = (D3DPOOL)D3DPOOL_DEFAULT;
}
hResult = m_lpD3D9Device->CreateOffscreenPlainSurface (iWidth, iHeight, (D3DFORMAT)D3Dformat, (D3DPOOL)D3Dpool,
&m_lpD3D9RawSurfaceShare, NULL);
}
if (m_lpD3D9Device == NULL || m_lpD3D9RawSurfaceShare == NULL)
hResult = E_FAIL;
return hResult;
}
CD3D9ExUtils::CD3D9ExUtils() {
m_hDll = NULL;
m_hWnd = NULL;
m_pDumpYUV = NULL;
m_bInitDone = FALSE;
m_lpD3D9 = NULL;
m_lpD3D9Device = NULL;
m_lpD3D9RawSurfaceShare = NULL;
// coverity scan uninitial
ZeroMemory (&m_d3dpp, sizeof (m_d3dpp));
}
CD3D9ExUtils::~CD3D9ExUtils() {
Uninit();
}
HRESULT CD3D9ExUtils::Init (BOOL bWindowed) {
if (m_bInitDone)
return S_OK;
m_hDll = LoadLibrary (TEXT ("d3d9.dll"));
pFnCreateD3D9Ex pCreateD3D9Ex = NULL;
if (m_hDll)
pCreateD3D9Ex = (pFnCreateD3D9Ex) GetProcAddress (m_hDll, TEXT ("Direct3DCreate9Ex"));
else
return E_FAIL;
pCreateD3D9Ex (D3D_SDK_VERSION, &m_lpD3D9);
return bWindowed ? InitWindow (&m_hWnd) : S_OK;
}
HRESULT CD3D9ExUtils::Uninit() {
SAFE_RELEASE (m_lpD3D9RawSurfaceShare);
SAFE_RELEASE (m_lpD3D9Device);
SAFE_RELEASE (m_lpD3D9);
SAFE_FREE (m_pDumpYUV);
if (m_hDll) {
FreeLibrary (m_hDll);
m_hDll = NULL;
}
return S_OK;
}
HRESULT CD3D9ExUtils::Process (void* pDst[3], SBufferInfo* pInfo, FILE* pFp) {
HRESULT hResult = E_FAIL;
if (pDst == NULL || pInfo == NULL)
return hResult;
BOOL bWindowed = pFp ? FALSE : TRUE;
BOOL bNeedD3D9 = ! (!bWindowed && pInfo->eBufferProperty == BUFFER_HOST);
if (!m_bInitDone)
m_bInitDone = !bNeedD3D9;
if (!m_bInitDone) {
hResult = Init (bWindowed);
if (SUCCEEDED (hResult))
m_bInitDone = TRUE;
}
if (m_bInitDone) {
if (bWindowed) {
hResult = Render (pDst, pInfo);
Sleep (30); // set a simple time controlling with default of 30fps
} else if (pFp) {
hResult = Dump (pDst, pInfo, pFp);
Sleep (0);
}
}
return hResult;
}
HRESULT CD3D9ExUtils::Render (void* pDst[3], SBufferInfo* pInfo) {
HRESULT hResult = E_FAIL;
EBufferProperty eBufferProperty = pInfo->eBufferProperty;
if (eBufferProperty == BUFFER_HOST) {
hResult = InitResource (NULL, pInfo);
if (SUCCEEDED (hResult))
hResult = Dump2Surface (pDst, m_lpD3D9RawSurfaceShare, pInfo->UsrData.sSystemBuffer.iWidth,
pInfo->UsrData.sSystemBuffer.iHeight, pInfo->UsrData.sSystemBuffer.iStride);
} else if (eBufferProperty == BUFFER_DEVICE) {
VOID* pSharedHandle = pDst[0];
hResult = InitResource (pSharedHandle, pInfo);
}
if (SUCCEEDED (hResult)) {
IDirect3DSurface9* pBackBuffer = NULL;
hResult = m_lpD3D9Device->GetBackBuffer (0, 0, D3DBACKBUFFER_TYPE_MONO, &pBackBuffer);
hResult = m_lpD3D9Device->StretchRect (m_lpD3D9RawSurfaceShare, NULL, pBackBuffer, NULL, D3DTEXF_NONE);
hResult = m_lpD3D9Device->PresentEx (0, 0, NULL, NULL, 0);
}
return hResult;
}
HRESULT CD3D9ExUtils::Dump (void* pDst[3], SBufferInfo* pInfo, FILE* pFp) {
HRESULT hResult = E_FAIL;
EBufferProperty eBufferProperty = pInfo->eBufferProperty;
int iStride[2];
int iWidth;
int iHeight;
if (eBufferProperty != BUFFER_HOST) {
iWidth = pInfo->UsrData.sVideoBuffer.iSurfaceWidth;
iHeight = pInfo->UsrData.sVideoBuffer.iSurfaceHeight;
iStride[0] = iWidth;
iStride[1] = iWidth / 2;
if (m_pDumpYUV == NULL) {
m_pDumpYUV = (unsigned char*)malloc (iWidth * iHeight * 3 / 2 * sizeof (unsigned char));
}
if (m_pDumpYUV) {
void* pSurface = pDst[1];
pDst[0] = m_pDumpYUV;
pDst[1] = m_pDumpYUV + iHeight * iStride[0] * sizeof (unsigned char);
pDst[2] = m_pDumpYUV + iHeight * iStride[0] * 5 / 4 * sizeof (unsigned char);
hResult = Dump2YUV (pDst, pSurface, iWidth, iHeight, iStride);
}
} else {
iWidth = pInfo->UsrData.sSystemBuffer.iWidth;
iHeight = pInfo->UsrData.sSystemBuffer.iHeight;
iStride[0] = pInfo->UsrData.sSystemBuffer.iStride[0];
iStride[1] = pInfo->UsrData.sSystemBuffer.iStride[1];
}
if (pDst[0] && pDst[1] && pDst[2])
Write2File (pFp, (unsigned char**)pDst, iStride, iWidth, iHeight);
return hResult;
}
HRESULT CD3D9ExUtils::InitResource (void* pSharedHandle, SBufferInfo* pInfo) {
HRESULT hResult = S_OK;
int iWidth;
int iHeight;
D3DFORMAT D3Dformat;
D3DPOOL D3Dpool;
if (pInfo == NULL)
return E_FAIL;
if (m_lpD3D9Device == NULL && m_lpD3D9RawSurfaceShare == NULL) {
HMONITOR hMonitorWnd = MonitorFromWindow (m_hWnd, MONITOR_DEFAULTTONULL);
UINT uiAdapter = D3DADAPTER_DEFAULT;
UINT uiCnt = m_lpD3D9->GetAdapterCount();
for (UINT i = 0; i < uiCnt; i++) {
HMONITOR hMonitor = m_lpD3D9->GetAdapterMonitor (i);
if (hMonitor == hMonitorWnd) {
uiAdapter = i;
break;
}
}
D3DDISPLAYMODEEX D3DDisplayMode;
D3DDisplayMode.Size = sizeof (D3DDISPLAYMODEEX);
hResult = m_lpD3D9->GetAdapterDisplayModeEx (uiAdapter, &D3DDisplayMode, NULL);
D3DDEVTYPE D3DDevType = D3DDEVTYPE_HAL;
DWORD dwBehaviorFlags = D3DCREATE_SOFTWARE_VERTEXPROCESSING | D3DCREATE_MULTITHREADED;
ZeroMemory (&m_d3dpp, sizeof (m_d3dpp));
m_d3dpp.Flags = D3DPRESENTFLAG_VIDEO;
m_d3dpp.SwapEffect = D3DSWAPEFFECT_DISCARD;
m_d3dpp.BackBufferFormat = D3DDisplayMode.Format;
m_d3dpp.Windowed = TRUE;
m_d3dpp.hDeviceWindow = m_hWnd;
m_d3dpp.PresentationInterval = D3DPRESENT_INTERVAL_IMMEDIATE;
hResult = m_lpD3D9->CreateDeviceEx (uiAdapter, D3DDevType, NULL, dwBehaviorFlags, &m_d3dpp, NULL, &m_lpD3D9Device);
if (pInfo->eBufferProperty == BUFFER_HOST) {
iWidth = pInfo->UsrData.sSystemBuffer.iWidth;
iHeight = pInfo->UsrData.sSystemBuffer.iHeight;
D3Dformat = (D3DFORMAT)NV12_FORMAT;
D3Dpool = (D3DPOOL)D3DPOOL_DEFAULT;
} else {
iWidth = pInfo->UsrData.sVideoBuffer.iSurfaceWidth;
iHeight = pInfo->UsrData.sVideoBuffer.iSurfaceHeight;
D3Dformat = (D3DFORMAT)pInfo->UsrData.sVideoBuffer.D3Dformat;
D3Dpool = (D3DPOOL)pInfo->UsrData.sVideoBuffer.D3DPool;
}
hResult = m_lpD3D9Device->CreateOffscreenPlainSurface (iWidth, iHeight, (D3DFORMAT)D3Dformat, (D3DPOOL)D3Dpool,
&m_lpD3D9RawSurfaceShare, &pSharedHandle);
}
if (m_lpD3D9Device == NULL || m_lpD3D9RawSurfaceShare == NULL)
hResult = E_FAIL;
return hResult;
}
HRESULT Dump2YUV (void* pDst[3], void* pSurface, int iWidth, int iHeight, int iStride[2]) {
HRESULT hResult = E_FAIL;
if (!pDst[0] || !pDst[1] || !pDst[2] || !pSurface)
return hResult;
IDirect3DSurface9* pSurfaceData = (IDirect3DSurface9*)pSurface;
D3DLOCKED_RECT sD3DLockedRect = {0};
hResult = pSurfaceData->LockRect (&sD3DLockedRect, NULL, 0);
unsigned char* pInY = (unsigned char*)sD3DLockedRect.pBits;
unsigned char* pOutY = (unsigned char*)pDst[0];
int iInStride = sD3DLockedRect.Pitch;
int iOutStride = iStride[0];
for (int j = 0; j < iHeight; j++)
memcpy (pOutY + j * iOutStride, pInY + j * iInStride, iWidth); //confirmed_safe_unsafe_usage
unsigned char* pOutV = (unsigned char*)pDst[1];
unsigned char* pOutU = (unsigned char*)pDst[2];
unsigned char* pInC = pInY + iInStride * iHeight;
iOutStride = iStride[1];
for (int i = 0; i < iHeight / 2; i++) {
for (int j = 0; j < iWidth; j += 2) {
pOutV[i * iOutStride + j / 2] = pInC[i * iInStride + j ];
pOutU[i * iOutStride + j / 2] = pInC[i * iInStride + j + 1];
}
}
pSurfaceData->UnlockRect();
return hResult;
}
HRESULT Dump2Surface (void* pDst[3], void* pSurface, int iWidth, int iHeight, int iStride[2]) {
HRESULT hResult = E_FAIL;
if (!pDst[0] || !pDst[1] || !pDst[2] || !pSurface)
return hResult;
IDirect3DSurface9* pSurfaceData = (IDirect3DSurface9*)pSurface;
D3DLOCKED_RECT sD3DLockedRect = {0};
hResult = pSurfaceData->LockRect (&sD3DLockedRect, NULL, 0);
unsigned char* pInY = (unsigned char*)pDst[0];
unsigned char* pOutY = (unsigned char*)sD3DLockedRect.pBits;
int iOutStride = sD3DLockedRect.Pitch;
for (int j = 0; j < iHeight; j++)
memcpy (pOutY + j * iOutStride, pInY + j * iStride[0], iWidth); //confirmed_safe_unsafe_usage
unsigned char* pInV = (unsigned char*)pDst[1];
unsigned char* pInU = (unsigned char*)pDst[2];
unsigned char* pOutC = pOutY + iOutStride * iHeight;
for (int i = 0; i < iHeight / 2; i++) {
for (int j = 0; j < iWidth; j += 2) {
pOutC[i * iOutStride + j ] = pInV[i * iStride[1] + j / 2];
pOutC[i * iOutStride + j + 1] = pInU[i * iStride[1] + j / 2];
}
}
pSurfaceData->UnlockRect();
return hResult;
}
HRESULT InitWindow (HWND* hWnd) {
const TCHAR kszWindowTitle[] = TEXT ("Wels Decoder Application");
const TCHAR kszWindowClass[] = TEXT ("Wels Decoder Class");
WNDCLASSEX sWndClassEx = {0};
sWndClassEx.cbSize = sizeof (WNDCLASSEX);
sWndClassEx.style = CS_HREDRAW | CS_VREDRAW;
sWndClassEx.lpfnWndProc = (WNDPROC)WndProc;
sWndClassEx.cbClsExtra = 0;
sWndClassEx.cbWndExtra = 0;
sWndClassEx.hInstance = GetModuleHandle (NULL);
sWndClassEx.hIcon = LoadIcon (sWndClassEx.hInstance, (LPCTSTR)IDI_TESTSHARESURFACE);
sWndClassEx.hCursor = LoadCursor (NULL, IDC_ARROW);
sWndClassEx.hbrBackground = (HBRUSH) (COLOR_WINDOW + 1);
sWndClassEx.lpszMenuName = (LPCSTR)IDC_TESTSHARESURFACE;
sWndClassEx.lpszClassName = kszWindowClass;
sWndClassEx.hIconSm = LoadIcon (sWndClassEx.hInstance, (LPCTSTR)IDI_SMALL);
if (!RegisterClassEx (&sWndClassEx))
return E_FAIL;
HWND hTmpWnd = CreateWindow (kszWindowClass, kszWindowTitle, WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, sWndClassEx.hInstance, NULL);
*hWnd = hTmpWnd;
if (!hTmpWnd)
return E_FAIL;
ShowWindow (hTmpWnd, SW_SHOWDEFAULT);
UpdateWindow (hTmpWnd);
return S_OK;
}
LRESULT CALLBACK WndProc (HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam) {
INT wmId, wmEvent;
switch (message) {
case WM_COMMAND:
wmId = LOWORD (wParam);
wmEvent = HIWORD (wParam);
switch (wmId) {
case IDM_ABOUT:
break;
case IDM_EXIT:
DestroyWindow (hWnd);
break;
default:
return DefWindowProc (hWnd, message, wParam, lParam);
}
break;
case WM_PAINT:
ValidateRect (hWnd , NULL);
break;
case WM_DESTROY:
PostQuitMessage (0);
break;
default:
return DefWindowProc (hWnd, message, wParam, lParam);
}
return 0;
}
#endif
CUtils::CUtils() {
hHandle = NULL;
iOSType = CheckOS();
#ifdef ENABLE_DISPLAY_MODULE
if (iOSType == OS_XP)
hHandle = (void*) new CD3D9Utils;
else if (iOSType == OS_VISTA_UPPER)
hHandle = (void*) new CD3D9ExUtils;
#endif
if (hHandle == NULL)
iOSType = OS_UNSUPPORTED;
}
CUtils::~CUtils() {
#ifdef ENABLE_DISPLAY_MODULE
if (hHandle) {
if (iOSType == OS_XP) {
CD3D9Utils* hTmp = (CD3D9Utils*) hHandle;
delete hTmp;
} else if (iOSType == OS_VISTA_UPPER) {
CD3D9ExUtils* hTmp = (CD3D9ExUtils*) hHandle;
delete hTmp;
}
hHandle = NULL;
}
#endif
}
int CUtils::Process (void* pDst[3], SBufferInfo* pInfo, FILE* pFp) {
int iRet = 0;
if (iOSType == OS_UNSUPPORTED) {
if (pFp && pDst[0] && pDst[1] && pDst[2] && pInfo) {
int iStride[2];
int iWidth = pInfo->UsrData.sSystemBuffer.iWidth;
int iHeight = pInfo->UsrData.sSystemBuffer.iHeight;
iStride[0] = pInfo->UsrData.sSystemBuffer.iStride[0];
iStride[1] = pInfo->UsrData.sSystemBuffer.iStride[1];
Write2File (pFp, (unsigned char**)pDst, iStride, iWidth, iHeight);
}
}
#ifdef ENABLE_DISPLAY_MODULE
else {
MSG msg;
ZeroMemory (&msg, sizeof (msg));
while (msg.message != WM_QUIT) {
if (PeekMessage (&msg, NULL, 0U, 0U, PM_REMOVE)) {
TranslateMessage (&msg);
DispatchMessage (&msg);
} else {
HRESULT hResult = S_OK;
if (iOSType == OS_XP)
hResult = ((CD3D9Utils*)hHandle)->Process (pDst, pInfo, pFp);
else if (iOSType == OS_VISTA_UPPER)
hResult = ((CD3D9ExUtils*)hHandle)->Process (pDst, pInfo, pFp);
iRet = !SUCCEEDED (hResult);
break;
}
}
}
#endif
return iRet;
}
int CUtils::CheckOS() {
int iType = OS_UNSUPPORTED;
#ifdef ENABLE_DISPLAY_MODULE
OSVERSIONINFOEX osvi;
ZeroMemory (&osvi, sizeof (OSVERSIONINFOEX));
osvi.dwOSVersionInfoSize = sizeof (OSVERSIONINFOEX);
if (!GetVersionEx ((OSVERSIONINFO*) &osvi)) {
osvi.dwOSVersionInfoSize = sizeof (OSVERSIONINFO);
if (! GetVersionEx ((OSVERSIONINFO*) &osvi))
return iType;
}
switch (osvi.dwPlatformId) {
case VER_PLATFORM_WIN32_NT:
if (osvi.dwMajorVersion >= 6)
iType = OS_VISTA_UPPER;
else if (osvi.dwMajorVersion == 5)
iType = OS_XP;
break;
default:
break;
}
#endif
return iType;
}
void Write2File (FILE* pFp, unsigned char* pData[3], int iStride[2], int iWidth, int iHeight) {
int i;
unsigned char* pPtr = NULL;
pPtr = pData[0];
for (i = 0; i < iHeight; i++) {
fwrite (pPtr, 1, iWidth, pFp);
pPtr += iStride[0];
}
iHeight = iHeight / 2;
iWidth = iWidth / 2;
pPtr = pData[1];
for (i = 0; i < iHeight; i++) {
fwrite (pPtr, 1, iWidth, pFp);
pPtr += iStride[1];
}
pPtr = pData[2];
for (i = 0; i < iHeight; i++) {
fwrite (pPtr, 1, iWidth, pFp);
pPtr += iStride[1];
}
}