example.cpp

3d 游戏 入门教程之例子源码-图像渲染

//-----------------------------------------------------------------------------
// File: Example.cpp
//
// Desc: Example code shows how to use a HLSL shader in an effect file
//
// Last modification: February 2nd, 2003
//
// Copyright (c) Wolfgang F. Engel ([e-mail])
// Copyright (c) Microsoft Corporation. All rights reserved.
//-----------------------------------------------------------------------------
#define STRICT
#include <Windows.h>
#include <commctrl.h>
#include <stdio.h>
#include <D3DX9.h>
#include "DXUtil.h"
#include "D3DEnumeration.h"
#include "D3DSettings.h"
#include "D3DApp.h"
#include "D3DFont.h"
#include "D3DUtil.h"
#include "resource.h"

#include "Trackball.h"

//-----------------------------------------------------------------------------
// Name: class CMyD3DApplication
// Desc: Application class. The base class (CD3DApplication) provides the 
//       generic functionality needed in all Direct3D samples. CMyD3DApplication 
//       adds functionality specific to this sample program.
//-----------------------------------------------------------------------------
class CMyD3DApplication : public CD3DApplication
{
    CD3DFont* m_pFont;
	CD3DFont* m_pFontSmall;

	LPD3DXEFFECT     m_pEffect;    

	ID3DXMesh*  m_pD3DXMesh;            // D3DX mesh to store teapot
	D3DXMATRIX  m_matWorld;
	D3DXMATRIX  m_matView;
	D3DXMATRIX  m_matProj;

	D3DXVECTOR4 m_LightDir;

	BYTE  m_bKey[256];

	LPDIRECT3DTEXTURE9		m_pBumpMap;
	LPDIRECT3DTEXTURE9		m_pColorMap;

	TCHAR m_cColorMap[64];
	TCHAR m_cBumpMap[64];
	TCHAR m_cCubeNormalizationMap[64];


	trackball_t m_pTrackball;	// trackball to turn the model
	bool m_bZoomDrag;			// zooming
	float m_fZoom;
	FLOAT m_fTheta;
	FLOAT m_fTheta2;
	BOOL m_bLMouseDown;
	BOOL m_bRMouseDown;
	FLOAT m_xPos, m_yPos;
	FLOAT m_oPosX, m_oPosY;

	TCHAR m_strStats[1024];	// help string

protected:
    HRESULT OneTimeSceneInit();
    HRESULT InitDeviceObjects();
    HRESULT RestoreDeviceObjects();
    HRESULT InvalidateDeviceObjects();
    HRESULT DeleteDeviceObjects();
    HRESULT FinalCleanup();
    HRESULT Render();
    HRESULT FrameMove();
    HRESULT ConfirmDevice( D3DCAPS9* pCaps, DWORD dwBehavior, 
		D3DFORMAT adapterFormat, D3DFORMAT backBufferFormat );
	LRESULT MsgProc( HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam );

	HRESULT LoadXFile(TCHAR*);


public:
    CMyD3DApplication();
	~CMyD3DApplication();
};

//-----------------------------------------------------------------------------
// Name: WinMain()
// Desc: Entry point to the program. Initializes everything, and goes into a
//       message-processing loop. Idle time is used to render the scene.
//-----------------------------------------------------------------------------
INT WINAPI WinMain( HINSTANCE hInst, HINSTANCE, LPSTR, INT )
{
    CMyD3DApplication d3dApp;

    InitCommonControls();
    if( FAILED( d3dApp.Create( hInst ) ) )
        return 0;

    return d3dApp.Run();
}

//-----------------------------------------------------------------------------
// Name: CMyD3DApplication()
// Desc: Application constructor. Sets attributes for the app.
//-----------------------------------------------------------------------------
CMyD3DApplication::CMyD3DApplication()
{
    m_strWindowTitle  = _T("High-Level-Shading Language");
    m_d3dEnumeration.AppUsesDepthBuffer = TRUE;
	m_dwCreationWidth		= 800;   // Width used to create window
	m_dwCreationHeight		= 600;
	m_bShowCursorWhenFullscreen = TRUE;
	m_pFont					= new CD3DFont( _T("Arial"), 12, D3DFONT_BOLD );
	m_pFontSmall			= new CD3DFont( _T("Arial"),  9, D3DFONT_BOLD );
	memset (m_bKey, 0, sizeof (m_bKey));
	m_bLMouseDown			= FALSE;
	m_bRMouseDown			= FALSE;
	m_bZoomDrag				= false;
	m_fZoom					= -50.0f;			// initial zoom
	m_fTheta				= 0;
	m_fTheta2				= 0;
	m_LightDir				= D3DXVECTOR4( 0.0f, 0.0f, 1.0f, 1.0f );
	D3DXMatrixIdentity(&m_matWorld);
	m_pEffect				= NULL;
	m_pD3DXMesh             = NULL;
}

//-----------------------------------------------------------------------------
// Name: CMyD3DApplication()
// Desc: Application denstructor
//-----------------------------------------------------------------------------
CMyD3DApplication::~CMyD3DApplication()
{
	SAFE_DELETE(m_pFont);
	SAFE_DELETE(m_pFontSmall);
}

//-----------------------------------------------------------------------------
// Name: OneTimeSceneInit()
// Desc: Called during initial app startup, this function performs all the
//       permanent initialization.
//-----------------------------------------------------------------------------
HRESULT CMyD3DApplication::OneTimeSceneInit()
{
	strcpy(m_cBumpMap, _T("media/normal.tga"));
	strcpy(m_cColorMap,_T("media/earth.tga"));

    return S_OK;
}

//-----------------------------------------------------------------------------
// Name: FrameMove()
// Desc: Called once per frame, the call is the entry point for animating
//       the scene.
//-----------------------------------------------------------------------------
HRESULT CMyD3DApplication::FrameMove()
{
	m_pTrackball.TrackMouse(m_hWnd, m_bZoomDrag, &m_fZoom); // track mouse movements and move object
	m_pTrackball.SetMatrix(m_matWorld); // get the world matrix for the trackball

	// setup view matrix
	D3DXVECTOR3 vUpVec(0,1,0);
	D3DXVECTOR3 vEyePt(0,0,m_fZoom);
	D3DXVECTOR3 vLookatPt(0,0,0);
	D3DXMatrixLookAtLH(&m_matView, &vEyePt, &vLookatPt, &vUpVec);

	if(m_bKey['W']) m_fZoom += 10.0f * m_fElapsedTime; // Move Forward
	if(m_bKey['S']) m_fZoom -= 10.0f * m_fElapsedTime; // Move Backward

	// move light
	if(m_bKey[VK_UP])	m_LightDir.y -= 1.5f  * m_fElapsedTime;
	if(m_bKey[VK_DOWN]) m_LightDir.y += 1.5f  * m_fElapsedTime;
	if(m_bKey[VK_LEFT])	m_LightDir.x += 1.5f  * m_fElapsedTime;
	if(m_bKey[VK_RIGHT]) m_LightDir.x -= 1.5f  * m_fElapsedTime;
	if(m_bKey[VK_HOME])	m_LightDir.z += 1.0f  * m_fElapsedTime;
	if(m_bKey[VK_END]) m_LightDir.z -= 1.0f  * m_fElapsedTime;
	if (m_LightDir.z >= 2.7f) m_LightDir.z = 2.7f; 
	if (m_LightDir.z <= -5.7f) m_LightDir.z = -5.7f; 

	// set light direction
	m_pEffect->SetVector("vecLightDir", &-m_LightDir);
	m_pEffect->SetVector("vecEye", &D3DXVECTOR4(vEyePt.x,vEyePt.y,vEyePt.z,0));

	D3DXMATRIX mWorldViewProj = m_matWorld * m_matView * m_matProj;
	m_pEffect->SetMatrix( "matWorldViewProj", &mWorldViewProj );
	m_pEffect->SetMatrix( "matWorld", &m_matWorld);

    return S_OK;
}

//-----------------------------------------------------------------------------
// Name: Render()
// Desc: Called once per frame, the call is the entry point for 3d
//       rendering. This function sets up render states, clears the
//       viewport, and renders the scene.
//-----------------------------------------------------------------------------
HRESULT CMyD3DApplication::Render()
{	
		m_pd3dDevice->Clear(0, NULL, D3DCLEAR_TARGET | D3DCLEAR_ZBUFFER, 
							D3DCOLOR_ARGB( 0x00, 0x00, 0x00, 0x00), 1.0, 0);
        UINT nPasses;
        UINT iPass;

		// Begin the scene
		if( SUCCEEDED( m_pd3dDevice->BeginScene() ))
		{
			if( m_pEffect != NULL )
			{
				D3DXHANDLE hTechnique = m_pEffect->GetTechniqueByName( "TShader" );
				m_pEffect->SetTechnique( hTechnique );

				m_pEffect->SetTexture("ColorMap", m_pColorMap);
				m_pEffect->SetTexture("BumpMap", m_pBumpMap);

				m_pEffect->Begin( &nPasses, 0 );

				for( iPass = 0; iPass < nPasses; iPass ++ )
				{
					m_pEffect->Pass( iPass );
					m_pD3DXMesh->DrawSubset( 0 );
				}
				m_pEffect->End();
			}

			// Output statistics
			m_pFont->DrawText( 2,  0, D3DCOLOR_ARGB(255,255,255,0), m_strFrameStats );
			m_pFont->DrawText( 2, 20, D3DCOLOR_ARGB(255,255,255,0), m_strDeviceStats );
			sprintf(m_strStats, 
				"W -     Zoom In \
				\nS -     Zoom Out \
				\nHome/End -     Moves Light -z/+z\
				\nUp/Down -     Moves Light -y/+y\
				\nLeft/Right -     Moves Light -x/+x");
			m_pFontSmall->DrawText( 2, 40, D3DCOLOR_ARGB(255,255,255,255), m_strStats);