mirror of
https://github.com/TheRedShip/RT_GPU.git
synced 2025-09-28 02:58:35 +02:00
+ | Shader class system better working
This commit is contained in:
TheRedShip
187
srcs/RT.cpp
187
srcs/RT.cpp
@ -12,172 +12,77 @@
|
||||
|
||||
#include "RT.hpp"
|
||||
|
||||
void setupScreenTriangle(GLuint *VAO);
|
||||
void drawScreenTriangle(GLuint VAO, GLuint output_texture, GLuint program);
|
||||
std::vector<GLuint> generateTextures(unsigned int textures_count);
|
||||
std::vector<Buffer *> createDataOnGPU(Scene &scene);
|
||||
void updateDataOnGPU(Scene &scene, std::vector<Buffer *> buffers);
|
||||
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
Arguments args(argc, argv);
|
||||
if (args.error())
|
||||
return (1);
|
||||
|
||||
Scene scene(args.getSceneName());
|
||||
if (scene.fail())
|
||||
return (1);
|
||||
|
||||
Window window(&scene, WIDTH, HEIGHT, "RT_GPU", 0, args);
|
||||
Shader shader("shaders/vertex.vert", "shaders/frag.frag", "shaders/compute.glsl", "shaders/denoising.glsl");
|
||||
// Shader shader("shaders/vertex.vert", "shaders/frag.frag", "shaders/debug.glsl");
|
||||
|
||||
GLuint VAO;
|
||||
setupScreenTriangle(&VAO);
|
||||
|
||||
std::vector<GLuint> textures = generateTextures(2);
|
||||
GLuint output_texture = textures[0];
|
||||
|
||||
ShaderProgram raytracing_program;
|
||||
Shader compute = Shader(GL_COMPUTE_SHADER, "shaders/compute.glsl");
|
||||
|
||||
raytracing_program.attachShader(&compute);
|
||||
raytracing_program.link();
|
||||
|
||||
raytracing_program.use();
|
||||
// raytracing_program.bindImageTexture(output_texture, 0, GL_READ_WRITE, GL_RGBA32F);
|
||||
// raytracing_program.bindImageTexture(textures[1], 1, GL_READ_WRITE, GL_RGBA32F);
|
||||
|
||||
|
||||
GLint max_gpu_size;
|
||||
glGetIntegerv(GL_MAX_SHADER_STORAGE_BLOCK_SIZE, &max_gpu_size);
|
||||
ShaderProgram render_program;
|
||||
Shader vertex = Shader(GL_VERTEX_SHADER, "shaders/vertex.vert");
|
||||
Shader frag = Shader(GL_FRAGMENT_SHADER, "shaders/frag.frag");
|
||||
|
||||
const std::vector<GPUObject> &object_data = scene.getObjectData();
|
||||
const std::vector<GPUTriangle> &triangle_data = scene.getTriangleData();
|
||||
const std::vector<GPUBvh> &bvh_nodes = scene.getBvh();
|
||||
const std::vector<GPUBvhData> &bvh_data = scene.getBvhData();
|
||||
const std::vector<GPUMaterial> &material_data = scene.getMaterialData();
|
||||
render_program.attachShader(&vertex);
|
||||
render_program.attachShader(&frag);
|
||||
render_program.link();
|
||||
|
||||
std::cout << "Sending " << object_data.size() << " objects for " << \
|
||||
object_data.size() * sizeof(GPUObject) + \
|
||||
triangle_data.size() * sizeof(GPUTriangle) + \
|
||||
bvh_nodes.size() * sizeof(GPUBvh) + \
|
||||
material_data.size() * sizeof(GPUMaterial) \
|
||||
<< " / " << max_gpu_size << " bytes" << std::endl;
|
||||
|
||||
GLuint objectSSBO;
|
||||
glGenBuffers(1, &objectSSBO);
|
||||
glBindBuffer(GL_SHADER_STORAGE_BUFFER, objectSSBO);
|
||||
glBufferData(GL_SHADER_STORAGE_BUFFER, sizeof(GPUObject) * object_data.size(), object_data.data(), GL_STATIC_DRAW);
|
||||
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, objectSSBO);
|
||||
|
||||
GLuint trianglesSSBO;
|
||||
glGenBuffers(1, &trianglesSSBO);
|
||||
glBindBuffer(GL_SHADER_STORAGE_BUFFER, trianglesSSBO);
|
||||
glBufferData(GL_SHADER_STORAGE_BUFFER, sizeof(GPUTriangle) * triangle_data.size(), triangle_data.data(), GL_STATIC_DRAW);
|
||||
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 2, trianglesSSBO);
|
||||
|
||||
GLuint bvh_nodesSSBO;
|
||||
glGenBuffers(1, &bvh_nodesSSBO);
|
||||
glBindBuffer(GL_SHADER_STORAGE_BUFFER, bvh_nodesSSBO);
|
||||
glBufferData(GL_SHADER_STORAGE_BUFFER, sizeof(GPUBvhData) * bvh_data.size(), bvh_data.data(), GL_STATIC_DRAW);
|
||||
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 3, bvh_nodesSSBO);
|
||||
|
||||
GLuint bvhSSBO;
|
||||
glGenBuffers(1, &bvhSSBO);
|
||||
glBindBuffer(GL_SHADER_STORAGE_BUFFER, bvhSSBO);
|
||||
glBufferData(GL_SHADER_STORAGE_BUFFER, sizeof(GPUBvh) * bvh_nodes.size(), bvh_nodes.data(), GL_STATIC_DRAW);
|
||||
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 4, bvhSSBO);
|
||||
|
||||
GLuint materialSSBO;
|
||||
glGenBuffers(1, &materialSSBO);
|
||||
glBindBuffer(GL_SHADER_STORAGE_BUFFER, materialSSBO);
|
||||
glBufferData(GL_SHADER_STORAGE_BUFFER, sizeof(GPUMaterial) * material_data.size(), nullptr, GL_STATIC_DRAW);
|
||||
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 5, materialSSBO);
|
||||
|
||||
GLuint lightSSBO;
|
||||
glGenBuffers(1, &lightSSBO);
|
||||
glBindBuffer(GL_SHADER_STORAGE_BUFFER, lightSSBO);
|
||||
glBufferData(GL_SHADER_STORAGE_BUFFER, scene.getGPULights().size() * sizeof(int), nullptr, GL_STATIC_DRAW);
|
||||
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 6, lightSSBO);
|
||||
|
||||
GLuint cameraUBO;
|
||||
glGenBuffers(1, &cameraUBO);
|
||||
glBindBuffer(GL_UNIFORM_BUFFER, cameraUBO);
|
||||
glBufferData(GL_UNIFORM_BUFFER, sizeof(GPUCamera), nullptr, GL_DYNAMIC_DRAW);
|
||||
glBindBufferBase(GL_UNIFORM_BUFFER, 0, cameraUBO);
|
||||
|
||||
GLuint volumeUBO;
|
||||
glGenBuffers(1, &volumeUBO);
|
||||
glBindBuffer(GL_UNIFORM_BUFFER, volumeUBO);
|
||||
glBufferData(GL_UNIFORM_BUFFER, sizeof(GPUVolume), nullptr, GL_STATIC_DRAW);
|
||||
glBindBufferBase(GL_UNIFORM_BUFFER, 1, volumeUBO);
|
||||
|
||||
GLuint debugUBO;
|
||||
glGenBuffers(1, &debugUBO);
|
||||
glBindBuffer(GL_UNIFORM_BUFFER, debugUBO);
|
||||
glBufferData(GL_UNIFORM_BUFFER, sizeof(GPUDebug), nullptr, GL_STATIC_DRAW);
|
||||
glBindBufferBase(GL_UNIFORM_BUFFER, 2, debugUBO);
|
||||
|
||||
if (!scene.loadTextures())
|
||||
return (1);
|
||||
|
||||
shader.attach();
|
||||
|
||||
shader.setupVertexBuffer();
|
||||
std::vector<Buffer *> buffers = createDataOnGPU(scene);
|
||||
|
||||
while (!window.shouldClose())
|
||||
{
|
||||
window.updateDeltaTime();
|
||||
|
||||
glUseProgram(shader.getProgramCompute());
|
||||
|
||||
glBindBuffer(GL_SHADER_STORAGE_BUFFER, materialSSBO);
|
||||
glBufferSubData(GL_SHADER_STORAGE_BUFFER, 0, material_data.size() * sizeof(GPUMaterial), material_data.data());
|
||||
|
||||
std::set<int> gpu_lights = scene.getGPULights();
|
||||
std::vector<int> gpu_lights_array(gpu_lights.begin(), gpu_lights.end());
|
||||
glBindBuffer(GL_SHADER_STORAGE_BUFFER, lightSSBO);
|
||||
glBufferSubData(GL_SHADER_STORAGE_BUFFER, 0, gpu_lights_array.size() * sizeof(int), gpu_lights_array.data());
|
||||
window.rendererUpdate(shader);
|
||||
|
||||
GPUCamera camera_data = scene.getCamera()->getGPUData();
|
||||
|
||||
glBindBuffer(GL_UNIFORM_BUFFER, cameraUBO);
|
||||
glBufferSubData(GL_UNIFORM_BUFFER, 0, sizeof(GPUCamera), &camera_data);
|
||||
|
||||
glBindBuffer(GL_UNIFORM_BUFFER, volumeUBO);
|
||||
glBufferSubData(GL_UNIFORM_BUFFER, 0, sizeof(GPUVolume), &scene.getVolume());
|
||||
|
||||
glBindBuffer(GL_UNIFORM_BUFFER, debugUBO);
|
||||
glBufferSubData(GL_UNIFORM_BUFFER, 0, sizeof(GPUDebug), &scene.getDebug());
|
||||
|
||||
shader.set_int("u_frameCount", window.getFrameCount());
|
||||
shader.set_int("u_objectsNum", object_data.size());
|
||||
shader.set_int("u_bvhNum", bvh_data.size());
|
||||
shader.set_int("u_lightsNum", gpu_lights.size());
|
||||
shader.set_int("u_pixelisation", window.getPixelisation());
|
||||
shader.set_float("u_time", (float)(glfwGetTime()));
|
||||
shader.set_vec2("u_resolution", glm::vec2(WIDTH, HEIGHT));
|
||||
|
||||
shader.set_textures(scene.getTextureIDs(), scene.getEmissionTextureIDs());
|
||||
|
||||
glDispatchCompute((WIDTH + 15) / 16, (HEIGHT + 15) / 16, 1);
|
||||
glMemoryBarrier(GL_SHADER_IMAGE_ACCESS_BARRIER_BIT);
|
||||
|
||||
GPUDenoise denoise = scene.getDenoise();
|
||||
if (denoise.enabled)
|
||||
{
|
||||
glUseProgram(shader.getProgramComputeDenoising());
|
||||
|
||||
glUniform2fv(glGetUniformLocation(shader.getProgramComputeDenoising(), "u_resolution"), 1, glm::value_ptr(glm::vec2(WIDTH, HEIGHT)));
|
||||
glUniform1f(glGetUniformLocation(shader.getProgramComputeDenoising(), "u_c_phi"), denoise.c_phi);
|
||||
glUniform1f(glGetUniformLocation(shader.getProgramComputeDenoising(), "u_p_phi"), denoise.p_phi);
|
||||
glUniform1f(glGetUniformLocation(shader.getProgramComputeDenoising(), "u_n_phi"), denoise.n_phi);
|
||||
|
||||
for (int pass = 0; pass < denoise.pass ; ++pass)
|
||||
{
|
||||
shader.flipOutputDenoising(pass % 2 == 0);
|
||||
|
||||
glUniform1i(glGetUniformLocation(shader.getProgramComputeDenoising(), "u_pass"), pass);
|
||||
|
||||
glDispatchCompute((WIDTH + 15) / 16, (HEIGHT + 15) / 16, 1);
|
||||
glMemoryBarrier(GL_SHADER_IMAGE_ACCESS_BARRIER_BIT);
|
||||
}
|
||||
shader.flipOutputDenoising(true);
|
||||
}
|
||||
|
||||
|
||||
glClear(GL_COLOR_BUFFER_BIT);
|
||||
|
||||
|
||||
updateDataOnGPU(scene, buffers);
|
||||
|
||||
raytracing_program.use();
|
||||
raytracing_program.set_int("u_frameCount", window.getFrameCount());
|
||||
raytracing_program.set_int("u_objectsNum", scene.getObjectData().size());
|
||||
raytracing_program.set_int("u_bvhNum", scene.getBvhData().size());
|
||||
raytracing_program.set_int("u_lightsNum", scene.getGPULights().size());
|
||||
raytracing_program.set_int("u_pixelisation", window.getPixelisation());
|
||||
raytracing_program.set_float("u_time", (float)(glfwGetTime()));
|
||||
raytracing_program.set_vec2("u_resolution", glm::vec2(WIDTH, HEIGHT));
|
||||
raytracing_program.dispathCompute((WIDTH + 15) / 16, (HEIGHT + 15) / 16, 1);
|
||||
|
||||
window.imGuiNewFrame();
|
||||
|
||||
glUseProgram(shader.getProgram());
|
||||
shader.drawTriangles();
|
||||
render_program.use();
|
||||
drawScreenTriangle(VAO, output_texture, render_program.getProgram());
|
||||
|
||||
window.imGuiRender();
|
||||
|
||||
window.display();
|
||||
window.pollEvents();
|
||||
|
||||
glClearTexImage(shader.getNormalTexture(), 0, GL_RGBA, GL_UNSIGNED_BYTE, NULL);
|
||||
glClearTexImage(shader.getPositionTexture(), 0, GL_RGBA, GL_UNSIGNED_BYTE, NULL);
|
||||
}
|
||||
|
||||
ImGui_ImplOpenGL3_Shutdown();
|
||||
|
||||
Reference in New Issue
Block a user