1 | //--------------------------------------------------------------------------------------
|
---|
2 | // File: DXUTMisc.h
|
---|
3 | //
|
---|
4 | // Helper functions for Direct3D programming.
|
---|
5 | //
|
---|
6 | // Copyright (c) Microsoft Corporation. All rights reserved
|
---|
7 | //--------------------------------------------------------------------------------------
|
---|
8 | #pragma once
|
---|
9 | #ifndef DXUT_MISC_H
|
---|
10 | #define DXUT_MISC_H
|
---|
11 |
|
---|
12 |
|
---|
13 | //--------------------------------------------------------------------------------------
|
---|
14 | // A growable array
|
---|
15 | //--------------------------------------------------------------------------------------
|
---|
16 | template< typename TYPE >
|
---|
17 | class CGrowableArray
|
---|
18 | {
|
---|
19 | public:
|
---|
20 | CGrowableArray() { m_pData = NULL; m_nSize = 0; m_nMaxSize = 0; }
|
---|
21 | CGrowableArray( const CGrowableArray<TYPE>& a ) { for( int i=0; i < a.m_nSize; i++ ) Add( a.m_pData[i] ); }
|
---|
22 | ~CGrowableArray() { RemoveAll(); }
|
---|
23 |
|
---|
24 | const TYPE& operator[]( int nIndex ) const { return GetAt( nIndex ); }
|
---|
25 | TYPE& operator[]( int nIndex ) { return GetAt( nIndex ); }
|
---|
26 |
|
---|
27 | CGrowableArray& operator=( const CGrowableArray<TYPE>& a ) { if( this == &a ) return *this; RemoveAll(); for( int i=0; i < a.m_nSize; i++ ) Add( a.m_pData[i] ); return *this; }
|
---|
28 |
|
---|
29 | HRESULT SetSize( int nNewMaxSize );
|
---|
30 | HRESULT Add( const TYPE& value );
|
---|
31 | HRESULT Insert( int nIndex, const TYPE& value );
|
---|
32 | HRESULT SetAt( int nIndex, const TYPE& value );
|
---|
33 | TYPE& GetAt( int nIndex ) { assert( nIndex >= 0 && nIndex < m_nSize ); return m_pData[nIndex]; }
|
---|
34 | int GetSize() const { return m_nSize; }
|
---|
35 | TYPE* GetData() { return m_pData; }
|
---|
36 | bool Contains( const TYPE& value ){ return ( -1 != IndexOf( value ) ); }
|
---|
37 |
|
---|
38 | int IndexOf( const TYPE& value ) { return ( m_nSize > 0 ) ? IndexOf( value, 0, m_nSize ) : -1; }
|
---|
39 | int IndexOf( const TYPE& value, int iStart ) { return IndexOf( value, iStart, m_nSize - iStart ); }
|
---|
40 | int IndexOf( const TYPE& value, int nIndex, int nNumElements );
|
---|
41 |
|
---|
42 | int LastIndexOf( const TYPE& value ) { return ( m_nSize > 0 ) ? LastIndexOf( value, m_nSize-1, m_nSize ) : -1; }
|
---|
43 | int LastIndexOf( const TYPE& value, int nIndex ) { return LastIndexOf( value, nIndex, nIndex+1 ); }
|
---|
44 | int LastIndexOf( const TYPE& value, int nIndex, int nNumElements );
|
---|
45 |
|
---|
46 | HRESULT Remove( int nIndex );
|
---|
47 | void RemoveAll() { SetSize(0); }
|
---|
48 |
|
---|
49 | protected:
|
---|
50 | TYPE* m_pData; // the actual array of data
|
---|
51 | int m_nSize; // # of elements (upperBound - 1)
|
---|
52 | int m_nMaxSize; // max allocated
|
---|
53 |
|
---|
54 | HRESULT SetSizeInternal( int nNewMaxSize ); // This version doesn't call ctor or dtor.
|
---|
55 | };
|
---|
56 |
|
---|
57 |
|
---|
58 | //--------------------------------------------------------------------------------------
|
---|
59 | // Performs timer operations
|
---|
60 | // Use DXUTGetGlobalTimer() to get the global instance
|
---|
61 | //--------------------------------------------------------------------------------------
|
---|
62 | class CDXUTTimer
|
---|
63 | {
|
---|
64 | public:
|
---|
65 | CDXUTTimer();
|
---|
66 |
|
---|
67 | void Reset(); // resets the timer
|
---|
68 | void Start(); // starts the timer
|
---|
69 | void Stop(); // stop (or pause) the timer
|
---|
70 | void Advance(); // advance the timer by 0.1 seconds
|
---|
71 | double GetAbsoluteTime(); // get the absolute system time
|
---|
72 | double GetTime(); // get the current time
|
---|
73 | double GetElapsedTime(); // get the time that elapsed between GetElapsedTime() calls
|
---|
74 | bool IsStopped(); // returns true if timer stopped
|
---|
75 |
|
---|
76 | protected:
|
---|
77 | bool m_bUsingQPF;
|
---|
78 | bool m_bTimerStopped;
|
---|
79 | LONGLONG m_llQPFTicksPerSec;
|
---|
80 |
|
---|
81 | LONGLONG m_llStopTime;
|
---|
82 | LONGLONG m_llLastElapsedTime;
|
---|
83 | LONGLONG m_llBaseTime;
|
---|
84 | };
|
---|
85 |
|
---|
86 | CDXUTTimer* DXUTGetGlobalTimer();
|
---|
87 |
|
---|
88 |
|
---|
89 | //-----------------------------------------------------------------------------
|
---|
90 | // Resource cache for textures, fonts, meshs, and effects.
|
---|
91 | // Use DXUTGetGlobalResourceCache() to access the global cache
|
---|
92 | //-----------------------------------------------------------------------------
|
---|
93 |
|
---|
94 | enum DXUTCACHE_SOURCELOCATION { DXUTCACHE_LOCATION_FILE, DXUTCACHE_LOCATION_RESOURCE };
|
---|
95 |
|
---|
96 | struct DXUTCache_Texture
|
---|
97 | {
|
---|
98 | DXUTCACHE_SOURCELOCATION Location;
|
---|
99 | WCHAR wszSource[MAX_PATH];
|
---|
100 | HMODULE hSrcModule;
|
---|
101 | UINT Width;
|
---|
102 | UINT Height;
|
---|
103 | UINT Depth;
|
---|
104 | UINT MipLevels;
|
---|
105 | DWORD Usage;
|
---|
106 | D3DFORMAT Format;
|
---|
107 | D3DPOOL Pool;
|
---|
108 | D3DRESOURCETYPE Type;
|
---|
109 | IDirect3DBaseTexture9 *pTexture;
|
---|
110 | };
|
---|
111 |
|
---|
112 | struct DXUTCache_Font : public D3DXFONT_DESC
|
---|
113 | {
|
---|
114 | ID3DXFont *pFont;
|
---|
115 | };
|
---|
116 |
|
---|
117 | struct DXUTCache_Effect
|
---|
118 | {
|
---|
119 | DXUTCACHE_SOURCELOCATION Location;
|
---|
120 | WCHAR wszSource[MAX_PATH];
|
---|
121 | HMODULE hSrcModule;
|
---|
122 | DWORD dwFlags;
|
---|
123 | ID3DXEffect *pEffect;
|
---|
124 | };
|
---|
125 |
|
---|
126 |
|
---|
127 | class CDXUTResourceCache
|
---|
128 | {
|
---|
129 | public:
|
---|
130 | ~CDXUTResourceCache();
|
---|
131 |
|
---|
132 | HRESULT CreateTextureFromFile( LPDIRECT3DDEVICE9 pDevice, LPCTSTR pSrcFile, LPDIRECT3DTEXTURE9 *ppTexture );
|
---|
133 | HRESULT CreateTextureFromFileEx( LPDIRECT3DDEVICE9 pDevice, LPCTSTR pSrcFile, UINT Width, UINT Height, UINT MipLevels, DWORD Usage, D3DFORMAT Format, D3DPOOL Pool, DWORD Filter, DWORD MipFilter, D3DCOLOR ColorKey, D3DXIMAGE_INFO *pSrcInfo, PALETTEENTRY *pPalette, LPDIRECT3DTEXTURE9 *ppTexture );
|
---|
134 | HRESULT CreateTextureFromResource( LPDIRECT3DDEVICE9 pDevice, HMODULE hSrcModule, LPCTSTR pSrcResource, LPDIRECT3DTEXTURE9 *ppTexture );
|
---|
135 | HRESULT CreateTextureFromResourceEx( LPDIRECT3DDEVICE9 pDevice, HMODULE hSrcModule, LPCTSTR pSrcResource, UINT Width, UINT Height, UINT MipLevels, DWORD Usage, D3DFORMAT Format, D3DPOOL Pool, DWORD Filter, DWORD MipFilter, D3DCOLOR ColorKey, D3DXIMAGE_INFO *pSrcInfo, PALETTEENTRY *pPalette, LPDIRECT3DTEXTURE9 *ppTexture );
|
---|
136 | HRESULT CreateCubeTextureFromFile( LPDIRECT3DDEVICE9 pDevice, LPCTSTR pSrcFile, LPDIRECT3DCUBETEXTURE9 *ppCubeTexture );
|
---|
137 | HRESULT CreateCubeTextureFromFileEx( LPDIRECT3DDEVICE9 pDevice, LPCTSTR pSrcFile, UINT Size, UINT MipLevels, DWORD Usage, D3DFORMAT Format, D3DPOOL Pool, DWORD Filter, DWORD MipFilter, D3DCOLOR ColorKey, D3DXIMAGE_INFO *pSrcInfo, PALETTEENTRY *pPalette, LPDIRECT3DCUBETEXTURE9 *ppCubeTexture );
|
---|
138 | HRESULT CreateCubeTextureFromResource( LPDIRECT3DDEVICE9 pDevice, HMODULE hSrcModule, LPCTSTR pSrcResource, LPDIRECT3DCUBETEXTURE9 *ppCubeTexture );
|
---|
139 | HRESULT CreateCubeTextureFromResourceEx( LPDIRECT3DDEVICE9 pDevice, HMODULE hSrcModule, LPCTSTR pSrcResource, UINT Size, UINT MipLevels, DWORD Usage, D3DFORMAT Format, D3DPOOL Pool, DWORD Filter, DWORD MipFilter, D3DCOLOR ColorKey, D3DXIMAGE_INFO *pSrcInfo, PALETTEENTRY *pPalette, LPDIRECT3DCUBETEXTURE9 *ppCubeTexture );
|
---|
140 | HRESULT CreateVolumeTextureFromFile( LPDIRECT3DDEVICE9 pDevice, LPCTSTR pSrcFile, LPDIRECT3DVOLUMETEXTURE9 *ppVolumeTexture );
|
---|
141 | HRESULT CreateVolumeTextureFromFileEx( LPDIRECT3DDEVICE9 pDevice, LPCTSTR pSrcFile, UINT Width, UINT Height, UINT Depth, UINT MipLevels, DWORD Usage, D3DFORMAT Format, D3DPOOL Pool, DWORD Filter, DWORD MipFilter, D3DCOLOR ColorKey, D3DXIMAGE_INFO *pSrcInfo, PALETTEENTRY *pPalette, LPDIRECT3DVOLUMETEXTURE9 *ppTexture );
|
---|
142 | HRESULT CreateVolumeTextureFromResource( LPDIRECT3DDEVICE9 pDevice, HMODULE hSrcModule, LPCTSTR pSrcResource, LPDIRECT3DVOLUMETEXTURE9 *ppVolumeTexture );
|
---|
143 | HRESULT CreateVolumeTextureFromResourceEx( LPDIRECT3DDEVICE9 pDevice, HMODULE hSrcModule, LPCTSTR pSrcResource, UINT Width, UINT Height, UINT Depth, UINT MipLevels, DWORD Usage, D3DFORMAT Format, D3DPOOL Pool, DWORD Filter, DWORD MipFilter, D3DCOLOR ColorKey, D3DXIMAGE_INFO *pSrcInfo, PALETTEENTRY *pPalette, LPDIRECT3DVOLUMETEXTURE9 *ppVolumeTexture );
|
---|
144 | HRESULT CreateFont( LPDIRECT3DDEVICE9 pDevice, UINT Height, UINT Width, UINT Weight, UINT MipLevels, BOOL Italic, DWORD CharSet, DWORD OutputPrecision, DWORD Quality, DWORD PitchAndFamily, LPCTSTR pFacename, LPD3DXFONT *ppFont );
|
---|
145 | HRESULT CreateFontIndirect( LPDIRECT3DDEVICE9 pDevice, CONST D3DXFONT_DESC *pDesc, LPD3DXFONT *ppFont );
|
---|
146 | HRESULT CreateEffectFromFile( LPDIRECT3DDEVICE9 pDevice, LPCTSTR pSrcFile, const D3DXMACRO *pDefines, LPD3DXINCLUDE pInclude, DWORD Flags, LPD3DXEFFECTPOOL pPool, LPD3DXEFFECT *ppEffect, LPD3DXBUFFER *ppCompilationErrors );
|
---|
147 | HRESULT CreateEffectFromResource( LPDIRECT3DDEVICE9 pDevice, HMODULE hSrcModule, LPCTSTR pSrcResource, const D3DXMACRO *pDefines, LPD3DXINCLUDE pInclude, DWORD Flags, LPD3DXEFFECTPOOL pPool, LPD3DXEFFECT *ppEffect, LPD3DXBUFFER *ppCompilationErrors );
|
---|
148 |
|
---|
149 | public:
|
---|
150 | HRESULT OnCreateDevice( IDirect3DDevice9 *pd3dDevice );
|
---|
151 | HRESULT OnResetDevice( IDirect3DDevice9 *pd3dDevice );
|
---|
152 | HRESULT OnLostDevice();
|
---|
153 | HRESULT OnDestroyDevice();
|
---|
154 |
|
---|
155 | protected:
|
---|
156 | friend CDXUTResourceCache& DXUTGetGlobalResourceCache();
|
---|
157 | friend HRESULT DXUTInitialize3DEnvironment();
|
---|
158 | friend HRESULT DXUTReset3DEnvironment();
|
---|
159 | friend void DXUTCleanup3DEnvironment( bool bReleaseSettings );
|
---|
160 |
|
---|
161 | CDXUTResourceCache() { }
|
---|
162 |
|
---|
163 | CGrowableArray< DXUTCache_Texture > m_TextureCache;
|
---|
164 | CGrowableArray< DXUTCache_Effect > m_EffectCache;
|
---|
165 | CGrowableArray< DXUTCache_Font > m_FontCache;
|
---|
166 | };
|
---|
167 |
|
---|
168 | CDXUTResourceCache& DXUTGetGlobalResourceCache();
|
---|
169 |
|
---|
170 |
|
---|
171 | //--------------------------------------------------------------------------------------
|
---|
172 | class CD3DArcBall
|
---|
173 | {
|
---|
174 | public:
|
---|
175 | CD3DArcBall();
|
---|
176 |
|
---|
177 | // Functions to change behavior
|
---|
178 | void Reset();
|
---|
179 | void SetTranslationRadius( FLOAT fRadiusTranslation ) { m_fRadiusTranslation = fRadiusTranslation; }
|
---|
180 | void SetWindow( INT nWidth, INT nHeight, FLOAT fRadius = 0.9f ) { m_nWidth = nWidth; m_nHeight = nHeight; m_fRadius = fRadius; m_vCenter = D3DXVECTOR2(m_nWidth/2.0f,m_nHeight/2.0f); }
|
---|
181 | void SetOffset( INT nX, INT nY ) { m_Offset.x = nX; m_Offset.y = nY; }
|
---|
182 |
|
---|
183 | // Call these from client and use GetRotationMatrix() to read new rotation matrix
|
---|
184 | void OnBegin( int nX, int nY ); // start the rotation (pass current mouse position)
|
---|
185 | void OnMove( int nX, int nY ); // continue the rotation (pass current mouse position)
|
---|
186 | void OnEnd(); // end the rotation
|
---|
187 |
|
---|
188 | // Or call this to automatically handle left, middle, right buttons
|
---|
189 | LRESULT HandleMessages( HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam );
|
---|
190 |
|
---|
191 | // Functions to get/set state
|
---|
192 | D3DXMATRIX* GetRotationMatrix() { return D3DXMatrixRotationQuaternion(&m_mRotation, &m_qNow); };
|
---|
193 | D3DXMATRIX* GetTranslationMatrix() { return &m_mTranslation; }
|
---|
194 | D3DXMATRIX* GetTranslationDeltaMatrix() { return &m_mTranslationDelta; }
|
---|
195 | bool IsBeingDragged() { return m_bDrag; }
|
---|
196 | D3DXQUATERNION GetQuatNow() { return m_qNow; }
|
---|
197 | void SetQuatNow( D3DXQUATERNION q ) { m_qNow = q; }
|
---|
198 |
|
---|
199 | static D3DXQUATERNION QuatFromBallPoints( const D3DXVECTOR3 &vFrom, const D3DXVECTOR3 &vTo );
|
---|
200 |
|
---|
201 |
|
---|
202 | protected:
|
---|
203 | D3DXMATRIXA16 m_mRotation; // Matrix for arc ball's orientation
|
---|
204 | D3DXMATRIXA16 m_mTranslation; // Matrix for arc ball's position
|
---|
205 | D3DXMATRIXA16 m_mTranslationDelta; // Matrix for arc ball's position
|
---|
206 |
|
---|
207 | POINT m_Offset; // window offset, or upper-left corner of window
|
---|
208 | INT m_nWidth; // arc ball's window width
|
---|
209 | INT m_nHeight; // arc ball's window height
|
---|
210 | D3DXVECTOR2 m_vCenter; // center of arc ball
|
---|
211 | FLOAT m_fRadius; // arc ball's radius in screen coords
|
---|
212 | FLOAT m_fRadiusTranslation; // arc ball's radius for translating the target
|
---|
213 |
|
---|
214 | D3DXQUATERNION m_qDown; // Quaternion before button down
|
---|
215 | D3DXQUATERNION m_qNow; // Composite quaternion for current drag
|
---|
216 | bool m_bDrag; // Whether user is dragging arc ball
|
---|
217 |
|
---|
218 | POINT m_ptLastMouse; // position of last mouse point
|
---|
219 | D3DXVECTOR3 m_vDownPt; // starting point of rotation arc
|
---|
220 | D3DXVECTOR3 m_vCurrentPt; // current point of rotation arc
|
---|
221 |
|
---|
222 | D3DXVECTOR3 ScreenToVector( float fScreenPtX, float fScreenPtY );
|
---|
223 | };
|
---|
224 |
|
---|
225 |
|
---|
226 | //--------------------------------------------------------------------------------------
|
---|
227 | // used by CCamera to map WM_KEYDOWN keys
|
---|
228 | //--------------------------------------------------------------------------------------
|
---|
229 | enum D3DUtil_CameraKeys
|
---|
230 | {
|
---|
231 | CAM_STRAFE_LEFT = 0,
|
---|
232 | CAM_STRAFE_RIGHT,
|
---|
233 | CAM_MOVE_FORWARD,
|
---|
234 | CAM_MOVE_BACKWARD,
|
---|
235 | CAM_MOVE_UP,
|
---|
236 | CAM_MOVE_DOWN,
|
---|
237 | CAM_RESET,
|
---|
238 | CAM_CONTROLDOWN,
|
---|
239 | CAM_MAX_KEYS,
|
---|
240 | CAM_UNKNOWN = 0xFF
|
---|
241 | };
|
---|
242 |
|
---|
243 | #define KEY_WAS_DOWN_MASK 0x80
|
---|
244 | #define KEY_IS_DOWN_MASK 0x01
|
---|
245 |
|
---|
246 | #define MOUSE_LEFT_BUTTON 0x01
|
---|
247 | #define MOUSE_MIDDLE_BUTTON 0x02
|
---|
248 | #define MOUSE_RIGHT_BUTTON 0x04
|
---|
249 | #define MOUSE_WHEEL 0x08
|
---|
250 |
|
---|
251 |
|
---|
252 | //--------------------------------------------------------------------------------------
|
---|
253 | // Simple base camera class that moves and rotates. The base class
|
---|
254 | // records mouse and keyboard input for use by a derived class, and
|
---|
255 | // keeps common state.
|
---|
256 | //--------------------------------------------------------------------------------------
|
---|
257 | class CBaseCamera
|
---|
258 | {
|
---|
259 | public:
|
---|
260 | CBaseCamera();
|
---|
261 |
|
---|
262 | // Call these from client and use Get*Matrix() to read new matrices
|
---|
263 | virtual LRESULT HandleMessages( HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam );
|
---|
264 | virtual void FrameMove( FLOAT fElapsedTime ) = 0;
|
---|
265 |
|
---|
266 | // Functions to change camera matrices
|
---|
267 | virtual void Reset();
|
---|
268 | virtual void SetViewParams( D3DXVECTOR3* pvEyePt, D3DXVECTOR3* pvLookatPt );
|
---|
269 | virtual void SetProjParams( FLOAT fFOV, FLOAT fAspect, FLOAT fNearPlane, FLOAT fFarPlane );
|
---|
270 |
|
---|
271 | // Functions to change behavior
|
---|
272 | virtual void SetDragRect( RECT &rc ) { m_rcDrag = rc; }
|
---|
273 | void SetInvertPitch( bool bInvertPitch ) { m_bInvertPitch = bInvertPitch; }
|
---|
274 | void SetDrag( bool bMovementDrag, FLOAT fTotalDragTimeToZero = 0.25f ) { m_bMovementDrag = bMovementDrag; m_fTotalDragTimeToZero = fTotalDragTimeToZero; }
|
---|
275 | void SetEnableYAxisMovement( bool bEnableYAxisMovement ) { m_bEnableYAxisMovement = bEnableYAxisMovement; }
|
---|
276 | void SetEnablePositionMovement( bool bEnablePositionMovement ) { m_bEnablePositionMovement = bEnablePositionMovement; }
|
---|
277 | void SetClipToBoundary( bool bClipToBoundary, D3DXVECTOR3* pvMinBoundary, D3DXVECTOR3* pvMaxBoundary ) { m_bClipToBoundary = bClipToBoundary; if( pvMinBoundary ) m_vMinBoundary = *pvMinBoundary; if( pvMaxBoundary ) m_vMaxBoundary = *pvMaxBoundary; }
|
---|
278 | void SetScalers( FLOAT fRotationScaler = 0.01f, FLOAT fMoveScaler = 5.0f ) { m_fRotationScaler = fRotationScaler; m_fMoveScaler = fMoveScaler; }
|
---|
279 | void SetNumberOfFramesToSmoothMouseData( int nFrames ) { if( nFrames > 0 ) m_fFramesToSmoothMouseData = (float)nFrames; }
|
---|
280 | void SetResetCursorAfterMove( bool bResetCursorAfterMove ) { m_bResetCursorAfterMove = bResetCursorAfterMove; }
|
---|
281 |
|
---|
282 | // Functions to get state
|
---|
283 | const D3DXMATRIX* GetViewMatrix() const { return &m_mView; }
|
---|
284 | const D3DXMATRIX* GetProjMatrix() const { return &m_mProj; }
|
---|
285 | const D3DXVECTOR3* GetEyePt() const { return &m_vEye; }
|
---|
286 | const D3DXVECTOR3* GetLookAtPt() const { return &m_vLookAt; }
|
---|
287 |
|
---|
288 | bool IsBeingDragged() { return (m_bMouseLButtonDown || m_bMouseMButtonDown || m_bMouseRButtonDown); }
|
---|
289 | bool IsMouseLButtonDown() { return m_bMouseLButtonDown; }
|
---|
290 | bool IsMouseMButtonDown() { return m_bMouseMButtonDown; }
|
---|
291 | bool IsMouseRButtonDown() { return m_bMouseRButtonDown; }
|
---|
292 |
|
---|
293 | protected:
|
---|
294 | // Functions to map a WM_KEYDOWN key to a D3DUtil_CameraKeys enum
|
---|
295 | virtual D3DUtil_CameraKeys MapKey( UINT nKey );
|
---|
296 | bool IsKeyDown( BYTE key ) { return( (key & KEY_IS_DOWN_MASK) == KEY_IS_DOWN_MASK ); }
|
---|
297 | bool WasKeyDown( BYTE key ) { return( (key & KEY_WAS_DOWN_MASK) == KEY_WAS_DOWN_MASK ); }
|
---|
298 |
|
---|
299 | void ConstrainToBoundary( D3DXVECTOR3* pV );
|
---|
300 | void UpdateMouseDelta( float fElapsedTime );
|
---|
301 | void UpdateVelocity( float fElapsedTime );
|
---|
302 |
|
---|
303 | D3DXMATRIX m_mView; // View matrix
|
---|
304 | D3DXMATRIX m_mProj; // Projection matrix
|
---|
305 |
|
---|
306 | BYTE m_aKeys[CAM_MAX_KEYS]; // State of input - KEY_WAS_DOWN_MASK|KEY_IS_DOWN_MASK
|
---|
307 | POINT m_ptLastMousePosition; // Last absolute position of mouse cursor
|
---|
308 | bool m_bMouseLButtonDown; // True if left button is down
|
---|
309 | bool m_bMouseMButtonDown; // True if middle button is down
|
---|
310 | bool m_bMouseRButtonDown; // True if right button is down
|
---|
311 | int m_nCurrentButtonMask; // mask of which buttons are down
|
---|
312 | int m_nMouseWheelDelta; // Amount of middle wheel scroll (+/-)
|
---|
313 | D3DXVECTOR2 m_vMouseDelta; // Mouse relative delta smoothed over a few frames
|
---|
314 | float m_fFramesToSmoothMouseData; // Number of frames to smooth mouse data over
|
---|
315 |
|
---|
316 | D3DXVECTOR3 m_vDefaultEye; // Default camera eye position
|
---|
317 | D3DXVECTOR3 m_vDefaultLookAt; // Default LookAt position
|
---|
318 | D3DXVECTOR3 m_vEye; // Camera eye position
|
---|
319 | D3DXVECTOR3 m_vLookAt; // LookAt position
|
---|
320 | float m_fCameraYawAngle; // Yaw angle of camera
|
---|
321 | float m_fCameraPitchAngle; // Pitch angle of camera
|
---|
322 |
|
---|
323 | RECT m_rcDrag; // Rectangle within which a drag can be initiated.
|
---|
324 | D3DXVECTOR3 m_vVelocity; // Velocity of camera
|
---|
325 | bool m_bMovementDrag; // If true, then camera movement will slow to a stop otherwise movement is instant
|
---|
326 | D3DXVECTOR3 m_vVelocityDrag; // Velocity drag force
|
---|
327 | FLOAT m_fDragTimer; // Countdown timer to apply drag
|
---|
328 | FLOAT m_fTotalDragTimeToZero; // Time it takes for velocity to go from full to 0
|
---|
329 | D3DXVECTOR2 m_vRotVelocity; // Velocity of camera
|
---|
330 |
|
---|
331 | float m_fFOV; // Field of view
|
---|
332 | float m_fAspect; // Aspect ratio
|
---|
333 | float m_fNearPlane; // Near plane
|
---|
334 | float m_fFarPlane; // Far plane
|
---|
335 |
|
---|
336 | float m_fRotationScaler; // Scaler for rotation
|
---|
337 | float m_fMoveScaler; // Scaler for movement
|
---|
338 |
|
---|
339 | bool m_bInvertPitch; // Invert the pitch axis
|
---|
340 | bool m_bEnablePositionMovement; // If true, then the user can translate the camera/model
|
---|
341 | bool m_bEnableYAxisMovement; // If true, then camera can move in the y-axis
|
---|
342 |
|
---|
343 | bool m_bClipToBoundary; // If true, then the camera will be clipped to the boundary
|
---|
344 | D3DXVECTOR3 m_vMinBoundary; // Min point in clip boundary
|
---|
345 | D3DXVECTOR3 m_vMaxBoundary; // Max point in clip boundary
|
---|
346 |
|
---|
347 | bool m_bResetCursorAfterMove;// If true, the class will reset the cursor position so that the cursor always has space to move
|
---|
348 | };
|
---|
349 |
|
---|
350 |
|
---|
351 | //--------------------------------------------------------------------------------------
|
---|
352 | // Simple first person camera class that moves and rotates.
|
---|
353 | // It allows yaw and pitch but not roll. It uses WM_KEYDOWN and
|
---|
354 | // GetCursorPos() to respond to keyboard and mouse input and updates the
|
---|
355 | // view matrix based on input.
|
---|
356 | //--------------------------------------------------------------------------------------
|
---|
357 | class CFirstPersonCamera : public CBaseCamera
|
---|
358 | {
|
---|
359 | public:
|
---|
360 | CFirstPersonCamera();
|
---|
361 |
|
---|
362 | // Call these from client and use Get*Matrix() to read new matrices
|
---|
363 | virtual void FrameMove( FLOAT fElapsedTime );
|
---|
364 |
|
---|
365 | // Functions to change behavior
|
---|
366 | void SetRotateButtons( bool bLeft, bool bMiddle, bool bRight );
|
---|
367 |
|
---|
368 | // Functions to get state
|
---|
369 | D3DXMATRIX* GetWorldMatrix() { return &m_mCameraWorld; }
|
---|
370 |
|
---|
371 | const D3DXVECTOR3* GetWorldRight() const { return (D3DXVECTOR3*)&m_mCameraWorld._11; }
|
---|
372 | const D3DXVECTOR3* GetWorldUp() const { return (D3DXVECTOR3*)&m_mCameraWorld._21; }
|
---|
373 | const D3DXVECTOR3* GetWorldAhead() const { return (D3DXVECTOR3*)&m_mCameraWorld._31; }
|
---|
374 | const D3DXVECTOR3* GetEyePt() const { return (D3DXVECTOR3*)&m_mCameraWorld._41; }
|
---|
375 |
|
---|
376 | protected:
|
---|
377 | D3DXMATRIX m_mCameraWorld; // World matrix of the camera (inverse of the view matrix)
|
---|
378 |
|
---|
379 | int m_nActiveButtonMask; // Mask to determine which button to enable for rotation
|
---|
380 | };
|
---|
381 |
|
---|
382 |
|
---|
383 | //--------------------------------------------------------------------------------------
|
---|
384 | // Simple model viewing camera class that rotates around the object.
|
---|
385 | //--------------------------------------------------------------------------------------
|
---|
386 | class CModelViewerCamera : public CBaseCamera
|
---|
387 | {
|
---|
388 | public:
|
---|
389 | CModelViewerCamera();
|
---|
390 |
|
---|
391 | // Call these from client and use Get*Matrix() to read new matrices
|
---|
392 | virtual LRESULT HandleMessages( HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam );
|
---|
393 | virtual void FrameMove( FLOAT fElapsedTime );
|
---|
394 |
|
---|
395 |
|
---|
396 | // Functions to change behavior
|
---|
397 | virtual void SetDragRect( RECT &rc );
|
---|
398 | void Reset();
|
---|
399 | void SetViewParams( D3DXVECTOR3* pvEyePt, D3DXVECTOR3* pvLookatPt );
|
---|
400 | void SetButtonMasks( int nRotateModelButtonMask = MOUSE_LEFT_BUTTON, int nZoomButtonMask = MOUSE_WHEEL, int nRotateCameraButtonMask = MOUSE_RIGHT_BUTTON ) { m_nRotateModelButtonMask = nRotateModelButtonMask, m_nZoomButtonMask = nZoomButtonMask; m_nRotateCameraButtonMask = nRotateCameraButtonMask; }
|
---|
401 | void SetAttachCameraToModel( bool bEnable = false ) { m_bAttachCameraToModel = bEnable; }
|
---|
402 | void SetWindow( int nWidth, int nHeight, float fArcballRadius=0.9f ) { m_WorldArcBall.SetWindow( nWidth, nHeight, fArcballRadius ); m_ViewArcBall.SetWindow( nWidth, nHeight, fArcballRadius ); }
|
---|
403 | void SetRadius( float fDefaultRadius=5.0f, float fMinRadius=1.0f, float fMaxRadius=FLT_MAX ) { m_fDefaultRadius = m_fRadius = fDefaultRadius; m_fMinRadius = fMinRadius; m_fMaxRadius = fMaxRadius; }
|
---|
404 | void SetModelCenter( D3DXVECTOR3 vModelCenter ) { m_vModelCenter = vModelCenter; }
|
---|
405 | void SetLimitPitch( bool bLimitPitch ) { m_bLimitPitch = bLimitPitch; }
|
---|
406 | void SetViewQuat( D3DXQUATERNION q ) { m_ViewArcBall.SetQuatNow( q ); }
|
---|
407 | void SetWorldQuat( D3DXQUATERNION q ) { m_WorldArcBall.SetQuatNow( q ); }
|
---|
408 |
|
---|
409 | // Functions to get state
|
---|
410 | D3DXMATRIX* GetWorldMatrix() { return &m_mWorld; }
|
---|
411 |
|
---|
412 | protected:
|
---|
413 | CD3DArcBall m_WorldArcBall;
|
---|
414 | CD3DArcBall m_ViewArcBall;
|
---|
415 | D3DXVECTOR3 m_vModelCenter;
|
---|
416 | D3DXMATRIX m_mModelLastRot; // Last arcball rotation matrix for model
|
---|
417 | D3DXMATRIX m_mModelRot; // Rotation matrix of model
|
---|
418 | D3DXMATRIX m_mWorld; // World matrix of model
|
---|
419 |
|
---|
420 | int m_nRotateModelButtonMask;
|
---|
421 | int m_nZoomButtonMask;
|
---|
422 | int m_nRotateCameraButtonMask;
|
---|
423 |
|
---|
424 | bool m_bAttachCameraToModel;
|
---|
425 | bool m_bLimitPitch;
|
---|
426 | float m_fRadius; // Distance from the camera to model
|
---|
427 | float m_fDefaultRadius; // Distance from the camera to model
|
---|
428 | float m_fMinRadius; // Min radius
|
---|
429 | float m_fMaxRadius; // Max radius
|
---|
430 |
|
---|
431 | D3DXMATRIX m_mCameraRotLast;
|
---|
432 |
|
---|
433 | };
|
---|
434 |
|
---|
435 |
|
---|
436 | //--------------------------------------------------------------------------------------
|
---|
437 | // Manages the intertion point when drawing text
|
---|
438 | //--------------------------------------------------------------------------------------
|
---|
439 | class CDXUTTextHelper
|
---|
440 | {
|
---|
441 | public:
|
---|
442 | CDXUTTextHelper( ID3DXFont* pFont, ID3DXSprite* pSprite, int nLineHeight );
|
---|
443 |
|
---|
444 | void SetInsertionPos( int x, int y ) { m_pt.x = x; m_pt.y = y; }
|
---|
445 | void SetForegroundColor( D3DXCOLOR clr ) { m_clr = clr; }
|
---|
446 |
|
---|
447 | void Begin();
|
---|
448 | HRESULT DrawFormattedTextLine( const WCHAR* strMsg, ... );
|
---|
449 | HRESULT DrawTextLine( const WCHAR* strMsg );
|
---|
450 | HRESULT DrawFormattedTextLine( RECT &rc, DWORD dwFlags, const WCHAR* strMsg, ... );
|
---|
451 | HRESULT DrawTextLine( RECT &rc, DWORD dwFlags, const WCHAR* strMsg );
|
---|
452 | void End();
|
---|
453 |
|
---|
454 | protected:
|
---|
455 | ID3DXFont* m_pFont;
|
---|
456 | ID3DXSprite* m_pSprite;
|
---|
457 | D3DXCOLOR m_clr;
|
---|
458 | POINT m_pt;
|
---|
459 | int m_nLineHeight;
|
---|
460 | };
|
---|
461 |
|
---|
462 |
|
---|
463 | //--------------------------------------------------------------------------------------
|
---|
464 | // Manages a persistent list of lines and draws them using ID3DXLine
|
---|
465 | //--------------------------------------------------------------------------------------
|
---|
466 | class CDXUTLineManager
|
---|
467 | {
|
---|
468 | public:
|
---|
469 | CDXUTLineManager();
|
---|
470 | ~CDXUTLineManager();
|
---|
471 |
|
---|
472 | HRESULT OnCreatedDevice( IDirect3DDevice9* pd3dDevice );
|
---|
473 | HRESULT OnResetDevice();
|
---|
474 | HRESULT OnRender();
|
---|
475 | HRESULT OnLostDevice();
|
---|
476 | HRESULT OnDeletedDevice();
|
---|
477 |
|
---|
478 | HRESULT AddLine( int* pnLineID, D3DXVECTOR2* pVertexList, DWORD dwVertexListCount, D3DCOLOR Color, float fWidth, float fScaleRatio, bool bAntiAlias );
|
---|
479 | HRESULT AddRect( int* pnLineID, RECT rc, D3DCOLOR Color, float fWidth, float fScaleRatio, bool bAntiAlias );
|
---|
480 | HRESULT RemoveLine( int nLineID );
|
---|
481 | HRESULT RemoveAllLines();
|
---|
482 |
|
---|
483 | protected:
|
---|
484 | struct LINE_NODE
|
---|
485 | {
|
---|
486 | int nLineID;
|
---|
487 | D3DCOLOR Color;
|
---|
488 | float fWidth;
|
---|
489 | bool bAntiAlias;
|
---|
490 | float fScaleRatio;
|
---|
491 | D3DXVECTOR2* pVertexList;
|
---|
492 | DWORD dwVertexListCount;
|
---|
493 | };
|
---|
494 |
|
---|
495 | CGrowableArray<LINE_NODE*> m_LinesList;
|
---|
496 | IDirect3DDevice9* m_pd3dDevice;
|
---|
497 | ID3DXLine* m_pD3DXLine;
|
---|
498 | };
|
---|
499 |
|
---|
500 |
|
---|
501 | //--------------------------------------------------------------------------------------
|
---|
502 | // Manages the mesh, direction, mouse events of a directional arrow that
|
---|
503 | // rotates around a radius controlled by an arcball
|
---|
504 | //--------------------------------------------------------------------------------------
|
---|
505 | class CDXUTDirectionWidget
|
---|
506 | {
|
---|
507 | public:
|
---|
508 | CDXUTDirectionWidget();
|
---|
509 |
|
---|
510 | static HRESULT StaticOnCreateDevice( IDirect3DDevice9* pd3dDevice );
|
---|
511 | HRESULT OnResetDevice( const D3DSURFACE_DESC* pBackBufferSurfaceDesc );
|
---|
512 | HRESULT OnRender( D3DXCOLOR color, const D3DXMATRIX* pmView, const D3DXMATRIX* pmProj, const D3DXVECTOR3* pEyePt );
|
---|
513 | LRESULT HandleMessages( HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam );
|
---|
514 | static void StaticOnLostDevice();
|
---|
515 | static void StaticOnDestroyDevice();
|
---|
516 |
|
---|
517 | D3DXVECTOR3 GetLightDirection() { return m_vCurrentDir; };
|
---|
518 | void SetLightDirection( D3DXVECTOR3 vDir ) { m_vDefaultDir = m_vCurrentDir = vDir; };
|
---|
519 | void SetButtonMask( int nRotate = MOUSE_RIGHT_BUTTON ) { m_nRotateMask = nRotate; }
|
---|
520 |
|
---|
521 | float GetRadius() { return m_fRadius; };
|
---|
522 | void SetRadius( float fRadius ) { m_fRadius = fRadius; };
|
---|
523 |
|
---|
524 | bool IsBeingDragged() { return m_ArcBall.IsBeingDragged(); };
|
---|
525 |
|
---|
526 | protected:
|
---|
527 | HRESULT UpdateLightDir();
|
---|
528 |
|
---|
529 | static IDirect3DDevice9* s_pd3dDevice;
|
---|
530 | static ID3DXEffect* s_pEffect;
|
---|
531 | static ID3DXMesh* s_pMesh;
|
---|
532 |
|
---|
533 | float m_fRadius;
|
---|
534 | int m_nRotateMask;
|
---|
535 | CD3DArcBall m_ArcBall;
|
---|
536 | D3DXVECTOR3 m_vDefaultDir;
|
---|
537 | D3DXVECTOR3 m_vCurrentDir;
|
---|
538 | D3DXMATRIX m_mView;
|
---|
539 | D3DXMATRIXA16 m_mRot;
|
---|
540 | D3DXMATRIXA16 m_mRotSnapshot;
|
---|
541 | };
|
---|
542 |
|
---|
543 |
|
---|
544 | //--------------------------------------------------------------------------------------
|
---|
545 | // Returns the DirectX SDK path, as stored in the system registry
|
---|
546 | // during the SDK install.
|
---|
547 | //--------------------------------------------------------------------------------------
|
---|
548 | HRESULT DXUTGetDXSDKMediaPathCch( WCHAR* strDest, int cchDest );
|
---|
549 | HRESULT DXUTFindDXSDKMediaFileCch( WCHAR* strDestPath, int cchDest, LPCWSTR strFilename );
|
---|
550 |
|
---|
551 |
|
---|
552 | //--------------------------------------------------------------------------------------
|
---|
553 | // Returns the string for the given D3DFORMAT.
|
---|
554 | // bWithPrefix determines whether the string should include the "D3DFMT_"
|
---|
555 | //--------------------------------------------------------------------------------------
|
---|
556 | LPCWSTR DXUTD3DFormatToString( D3DFORMAT format, bool bWithPrefix );
|
---|
557 |
|
---|
558 |
|
---|
559 | //--------------------------------------------------------------------------------------
|
---|
560 | // Builds and sets a cursor for the D3D device based on hCursor.
|
---|
561 | //--------------------------------------------------------------------------------------
|
---|
562 | HRESULT DXUTSetDeviceCursor( IDirect3DDevice9* pd3dDevice, HCURSOR hCursor, bool bAddWatermark );
|
---|
563 |
|
---|
564 |
|
---|
565 | //--------------------------------------------------------------------------------------
|
---|
566 | // Returns a view matrix for rendering to a face of a cubemap.
|
---|
567 | //--------------------------------------------------------------------------------------
|
---|
568 | D3DXMATRIX DXUTGetCubeMapViewMatrix( DWORD dwFace );
|
---|
569 |
|
---|
570 |
|
---|
571 | //--------------------------------------------------------------------------------------
|
---|
572 | // Debug printing support
|
---|
573 | // See dxerr9.h for more debug printing support
|
---|
574 | //--------------------------------------------------------------------------------------
|
---|
575 | void DXUTOutputDebugStringW( LPCWSTR strMsg, ... );
|
---|
576 | void DXUTOutputDebugStringA( LPCSTR strMsg, ... );
|
---|
577 | HRESULT WINAPI DXUTTrace( const CHAR* strFile, DWORD dwLine, HRESULT hr, const WCHAR* strMsg, bool bPopMsgBox );
|
---|
578 |
|
---|
579 | #ifdef UNICODE
|
---|
580 | #define DXUTOutputDebugString DXUTOutputDebugStringW
|
---|
581 | #else
|
---|
582 | #define DXUTOutputDebugString DXUTOutputDebugStringA
|
---|
583 | #endif
|
---|
584 |
|
---|
585 | // These macros are very similar to dxerr9's but it special cases the HRESULT defined
|
---|
586 | // by the sample framework pop better message boxes.
|
---|
587 | #if defined(DEBUG) | defined(_DEBUG)
|
---|
588 | #define DXUT_ERR(str,hr) DXUTTrace( __FILE__, (DWORD)__LINE__, hr, str, false )
|
---|
589 | #define DXUT_ERR_MSGBOX(str,hr) DXUTTrace( __FILE__, (DWORD)__LINE__, hr, str, true )
|
---|
590 | #define DXUTTRACE DXUTOutputDebugString
|
---|
591 | #else
|
---|
592 | #define DXUT_ERR(str,hr) (hr)
|
---|
593 | #define DXUT_ERR_MSGBOX(str,hr) (hr)
|
---|
594 | #define DXUTTRACE (__noop)
|
---|
595 | #endif
|
---|
596 |
|
---|
597 |
|
---|
598 | //--------------------------------------------------------------------------------------
|
---|
599 | // Direct3D9 dynamic linking support -- calls top-level D3D9 APIs with graceful
|
---|
600 | // failure if APIs are not present.
|
---|
601 | //--------------------------------------------------------------------------------------
|
---|
602 |
|
---|
603 | IDirect3D9 * WINAPI DXUT_Dynamic_Direct3DCreate9(UINT SDKVersion);
|
---|
604 | int WINAPI DXUT_Dynamic_D3DPERF_BeginEvent( D3DCOLOR col, LPCWSTR wszName );
|
---|
605 | int WINAPI DXUT_Dynamic_D3DPERF_EndEvent( void );
|
---|
606 | void WINAPI DXUT_Dynamic_D3DPERF_SetMarker( D3DCOLOR col, LPCWSTR wszName );
|
---|
607 | void WINAPI DXUT_Dynamic_D3DPERF_SetRegion( D3DCOLOR col, LPCWSTR wszName );
|
---|
608 | BOOL WINAPI DXUT_Dynamic_D3DPERF_QueryRepeatFrame( void );
|
---|
609 | void WINAPI DXUT_Dynamic_D3DPERF_SetOptions( DWORD dwOptions );
|
---|
610 | DWORD WINAPI DXUT_Dynamic_D3DPERF_GetStatus( void );
|
---|
611 |
|
---|
612 |
|
---|
613 | //--------------------------------------------------------------------------------------
|
---|
614 | // Profiling/instrumentation support
|
---|
615 | //--------------------------------------------------------------------------------------
|
---|
616 |
|
---|
617 | //--------------------------------------------------------------------------------------
|
---|
618 | // Some D3DPERF APIs take a color that can be used when displaying user events in
|
---|
619 | // performance analysis tools. The following constants are provided for your
|
---|
620 | // convenience, but you can use any colors you like.
|
---|
621 | //--------------------------------------------------------------------------------------
|
---|
622 | const D3DCOLOR DXUT_PERFEVENTCOLOR = D3DCOLOR_XRGB(200,100,100);
|
---|
623 | const D3DCOLOR DXUT_PERFEVENTCOLOR2 = D3DCOLOR_XRGB(100,200,100);
|
---|
624 | const D3DCOLOR DXUT_PERFEVENTCOLOR3 = D3DCOLOR_XRGB(100,100,200);
|
---|
625 |
|
---|
626 | //--------------------------------------------------------------------------------------
|
---|
627 | // The following macros provide a convenient way for your code to call the D3DPERF
|
---|
628 | // functions only when PROFILE is defined. If PROFILE is not defined (as for the final
|
---|
629 | // release version of a program), these macros evaluate to nothing, so no detailed event
|
---|
630 | // information is embedded in your shipping program. It is recommended that you create
|
---|
631 | // and use three build configurations for your projects:
|
---|
632 | // Debug (nonoptimized code, asserts active, PROFILE defined to assist debugging)
|
---|
633 | // Profile (optimized code, asserts disabled, PROFILE defined to assist optimization)
|
---|
634 | // Release (optimized code, asserts disabled, PROFILE not defined)
|
---|
635 | //--------------------------------------------------------------------------------------
|
---|
636 | #ifdef PROFILE
|
---|
637 | // PROFILE is defined, so these macros call the D3DPERF functions
|
---|
638 | #define DXUT_BeginPerfEvent( color, pstrMessage ) DXUT_Dynamic_D3DPERF_BeginEvent( color, pstrMessage )
|
---|
639 | #define DXUT_EndPerfEvent() DXUT_Dynamic_D3DPERF_EndEvent()
|
---|
640 | #define DXUT_SetPerfMarker( color, pstrMessage ) DXUT_Dynamic_D3DPERF_SetMarker( color, pstrMessage )
|
---|
641 | #else
|
---|
642 | // PROFILE is not defined, so these macros do nothing
|
---|
643 | #define DXUT_BeginPerfEvent( color, pstrMessage ) (__noop)
|
---|
644 | #define DXUT_EndPerfEvent() (__noop)
|
---|
645 | #define DXUT_SetPerfMarker( color, pstrMessage ) (__noop)
|
---|
646 | #endif
|
---|
647 |
|
---|
648 | //--------------------------------------------------------------------------------------
|
---|
649 | // CDXUTPerfEventGenerator is a helper class that makes it easy to attach begin and end
|
---|
650 | // events to a block of code. Simply define a CDXUTPerfEventGenerator variable anywhere
|
---|
651 | // in a block of code, and the class's constructor will call DXUT_BeginPerfEvent when
|
---|
652 | // the block of code begins, and the class's destructor will call DXUT_EndPerfEvent when
|
---|
653 | // the block ends.
|
---|
654 | //--------------------------------------------------------------------------------------
|
---|
655 | class CDXUTPerfEventGenerator
|
---|
656 | {
|
---|
657 | public:
|
---|
658 | CDXUTPerfEventGenerator( D3DCOLOR color, WCHAR* pstrMessage ) { DXUT_BeginPerfEvent( color, pstrMessage ); }
|
---|
659 | ~CDXUTPerfEventGenerator( void ) { DXUT_EndPerfEvent(); }
|
---|
660 | };
|
---|
661 |
|
---|
662 | //--------------------------------------------------------------------------------------
|
---|
663 | // Implementation of CGrowableArray
|
---|
664 | //--------------------------------------------------------------------------------------
|
---|
665 |
|
---|
666 | // This version doesn't call ctor or dtor.
|
---|
667 | template< typename TYPE >
|
---|
668 | HRESULT CGrowableArray<TYPE>::SetSizeInternal( int nNewMaxSize )
|
---|
669 | {
|
---|
670 | if( nNewMaxSize < 0 )
|
---|
671 | {
|
---|
672 | assert( false );
|
---|
673 | return E_INVALIDARG;
|
---|
674 | }
|
---|
675 |
|
---|
676 | if( nNewMaxSize == 0 )
|
---|
677 | {
|
---|
678 | // Shrink to 0 size & cleanup
|
---|
679 | if( m_pData )
|
---|
680 | {
|
---|
681 | free( m_pData );
|
---|
682 | m_pData = NULL;
|
---|
683 | }
|
---|
684 |
|
---|
685 | m_nMaxSize = 0;
|
---|
686 | m_nSize = 0;
|
---|
687 | }
|
---|
688 | else if( m_pData == NULL || nNewMaxSize > m_nMaxSize )
|
---|
689 | {
|
---|
690 | // Grow array
|
---|
691 | int nGrowBy = ( m_nMaxSize == 0 ) ? 16 : m_nMaxSize;
|
---|
692 | nNewMaxSize = max( nNewMaxSize, m_nMaxSize + nGrowBy );
|
---|
693 |
|
---|
694 | TYPE* pDataNew = (TYPE*) realloc( m_pData, nNewMaxSize * sizeof(TYPE) );
|
---|
695 | if( pDataNew == NULL )
|
---|
696 | return E_OUTOFMEMORY;
|
---|
697 |
|
---|
698 | m_pData = pDataNew;
|
---|
699 | m_nMaxSize = nNewMaxSize;
|
---|
700 | }
|
---|
701 |
|
---|
702 | return S_OK;
|
---|
703 | }
|
---|
704 |
|
---|
705 |
|
---|
706 | //--------------------------------------------------------------------------------------
|
---|
707 | template< typename TYPE >
|
---|
708 | HRESULT CGrowableArray<TYPE>::SetSize( int nNewMaxSize )
|
---|
709 | {
|
---|
710 | int nOldSize = m_nSize;
|
---|
711 |
|
---|
712 | if( nOldSize > nNewMaxSize )
|
---|
713 | {
|
---|
714 | // Removing elements. Call dtor.
|
---|
715 |
|
---|
716 | for( int i = nNewMaxSize; i < nOldSize; ++i )
|
---|
717 | m_pData[i].~TYPE();
|
---|
718 | }
|
---|
719 |
|
---|
720 | // Adjust buffer. Note that there's no need to check for error
|
---|
721 | // since if it happens, nOldSize == nNewMaxSize will be true.)
|
---|
722 | HRESULT hr = SetSizeInternal( nNewMaxSize );
|
---|
723 |
|
---|
724 | if( nOldSize < nNewMaxSize )
|
---|
725 | {
|
---|
726 | // Adding elements. Call ctor.
|
---|
727 |
|
---|
728 | for( int i = nOldSize; i < nNewMaxSize; ++i )
|
---|
729 | ::new (&m_pData[i]) TYPE;
|
---|
730 | }
|
---|
731 |
|
---|
732 | return hr;
|
---|
733 | }
|
---|
734 |
|
---|
735 |
|
---|
736 | //--------------------------------------------------------------------------------------
|
---|
737 | template< typename TYPE >
|
---|
738 | HRESULT CGrowableArray<TYPE>::Add( const TYPE& value )
|
---|
739 | {
|
---|
740 | HRESULT hr;
|
---|
741 | if( FAILED( hr = SetSizeInternal( m_nSize + 1 ) ) )
|
---|
742 | return hr;
|
---|
743 |
|
---|
744 | // Construct the new element
|
---|
745 | ::new (&m_pData[m_nSize]) TYPE;
|
---|
746 |
|
---|
747 | // Assign
|
---|
748 | m_pData[m_nSize] = value;
|
---|
749 | ++m_nSize;
|
---|
750 |
|
---|
751 | return S_OK;
|
---|
752 | }
|
---|
753 |
|
---|
754 |
|
---|
755 | //--------------------------------------------------------------------------------------
|
---|
756 | template< typename TYPE >
|
---|
757 | HRESULT CGrowableArray<TYPE>::Insert( int nIndex, const TYPE& value )
|
---|
758 | {
|
---|
759 | HRESULT hr;
|
---|
760 |
|
---|
761 | // Validate index
|
---|
762 | if( nIndex < 0 ||
|
---|
763 | nIndex > m_nSize )
|
---|
764 | {
|
---|
765 | assert( false );
|
---|
766 | return E_INVALIDARG;
|
---|
767 | }
|
---|
768 |
|
---|
769 | // Prepare the buffer
|
---|
770 | if( FAILED( hr = SetSizeInternal( m_nSize + 1 ) ) )
|
---|
771 | return hr;
|
---|
772 |
|
---|
773 | // Shift the array
|
---|
774 | MoveMemory( &m_pData[nIndex+1], &m_pData[nIndex], sizeof(TYPE) * (m_nSize - nIndex) );
|
---|
775 |
|
---|
776 | // Construct the new element
|
---|
777 | ::new (&m_pData[nIndex]) TYPE;
|
---|
778 |
|
---|
779 | // Set the value and increase the size
|
---|
780 | m_pData[nIndex] = value;
|
---|
781 | ++m_nSize;
|
---|
782 |
|
---|
783 | return S_OK;
|
---|
784 | }
|
---|
785 |
|
---|
786 |
|
---|
787 | //--------------------------------------------------------------------------------------
|
---|
788 | template< typename TYPE >
|
---|
789 | HRESULT CGrowableArray<TYPE>::SetAt( int nIndex, const TYPE& value )
|
---|
790 | {
|
---|
791 | // Validate arguments
|
---|
792 | if( nIndex < 0 ||
|
---|
793 | nIndex >= m_nSize )
|
---|
794 | {
|
---|
795 | assert( false );
|
---|
796 | return E_INVALIDARG;
|
---|
797 | }
|
---|
798 |
|
---|
799 | m_pData[nIndex] = value;
|
---|
800 | return S_OK;
|
---|
801 | }
|
---|
802 |
|
---|
803 |
|
---|
804 | //--------------------------------------------------------------------------------------
|
---|
805 | // Searches for the specified value and returns the index of the first occurrence
|
---|
806 | // within the section of the data array that extends from iStart and contains the
|
---|
807 | // specified number of elements. Returns -1 if value is not found within the given
|
---|
808 | // section.
|
---|
809 | //--------------------------------------------------------------------------------------
|
---|
810 | template< typename TYPE >
|
---|
811 | int CGrowableArray<TYPE>::IndexOf( const TYPE& value, int iStart, int nNumElements )
|
---|
812 | {
|
---|
813 | // Validate arguments
|
---|
814 | if( iStart < 0 ||
|
---|
815 | iStart >= m_nSize ||
|
---|
816 | nNumElements < 0 ||
|
---|
817 | iStart + nNumElements > m_nSize )
|
---|
818 | {
|
---|
819 | assert( false );
|
---|
820 | return -1;
|
---|
821 | }
|
---|
822 |
|
---|
823 | // Search
|
---|
824 | for( int i = iStart; i < (iStart + nNumElements); i++ )
|
---|
825 | {
|
---|
826 | if( value == m_pData[i] )
|
---|
827 | return i;
|
---|
828 | }
|
---|
829 |
|
---|
830 | // Not found
|
---|
831 | return -1;
|
---|
832 | }
|
---|
833 |
|
---|
834 |
|
---|
835 | //--------------------------------------------------------------------------------------
|
---|
836 | // Searches for the specified value and returns the index of the last occurrence
|
---|
837 | // within the section of the data array that contains the specified number of elements
|
---|
838 | // and ends at iEnd. Returns -1 if value is not found within the given section.
|
---|
839 | //--------------------------------------------------------------------------------------
|
---|
840 | template< typename TYPE >
|
---|
841 | int CGrowableArray<TYPE>::LastIndexOf( const TYPE& value, int iEnd, int nNumElements )
|
---|
842 | {
|
---|
843 | // Validate arguments
|
---|
844 | if( iEnd < 0 ||
|
---|
845 | iEnd >= m_nSize ||
|
---|
846 | nNumElements < 0 ||
|
---|
847 | iEnd - nNumElements < 0 )
|
---|
848 | {
|
---|
849 | assert( false );
|
---|
850 | return -1;
|
---|
851 | }
|
---|
852 |
|
---|
853 | // Search
|
---|
854 | for( int i = iEnd; i > (iEnd - nNumElements); i-- )
|
---|
855 | {
|
---|
856 | if( value == m_pData[i] )
|
---|
857 | return i;
|
---|
858 | }
|
---|
859 |
|
---|
860 | // Not found
|
---|
861 | return -1;
|
---|
862 | }
|
---|
863 |
|
---|
864 |
|
---|
865 |
|
---|
866 | //--------------------------------------------------------------------------------------
|
---|
867 | template< typename TYPE >
|
---|
868 | HRESULT CGrowableArray<TYPE>::Remove( int nIndex )
|
---|
869 | {
|
---|
870 | if( nIndex < 0 ||
|
---|
871 | nIndex >= m_nSize )
|
---|
872 | {
|
---|
873 | assert( false );
|
---|
874 | return E_INVALIDARG;
|
---|
875 | }
|
---|
876 |
|
---|
877 | // Destruct the element to be removed
|
---|
878 | m_pData[nIndex].~TYPE();
|
---|
879 |
|
---|
880 | // Compact the array and decrease the size
|
---|
881 | MoveMemory( &m_pData[nIndex], &m_pData[nIndex+1], sizeof(TYPE) * (m_nSize - (nIndex+1)) );
|
---|
882 | --m_nSize;
|
---|
883 |
|
---|
884 | return S_OK;
|
---|
885 | }
|
---|
886 |
|
---|
887 |
|
---|
888 | #endif
|
---|