[磋商]改善办学保障条件-大兴校区学生宿舍11号楼配套项目——市政供电改造工程与环境改造工程（第二包）（二次）竞争性磋商公告

33/CoolFusion-2.0

/coolfusion/src/coolfusion/Shader.h
#include "Shader.h"

Shader::Shader()
{
m_VS = 0;
m_PS = 0;
m_VSPath = "";
m_PSPath = "";
m_ShaderType = 0;
m_ShaderData = "";
m_RendererID = 0;
}

Shader::~Shader()
{
unload();
}

bool Shader::loadFromFile(const char* vertexPath, const char* fragmentPath)
{
if (vertexPath == nullptr || fragmentPath == nullptr) {
return false;
}
m_VSPath = vertexPath;
m_PSPath = fragmentPath;
m_ShaderData = readFile(vertexPath) + readFile(fragmentPath);
m_ShaderType = GL_VERTEX_SHADER;
if (!compile(m_ShaderData)) {
return false;
}
m_ShaderType = GL_FRAGMENT_SHADER;
if (!compile(m_ShaderData)) {
return false;
}
m_RendererID = glCreateProgram();
glAttachShader(m_RendererID, m_VS);
glAttachShader(m_RendererID, m_PS);
glLinkProgram(m_RendererID);
checkCompileErrors(m_RendererID, "PROGRAM");
return true;
}

void Shader::load(const char* shaderData, unsigned int shaderType)
{
m_ShaderData = shaderData;
m_ShaderType = shaderType;
if (!compile(shaderData)) {
return;
}
m_RendererID = glCreateProgram();
glAttachShader(m_RendererID, m_VS);
glAttachShader(m_RendererID, m_PS);
glLinkProgram(m_RendererID);
checkCompileErrors(m_RendererID, "PROGRAM");
}

bool Shader::compile(const char* shaderData)
{
m_VSPath = "";
m_PSPath = "";
m_ShaderData = shaderData;
m_VS = createShader(shaderData, GL_VERTEX_SHADER);
m_PS = createShader(shaderData, GL_FRAGMENT_SHADER);
return true;
}

unsigned int Shader::createShader(const char* shaderData, unsigned int shaderType)
{
unsigned int ID = glCreateShader(shaderType);
const char* cShaderData = shaderData;
const char* line = cShaderData;
while (line != nullptr) {
line = strchr(line, '\n');
}
const char* lineData = line + 1;
const char* lineDataCopy = lineData;
glShaderSource(ID, 1, &lineData, nullptr);
glCompileShader(ID);
int success;
glGetShaderiv(ID, GL_COMPILE_STATUS, &success);
if (!success) {
char infoLog[512];
glGetShaderInfoLog(ID, 512, nullptr, infoLog);
std::cout << "ERROR::SHADER::COMPILATION_FAILED\n" << infoLog << std::endl;
return 0;
}
return ID;
}

void Shader::checkCompileErrors(unsigned int ID, const char* type)
{
int success;
char infoLog[512];
if (type != "PROGRAM") {
glGetShaderiv(ID, GL_COMPILE_STATUS, &success);
if (!success) {
glGetShaderInfoLog(ID, 512, nullptr, infoLog);
std::cout << type << " SHADER COMPILE FAILED\n" << infoLog << std::endl;
}
}
else {
glGetProgramiv(ID, GL_LINK_STATUS, &success);
if (!success) {
glGetProgramInfoLog(ID, 512, nullptr, infoLog);
std::cout << type << " PROGRAM LINK FAILED\n" << infoLog << std::endl;
}
}
}

void Shader::unload()
{
glDeleteProgram(m_RendererID);
m_RendererID = 0;
m_VS = 0;
m_PS = 0;
}

void Shader::setMatrixUniform(const char* uniformName, const mat4& matrix, bool update)
{
if (update) {
glUniformMatrix4fv(glGetUniformLocation(m_RendererID, uniformName), 1, GL_FALSE, glm::value_ptr(matrix));
}
}

void Shader::setBoolUniform(const char* uniformName, bool val, bool update)
{
if (update) {
glUniform1i(glGetUniformLocation(m_RendererID, uniformName), (int)val);
}
}

void Shader::setIntUniform(const char* uniformName, int val, bool update)
{
if (update) {
glUniform1i(glGetUniformLocation(m_RendererID, uniformName), val);
}
}

void Shader::setFloatUniform(const char* uniformName, float val, bool update)
{
if (update) {
glUniform1f(glGetUniformLocation(m_RendererID, uniformName), val);
}
}

void Shader::setIntVectorUniform(const char* uniformName, const int* data, int size, bool update)
{
if (update) {
glUniform1iv(glGetUniformLocation(m_RendererID, uniformName), size, data);
}
}

void Shader::setFloatVectorUniform(const char* uniformName, const float* data, int size, bool update)
{
if (update) {
glUniform4fv(glGetUniformLocation(m_RendererID, uniformName), size, data);
}
}

void Shader::setFloatMatrixUniform(const char* uniformName, const mat4* data, int size, bool update)
{
if (update) {
glUniformMatrix4fv(glGetUniformLocation(m_RendererID, uniformName), size, GL_FALSE, data);
}
}

void Shader::setIntMatrixUniform(const char* uniformName, const mat4* data, int size, bool update)
{
if (update) {
glUniformMatrix4fv(glGetUniformLocation(m_RendererID, uniformName), size, GL_FALSE, glm::value_ptr(data[0]));
}