Compare commits

...

2 Commits
v1.3 ... main

  1. 9
      src/GLWidget.cpp
  2. 4
      src/GLWidget.h
  3. 9
      src/MainWindow.cpp
  4. 2
      src/MainWindow.h
  5. 102
      src/Simulation.cpp
  6. 12
      src/Simulation.h
  7. 5
      src/main.cpp

@ -46,6 +46,7 @@ void GLWidget::initializeGL() {
simulation->semaphore.acquire(); simulation->semaphore.acquire();
uploadStaticDataToGPU(); uploadStaticDataToGPU();
makeCurrent();
overlay->init(); overlay->init();
} }
@ -102,6 +103,8 @@ bool GLWidget::AnyDialogOpen() {
} }
void GLWidget::uploadStaticDataToGPU() { void GLWidget::uploadStaticDataToGPU() {
makeCurrent();
auto pendula = &simulation->pendula; auto pendula = &simulation->pendula;
size_t pointCount = std::transform_reduce(pendula->begin(), pendula->end(), 0, [](size_t prev, size_t curr){ size_t pointCount = std::transform_reduce(pendula->begin(), pendula->end(), 0, [](size_t prev, size_t curr){
@ -183,10 +186,14 @@ void GLWidget::uploadStaticDataToGPU() {
glBindVertexArray(0); glBindVertexArray(0);
doneCurrent();
simulation->semaphore.release(); simulation->semaphore.release();
} }
void GLWidget::changePosition() { void GLWidget::changePosition() {
makeCurrent();
auto t1 = system_clock::now(); auto t1 = system_clock::now();
glBindBuffer(GL_ARRAY_BUFFER, positionVBO); glBindBuffer(GL_ARRAY_BUFFER, positionVBO);
auto positions = (GLfloat*) glMapBufferRange(GL_ARRAY_BUFFER, 0, GLsizeiptr(positionCount * sizeof(float)), GL_MAP_WRITE_BIT); auto positions = (GLfloat*) glMapBufferRange(GL_ARRAY_BUFFER, 0, GLsizeiptr(positionCount * sizeof(float)), GL_MAP_WRITE_BIT);
@ -201,6 +208,8 @@ void GLWidget::changePosition() {
} }
glUnmapBuffer(GL_ARRAY_BUFFER); glUnmapBuffer(GL_ARRAY_BUFFER);
// std::cout << duration_cast<microseconds>(system_clock::now() - t1).count() << std::endl; // std::cout << duration_cast<microseconds>(system_clock::now() - t1).count() << std::endl;
doneCurrent();
simulation->semaphore.release(); simulation->semaphore.release();
} }

@ -1,7 +1,7 @@
#pragma once #pragma once
#include <QOpenGLWidget> #include <QOpenGLWidget>
#include <QOpenGLExtraFunctions> #include <QOpenGLFunctions_3_3_Core>
#include <QMatrix3x3> #include <QMatrix3x3>
class Pendulum; class Pendulum;
@ -9,7 +9,7 @@ class Simulation;
class QOpenGLShaderProgram; class QOpenGLShaderProgram;
class Overlay; class Overlay;
class GLWidget : public QOpenGLWidget, protected QOpenGLExtraFunctions { class GLWidget : public QOpenGLWidget, protected QOpenGLFunctions_3_3_Core {
public: public:
explicit GLWidget(Simulation *); explicit GLWidget(Simulation *);
Overlay * overlay; Overlay * overlay;

@ -18,13 +18,14 @@
#include <QTimer> #include <QTimer>
#include "Overlay.h" #include "Overlay.h"
#include "Slider.h" #include "Slider.h"
#include <QOffscreenSurface>
MainWindow::MainWindow() { MainWindow::MainWindow() {
simulationThread = new QThread(this); simulationThread = new QThread(this);
simulationThread->setObjectName("Simulation"); simulationThread->setObjectName("Simulation");
simulation = new Simulation; simulation = new Simulation;
simulation->moveToThread(simulationThread); simulation->moveToThread(simulationThread);
simulationThread->start(); connect(simulationThread, &QThread::started, simulation, &Simulation::initialize);
masses = std::vector<double>(MaxSegments); masses = std::vector<double>(MaxSegments);
lengths = std::vector<double>(MaxSegments); lengths = std::vector<double>(MaxSegments);
@ -36,7 +37,9 @@ MainWindow::MainWindow() {
normalizeLengths(); normalizeLengths();
QMetaObject::invokeMethod(simulation, &Simulation::initialize); auto offSurface = new QOffscreenSurface;
offSurface->create();
simulation->offSurface = offSurface;
} }
void MainWindow::buildUI() { void MainWindow::buildUI() {
@ -383,13 +386,13 @@ QWidget * MainWindow::buildSimulationUI() {
// GPU-Acceleration // GPU-Acceleration
{ {
auto useGPU = new QCheckBox("Use GPU-Acceleration"); auto useGPU = new QCheckBox("Use GPU-Acceleration");
useGPU->setCheckState(Qt::Checked);
connect(useGPU, &QCheckBox::stateChanged, simulation, &Simulation::useGPUChanged); connect(useGPU, &QCheckBox::stateChanged, simulation, &Simulation::useGPUChanged);
connect(simulation, &Simulation::gpuNotSupported, [useGPU](const std::string &message){ connect(simulation, &Simulation::gpuNotSupported, [useGPU](const std::string &message){
useGPU->setCheckState(Qt::Unchecked); useGPU->setCheckState(Qt::Unchecked);
useGPU->setEnabled(false); useGPU->setEnabled(false);
useGPU->setToolTip(QString::fromStdString(message)); useGPU->setToolTip(QString::fromStdString(message));
}); });
useGPU->setCheckState(Qt::Unchecked);
lyt->addWidget(useGPU); lyt->addWidget(useGPU);
} }

@ -15,12 +15,12 @@ class MainWindow : public QWidget {
public: public:
explicit MainWindow(); explicit MainWindow();
~MainWindow() override; ~MainWindow() override;
QThread * simulationThread;
private: private:
void buildUI(); void buildUI();
QWidget * buildAddUI(); QWidget * buildAddUI();
QWidget * buildSimulationUI(); QWidget * buildSimulationUI();
Simulation * simulation; Simulation * simulation;
QThread * simulationThread;
GLWidget * glWidget = nullptr; GLWidget * glWidget = nullptr;

@ -23,15 +23,19 @@ void Simulation::initialize() {
// OpenGL Offscreen functions // OpenGL Offscreen functions
{ {
auto context = new QOpenGLContext(this); context = new QOpenGLContext;
context->setFormat(QSurfaceFormat::defaultFormat()); auto format = QSurfaceFormat::defaultFormat();
format.setVersion(4, 3);
context->setFormat(format);
context->create(); context->create();
auto surface = new QOffscreenSurface(nullptr, this); context->makeCurrent(offSurface);
surface->setFormat(context->format()); f = new QOpenGLFunctions_4_3_Core;
surface->create(); if (!f->initializeOpenGLFunctions()){
context->makeCurrent(surface); std::string message = "OpenGL 4.3 is required for Compute Shaders! Falling back to CPU-Multithreading.";
initializeOpenGLFunctions(); printf("%s\n", message.c_str());
if (!context->hasExtension("GL_ARB_gpu_shader_fp64")){ useGPUAcceleration = false;
emit gpuNotSupported(message);
} else if (!context->hasExtension("GL_ARB_gpu_shader_fp64")){
std::string message = "Double precision not supported by OpenGL! Falling back to CPU-Multithreading."; std::string message = "Double precision not supported by OpenGL! Falling back to CPU-Multithreading.";
printf("%s\n", message.c_str()); printf("%s\n", message.c_str());
useGPUAcceleration = false; useGPUAcceleration = false;
@ -39,13 +43,19 @@ void Simulation::initialize() {
} }
} }
if (!useGPUAcceleration){
context->doneCurrent();
return;
}
// Read GPU Limits // Read GPU Limits
{ {
for (int i = 0; i < 3; i++){ for (int i = 0; i < 3; i++){
glGetIntegeri_v(GL_MAX_COMPUTE_WORK_GROUP_COUNT, i, gpuLimits.maxWGCount + i); f->glGetIntegeri_v(GL_MAX_COMPUTE_WORK_GROUP_COUNT, i, gpuLimits.maxWGCount + i);
glGetIntegeri_v(GL_MAX_COMPUTE_WORK_GROUP_SIZE, i, gpuLimits.maxWGSize + i); f->glGetIntegeri_v(GL_MAX_COMPUTE_WORK_GROUP_SIZE, i, gpuLimits.maxWGSize + i);
} }
glGetIntegerv(GL_MAX_COMPUTE_WORK_GROUP_INVOCATIONS, &gpuLimits.maxWGInvocations); f->glGetIntegerv(GL_MAX_COMPUTE_WORK_GROUP_INVOCATIONS, &gpuLimits.maxWGInvocations);
printf("Max work group count: (%d, %d, %d)\n", gpuLimits.maxWGCount[0], gpuLimits.maxWGCount[1], gpuLimits.maxWGCount[2]); printf("Max work group count: (%d, %d, %d)\n", gpuLimits.maxWGCount[0], gpuLimits.maxWGCount[1], gpuLimits.maxWGCount[2]);
printf("Max work group size: (%d, %d, %d)\n", gpuLimits.maxWGSize[0], gpuLimits.maxWGSize[1], gpuLimits.maxWGSize[2]); printf("Max work group size: (%d, %d, %d)\n", gpuLimits.maxWGSize[0], gpuLimits.maxWGSize[1], gpuLimits.maxWGSize[2]);
@ -58,19 +68,22 @@ void Simulation::initialize() {
program->addShaderFromSourceFile(QOpenGLShader::Compute, ":/shaders/compute.glsl"); program->addShaderFromSourceFile(QOpenGLShader::Compute, ":/shaders/compute.glsl");
program->link(); program->link();
glGenBuffers(6, &SSBO.positions); f->glGenBuffers(6, &SSBO.positions);
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 0, SSBO.positions); f->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 0, SSBO.positions);
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, SSBO.velocities); f->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 1, SSBO.velocities);
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 2, SSBO.invMasses); f->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 2, SSBO.invMasses);
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 3, SSBO.lengths); f->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 3, SSBO.lengths);
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 4, SSBO.indices); f->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 4, SSBO.indices);
glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 5, SSBO.segmentCounts); f->glBindBufferBase(GL_SHADER_STORAGE_BUFFER, 5, SSBO.segmentCounts);
} }
context->doneCurrent();
} }
// When the layout changes // When the layout changes
void Simulation::updateGPUData() { void Simulation::updateGPUData() {
context->makeCurrent(offSurface);
std::vector<GLdouble> positions; std::vector<GLdouble> positions;
std::vector<GLdouble> velocities; std::vector<GLdouble> velocities;
std::vector<GLdouble> invMasses; std::vector<GLdouble> invMasses;
@ -100,18 +113,20 @@ void Simulation::updateGPUData() {
auto indicesSize = GLsizeiptr(indices.size() * sizeof(GLuint)); auto indicesSize = GLsizeiptr(indices.size() * sizeof(GLuint));
auto segmentCountsSize = GLsizeiptr(segmentCounts.size() * sizeof(GLuint)); auto segmentCountsSize = GLsizeiptr(segmentCounts.size() * sizeof(GLuint));
glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.positions); f->glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.positions);
glBufferData(GL_SHADER_STORAGE_BUFFER, posSize, positions.data(), GL_DYNAMIC_DRAW); f->glBufferData(GL_SHADER_STORAGE_BUFFER, posSize, positions.data(), GL_DYNAMIC_DRAW);
glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.velocities); f->glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.velocities);
glBufferData(GL_SHADER_STORAGE_BUFFER, velSize, velocities.data(), GL_DYNAMIC_DRAW); f->glBufferData(GL_SHADER_STORAGE_BUFFER, velSize, velocities.data(), GL_DYNAMIC_DRAW);
glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.invMasses); f->glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.invMasses);
glBufferData(GL_SHADER_STORAGE_BUFFER, invMassSize, invMasses.data(), GL_STATIC_DRAW); f->glBufferData(GL_SHADER_STORAGE_BUFFER, invMassSize, invMasses.data(), GL_STATIC_DRAW);
glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.lengths); f->glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.lengths);
glBufferData(GL_SHADER_STORAGE_BUFFER, lengthSize, lengths.data(), GL_STATIC_DRAW); f->glBufferData(GL_SHADER_STORAGE_BUFFER, lengthSize, lengths.data(), GL_STATIC_DRAW);
glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.indices); f->glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.indices);
glBufferData(GL_SHADER_STORAGE_BUFFER, indicesSize, indices.data(), GL_STATIC_DRAW); f->glBufferData(GL_SHADER_STORAGE_BUFFER, indicesSize, indices.data(), GL_STATIC_DRAW);
glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.segmentCounts); f->glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.segmentCounts);
glBufferData(GL_SHADER_STORAGE_BUFFER, segmentCountsSize, segmentCounts.data(), GL_STATIC_DRAW); f->glBufferData(GL_SHADER_STORAGE_BUFFER, segmentCountsSize, segmentCounts.data(), GL_STATIC_DRAW);
context->doneCurrent();
} }
// Every few milliseconds // Every few milliseconds
@ -129,18 +144,19 @@ void Simulation::update() {
double newKineticEnergy = 0; double newKineticEnergy = 0;
if (useGPUAcceleration) { if (useGPUAcceleration) {
context->makeCurrent(offSurface);
program->bind(); program->bind();
glUniform1d(program->uniformLocation("h"), h); f->glUniform1d(program->uniformLocation("h"), h);
glUniform1d(program->uniformLocation("gravity"), gravity); f->glUniform1d(program->uniformLocation("gravity"), gravity);
glUniform1ui(program->uniformLocation("substeps"), substeps); f->glUniform1ui(program->uniformLocation("substeps"), substeps);
glDispatchCompute(pendula.size(), 1, 1); f->glDispatchCompute(pendula.size(), 1, 1);
glMemoryBarrier(GL_SHADER_STORAGE_BARRIER_BIT); f->glMemoryBarrier(GL_SHADER_STORAGE_BARRIER_BIT);
// Read updated positions // Read updated positions
{ {
glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.positions); f->glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.positions);
auto * newPositions = (double*) glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY); auto * newPositions = (double*) f->glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int index = 0; int index = 0;
for (Pendulum * p : pendula){ for (Pendulum * p : pendula){
for (Vector &point : p->X){ for (Vector &point : p->X){
@ -148,13 +164,13 @@ void Simulation::update() {
point.y = newPositions[index++]; point.y = newPositions[index++];
} }
} }
glUnmapBuffer(GL_SHADER_STORAGE_BUFFER); f->glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
} }
// Read updated velocities // Read updated velocities
{ {
glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.velocities); f->glBindBuffer(GL_SHADER_STORAGE_BUFFER, SSBO.velocities);
auto * newVelocities = (double*) glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY); auto * newVelocities = (double*) f->glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int index = 0; int index = 0;
for (Pendulum * p : pendula){ for (Pendulum * p : pendula){
for (Vector &velocity : p->V){ for (Vector &velocity : p->V){
@ -162,7 +178,7 @@ void Simulation::update() {
velocity.y = newVelocities[index++]; velocity.y = newVelocities[index++];
} }
} }
glUnmapBuffer(GL_SHADER_STORAGE_BUFFER); f->glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
} }
#pragma omp parallel for #pragma omp parallel for
@ -174,7 +190,7 @@ void Simulation::update() {
#pragma omp atomic #pragma omp atomic
newKineticEnergy += localKineticEnergy; newKineticEnergy += localKineticEnergy;
} }
context->doneCurrent();
} else { } else {
// CPU-Multithreading // CPU-Multithreading
#pragma omp parallel for #pragma omp parallel for
@ -208,6 +224,7 @@ void Simulation::clearPendula() {
pendula.shrink_to_fit(); pendula.shrink_to_fit();
updateEnergy(); updateEnergy();
if (useGPUAcceleration)
updateGPUData(); updateGPUData();
emit layoutChanged(); emit layoutChanged();
@ -220,6 +237,7 @@ void Simulation::addPendula(const std::vector<Pendulum *> &add) {
pendula.push_back(p); pendula.push_back(p);
updateEnergy(); updateEnergy();
if (useGPUAcceleration)
updateGPUData(); updateGPUData();
emit layoutChanged(); emit layoutChanged();

@ -5,11 +5,13 @@
#include <semaphore> #include <semaphore>
#include <QOpenGLFunctions_4_3_Core> #include <QOpenGLFunctions_4_3_Core>
class QOffscreenSurface;
class QTimer;
class FPS; class FPS;
class Pendulum; class Pendulum;
class QOpenGLShaderProgram; class QOpenGLShaderProgram;
class Simulation : public QObject, protected QOpenGLFunctions_4_3_Core { class Simulation : public QObject {
Q_OBJECT Q_OBJECT
public: public:
explicit Simulation(); explicit Simulation();
@ -29,19 +31,23 @@ public:
std::binary_semaphore semaphore = std::binary_semaphore(1); std::binary_semaphore semaphore = std::binary_semaphore(1);
FPS * ups; FPS * ups;
void initialize(); QOffscreenSurface* offSurface;
QOpenGLContext* context;
void updateEnergy(); void updateEnergy();
signals: signals:
void gpuNotSupported(const std::string &message); void gpuNotSupported(const std::string &message);
void layoutChanged(); void layoutChanged();
void positionChanged(); void positionChanged();
public slots: public slots:
void initialize();
void clearPendula(); void clearPendula();
void addPendula(const std::vector<Pendulum *> &add); void addPendula(const std::vector<Pendulum *> &add);
void useGPUChanged(int); void useGPUChanged(int);
private slots: private slots:
void update(); void update();
private: private:
QOpenGLFunctions_4_3_Core * f;
void updateGPUData(); void updateGPUData();
struct GPULimits { struct GPULimits {
int maxWGCount[3]; int maxWGCount[3];
@ -59,5 +65,5 @@ private:
} SSBO; } SSBO;
QTimer * timer; QTimer * timer;
int updateInterval = 16; int updateInterval = 16;
bool useGPUAcceleration = false; bool useGPUAcceleration = true;
}; };

@ -2,13 +2,14 @@
#include <QSurfaceFormat> #include <QSurfaceFormat>
#include "MainWindow.h" #include "MainWindow.h"
#include <QIcon> #include <QIcon>
#include <QThread>
int main(int argc, char* argv[]) { int main(int argc, char* argv[]) {
QSurfaceFormat fmt = QSurfaceFormat::defaultFormat(); QSurfaceFormat fmt = QSurfaceFormat::defaultFormat();
fmt.setDepthBufferSize(24); fmt.setDepthBufferSize(24);
fmt.setSamples(8); fmt.setSamples(8);
fmt.setSwapInterval(1); fmt.setSwapInterval(1);
fmt.setVersion(4, 3); fmt.setVersion(3, 3);
fmt.setProfile(QSurfaceFormat::CoreProfile); fmt.setProfile(QSurfaceFormat::CoreProfile);
QSurfaceFormat::setDefaultFormat(fmt); QSurfaceFormat::setDefaultFormat(fmt);
@ -19,5 +20,7 @@ int main(int argc, char* argv[]) {
w.setWindowIcon(QIcon(":/icons/app_icon.ico")); w.setWindowIcon(QIcon(":/icons/app_icon.ico"));
w.show(); w.show();
w.simulationThread->start();
return QApplication::exec(); return QApplication::exec();
} }

Loading…
Cancel
Save