#include "Shader.h"
#include "Renderer.h"

#include "Platform/OpenGL/OpenGLShader.h"
#include "Platform/Vulkan/VulkanShader.h"

namespace Donut
{
    Shader* Shader::Create(const std::string& filepath) 
    {
        switch (Renderer::GetAPI())
        {
            case RendererAPI::API::OpenGL:
                return new OpenGLShader(filepath);
            case RendererAPI::API::Vulkan:
                return new VulkanShader(filepath);
            default:
                return nullptr;
        }
    }

    Shader* Shader::Create(const std::string& name, const std::string& vertexSrc, const std::string& fragmentSrc)
    {
        switch (Renderer::GetAPI()) 
        {
            case RendererAPI::API::OpenGL:
                return new OpenGLShader(name, vertexSrc, fragmentSrc);
            case RendererAPI::API::Vulkan:
                return new VulkanShader(name, vertexSrc, fragmentSrc);
            default:
                return nullptr;
        }
    }

    Shader* Shader::CreateCompute(const std::string& name, const std::string& computeSrc)
    {
        switch (Renderer::GetAPI()) 
        {
            case RendererAPI::API::OpenGL:
                return new OpenGLShader(name, computeSrc);
            case RendererAPI::API::Vulkan:
                return new VulkanShader(name, computeSrc);
            default:
                return nullptr;
        }
    }

    void ShaderLibrary::Add(const Ref<Shader>& shader)
    {
        auto& name = shader->GetName();
        Add(name, shader);
    }

    void ShaderLibrary::Add(const std::string& name, const Ref<Shader>& shader)
    {
        m_Shaders[name] = shader;
    }

    Ref<Shader> ShaderLibrary::Load(const std::string& filepath)
    {
        auto shader = Ref<Shader>(Shader::Create(filepath));
        Add(shader);
        return shader;
    }

    Ref<Shader> ShaderLibrary::Load(const std::string& name, const std::string& filepath)
    {
        auto shader = Ref<Shader>(Shader::Create(filepath));
        Add(name, shader);
        return shader;
    }

    Ref<Shader> ShaderLibrary::Get(const std::string& name)
    {
        if (Exists(name))
            return m_Shaders[name];
        return nullptr;
    }

    bool ShaderLibrary::Exists(const std::string& name) const
    {
        return m_Shaders.find(name) != m_Shaders.end();
    }
};