2.12.2021, 9:00 - 11:00: Due to updates GitLab may be unavailable for some minutes between 09:00 and 11:00.

mainwindow.cpp 20.2 KB
Newer Older
1
2
// ================================================================================================
// 
schultezub's avatar
schultezub committed
3
// This file is part of the CAMPVis Software Framework.
4
// 
5
// If not explicitly stated otherwise: Copyright (C) 2012-2014, all rights reserved,
schultezub's avatar
schultezub committed
6
//      Christian Schulte zu Berge <christian.szb@in.tum.de>
7
//      Chair for Computer Aided Medical Procedures
8
9
//      Technische Universitaet Muenchen
//      Boltzmannstr. 3, 85748 Garching b. Muenchen, Germany
10
// 
schultezub's avatar
schultezub committed
11
// For a full list of authors and contributors, please refer to the file "AUTHORS.txt".
12
// 
13
14
15
16
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file 
// except in compliance with the License. You may obtain a copy of the License at
// 
// http://www.apache.org/licenses/LICENSE-2.0
17
// 
18
19
20
21
// Unless required by applicable law or agreed to in writing, software distributed under the 
// License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, 
// either express or implied. See the License for the specific language governing permissions 
// and limitations under the License.
22
23
24
// 
// ================================================================================================

25
26
#include "mainwindow.h"

27
#include "cgt/assert.h"
28
29
#include "cgt/opengljobprocessor.h"

schultezub's avatar
schultezub committed
30
#include "application/campvisapplication.h"
31
#include "application/gui/datacontainerinspectorwidget.h"
32
#include "application/gui/datacontainerinspectorcanvas.h"
33
#include "application/gui/mdi/mdidockablewindow.h"
34
#include "application/gui/qtdatahandle.h"
35
#include "core/datastructures/datacontainer.h"
36
37
#include "core/pipeline/abstractpipeline.h"
#include "core/pipeline/abstractprocessor.h"
38
39
#include "core/tools/stringutils.h"
#include "modules/pipelinefactory.h"
40

41
#include <QScrollBar>
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
42
43
44
45
#include <QFileDialog>

#include <fstream>

Hossain Mahmud's avatar
Hossain Mahmud committed
46
#include "scripting/luagen/properties/propertycollectionluascriptgenerator.h"
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
47
#include "scripting/luagen/properties/abstractpropertylua.h"
Christian Schulte zu Berge's avatar
Christian Schulte zu Berge committed
48

49

schultezub's avatar
schultezub committed
50
namespace campvis {
51

52
    MainWindow::MainWindow(CampVisApplication* application)
53
54
        : QMainWindow()
        , _application(application)
55
        , _mdiArea(0)
56
57
58
        , _containerWidget(0)
        , _cbPipelineFactory(0)
        , _btnPipelineFactory(0)
59
        , _pipelineWidget(0)
60
61
        , _propCollectionWidget(0)
        , _dcInspectorWidget(0)
62
        , _dcInspectorWindow(0)
63
64
65
66
        , _btnExecute(0)
        , _btnShowDataContainerInspector(0)
        , _selectedPipeline(0)
        , _selectedProcessor(0)
67
        , _selectedDataContainer(0)
68
        , _logViewer(0)
69
        , _scriptingConsoleWidget(nullptr)
70
        , _workflowWidget(nullptr)
71
    {
72
        cgtAssert(_application != 0, "Application must not be 0.");
73
        ui.setupUi(this);
74
75
76
77
        setup();
    }

    MainWindow::~MainWindow() {
schultezub's avatar
schultezub committed
78
        _application->s_PipelinesChanged.disconnect(this);
79
        _application->s_DataContainersChanged.disconnect(this);
80
        delete _dcInspectorWidget;
81
82
83
    }

    void MainWindow::setup() {
84
85
        qRegisterMetaType<QtDataHandle>("QtDataHandle");

86
        setCorner(Qt::BottomLeftCorner, Qt::LeftDockWidgetArea);
87
        setCorner(Qt::TopLeftCorner, Qt::LeftDockWidgetArea);
88
        setCorner(Qt::BottomRightCorner, Qt::RightDockWidgetArea);
89
        setCorner(Qt::TopRightCorner, Qt::RightDockWidgetArea);
90

91
        setTabPosition(Qt::TopDockWidgetArea, QTabWidget::North);
92

93
        _mdiArea = new MdiDockArea();
94
95
96
        _mdiArea->tileSubWindows();
        setCentralWidget(_mdiArea);

97
98
99
        _containerWidget = new QWidget(this);
        QGridLayout* _cwLayout = new QGridLayout(_containerWidget);

100
        int rowPosition = 0;
101
102
103
104
        _cbPipelineFactory = new QComboBox(_containerWidget);
        std::vector<std::string> registeredPipelines = PipelineFactory::getRef().getRegisteredPipelines();
        for (std::vector<std::string>::const_iterator it = registeredPipelines.begin(); it != registeredPipelines.end(); ++it)
            _cbPipelineFactory->addItem(QString::fromStdString(*it));
105
        _cwLayout->addWidget(_cbPipelineFactory, rowPosition, 0);
106
107

        _btnPipelineFactory = new QPushButton("Add Pipeline", _containerWidget);
108
        _cwLayout->addWidget(_btnPipelineFactory, rowPosition++, 1);
109

110
        _pipelineWidget = new PipelineTreeWidget(this);
111
        _containerWidget->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Maximum);
112
        _cwLayout->addWidget(_pipelineWidget, rowPosition++, 0, 1, 2);
113

114
        _btnExecute = new QPushButton("Execute Selected Pipeline/Processor", _containerWidget);
115
116
117
118
119
120
121
122
123
124
        _cwLayout->addWidget(_btnExecute, rowPosition++, 0, 1, 2);

        _cbProcessorFactory = new QComboBox(_containerWidget);
        std::vector<std::string> registeredProcessors = ProcessorFactory::getRef().getRegisteredProcessors();
        for (std::vector<std::string>::const_iterator it = registeredProcessors.begin(); it != registeredProcessors.end(); ++it)
            _cbProcessorFactory->addItem(QString::fromStdString(*it));
        _cwLayout->addWidget(_cbProcessorFactory, rowPosition, 0);

        _btnProcessorFactory = new QPushButton("Add Processor", _containerWidget);
        _cwLayout->addWidget(_btnProcessorFactory, rowPosition++, 1);
125
126

        _btnShowDataContainerInspector = new QPushButton("Inspect DataContainer of Selected Pipeline", _containerWidget);
127
        _cwLayout->addWidget(_btnShowDataContainerInspector, rowPosition++, 0, 1, 2);
128
129
130

#ifdef CAMPVIS_HAS_SCRIPTING
        _btnLuaLoad = new QPushButton("Load Script", _containerWidget);
131
        _cwLayout->addWidget(_btnLuaLoad, rowPosition++, 0, 1, 2);
132
        _btnLuaSave = new QPushButton("Save Script", _containerWidget);
133
        _cwLayout->addWidget(_btnLuaSave, rowPosition++, 0, 1, 2);
134
135
136
137
138
139
140
141
142
        connect(
            _btnLuaLoad, SIGNAL(clicked()), 
            this, SLOT(onBtnLuaLoadClicked()));
        connect(
            _btnLuaSave, SIGNAL(clicked()), 
            this, SLOT(onBtnLuaSaveClicked()));
#else
#endif

143
144
        _containerWidget->setLayout(_cwLayout);
        ui.pipelineTreeDock->setWidget(_containerWidget);
145

146
        _pipelinePropertiesScrollArea = new QScrollArea(this);
147
148
149
150
        _pipelinePropertiesScrollArea->setWidgetResizable(true);
        _pipelinePropertiesScrollArea->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
        _pipelinePropertiesScrollArea->setFrameStyle(QScrollArea::NoFrame);

151
        _pipelinePropertiesWidget = new QWidget(_pipelinePropertiesScrollArea);
152
153
154
        _pipelinePropertiesWidget->installEventFilter(this);
        _pipelinePropertiesScrollArea->setWidget(_pipelinePropertiesWidget);
        ui.pipelinePropertiesDock->setWidget(_pipelinePropertiesScrollArea);
155

156
        QVBoxLayout* rightLayout = new QVBoxLayout(_pipelinePropertiesWidget);
157
        rightLayout->setSpacing(4);
158
        _pipelinePropertiesWidget->setLayout(rightLayout);
159

160
        _propCollectionWidget = new PropertyCollectionWidget(this);
161
        rightLayout->addWidget(_propCollectionWidget);
162
        rightLayout->addStretch();
163

164
        _logViewer = new LogViewerWidget(this);
165
        ui.logViewerDock->setWidget(_logViewer);
166

167
168
169
170
#ifdef CAMPVIS_HAS_SCRIPTING
        _scriptingConsoleWidget = new ScriptingWidget(this);
        ui.scriptingConsoleDock->setWidget(_scriptingConsoleWidget);
        connect(_scriptingConsoleWidget, SIGNAL(s_commandExecuted(const QString&)), this, SLOT(onLuaCommandExecuted(const QString&)));
171
172
#else
        ui.scriptingConsoleDock->setVisible(false);
173
174
#endif

175
        _workflowWidget = new WorkflowControllerWidget(_application, this);
176
        ui.workflowDock->setWidget(_workflowWidget);
177
        ui.workflowDock->setVisible(! _application->_workflows.empty());
178

179
        _dcInspectorWidget = new DataContainerInspectorWidget();
180
        this->populateMainMenu();
181

182
183
        qRegisterMetaType< std::vector<DataContainer*> >("std::vector<DataContainer*>");
        qRegisterMetaType< std::vector<AbstractPipeline*> >("std::vector<AbstractPipeline*>");
Christian Schulte zu Berge's avatar
Christian Schulte zu Berge committed
184
        qRegisterMetaType< std::map<AbstractProperty*, QWidget*>::iterator >("PropertyWidgetMapIterator");
185
        connect(
186
187
            this, SIGNAL(updatePipelineWidget(const std::vector<DataContainer*>&, const std::vector<AbstractPipeline*>&)), 
            _pipelineWidget, SLOT(update(const std::vector<DataContainer*>&, const std::vector<AbstractPipeline*>&)));
188
189
190
191
        connect(
            _pipelineWidget, SIGNAL(clicked(const QModelIndex&)), 
            this, SLOT(onPipelineWidgetItemClicked(const QModelIndex&)));
        connect(
192
193
            this, SIGNAL(updatePropCollectionWidget(HasPropertyCollection*, DataContainer*)),
            _propCollectionWidget, SLOT(updatePropCollection(HasPropertyCollection*, DataContainer*)));
194
195
196
197
198
199
        connect(
            _btnExecute, SIGNAL(clicked()), 
            this, SLOT(onBtnExecuteClicked()));
        connect(
            _btnShowDataContainerInspector, SIGNAL(clicked()), 
            this, SLOT(onBtnShowDataContainerInspectorClicked()));
200
201
202
        connect(
            _btnPipelineFactory, SIGNAL(clicked()), 
            this, SLOT(onBtnPipelineFactoryClicked()));
203
204
205
        connect(
            _btnProcessorFactory, SIGNAL(clicked()), 
            this, SLOT(onBtnProcessorFactoryClicked()));
206

207
        _application->s_PipelinesChanged.connect(this, &MainWindow::onPipelinesChanged);
208
        _application->s_DataContainersChanged.connect(this, &MainWindow::onDataContainersChanged);
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
209

210
211
    }

212
    void MainWindow::populateMainMenu() {
213
        // Populate the file menu
214
215
        QMenuBar* menuBar = this->menuBar();
        QMenu* fileMenu = menuBar->addMenu(tr("&File"));
216
        fileMenu->addAction(tr("&Rebuild all Shaders from File"), this, SLOT(onRebuildShadersClicked()), QKeySequence(Qt::CTRL + Qt::Key_F5));
217
218
        fileMenu->addAction(tr("&Quit"), qApp, SLOT(closeAllWindows()), QKeySequence(Qt::CTRL + Qt::Key_Q));

219
220
221
222
223
224
225
226
227
228
        // Populate the visualizations menu
        QMenu* visualizationsMenu = _mdiArea->menu();
        visualizationsMenu->setTitle(tr("&Visualizations"));
        menuBar->addMenu(visualizationsMenu);

        // Populate the tools menu
        QMenu* toolsMenu = menuBar->addMenu(tr("&Tools"));
        toolsMenu->addAction(ui.pipelineTreeDock->toggleViewAction());
        toolsMenu->addAction(ui.pipelinePropertiesDock->toggleViewAction());
        toolsMenu->addAction(ui.logViewerDock->toggleViewAction());
229
        toolsMenu->addAction(ui.workflowDock->toggleViewAction());
230
231
    }

232
233
234
235
236
237
238
239
240
    bool MainWindow::eventFilter(QObject* watched, QEvent* event) {
        if (watched == _pipelinePropertiesWidget && event->type() == QEvent::Resize) {
            _pipelinePropertiesScrollArea->setMinimumWidth(_pipelinePropertiesWidget->minimumSizeHint().width() +
                                                           _pipelinePropertiesScrollArea->verticalScrollBar()->width());
        }

        return false;
    }

241
    void MainWindow::onPipelinesChanged() {
242
243
244
245
        std::vector<AbstractPipeline*> pipelines;
        std::for_each(_application->_pipelines.begin(), _application->_pipelines.end(), [&] (const CampVisApplication::PipelineRecord& pr) { pipelines.push_back(pr._pipeline); });

        emit updatePipelineWidget(_application->_dataContainers, pipelines);
246
247
248
    }

    void MainWindow::onDataContainersChanged() {
249
250
        std::vector<AbstractPipeline*> pipelines;
        std::for_each(_application->_pipelines.begin(), _application->_pipelines.end(), [&] (const CampVisApplication::PipelineRecord& pr) { pipelines.push_back(pr._pipeline); });
251
        
252
        emit updatePipelineWidget(_application->_dataContainers, pipelines);
253
254
    }

255
256
257
258
    void MainWindow::onPipelineWidgetItemClicked(const QModelIndex& index) {
        if (index.isValid()) {
            // Yak, this is so ugly - another reason why GUI programming sucks...
            QVariant item = index.data(Qt::UserRole);
259
260
            if (item.isValid()) {
                HasPropertyCollection* ptr = static_cast<HasPropertyCollection*>(item.value<void*>());
261

262
                if (AbstractPipeline* pipeline = dynamic_cast<AbstractPipeline*>(ptr)) {
263
                    _selectedPipeline = pipeline;
264
                    _selectedProcessor = 0;
265
                    _selectedDataContainer = &pipeline->getDataContainer();
266
267
268
269
270
271
272
273
                }
                else if (AbstractProcessor* processor = dynamic_cast<AbstractProcessor*>(ptr)) {
                    _selectedProcessor = processor;

                    QVariant parentItem = index.parent().data(Qt::UserRole);
                    HasPropertyCollection* pptr = static_cast<HasPropertyCollection*>(parentItem.value<void*>());
                    if (AbstractPipeline* pipeline = dynamic_cast<AbstractPipeline*>(pptr)) {
                        _selectedPipeline = pipeline;
274
                        _selectedDataContainer = &pipeline->getDataContainer();
275
                    }
276
277
                }

278
                emit updatePropCollectionWidget(ptr, &_selectedPipeline->getDataContainer());
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
279

280
281
282
            }
            else {
                emit updatePropCollectionWidget(0, 0);
283
                _selectedDataContainer = 0;
284
            }
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
285

286
287
        }
        else {
288
            emit updatePropCollectionWidget(0, 0);
289
            _selectedDataContainer = 0;
290
291
292
        }
    }

293
    QSize MainWindow::sizeHint() const {
294
        return QSize(1000, 600);
295
296
    }

297
298
299
300
    void MainWindow::onBtnExecuteClicked() {
        if (_selectedProcessor != 0 && _selectedPipeline != 0) {
            // this is not as trivial as it seems:
            // We need the pipeline, probably an OpenGL context...
301
            _selectedProcessor->invalidate(AbstractProcessor::INVALID_RESULT);
302
303
        }
        else if (_selectedPipeline != 0) {
304
305
306
            for (std::vector<AbstractProcessor*>::const_iterator it = _selectedPipeline->getProcessors().begin(); it != _selectedPipeline->getProcessors().end(); ++it) {
                (*it)->invalidate(AbstractProcessor::INVALID_RESULT);
            }
307
308
309
        }
    }

Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
310
311
312
313
314
315
316
317
    void MainWindow::onBtnLuaLoadClicked() {
#ifdef CAMPVIS_HAS_SCRIPTING
        const QString dialogCaption = QString::fromStdString("Select File");
        const QString directory = QString::fromStdString(".");
        const QString fileFilter = tr("All files (*)");

        QString filename = QFileDialog::getOpenFileName(QWidget::parentWidget(), dialogCaption, directory, fileFilter);
        if (filename != nullptr && _application->getLuaVmState() != nullptr) {
318
            _application->getLuaVmState()->execFile(filename.toStdString());
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
319
320
321
322
323
324
325
326
327
328
329
330
331
332
        }
#endif
    }

    void MainWindow::onBtnLuaSaveClicked() {
#ifdef CAMPVIS_HAS_SCRIPTING
        const QString dialogCaption = QString::fromStdString("Save File as");
        const QString directory = QString::fromStdString(".");
        const QString fileFilter = tr("All files (*)");

        QString filename = QFileDialog::getSaveFileName(QWidget::parentWidget(), dialogCaption, directory, fileFilter);

        if (filename != nullptr) {
            if (_selectedProcessor != 0 && _selectedPipeline != 0) {
Hossain Mahmud's avatar
Hossain Mahmud committed
333
                PropertyCollectionLuaScriptGenerator* _pcLua = new PropertyCollectionLuaScriptGenerator();
334
335
                
                std::string pipeScript = "pipeline = pipelines[\"" + _selectedPipeline->getName()+"\"]\n\n";
336
                for (size_t i = 0; i < _selectedPipeline->getProcessors().size(); i++) {
337
                    pipeScript += "proc = pipeline:getProcessor(" + StringUtils::toString(i) + ")\n";
338
                    AbstractProcessor* proc = _selectedPipeline->getProcessor(int(i));
339
340

                    _pcLua->updatePropCollection(proc, &_selectedPipeline->getDataContainer());
Hossain Mahmud's avatar
Hossain Mahmud committed
341
                    std::string res = _pcLua->getLuaScript(std::string(""), std::string("proc:"));
342
                    pipeScript += res;
343
344
                }
                if (pipeScript != "pipeline = pipelines[\"" + _selectedPipeline->getName()+"\"]\n\n") {
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
345
346
                    std::ofstream file;
                    file.open(filename.toStdString());
347
                    file << pipeScript.c_str();
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
348
349
                    file.close();
                }
350
351
                
                delete _pcLua;
Hossain Mahmud's avatar
stuck!    
Hossain Mahmud committed
352
353
354
355
356
            }
        }
#endif
    }

357
358
    void MainWindow::onBtnShowDataContainerInspectorClicked() {
        if (_selectedPipeline != 0) {
359
360
361
            if (_dcInspectorWindow == 0) {
                _dcInspectorWindow = _mdiArea->addWidget(_dcInspectorWidget);
                _dcInspectorWindow->setWindowTitle(tr("Data Container Inspector"));
362
363
            }

364
            _dcInspectorWidget->setDataContainer(&(_selectedPipeline->getDataContainer()));
365
366
            _dcInspectorWindow->show();
            _dcInspectorWindow->activateWindow();
367
368
369
        }
    }

370
    void MainWindow::init() {
371
372
        if (_dcInspectorWidget != 0)
            _dcInspectorWidget->init();
373

374
        _logViewer->init();
375
376
377

        if (_scriptingConsoleWidget)
            _scriptingConsoleWidget->init();
378
379
380
    }

    void MainWindow::deinit() {
381
382
        if (_dcInspectorWidget != 0)
            _dcInspectorWidget->deinit();
383

384
        _logViewer->deinit();
385
386
387

        if (_scriptingConsoleWidget)
            _scriptingConsoleWidget->deinit();
388
389
    }

390
    MdiDockableWindow * MainWindow::addVisualizationPipelineWidget(const std::string& name, QWidget* canvas) {
391
        MdiDockableWindow* dockableWindow = _mdiArea->addWidget(canvas);
392
        QString windowTitle = QString::fromStdString(name);
393
394
        dockableWindow->setWindowTitle(windowTitle);
        dockableWindow->show();
395
        return dockableWindow;
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
    }

    QDockWidget* MainWindow::dockPrimaryWidget(const std::string& name, QWidget* widget) {
        QDockWidget* dockWidget = new QDockWidget(QString::fromStdString(name));
        dockWidget->setWidget(widget);
        dockWidget->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Expanding);

        if (_primaryDocks.empty()) {
            addDockWidget(Qt::TopDockWidgetArea, dockWidget);
        } else {
            tabifyDockWidget(_primaryDocks.back(), dockWidget);
            // Activate the dock's tab
            dockWidget->setVisible(true);
            dockWidget->raise();
        }

        _primaryDocks.push_back(dockWidget);
        return dockWidget;
414
415
    }

416
417
418
419
420
421
422
423
424
425
    void MainWindow::onBtnPipelineFactoryClicked() {
        std::string name = this->_cbPipelineFactory->currentText().toStdString();
        DataContainer* dc = _selectedDataContainer;
        if (dc == 0) {
            dc = _application->createAndAddDataContainer("DataContainer #" + StringUtils::toString(_application->_dataContainers.size() + 1));
        }
        AbstractPipeline* p = PipelineFactory::getRef().createPipeline(name, dc);
        _application->addPipeline(name, p);
    }

426
427
428
429
430
431
432
    void MainWindow::onBtnProcessorFactoryClicked() {
        cgt::OpenGLJobProcessor::ScopedSynchronousGlJobExecution jobGuard;

        std::string name = this->_cbProcessorFactory->currentText().toStdString();
        if (_selectedPipeline == nullptr) 
            return;

Hossain Mahmud's avatar
Hossain Mahmud committed
433
434
435
        IVec2Property temp = _selectedPipeline->getCanvasSize();
        IVec2Property *viewPort = new IVec2Property(temp.getName(), temp.getTitle(), temp.getValue(), temp.getMinValue(), temp.getMaxValue(), temp.getStepValue());
        AbstractProcessor* p = ProcessorFactory::getRef().createProcessor(name, viewPort);
436
437
        if (p == nullptr)
            return;
438
        
439
        p->init();
440
441
442
443
444
445
446
447
        _selectedPipeline->addProcessor(p);

        std::vector<AbstractPipeline*> pipelines;
        std::for_each(_application->_pipelines.begin(), _application->_pipelines.end(), [&] (const CampVisApplication::PipelineRecord& pr) { pipelines.push_back(pr._pipeline); });

        emit updatePipelineWidget(_application->_dataContainers, pipelines);
    }

448
449
450
451
    void MainWindow::onRebuildShadersClicked() {
        _application->rebuildAllShadersFromFiles();
    }

452
    void MainWindow::onLuaCommandExecuted(const QString& cmd) {
453
454
#ifdef CAMPVIS_HAS_SCRIPTING
        if (_application->getLuaVmState() != nullptr) {
455
            cgt::OpenGLJobProcessor::ScopedSynchronousGlJobExecution jobGuard;
456
            _application->getLuaVmState()->execString(cmd.toStdString());
457
        }
458
#endif
459
460
    }

461
462
463
464
465
    void MainWindow::setWorkflow(AbstractWorkflow* w) {
        ui.workflowDock->setVisible(true);
        _workflowWidget->setWorkflow(w);
    }

466
}