eepgenerator.cpp 5.68 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/geometrydata.h"
39
40
41
42
43
44
45

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

    EEPGenerator::EEPGenerator(GenericProperty<tgt::ivec2>& canvasSize)
        : VisualizationProcessor(canvasSize)
        , _sourceImageID("sourceImageID", "Input Image", "")
46
        , _geometryID("geometryID", "Input Geometry ID", "pg.input")
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
        , _entryImageID("entryImageID", "Output Entry Points Image", "")
        , _exitImageID("exitImageID", "Output Exit Points Image", "")
        , _camera("camera", "Camera")
        , _shader(0)
    {
        addProperty(&_sourceImageID);
        addProperty(&_entryImageID);
        addProperty(&_exitImageID);
        addProperty(&_camera);
    }

    EEPGenerator::~EEPGenerator() {

    }

    void EEPGenerator::init() {
        VisualizationProcessor::init();
schultezub's avatar
schultezub committed
64
        _shader = ShdrMgr.loadSeparate("core/glsl/passthrough.vert", "modules/vis/eepgenerator.frag", "", false);
65
66
67
68
    }

    void EEPGenerator::deinit() {
        ShdrMgr.dispose(_shader);
schultezub's avatar
schultezub committed
69
        _shader = 0;
70
71
72
73
74
        VisualizationProcessor::deinit();
    }

    void EEPGenerator::process(DataContainer& data) {
        DataContainer::ScopedTypedData<ImageDataGL> img(data, _sourceImageID.getValue());
75
        DataContainer::ScopedTypedData<GeometryData> proxyGeometry(data, _geometryID.getValue());
76

77
        if (img != 0 && proxyGeometry != 0) {
78
            if (img->getDimensionality() == 3) {
79
                // TODO: implement some kind of cool proxy geometry supporting clipping (camera in volume)...
80
81
82
83
84
85
86
                tgt::Bounds volumeExtent = img->getWorldBounds();
                tgt::Bounds textureBounds(tgt::vec3(0.f), tgt::vec3(1.f));

                // set modelview and projection matrices
                glPushAttrib(GL_ALL_ATTRIB_BITS);

                _shader->activate();
schultezub's avatar
schultezub committed
87
88
89
90
                _shader->setUniform("_projectionMatrix", _camera.getValue().getProjectionMatrix());
                _shader->setUniform("_viewMatrix", _camera.getValue().getViewMatrix());
                _shader->setAttributeLocation(0, "in_Position");
                _shader->setAttributeLocation(1, "in_TexCoord");
91
                glEnable(GL_CULL_FACE);
92
93

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

97
98
                glDepthFunc(GL_LESS);
                glClearDepth(1.0f);
99
                glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
100
                glCullFace(GL_BACK);
101
                proxyGeometry->render();
102
103
104
105

                entrypoints->deactivate();

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

109
110
                glDepthFunc(GL_GREATER);
                glClearDepth(0.0f);
111
                glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
112
                glCullFace(GL_FRONT);
113
                proxyGeometry->render();
114
115
116
117
118

                exitpoints->deactivate();

                _shader->deactivate();
                glPopAttrib();
schultezub's avatar
schultezub committed
119
                LGL_ERROR;
120
121
122
123
124
125
126
127
128

                data.addData(_entryImageID.getValue(), entrypoints);
                data.addData(_exitImageID.getValue(), exitpoints);
            }
            else {
                LERROR("Input image must have dimensionality of 3.");
            }
        }
        else {
129
            LERROR("No suitable input image or proxy geometry found.");
130
131
132
133
134
135
        }

        _invalidationLevel.setValid();
    }

}