datacontainerinspectorcanvas.cpp 15.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, 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
33
// 
// 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 "datacontainerinspectorcanvas.h"

#include "tgt/assert.h"
#include "tgt/shadermanager.h"
34
35
#include "tgt/textureunit.h"

36
37
#include "core/datastructures/datacontainer.h"
#include "core/datastructures/datahandle.h"
38
#include "core/datastructures/renderdata.h"
39
#include "core/datastructures/imagerepresentationgl.h"
40
#include "core/datastructures/facegeometry.h"
41
#include "core/datastructures/meshgeometry.h"
42
#include "core/tools/job.h"
43
44
#include "core/classification/tfgeometry1d.h"
#include "core/classification/geometry1dtransferfunction.h"
45

46
47
#include "datacontainerinspectorwidget.h"

48

schultezub's avatar
schultezub committed
49
namespace campvis {
50
51
52

    DataContainerInspectorCanvas::DataContainerInspectorCanvas(QWidget* parent /*= 0*/) 
        : tgt::QtThreadedCanvas("DataContainer Inspector", tgt::ivec2(640, 480), tgt::GLCanvas::RGBA_BUFFER, parent, true)
53
        , p_currentSlice("CurrentSlice", "Slice", -1, -1, -1)
54
        , p_transferFunction("TransferFunction", "Transfer Function", new Geometry1DTransferFunction(256, tgt::vec2(0.f, 1.f)))
55
56
        , _dataContainer(0)
        , _paintShader(0)
57
        , _quad(0)
58
59
60
61
        , _numTiles(0, 0)
        , _quadSize(0, 0)
        , _selectedTexture(0)
        , _renderFullscreen(false)
62
        , _currentSlice(-1)
63
		, _color(0.0f, 0.0f, 0.0f, 0.0f)
64
    {
65
        static_cast<Geometry1DTransferFunction*>(p_transferFunction.getTF())->addGeometry(TFGeometry1D::createQuad(tgt::vec2(0.f, 1.f), tgt::col4(0, 0, 0, 255), tgt::col4(255, 255, 255, 255)));
66
67
68
69
70
71
72
73
74
75

        makeCurrent();
        // Init GLEW for this context
        GLenum err = glewInit();
        if (err != GLEW_OK) {
            // Problem: glewInit failed, something is seriously wrong.
            tgtAssert(false, "glewInit failed");
            std::cerr << "glewInit failed, error: " << glewGetErrorString(err) << std::endl;
            exit(EXIT_FAILURE);
        }
76
77
78

        addProperty(&p_currentSlice);
        addProperty(&p_transferFunction);
79
80
81
82
83
84
    }

    DataContainerInspectorCanvas::~DataContainerInspectorCanvas() {

    }

85
    void DataContainerInspectorCanvas::init(DataContainerInspectorWidget* _pWidget) {
schultezub's avatar
schultezub committed
86
87
        initAllProperties();

88
89
		_widget = _pWidget;

90
91
92
93
        GLJobProc.registerContext(this);
        _paintShader = ShdrMgr.loadSeparate("core/glsl/passthrough.vert", "application/glsl/datacontainerinspector.frag", "", false);
        _paintShader->setAttributeLocation(0, "in_Position");
        _paintShader->setAttributeLocation(1, "in_TexCoords");
94

95
96
        createQuad();

97
        // set this as painter to get notified when window size changes
98
        setPainter(this, false);
99
        getEventHandler()->addEventListenerToFront(this);
100
101
102
    }

    void DataContainerInspectorCanvas::deinit() {
schultezub's avatar
schultezub committed
103
104
        deinitAllProperties();

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

109
        _handles.clear();
110
        GLJobProc.deregisterContext(this);
111
        ShdrMgr.dispose(_paintShader);
112
        delete _quad;
113
114
115
116
117
118
119
120
121
    }

    QSize DataContainerInspectorCanvas::sizeHint() const {
        return QSize(640, 480);
    }


    void DataContainerInspectorCanvas::paint() {
        tbb::mutex::scoped_lock lock(_localMutex);
122
123
124
        if (_texturesDirty)
            updateTextures();

125
126
        if (_textures.empty())
            return;
127
128

        glPushAttrib(GL_ALL_ATTRIB_BITS);
129
        glViewport(0, 0, size_.x, size_.y);
130
131
132
133
134
        glClearColor(0.7f, 0.7f, 0.7f, 1.f);
        glClear(GL_COLOR_BUFFER_BIT);
        LGL_ERROR;

        // update layout dimensions
135
136
        _numTiles.x = ceil(sqrt(static_cast<float>(_textures.size())));
        _numTiles.y = ceil(static_cast<float>(_textures.size()) / _numTiles.x);
137
        _quadSize = size_ / _numTiles;
138
139
140
141
142
143

        _paintShader->activate();

        tgt::mat4 projection = tgt::mat4::createOrtho(0, size_.x, 0, size_.y, -1, 1);
        _paintShader->setUniform("_projectionMatrix", projection);

144
145
        tgt::TextureUnit tfUnit, unit2d, unit3d;
        p_transferFunction.getTF()->bind(_paintShader, tfUnit);
146
147
        _paintShader->setUniform("_texture2d", unit2d.getUnitNumber());
        _paintShader->setUniform("_texture3d", unit3d.getUnitNumber());
148

149
150
151
152
153
154
155
156
157
158
        for (int y = 0; y < _numTiles.y; ++y) {
            for (int x = 0; x < _numTiles.x; ++x) {
                int index = (_numTiles.x * y) + x;
                if (index >= static_cast<int>(_textures.size()))
                    break;

                tgt::mat4 scaleMatrix = tgt::mat4::createScale(tgt::vec3(_quadSize, 1.f));
                tgt::mat4 translation = tgt::mat4::createTranslation(tgt::vec3(_quadSize.x * x, _quadSize.y * y, 0.f));
                _paintShader->setUniform("_modelMatrix", translation * scaleMatrix);
                paintTexture(_textures[index], unit2d, unit3d);
159
160
161
            }
        }

162
        _paintShader->deactivate();
163
164
165
166
        LGL_ERROR;
        glPopAttrib();
    }

167
    void DataContainerInspectorCanvas::paintTexture(const tgt::Texture* texture, const tgt::TextureUnit& unit2d, const tgt::TextureUnit& unit3d) {
168
169

        _paintShader->setIgnoreUniformLocationError(true);
170
171
172
173
        if (texture->getDimensions().z == 1) {
            unit2d.activate();
            texture->bind();
            _paintShader->setUniform("_is3d", false);
174
            _paintShader->setUniform("_isDepthTexture", texture->isDepthTexture());
175
176
177
            _paintShader->setUniform("_2dTextureParams._size", tgt::vec2(texture->getDimensions().xy()));
            _paintShader->setUniform("_2dTextureParams._sizeRCP", tgt::vec2(1.f) / tgt::vec2(texture->getDimensions().xy()));
            _paintShader->setUniform("_2dTextureParams._numChannels", static_cast<int>(texture->getNumChannels()));
178
179
        }
        else {
180
181
182
            // clamp current slice to texture size, since this can't be done in event handler:
            _currentSlice = tgt::clamp(_currentSlice, -1, texture->getDimensions().z);

183
184
185
            unit3d.activate();
            texture->bind();
            _paintShader->setUniform("_is3d", true);
186
            _paintShader->setUniform("_sliceNumber", p_currentSlice.getValue());
187
188
189
            _paintShader->setUniform("_3dTextureParams._size", tgt::vec3(texture->getDimensions()));
            _paintShader->setUniform("_3dTextureParams._sizeRCP", tgt::vec3(1.f) / tgt::vec3(texture->getDimensions()));
            _paintShader->setUniform("_3dTextureParams._numChannels", static_cast<int>(texture->getNumChannels()));
190
        }
191
192
        _paintShader->setIgnoreUniformLocationError(false);

193
        _quad->render(GL_POLYGON);
194
195
196
    }

    void DataContainerInspectorCanvas::invalidate() {
Christian Schulte zu Berge's avatar
Christian Schulte zu Berge committed
197
198
199
        // only if inited
        if (_quad != 0 && _paintShader != 0)
            GLJobProc.enqueueJob(this, makeJobOnHeap(this, &DataContainerInspectorCanvas::paint), OpenGLJobProcessor::PaintJob);
200
201
    }

202
    void DataContainerInspectorCanvas::createQuad() {
203
204
205
        std::vector<tgt::vec3> vertices, texCorods;

        vertices.push_back(tgt::vec3( 0.f,  0.f, 0.f));
206
207
208
        vertices.push_back(tgt::vec3(1.f, 0.f, 0.f));
        vertices.push_back(tgt::vec3(1.f, 1.f, 0.f));
        vertices.push_back(tgt::vec3( 0.f,  1.f, 0.f));
209
210
211
212
213
214
215
216
217
218
        texCorods.push_back(tgt::vec3(0.f, 1.f, 0.f));
        texCorods.push_back(tgt::vec3(1.f, 1.f, 0.f));
        texCorods.push_back(tgt::vec3(1.f, 0.f, 0.f));
        texCorods.push_back(tgt::vec3(0.f, 0.f, 0.f));

        delete _quad;
        _quad = new FaceGeometry(vertices, texCorods);
        _quad->createGLBuffers();
    }

219
220
221
222
    void DataContainerInspectorCanvas::repaint() {
        invalidate();
    }

223
224
225
226
    void DataContainerInspectorCanvas::sizeChanged(const tgt::ivec2&) {
        invalidate();
    }

227
228
229
230
231
232
233
234
235
    void DataContainerInspectorCanvas::mouseDoubleClickEvent(tgt::MouseEvent* e) {
        if (_renderFullscreen) {
            _renderFullscreen = false;
        }
        else {
            tgt::ivec2 selectedIndex(e->x() / _quadSize.x, e->y() / _quadSize.y);
            _selectedTexture = (selectedIndex.y * _numTiles.x) + selectedIndex.x;
            _renderFullscreen = true;
        }
236
        e->ignore();
237
238
239
        invalidate();
    }

240
	void DataContainerInspectorCanvas::mouseMoveEvent(tgt::MouseEvent* e)
241
242
243
244
245
246
247
248
249
250
251
	{
        /*if (_renderFullscreen) {
            _renderFullscreen = false;
        }
        else {
            tgt::ivec2 selectedIndex(e->x() / _quadSize.x, e->y() / _quadSize.y);
            _selectedTexture = (selectedIndex.y * _numTiles.x) + selectedIndex.x;
            _renderFullscreen = true;
        }
        e->ignore();
        invalidate();*/
252
253
254
255
256
		static clock_t lastRunTime = clock();
		if(!(clock() - lastRunTime > 0))
		{
			return;
		}
257
		
258
259
		if(e->button() == tgt::MouseEvent::MOUSE_BUTTON_RIGHT)
		{
260

261
			tgt::ivec2 dimCanvas = tgt::ivec2(_quadSize.x * _numTiles.x, _quadSize.y * _numTiles.y);	
262

263
			if(e->x() >= dimCanvas.x || e->y() >= dimCanvas.y || e->x() < 0 || e->y() < 0)
264
265
266
				return;

			int texIndx = (e->y() / _quadSize.y) * _numTiles.x + (e->x() / _quadSize.x);
267
			
268
269
270
			const tgt::Texture* texturePtr = _textures[texIndx];
			const int texWidth  = texturePtr->getWidth();
			const int texHeight = texturePtr->getHeight();
271

272
273
			int cursorPosX = (float)(e->x() % _quadSize.x) / _quadSize.x * texWidth;
			int cursorPosY = (float)(e->y() % _quadSize.y) / _quadSize.y * texHeight;
274

275
			_color = _textures[texIndx]->texelAsFloat(cursorPosX, texHeight - cursorPosY - 1);
276
277
			
			_widget->updateColor();
278
		}
279
280
    }

281
282
283
284
285
286
287
288
289
290
291
292
    void DataContainerInspectorCanvas::wheelEvent(tgt::MouseEvent* e) {
        if (_renderFullscreen) {
            switch (e->button()) {
                case tgt::MouseEvent::MOUSE_WHEEL_UP:
                    ++_currentSlice; // we cant clamp the value here to the number of slices - we do this during rendering
                    e->ignore();
                    break;
                case tgt::MouseEvent::MOUSE_WHEEL_DOWN:
                    if (_currentSlice >= -1)
                        --_currentSlice;
                    e->ignore();
                    break;
293
294
                default:
                    break;
295
296
297
298
299
            }
            invalidate();
        }
    }

300
301
302
303
304
305
306
307
308
309
310
311
312
    void DataContainerInspectorCanvas::onDataContainerChanged(const QString& key, QtDataHandle dh) {
        {
            tbb::mutex::scoped_lock lock(_localMutex);

            // check whether DataHandle is already existing
            std::map<QString, QtDataHandle>::iterator lb = _handles.lower_bound(key);
            if (lb == _handles.end() || lb->first != key) {
                // not existant -> do nothing
            }
            else {
                // existant -> replace
                lb->second = QtDataHandle(dh);
                // update _textures array
313
                _texturesDirty = true;
314
315
            }
        }
316
317
318

        if (_texturesDirty)
            invalidate();
319
320
321
322
323
324
325
326
327
    }

    void DataContainerInspectorCanvas::setDataHandles(const std::vector< std::pair<QString, QtDataHandle> >& handles) {
        {
            tbb::mutex::scoped_lock lock(_localMutex);
            _handles.clear();
            for (std::vector< std::pair<QString, QtDataHandle> >::const_iterator it = handles.begin(); it != handles.end(); ++it)
                _handles.insert(*it);

328
            _texturesDirty = true;
329
330
331
332
333
        }

        invalidate();
    }

334
335
336
337
338
	tgt::Color DataContainerInspectorCanvas::getCapturedColor()
	{
		return _color;
	}

339
    void DataContainerInspectorCanvas::updateTextures() {
340
		/// Clear the textures Array
341
        _textures.clear();
342
343

		/// Calculate the maximum slices of the textures and fill the textures array
344
        int maxSlices = 1;
345
346
        for (std::map<QString, QtDataHandle>::iterator it = _handles.begin(); it != _handles.end(); ++it) {
            if (const ImageData* img = dynamic_cast<const ImageData*>(it->second.getData())) {
347
                if (const ImageRepresentationGL* imgGL = img->getRepresentation<ImageRepresentationGL>()) {
348
                    _textures.push_back(imgGL->getTexture());
349
                    maxSlices = std::max(maxSlices, imgGL->getTexture()->getDimensions().z);
350
351
                }
            }
352
353
354
355
356
357
358
359
360
361
362
363
364
            else if (const RenderData* rd = dynamic_cast<const RenderData*>(it->second.getData())) {
                for (size_t i = 0; i < rd->getNumColorTextures(); ++i) {
                    const ImageRepresentationGL* imgGL = rd->getColorTexture(i)->getRepresentation<ImageRepresentationGL>();
                    if (imgGL)
                        _textures.push_back(imgGL->getTexture());
                }
                if (rd->hasDepthTexture()) {
                    const ImageRepresentationGL* imgGL = rd->getDepthTexture()->getRepresentation<ImageRepresentationGL>();
                    if (imgGL)
                        _textures.push_back(imgGL->getTexture());

                }
            }
365
366
367
368
369
370
371
			else if(const campvis::MeshGeometry* mg = dynamic_cast<const campvis::MeshGeometry*>(it->second.getData())){

				// Here the object will be rendered into a texture and the texture will be shown on the output buffer.

				//mg->render(GL_POLYGON);
				//const tgt::BufferObject* buffer = mg->getColorsBuffer();
			}
372
        }
373
374
375
376

        if (maxSlices == 1)
            maxSlices = -1;
        p_currentSlice.setMaxValue(maxSlices);
377
        _texturesDirty = false;
378
379
380
381
382
383
384
    }

    void DataContainerInspectorCanvas::onPropertyChanged(const AbstractProperty* prop) {
        HasPropertyCollection::onPropertyChanged(prop);
        invalidate();
    }

385
}