开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 3D图形编程 > 查看源码
d3dutil.cpp
资源名称：estereo2.zip　[点击查看]
上传用户：fengshi120
上传日期：2014-07-17
资源大小：6155k
文件大小：25k
源码类别：

3D图形编程

开发平台：

C/C++

d3dutil.cpp：源码内容
  1. //-----------------------------------------------------------------------------
  2. // File: D3DUtil.cpp
  3. //
  4. // Desc: Shortcut macros and functions for using DX objects
  5. //-----------------------------------------------------------------------------
  6. #define STRICT
  7. #include "stdafx.h"
  8. #include <Windows.h>
  9. #include <WindowsX.h>
  10. #include <tchar.h>
  11. #include <stdio.h>
  12. #include "D3DUtil.h"
  13. #include "DXUtil.h"
  14. #include "D3DX9.h"
  19. //-----------------------------------------------------------------------------
  20. // Name: D3DUtil_InitMaterial()
  21. // Desc: Initializes a D3DMATERIAL9 structure, setting the diffuse and ambient
  22. //       colors. It does not set emissive or specular colors.
  23. //-----------------------------------------------------------------------------
  24. VOID D3DUtil_InitMaterial( D3DMATERIAL9& mtrl, FLOAT r, FLOAT g, FLOAT b,
  25.                            FLOAT a )
  26. {
  27.     ZeroMemory( &mtrl, sizeof(D3DMATERIAL9) );
  28.     mtrl.Diffuse.r = mtrl.Ambient.r = r;
  29.     mtrl.Diffuse.g = mtrl.Ambient.g = g;
  30.     mtrl.Diffuse.b = mtrl.Ambient.b = b;
  31.     mtrl.Diffuse.a = mtrl.Ambient.a = a;
  32. }
  37. //-----------------------------------------------------------------------------
  38. // Name: D3DUtil_InitLight()
  39. // Desc: Initializes a D3DLIGHT structure, setting the light position. The
  40. //       diffuse color is set to white; specular and ambient are left as black.
  41. //-----------------------------------------------------------------------------
  42. VOID D3DUtil_InitLight( D3DLIGHT9& light, D3DLIGHTTYPE ltType,
  43.                         FLOAT x, FLOAT y, FLOAT z )
  44. {
  45.     D3DXVECTOR3 vecLightDirUnnormalized(x, y, z);
  46.     ZeroMemory( &light, sizeof(D3DLIGHT9) );
  47.     light.Type        = ltType;
  48.     light.Diffuse.r   = 1.0f;
  49.     light.Diffuse.g   = 1.0f;
  50.     light.Diffuse.b   = 1.0f;
  51.     D3DXVec3Normalize( (D3DXVECTOR3*)&light.Direction, &vecLightDirUnnormalized );
  52.     light.Position.x   = x;
  53.     light.Position.y   = y;
  54.     light.Position.z   = z;
  55.     light.Range        = 1000.0f;
  56. }
  61. //-----------------------------------------------------------------------------
  62. // Name: D3DUtil_CreateTexture()
  63. // Desc: Helper function to create a texture. It checks the root path first,
  64. //       then tries the DXSDK media path (as specified in the system registry).
  65. //-----------------------------------------------------------------------------
  66. HRESULT D3DUtil_CreateTexture( LPDIRECT3DDEVICE9 pd3dDevice, TCHAR* strTexture,
  67.                                LPDIRECT3DTEXTURE9* ppTexture, D3DFORMAT d3dFormat )
  68. {
  69.     HRESULT hr;
  70.     TCHAR strPath[MAX_PATH];
  72.     // Get the path to the texture
  73.     if( FAILED( hr = DXUtil_FindMediaFileCb( strPath, sizeof(strPath), strTexture ) ) )
  74.         return hr;
  76.     // Create the texture using D3DX
  77.     return D3DXCreateTextureFromFileEx( pd3dDevice, strPath, 
  78.                 D3DX_DEFAULT, D3DX_DEFAULT, D3DX_DEFAULT, 0, d3dFormat, 
  79.                 D3DPOOL_MANAGED, D3DX_FILTER_TRIANGLE|D3DX_FILTER_MIRROR, 
  80.                 D3DX_FILTER_TRIANGLE|D3DX_FILTER_MIRROR, 0, NULL, NULL, ppTexture );
  81. }
  86. //-----------------------------------------------------------------------------
  87. // Name: D3DUtil_GetCubeMapViewMatrix()
  88. // Desc: Returns a view matrix for rendering to a face of a cubemap.
  89. //-----------------------------------------------------------------------------
  90. D3DXMATRIX D3DUtil_GetCubeMapViewMatrix( DWORD dwFace )
  91. {
  92.     D3DXVECTOR3 vEyePt   = D3DXVECTOR3( 0.0f, 0.0f, 0.0f );
  93.     D3DXVECTOR3 vLookDir;
  94.     D3DXVECTOR3 vUpDir;
  96.     switch( dwFace )
  97.     {
  98.         case D3DCUBEMAP_FACE_POSITIVE_X:
  99.             vLookDir = D3DXVECTOR3( 1.0f, 0.0f, 0.0f );
  100.             vUpDir   = D3DXVECTOR3( 0.0f, 1.0f, 0.0f );
  101.             break;
  102.         case D3DCUBEMAP_FACE_NEGATIVE_X:
  103.             vLookDir = D3DXVECTOR3(-1.0f, 0.0f, 0.0f );
  104.             vUpDir   = D3DXVECTOR3( 0.0f, 1.0f, 0.0f );
  105.             break;
  106.         case D3DCUBEMAP_FACE_POSITIVE_Y:
  107.             vLookDir = D3DXVECTOR3( 0.0f, 1.0f, 0.0f );
  108.             vUpDir   = D3DXVECTOR3( 0.0f, 0.0f,-1.0f );
  109.             break;
  110.         case D3DCUBEMAP_FACE_NEGATIVE_Y:
  111.             vLookDir = D3DXVECTOR3( 0.0f,-1.0f, 0.0f );
  112.             vUpDir   = D3DXVECTOR3( 0.0f, 0.0f, 1.0f );
  113.             break;
  114.         case D3DCUBEMAP_FACE_POSITIVE_Z:
  115.             vLookDir = D3DXVECTOR3( 0.0f, 0.0f, 1.0f );
  116.             vUpDir   = D3DXVECTOR3( 0.0f, 1.0f, 0.0f );
  117.             break;
  118.         case D3DCUBEMAP_FACE_NEGATIVE_Z:
  119.             vLookDir = D3DXVECTOR3( 0.0f, 0.0f,-1.0f );
  120.             vUpDir   = D3DXVECTOR3( 0.0f, 1.0f, 0.0f );
  121.             break;
  122.     }
  124.     // Set the view transform for this cubemap surface
  125.     D3DXMATRIXA16 matView;
  126.     D3DXMatrixLookAtLH( &matView, &vEyePt, &vLookDir, &vUpDir );
  127.     return matView;
  128. }
  133. //-----------------------------------------------------------------------------
  134. // Name: D3DUtil_GetRotationFromCursor()
  135. // Desc: Returns a quaternion for the rotation implied by the window's cursor
  136. //       position.
  137. //-----------------------------------------------------------------------------
  138. D3DXQUATERNION D3DUtil_GetRotationFromCursor( HWND hWnd,
  139.                                               FLOAT fTrackBallRadius )
  140. {
  141.     POINT pt;
  142.     RECT  rc;
  143.     GetCursorPos( &pt );
  144.     GetClientRect( hWnd, &rc );
  145.     ScreenToClient( hWnd, &pt );
  146.     FLOAT sx = ( ( ( 2.0f * pt.x ) / (rc.right-rc.left) ) - 1 );
  147.     FLOAT sy = ( ( ( 2.0f * pt.y ) / (rc.bottom-rc.top) ) - 1 );
  148.     FLOAT sz;
  150.     if( sx == 0.0f && sy == 0.0f )
  151.         return D3DXQUATERNION( 0.0f, 0.0f, 0.0f, 1.0f );
  153.     FLOAT d2 = sqrtf( sx*sx + sy*sy );
  155.     if( d2 < fTrackBallRadius * 0.70710678118654752440 ) // Inside sphere
  156.         sz = sqrtf( fTrackBallRadius*fTrackBallRadius - d2*d2 );
  157.     else                                                 // On hyperbola
  158.         sz = (fTrackBallRadius*fTrackBallRadius) / (2.0f*d2);
  160.     // Get two points on trackball's sphere
  161.     D3DXVECTOR3 p1( sx, sy, sz );
  162.     D3DXVECTOR3 p2( 0.0f, 0.0f, fTrackBallRadius );
  164.     // Get axis of rotation, which is cross product of p1 and p2
  165.     D3DXVECTOR3 vAxis;
  166.     D3DXVec3Cross( &vAxis, &p1, &p2);
  168.     // Calculate angle for the rotation about that axis
  169.     D3DXVECTOR3 vecDiff = p2-p1;
  170.     FLOAT t = D3DXVec3Length( &vecDiff ) / ( 2.0f*fTrackBallRadius );
  171.     if( t > +1.0f) t = +1.0f;
  172.     if( t < -1.0f) t = -1.0f;
  173.     FLOAT fAngle = 2.0f * asinf( t );
  175.     // Convert axis to quaternion
  176.     D3DXQUATERNION quat;
  177.     D3DXQuaternionRotationAxis( &quat, &vAxis, fAngle );
  178.     return quat;
  179. }
  184. //-----------------------------------------------------------------------------
  185. // Name: D3DUtil_SetDeviceCursor
  186. // Desc: Gives the D3D device a cursor with image and hotspot from hCursor.
  187. //-----------------------------------------------------------------------------
  188. HRESULT D3DUtil_SetDeviceCursor( LPDIRECT3DDEVICE9 pd3dDevice, HCURSOR hCursor,
  189.                                  BOOL bAddWatermark )
  190. {
  191.     HRESULT hr = E_FAIL;
  192.     ICONINFO iconinfo;
  193.     BOOL bBWCursor;
  194.     LPDIRECT3DSURFACE9 pCursorSurface = NULL;
  195.     HDC hdcColor = NULL;
  196.     HDC hdcMask = NULL;
  197.     HDC hdcScreen = NULL;
  198.     BITMAP bm;
  199.     DWORD dwWidth;
  200.     DWORD dwHeightSrc;
  201.     DWORD dwHeightDest;
  202.     COLORREF crColor;
  203.     COLORREF crMask;
  204.     UINT x;
  205.     UINT y;
  206.     BITMAPINFO bmi;
  207.     COLORREF* pcrArrayColor = NULL;
  208.     COLORREF* pcrArrayMask = NULL;
  209.     DWORD* pBitmap;
  210.     HGDIOBJ hgdiobjOld;
  212.     ZeroMemory( &iconinfo, sizeof(iconinfo) );
  213.     if( !GetIconInfo( hCursor, &iconinfo ) )
  214.         goto End;
  216.     if (0 == GetObject((HGDIOBJ)iconinfo.hbmMask, sizeof(BITMAP), (LPVOID)&bm))
  217.         goto End;
  218.     dwWidth = bm.bmWidth;
  219.     dwHeightSrc = bm.bmHeight;
  221.     if( iconinfo.hbmColor == NULL )
  222.     {
  223.         bBWCursor = TRUE;
  224.         dwHeightDest = dwHeightSrc / 2;
  225.     }
  226.     else 
  227.     {
  228.         bBWCursor = FALSE;
  229.         dwHeightDest = dwHeightSrc;
  230.     }
  232.     // Create a surface for the fullscreen cursor
  233.     if( FAILED( hr = pd3dDevice->CreateOffscreenPlainSurface( dwWidth, dwHeightDest, 
  234.         D3DFMT_A8R8G8B8, D3DPOOL_SCRATCH, &pCursorSurface, NULL ) ) )
  235.     {
  236.         goto End;
  237.     }
  239.     pcrArrayMask = new DWORD[dwWidth * dwHeightSrc];
  241.     ZeroMemory(&bmi, sizeof(bmi));
  242.     bmi.bmiHeader.biSize = sizeof(bmi.bmiHeader);
  243.     bmi.bmiHeader.biWidth = dwWidth;
  244.     bmi.bmiHeader.biHeight = dwHeightSrc;
  245.     bmi.bmiHeader.biPlanes = 1;
  246.     bmi.bmiHeader.biBitCount = 32;
  247.     bmi.bmiHeader.biCompression = BI_RGB;
  249.     hdcScreen = GetDC( NULL );
  250.     hdcMask = CreateCompatibleDC( hdcScreen );
  251.     if( hdcMask == NULL )
  252.     {
  253.         hr = E_FAIL;
  254.         goto End;
  255.     }
  256.     hgdiobjOld = SelectObject(hdcMask, iconinfo.hbmMask);
  257.     GetDIBits(hdcMask, iconinfo.hbmMask, 0, dwHeightSrc, 
  258.         pcrArrayMask, &bmi, DIB_RGB_COLORS);
  259.     SelectObject(hdcMask, hgdiobjOld);
  261.     if (!bBWCursor)
  262.     {
  263.         pcrArrayColor = new DWORD[dwWidth * dwHeightDest];
  264.         hdcColor = CreateCompatibleDC( hdcScreen );
  265.         if( hdcColor == NULL )
  266.         {
  267.             hr = E_FAIL;
  268.             goto End;
  269.         }
  270.         SelectObject(hdcColor, iconinfo.hbmColor);
  271.         GetDIBits(hdcColor, iconinfo.hbmColor, 0, dwHeightDest, 
  272.             pcrArrayColor, &bmi, DIB_RGB_COLORS);
  273.     }
  275.     // Transfer cursor image into the surface
  276.     D3DLOCKED_RECT lr;
  277.     pCursorSurface->LockRect( &lr, NULL, 0 );
  278.     pBitmap = (DWORD*)lr.pBits;
  279.     for( y = 0; y < dwHeightDest; y++ )
  280.     {
  281.         for( x = 0; x < dwWidth; x++ )
  282.         {
  283.             if (bBWCursor)
  284.             {
  285.                 crColor = pcrArrayMask[dwWidth*(dwHeightDest-1-y) + x];
  286.                 crMask = pcrArrayMask[dwWidth*(dwHeightSrc-1-y) + x];
  287.             }
  288.             else
  289.             {
  290.                 crColor = pcrArrayColor[dwWidth*(dwHeightDest-1-y) + x];
  291.                 crMask = pcrArrayMask[dwWidth*(dwHeightDest-1-y) + x];
  292.             }
  293.             if (crMask == 0)
  294.                 pBitmap[dwWidth*y + x] = 0xff000000 | crColor;
  295.             else
  296.                 pBitmap[dwWidth*y + x] = 0x00000000;
  298.             // It may be helpful to make the D3D cursor look slightly 
  299.             // different from the Windows cursor so you can distinguish 
  300.             // between the two when developing/testing code.  When
  301.             // bAddWatermark is TRUE, the following code adds some
  302.             // small grey "D3D" characters to the upper-left corner of
  303.             // the D3D cursor image.
  304.             if( bAddWatermark && x < 12 && y < 5 )
  305.             {
  306.                 // 11.. 11.. 11.. .... CCC0
  307.                 // 1.1. ..1. 1.1. .... A2A0
  308.                 // 1.1. .1.. 1.1. .... A4A0
  309.                 // 1.1. ..1. 1.1. .... A2A0
  310.                 // 11.. 11.. 11.. .... CCC0
  312.                 const WORD wMask[5] = { 0xccc0, 0xa2a0, 0xa4a0, 0xa2a0, 0xccc0 };
  313.                 if( wMask[y] & (1 << (15 - x)) )
  314.                 {
  315.                     pBitmap[dwWidth*y + x] |= 0xff808080;
  316.                 }
  317.             }
  318.         }
  319.     }
  320.     pCursorSurface->UnlockRect();
  322.     // Set the device cursor
  323.     if( FAILED( hr = pd3dDevice->SetCursorProperties( iconinfo.xHotspot, 
  324.         iconinfo.yHotspot, pCursorSurface ) ) )
  325.     {
  326.         goto End;
  327.     }
  329.     hr = S_OK;
  331. End:
  332.     if( iconinfo.hbmMask != NULL )
  333.         DeleteObject( iconinfo.hbmMask );
  334.     if( iconinfo.hbmColor != NULL )
  335.         DeleteObject( iconinfo.hbmColor );
  336.     if( hdcScreen != NULL )
  337.         ReleaseDC( NULL, hdcScreen );
  338.     if( hdcColor != NULL )
  339.         DeleteDC( hdcColor );
  340.     if( hdcMask != NULL )
  341.         DeleteDC( hdcMask );
  342.     SAFE_DELETE_ARRAY( pcrArrayColor );
  343.     SAFE_DELETE_ARRAY( pcrArrayMask );
  344.     SAFE_RELEASE( pCursorSurface );
  345.     return hr;
  346. }
  350. //-----------------------------------------------------------------------------
  351. // Name: D3DFormatToString
  352. // Desc: Returns the string for the given D3DFORMAT.
  353. //-----------------------------------------------------------------------------
  354. TCHAR* D3DUtil_D3DFormatToString( D3DFORMAT format, bool bWithPrefix )
  355. {
  356.     TCHAR* pstr = NULL;
  357.     switch( format )
  358.     {
  359.     case D3DFMT_UNKNOWN:         pstr = TEXT("D3DFMT_UNKNOWN"); break;
  360.     case D3DFMT_R8G8B8:          pstr = TEXT("D3DFMT_R8G8B8"); break;
  361.     case D3DFMT_A8R8G8B8:        pstr = TEXT("D3DFMT_A8R8G8B8"); break;
  362.     case D3DFMT_X8R8G8B8:        pstr = TEXT("D3DFMT_X8R8G8B8"); break;
  363.     case D3DFMT_R5G6B5:          pstr = TEXT("D3DFMT_R5G6B5"); break;
  364.     case D3DFMT_X1R5G5B5:        pstr = TEXT("D3DFMT_X1R5G5B5"); break;
  365.     case D3DFMT_A1R5G5B5:        pstr = TEXT("D3DFMT_A1R5G5B5"); break;
  366.     case D3DFMT_A4R4G4B4:        pstr = TEXT("D3DFMT_A4R4G4B4"); break;
  367.     case D3DFMT_R3G3B2:          pstr = TEXT("D3DFMT_R3G3B2"); break;
  368.     case D3DFMT_A8:              pstr = TEXT("D3DFMT_A8"); break;
  369.     case D3DFMT_A8R3G3B2:        pstr = TEXT("D3DFMT_A8R3G3B2"); break;
  370.     case D3DFMT_X4R4G4B4:        pstr = TEXT("D3DFMT_X4R4G4B4"); break;
  371.     case D3DFMT_A2B10G10R10:     pstr = TEXT("D3DFMT_A2B10G10R10"); break;
  372.     case D3DFMT_A8B8G8R8:        pstr = TEXT("D3DFMT_A8B8G8R8"); break;
  373.     case D3DFMT_X8B8G8R8:        pstr = TEXT("D3DFMT_X8B8G8R8"); break;
  374.     case D3DFMT_G16R16:          pstr = TEXT("D3DFMT_G16R16"); break;
  375.     case D3DFMT_A2R10G10B10:     pstr = TEXT("D3DFMT_A2R10G10B10"); break;
  376.     case D3DFMT_A16B16G16R16:    pstr = TEXT("D3DFMT_A16B16G16R16"); break;
  377.     case D3DFMT_A8P8:            pstr = TEXT("D3DFMT_A8P8"); break;
  378.     case D3DFMT_P8:              pstr = TEXT("D3DFMT_P8"); break;
  379.     case D3DFMT_L8:              pstr = TEXT("D3DFMT_L8"); break;
  380.     case D3DFMT_A8L8:            pstr = TEXT("D3DFMT_A8L8"); break;
  381.     case D3DFMT_A4L4:            pstr = TEXT("D3DFMT_A4L4"); break;
  382.     case D3DFMT_V8U8:            pstr = TEXT("D3DFMT_V8U8"); break;
  383.     case D3DFMT_L6V5U5:          pstr = TEXT("D3DFMT_L6V5U5"); break;
  384.     case D3DFMT_X8L8V8U8:        pstr = TEXT("D3DFMT_X8L8V8U8"); break;
  385.     case D3DFMT_Q8W8V8U8:        pstr = TEXT("D3DFMT_Q8W8V8U8"); break;
  386.     case D3DFMT_V16U16:          pstr = TEXT("D3DFMT_V16U16"); break;
  387.     case D3DFMT_A2W10V10U10:     pstr = TEXT("D3DFMT_A2W10V10U10"); break;
  388.     case D3DFMT_UYVY:            pstr = TEXT("D3DFMT_UYVY"); break;
  389.     case D3DFMT_YUY2:            pstr = TEXT("D3DFMT_YUY2"); break;
  390.     case D3DFMT_DXT1:            pstr = TEXT("D3DFMT_DXT1"); break;
  391.     case D3DFMT_DXT2:            pstr = TEXT("D3DFMT_DXT2"); break;
  392.     case D3DFMT_DXT3:            pstr = TEXT("D3DFMT_DXT3"); break;
  393.     case D3DFMT_DXT4:            pstr = TEXT("D3DFMT_DXT4"); break;
  394.     case D3DFMT_DXT5:            pstr = TEXT("D3DFMT_DXT5"); break;
  395.     case D3DFMT_D16_LOCKABLE:    pstr = TEXT("D3DFMT_D16_LOCKABLE"); break;
  396.     case D3DFMT_D32:             pstr = TEXT("D3DFMT_D32"); break;
  397.     case D3DFMT_D15S1:           pstr = TEXT("D3DFMT_D15S1"); break;
  398.     case D3DFMT_D24S8:           pstr = TEXT("D3DFMT_D24S8"); break;
  399.     case D3DFMT_D24X8:           pstr = TEXT("D3DFMT_D24X8"); break;
  400.     case D3DFMT_D24X4S4:         pstr = TEXT("D3DFMT_D24X4S4"); break;
  401.     case D3DFMT_D16:             pstr = TEXT("D3DFMT_D16"); break;
  402.     case D3DFMT_L16:             pstr = TEXT("D3DFMT_L16"); break;
  403.     case D3DFMT_VERTEXDATA:      pstr = TEXT("D3DFMT_VERTEXDATA"); break;
  404.     case D3DFMT_INDEX16:         pstr = TEXT("D3DFMT_INDEX16"); break;
  405.     case D3DFMT_INDEX32:         pstr = TEXT("D3DFMT_INDEX32"); break;
  406.     case D3DFMT_Q16W16V16U16:    pstr = TEXT("D3DFMT_Q16W16V16U16"); break;
  407.     case D3DFMT_MULTI2_ARGB8:    pstr = TEXT("D3DFMT_MULTI2_ARGB8"); break;
  408.     case D3DFMT_R16F:            pstr = TEXT("D3DFMT_R16F"); break;
  409.     case D3DFMT_G16R16F:         pstr = TEXT("D3DFMT_G16R16F"); break;
  410.     case D3DFMT_A16B16G16R16F:   pstr = TEXT("D3DFMT_A16B16G16R16F"); break;
  411.     case D3DFMT_R32F:            pstr = TEXT("D3DFMT_R32F"); break;
  412.     case D3DFMT_G32R32F:         pstr = TEXT("D3DFMT_G32R32F"); break;
  413.     case D3DFMT_A32B32G32R32F:   pstr = TEXT("D3DFMT_A32B32G32R32F"); break;
  414.     case D3DFMT_CxV8U8:          pstr = TEXT("D3DFMT_CxV8U8"); break;
  415.     default:                     pstr = TEXT("Unknown format"); break;
  416.     }
  417.     if( bWithPrefix || _tcsstr( pstr, TEXT("D3DFMT_") )== NULL )
  418.         return pstr;
  419.     else
  420.         return pstr + lstrlen( TEXT("D3DFMT_") );
  421. }
  424. //-----------------------------------------------------------------------------
  425. // Name: D3DXQuaternionUnitAxisToUnitAxis2
  426. // Desc: Axis to axis quaternion double angle (no normalization)
  427. //       Takes two points on unit sphere an angle THETA apart, returns
  428. //       quaternion that represents a rotation around cross product by 2*THETA.
  429. //-----------------------------------------------------------------------------
  430. inline D3DXQUATERNION* WINAPI D3DXQuaternionUnitAxisToUnitAxis2
  431. ( D3DXQUATERNION *pOut, const D3DXVECTOR3 *pvFrom, const D3DXVECTOR3 *pvTo)
  432. {
  433.     D3DXVECTOR3 vAxis;
  434.     D3DXVec3Cross(&vAxis, pvFrom, pvTo);    // proportional to sin(theta)
  435.     pOut->x = vAxis.x;
  436.     pOut->y = vAxis.y;
  437.     pOut->z = vAxis.z;
  438.     pOut->w = D3DXVec3Dot( pvFrom, pvTo );
  439.     return pOut;
  440. }
  445. //-----------------------------------------------------------------------------
  446. // Name: D3DXQuaternionAxisToAxis
  447. // Desc: Axis to axis quaternion 
  448. //       Takes two points on unit sphere an angle THETA apart, returns
  449. //       quaternion that represents a rotation around cross product by theta.
  450. //-----------------------------------------------------------------------------
  451. inline D3DXQUATERNION* WINAPI D3DXQuaternionAxisToAxis
  452. ( D3DXQUATERNION *pOut, const D3DXVECTOR3 *pvFrom, const D3DXVECTOR3 *pvTo)
  453. {
  454.     D3DXVECTOR3 vA, vB;
  455.     D3DXVec3Normalize(&vA, pvFrom);
  456.     D3DXVec3Normalize(&vB, pvTo);
  457.     D3DXVECTOR3 vHalf(vA + vB);
  458.     D3DXVec3Normalize(&vHalf, &vHalf);
  459.     return D3DXQuaternionUnitAxisToUnitAxis2(pOut, &vA, &vHalf);
  460. }
  465. //-----------------------------------------------------------------------------
  466. // Name:
  467. // Desc:
  468. //-----------------------------------------------------------------------------
  469. CD3DArcBall::CD3DArcBall()
  470. {
  471.     Init();
  472. }
  477. //-----------------------------------------------------------------------------
  478. // Name:
  479. // Desc:
  480. //-----------------------------------------------------------------------------
  481. void CD3DArcBall::Init()
  482. {
  483.     D3DXQuaternionIdentity( &m_qDown );
  484.     D3DXQuaternionIdentity( &m_qNow );
  486.     D3DXMatrixIdentity( &m_matRotation );
  487.     D3DXMatrixIdentity( &m_matRotationDelta );
  488.     D3DXMatrixIdentity( &m_matTranslation );
  489.     D3DXMatrixIdentity( &m_matTranslationDelta );
  490.     m_bDrag = FALSE;
  491.     m_fRadiusTranslation = 1.0f;
  492.     m_bRightHanded = FALSE;
  493. }
  498. //-----------------------------------------------------------------------------
  499. // Name:
  500. // Desc:
  501. //-----------------------------------------------------------------------------
  502. VOID CD3DArcBall::SetWindow( int iWidth, int iHeight, float fRadius )
  503. {
  504.     // Set ArcBall info
  505.     m_iWidth  = iWidth;
  506.     m_iHeight = iHeight;
  507.     m_fRadius = fRadius;
  508. }
  513. //-----------------------------------------------------------------------------
  514. // Name:
  515. // Desc:
  516. //-----------------------------------------------------------------------------
  517. D3DXVECTOR3 CD3DArcBall::ScreenToVector( int sx, int sy )
  518. {
  519.     // Scale to screen
  520.     FLOAT x   = -(sx - m_iWidth/2)  / (m_fRadius*m_iWidth/2);
  521.     FLOAT y   =  (sy - m_iHeight/2) / (m_fRadius*m_iHeight/2);
  523.     if( m_bRightHanded )
  524.     {
  525.         x = -x;
  526.         y = -y;
  527.     }
  529.     FLOAT z   = 0.0f;
  530.     FLOAT mag = x*x + y*y;
  532.     if( mag > 1.0f )
  533.     {
  534.         FLOAT scale = 1.0f/sqrtf(mag);
  535.         x *= scale;
  536.         y *= scale;
  537.     }
  538.     else
  539.         z = sqrtf( 1.0f - mag );
  541.     // Return vector
  542.     return D3DXVECTOR3( x, y, z );
  543. }
  548. //-----------------------------------------------------------------------------
  549. // Name:
  550. // Desc:
  551. //-----------------------------------------------------------------------------
  552. VOID CD3DArcBall::SetRadius( FLOAT fRadius )
  553. {
  554.     m_fRadiusTranslation = fRadius;
  555. }
  560. //-----------------------------------------------------------------------------
  561. // Name:
  562. // Desc:
  563. //-----------------------------------------------------------------------------
  564. LRESULT CD3DArcBall::HandleMouseMessages( HWND hWnd, UINT uMsg, WPARAM wParam,
  565.                                           LPARAM lParam )
  566. {
  567.     UNREFERENCED_PARAMETER( hWnd );
  569.     static int         iCurMouseX;      // Saved mouse position
  570.     static int         iCurMouseY;
  571.     static D3DXVECTOR3 s_vDown;         // Button down vector
  573.     // Current mouse position
  574.     int iMouseX = GET_X_LPARAM(lParam);
  575.     int iMouseY = GET_Y_LPARAM(lParam);
  577.     switch( uMsg )
  578.     {
  579.         case WM_RBUTTONDOWN:
  580.         case WM_MBUTTONDOWN:
  581.             // Store off the position of the cursor when the button is pressed
  582.             iCurMouseX = iMouseX;
  583.             iCurMouseY = iMouseY;
  584.             return TRUE;
  586.         case WM_LBUTTONDOWN:
  587.             // Start drag mode
  588.             m_bDrag = TRUE;
  589.             s_vDown = ScreenToVector( iMouseX, iMouseY );
  590.             m_qDown = m_qNow;
  591.             return TRUE;
  593.         case WM_LBUTTONUP:
  594.             // End drag mode
  595.             m_bDrag = FALSE;
  596.             return TRUE;
  598.         case WM_MOUSEMOVE:
  599.             // Drag object
  600.             if( MK_LBUTTON&wParam )
  601.             {
  602.                 if( m_bDrag )
  603.                 {
  604.                     // recompute m_qNow
  605.                     D3DXVECTOR3 vCur = ScreenToVector( iMouseX, iMouseY );
  606.                     D3DXQUATERNION qAxisToAxis;
  607.                     D3DXQuaternionAxisToAxis(&qAxisToAxis, &s_vDown, &vCur);
  608.                     m_qNow = m_qDown;
  609.                     m_qNow *= qAxisToAxis;
  610.                     D3DXMatrixRotationQuaternion(&m_matRotationDelta, &qAxisToAxis);
  611.                 }
  612.                 else
  613.                     D3DXMatrixIdentity(&m_matRotationDelta);
  614.                 D3DXMatrixRotationQuaternion(&m_matRotation, &m_qNow);
  615.                 m_bDrag = TRUE;
  616.             }
  617.             else if( (MK_RBUTTON&wParam) || (MK_MBUTTON&wParam) )
  618.             {
  619.                 // Normalize based on size of window and bounding sphere radius
  620.                 FLOAT fDeltaX = ( iCurMouseX-iMouseX ) * m_fRadiusTranslation / m_iWidth;
  621.                 FLOAT fDeltaY = ( iCurMouseY-iMouseY ) * m_fRadiusTranslation / m_iHeight;
  623.                 if( wParam & MK_RBUTTON )
  624.                 {
  625.                     D3DXMatrixTranslation( &m_matTranslationDelta, -2*fDeltaX, 2*fDeltaY, 0.0f );
  626.                     D3DXMatrixMultiply( &m_matTranslation, &m_matTranslation, &m_matTranslationDelta );
  627.                 }
  628.                 else  // wParam & MK_MBUTTON
  629.                 {
  630.                     D3DXMatrixTranslation( &m_matTranslationDelta, 0.0f, 0.0f, 5*fDeltaY );
  631.                     D3DXMatrixMultiply( &m_matTranslation, &m_matTranslation, &m_matTranslationDelta );
  632.                 }
  634.                 // Store mouse coordinate
  635.                 iCurMouseX = iMouseX;
  636.                 iCurMouseY = iMouseY;
  637.             }
  638.             return TRUE;
  639.     }
  641.     return FALSE;
  642. }
  647. //-----------------------------------------------------------------------------
  648. // Name:
  649. // Desc:
  650. //-----------------------------------------------------------------------------
  651. CD3DCamera::CD3DCamera()
  652. {
  653.     // Set attributes for the view matrix
  654.     D3DXVECTOR3 vEyePt(0.0f,0.0f,0.0f);
  655.     D3DXVECTOR3 vLookatPt(0.0f,0.0f,1.0f);
  656.     D3DXVECTOR3 vUpVec(0.0f,1.0f,0.0f);
  657.     SetViewParams( vEyePt, vLookatPt, vUpVec );
  659.     // Set attributes for the projection matrix
  660.     SetProjParams( D3DX_PI/4, 1.0f, 1.0f, 1000.0f );
  661. }
  666. //-----------------------------------------------------------------------------
  667. // Name:
  668. // Desc:
  669. //-----------------------------------------------------------------------------
  670. VOID CD3DCamera::SetViewParams( D3DXVECTOR3 &vEyePt, D3DXVECTOR3& vLookatPt,
  671.                                 D3DXVECTOR3& vUpVec )
  672. {
  673.     // Set attributes for the view matrix
  674.     m_vEyePt    = vEyePt;
  675.     m_vLookatPt = vLookatPt;
  676.     m_vUpVec    = vUpVec;
  677.     D3DXVECTOR3 vDir = m_vLookatPt - m_vEyePt;
  678.     D3DXVec3Normalize( &m_vView, &vDir );
  679.     D3DXVec3Cross( &m_vCross, &m_vView, &m_vUpVec );
  681.     D3DXMatrixLookAtLH( &m_matView, &m_vEyePt, &m_vLookatPt, &m_vUpVec );
  682.     D3DXMatrixInverse( &m_matBillboard, NULL, &m_matView );
  683.     m_matBillboard._41 = 0.0f;
  684.     m_matBillboard._42 = 0.0f;
  685.     m_matBillboard._43 = 0.0f;
  686. }
  691. //-----------------------------------------------------------------------------
  692. // Name:
  693. // Desc:
  694. //-----------------------------------------------------------------------------
  695. VOID CD3DCamera::SetProjParams( FLOAT fFOV, FLOAT fAspect, FLOAT fNearPlane,
  696.                                 FLOAT fFarPlane )
  697. {
  698.     // Set attributes for the projection matrix
  699.     m_fFOV        = fFOV;
  700.     m_fAspect     = fAspect;
  701.     m_fNearPlane  = fNearPlane;
  702.     m_fFarPlane   = fFarPlane;
  704.     D3DXMatrixPerspectiveFovLH( &m_matProj, fFOV, fAspect, fNearPlane, fFarPlane );
  705. }