mirror of
https://github.com/TheRedShip/RT_GPU.git
synced 2025-09-27 10:48:34 +02:00
132 lines
4.6 KiB
C++
132 lines
4.6 KiB
C++
/* ************************************************************************** */
|
|
/* */
|
|
/* ::: :::::::: */
|
|
/* RT.cpp :+: :+: :+: */
|
|
/* +:+ +:+ +:+ */
|
|
/* By: ycontre <ycontre@student.42.fr> +#+ +:+ +#+ */
|
|
/* +#+#+#+#+#+ +#+ */
|
|
/* Created: 2024/09/27 14:51:49 by TheRed #+# #+# */
|
|
/* Updated: 2025/02/13 19:22:02 by ycontre ### ########.fr */
|
|
/* */
|
|
/* ************************************************************************** */
|
|
|
|
#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);
|
|
|
|
void shaderDenoise(ShaderProgram &denoising_program, GPUDenoise &denoise, std::vector<GLuint> textures)
|
|
{
|
|
denoising_program.use();
|
|
|
|
denoising_program.set_vec2("u_resolution", glm::vec2(WIDTH, HEIGHT));
|
|
denoising_program.set_float("u_c_phi", denoise.c_phi);
|
|
denoising_program.set_float("u_p_phi", denoise.p_phi);
|
|
denoising_program.set_float("u_n_phi", denoise.n_phi);
|
|
|
|
int output_texture = 0;
|
|
int denoising_texture = 2;
|
|
|
|
for (int pass = 0; pass < denoise.pass ; ++pass)
|
|
{
|
|
glBindImageTexture(0, textures[output_texture], 0, GL_FALSE, 0, GL_READ_WRITE, GL_RGBA32F);
|
|
glBindImageTexture(2, textures[denoising_texture], 0, GL_FALSE, 0, GL_READ_WRITE, GL_RGBA32F);
|
|
|
|
denoising_program.set_int("u_pass", pass);
|
|
denoising_program.dispathCompute((WIDTH + 15) / 16, (HEIGHT + 15) / 16, 1);
|
|
|
|
std::swap(output_texture, denoising_texture);
|
|
}
|
|
|
|
glBindImageTexture(0, textures[output_texture], 0, GL_FALSE, 0, GL_READ_WRITE, GL_RGBA32F);
|
|
}
|
|
|
|
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);
|
|
|
|
GLuint VAO;
|
|
setupScreenTriangle(&VAO);
|
|
|
|
std::vector<GLuint> textures = generateTextures(5);
|
|
|
|
ShaderProgram raytracing_program;
|
|
Shader compute = Shader(GL_COMPUTE_SHADER, "shaders/compute.glsl");
|
|
raytracing_program.attachShader(&compute);
|
|
raytracing_program.link();
|
|
|
|
ShaderProgram denoising_program;
|
|
Shader denoise = Shader(GL_COMPUTE_SHADER, "shaders/denoising.glsl");
|
|
denoising_program.attachShader(&denoise);
|
|
denoising_program.link();
|
|
|
|
ShaderProgram render_program;
|
|
Shader vertex = Shader(GL_VERTEX_SHADER, "shaders/vertex.vert");
|
|
Shader frag = Shader(GL_FRAGMENT_SHADER, "shaders/frag.frag");
|
|
render_program.attachShader(&vertex);
|
|
render_program.attachShader(&frag);
|
|
render_program.link();
|
|
|
|
std::vector<Buffer *> buffers = createDataOnGPU(scene);
|
|
|
|
if (!scene.loadTextures())
|
|
return (-1);
|
|
|
|
while (!window.shouldClose())
|
|
{
|
|
window.updateDeltaTime();
|
|
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));
|
|
|
|
std::map<std::string, std::vector<GLuint>> object_textures;
|
|
object_textures["textures"] = scene.getTextureIDs();
|
|
object_textures["emissive_textures"] = scene.getEmissionTextureIDs();
|
|
raytracing_program.set_textures(object_textures);
|
|
|
|
raytracing_program.dispathCompute((WIDTH + 15) / 16, (HEIGHT + 15) / 16, 1);
|
|
|
|
if (scene.getDenoise().enabled)
|
|
shaderDenoise(denoising_program, scene.getDenoise(), textures);
|
|
|
|
window.imGuiNewFrame();
|
|
|
|
render_program.use();
|
|
drawScreenTriangle(VAO, textures[0], render_program.getProgram());
|
|
|
|
window.imGuiRender(raytracing_program);
|
|
|
|
window.display();
|
|
window.pollEvents();
|
|
|
|
glClearTexImage(textures[3], 0, GL_RGBA, GL_FLOAT, nullptr);
|
|
glClearTexImage(textures[4], 0, GL_RGBA, GL_FLOAT, nullptr);
|
|
}
|
|
|
|
ImGui_ImplOpenGL3_Shutdown();
|
|
ImGui_ImplGlfw_Shutdown();
|
|
ImGui::DestroyContext();
|
|
|
|
return (0);
|
|
}
|