reed-framework/mesh.cpp at master · Reedbeta/reed-framework · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#include "framework.h"

namespace Framework
{
	Mesh::Mesh()
	:	m_pVerts(nullptr),
		m_pIndices(nullptr),
		m_vertCount(0),
		m_indexCount(0),
		m_vtxStrideBytes(0),
		m_primtopo(D3D11_PRIMITIVE_TOPOLOGY_UNDEFINED),
		m_bounds(empty)
	{
	}

	void Mesh::Draw(ID3D11DeviceContext * pCtx)
	{
		ASSERT_ERR(pCtx);

		UINT zero = 0;
		pCtx->IASetVertexBuffers(0, 1, &m_pVtxBuffer, (UINT *)&m_vtxStrideBytes, &zero);
		pCtx->IASetIndexBuffer(m_pIdxBuffer, DXGI_FORMAT_R32_UINT, 0);
		pCtx->IASetPrimitiveTopology(m_primtopo);
		pCtx->DrawIndexed(m_indexCount, 0, 0);
	}

	void Mesh::DrawMtlRange(ID3D11DeviceContext * pCtx, int iMtlRange)
	{
		ASSERT_ERR(pCtx);
		ASSERT_ERR(iMtlRange >= 0 && iMtlRange < int(m_mtlRanges.size()));

		const MtlRange * pRange = &m_mtlRanges[iMtlRange];

		UINT zero = 0;
		pCtx->IASetVertexBuffers(0, 1, &m_pVtxBuffer, (UINT *)&m_vtxStrideBytes, &zero);
		pCtx->IASetIndexBuffer(m_pIdxBuffer, DXGI_FORMAT_R32_UINT, 0);
		pCtx->IASetPrimitiveTopology(m_primtopo);
		pCtx->DrawIndexed(pRange->m_indexCount, pRange->m_indexStart, 0);
	}

	void Mesh::Reset()
	{
		m_pPack.release();
		m_pVerts = nullptr;
		m_pIndices = nullptr;
		m_vertCount = 0;
		m_indexCount = 0;
		m_pVtxBuffer.release();
		m_pIdxBuffer.release();
		m_vtxStrideBytes = 0;
		m_primtopo = D3D11_PRIMITIVE_TOPOLOGY_UNDEFINED;
		m_bounds = box3(empty);
	}

	void Mesh::UploadToGPU(ID3D11Device * pDevice)
	{
		ASSERT_ERR(pDevice);

		m_pVtxBuffer.release();
		m_pIdxBuffer.release();

		D3D11_BUFFER_DESC vtxBufferDesc =
		{
			sizeof(Vertex) * m_vertCount,
			D3D11_USAGE_IMMUTABLE,
			D3D11_BIND_VERTEX_BUFFER,
			0,	// no cpu access
			0,	// no misc flags
			0,	// structured buffer stride
		};
		D3D11_SUBRESOURCE_DATA vtxBufferData = { m_pVerts, 0, 0 };
		CHECK_D3D(pDevice->CreateBuffer(&vtxBufferDesc, &vtxBufferData, &m_pVtxBuffer));

		D3D11_BUFFER_DESC idxBufferDesc =
		{
			sizeof(int) * m_indexCount,
			D3D11_USAGE_IMMUTABLE,
			D3D11_BIND_INDEX_BUFFER,
			0,	// no cpu access
			0,	// no misc flags
			0,	// structured buffer stride
		};
		D3D11_SUBRESOURCE_DATA idxBufferData = { m_pIndices, 0, 0 };
		CHECK_D3D(pDevice->CreateBuffer(&idxBufferDesc, &idxBufferData, &m_pIdxBuffer));

		m_vtxStrideBytes = sizeof(Vertex);
		m_primtopo = D3D11_PRIMITIVE_TOPOLOGY_TRIANGLELIST;
	}
}