eepgenerator.cpp 7.93 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
// ================================================================================================
// 
// This file is part of the TUMVis Visualization Framework.
// 
// If not explicitly stated otherwise: Copyright (C) 2012, all rights reserved,
//      Christian Schulte zu Berge (christian.szb@in.tum.de)
//      Chair for Computer Aided Medical Procedures
//      Technische Universitt Mnchen
//      Boltzmannstr. 3, 85748 Garching b. Mnchen, Germany
// 
// 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.
// 
// ================================================================================================

29
30
31
32
33
34
35
36
37
#include "eepgenerator.h"

#include "tgt/glmath.h"
#include "tgt/logmanager.h"
#include "tgt/shadermanager.h"
#include "tgt/textureunit.h"

#include "core/datastructures/imagedatagl.h"
#include "core/datastructures/imagedatarendertarget.h"
38
#include "core/datastructures/meshgeometry.h"
39
40
41
42
43
44

namespace TUMVis {
    const std::string EEPGenerator::loggerCat_ = "TUMVis.modules.vis.EEPGenerator";

    EEPGenerator::EEPGenerator(GenericProperty<tgt::ivec2>& canvasSize)
        : VisualizationProcessor(canvasSize)
schultezub's avatar
schultezub committed
45
46
47
48
49
        , _sourceImageID("sourceImageID", "Input Image", "", DataNameProperty::READ)
        , _geometryID("geometryID", "Input Geometry ID", "proxygeometry", DataNameProperty::READ)
        , _mirrorID("mirrorID", "Input Mirror ID", "mirror", DataNameProperty::READ)
        , _entryImageID("entryImageID", "Output Entry Points Image", "eep.entry", DataNameProperty::WRITE)
        , _exitImageID("exitImageID", "Output Exit Points Image", "eep.exit", DataNameProperty::WRITE)
50
        , _camera("camera", "Camera")
51
        , _enableMirror("enableMirror", "Enable Virtual Mirror Feature", false)
52
53
54
        , _shader(0)
    {
        addProperty(&_sourceImageID);
schultezub's avatar
schultezub committed
55
        addProperty(&_geometryID);
56
        addProperty(&_mirrorID);
57
58
59
        addProperty(&_entryImageID);
        addProperty(&_exitImageID);
        addProperty(&_camera);
60
        addProperty(&_enableMirror);
61
62
63
64
65
66
67
68
    }

    EEPGenerator::~EEPGenerator() {

    }

    void EEPGenerator::init() {
        VisualizationProcessor::init();
schultezub's avatar
schultezub committed
69
        _shader = ShdrMgr.loadSeparate("core/glsl/passthrough.vert", "modules/vis/eepgenerator.frag", "", false);
70
71
72
73
        if (_shader != 0) {
            _shader->setAttributeLocation(0, "in_Position");
            _shader->setAttributeLocation(1, "in_TexCoord");
        }
74
75
76
77
    }

    void EEPGenerator::deinit() {
        ShdrMgr.dispose(_shader);
schultezub's avatar
schultezub committed
78
        _shader = 0;
79
80
81
82
83
        VisualizationProcessor::deinit();
    }

    void EEPGenerator::process(DataContainer& data) {
        DataContainer::ScopedTypedData<ImageDataGL> img(data, _sourceImageID.getValue());
84
        DataContainer::ScopedTypedData<MeshGeometry> proxyGeometry(data, _geometryID.getValue());
85

86
        if (img != 0 && proxyGeometry != 0 && _shader != 0) {
87
88
89
            if (img->getDimensionality() == 3) {
                tgt::Bounds volumeExtent = img->getWorldBounds();
                tgt::Bounds textureBounds(tgt::vec3(0.f), tgt::vec3(1.f));
90

91
92
93
94
                // clip proxy geometry against near-plane to support camera in volume
                // FIXME:   In some cases, the near plane is not rendered correctly...
                float nearPlaneDistToOrigin = tgt::dot(_camera.getValue().getPosition(), -_camera.getValue().getLook()) - _camera.getValue().getNearDist() - .001f;
                MeshGeometry clipped = proxyGeometry->clipAgainstPlane(nearPlaneDistToOrigin, -_camera.getValue().getLook(), true, 0.02f);
95

96
97
98
99
                // set modelview and projection matrices
                glPushAttrib(GL_ALL_ATTRIB_BITS);

                _shader->activate();
100

schultezub's avatar
schultezub committed
101
                tgt::mat4 mirrorMatrix = tgt::mat4::identity;
102
                if (_enableMirror.getValue()) {
schultezub's avatar
schultezub committed
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
                    DataContainer::ScopedTypedData<FaceGeometry> mirrorGeometry(data, _mirrorID.getValue());
                    if (mirrorGeometry && mirrorGeometry->size() > 0) {
                        const tgt::vec3& p = mirrorGeometry->getVertices()[0];
                        tgt::vec3 n = tgt::normalize(tgt::cross(mirrorGeometry->getVertices()[1] - mirrorGeometry->getVertices()[0], mirrorGeometry->getVertices()[2] - mirrorGeometry->getVertices()[0]));
                        float k = tgt::dot(p, n);

                        // mirror matrix sponsored by:
                        // Jiang 
                        mirrorMatrix = tgt::mat4(
                            1 - 2*n.x*n.x, -2*n.y*n.x   , -2*n.z*n.x   , 0, 
                            -2*n.x*n.y   , 1 - 2*n.y*n.y, -2*n.z*n.y   , 0, 
                            -2*n.x*n.z   , -2*n.y*n.z   , 1 - 2*n.z*n.z, 0, 
                            2*n.x*k      , 2*n.y*k      , 2*n.z*k      , 1);

                        // TODO: double check, whether matrix transpose is necessary
                    }
                    else {
                        LERROR("No suitable virtual mirror geometry found.");
                    }
122
                }
schultezub's avatar
schultezub committed
123
124
                
                _shader->setUniform("_modelMatrix", mirrorMatrix);
schultezub's avatar
schultezub committed
125
126
                _shader->setUniform("_projectionMatrix", _camera.getValue().getProjectionMatrix());
                _shader->setUniform("_viewMatrix", _camera.getValue().getViewMatrix());
127
                glEnable(GL_CULL_FACE);
128
129

                // create entry points texture
130
                ImageDataRenderTarget* entrypoints = new ImageDataRenderTarget(tgt::svec3(_renderTargetSize.getValue(), 1), GL_RGBA16);
131
132
                entrypoints->activate();

133
134
                glDepthFunc(GL_LESS);
                glClearDepth(1.0f);
135
                glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
136
                glCullFace(_enableMirror.getValue() ? GL_FRONT : GL_BACK);
137
                clipped.render();
138
139
140
141

                entrypoints->deactivate();

                // create exit points texture
142
                ImageDataRenderTarget* exitpoints = new ImageDataRenderTarget(tgt::svec3(_renderTargetSize.getValue(), 1), GL_RGBA16);
143
144
                exitpoints->activate();

145
146
                glDepthFunc(GL_GREATER);
                glClearDepth(0.0f);
147
                glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
148
                glCullFace(_enableMirror.getValue() ? GL_BACK : GL_FRONT);
149
                clipped.render();
150
151
152
153
154

                exitpoints->deactivate();

                _shader->deactivate();
                glPopAttrib();
schultezub's avatar
schultezub committed
155
                LGL_ERROR;
156
157
158
159
160
161
162
163
164

                data.addData(_entryImageID.getValue(), entrypoints);
                data.addData(_exitImageID.getValue(), exitpoints);
            }
            else {
                LERROR("Input image must have dimensionality of 3.");
            }
        }
        else {
165
            LERROR("No suitable input image or proxy geometry found.");
166
167
168
169
170
171
        }

        _invalidationLevel.setValid();
    }

}