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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > DirextX编程 > 查看源码
AllocateHierarchy.cpp
资源名称：Direct3D.rar　[点击查看]
上传用户：junlon
上传日期：2022-01-05
资源大小：39075k
文件大小：12k
源码类别：

DirextX编程

开发平台：

Visual C++

AllocateHierarchy.cpp：源码内容
  1. //=============================================================================
  2. //Desc: AllocateHierarchy.cpp，CAllocateHierarchy类的实现
  3. //=============================================================================
  5. #include "AllocateHierarchy.h"
  8. //-----------------------------------------------------------------------------
  9. // 拷贝字符串
  10. //-----------------------------------------------------------------------------
  11. HRESULT CAllocateHierarchy::AllocateName( LPCSTR Name, LPSTR *pNewName )
  12. {
  13.     UINT cbLength;
  15.     if( Name != NULL )
  16.     {
  17.         cbLength = (UINT)strlen(Name) + 1;
  18.         *pNewName = new CHAR[cbLength];
  19.         if (*pNewName == NULL)
  20.             return E_OUTOFMEMORY;
  21.         memcpy( *pNewName, Name, cbLength*sizeof(CHAR) );
  22.     }
  23.     else
  24.     {
  25.         *pNewName = NULL;
  26.     }
  28.     return S_OK;
  29. }
  32. //-----------------------------------------------------------------------------
  33. // 生成蒙皮网格模型(含有每个顶点的混合权重、索引和一个骨骼组合表)
  34. //-----------------------------------------------------------------------------
  35. HRESULT CAllocateHierarchy::GenerateSkinnedMesh(D3DXMESHCONTAINER_DERIVED *pMeshContainer, LPDIRECT3DDEVICE9 pd3dDevice)
  36. {
  37. HRESULT hr = S_OK;
  39. if (pMeshContainer->pSkinInfo == NULL)
  40. return hr;
  42. SAFE_RELEASE( pMeshContainer->MeshData.pMesh );
  43. SAFE_RELEASE( pMeshContainer->pBoneCombinationBuf );
  45. //获取网格模型索引缓冲区
  46. LPDIRECT3DINDEXBUFFER9 pIB;
  47. if (FAILED(hr = pMeshContainer->pOrigMesh->GetIndexBuffer(&pIB)))
  48. return E_FAIL;
  50. //获取影响一个面(三角形)的矩阵数量
  51. DWORD NumMaxFaceInfl;
  52. hr = pMeshContainer->pSkinInfo->GetMaxFaceInfluences(pIB, pMeshContainer->pOrigMesh->GetNumFaces(), &NumMaxFaceInfl);
  53. pIB->Release();
  54. if (FAILED(hr))
  55. return E_FAIL;
  57. //影响一个面的矩阵数量不会超过12
  58. NumMaxFaceInfl = min(NumMaxFaceInfl, 12);
  60. //获取当前设备的能力
  61. D3DCAPS9 d3dCaps;
  62.     pd3dDevice->GetDeviceCaps( &d3dCaps );
  64. //根据当前设备能力和NumMaxFaceInfl, 设置在进行骨骼蒙皮调色时使用骨骼数量的上限
  65. if( (d3dCaps.MaxVertexBlendMatrixIndex+1)/2 < NumMaxFaceInfl )
  66. {
  67. pMeshContainer->NumPaletteEntries = min(256, pMeshContainer->pSkinInfo->GetNumBones());
  68. pMeshContainer->UseSoftwareVP = true;
  69. }
  70. else
  71. {
  72. pMeshContainer->NumPaletteEntries = min( (d3dCaps.MaxVertexBlendMatrixIndex+1) / 2, 
  73.                                      pMeshContainer->pSkinInfo->GetNumBones() );
  74. pMeshContainer->UseSoftwareVP = false;
  75. }
  77. //生成蒙皮网格模型
  78. hr = pMeshContainer->pSkinInfo->ConvertToIndexedBlendedMesh
  79.                               ( pMeshContainer->pOrigMesh,
  80.                                 0,
  81.                                 pMeshContainer->NumPaletteEntries,
  82.                                 pMeshContainer->pAdjacency,
  83.                                 NULL, NULL, NULL,
  84.                                 &pMeshContainer->NumInfl,
  85.                                 &pMeshContainer->NumAttributeGroups,
  86.                                 &pMeshContainer->pBoneCombinationBuf,
  87.                                 &pMeshContainer->MeshData.pMesh);
  89. return hr;
  91. }
  94. //-----------------------------------------------------------------------------
  95. // 创建框架, 仅仅是分配内存和初始化,还没有对其成员赋予合适的值
  96. //-----------------------------------------------------------------------------
  97. HRESULT CAllocateHierarchy::CreateFrame(LPCSTR Name, LPD3DXFRAME *ppNewFrame)
  98. {
  99.     HRESULT hr = S_OK;
  100.     D3DXFRAME_DERIVED *pFrame;
  102.     *ppNewFrame = NULL;
  104.     pFrame = new D3DXFRAME_DERIVED;  //创建框架结构对象
  105.     if (pFrame == NULL) 
  106.     {
  107.         return E_OUTOFMEMORY;
  108.     }
  110. //为框架指定名称
  111.     hr = AllocateName(Name, (LPSTR*)&pFrame->Name);
  112. if (FAILED(hr))
  113. {
  114. delete pFrame;
  115. return hr;
  116. }
  118. //初始化D3DXFRAME_DERIVED结构其它成员变量
  119.     D3DXMatrixIdentity(&pFrame->TransformationMatrix);
  120.     D3DXMatrixIdentity(&pFrame->CombinedTransformationMatrix);
  122.     pFrame->pMeshContainer = NULL;
  123.     pFrame->pFrameSibling = NULL;
  124.     pFrame->pFrameFirstChild = NULL;
  126.     *ppNewFrame = pFrame;
  127.     pFrame = NULL;
  129. return hr;
  130. }
  133. //-----------------------------------------------------------------------------
  134. // 在这里是调用了成员函数 GenerateSkinnedMesh(pMeshContainer);
  135. // 是在这里加载了蒙皮信息
  136. //-----------------------------------------------------------------------------
  137. HRESULT CAllocateHierarchy::CreateMeshContainer(LPCSTR Name, 
  138.                                                 CONST D3DXMESHDATA *pMeshData,
  139.                                                 CONST D3DXMATERIAL *pMaterials, 
  140.                                                 CONST D3DXEFFECTINSTANCE *pEffectInstances, 
  141.                                                 DWORD NumMaterials, 
  142.                                                 CONST DWORD *pAdjacency, 
  143.                                                 LPD3DXSKININFO pSkinInfo, 
  144.                                                 LPD3DXMESHCONTAINER *ppNewMeshContainer) 
  145. {
  146.     HRESULT hr;
  147.     D3DXMESHCONTAINER_DERIVED *pMeshContainer = NULL;
  148.     UINT NumFaces;       //网格中的面数,在填充网格容器结构的邻接信息成员时使用
  149.     UINT iMaterial;      //纹理操作时的循环变量
  150.     UINT cBones;         //当前网格模型骨骼总数
  151.     LPDIRECT3DDEVICE9 pd3dDevice = NULL;
  152.     LPD3DXMESH pMesh    = NULL;
  153.     *ppNewMeshContainer = NULL;
  155.     if (pMeshData->Type != D3DXMESHTYPE_MESH)
  156.     {
  157.         return E_FAIL;
  158.     }
  160.     pMesh = pMeshData->pMesh;
  162.     if (pMesh->GetFVF() == 0)
  163.     {
  164.         return E_FAIL;
  165.     }
  167. //为网格容器分配内存
  168.     pMeshContainer = new D3DXMESHCONTAINER_DERIVED;
  169.     if (pMeshContainer == NULL)
  170.     {
  171.         return E_OUTOFMEMORY;
  172.     }
  173.     memset(pMeshContainer, 0, sizeof(D3DXMESHCONTAINER_DERIVED));
  175. //填充网格容器结构D3DXMESHCONTAINER_DERIVED的成员
  177. //为网格指定名称
  178. hr = AllocateName(Name, &pMeshContainer->Name);
  179. if (FAILED(hr))
  180. {
  181. DestroyMeshContainer(pMeshContainer);
  182. return hr;
  183. }      
  185.     pMesh->GetDevice(&pd3dDevice);
  186.     NumFaces = pMesh->GetNumFaces();
  188. //确保网格顶点包含法线
  189.     if (!(pMesh->GetFVF() & D3DFVF_NORMAL))
  190.     {
  191.         pMeshContainer->MeshData.Type = D3DXMESHTYPE_MESH;
  192.         hr = pMesh->CloneMeshFVF( pMesh->GetOptions(), 
  193.                       pMesh->GetFVF() | D3DFVF_NORMAL, 
  194.                       pd3dDevice, 
  195.                       &pMeshContainer->MeshData.pMesh );
  196. if (FAILED(hr))
  197. {
  198. SAFE_RELEASE(pd3dDevice);
  199. DestroyMeshContainer(pMeshContainer);
  200. return hr;
  201. }
  203. pMesh = pMeshContainer->MeshData.pMesh;
  204.         D3DXComputeNormals( pMesh, NULL );
  205.     }
  206.     else 
  207.     {
  208.         pMeshContainer->MeshData.pMesh = pMesh;
  209.         pMeshContainer->MeshData.Type = D3DXMESHTYPE_MESH;
  210.         pMesh->AddRef();
  211.     }
  213.     //为网格模型准备材质和纹理
  214.     pMeshContainer->NumMaterials = max(1, NumMaterials); 
  215.     pMeshContainer->pMaterials = new D3DXMATERIAL[pMeshContainer->NumMaterials];
  216.     pMeshContainer->ppTextures = new LPDIRECT3DTEXTURE9[pMeshContainer->NumMaterials];
  217.     pMeshContainer->pAdjacency = new DWORD[NumFaces*3];
  218.     if ((pMeshContainer->pAdjacency == NULL) || (pMeshContainer->pMaterials == NULL)
  219. || (pMeshContainer->ppTextures == NULL))
  220. {
  221. hr = E_OUTOFMEMORY;
  222. SAFE_RELEASE(pd3dDevice);
  223. DestroyMeshContainer(pMeshContainer);
  224. return hr;
  225. }
  227. memcpy(pMeshContainer->pAdjacency, pAdjacency, sizeof(DWORD) * NumFaces*3);  
  228.     memset(pMeshContainer->ppTextures, 0, sizeof(LPDIRECT3DTEXTURE9) * pMeshContainer->NumMaterials);
  230.    if (NumMaterials > 0)            
  231.     {    
  232.         //复制材质属性, 设置材质环境光属性
  233.         memcpy(pMeshContainer->pMaterials, pMaterials, sizeof(D3DXMATERIAL) * NumMaterials); 
  234. pMeshContainer->pMaterials->MatD3D.Ambient = pMeshContainer->pMaterials->MatD3D.Diffuse;
  235. // pMeshContainer->ppTextures = new LPDIRECT3DTEXTURE9[NumMaterials];
  236.        
  237. //提取纹理文件, 创建纹理对象
  238.         for (iMaterial = 0; iMaterial < NumMaterials; iMaterial++) 
  239.         {
  240.             if (pMeshContainer->pMaterials[iMaterial].pTextureFilename != NULL)
  241.             {
  242.                 if( FAILED( D3DXCreateTextureFromFileA( pd3dDevice, 
  243. (pMeshContainer->pMaterials[iMaterial].pTextureFilename),
  244.                     &pMeshContainer->ppTextures[iMaterial] ) ) )
  245.                     pMeshContainer->ppTextures[iMaterial] = NULL;
  247.                 pMeshContainer->pMaterials[iMaterial].pTextureFilename = NULL;
  248. }
  249.         }
  250.     }
  251.     else
  252.     {
  253.         pMeshContainer->pMaterials[0].pTextureFilename = NULL;
  254.         memset(&pMeshContainer->pMaterials[0].MatD3D, 0, sizeof(D3DMATERIAL9));
  255.         pMeshContainer->pMaterials[0].MatD3D.Diffuse.r = 0.5f;
  256.         pMeshContainer->pMaterials[0].MatD3D.Diffuse.g = 0.5f;
  257.         pMeshContainer->pMaterials[0].MatD3D.Diffuse.b = 0.5f;
  258.         pMeshContainer->pMaterials[0].MatD3D.Specular = pMeshContainer->pMaterials[0].MatD3D.Diffuse;
  259.     }
  261. // SAFE_RELEASE(pd3dDevice);
  263.     //如果当前网格包含蒙皮信息
  264.     if (pSkinInfo != NULL)
  265.     {
  266. //加载蒙皮网格信息
  267.         pMeshContainer->pSkinInfo = pSkinInfo; 
  268.         pSkinInfo->AddRef();
  270. //保留原网格信息
  271.         pMeshContainer->pOrigMesh = pMesh;
  272.         pMesh->AddRef();
  274. //获取骨骼数量
  275.         cBones = pSkinInfo->GetNumBones();
  277. //为每块骨骼分配保存初始变换矩阵的内存空间
  278.         pMeshContainer->pBoneOffsetMatrices = new D3DXMATRIX[cBones];
  279.         if (pMeshContainer->pBoneOffsetMatrices == NULL) 
  280.         {
  281. hr = E_OUTOFMEMORY;
  282. DestroyMeshContainer(pMeshContainer);
  283. return hr;
  284. }
  285.         
  286. //保存每块骨骼的初始变换矩阵
  287.         for (UINT iBone = 0; iBone < cBones; iBone++)
  288.         {
  289.             pMeshContainer->pBoneOffsetMatrices[iBone] = *(pMeshContainer->pSkinInfo->GetBoneOffsetMatrix(iBone));
  290.         }
  292. //生成蒙皮网格模型
  293.         hr = GenerateSkinnedMesh(pMeshContainer, pd3dDevice); 
  294.         if (FAILED(hr))
  295. {
  296. DestroyMeshContainer(pMeshContainer);
  297. return hr;
  298. }
  299.     }
  301.     *ppNewMeshContainer = pMeshContainer;
  302.     pMeshContainer = NULL;
  303.     return hr;
  304. }
  307. //-----------------------------------------------------------------------------
  308. // 释放框架
  309. //-----------------------------------------------------------------------------
  310. HRESULT CAllocateHierarchy::DestroyFrame(LPD3DXFRAME pFrameToFree) 
  311. {
  312. if(pFrameToFree != NULL)
  313. {
  314. SAFE_DELETE_ARRAY( pFrameToFree->Name );
  315. SAFE_DELETE( pFrameToFree );
  316. }
  317.     return S_OK; 
  318. }
  321. //-----------------------------------------------------------------------------
  322. // 释放模型容器
  323. //-----------------------------------------------------------------------------
  324. HRESULT CAllocateHierarchy::DestroyMeshContainer(LPD3DXMESHCONTAINER pMeshContainerBase)
  325. {
  326. if(pMeshContainerBase == NULL)
  327. return S_OK;
  329.     UINT iMaterial;
  330. // 先转为扩展型以免内存泄漏
  331.     D3DXMESHCONTAINER_DERIVED *pMeshContainer = (D3DXMESHCONTAINER_DERIVED*)pMeshContainerBase;
  333.     SAFE_DELETE_ARRAY( pMeshContainer->Name );
  334.     SAFE_DELETE_ARRAY( pMeshContainer->pAdjacency );
  335.     SAFE_DELETE_ARRAY( pMeshContainer->pMaterials );
  336.     SAFE_DELETE_ARRAY( pMeshContainer->pBoneOffsetMatrices );
  338.     if (pMeshContainer->ppTextures != NULL)
  339.     {
  340.         for (iMaterial = 0; iMaterial < pMeshContainer->NumMaterials; iMaterial++)
  341.         {
  342.             SAFE_RELEASE( pMeshContainer->ppTextures[iMaterial] );
  343.         }
  344.     }
  345.     SAFE_DELETE_ARRAY( pMeshContainer->ppTextures );
  347.     SAFE_DELETE_ARRAY( pMeshContainer->ppBoneMatrixPtrs );
  348.     SAFE_RELEASE( pMeshContainer->pBoneCombinationBuf );
  349.     SAFE_RELEASE( pMeshContainer->MeshData.pMesh );
  350.     SAFE_RELEASE( pMeshContainer->pSkinInfo );
  351.     SAFE_RELEASE( pMeshContainer->pOrigMesh );
  352.     SAFE_DELETE( pMeshContainer );
  353.     return S_OK;
  354. }