MyGame/Material.cpp at master · saladas3/MyGame · 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
#include "Material.h"

Material::Material(const wchar_t* vertex_shader_path, const wchar_t* pixel_shader_path)
{
    void* shader_byte_code = nullptr;
    size_t size_shader = 0;

    // Create vertex shader for material
    GraphicsEngine::get()->getRenderSystem()->compileVertexShader(
        vertex_shader_path, "vsmain", &shader_byte_code, &size_shader
    );
    this->mVertexShader = GraphicsEngine::get()->getRenderSystem()->createVertexShader(
        shader_byte_code, size_shader
    );
    GraphicsEngine::get()->getRenderSystem()->releaseCompiledShader();
    if (!mVertexShader) throw std::runtime_error("Could not create vertex shader for material");

    // Create pixel shader for material
    GraphicsEngine::get()->getRenderSystem()->compilePixelShader(
        pixel_shader_path, "psmain", &shader_byte_code, &size_shader
    );
    this->mPixelShader = GraphicsEngine::get()->getRenderSystem()->createPixelShader(
        shader_byte_code, size_shader
    );
    GraphicsEngine::get()->getRenderSystem()->releaseCompiledShader();
    if (!mPixelShader) throw std::runtime_error("Could not create pixel shader for material");
}

Material::Material(const MaterialPtr& material)
{
    this->mVertexShader = material->mVertexShader;
    this->mPixelShader = material->mPixelShader;
}

Material::~Material() = default;

void Material::addTexture(const TexturePtr& texture)
{
    this->mVecTextures.push_back(texture);
}

void Material::removeTexture(int index)
{
    if (index >= this->mVecTextures.size()) return;
    mVecTextures.erase(mVecTextures.begin() + index);
}

void Material::setData(void* data, unsigned int size)
{
    if (!mConstantBuffer)
        this->mConstantBuffer = GraphicsEngine::get()->getRenderSystem()->createConstantBuffer(
            data, size
        );
    else
        this->mConstantBuffer->update(
            GraphicsEngine::get()->getRenderSystem()->getImmediateDeviceContext(), data
        );
}

void Material::setCullMode(CULL_MODE mode)
{
    this->mCullMode = mode;
}

CULL_MODE Material::getCullMode(CULL_MODE mode)
{
	return this->mCullMode;
}