mainwindow.cpp 19.9 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/pipeline/pipelinefactory.h"
#include "core/pipeline/processorfactory.h"
40
#include "core/tools/stringutils.h"
41

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

#include <fstream>

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

50

schultezub's avatar
schultezub committed
51
namespace campvis {
52

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

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

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

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

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

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

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

101
        int rowPosition = 0;
102 103 104 105
        _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));
106
        _cwLayout->addWidget(_cbPipelineFactory, rowPosition, 0);
107 108

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

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

115
        _btnExecute = new QPushButton("Execute Selected Pipeline/Processor", _containerWidget);
116 117 118 119 120 121 122 123 124 125
        _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);
126 127

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

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

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

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

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

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

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

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

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

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

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

183 184
        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
185
        qRegisterMetaType< std::map<AbstractProperty*, QWidget*>::iterator >("PropertyWidgetMapIterator");
186
        connect(
187 188
            this, SIGNAL(updatePipelineWidget(const std::vector<DataContainer*>&, const std::vector<AbstractPipeline*>&)), 
            _pipelineWidget, SLOT(update(const std::vector<DataContainer*>&, const std::vector<AbstractPipeline*>&)));
189
        connect(
190 191
            _pipelineWidget, SIGNAL(itemChanged(const QModelIndex&)), 
            this, SLOT(onPipelineWidgetItemChanged(const QModelIndex&)));
192
        connect(
193 194
            this, SIGNAL(updatePropCollectionWidget(HasPropertyCollection*, DataContainer*)),
            _propCollectionWidget, SLOT(updatePropCollection(HasPropertyCollection*, DataContainer*)));
195 196 197 198 199 200
        connect(
            _btnExecute, SIGNAL(clicked()), 
            this, SLOT(onBtnExecuteClicked()));
        connect(
            _btnShowDataContainerInspector, SIGNAL(clicked()), 
            this, SLOT(onBtnShowDataContainerInspectorClicked()));
201 202 203
        connect(
            _btnPipelineFactory, SIGNAL(clicked()), 
            this, SLOT(onBtnPipelineFactoryClicked()));
204 205 206
        connect(
            _btnProcessorFactory, SIGNAL(clicked()), 
            this, SLOT(onBtnProcessorFactoryClicked()));
207

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

211 212
    }

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

220 221 222 223 224 225 226 227 228 229
        // 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());
230
        toolsMenu->addAction(ui.workflowDock->toggleViewAction());
231 232
    }

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

        return false;
    }

242
    void MainWindow::onPipelinesChanged() {
243
        emit updatePipelineWidget(_application->_dataContainers, _application->_pipelines);
244 245 246
    }

    void MainWindow::onDataContainersChanged() {
247
        emit updatePipelineWidget(_application->_dataContainers, _application->_pipelines);
248 249
    }

250
    void MainWindow::onPipelineWidgetItemChanged(const QModelIndex& index) {
251 252 253
        if (index.isValid()) {
            // Yak, this is so ugly - another reason why GUI programming sucks...
            QVariant item = index.data(Qt::UserRole);
254 255
            if (item.isValid()) {
                HasPropertyCollection* ptr = static_cast<HasPropertyCollection*>(item.value<void*>());
256

257
                if (AbstractPipeline* pipeline = dynamic_cast<AbstractPipeline*>(ptr)) {
258
                    _selectedPipeline = pipeline;
259
                    _selectedProcessor = 0;
260
                    _selectedDataContainer = &pipeline->getDataContainer();
261 262 263 264 265 266 267 268
                }
                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;
269
                        _selectedDataContainer = &pipeline->getDataContainer();
270
                    }
271 272
                }

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

275 276 277
            }
            else {
                emit updatePropCollectionWidget(0, 0);
278
                _selectedDataContainer = 0;
279
            }
Hossain Mahmud's avatar
stuck!  
Hossain Mahmud committed
280

281 282
        }
        else {
283
            emit updatePropCollectionWidget(0, 0);
284
            _selectedDataContainer = 0;
285 286 287
        }
    }

288
    QSize MainWindow::sizeHint() const {
289
        return QSize(1000, 600);
290 291
    }

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

Hossain Mahmud's avatar
stuck!  
Hossain Mahmud committed
305 306 307 308 309 310 311 312
    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) {
313
            _application->getLuaVmState()->execFile(filename.toStdString());
Hossain Mahmud's avatar
stuck!  
Hossain Mahmud committed
314 315 316 317 318 319 320 321 322 323 324 325 326 327
        }
#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
328
                PropertyCollectionLuaScriptGenerator* _pcLua = new PropertyCollectionLuaScriptGenerator();
329 330
                
                std::string pipeScript = "pipeline = pipelines[\"" + _selectedPipeline->getName()+"\"]\n\n";
331
                for (size_t i = 0; i < _selectedPipeline->getProcessors().size(); i++) {
332
                    pipeScript += "proc = pipeline:getProcessor(" + StringUtils::toString(i) + ")\n";
333
                    AbstractProcessor* proc = _selectedPipeline->getProcessor(int(i));
334 335

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

352 353
    void MainWindow::onBtnShowDataContainerInspectorClicked() {
        if (_selectedPipeline != 0) {
354 355 356
            if (_dcInspectorWindow == 0) {
                _dcInspectorWindow = _mdiArea->addWidget(_dcInspectorWidget);
                _dcInspectorWindow->setWindowTitle(tr("Data Container Inspector"));
357 358
            }

359
            _dcInspectorWidget->setDataContainer(&(_selectedPipeline->getDataContainer()));
360 361
            _dcInspectorWindow->show();
            _dcInspectorWindow->activateWindow();
362 363 364
        }
    }

365
    void MainWindow::init() {
366 367
        if (_dcInspectorWidget != 0)
            _dcInspectorWidget->init();
368

369
        _logViewer->init();
370 371 372

        if (_scriptingConsoleWidget)
            _scriptingConsoleWidget->init();
373 374 375
    }

    void MainWindow::deinit() {
376 377
        if (_dcInspectorWidget != 0)
            _dcInspectorWidget->deinit();
378

379
        _logViewer->deinit();
380 381 382

        if (_scriptingConsoleWidget)
            _scriptingConsoleWidget->deinit();
383 384
    }

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

    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;
409 410
    }

411 412 413 414 415 416 417 418 419 420
    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);
    }

421 422 423 424 425 426 427
    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
428 429 430
        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);
431 432
        if (p == nullptr)
            return;
433
        
434
        p->init();
435 436
        _selectedPipeline->addProcessor(p);

437
        emit updatePipelineWidget(_application->_dataContainers, _application->_pipelines);
438 439
    }

440 441 442 443
    void MainWindow::onRebuildShadersClicked() {
        _application->rebuildAllShadersFromFiles();
    }

444
    void MainWindow::onLuaCommandExecuted(const QString& cmd) {
445 446
#ifdef CAMPVIS_HAS_SCRIPTING
        if (_application->getLuaVmState() != nullptr) {
447
            cgt::OpenGLJobProcessor::ScopedSynchronousGlJobExecution jobGuard;
448
            _application->getLuaVmState()->execString(cmd.toStdString());
449
        }
450
#endif
451 452
    }

453 454 455 456 457
    void MainWindow::setWorkflow(AbstractWorkflow* w) {
        ui.workflowDock->setVisible(true);
        _workflowWidget->setWorkflow(w);
    }

458 459 460 461 462 463 464 465 466
    void MainWindow::enableKioskMode() {
        ui.logViewerDock->hide();
        ui.pipelineTreeDock->hide();
        ui.pipelinePropertiesDock->hide();
        ui.scriptingConsoleDock->hide();
        ui.workflowDock->show();

    }

467
}