Files
SteamworksSDK/SteamworksExample/GameEngine.cpp
T
Riley Labrecque 1383b358e4 v1.00
2016-07-07 05:54:47 -06:00

1631 lines
45 KiB
C++

//========= Copyright © 1996-2008, Valve LLC, All rights reserved. ============
//
// Purpose: Main class for the game engine
//
// $NoKeywords: $
//=============================================================================
#include "stdafx.h"
#include "GameEngine.h"
// Allocate static member
std::map<HWND, CGameEngine* > CGameEngine::m_MapEngineInstances;
//-----------------------------------------------------------------------------
// Purpose: WndProc
//-----------------------------------------------------------------------------
LRESULT CALLBACK GameWndProc( HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam )
{
switch ( msg )
{
case WM_CLOSE:
case WM_DESTROY:
case WM_QUIT:
{
CGameEngine *pGameEngine = CGameEngine::FindEngineInstanceForHWND( hWnd );
if ( pGameEngine )
pGameEngine->Shutdown();
else
OutputDebugString( "Failed to find game engine instance for hwnd\n" );
PostQuitMessage( 0 );
return(0);
} break;
case WM_KEYDOWN:
case WM_SYSKEYDOWN:
{
CGameEngine *pGameEngine = CGameEngine::FindEngineInstanceForHWND( hWnd );
if ( pGameEngine )
{
pGameEngine->RecordKeyDown( (DWORD) wParam );
return 0;
}
else
{
OutputDebugString( "Failed to find game engine for hwnd, key down event lost\n" );
}
}
break;
case WM_KEYUP:
case WM_SYSKEYUP:
{
CGameEngine *pGameEngine = CGameEngine::FindEngineInstanceForHWND( hWnd );
if ( pGameEngine )
{
pGameEngine->RecordKeyUp( (DWORD) wParam );
return 0;
}
else
{
OutputDebugString( "Failed to find game engine for hwnd, key up event lost\n" );
}
}
break;
// Add additional handlers for things like input here...
default:
break;
}
return DefWindowProc(hWnd, msg, wParam, lParam);
}
//-----------------------------------------------------------------------------
// Purpose: Constructor for game engine instance
//-----------------------------------------------------------------------------
CGameEngine::CGameEngine( HINSTANCE hInstance, int nShowCommand, int32 nWindowWidth, int32 nWindowHeight )
{
m_bEngineReadyForUse = false;
m_bShuttingDown = false;
m_hInstance = hInstance;
m_hWnd = NULL;
m_pD3D9Interface = NULL;
m_pD3D9Device = NULL;
m_nWindowWidth = nWindowWidth;
m_nWindowHeight = nWindowHeight;
m_nNextFontHandle = 1;
m_nNextVertBufferHandle = 1;
m_nNextTextureHandle = 1;
m_hLineBuffer = NULL;
m_pLineVertexes = NULL;
m_dwLinesToFlush = 0;
m_dwLineBufferBatchPos = 0;
m_hPointBuffer = NULL;
m_pPointVertexes = NULL;
m_dwPointsToFlush = 0;
m_dwPointBufferBatchPos = 0;
m_hQuadBuffer = NULL;
m_pQuadVertexes = NULL;
m_dwQuadsToFlush = 0;
m_dwQuadBufferBatchPos = 0;
m_hTextureWhite = NULL;
m_bDeviceLost = false;
m_bFirstFrame = true;
m_flMaxFPS = 120.0f;
m_hLastTexture = NULL;
m_ulPreviousGameTickCount = 0;
m_ulGameTickCount = 0;
// restrict this main game thread to the first processor, so queryperformance counter won't jump on crappy AMD cpus
DWORD dwThreadAffinityMask = 0x01;
::SetThreadAffinityMask( ::GetCurrentThread(), dwThreadAffinityMask );
LARGE_INTEGER l;
if ( QueryPerformanceFrequency( &l ) )
{
m_ulPerfCounterToMillisecondsDivisor = l.QuadPart/1000;
}
else
{
m_ulPerfCounterToMillisecondsDivisor = 0;
}
m_dwBackgroundColor = D3DCOLOR_ARGB(0, 255, 255, 255 );
if ( !BCreateGameWindow( nShowCommand ) || !m_hWnd )
{
OutputDebugString( "Failed creating game window\n" );
return;
}
CGameEngine::AddInstanceToHWNDMap( this, m_hWnd );
if ( !BInitializeD3D9() )
{
::MessageBoxA( NULL, "Failed to initialize D3D9.\n\nGame will now exit.", "SteamworksExample - Fatal error", MB_OK | MB_ICONERROR );
return;
}
RECT r;
::GetClientRect( m_hWnd, &r );
m_nViewportWidth = r.right-r.left;
m_nViewportHeight = r.bottom-r.top;
m_bEngineReadyForUse = true;
}
//-----------------------------------------------------------------------------
// Purpose: Shutdown the game engine
//-----------------------------------------------------------------------------
void CGameEngine::Shutdown()
{
// Flag that we are shutting down so the frame loop will stop running
m_bShuttingDown = true;
// Cleanup D3D fonts
{
std::map<HGAMEFONT, ID3DXFont *>::iterator iter;
for( iter = m_MapFontInstances.begin(); iter != m_MapFontInstances.end(); ++iter )
{
iter->second->Release();
iter->second = NULL;
}
m_MapFontInstances.clear();
}
// Cleanup D3D vertex buffers
{
std::map<HGAMEVERTBUF, VertBufData_t>::iterator iter;
for( iter = m_MapVertexBuffers.begin(); iter != m_MapVertexBuffers.end(); ++iter )
{
if ( iter->second.m_bIsLocked )
iter->second.m_pBuffer->Unlock();
iter->second.m_pBuffer->Release();
iter->second.m_pBuffer = NULL;
}
m_MapVertexBuffers.clear();
}
// Cleanup D3D textures
{
std::map<HGAMETEXTURE, TextureData_t>::iterator iter;
for( iter = m_MapTextures.begin(); iter != m_MapTextures.end(); ++iter )
{
if ( iter->second.m_pRGBAData )
{
delete[] iter->second.m_pRGBAData;
iter->second.m_pRGBAData = NULL;
}
if ( iter->second.m_pTexture )
{
iter->second.m_pTexture->Release();
iter->second.m_pTexture = NULL;
}
}
m_MapTextures.clear();
}
// Cleanup D3D
if ( m_pD3D9Device )
{
m_pD3D9Device->Release();
m_pD3D9Device = NULL;
}
if ( m_pD3D9Interface )
{
m_pD3D9Interface->Release();
m_pD3D9Interface = NULL;
}
// Destroy our window
if ( m_hWnd )
{
if ( !DestroyWindow( m_hWnd ) )
{
// We failed to destroy our window. This shouldn't ever happen.
OutputDebugString( "Failed destroying window\n" );
}
else
{
// Clean up any pending messages.
MSG msg;
while( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
{
DispatchMessage(&msg);
}
}
CGameEngine::RemoveInstanceFromHWNDMap( m_hWnd );
m_hWnd = NULL;
}
// Unregister our window class
if ( m_hInstance )
{
if ( !UnregisterClass( "SteamworksExample", m_hInstance ) )
{
OutputDebugString( "Failed unregistering window class\n" );
}
m_hInstance = NULL;
}
}
//-----------------------------------------------------------------------------
// Purpose: Handle losing the d3d device (ie, release resources)
//-----------------------------------------------------------------------------
bool CGameEngine::BHandleLostDevice()
{
bool bFullySuccessful = true;
// Clear our saved FVF so that we will manually reset it once we re-acquire the device
m_dwCurrentFVF = NULL;
// Clear our internal buffers and batching data
m_dwLinesToFlush = 0;
m_dwLineBufferBatchPos = 0;
m_pLineVertexes = NULL;
m_hLineBuffer = NULL;
m_dwPointsToFlush = 0;
m_dwPointBufferBatchPos = 0;
m_pPointVertexes = NULL;
m_hPointBuffer = NULL;
m_dwQuadsToFlush = 0;
m_dwQuadBufferBatchPos = 0;
m_pQuadVertexes = NULL;
m_hQuadBuffer = NULL;
// Fonts are easy since we used d3dx, they have their own handlers
{
std::map<HGAMEFONT, ID3DXFont *>::iterator iter;
for( iter = m_MapFontInstances.begin(); iter != m_MapFontInstances.end(); ++iter )
{
if ( FAILED( iter->second->OnLostDevice() ) )
{
bFullySuccessful = false;
OutputDebugString( "Failed OnLostDevice on a font object\n" );
}
}
}
// Vertex buffers we have to release and then re-create later, since we only use them internal
// to the engine we can just free them all, and we'll know how to recreate them later on demand
{
std::map<HGAMEVERTBUF, VertBufData_t>::iterator iter;
for( iter = m_MapVertexBuffers.begin(); iter != m_MapVertexBuffers.end(); ++iter )
{
if ( iter->second.m_pBuffer )
{
if ( iter->second.m_bIsLocked )
iter->second.m_pBuffer->Unlock();
iter->second.m_bIsLocked = false;
iter->second.m_pBuffer->Release();
iter->second.m_pBuffer = NULL;
}
}
m_MapVertexBuffers.clear();
}
// Textures we can just release, and they will be recreated on demand when used again
{
std::map<HGAMETEXTURE, TextureData_t>::iterator iter;
for( iter = m_MapTextures.begin(); iter != m_MapTextures.end(); ++iter )
{
if ( iter->second.m_pTexture )
{
iter->second.m_pTexture->Release();
iter->second.m_pTexture = NULL;
}
}
}
return bFullySuccessful;
}
//-----------------------------------------------------------------------------
// Purpose: Handle device reset after losing it
//-----------------------------------------------------------------------------
bool CGameEngine::BHandleResetDevice()
{
bool bFullySuccessful = true;
ResetRenderStates();
// Fonts are easy since we used d3dx, they have their own handlers
{
std::map<HGAMEFONT, ID3DXFont *>::iterator iter;
for( iter = m_MapFontInstances.begin(); iter != m_MapFontInstances.end(); ++iter )
{
if ( FAILED( iter->second->OnResetDevice() ) )
{
OutputDebugString( "Reset for a font object failed\n" );
bFullySuccessful = false;
}
}
}
// Vertex buffers we only use internal to the class, so we know
// how to recreate them on demand. Nothing to do here for them.
return bFullySuccessful;
}
//-----------------------------------------------------------------------------
// Purpose: Resets all the render, texture, and sampler states to our defaults
//-----------------------------------------------------------------------------
void CGameEngine::ResetRenderStates()
{
// Since we are just a really basic rendering engine we'll setup our initial
// render states here and we can just assume that they don't change later
m_pD3D9Device->SetRenderState( D3DRS_LIGHTING, FALSE );
m_pD3D9Device->SetRenderState( D3DRS_ALPHABLENDENABLE, TRUE );
m_pD3D9Device->SetRenderState( D3DRS_SRCBLEND, D3DBLEND_BOTHSRCALPHA );
m_pD3D9Device->SetRenderState( D3DRS_DESTBLEND, D3DBLEND_INVSRCALPHA );
// texture stage state
m_pD3D9Device->SetTextureStageState( 0, D3DTSS_COLOROP, D3DTOP_MODULATE );
m_pD3D9Device->SetTextureStageState( 0, D3DTSS_COLORARG1, D3DTA_TEXTURE );
m_pD3D9Device->SetTextureStageState( 0, D3DTSS_COLORARG2, D3DTA_CURRENT );
m_pD3D9Device->SetTextureStageState( 0, D3DTSS_ALPHAOP, D3DTOP_MODULATE );
m_pD3D9Device->SetTextureStageState( 0, D3DTSS_ALPHAARG1, D3DTA_TEXTURE );
m_pD3D9Device->SetTextureStageState( 0, D3DTSS_ALPHAARG2, D3DTA_CURRENT );
// sampler state
m_pD3D9Device->SetSamplerState( 0, D3DSAMP_MINFILTER, D3DTEXF_LINEAR );
m_pD3D9Device->SetSamplerState( 0, D3DSAMP_MAGFILTER, D3DTEXF_LINEAR );
m_pD3D9Device->SetSamplerState( 0, D3DSAMP_MIPFILTER, D3DTEXF_LINEAR );
}
//-----------------------------------------------------------------------------
// Purpose: Creates the window for the game to use
//-----------------------------------------------------------------------------
bool CGameEngine::BCreateGameWindow( int nShowCommand )
{
WNDCLASS wc;
DWORD style;
// Register the window class.
wc.lpszClassName = "SteamworksExample";
wc.lpfnWndProc = GameWndProc;
wc.style = 0;
wc.cbClsExtra = 0;
wc.cbWndExtra = 0;
wc.hInstance = m_hInstance;
wc.hIcon = LoadIcon(NULL,IDI_APPLICATION);
wc.hCursor = LoadCursor(NULL,IDC_ARROW);
wc.hbrBackground = (HBRUSH)GetStockObject(BLACK_BRUSH);
wc.lpszMenuName = NULL;
if ( !RegisterClass( &wc ) )
{
OutputDebugString( "Failure registering window class\n" );
return false;
}
// Set parent window mode (normal system window with overlap/draw-ordering)
style = WS_OVERLAPPED|WS_SYSMENU;
// Create actual window
m_hWnd = CreateWindow( "SteamworksExample",
"SteamworksExample",
style,
0,
0,
m_nWindowWidth,
m_nWindowHeight,
NULL,
NULL,
m_hInstance,
NULL );
if ( m_hWnd == NULL )
{
OutputDebugString( "Failed to create window for CGameEngine\n" );
return false;
}
// Give focus to newly created app window.
::ShowWindow( m_hWnd, nShowCommand );
::UpdateWindow( m_hWnd );
::SetFocus( m_hWnd );
return true;
}
bool CGameEngine::BInitializeD3D9()
{
if ( !m_pD3D9Interface )
{
// Initialize the d3d interface
m_pD3D9Interface = Direct3DCreate9( D3D_SDK_VERSION );
if ( m_pD3D9Interface == NULL )
{
OutputDebugString( "Direct3DCreate9 failed\n" );
return false;
}
}
if ( !m_pD3D9Device )
{
D3DDISPLAYMODE d3ddisplaymode;
// Get the current desktop display mode, only needed if running in a window.
HRESULT hRes = m_pD3D9Interface->GetAdapterDisplayMode( D3DADAPTER_DEFAULT, &d3ddisplaymode );
if (FAILED(hRes))
{
OutputDebugString( "GetAdapterDisplayMode failed\n");
return false;
}
// Setup presentation parameters
ZeroMemory( &m_d3dpp, sizeof( m_d3dpp ) );
m_d3dpp.SwapEffect = D3DSWAPEFFECT_DISCARD;
m_d3dpp.PresentationInterval = D3DPRESENT_INTERVAL_IMMEDIATE; // no v-sync
m_d3dpp.hDeviceWindow = m_hWnd;
m_d3dpp.BackBufferCount = 1;
m_d3dpp.EnableAutoDepthStencil = TRUE;
m_d3dpp.AutoDepthStencilFormat = D3DFMT_D16;
m_d3dpp.Windowed = TRUE; // bugbug jmccaskey - make a parameter?
m_d3dpp.BackBufferFormat = d3ddisplaymode.Format;
// Create Direct3D9 device
// (if it fails to create hardware vertex processing, then go with the software alternative).
hRes = m_pD3D9Interface->CreateDevice(
D3DADAPTER_DEFAULT,
D3DDEVTYPE_HAL,
m_hWnd,
D3DCREATE_HARDWARE_VERTEXPROCESSING,
&m_d3dpp,
&m_pD3D9Device );
// Could not create a hardware device, create a software one instead (slow....)
if ( FAILED( hRes ) )
{
hRes = m_pD3D9Interface->CreateDevice(
D3DADAPTER_DEFAULT,
D3DDEVTYPE_HAL,
m_hWnd,
D3DCREATE_SOFTWARE_VERTEXPROCESSING,
&m_d3dpp,
&m_pD3D9Device );
}
// If we couldn't create a device even with software vertex processing then
// it's a fatal error
if ( FAILED( hRes ) )
{
// Make sure the pointer is NULL after failures (seems it sometimes gets modified even when failing)
m_pD3D9Device = NULL;
OutputDebugString( "Failed to create D3D9 device\n" );
return false;
}
//Initialize our render, texture, and sampler stage states
ResetRenderStates();
}
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Set the background color to clear to
//-----------------------------------------------------------------------------
void CGameEngine::SetBackgroundColor( short a, short r, short g, short b )
{
m_dwBackgroundColor = D3DCOLOR_ARGB( a, r, g, b );
}
//-----------------------------------------------------------------------------
// Purpose: Start a new frame
//-----------------------------------------------------------------------------
bool CGameEngine::StartFrame()
{
// Track current game time
// bugbug jmccaskey - use a more reliable timer that won't wrap and has better precision!
if ( m_bFirstFrame )
{
LARGE_INTEGER l;
if ( !QueryPerformanceCounter( &l ) )
OutputDebugString( "QueryPerformanceCounter failed\n" );
m_ulLastQueryPerformanceCounterValue = l.QuadPart;
m_ulPreviousGameTickCount = 0;
m_ulGameTickCount = 0;
m_bFirstFrame = false;
}
else
{
m_ulPreviousGameTickCount = m_ulGameTickCount;
LARGE_INTEGER l;
if ( !QueryPerformanceCounter( &l ) )
OutputDebugString( "QueryPerformanceCounter failed\n" );
m_ulGameTickCount += (l.QuadPart - m_ulLastQueryPerformanceCounterValue)/m_ulPerfCounterToMillisecondsDivisor;
m_ulLastQueryPerformanceCounterValue = l.QuadPart;
}
// Before doing anything else pump messages
MessagePump();
// Detect if we lose focus and cause all keys do go up so they aren't stuck
if ( ::GetForegroundWindow() != m_hWnd )
{
m_SetKeysDown.clear();
}
// Message pump may have lead us to shutdown, check
if ( BShuttingDown() )
return false;
if ( !m_pD3D9Device )
return false;
// Test that we haven't lost the device
HRESULT hRes = m_pD3D9Device->TestCooperativeLevel();
if ( hRes == D3DERR_DEVICELOST )
{
// Device is currently lost, can't render frames, but lets just let things continue so the
// simulation keeps running, the game should probably pause itself
// If it is newly lost then release resources
if ( !m_bDeviceLost )
{
OutputDebugString( "Device lost\n" );
// HandleLostDevice() will free all our resources
if ( !BHandleLostDevice() )
OutputDebugString( "Failed to release all resources for lost device\n" );
}
m_bDeviceLost = true;
}
else if ( hRes == D3DERR_DEVICENOTRESET )
{
OutputDebugString( "Getting ready to reset device\n" );
// Reset the device
hRes = m_pD3D9Device->Reset( &m_d3dpp );
if ( !FAILED( hRes ) )
{
m_bDeviceLost = false;
// Acquire all our resources again
if ( !BHandleResetDevice() )
{
OutputDebugString( "Failed to acquire all resources again after device reset\n" );
}
}
else
{
OutputDebugString( "Reset() call on device failed\n" );
::MessageBox( m_hWnd, "m_pD3D9Device->Reset() call has failed unexpectedly\n", "Fatal Error", MB_OK );
Shutdown();
}
}
// Return true even though we can't render, frames can still run otherwise
// and the game should continue its simulation or choose to pause on its own
if ( m_bDeviceLost )
return true;
hRes = m_pD3D9Device->BeginScene();
if ( FAILED( hRes ) )
return false;
// Clear the back buffer and z-buffer
hRes = m_pD3D9Device->Clear( 0, NULL, D3DCLEAR_TARGET|D3DCLEAR_ZBUFFER, m_dwBackgroundColor, 1.0f, 0 );
if ( FAILED( hRes ) )
return false;
return true;
}
//-----------------------------------------------------------------------------
// Purpose: End the current frame
//-----------------------------------------------------------------------------
void CGameEngine::EndFrame()
{
if ( BShuttingDown() )
return;
if ( !m_pD3D9Device )
return;
if ( m_bDeviceLost )
return;
// See if we lost the device in the middle of the frame
HRESULT hRes = m_pD3D9Device->TestCooperativeLevel();
if ( hRes == D3DERR_DEVICELOST )
{
// Ok, StartFrame will handle this next time through, but bail out early here
return;
}
// Flush point buffer
BFlushPointBuffer();
// Flush line buffer
BFlushLineBuffer();
// Flush quad buffer
BFlushQuadBuffer();
hRes = m_pD3D9Device->EndScene();
if ( FAILED( hRes ) )
{
OutputDebugString( "EndScene() call failed\n" );
return;
}
hRes = m_pD3D9Device->Present( NULL, NULL, NULL, NULL );
if ( FAILED( hRes ) )
{
OutputDebugString( "Present() call failed\n" );
return;
}
// Frame rate limiting
float flDesiredFrameMilliseconds = 1000.0f/m_flMaxFPS;
while( 1 )
{
uint64 ulCurrentTickCount = m_ulGameTickCount;
LARGE_INTEGER l;
if ( !QueryPerformanceCounter( &l ) )
OutputDebugString( "QueryPerformanceCounter failed\n" );
ulCurrentTickCount += (l.QuadPart-m_ulLastQueryPerformanceCounterValue)/m_ulPerfCounterToMillisecondsDivisor;
float flMillisecondsElapsed = (float)(ulCurrentTickCount - m_ulGameTickCount);
if ( flMillisecondsElapsed < flDesiredFrameMilliseconds )
{
// If enough time is left sleep, otherwise just keep spinning so we don't go over the limit...
if ( flDesiredFrameMilliseconds - flMillisecondsElapsed > 2.0f )
Sleep( 2 );
else
continue;
}
else
break;
}
}
//-----------------------------------------------------------------------------
// Purpose: Creates a new vertex buffer
//-----------------------------------------------------------------------------
HGAMEVERTBUF CGameEngine::HCreateVertexBuffer( uint32 nSizeInBytes, DWORD dwUsage, DWORD dwFVF )
{
if ( !m_pD3D9Device )
return false;
// Create a vertex buffer object
IDirect3DVertexBuffer9 *pVertBuffer;
HRESULT hRes = m_pD3D9Device->CreateVertexBuffer( nSizeInBytes, dwUsage,
dwFVF, D3DPOOL_DEFAULT, &pVertBuffer, NULL );
if ( FAILED( hRes ) )
{
OutputDebugString( "Failed creating vertex buffer\n" );
return 0;
}
HGAMEFONT hVertBuf = m_nNextVertBufferHandle;
++m_nNextVertBufferHandle;
VertBufData_t data;
data.m_bIsLocked = false;
data.m_pBuffer = pVertBuffer;
m_MapVertexBuffers[ hVertBuf ] = data;
return hVertBuf;
}
//-----------------------------------------------------------------------------
// Purpose: Locks an entire vertex buffer with the specified flags into memory
//-----------------------------------------------------------------------------
bool CGameEngine::BLockEntireVertexBuffer( HGAMEVERTBUF hVertBuf, void **ppVoid, DWORD dwFlags )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return false;
if ( !hVertBuf )
{
OutputDebugString( "Someone is calling BLockEntireVertexBuffer() with a null handle\n" );
return false;
}
// Find the font object for the passed handle
std::map<HGAMEVERTBUF, VertBufData_t>::iterator iter;
iter = m_MapVertexBuffers.find( hVertBuf );
if ( iter == m_MapVertexBuffers.end() )
{
OutputDebugString( "Invalid vertex buffer handle passed to BLockEntireVertexBuffer()\n" );
return false;
}
// Make sure the pointer is valid
if ( !iter->second.m_pBuffer )
{
OutputDebugString( "Pointer to vertex buffer is invalid (lost device and not recreated?)!\n" );
return false;
}
// Make sure its not already locked
if ( iter->second.m_bIsLocked )
{
OutputDebugString( "Trying to lock an already locked vertex buffer!\n" );
return false;
}
// we have the buffer, try to lock it
if( FAILED( iter->second.m_pBuffer->Lock( 0, 0, ppVoid, dwFlags ) ) )
{
OutputDebugString( "BLockEntireVertexBuffer call failed\n" );
return false;
}
// Track that we are now locked
iter->second.m_bIsLocked = true;
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Unlocks a vertex buffer
//-----------------------------------------------------------------------------
bool CGameEngine::BUnlockVertexBuffer( HGAMEVERTBUF hVertBuf )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return false;
if ( !hVertBuf )
{
OutputDebugString( "Someone is calling BUnlockVertexBuffer() with a null handle\n" );
return false;
}
// Find the vertex buffer for the passed handle
std::map<HGAMEVERTBUF, VertBufData_t>::iterator iter;
iter = m_MapVertexBuffers.find( hVertBuf );
if ( iter == m_MapVertexBuffers.end() )
{
OutputDebugString( "Invalid vertex buffer handle passed to BUnlockVertexBuffer()\n" );
return false;
}
// Make sure the pointer is valid
if ( !iter->second.m_pBuffer )
{
OutputDebugString( "Pointer to vertex buffer is invalid (lost device and not recreated?)!\n" );
return false;
}
// Make sure we are locked if someone is trying to unlock
if ( !iter->second.m_bIsLocked )
{
OutputDebugString( "Trying to unlock a vertex buffer that is not locked!\n" );
return false;
}
// we have the buffer, try to lock it
if( FAILED( iter->second.m_pBuffer->Unlock() ) )
{
OutputDebugString( "BUnlockVertexBuffer call failed\n" );
return false;
}
// Track that we are now unlocked
iter->second.m_bIsLocked = false;
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Release a vertex buffer and free its resources
//-----------------------------------------------------------------------------
bool CGameEngine::BReleaseVertexBuffer( HGAMEVERTBUF hVertBuf )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return false;
if ( !hVertBuf )
{
OutputDebugString( "Someone is calling BReleaseVertexBuffer() with a null handle\n" );
return false;
}
// Find the vertex buffer object for the passed handle
std::map<HGAMEVERTBUF, VertBufData_t>::iterator iter;
iter = m_MapVertexBuffers.find( hVertBuf );
if ( iter == m_MapVertexBuffers.end() )
{
OutputDebugString( "Invalid vertex buffer handle passed to BReleaseVertexBuffer()\n" );
return false;
}
// Make sure the pointer is valid
if ( !iter->second.m_pBuffer )
{
OutputDebugString( "Pointer to vertex buffer is invalid (lost device and not recreated?)!\n" );
return false;
}
// Make sure its unlocked, if it isn't locked this will just fail quietly
if ( iter->second.m_bIsLocked )
iter->second.m_pBuffer->Unlock();
// Release the resources
iter->second.m_pBuffer->Release();
// Remove from the map
m_MapVertexBuffers.erase( iter );
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Set the FVF
//-----------------------------------------------------------------------------
bool CGameEngine::BSetFVF( DWORD dwFormat )
{
if ( !m_pD3D9Device )
return false;
// Can't set the FVF when the device is lost
if ( m_bDeviceLost )
return false;
// Short circuit if the request is a noop
if ( m_dwCurrentFVF == dwFormat )
return true;
if ( FAILED( m_pD3D9Device->SetFVF( dwFormat ) ) )
{
OutputDebugString( "SetFVF() call failed\n" );
return false;
}
m_dwCurrentFVF = dwFormat;
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Draw a line, the engine internally manages a vertex buffer for batching these
//-----------------------------------------------------------------------------
bool CGameEngine::BDrawLine( float xPos0, float yPos0, DWORD dwColor0, float xPos1, float yPos1, DWORD dwColor1 )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return true; // Fail silently in this case
if ( !m_hLineBuffer )
{
// Create the line buffer
m_hLineBuffer = HCreateVertexBuffer( sizeof( LineVertex_t ) * LINE_BUFFER_TOTAL_SIZE * 2, D3DUSAGE_DYNAMIC | D3DUSAGE_WRITEONLY, D3DFVF_XYZRHW | D3DFVF_DIFFUSE );
if ( !m_hLineBuffer )
{
OutputDebugString( "Can't BDrawLine() because vertex buffer creation failed\n" );
return false;
}
}
// Check if we are out of room and need to flush the buffer
if ( m_dwLinesToFlush == LINE_BUFFER_BATCH_SIZE )
{
BFlushLineBuffer();
}
// Set FVF
if ( !BSetFVF( D3DFVF_XYZRHW | D3DFVF_DIFFUSE ) )
return false;
// Lock the vertex buffer into memory
if ( !m_pLineVertexes )
{
if ( !BLockEntireVertexBuffer( m_hLineBuffer, (void**)&m_pLineVertexes, m_dwLineBufferBatchPos ? D3DLOCK_NOOVERWRITE : D3DLOCK_DISCARD ) )
{
m_pLineVertexes = NULL;
OutputDebugString( "BDrawLine failed because locking vertex buffer failed\n" );
return false;
}
}
LineVertex_t *pVertData = &m_pLineVertexes[ m_dwLineBufferBatchPos*2+m_dwLinesToFlush*2 ];
pVertData[0].rhw = 1.0;
pVertData[0].z = 1.0;
pVertData[0].x = xPos0;
pVertData[0].y = yPos0;
pVertData[0].color = dwColor0;
pVertData[1].rhw = 1.0;
pVertData[1].z = 1.0;
pVertData[1].x = xPos1;
pVertData[1].y = yPos1;
pVertData[1].color = dwColor1;
++m_dwLinesToFlush;
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Flush batched lines to the screen
//-----------------------------------------------------------------------------
bool CGameEngine::BFlushLineBuffer()
{
// If the vert buffer isn't already locked into memory, then there is nothing to flush
if ( m_pLineVertexes == NULL )
return true; // consider this successful since there was no error
// OK, it is locked, so unlock now
if ( !BUnlockVertexBuffer( m_hLineBuffer ) )
{
OutputDebugString( "Failed flushing line buffer because BUnlockVertexBuffer failed\n" );
return false;
}
// Clear the memory pointer as its invalid now that we unlocked
m_pLineVertexes = NULL;
// If there is nothing to actual flush, we are done
if ( m_dwLinesToFlush == 0 )
return true;
// Set FVF (will short circuit if this is already the set FVF)
if ( !BSetFVF( D3DFVF_XYZRHW | D3DFVF_DIFFUSE ) )
{
OutputDebugString( "Failed flushing line buffer because BSetFVF failed\n" );
return false;
}
if ( !BSetStreamSource( m_hLineBuffer, 0, sizeof( LineVertex_t ) ) )
{
OutputDebugString( "Failed flushing line buffer because BSetStreamSource failed\n" );
return false;
}
// Actual render calls
if ( !BRenderPrimitive( D3DPT_LINELIST, m_dwLineBufferBatchPos*2, m_dwLinesToFlush ) )
{
OutputDebugString( "Failed flushing line buffer because BRenderPrimitive failed\n" );
return false;
}
m_dwLinesToFlush = 0;
m_dwLineBufferBatchPos += LINE_BUFFER_BATCH_SIZE;
if ( m_dwLineBufferBatchPos >= LINE_BUFFER_TOTAL_SIZE )
{
m_dwLineBufferBatchPos = 0;
}
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Draw a point, the engine internally manages a vertex buffer for batching these
//-----------------------------------------------------------------------------
bool CGameEngine::BDrawPoint( float xPos, float yPos, DWORD dwColor )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return true; // Fail silently in this case
if ( !m_hPointBuffer )
{
// Create the point buffer
m_hPointBuffer = HCreateVertexBuffer( sizeof( PointVertex_t ) * POINT_BUFFER_TOTAL_SIZE * 2, D3DUSAGE_DYNAMIC | D3DUSAGE_WRITEONLY, D3DFVF_XYZRHW | D3DFVF_DIFFUSE );
if ( !m_hPointBuffer )
{
OutputDebugString( "Can't BDrawPoint() because vertex buffer creation failed\n" );
return false;
}
}
// Check if we are out of room and need to flush the buffer
if ( m_dwPointsToFlush == POINT_BUFFER_BATCH_SIZE )
{
BFlushPointBuffer();
}
// Set FVF
if ( !BSetFVF( D3DFVF_XYZRHW | D3DFVF_DIFFUSE ) )
return false;
// Lock the vertex buffer into memory
if ( !m_pPointVertexes )
{
if ( !BLockEntireVertexBuffer( m_hPointBuffer, (void**)&m_pPointVertexes, m_dwPointBufferBatchPos+m_dwPointsToFlush ? D3DLOCK_NOOVERWRITE : D3DLOCK_DISCARD ) )
{
m_pPointVertexes = NULL;
OutputDebugString( "BDrawPoint failed because locking vertex buffer failed\n" );
return false;
}
}
PointVertex_t *pVertData = &m_pPointVertexes[ m_dwPointBufferBatchPos+m_dwPointsToFlush ];
pVertData[0].rhw = 1.0;
pVertData[0].z = 1.0;
pVertData[0].x = xPos;
pVertData[0].y = yPos;
pVertData[0].color = dwColor;
++m_dwPointsToFlush;
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Flush batched points to the screen
//-----------------------------------------------------------------------------
bool CGameEngine::BFlushPointBuffer()
{
// If the vert buffer isn't already locked into memory, then there is nothing to flush
if ( m_pPointVertexes == NULL )
return true; // consider this successful since there was no error
// OK, it is locked, so unlock now
if ( !BUnlockVertexBuffer( m_hPointBuffer ) )
{
OutputDebugString( "Failed flushing point buffer because BUnlockVertexBuffer failed\n" );
return false;
}
// Clear the memory pointer as its invalid now that we unlocked
m_pPointVertexes = NULL;
// If there is nothing to actual flush, we are done
if ( m_dwPointsToFlush == 0 )
return true;
// Set FVF (will short circuit if this is already the set FVF)
if ( !BSetFVF( D3DFVF_XYZRHW | D3DFVF_DIFFUSE ) )
{
OutputDebugString( "Failed flushing point buffer because BSetFVF failed\n" );
return false;
}
if ( !BSetStreamSource( m_hPointBuffer, 0, sizeof( PointVertex_t ) ) )
{
OutputDebugString( "Failed flushing point buffer because BSetStreamSource failed\n" );
return false;
}
// Actual render calls
if ( !BRenderPrimitive( D3DPT_POINTLIST, m_dwPointBufferBatchPos, m_dwPointsToFlush ) )
{
OutputDebugString( "Failed flushing point buffer because BRenderPrimitive failed\n" );
return false;
}
m_dwPointsToFlush = 0;
m_dwPointBufferBatchPos += POINT_BUFFER_BATCH_SIZE;
if ( m_dwPointBufferBatchPos >= POINT_BUFFER_TOTAL_SIZE )
{
m_dwPointBufferBatchPos = 0;
}
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Draw a filled quad
//-----------------------------------------------------------------------------
bool CGameEngine::BDrawFilledQuad( float xPos0, float yPos0, float xPos1, float yPos1, DWORD dwColor )
{
if ( !m_hTextureWhite )
{
byte *pRGBAData = new byte[ 1 * 1 * 4 ];
memset( pRGBAData, 255, 1*1*4 );
m_hTextureWhite = HCreateTexture( pRGBAData, 1, 1 );
delete[] pRGBAData;
}
return BDrawTexturedQuad( xPos0, yPos0, xPos1, yPos1, 0.0f, 0.0f, 1.0f, 1.0f, dwColor, m_hTextureWhite );
}
//-----------------------------------------------------------------------------
// Purpose: Draw a textured quad
//-----------------------------------------------------------------------------
bool CGameEngine::BDrawTexturedQuad( float xPos0, float yPos0, float xPos1, float yPos1, float u0, float v0, float u1, float v1, DWORD dwColor, HGAMETEXTURE hTexture )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return true; // Fail silently in this case
// Find the texture
std::map<HGAMETEXTURE, TextureData_t>::iterator iter;
iter = m_MapTextures.find( hTexture );
if ( iter == m_MapTextures.end() )
{
OutputDebugString( "BDrawTexturedQuad called with invalid hTexture value\n" );
return false;
}
if ( !m_hQuadBuffer )
{
// Create the line buffer
m_hQuadBuffer = HCreateVertexBuffer( sizeof( TexturedQuadVertex_t ) * QUAD_BUFFER_TOTAL_SIZE * 4, D3DUSAGE_DYNAMIC | D3DUSAGE_WRITEONLY, D3DFVF_XYZRHW | D3DFVF_DIFFUSE | D3DFVF_TEX1 );
if ( !m_hQuadBuffer )
{
OutputDebugString( "Can't BDrawTexturedQuad() because vertex buffer creation failed\n" );
return false;
}
}
// Check if we are out of room and need to flush the buffer
if ( m_dwQuadsToFlush == QUAD_BUFFER_BATCH_SIZE )
{
BFlushQuadBuffer();
}
// Check if the texture changed so we need to flush the buffer
if ( m_hLastTexture != hTexture )
{
BFlushQuadBuffer();
}
// Save the texture to use for next flush
m_hLastTexture = hTexture;
if ( !BSetFVF( D3DFVF_XYZRHW | D3DFVF_DIFFUSE | D3DFVF_TEX1 ) )
{
OutputDebugString( "Setting FVF failed for textured rect drawing\n" );
return false;
}
// Lock the vertex buffer into memory
if ( !m_pQuadVertexes )
{
if ( !BLockEntireVertexBuffer( m_hQuadBuffer, (void**)&m_pQuadVertexes, m_dwQuadBufferBatchPos ? D3DLOCK_NOOVERWRITE : D3DLOCK_DISCARD ) )
{
m_pQuadVertexes = NULL;
OutputDebugString( "BDrawTexturedQuad failed because locking vertex buffer failed\n" );
return false;
}
}
TexturedQuadVertex_t *pVertData = &m_pQuadVertexes[ m_dwQuadBufferBatchPos*4+m_dwQuadsToFlush*4 ];
pVertData[0].color = dwColor;
pVertData[0].rhw = 1.0f;
pVertData[0].z = 1.0f;
pVertData[0].x = xPos0;
pVertData[0].y = yPos0;
pVertData[0].u = u0;
pVertData[0].v = v0;
pVertData[1].color = dwColor;
pVertData[1].rhw = 1.0f;
pVertData[1].z = 1.0f;
pVertData[1].x = xPos1;
pVertData[1].y = yPos0;
pVertData[1].u = u1;
pVertData[1].v = v0;
pVertData[2].color = dwColor;
pVertData[2].rhw = 1.0f;
pVertData[2].z = 1.0f;
pVertData[2].x = xPos0;
pVertData[2].y = yPos1;
pVertData[2].u = u0;
pVertData[2].v = v1;
pVertData[3].color = dwColor;
pVertData[3].rhw = 1.0f;
pVertData[3].z = 1.0f;
pVertData[3].x = xPos1;
pVertData[3].y = yPos1;
pVertData[3].u = u1;
pVertData[3].v = v1;
++m_dwQuadsToFlush;
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Flush buffered quads
//-----------------------------------------------------------------------------
bool CGameEngine::BFlushQuadBuffer()
{
// If the vert buffer isn't already locked into memory, then there is nothing to flush
if ( m_pQuadVertexes == NULL )
return true; // consider this successful since there was no error
// OK, it is locked, so unlock now
if ( !BUnlockVertexBuffer( m_hQuadBuffer ) )
{
OutputDebugString( "Failed flushing quad buffer because BUnlockVertexBuffer failed\n" );
return false;
}
// Clear the memory pointer as its invalid now that we unlocked
m_pQuadVertexes = NULL;
// If there is nothing to actual flush, we are done
if ( m_dwQuadsToFlush == 0 )
return true;
// Set FVF (will short circuit if this is already the set FVF)
if ( !BSetFVF( D3DFVF_XYZRHW | D3DFVF_DIFFUSE | D3DFVF_TEX1 ) )
{
OutputDebugString( "Failed flushing quad buffer because BSetFVF failed\n" );
return false;
}
// Find the texture
std::map<HGAMETEXTURE, TextureData_t>::iterator iter;
iter = m_MapTextures.find( m_hLastTexture );
if ( iter == m_MapTextures.end() )
{
OutputDebugString( "BFlushQuadBuffer failed with invalid m_hLastTexture value\n" );
return false;
}
// See if we need to actually create the d3d texture
if ( !iter->second.m_pTexture )
{
HRESULT hRes = m_pD3D9Device->CreateTexture(
iter->second.m_uWidth,
iter->second.m_uHeight,
1, // mip map levels (0 = generate all levels down to 1x1 if supported)
0, // have to set the right flag here if you want to autogen mipmaps... we don't
D3DFMT_A8R8G8B8,
D3DPOOL_MANAGED,
&iter->second.m_pTexture,
NULL );
if( FAILED( hRes ) )
{
OutputDebugString( "BFlushQuadBuffer failed because CreateTexture() call failed\n" );
iter->second.m_pTexture = NULL;
return false;
}
// Put the data into the texture
D3DLOCKED_RECT rect;
hRes = iter->second.m_pTexture->LockRect( 0, &rect, NULL, 0 );
if( FAILED( hRes ) )
{
OutputDebugString( "LockRect call failed\n" );
iter->second.m_pTexture->Release();
iter->second.m_pTexture = NULL;
return false;
}
DWORD *pARGB = (DWORD *) rect.pBits;
byte *pRGBA = (byte *) iter->second.m_pRGBAData;
byte r,g,b,a;
for( uint32 y = 0; y < iter->second.m_uHeight; ++y )
{
for( uint32 x = 0; x < iter->second.m_uWidth; ++x )
{
// swap position of alpha value from back to front to be in correct format for d3d...
r = *pRGBA++;
g = *pRGBA++;
b = *pRGBA++;
a = *pRGBA++;
*pARGB++ = D3DCOLOR_ARGB( a, r, g, b );
}
}
hRes = iter->second.m_pTexture->UnlockRect( 0 );
if( FAILED( hRes ) )
{
OutputDebugString( "UnlockRect call failed\n" );
iter->second.m_pTexture->Release();
iter->second.m_pTexture = NULL;
return false;
}
}
// Ok, texture should be created ok, do the drawing work
if ( FAILED( m_pD3D9Device->SetTexture( 0, iter->second.m_pTexture ) ) )
{
OutputDebugString( "BFlushQuadBuffer failed setting texture\n" );
return false;
}
if ( !BSetStreamSource( m_hQuadBuffer, 0, sizeof( TexturedQuadVertex_t ) ) )
{
OutputDebugString( "Failed flushing quad buffer because BSetStreamSource failed\n" );
m_pD3D9Device->SetTexture( 0, NULL ); // need to clear the texture before other drawing ops
return false;
}
// Actual render calls
for ( DWORD i=0; i < m_dwQuadsToFlush*4; i += 4 )
{
if ( !BRenderPrimitive( D3DPT_TRIANGLESTRIP, (m_dwQuadBufferBatchPos*4)+i, 2 ) )
{
OutputDebugString( "Failed flushing line buffer because BRenderPrimitive failed\n" );
m_pD3D9Device->SetTexture( 0, NULL ); // need to clear the texture before other drawing ops
return false;
}
}
m_pD3D9Device->SetTexture( 0, NULL ); // need to clear the texture before other drawing ops
m_dwQuadsToFlush = 0;
m_dwQuadBufferBatchPos += QUAD_BUFFER_BATCH_SIZE;
if ( m_dwQuadBufferBatchPos >= QUAD_BUFFER_TOTAL_SIZE )
{
m_dwQuadBufferBatchPos = 0;
}
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Set the current stream source (this always set stream 0, we don't support more than 1 stream presently)
//-----------------------------------------------------------------------------
bool CGameEngine::BSetStreamSource( HGAMEVERTBUF hVertBuf, uint32 uOffset, uint32 uStride )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return false;
if ( !hVertBuf )
{
OutputDebugString( "Someone is calling BSetStreamSource() with a null handle\n" );
return false;
}
// Find the vertex buffer for the passed handle
std::map<HGAMEVERTBUF, VertBufData_t>::iterator iter;
iter = m_MapVertexBuffers.find( hVertBuf );
if ( iter == m_MapVertexBuffers.end() )
{
OutputDebugString( "Invalid vertex buffer handle passed to BSetStreamSource()\n" );
return false;
}
// Make sure the pointer is valid
if ( !iter->second.m_pBuffer )
{
OutputDebugString( "Pointer to vertex buffer is invalid (lost device and not recreated?)!\n" );
return false;
}
if ( FAILED( m_pD3D9Device->SetStreamSource( 0, iter->second.m_pBuffer, uOffset, uStride ) ) )
{
OutputDebugString( "SetStreamSource() call failed\n" );
return false;
}
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Renders primitives using the current stream source
//-----------------------------------------------------------------------------
bool CGameEngine::BRenderPrimitive( D3DPRIMITIVETYPE primType, uint32 uStartVertex, uint32 uCount )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return true; // Fail silently in this case
if ( FAILED( m_pD3D9Device->DrawPrimitive( primType, uStartVertex, uCount ) ) )
{
OutputDebugString( "BRenderPrimtive() call failed\n" );
return false;
}
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Creates a new texture
//-----------------------------------------------------------------------------
HGAMETEXTURE CGameEngine::HCreateTexture( byte *pRGBAData, uint32 uWidth, uint32 uHeight )
{
if ( !m_pD3D9Device )
return 0;
TextureData_t TexData;
TexData.m_uWidth = uWidth;
TexData.m_uHeight = uHeight;
TexData.m_pRGBAData = new byte[uWidth*uHeight*4];
memcpy( TexData.m_pRGBAData, pRGBAData, uWidth*uHeight*4 );
TexData.m_pTexture = NULL;
int nHandle = m_nNextTextureHandle;
++m_nNextTextureHandle;
m_MapTextures[nHandle] = TexData;
return nHandle;
}
//-----------------------------------------------------------------------------
// Purpose: Creates a new font
//-----------------------------------------------------------------------------
HGAMEFONT CGameEngine::HCreateFont( int nHeight, int nFontWeight, bool bItalic, char * pchFont )
{
if ( !m_pD3D9Device )
return 0;
// Create a D3DX font object
LPD3DXFONT pFont;
HRESULT hRes = D3DXCreateFont( m_pD3D9Device, nHeight, 0, nFontWeight, 0, bItalic, DEFAULT_CHARSET, OUT_DEFAULT_PRECIS, DEFAULT_QUALITY, DEFAULT_PITCH | FF_DONTCARE, pchFont, &pFont );
if ( FAILED( hRes ) )
{
OutputDebugString( "Failed creating font via D3DXCreateFont\n" );
return 0;
}
HGAMEFONT hFont = m_nNextFontHandle;
++m_nNextFontHandle;
m_MapFontInstances[ hFont ] = pFont;
return hFont;
}
//-----------------------------------------------------------------------------
// Purpose: Draws text to the screen inside the given rectangular region, using the given font
//-----------------------------------------------------------------------------
bool CGameEngine::BDrawString( HGAMEFONT hFont, RECT rect, DWORD dwColor, DWORD dwFormat, const char *pchText )
{
if ( !m_pD3D9Device )
return false;
if ( m_bDeviceLost )
return true; // Fail silently in this case
if ( !hFont )
{
OutputDebugString( "Someone is calling BDrawString with a null font handle\n" );
return false;
}
// Find the font object for the passed handle
std::map<HGAMEFONT, LPD3DXFONT>::iterator iter;
iter = m_MapFontInstances.find( hFont );
if ( iter == m_MapFontInstances.end() )
{
OutputDebugString( "Invalid font handle passed to BDrawString()\n" );
return false;
}
// we have the font, try to draw with it
if( !iter->second->DrawText( NULL, pchText, -1, &rect, dwFormat, dwColor ) )
{
OutputDebugString( "DrawText call failed\n" );
return false;
}
return true;
}
//-----------------------------------------------------------------------------
// Purpose: Message pump for OS messages
//-----------------------------------------------------------------------------
void CGameEngine::MessagePump()
{
MSG msg;
BOOL bRet;
while( PeekMessage( &msg, m_hWnd, 0, 0, PM_NOREMOVE ) )
{
bRet = GetMessage( &msg, m_hWnd, 0, 0 );
if( bRet != 0 && bRet != -1 )
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}
}
//-----------------------------------------------------------------------------
// Purpose: Track keys which are down
//-----------------------------------------------------------------------------
void CGameEngine::RecordKeyDown( DWORD dwVK )
{
m_SetKeysDown.insert( dwVK );
}
//-----------------------------------------------------------------------------
// Purpose: Track keys which are up
//-----------------------------------------------------------------------------
void CGameEngine::RecordKeyUp( DWORD dwVK )
{
std::set<DWORD>::iterator iter;
iter = m_SetKeysDown.find( dwVK );
if ( iter != m_SetKeysDown.end() )
m_SetKeysDown.erase( iter );
}
//-----------------------------------------------------------------------------
// Purpose: Find out if a key is currently down
//-----------------------------------------------------------------------------
bool CGameEngine::BIsKeyDown( DWORD dwVK )
{
std::set<DWORD>::iterator iter;
iter = m_SetKeysDown.find( dwVK );
if ( iter != m_SetKeysDown.end() )
return true;
return false;
}
//-----------------------------------------------------------------------------
// Purpose: Find the engine instance tied to a given hwnd
//-----------------------------------------------------------------------------
CGameEngine * CGameEngine::FindEngineInstanceForHWND( HWND hWnd )
{
std::map<HWND, CGameEngine *>::iterator iter;
iter = m_MapEngineInstances.find( hWnd );
if ( iter == m_MapEngineInstances.end() )
return NULL;
else
return iter->second;
}
//-----------------------------------------------------------------------------
// Purpose: Add the engine instance tied to a given hwnd to our static map
//-----------------------------------------------------------------------------
void CGameEngine::AddInstanceToHWNDMap( CGameEngine* pInstance, HWND hWnd )
{
m_MapEngineInstances[hWnd] = pInstance;
}
//-----------------------------------------------------------------------------
// Purpose: Removes the instance associated with a given HWND from the map
//-----------------------------------------------------------------------------
void CGameEngine::RemoveInstanceFromHWNDMap( HWND hWnd )
{
std::map<HWND, CGameEngine *>::iterator iter;
iter = m_MapEngineInstances.find( hWnd );
if ( iter != m_MapEngineInstances.end() )
m_MapEngineInstances.erase( iter );
}