hl2_src-leak-2017/src/utils/scratchpad3dviewer/d3dapp.cpp

561 lines
10 KiB
C++

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//
//=============================================================================//
// TerrainBlend.cpp : Defines the entry point for the application.
//
#include "stdafx.h"
#include "resource.h"
#include "mathlib/mathlib.h"
#include "tier0/dbg.h"
#include "tier0/icommandline.h"
#include "tier1/strtools.h"
#define MAX_LOADSTRING 100
// Global Variables:
HINSTANCE hInst; // current instance
TCHAR szTitle[MAX_LOADSTRING]; // The title bar text
TCHAR szWindowClass[MAX_LOADSTRING]; // The title bar text
HWND g_hWnd;
// Foward declarations of functions included in this code module:
ATOM MyRegisterClass(HINSTANCE hInstance);
BOOL InitInstance(HINSTANCE, int);
LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
LRESULT CALLBACK About(HWND, UINT, WPARAM, LPARAM);
int g_nCapture = 0;
bool g_bFocus = false;
int g_ScreenWidth, g_ScreenHeight;
D3DPRESENT_PARAMETERS d3dpp;
IDirect3D8 *g_pDirect3D;
IDirect3DDevice8 *g_pDevice;
POINT GetWindowCenter()
{
RECT rc;
GetWindowRect(g_hWnd, &rc);
POINT ret;
ret.x = (rc.left + rc.right) / 2;
ret.y = (rc.top + rc.bottom) / 2;
return ret;
}
void CallAppRender( bool bInvalidRect )
{
static DWORD lastTime = 0;
static POINT lastMousePos = {0xFFFF, 0xFFFF};
// Sample time and mouse position and tell the app to render.
DWORD curTime = GetTickCount();
float frametime = (curTime - lastTime) / 1000.0f;
if( frametime > 0.1f )
frametime = 0.1f;
lastTime = curTime;
// Get the cursor delta.
POINT curMousePos;
GetCursorPos(&curMousePos);
int deltaX, deltaY;
if( lastMousePos.x == 0xFFFF )
{
deltaX = deltaY = 0;
}
else
{
deltaX = curMousePos.x - lastMousePos.x;
deltaY = curMousePos.y - lastMousePos.y;
}
// Recenter the cursor.
if( g_nCapture )
{
lastMousePos = GetWindowCenter();
SetCursorPos( lastMousePos.x, lastMousePos.y );
}
else
{
lastMousePos = curMousePos;
}
AppRender( frametime, (float)deltaX, (float)deltaY, bInvalidRect );
}
SpewRetval_t D3DAppSpewFunc( SpewType_t spewType, char const *pMsg )
{
switch (spewType)
{
case SPEW_ERROR:
MessageBox(NULL, pMsg, "FATAL ERROR", MB_OK);
return SPEW_ABORT;
default:
OutputDebugString(pMsg);
#ifdef _DEBUG
return spewType == SPEW_ASSERT ? SPEW_DEBUGGER : SPEW_CONTINUE;
#else
return SPEW_CONTINUE;
#endif
}
}
int APIENTRY WinMain(HINSTANCE hInstance,
HINSTANCE hPrevInstance,
LPSTR lpCmdLine,
int nCmdShow)
{
CommandLine()->CreateCmdLine( Plat_GetCommandLine() );
// TODO: Place code here.
SpewOutputFunc( D3DAppSpewFunc );
MathLib_Init( true, true, true, 2.2f, 2.2f, 0.0f, 2.0f );
MSG msg;
HACCEL hAccelTable;
// Initialize global strings
LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
strcpy( szWindowClass, "d3dapp" );
MyRegisterClass(hInstance);
// Perform application initialization:
if (!InitInstance (hInstance, nCmdShow))
{
return FALSE;
}
hAccelTable = LoadAccelerators(hInstance, (LPCTSTR)IDC_TERRAINBLEND);
InvalidateRect( g_hWnd, NULL, FALSE );
// Main message loop:
while(1)
{
while(PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
{
if(msg.message == WM_QUIT)
break;
if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}
if(msg.message == WM_QUIT)
break;
CallAppRender( false );
}
AppExit();
return msg.wParam;
}
//
// FUNCTION: MyRegisterClass()
//
// PURPOSE: Registers the window class.
//
// COMMENTS:
//
// This function and its usage is only necessary if you want this code
// to be compatible with Win32 systems prior to the 'RegisterClassEx'
// function that was added to Windows 95. It is important to call this function
// so that the application will get 'well formed' small icons associated
// with it.
//
ATOM MyRegisterClass(HINSTANCE hInstance)
{
WNDCLASSEX wcex;
wcex.cbSize = sizeof(WNDCLASSEX);
wcex.style = CS_HREDRAW | CS_VREDRAW;
wcex.lpfnWndProc = (WNDPROC)WndProc;
wcex.cbClsExtra = 0;
wcex.cbWndExtra = 0;
wcex.hInstance = hInstance;
wcex.hIcon = LoadIcon(hInstance, (LPCTSTR)IDI_TERRAINBLEND);
wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
wcex.lpszMenuName = NULL;
wcex.lpszClassName = szWindowClass;
wcex.hIconSm = LoadIcon(wcex.hInstance, (LPCTSTR)IDI_SMALL);
return RegisterClassEx(&wcex);
}
void ShutdownD3D()
{
if( g_pDevice )
{
g_pDevice->Release();
g_pDevice = NULL;
}
if( g_pDirect3D )
{
g_pDirect3D->Release();
g_pDirect3D = NULL;
}
}
void InitD3D()
{
ShutdownD3D();
RECT rcClient;
GetClientRect( g_hWnd, &rcClient );
g_ScreenWidth = rcClient.right - rcClient.left;
g_ScreenHeight = rcClient.bottom - rcClient.top;
// Initialize D3D.
g_pDirect3D = Direct3DCreate8( D3D_SDK_VERSION );
// Get the current desktop display mode, so we can set up a back
// buffer of the same format
D3DDISPLAYMODE d3ddm;
g_pDirect3D->GetAdapterDisplayMode( D3DADAPTER_DEFAULT, &d3ddm );
// Set up the structure used to create the D3DDevice
ZeroMemory( &d3dpp, sizeof(d3dpp) );
d3dpp.Windowed = TRUE;
d3dpp.SwapEffect = D3DSWAPEFFECT_DISCARD;
d3dpp.BackBufferFormat = d3ddm.Format;
d3dpp.EnableAutoDepthStencil = TRUE;
d3dpp.AutoDepthStencilFormat = D3DFMT_D24S8;
d3dpp.MultiSampleType = D3DMULTISAMPLE_NONE;
HRESULT hr = g_pDirect3D->CreateDevice(
D3DADAPTER_DEFAULT,
D3DDEVTYPE_HAL,
g_hWnd,
D3DCREATE_SOFTWARE_VERTEXPROCESSING,
&d3dpp,
&g_pDevice);
if( FAILED(hr) )
{
Sys_Error( "CreateDevice failed (%s)", DXGetErrorString8(hr) );
}
g_pDevice->SetRenderState( D3DRS_CULLMODE, D3DCULL_NONE );
g_pDevice->SetRenderState( D3DRS_LIGHTING, FALSE );
}
void DoResize()
{
AppPreResize();
InitD3D();
AppPostResize();
CallAppRender( true );
}
//
// FUNCTION: InitInstance(HANDLE, int)
//
// PURPOSE: Saves instance handle and creates main window
//
// COMMENTS:
//
// In this function, we save the instance handle in a global variable and
// create and display the main program window.
//
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
{
hInst = hInstance; // Store instance handle in our global variable
int x = Sys_FindArgInt( "-x", 0 );
int y = Sys_FindArgInt( "-y", 0 );
int w = Sys_FindArgInt( "-width", CW_USEDEFAULT );
int h = Sys_FindArgInt( "-height", CW_USEDEFAULT );
DWORD dwFlags = WS_OVERLAPPEDWINDOW;
// Get the 'work area' so we don't overlap the taskbar on the top or left.
RECT rcWorkArea;
SystemParametersInfo( SPI_GETWORKAREA, 0, &rcWorkArea, 0 );
// If they don't specify anything, maximize the window.
if( x == 0 && y == 0 && w == CW_USEDEFAULT && h == CW_USEDEFAULT )
{
x = rcWorkArea.left;
y = rcWorkArea.top;
w = rcWorkArea.right - rcWorkArea.left;
h = rcWorkArea.bottom - rcWorkArea.top;
dwFlags |= WS_MAXIMIZE;
}
else
{
x += rcWorkArea.left;
y += rcWorkArea.top;
}
g_hWnd = CreateWindow(
szWindowClass,
szTitle,
dwFlags, // window style
x, // x
y, // y
w, // width
h, // height
NULL,
NULL,
hInstance,
NULL);
if (!g_hWnd)
{
return FALSE;
}
ShowWindow(g_hWnd, nCmdShow);
UpdateWindow(g_hWnd);
InitD3D();
AppInit();
// Reinitialize D3D. For some reason, D3D point primitives are way too large
// unless we do this.
DoResize();
return TRUE;
}
//
// FUNCTION: WndProc(HWND, unsigned, WORD, LONG)
//
// PURPOSE: Processes messages for the main window.
//
// WM_COMMAND - process the application menu
// WM_PAINT - Paint the main window
// WM_DESTROY - post a quit message and return
//
//
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
switch (message)
{
case WM_PAINT:
{
PAINTSTRUCT ps;
BeginPaint( hWnd, &ps );
EndPaint( hWnd, &ps );
if( g_pDevice )
CallAppRender( true );
}
break;
case WM_KEYDOWN:
{
AppKey( (int)wParam, lParam&0xFFFF );
}
break;
case WM_KEYUP:
{
AppKey( (int)wParam, 0 );
}
break;
case WM_CHAR:
{
AppChar( (int)wParam );
}
break;
case WM_DESTROY:
PostQuitMessage(0);
break;
case WM_LBUTTONDOWN:
{
ShowCursor( FALSE );
SetCapture( g_hWnd );
g_nCapture++;
}
break;
case WM_LBUTTONUP:
{
ShowCursor( TRUE );
ReleaseCapture( );
g_nCapture--;
}
break;
case WM_RBUTTONDOWN:
{
ShowCursor( FALSE );
SetCapture( g_hWnd );
g_nCapture++;
}
break;
case WM_RBUTTONUP:
{
ShowCursor( TRUE );
ReleaseCapture( );
g_nCapture--;
}
break;
case WM_SIZE:
{
if( g_pDevice )
DoResize();
}
break;
case WM_SETFOCUS:
{
g_bFocus = true;
}
break;
case WM_KILLFOCUS:
{
g_bFocus = false;
}
break;
default:
return DefWindowProc(hWnd, message, wParam, lParam);
}
return 0;
}
bool Sys_Error(const char *pMsg, ...)
{
char msg[4096];
va_list marker;
va_start( marker, pMsg );
V_vsprintf_safe( msg, pMsg, marker );
va_end( marker );
MessageBox( NULL, msg, "Error!", MB_OK );
exit(1);
return true;
}
void Sys_Quit()
{
PostQuitMessage( 0 );
}
void Sys_SetWindowText( char const *pMsg, ... )
{
va_list marker;
char msg[4096];
va_start(marker, pMsg);
V_vsprintf_safe(msg, pMsg, marker);
va_end(marker);
SetWindowText( g_hWnd, msg );
}
bool Sys_GetKeyState( int key )
{
int keyTranslations[][2] =
{
{APPKEY_LBUTTON, VK_LBUTTON},
{APPKEY_RBUTTON, VK_RBUTTON},
{APPKEY_SPACE, VK_SPACE}
};
int nKeyTranslations = sizeof(keyTranslations) / sizeof(keyTranslations[0]);
for( int i=0; i < nKeyTranslations; i++ )
{
if( key == keyTranslations[i][0] )
{
key = keyTranslations[i][1];
break;
}
}
return !!( GetAsyncKeyState( key ) & 0x8000 );
}
void Sys_Sleep( int ms )
{
Sleep( (DWORD)ms );
}
bool Sys_HasFocus()
{
return g_bFocus;
}
char const* Sys_FindArg( char const *pArg, char const *pDefault )
{
for( int i=0; i < __argc; i++ )
{
if( stricmp( __argv[i], pArg ) == 0 )
return (i+1) < __argc ? __argv[i+1] : "";
}
return pDefault;
}
int Sys_FindArgInt( char const *pArg, int defaultVal )
{
char const *pVal = Sys_FindArg( pArg, NULL );
if( pVal )
return atoi( pVal );
else
return defaultVal;
}
int Sys_ScreenWidth()
{
return g_ScreenWidth;
}
int Sys_ScreenHeight()
{
return g_ScreenHeight;
}