DXUTmisc.h
上传用户:junlon
上传日期:2022-01-05
资源大小:39075k
文件大小:44k
源码类别:

DirextX编程

开发平台:

Visual C++

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