Currently job artifacts in CI/CD pipelines on LRZ GitLab never expire. Starting from Wed 26.1.2022 the default expiration time will be 30 days (GitLab default). Currently existing artifacts in already completed jobs will not be affected by the change. The latest artifacts for all jobs in the latest successful pipelines will be kept. More information: https://gitlab.lrz.de/help/user/admin_area/settings/continuous_integration.html#default-artifacts-expiration

datacontainerinspectorwidget.cpp 19.7 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-2013, 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 Universität München
//      Boltzmannstr. 3, 85748 Garching b. München, 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
27
// 
// ================================================================================================

#include "datacontainerinspectorwidget.h"

#include "tgt/assert.h"
28
#include "tgt/logmanager.h"
29
#include "tgt/filesystem.h"
30
31
32
33
34
#include "tgt/shadermanager.h"
#include "tgt/textureunit.h"
#include "tgt/qt/qtcontextmanager.h"
#include "tgt/qt/qtthreadedcanvas.h"

35
36
37
38
39
40
41
42
#ifdef CAMPVIS_HAS_MODULE_DEVIL
#include <IL/il.h>
#include <IL/ilu.h>
#endif

#include "core/tools/job.h"
#include "core/tools/opengljobprocessor.h"

43
#include "core/datastructures/abstractdata.h"
44
#include "core/datastructures/datacontainer.h"
45
#include "core/datastructures/facegeometry.h"
46
#include "core/datastructures/geometrydata.h"
47
#include "core/datastructures/imagerepresentationgl.h"
48
#include "core/datastructures/renderdata.h"
49

50
51
52
53
#ifdef CAMPVIS_HAS_MODULE_COLUMBIA
#include "modules/columbia/datastructures/fiberdata.h"
#endif

54
#include "application/gui/datacontainertreewidget.h"
55
#include "application/gui/qtdatahandle.h"
56
#include "application//gui/datacontainerfileloaderwidget.h"
57
#include "modules/io/processors/genericimagereader.h"
58

59
60
#include <QFileDialog>

schultezub's avatar
schultezub committed
61
namespace campvis {
62

63
64
    const std::string DataContainerInspectorWidget::loggerCat_ = "CAMPVis.application.DataContainerInspectorWidget";

65
66
    DataContainerInspectorWidget::DataContainerInspectorWidget(QWidget* parent) 
        : QWidget(parent)
67
        , _inited(false)
68
69
70
        , _dataContainer(0)
        , _dctWidget(0)
        , _canvas(0)
71
        , _pcWidget(0)
72
73
74
        , _mainLayout(0)
        , _infoWidget(0)
        , _infoWidgetLayout(0)
75
        , _propEditorWid(0)
76
77
78
79
80
    {
        setupGUI();
    }

    DataContainerInspectorWidget::~DataContainerInspectorWidget() {
81
82
83
        if (_dataContainer != 0) {
            _dataContainer->s_dataAdded.disconnect(this);
        }
84
85
86
87
    }

    void DataContainerInspectorWidget::setDataContainer(DataContainer* dataContainer) {
        if (_dataContainer != 0) {
88
            _dataContainer->s_dataAdded.disconnect(this);
89
90
91
92
        }

        _dataContainer = dataContainer;
        _dctWidget->update(dataContainer);
93
        updateInfoWidget();
94

95
        if (_dataContainer != 0) {
96
            _dataContainer->s_dataAdded.connect(this, &DataContainerInspectorWidget::onDataContainerDataAdded);
97
98
99
        }
    }

100
101
102
103
    DataContainer* DataContainerInspectorWidget::getDataContainer() {
        return _dataContainer;
    }

104
    void DataContainerInspectorWidget::onDataContainerDataAdded(const std::string& key, const DataHandle& dh) {
105
106
        // copy QtDataHandle because signal will be handled by a different thread an indefinite amount of time later:
        emit dataContainerChanged(QString::fromStdString(key), QtDataHandle(dh));
107
108
109
    }

    QSize DataContainerInspectorWidget::sizeHint() const {
110
        return QSize(800, 600);
111
112
    }

113

114
    void DataContainerInspectorWidget::setupGUI() {
schultezub's avatar
schultezub committed
115
116
        setWindowTitle(tr("DataContainer Inspector"));

117
118
119
120
121
        _mainLayout = new QHBoxLayout();
        _mainLayout->setSpacing(4);
        setLayout(_mainLayout);

        _dctWidget = new DataContainerTreeWidget(this);
122
123
        _dctWidget->setSelectionBehavior(QAbstractItemView::SelectRows);
        _dctWidget->setSelectionMode(QAbstractItemView::ExtendedSelection);
124
125
126
127
128
129
130
131
132
133
        _mainLayout->addWidget(_dctWidget);

        _infoWidget = new QWidget(this);
        _infoWidgetLayout = new QVBoxLayout();
        _infoWidgetLayout->setSpacing(4);
        _infoWidget->setLayout(_infoWidgetLayout);

        _lblName = new QLabel(QString("Name: "), _infoWidget);
        _infoWidgetLayout->addWidget(_lblName);

134
135
136
137
138
139
        _lblLocalMemoryFootprint = new QLabel(QString("Local Memory Footprint: "), _infoWidget);
        _infoWidgetLayout->addWidget(_lblLocalMemoryFootprint);

        _lblVideoMemoryFootprint = new QLabel(QString("Video Memory Footprint: "), _infoWidget);
        _infoWidgetLayout->addWidget(_lblVideoMemoryFootprint);

140
141
142
        _lblTimestamp = new QLabel("Timestamp: ", _infoWidget);
        _infoWidgetLayout->addWidget(_lblTimestamp);

143
144
145
        _lblSize = new QLabel(tr("Size: "), _infoWidget);
        _infoWidgetLayout->addWidget(_lblSize);

146
147
148
149
150
151
152
153
        _colorWidget = new QWidget(this);
        _lblColorVal = new QLabel(tr("Color: n/a"), _colorWidget);
        
        _colorValWidget = new QWidget(_colorWidget);
        _colorValWidget->setAutoFillBackground(true);
        _colorValWidget->setFixedSize(16, 16);
        
        _ColorValWidgetPalette = new QPalette(palette());
154
        _ColorValWidgetPalette->setColor(QPalette::Background, Qt::gray);
155
156
157
158
159
160
161
162
163
164
165
166
        _colorValWidget->setPalette(*(_ColorValWidgetPalette));
        
        _colorWidgetLayout = new QHBoxLayout();
        _colorWidgetLayout->setSpacing(0);
        _colorWidgetLayout->setMargin(0);
        _colorWidget->setLayout(_colorWidgetLayout);
        
        _colorWidgetLayout->addWidget(_lblColorVal);
        _colorWidgetLayout->addWidget(_colorValWidget);

        _infoWidgetLayout->addWidget(_colorWidget);
        
167
168
169
        _lblBounds = new QLabel(tr("World Bounds:"), _infoWidget);
        _infoWidgetLayout->addWidget(_lblBounds);

170
171
172
173
174
175

        QWidget* btnWidget = new QWidget(this);
        QGridLayout* gridLayout = new QGridLayout();
        btnWidget->setLayout(gridLayout);

#ifdef CAMPVIS_HAS_MODULE_DEVIL
176
        _btnSaveToFile = new QPushButton(tr("Save to File"), _infoWidget);
177
178
179
180
181
182
        gridLayout->addWidget(_btnSaveToFile, 0, 0);
        connect(
            _btnSaveToFile, SIGNAL(clicked()),
            this, SLOT(onBtnSaveToFileClicked()));
        _btnSaveToFile->setDisabled(true);
#endif
183

184
        _btnLoadFile = new QPushButton(tr("Load File"), _infoWidget);
185
186
187
188
        gridLayout->addWidget(_btnLoadFile, 0, 1);

        _infoWidgetLayout->addWidget(btnWidget);

189

190
        _canvas = new DataContainerInspectorCanvas(_infoWidget);
191
        _canvas->setMinimumSize(QSize(100, 100));
192
        _infoWidgetLayout->addWidget(_canvas, 1);
193

194
        _pcWidget = new PropertyCollectionWidget(_infoWidget);
195
        _pcWidget->updatePropCollection(_canvas, _dataContainer);
196
197
        _infoWidgetLayout->addWidget(_pcWidget);

schultezub's avatar
schultezub committed
198
        _mainLayout->addWidget(_infoWidget, 1);
199

200
        qRegisterMetaType<QtDataHandle>("QtDataHandle");
201
202
203
204
205
206
        connect(
            _dctWidget->selectionModel(), SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)), 
            this, SLOT(onDCTWidgetSelectionModelSelectionChanged(const QItemSelection&, const QItemSelection&)));
        connect(
            this, SIGNAL(dataContainerChanged(const QString&, QtDataHandle)),
            _canvas, SLOT(onDataContainerChanged(const QString&, QtDataHandle)));
207
        connect(
208
209
            this, SIGNAL(dataContainerChanged(const QString&, QtDataHandle)),
            _dctWidget->getTreeModel(), SLOT(onDataContainerChanged(const QString&, QtDataHandle)));
210
        connect(
211
212
            _btnLoadFile, SIGNAL(clicked()),
            this, SLOT(onBtnLoadFileClicked()));
213
214
    }

215
    void DataContainerInspectorWidget::updateColor(){
216
        const tgt::Color& color = _canvas->getCapturedColor();
217
218


219
        _lblColorVal->setText(QString("Color: [%1, %2, %3, %4]").arg(QString::number(color.r), QString::number(color.g), QString::number(color.b), QString::number(color.a)));
220
        
221
        _ColorValWidgetPalette->setColor(QPalette::Background, QColor(static_cast<int>(color.r * 255), static_cast<int>(color.g * 255), static_cast<int>(color.b * 255)));
222
223
        _colorValWidget->setPalette(*_ColorValWidgetPalette);
    }
224

225
    void DataContainerInspectorWidget::updateDepth(){
226
227
228
229
230

        float depth = _canvas->getCapturedDepth();

        _lblColorVal->setText(QString("Depth: %1").arg(QString::number(depth)));
        
231
        _ColorValWidgetPalette->setColor(QPalette::Background, QColor(static_cast<int>(depth * 255), static_cast<int>(depth * 255), static_cast<int>(depth * 255)));
232
233
234
        _colorValWidget->setPalette(*_ColorValWidgetPalette);
    }

235
    void DataContainerInspectorWidget::updateInfoWidget() {
236
237
238
        if (!_inited)
            return;

239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
        // get the selection from the tree widget
        const QModelIndexList& indices = _dctWidget->selectionModel()->selectedRows();
        std::vector< std::pair<QString, QtDataHandle> > handles;
        float _localFootprint = 0.f;
        float _videoFootprint = 0.f;

        // iterate through the indices of the selection
        for (QModelIndexList::const_iterator index = indices.begin(); index != indices.end(); ++index) {
            if (! index->isValid())
                continue;
            // get DataHandle and Handle name
            QVariant item = index->data(Qt::UserRole);
            QtDataHandle handle = item.value<QtDataHandle>();
            QModelIndex idxName = index->sibling(index->row(), 0);

            // only consider non-empty DataHandles
            if (handle.getData() != 0) {
                handles.push_back(std::make_pair(idxName.data(Qt::DisplayRole).toString(), handle));
                _localFootprint += handle.getData()->getLocalMemoryFootprint();
                _videoFootprint += handle.getData()->getVideoMemoryFootprint();
            }
schultezub's avatar
schultezub committed
260
        }
261

262
263
264
265
        // update labels
        if (handles.size() == 1) {
            _lblName->setText("Name: " + handles.front().first);
            _lblTimestamp->setText("Timestamp: " + QString::number(handles.front().second.getTimestamp()));
266

267
            if (const ImageData* tester = dynamic_cast<const ImageData*>(handles.front().second.getData())) {
268
                _canvas->p_transferFunction.setImageHandle(handles.front().second);
269
270
271
272
273
274
275
276
                std::ostringstream ss;

                ss << tester->getSize();
                _lblSize->setText(tr("Size: ") + QString::fromStdString(ss.str()));

                ss.str("");
                ss << tester->getWorldBounds();
                _lblBounds->setText(tr("World Bounds: ") + QString::fromStdString(ss.str())); 
277

278
279
280
281
            }
            else if (const GeometryData* tester = dynamic_cast<const GeometryData*>(handles.front().second.getData())) {
                _lblSize->setText(tr("Size: n/a"));

282
                std::ostringstream ss;
283
                ss << tester->getWorldBounds();
284
                _lblBounds->setText(tr("World Bounds: ") + QString::fromStdString(ss.str()));
285
            }
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
            else if (const RenderData* tester = dynamic_cast<const RenderData*>(handles.front().second.getData())) {
                const ImageData* id = tester->getNumColorTextures() > 0 ? tester->getColorTexture() : tester->getDepthTexture();
                if (id != 0) {
                    std::ostringstream ss;
                    ss << id->getSize();
                    _lblSize->setText(tr("Size: ") + QString::fromStdString(ss.str()));

                    ss.str("");
                    ss << id->getWorldBounds();
                    _lblBounds->setText(tr("World Bounds: ") + QString::fromStdString(ss.str())); 
                }
                else {
                    _lblSize->setText(tr("Size: n/a"));
                    _lblBounds->setText(tr("World Bounds: n/a")); 
                }
            }
302
303
304
305
306
307
308
309
310
311
312
313
#ifdef CAMPVIS_HAS_MODULE_COLUMBIA
            else if (const FiberData* tester = dynamic_cast<const FiberData*>(handles.front().second.getData())) {
                std::ostringstream ss;
                ss << "Size: " << tester->numFibers() << " Fibers with " << tester->numSegments() << " Segments.";
                _lblSize->setText(QString::fromStdString(ss.str()));

                ss.str("");
                ss << tester->getWorldBounds();
                _lblBounds->setText(tr("World Bounds: ") + QString::fromStdString(ss.str())); 
            }
#endif
            else {
314

315
                _lblSize->setText(tr("Size: n/a"));
316
317
                _lblBounds->setText(tr("World Bounds: n/a")); 
            }
318
319
320
321
        }
        else {
            _lblName->setText(QString::number(handles.size()) + " DataHandles selected");
            _lblTimestamp->setText("Timestamp: n/a");
322

323
            _canvas->p_transferFunction.setImageHandle(DataHandle(0));
324
        }
325
326
327
328
        _lblLocalMemoryFootprint->setText("Local Memory Footprint: " + humanizeBytes(_localFootprint));
        _lblVideoMemoryFootprint->setText("Video Memory Footprint: " + humanizeBytes(_videoFootprint));

        // update DataHandles for the DataContainerInspectorCanvas
329
        _canvas->setDataHandles(handles);
330
331
    }

332
333
334
335
336
    QString DataContainerInspectorWidget::humanizeBytes(size_t numBytes) const {
        QString units[5] = { tr(" Bytes"), tr(" KB"), tr(" MB"), tr(" GB"), tr(" TB") };
        size_t index = 0;
        size_t remainder = 0;

337
        while (numBytes > 1024 && index < 4) {
338
339
340
341
342
343
344
345
346
347
348
            remainder = numBytes % 1024;
            numBytes /= 1024;
            ++index;
        }

        if (remainder != 0)
            return QString::number(numBytes) + "." + QString::number(remainder) + units[index];
        else
            return QString::number(numBytes) + units[index];
    }

349
350
    void DataContainerInspectorWidget::init() {
        if (_canvas != 0)
351
            _canvas->init(this);
352
353

        _inited = true;
354
355
356
    }

    void DataContainerInspectorWidget::deinit() {
357
        _inited = false;
358
359
        if (_canvas != 0)
            _canvas->deinit();
schultezub's avatar
schultezub committed
360

361
        _pcWidget->updatePropCollection(0, 0);
362
363
364
365
366
367

        if (_dataContainer != 0) {
            _dataContainer->s_dataAdded.disconnect(this);
        }

        _dataContainer = 0;
368
        _dctWidget->update(0);
369
370
371

        if(_propEditorWid != nullptr)
            _propEditorWid->deinit();
372
373
    }

374
375
    void DataContainerInspectorWidget::onDCTWidgetSelectionModelSelectionChanged(const QItemSelection& selected, const QItemSelection& deselected) {
        updateInfoWidget();
376
377
378
379
380
381
382
383
384
385
386
387

        // get the selection from the tree widget
        const QModelIndexList& indices = _dctWidget->selectionModel()->selectedRows();

        // iterate through the indices of the selection
        for (QModelIndexList::const_iterator index = indices.begin(); index != indices.end(); ++index) {
            if (index->isValid()) {
                _btnSaveToFile->setDisabled(false);
                return;
            }
        }
        _btnSaveToFile->setDisabled(true);
388
389
    }

390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
    void DataContainerInspectorWidget::onBtnSaveToFileClicked() {
        // get the selection from the tree widget
        const QModelIndexList& indices = _dctWidget->selectionModel()->selectedRows();

        // iterate through the indices of the selection
        for (QModelIndexList::const_iterator index = indices.begin(); index != indices.end(); ++index) {
            if (! index->isValid())
                continue;

            // get DataHandle and Handle name
            QVariant item = index->data(Qt::UserRole);
            DataHandle handle = item.value<QtDataHandle>();
            QModelIndex idxName = index->sibling(index->row(), 0);

            // only consider non-empty DataHandles that are ImageData and have render target representations
            if (handle.getData() != 0) {
406
407
408
409
410
411
412
413
414
415
416
417
418
                if (dynamic_cast<const ImageData*>(handle.getData()) || dynamic_cast<const RenderData*>(handle.getData())) {
                    QString dialogCaption = "Export " + idxName.data(Qt::DisplayRole).toString() + " as Image";
                    QString directory = tr("");
                    const QString fileFilter = tr("*.png;;PNG images (*.png)");

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

                    if (! filename.isEmpty()) {
                        // Texture access needs OpenGL context - dispatch method call:
                        GLJobProc.enqueueJob(
                            _canvas, 
                            makeJobOnHeap(&DataContainerInspectorWidget::saveToFile, handle, filename.toStdString()), 
                            OpenGLJobProcessor::SerialJob);
419
420
421
422
423
424
425
426
427
428
429
430
431
                    }
                }
            }
        }
    }

    void DataContainerInspectorWidget::saveToFile(DataHandle handle, std::string filename) {
#ifdef CAMPVIS_HAS_MODULE_DEVIL
        if (tgt::FileSystem::fileExtension(filename).empty()) {
            LERRORC("CAMPVis.application.DataContainerInspectorWidget", "Filename has no extension");
            return;
        }

432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
        // get the ImageData object (either directly or from the RenderData)
        const ImageData* id = 0;
        if (const RenderData* tester = dynamic_cast<const RenderData*>(handle.getData())) {
            id = tester->getColorTexture(0);
        }
        else if (const ImageData* tester = dynamic_cast<const ImageData*>(handle.getData())) {
            id = tester;
        }
        else {
            LERROR("Could not extract image to save.");
            return;
        }

        // extract the data
        WeaklyTypedPointer wtp(WeaklyTypedPointer::UINT8, 1, 0);
        const ImageRepresentationGL* repGL = id->getRepresentation<ImageRepresentationGL>(false);
        if (repGL != 0) // if it's a GL texture, download it (we do not want to use the automatic conversion method here)
            wtp = repGL->getWeaklyTypedPointer();
        else {
            const ImageRepresentationLocal* repLocal = id->getRepresentation<ImageRepresentationLocal>(true);
            if (repLocal != 0)
                wtp = repLocal->getWeaklyTypedPointer();
        }

        if (wtp._pointer == 0) {
            LERROR("Could not extract image to save.");
            return;
        }
460
461
462
463
464
465
466
467


        // create Devil image from image data and write it to file
        ILuint img;
        ilGenImages(1, &img);
        ilBindImage(img);

        // put pixels into IL-Image
468
469
        tgt::ivec2 size = id->getSize().xy();
        ilTexImage(size.x, size.y, 1, static_cast<ILubyte>(wtp._numChannels), wtp.getIlFormat(), wtp.getIlDataType(), wtp._pointer);
470
471
472
473
474
475
476
477
478
479
480
481
482
483
        ilEnable(IL_FILE_OVERWRITE);
        ilResetWrite();
        ILboolean success = ilSaveImage(filename.c_str());
        ilDeleteImages(1, &img);

        if (!success) {
            LERRORC("CAMPVis.application.DataContainerInspectorWidget", "Could not save image to file: " << ilGetError());
        }
#else
        return;
#endif

    }

484
    void DataContainerInspectorWidget::onBtnLoadFileClicked() {
485
486
487
        // delete previous PropertyEditor, then create a new one
        // the final one will be deleted with deinit()
        if(nullptr != _propEditorWid)
488
            _propEditorWid->deinit();
489

490
        _propEditorWid = new DataContainerFileLoaderWidget(this, nullptr);
491
492
        _propEditorWid->setVisible(true);
        
493
494
    }

495
}