104 lines
2.6 KiB
C
104 lines
2.6 KiB
C
#include "main.h"
|
|
#include "gfx.h"
|
|
#include "clock.h"
|
|
#include "random.h"
|
|
|
|
#include "sphere.h"
|
|
#include "cam.h"
|
|
|
|
const int WIDTH = 420;
|
|
const int HEIGHT = 420;
|
|
|
|
void updateUniforms(GLuint shaderProgram);
|
|
|
|
int main()
|
|
{
|
|
randomInit();
|
|
|
|
// create a window and opengl context
|
|
SDL_Window* window = gfxInit(WIDTH, HEIGHT);
|
|
|
|
// generate noise
|
|
GLuint noise = createNoiseTexture(WIDTH, HEIGHT);
|
|
|
|
// create a texture for the compute shader to write to
|
|
GLuint textureOutput = createWriteOnlyTexture(WIDTH, HEIGHT);
|
|
printWorkGroupLimits();
|
|
|
|
// compile shader programs
|
|
unsigned int computeProgram = compileComputeShaderProgram(
|
|
"bin/rt.compute");
|
|
unsigned int quadProgram = compileQuadShaderProgram(
|
|
"bin/shader.vert",
|
|
"bin/shader.frag");
|
|
|
|
glBindTexture(GL_TEXTURE_2D, noise);
|
|
int noiseLoc = glGetUniformLocation(computeProgram, "_noise");
|
|
glUniform1i(noiseLoc, noise);
|
|
|
|
// initialise quad
|
|
initBuffers();
|
|
setVertexAttributes();
|
|
|
|
int frames = 0;
|
|
// render loop
|
|
while (!checkQuit())
|
|
{
|
|
// dispatch compute shader
|
|
glUseProgram(computeProgram);
|
|
updateUniforms(computeProgram);
|
|
glDispatchCompute((GLuint)WIDTH, (GLuint)HEIGHT, 1);
|
|
|
|
// make sure we're finished writing to the texture before trying to read it
|
|
glMemoryBarrier(GL_SHADER_IMAGE_ACCESS_BARRIER_BIT);
|
|
|
|
// normal drawing pass
|
|
glUseProgram(quadProgram);
|
|
|
|
// bind texture written to by compute stage to 2d target
|
|
glBindTexture(GL_TEXTURE_2D, textureOutput);
|
|
glDrawElements(GL_TRIANGLES, 6, GL_UNSIGNED_INT, 0);
|
|
|
|
// swip swap
|
|
SDL_GL_SwapWindow(window);
|
|
|
|
frames++;
|
|
}
|
|
|
|
float elapsed = now();
|
|
printf("%d frames in %f seconds (avg: %f fps)\n",
|
|
frames,
|
|
elapsed,
|
|
(float)frames/elapsed);
|
|
|
|
return 0;
|
|
}
|
|
|
|
void updateUniforms(GLuint shaderProgram)
|
|
{
|
|
// update random values
|
|
vec4 seed =
|
|
{
|
|
randomFloat(),
|
|
randomFloat(),
|
|
randomFloat(),
|
|
randomFloat()
|
|
};
|
|
int loc = glGetUniformLocation(shaderProgram, "_seed");
|
|
glUniform4fv(loc, 1, seed);
|
|
|
|
float t = now();
|
|
float sin_t = sin(t);
|
|
int tLocation = glGetUniformLocation(shaderProgram, "_t");
|
|
glUniform4f(tLocation, t, sin_t, (1.0 + sin_t)*0.5, 0.0f);
|
|
|
|
float aspect = (float)WIDTH/(float)HEIGHT;
|
|
updateCameraUniforms(shaderProgram, aspect);
|
|
|
|
const int sphereCount = 42;
|
|
struct Sphere spheres[sphereCount];
|
|
makeSpheres(spheres, sphereCount);
|
|
|
|
updateSphereUniforms(shaderProgram, spheres, sphereCount);
|
|
}
|