datacontainerinspectorwidget.cpp 18.3 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, 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
schultezub's avatar
schultezub committed
10
// For a full list of authors and contributors, please refer to the file "AUTHORS.txt".
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
// 
// The licensing of this softare is not yet resolved. Until then, redistribution in source or
// binary forms outside the CAMP chair is not permitted, unless explicitly stated in legal form.
// However, the names of the original authors and the above copyright notice must retain in its
// original state in any case.
// 
// Legal disclaimer provided by the BSD license:
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
// IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY 
// AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR 
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 
// OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
// 
// ================================================================================================

#include "datacontainerinspectorwidget.h"

#include "tgt/assert.h"
33
#include "tgt/logmanager.h"
34
#include "tgt/filesystem.h"
35
36
37
38
39
#include "tgt/shadermanager.h"
#include "tgt/textureunit.h"
#include "tgt/qt/qtcontextmanager.h"
#include "tgt/qt/qtthreadedcanvas.h"

40
41
42
43
44
45
46
47
#ifdef CAMPVIS_HAS_MODULE_DEVIL
#include <IL/il.h>
#include <IL/ilu.h>
#endif

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

48
#include "core/datastructures/abstractdata.h"
49
#include "core/datastructures/datacontainer.h"
50
#include "core/datastructures/facegeometry.h"
51
#include "core/datastructures/geometrydata.h"
52
#include "core/datastructures/imagerepresentationgl.h"
53
#include "core/datastructures/renderdata.h"
54

55
56
57
58
#ifdef CAMPVIS_HAS_MODULE_COLUMBIA
#include "modules/columbia/datastructures/fiberdata.h"
#endif

59
#include "application/gui/datacontainertreewidget.h"
60
#include "application/gui/qtdatahandle.h"
61

62
63
#include <QFileDialog>

schultezub's avatar
schultezub committed
64
namespace campvis {
65

66
67
    const std::string DataContainerInspectorWidget::loggerCat_ = "CAMPVis.application.DataContainerInspectorWidget";

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

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

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

        _dataContainer = dataContainer;
        _dctWidget->update(dataContainer);
95
        updateInfoWidget();
96

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

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

    QSize DataContainerInspectorWidget::sizeHint() const {
108
        return QSize(800, 600);
109
110
    }

111

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

115
116
117
118
119
        _mainLayout = new QHBoxLayout();
        _mainLayout->setSpacing(4);
        setLayout(_mainLayout);

        _dctWidget = new DataContainerTreeWidget(this);
120
121
        _dctWidget->setSelectionBehavior(QAbstractItemView::SelectRows);
        _dctWidget->setSelectionMode(QAbstractItemView::ExtendedSelection);
122
123
124
125
126
127
128
129
130
131
        _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);

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

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

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

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

144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
		
		_wdgtColor = new QWidget(this);
		//_lblColor = new QLabel(tr("Color:"), _wdgtColor);
		_lblColorVal = new QLabel(tr("Color: n/a"), _wdgtColor);
		
		_wdgtColorVal = new QWidget(this);
		_wdgtColorVal->setAutoFillBackground(true);
		_wdgtColorVal->setFixedSize(16, 16);
		
		_wdgtColorValPalette = new QPalette(palette());
		_wdgtColorValPalette->setColor(QPalette::ColorRole::Background, Qt::gray);
		_wdgtColorVal->setPalette(*(_wdgtColorValPalette));
		
		
		_wdgtColorLayout = new QHBoxLayout();
		_wdgtColorLayout->setSpacing(0);
		_wdgtColorLayout->setMargin(0);
		//_wdgtColorLayout->setAlignment(Qt::Alignment::enum_type::AlignLeft);
		_wdgtColor->setLayout(_wdgtColorLayout);
		
		//_wdgtColorLayout->addWidget(_lblColor);
		_wdgtColorLayout->addWidget(_lblColorVal);
		_wdgtColorLayout->addWidget(_wdgtColorVal);

		_infoWidgetLayout->addWidget(_wdgtColor);
		
170
171
172
        _lblBounds = new QLabel(tr("World Bounds:"), _infoWidget);
        _infoWidgetLayout->addWidget(_lblBounds);

173
174
175
        _btnSaveToFile = new QPushButton(tr("Save to File"), _infoWidget);
        _infoWidgetLayout->addWidget(_btnSaveToFile);

176
        _canvas = new DataContainerInspectorCanvas(_infoWidget);
177
        _canvas->setMinimumSize(QSize(100, 100));
178
        _infoWidgetLayout->addWidget(_canvas, 1);
179

180
        _pcWidget = new PropertyCollectionWidget(_infoWidget);
181
        _pcWidget->updatePropCollection(_canvas, _dataContainer);
182
183
        _infoWidgetLayout->addWidget(_pcWidget);

schultezub's avatar
schultezub committed
184
        _mainLayout->addWidget(_infoWidget, 1);
185

186
        qRegisterMetaType<QtDataHandle>("QtDataHandle");
187
188
189
190
191
192
        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)));
193
        connect(
194
195
            this, SIGNAL(dataContainerChanged(const QString&, QtDataHandle)),
            _dctWidget->getTreeModel(), SLOT(onDataContainerChanged(const QString&, QtDataHandle)));
196
197
198
        connect(
            _btnSaveToFile, SIGNAL(clicked()),
            this, SLOT(onBtnSaveToFileClicked()));
199
200
    }

201
202
203
204
205
206
207
208
209
210
211
	void DataContainerInspectorWidget::updateColor(){

		char bufferR[10];
		char bufferG[10];
		char bufferB[10];
		tgt::Color color = _canvas->getCapturedColor();

		itoa((int)(color.r * 255), bufferR, 10);
		itoa((int)(color.g * 255), bufferG, 10);
		itoa((int)(color.b * 255), bufferB, 10);

212
213
214
215
216
217
		_lblColorVal->setText(QString("Color: R = ") + QString(bufferR) + QString(" G = ") + QString(bufferG) + QString(" B = ") + QString(bufferB));
		QColor qtColor;
		qtColor.setRgb(color.r * 255, color.g * 255, color.b * 255);
		
		_wdgtColorValPalette->setColor(QPalette::ColorRole::Background, qtColor);
		_wdgtColorVal->setPalette(*_wdgtColorValPalette);
218
219
	}

220
    void DataContainerInspectorWidget::updateInfoWidget() {
221
222
223
        if (!_inited)
            return;

224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
        // 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
245
        }
246

247
248
249
250
        // update labels
        if (handles.size() == 1) {
            _lblName->setText("Name: " + handles.front().first);
            _lblTimestamp->setText("Timestamp: " + QString::number(handles.front().second.getTimestamp()));
251

252
            if (const ImageData* tester = dynamic_cast<const ImageData*>(handles.front().second.getData())) {
schultezub's avatar
schultezub committed
253
                _canvas->p_transferFunction.getTF()->setImageHandle(handles.front().second);
254
255
256
257
258
259
260
261
                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())); 
262

263
264
265
266
            }
            else if (const GeometryData* tester = dynamic_cast<const GeometryData*>(handles.front().second.getData())) {
                _lblSize->setText(tr("Size: n/a"));

267
				std::ostringstream ss;
268
269
                ss << tester->getWorldBounds();
                _lblBounds->setText(tr("World Bounds: ") + QString::fromStdString(ss.str())); 
270
271

				//tester->render();
272
            }
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
            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")); 
                }
            }
289
290
291
292
293
294
295
296
297
298
299
300
#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 {
301

302
				_lblSize->setText(tr("Size: n/a"));
303
304
                _lblBounds->setText(tr("World Bounds: n/a")); 
            }
305
306
307
308
        }
        else {
            _lblName->setText(QString::number(handles.size()) + " DataHandles selected");
            _lblTimestamp->setText("Timestamp: n/a");
309

310
            _canvas->p_transferFunction.getTF()->setImageHandle(DataHandle(0));
311
        }
312
313
314
315
        _lblLocalMemoryFootprint->setText("Local Memory Footprint: " + humanizeBytes(_localFootprint));
        _lblVideoMemoryFootprint->setText("Video Memory Footprint: " + humanizeBytes(_videoFootprint));

        // update DataHandles for the DataContainerInspectorCanvas
316
        _canvas->setDataHandles(handles);
317
318
    }

319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
    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;

        while (numBytes > 1024 && index < 5) {
            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];
    }

336
337
    void DataContainerInspectorWidget::init() {
        if (_canvas != 0)
338
            _canvas->init(this);
339
340

        _inited = true;
341
342
343
    }

    void DataContainerInspectorWidget::deinit() {
344
        _inited = false;
345
346
        if (_canvas != 0)
            _canvas->deinit();
schultezub's avatar
schultezub committed
347

348
        _pcWidget->updatePropCollection(0, 0);
349
350
351
352
353
354

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

        _dataContainer = 0;
355
        _dctWidget->update(0);
356
357
    }

358
359
360
361
    void DataContainerInspectorWidget::onDCTWidgetSelectionModelSelectionChanged(const QItemSelection& selected, const QItemSelection& deselected) {
        updateInfoWidget();
    }

362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
    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) {
378
379
380
381
382
383
384
385
386
387
388
389
390
                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);
391
392
393
394
395
396
397
398
399
400
401
402
403
                    }
                }
            }
        }
    }

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

404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
        // 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;
        }
432
433
434
435
436
437
438
439


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

        // put pixels into IL-Image
440
441
        tgt::ivec2 size = id->getSize().xy();
        ilTexImage(size.x, size.y, 1, static_cast<ILubyte>(wtp._numChannels), wtp.getIlFormat(), wtp.getIlDataType(), wtp._pointer);
442
443
444
445
446
447
448
449
450
451
452
453
454
455
        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

    }

456
}