NBody/mainwindow.cpp at master · MatejGomboc/NBody · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#include <QFile>
#include <QMessageBox>
#include "mainwindow.h"
#include "ui_mainwindow.h"

MainWindow::MainWindow(QWidget* parent) :
    QMainWindow(parent),
    m_ui(new Ui::MainWindow),
    m_rendering_timer(new QTimer(this))
{
    m_ui->setupUi(this);

    connect(m_ui->central_widget, &OpenGLSceneWidget::errorOccurred,
        this, &MainWindow::openglSceneWidget_errorOccurred,
        Qt::ConnectionType::QueuedConnection);

    connect(m_ui->central_widget, &OpenGLSceneWidget::openGlInitialized,
        this, &MainWindow::openglSceneWidget_openGlInitialized,
        Qt::ConnectionType::QueuedConnection);

    connect(m_ui->central_widget, &OpenGLSceneWidget::openGlDestroyed,
        this, &MainWindow::openglSceneWidget_openGlDestroyed,
        Qt::ConnectionType::QueuedConnection);

    connect(m_rendering_timer, &QTimer::timeout,
        this, &MainWindow::rendering_timer_timeout,
        Qt::ConnectionType::QueuedConnection);
}

MainWindow::~MainWindow()
{
    delete m_rendering_timer;
    delete m_ui;
}

void MainWindow::openglSceneWidget_errorOccurred(const QString& error_message)
{
    QMessageBox error_dialog(this);
    error_dialog.setWindowTitle("OpenGL error");
    error_dialog.setIcon(QMessageBox::Icon::Critical);
    error_dialog.setText(error_message);
    error_dialog.setModal(true);
    error_dialog.setTextInteractionFlags(Qt::TextSelectableByMouse);
    error_dialog.exec();
    QApplication::quit();
}

void MainWindow::openglSceneWidget_openGlInitialized()
{
    QMessageBox error_dialog(this);
    error_dialog.setIcon(QMessageBox::Icon::Critical);
    error_dialog.setModal(true);
    error_dialog.setTextInteractionFlags(Qt::TextSelectableByMouse);

    std::vector<float> vertices_data = NBodySim2D::generateRandomLocations(NUM_POINTS, MAX_START_DISTANCE);

    QString error_message_1;
    if (!m_ui->central_widget->initVertices(vertices_data, error_message_1)) {
        error_dialog.setWindowTitle("OpenGL error");
        error_dialog.setText(error_message_1);
        error_dialog.exec();
        QApplication::quit();
        return;
    }

    QFile opencl_source_file_gravity(":/gravity.cl");
    if (!opencl_source_file_gravity.open(QIODevice::ReadOnly)) {
        error_dialog.setWindowTitle("OpenCL error");
        error_dialog.setText("Cannot open OpenCL source file.");
        error_dialog.exec();
        QApplication::quit();
        return;
    }

    std::vector<std::string> opencl_sources;
    opencl_sources.push_back(opencl_source_file_gravity.readAll().toStdString());
    opencl_source_file_gravity.close();

    QFile opencl_source_file_leapfrog(":/leapfrog.cl");
    if (!opencl_source_file_leapfrog.open(QIODevice::ReadOnly)) {
        error_dialog.setWindowTitle("OpenCL error");
        error_dialog.setText("Cannot open OpenCL source file.");
        error_dialog.exec();
        QApplication::quit();
        return;
    }

    opencl_sources.push_back(opencl_source_file_leapfrog.readAll().toStdString());
    opencl_source_file_leapfrog.close();

    std::string error_message_2;
    if (!m_nbodysim.init(opencl_sources, m_ui->central_widget->getVertexBufferId(),
        static_cast<uint32_t>(vertices_data.size()), ATTRACTION, RADIUS, TIME_STEP,
        MAX_DISTANCE, MAX_VELOCITY, MAX_START_VELOCITY, error_message_2)) {
        error_dialog.setWindowTitle("OpenCL error");
        error_dialog.setText(error_message_2.c_str());
        error_dialog.exec();
        QApplication::quit();
        return;
    }

    m_rendering_timer->setSingleShot(false);
    m_rendering_timer->setInterval(RENDER_UPDATE_TIME_MS);
    m_rendering_timer->start();

    m_ui->central_widget->setZoom(MAX_DISTANCE);
}

void MainWindow::openglSceneWidget_openGlDestroyed()
{
    m_rendering_timer->stop();
    disconnect(m_rendering_timer, &QTimer::timeout, nullptr, nullptr);
}

void MainWindow::rendering_timer_timeout()
{
    std::string error_message;
    if (!m_nbodysim.updateLocations(NUM_POINTS, error_message)) {
        QMessageBox error_dialog(this);
        error_dialog.setIcon(QMessageBox::Icon::Critical);
        error_dialog.setModal(true);
        error_dialog.setTextInteractionFlags(Qt::TextSelectableByMouse);
        error_dialog.setWindowTitle("OpenCL error");
        error_dialog.setText(error_message.c_str());
        error_dialog.exec();
        QApplication::quit();
    }

    m_ui->central_widget->update();
}