Material.cpp

Go to the documentation of this file.

#include <Runtime/Material.hpp>
#include <Graphics/Common/Shader.hpp>
#include <Graphics/Common/ResourceManager.hpp>
#include <Graphics/Common/ConstantBuffer.hpp>
 
namespace Sleak {
 
namespace {
void AdoptTexture(RefPtr<Texture>& slot, Texture* texture) {
    if (slot.get() != texture) slot = RefPtr<Texture>(texture);
}
}  // namespace
 
    Material::Material() : Object("Material") {}
 
    Material::~Material() = default;
 
    /// Lazily allocates the material's constant buffer on first use.
    void Material::Initialize() {
        if (!m_materialBuffer) {
            m_materialBuffer = ObjectPtr<RenderEngine::BufferBase>(
                RenderEngine::ResourceManager::CreateBuffer(
                    RenderEngine::BufferType::Constant,
                    sizeof(RenderEngine::MaterialGPUData),
                    nullptr));
            m_materialBuffer->SetSlot(1);
        }
    }
 
    /// Binds shader, all set textures, and uploads the packed material CB for a forward draw.
    void Material::Bind() {
        if (m_shader) {
            m_shader->bind();
        }
 
        if (m_diffuseTexture)
            m_diffuseTexture->Bind(TEXTURE_SLOT_DIFFUSE);
        if (m_normalTexture)
            m_normalTexture->Bind(TEXTURE_SLOT_NORMAL);
        if (m_specularTexture)
            m_specularTexture->Bind(TEXTURE_SLOT_SPECULAR);
        if (m_roughnessTexture)
            m_roughnessTexture->Bind(TEXTURE_SLOT_ROUGHNESS);
        if (m_metallicTexture)
            m_metallicTexture->Bind(TEXTURE_SLOT_METALLIC);
        if (m_aoTexture)
            m_aoTexture->Bind(TEXTURE_SLOT_AO);
        if (m_emissiveTexture)
            m_emissiveTexture->Bind(TEXTURE_SLOT_EMISSIVE);
 
        if (m_materialBuffer) {
            RenderEngine::MaterialGPUData data = BuildGPUData();
            m_materialBuffer->Update(&data, sizeof(data));
            m_materialBuffer->Update();
        }
    }
 
    /// Packs texture-presence flags and material scalars into the GPU constant buffer layout.
    RenderEngine::MaterialGPUData Material::BuildGPUData() const {
        RenderEngine::MaterialGPUData data = {};
 
        data.HasDiffuseMap   = m_diffuseTexture.IsValid()   ? 1u : 0u;
        data.HasNormalMap    = m_normalTexture.IsValid()     ? 1u : 0u;
        data.HasSpecularMap  = m_specularTexture.IsValid()   ? 1u : 0u;
        data.HasRoughnessMap = m_roughnessTexture.IsValid()  ? 1u : 0u;
        data.HasMetallicMap  = m_metallicTexture.IsValid()   ? 1u : 0u;
        data.HasAOMap        = m_aoTexture.IsValid()         ? 1u : 0u;
        data.HasEmissiveMap  = m_emissiveTexture.IsValid()   ? 1u : 0u;
        data._pad0 = 0;
 
        Math::Vector4D diffNorm = m_diffuseColor.normalize();
        data.DiffuseR = diffNorm.GetX();
        data.DiffuseG = diffNorm.GetY();
        data.DiffuseB = diffNorm.GetZ();
        data.DiffuseA = diffNorm.GetW();
 
        Math::Vector4D specNorm = m_specularColor.normalize();
        data.SpecularR = specNorm.GetX();
        data.SpecularG = specNorm.GetY();
        data.SpecularB = specNorm.GetZ();
        data.Shininess = m_shininess;
 
        Math::Vector4D emitNorm = m_emissiveColor.normalize();
        data.EmissiveR = emitNorm.GetX();
        data.EmissiveG = emitNorm.GetY();
        data.EmissiveB = emitNorm.GetZ();
        data.EmissiveIntensity = m_emissiveIntensity;
 
        data.Metallic        = m_metallic;
        data.Roughness       = m_roughness;
        data.AO              = m_ao;
        data.NormalIntensity  = m_normalIntensity;
 
        data.TilingX = m_tilingX;
        data.TilingY = m_tilingY;
        data.OffsetX = m_offsetX;
        data.OffsetY = m_offsetY;
 
        data.Opacity     = m_opacity;
        data.AlphaCutoff = m_alphaCutoff;
        data._pad1 = 0.0f;
        data._pad2 = 0.0f;
 
        return data;
    }
 
    /// True when this material must skip the GBuffer and render in the forward transparent pass.
    bool Material::IsForwardRendered() const {
        return m_renderMode == MaterialRenderMode::Transparent
            || m_opacity < 1.0f;
    }
 
    /// Binds textures and uploads the material CB for the deferred geometry pass, skipping the shader bind.
    void Material::BindTexturesAndCB() {
        // Same as Bind() but without shader->bind() — geometry pass overrides the shader.
        if (m_diffuseTexture)   m_diffuseTexture->Bind(TEXTURE_SLOT_DIFFUSE);
        if (m_normalTexture)    m_normalTexture->Bind(TEXTURE_SLOT_NORMAL);
        if (m_specularTexture)  m_specularTexture->Bind(TEXTURE_SLOT_SPECULAR);
        if (m_roughnessTexture) m_roughnessTexture->Bind(TEXTURE_SLOT_ROUGHNESS);
        if (m_metallicTexture)  m_metallicTexture->Bind(TEXTURE_SLOT_METALLIC);
        if (m_aoTexture)        m_aoTexture->Bind(TEXTURE_SLOT_AO);
        if (m_emissiveTexture)  m_emissiveTexture->Bind(TEXTURE_SLOT_EMISSIVE);
 
        if (m_materialBuffer) {
            RenderEngine::MaterialGPUData data = BuildGPUData();
            m_materialBuffer->Update(&data, sizeof(data));
            m_materialBuffer->Update();
        }
    }
 
    void Material::SetShader(RenderEngine::Shader* shader) {
        m_shader = ObjectPtr<RenderEngine::Shader>(shader);
    }
 
    void Material::SetShader(const std::string& shaderPath) {
        auto* shader =
            RenderEngine::ResourceManager::CreateShader(shaderPath);
        if (shader)
            m_shader = ObjectPtr<RenderEngine::Shader>(shader);
    }
 
    RenderEngine::Shader* Material::GetShader() const {
        return m_shader.IsValid() ? m_shader.operator->() : nullptr;
    }
 
    void Material::SetDiffuseTexture(const RefPtr<Texture>& texture) {
        m_diffuseTexture = texture;
    }
 
    void Material::SetDiffuseTexture(Texture* texture) {
        AdoptTexture(m_diffuseTexture, texture);
    }
 
    void Material::SetDiffuseTexture(const std::string& path) {
        m_diffuseTexture =
            RefPtr<Texture>(RenderEngine::ResourceManager::CreateTexture(path));
    }
 
    Texture* Material::GetDiffuseTexture() const {
        return m_diffuseTexture.get();
    }
 
    bool Material::HasDiffuseTexture() const {
        return m_diffuseTexture.IsValid();
    }
 
    void Material::SetNormalTexture(const RefPtr<Texture>& texture) {
        m_normalTexture = texture;
    }
 
    void Material::SetNormalTexture(Texture* texture) {
        AdoptTexture(m_normalTexture, texture);
    }
 
    void Material::SetNormalTexture(const std::string& path) {
        m_normalTexture =
            RefPtr<Texture>(RenderEngine::ResourceManager::CreateTexture(path));
    }
 
    Texture* Material::GetNormalTexture() const {
        return m_normalTexture.get();
    }
 
    bool Material::HasNormalTexture() const {
        return m_normalTexture.IsValid();
    }
 
    void Material::SetSpecularTexture(const RefPtr<Texture>& texture) {
        m_specularTexture = texture;
    }
 
    void Material::SetSpecularTexture(Texture* texture) {
        AdoptTexture(m_specularTexture, texture);
    }
 
    void Material::SetSpecularTexture(const std::string& path) {
        m_specularTexture =
            RefPtr<Texture>(RenderEngine::ResourceManager::CreateTexture(path));
    }
 
    Texture* Material::GetSpecularTexture() const {
        return m_specularTexture.get();
    }
 
    bool Material::HasSpecularTexture() const {
        return m_specularTexture.IsValid();
    }
 
    void Material::SetRoughnessTexture(const RefPtr<Texture>& texture) {
        m_roughnessTexture = texture;
    }
 
    void Material::SetRoughnessTexture(Texture* texture) {
        AdoptTexture(m_roughnessTexture, texture);
    }
 
    void Material::SetRoughnessTexture(const std::string& path) {
        m_roughnessTexture =
            RefPtr<Texture>(RenderEngine::ResourceManager::CreateTexture(path));
    }
 
    Texture* Material::GetRoughnessTexture() const {
        return m_roughnessTexture.get();
    }
 
    bool Material::HasRoughnessTexture() const {
        return m_roughnessTexture.IsValid();
    }
 
    void Material::SetMetallicTexture(const RefPtr<Texture>& texture) {
        m_metallicTexture = texture;
    }
 
    void Material::SetMetallicTexture(Texture* texture) {
        AdoptTexture(m_metallicTexture, texture);
    }
 
    void Material::SetMetallicTexture(const std::string& path) {
        m_metallicTexture =
            RefPtr<Texture>(RenderEngine::ResourceManager::CreateTexture(path));
    }
 
    Texture* Material::GetMetallicTexture() const {
        return m_metallicTexture.get();
    }
 
    bool Material::HasMetallicTexture() const {
        return m_metallicTexture.IsValid();
    }
 
    void Material::SetAOTexture(const RefPtr<Texture>& texture) {
        m_aoTexture = texture;
    }
 
    void Material::SetAOTexture(Texture* texture) {
        AdoptTexture(m_aoTexture, texture);
    }
 
    void Material::SetAOTexture(const std::string& path) {
        m_aoTexture =
            RefPtr<Texture>(RenderEngine::ResourceManager::CreateTexture(path));
    }
 
    Texture* Material::GetAOTexture() const { return m_aoTexture.get(); }
 
    bool Material::HasAOTexture() const {
        return m_aoTexture.IsValid();
    }
 
    void Material::SetEmissiveTexture(const RefPtr<Texture>& texture) {
        m_emissiveTexture = texture;
    }
 
    void Material::SetEmissiveTexture(Texture* texture) {
        AdoptTexture(m_emissiveTexture, texture);
    }
 
    void Material::SetEmissiveTexture(const std::string& path) {
        m_emissiveTexture =
            RefPtr<Texture>(RenderEngine::ResourceManager::CreateTexture(path));
    }
 
    Texture* Material::GetEmissiveTexture() const {
        return m_emissiveTexture.get();
    }
 
    bool Material::HasEmissiveTexture() const {
        return m_emissiveTexture.IsValid();
    }
 
    void Material::SetDiffuseColor(Math::Color color) {
        m_diffuseColor = color;
    }
 
    void Material::SetDiffuseColor(uint8_t r, uint8_t g, uint8_t b,
                                   uint8_t a) {
        m_diffuseColor = Math::Color(r, g, b, a);
    }
 
    void Material::SetDiffuseColor(float r, float g, float b, float a) {
        m_diffuseColor = Math::Color(
            static_cast<uint8_t>(r * 255.0f),
            static_cast<uint8_t>(g * 255.0f),
            static_cast<uint8_t>(b * 255.0f),
            static_cast<uint8_t>(a * 255.0f));
    }
 
    Math::Color Material::GetDiffuseColor() const { return m_diffuseColor; }
 
    void Material::SetSpecularColor(Math::Color color) {
        m_specularColor = color;
    }
 
    void Material::SetSpecularColor(uint8_t r, uint8_t g, uint8_t b,
                                    uint8_t a) {
        m_specularColor = Math::Color(r, g, b, a);
    }
 
    Math::Color Material::GetSpecularColor() const {
        return m_specularColor;
    }
 
    void Material::SetEmissiveColor(Math::Color color) {
        m_emissiveColor = color;
    }
 
    void Material::SetEmissiveColor(uint8_t r, uint8_t g, uint8_t b) {
        m_emissiveColor = Math::Color(r, g, b, 255);
    }
 
    void Material::SetEmissiveColor(float r, float g, float b) {
        m_emissiveColor = Math::Color(
            static_cast<uint8_t>(r * 255.0f),
            static_cast<uint8_t>(g * 255.0f),
            static_cast<uint8_t>(b * 255.0f), 255);
    }
 
    Math::Color Material::GetEmissiveColor() const {
        return m_emissiveColor;
    }
 
    void Material::SetShininess(float shininess) {
        m_shininess = shininess;
    }
 
    float Material::GetShininess() const { return m_shininess; }
 
    void Material::SetMetallic(float metallic) {
        m_metallic = metallic;
    }
 
    float Material::GetMetallic() const { return m_metallic; }
 
    void Material::SetRoughness(float roughness) {
        m_roughness = roughness;
    }
 
    float Material::GetRoughness() const { return m_roughness; }
 
    void Material::SetAO(float ao) { m_ao = ao; }
 
    float Material::GetAO() const { return m_ao; }
 
    void Material::SetNormalIntensity(float intensity) {
        m_normalIntensity = intensity;
    }
 
    float Material::GetNormalIntensity() const { return m_normalIntensity; }
 
    void Material::SetEmissiveIntensity(float intensity) {
        m_emissiveIntensity = intensity;
    }
 
    float Material::GetEmissiveIntensity() const {
        return m_emissiveIntensity;
    }
 
    void Material::SetOpacity(float opacity) { m_opacity = opacity; }
 
    float Material::GetOpacity() const { return m_opacity; }
 
    void Material::SetAlphaCutoff(float cutoff) {
        m_alphaCutoff = cutoff;
    }
 
    float Material::GetAlphaCutoff() const { return m_alphaCutoff; }
 
    void Material::SetTiling(float x, float y) {
        m_tilingX = x;
        m_tilingY = y;
    }
 
    void Material::SetTiling(Math::Vector2D tiling) {
        m_tilingX = tiling.GetX();
        m_tilingY = tiling.GetY();
    }
 
    Math::Vector2D Material::GetTiling() const {
        return Math::Vector2D(m_tilingX, m_tilingY);
    }
 
    void Material::SetOffset(float x, float y) {
        m_offsetX = x;
        m_offsetY = y;
    }
 
    void Material::SetOffset(Math::Vector2D offset) {
        m_offsetX = offset.GetX();
        m_offsetY = offset.GetY();
    }
 
    Math::Vector2D Material::GetOffset() const {
        return Math::Vector2D(m_offsetX, m_offsetY);
    }
 
    void Material::SetRenderMode(MaterialRenderMode mode) {
        m_renderMode = mode;
    }
 
    MaterialRenderMode Material::GetRenderMode() const {
        return m_renderMode;
    }
 
    void Material::SetTwoSided(bool twoSided) {
        m_twoSided = twoSided;
    }
 
    bool Material::IsTwoSided() const { return m_twoSided; }
 
}