126 lines
2.9 KiB
C
126 lines
2.9 KiB
C
|
#pragma once
|
|||
|
|
|
|||
|
|
#include <resource/resource.h>
|
|||
|
|
#include <glad/glad.h>
|
|||
|
|
#include <types/math/vec2.h>
|
|||
|
|
#include <types/math/vec3.h>
|
|||
|
|
#include <types/math/color.h>
|
|||
|
|
#include <types/base/types.h>
|
|||
|
|
#include <string>
|
|||
|
|
#include <unordered_map>
|
|||
|
|
|
|||
|
|
namespace extra2d {
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 着色器类型枚举
|
|||
|
|
*/
|
|||
|
|
enum class ShaderType : uint8 {
|
|||
|
|
Vertex, // 顶点着色器
|
|||
|
|
Fragment, // 片段着色器
|
|||
|
|
Geometry, // 几何着色器
|
|||
|
|
Compute // 计算着色器
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 着色器资源类
|
|||
|
|
*
|
|||
|
|
* 封装 OpenGL ES 3.2 着色器程序
|
|||
|
|
*/
|
|||
|
|
class Shader : public Resource {
|
|||
|
|
public:
|
|||
|
|
Shader();
|
|||
|
|
~Shader() override;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取资源类型
|
|||
|
|
*/
|
|||
|
|
ResourceType getType() const override { return ResourceType::Shader; }
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 从文件加载着色器
|
|||
|
|
* @param vsPath 顶点着色器文件路径
|
|||
|
|
* @param fsPath 片段着色器文件路径
|
|||
|
|
* @return 是否加载成功
|
|||
|
|
*/
|
|||
|
|
bool loadFromFile(const std::string& vsPath, const std::string& fsPath);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 从源码加载着色器
|
|||
|
|
* @param vsSource 顶点着色器源码
|
|||
|
|
* @param fsSource 片段着色器源码
|
|||
|
|
* @return 是否加载成功
|
|||
|
|
*/
|
|||
|
|
bool loadFromSource(const std::string& vsSource, const std::string& fsSource);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 使用着色器程序
|
|||
|
|
*/
|
|||
|
|
void use() const;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 停止使用着色器程序
|
|||
|
|
*/
|
|||
|
|
void unuse() const;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置布尔 Uniform
|
|||
|
|
*/
|
|||
|
|
void setBool(const std::string& name, bool value);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置整数 Uniform
|
|||
|
|
*/
|
|||
|
|
void setInt(const std::string& name, int value);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置浮点数 Uniform
|
|||
|
|
*/
|
|||
|
|
void setFloat(const std::string& name, float value);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置 Vec2 Uniform
|
|||
|
|
*/
|
|||
|
|
void setVec2(const std::string& name, const Vec2& value);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置 Vec3 Uniform
|
|||
|
|
*/
|
|||
|
|
void setVec3(const std::string& name, const Vec3& value);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置颜色 Uniform
|
|||
|
|
*/
|
|||
|
|
void setColor(const std::string& name, const Color& value);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置 4x4 矩阵 Uniform
|
|||
|
|
*/
|
|||
|
|
void setMat4(const std::string& name, const float* value);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取 OpenGL 程序句柄
|
|||
|
|
*/
|
|||
|
|
GLuint getProgram() const { return program_; }
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
GLuint program_ = 0; // OpenGL 程序句柄
|
|||
|
|
std::unordered_map<std::string, GLint> uniformCache_; // Uniform 位置缓存
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 编译着色器
|
|||
|
|
* @param type 着色器类型
|
|||
|
|
* @param source 源码
|
|||
|
|
* @return 着色器句柄，失败返回 0
|
|||
|
|
*/
|
|||
|
|
GLuint compileShader(GLenum type, const std::string& source);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取 Uniform 位置
|
|||
|
|
* @param name Uniform 名称
|
|||
|
|
* @return Uniform 位置
|
|||
|
|
*/
|
|||
|
|
GLint getUniformLocation(const std::string& name);
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
} // namespace extra2d
|