65255f1321
Added the files provided for the tutorial Added the SplitMeshNode and SubMeshNode classes
93 lines
3.0 KiB
C++
93 lines
3.0 KiB
C++
#pragma once
|
|
#include "core.h"
|
|
#include "DirectXCore.h"
|
|
#include <vector>
|
|
|
|
// Core material class. Ideally, this should be extended to include more material attributes that can be
|
|
// recovered from Assimp, but this handles the basics.
|
|
|
|
class Material
|
|
{
|
|
public:
|
|
Material(wstring materialName, XMFLOAT4 diffuseColour, XMFLOAT4 specularColour, float shininess, float opacity, ComPtr<ID3D11ShaderResourceView> texture );
|
|
~Material();
|
|
|
|
inline wstring GetMaterialName() { return _materialName; }
|
|
inline XMFLOAT4 GetDiffuseColour() { return _diffuseColour; }
|
|
inline XMFLOAT4 GetSpecularColour() { return _specularColour; }
|
|
inline float GetShininess() { return _shininess; }
|
|
inline float GetOpacity() { return _opacity; }
|
|
inline ComPtr<ID3D11ShaderResourceView> GetTexture() { return _texture; }
|
|
|
|
private:
|
|
wstring _materialName;
|
|
XMFLOAT4 _diffuseColour;
|
|
XMFLOAT4 _specularColour;
|
|
float _shininess;
|
|
float _opacity;
|
|
ComPtr<ID3D11ShaderResourceView> _texture;
|
|
};
|
|
|
|
// Basic SubMesh class. A Mesh consists of one or more sub-meshes. The submesh provides everything that is needed to
|
|
// draw the sub-mesh.
|
|
|
|
class SubMesh
|
|
{
|
|
public:
|
|
SubMesh(ComPtr<ID3D11Buffer> vertexBuffer,
|
|
ComPtr<ID3D11Buffer> indexBuffer,
|
|
size_t vertexCount,
|
|
size_t indexCount,
|
|
shared_ptr<Material> material);
|
|
~SubMesh();
|
|
|
|
inline ComPtr<ID3D11Buffer> GetVertexBuffer() { return _vertexBuffer; }
|
|
inline ComPtr<ID3D11Buffer> GetIndexBuffer() { return _indexBuffer; }
|
|
inline shared_ptr<Material> GetMaterial() { return _material; }
|
|
inline size_t GetVertexCount() { return _vertexCount; }
|
|
inline size_t GetIndexCount() { return _indexCount; }
|
|
|
|
private:
|
|
ComPtr<ID3D11Buffer> _vertexBuffer;
|
|
ComPtr<ID3D11Buffer> _indexBuffer;
|
|
shared_ptr<Material> _material;
|
|
size_t _vertexCount;
|
|
size_t _indexCount;
|
|
};
|
|
|
|
// The core Mesh class. A Mesh corresponds to a scene in ASSIMP. A mesh consists of one or more sub-meshes.
|
|
|
|
class Node
|
|
{
|
|
public:
|
|
inline void SetName(wstring name) { _name = name; }
|
|
inline wstring GetName() { return _name; }
|
|
inline size_t GetMeshCount() { return _meshIndices.size(); }
|
|
inline unsigned int GetMesh(unsigned int index) { return _meshIndices[index]; }
|
|
inline void AddMesh(unsigned int meshIndex) { _meshIndices.push_back(meshIndex); }
|
|
inline size_t GetChildrenCount() { return _children.size(); }
|
|
inline shared_ptr<Node> GetChild(unsigned int index) { return _children[index]; }
|
|
inline void AddChild(shared_ptr<Node> node) { _children.push_back(node); }
|
|
|
|
private:
|
|
wstring _name;
|
|
vector<unsigned int> _meshIndices;
|
|
vector<shared_ptr<Node>> _children;
|
|
};
|
|
|
|
class Mesh
|
|
{
|
|
public:
|
|
size_t GetSubMeshCount();
|
|
shared_ptr<SubMesh> GetSubMesh(unsigned int i);
|
|
void AddSubMesh(shared_ptr<SubMesh> subMesh);
|
|
shared_ptr<Node> GetRootNode();
|
|
void SetRootNode(shared_ptr<Node> node);
|
|
|
|
private:
|
|
vector<shared_ptr<SubMesh>> _subMeshList;
|
|
shared_ptr<Node> _rootNode;
|
|
};
|
|
|
|
|