compute.cc

// clang-format off
// Must precede glfw/glad, to include OpenGL functions.
#include <qrk/quarkgl.h>
// clang-format on

#include <glm/glm.hpp>
#include <glm/gtc/matrix_transform.hpp>
#include <glm/gtc/type_ptr.hpp>

int main() {
  constexpr int width = 512, height = 512;

  qrk::Window win(width, height, "Compute shader", /* fullscreen */ false,
                  /* samples */ 4);
  win.setClearColor(glm::vec4(0.0f, 0.0f, 0.0f, 1.0f));
  win.setEscBehavior(qrk::EscBehavior::CLOSE);

  qrk::ComputeShader computeShader(
      qrk::ShaderPath("examples/shaders/compute.comp"));
  qrk::Texture computeTexture = qrk::Texture::create(width, height, GL_RGBA32F);
  qrk::ScreenQuadMesh screenQuad(computeTexture);
  qrk::ScreenShader screenShader;

  // Disable vsync to test speed.
  win.disableVsync();

  win.loop([&](float deltaTime) {
    // Run the compute shader.
    computeShader.updateUniforms();
    computeShader.dispatchToTexture(computeTexture);

    // Draw the results.
    screenQuad.draw(screenShader);

    // Print frame rate.
    if (win.getFrameCount() % 600 == 0) {
      printf("FPS: %.2f\n", 1 / deltaTime);
    }
  });

  return 0;
}