GPUCompute/App/Source/Shader.cpp at b8de310cde486de94566883f2064b656abbbc624 · TheCherno/GPUCompute · 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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
#include "Shader.h"

#include <iostream>
#include <fstream>
#include <vector>

#include <glad/gl.h>

uint32_t CreateComputeShader(const std::filesystem::path& path)
{
	std::ifstream file(path);

	if (!file.is_open())
	{
		std::cerr << "Failed to open file: " << path.string() << std::endl;
		return -1;
	}

	std::ostringstream contentStream;
	contentStream << file.rdbuf();
	std::string shaderSource = contentStream.str();

	GLuint shaderHandle = glCreateShader(GL_COMPUTE_SHADER);

	const GLchar* source = (const GLchar*)shaderSource.c_str();
	glShaderSource(shaderHandle, 1, &source, 0);

	glCompileShader(shaderHandle);

	GLint isCompiled = 0;
	glGetShaderiv(shaderHandle, GL_COMPILE_STATUS, &isCompiled);
	if (isCompiled == GL_FALSE)
	{
		GLint maxLength = 0;
		glGetShaderiv(shaderHandle, GL_INFO_LOG_LENGTH, &maxLength);

		std::vector<GLchar> infoLog(maxLength);
		glGetShaderInfoLog(shaderHandle, maxLength, &maxLength, &infoLog[0]);

		std::cerr << infoLog.data() << std::endl;

		glDeleteShader(shaderHandle);
		return -1;
	}

	GLuint program = glCreateProgram();
	glAttachShader(program, shaderHandle);
	glLinkProgram(program);

	GLint isLinked = 0;
	glGetProgramiv(program, GL_LINK_STATUS, (int*)&isLinked);
	if (isLinked == GL_FALSE)
	{
		GLint maxLength = 0;
		glGetProgramiv(program, GL_INFO_LOG_LENGTH, &maxLength);

		std::vector<GLchar> infoLog(maxLength);
		glGetProgramInfoLog(program, maxLength, &maxLength, &infoLog[0]);

		std::cerr << infoLog.data() << std::endl;

		glDeleteProgram(program);
		glDeleteShader(shaderHandle);

		return -1;
	}

	glDetachShader(program, shaderHandle);
	return program;
}

uint32_t ReloadComputeShader(uint32_t shaderHandle, const std::filesystem::path& path)
{
	uint32_t newShaderHandle = CreateComputeShader(path);

	// Return old shader if compilation failed
	if (newShaderHandle == -1)
		return shaderHandle;

	glDeleteProgram(shaderHandle);
	return newShaderHandle;
}