depthdarkening.cpp 7.5 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
8
9
//      Chair for Computer Aided Medical Procedures
//      Technische Universitt Mnchen
//      Boltzmannstr. 3, 85748 Garching b. Mnchen, 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
34
35
// 
// 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 "depthdarkening.h"
#include "tgt/logmanager.h"
#include "tgt/shadermanager.h"
#include "tgt/textureunit.h"

#include "core/datastructures/imagedata.h"
36
37
#include "core/datastructures/imagerepresentationgl.h"
#include "core/datastructures/imagerepresentationrendertarget.h"
38

39
40
41
42
43

#include "core/classification/simpletransferfunction.h"

#include "core/tools/quadrenderer.h"

schultezub's avatar
schultezub committed
44
45
namespace campvis {
    const std::string DepthDarkening::loggerCat_ = "CAMPVis.modules.vis.DepthDarkening";
46
47
48

    DepthDarkening::DepthDarkening(GenericProperty<tgt::ivec2>& canvasSize)
        : VisualizationProcessor(canvasSize)
49
50
51
52
53
54
55
        , p_inputImage("InputImage", "Input Image", "", DataNameProperty::READ)
        , p_outputImage("OutputImage", "Output Image", "dd.output", DataNameProperty::WRITE)
        , p_sigma("Sigma", "Sigma of Gaussian Filter", 2.f, 0.f, 10.f)
        , p_lambda("Lambda", "Strength of Depth Darkening Effect", 10.f, 0.f, 50.f)
        , p_useColorCoding("UseColorCoding", "Cold/Warm Color Coding", false, InvalidationLevel::INVALID_SHADER)
        , p_coldColor("ColdColor", "Cold Color (Far Objects)", tgt::vec3(0.f, 0.f, 1.f), tgt::vec3(0.f), tgt::vec3(1.f))
        , p_warmColor("WarmColor", "Warm Color (Near Objects)", tgt::vec3(1.f, 0.f, 0.f), tgt::vec3(0.f), tgt::vec3(1.f))
56
57
        , _shader(0)
    {
58
59
60
61
62
63
64
        addProperty(&p_inputImage);
        addProperty(&p_outputImage);
        addProperty(&p_sigma);
        addProperty(&p_lambda);
        addProperty(&p_useColorCoding);
        addProperty(&p_coldColor);
        addProperty(&p_warmColor);
65
66
67
68
69
70
71
72
    }

    DepthDarkening::~DepthDarkening() {

    }

    void DepthDarkening::init() {
        VisualizationProcessor::init();
73
        _shader = ShdrMgr.loadSeparate("core/glsl/passthrough.vert", "modules/vis/glsl/depthdarkening.frag", generateHeader(), false);
74
75
76
77
78
79
80
81
82
83
        _shader->setAttributeLocation(0, "in_Position");
        _shader->setAttributeLocation(1, "in_TexCoord");
    }

    void DepthDarkening::deinit() {
        VisualizationProcessor::deinit();
        ShdrMgr.dispose(_shader);
    }

    void DepthDarkening::process(DataContainer& data) {
84
        ImageRepresentationRenderTarget::ScopedRepresentation inputImage(data, p_inputImage.getValue());
85
86

        if (inputImage != 0) {
87
            if (getInvalidationLevel().isInvalidShader()) {
88
89
90
91
                _shader->setHeaders(generateHeader());
                _shader->rebuild();
            }

92
93
94
95
96
97
98
99
100
101
102
103
104
            // TODO: const cast is ugly...
            const_cast<tgt::Texture*>(inputImage->getDepthTexture())->downloadTexture();
            float* pixels = (float*)inputImage->getDepthTexture()->getPixelData();
            float curDepth = *(pixels);
            float minDepth = curDepth;
            float maxDepth = curDepth;
            size_t numPixels = inputImage->getNumElements();
            for (size_t i = 1; i < numPixels; ++i) {
                curDepth = pixels[i];
                minDepth = std::min(minDepth, curDepth);
                maxDepth = std::max(maxDepth, curDepth);
            }

105
106
            std::pair<ImageData*, ImageRepresentationRenderTarget*> tempTarget = ImageRepresentationRenderTarget::createWithImageData(_renderTargetSize.getValue());
            std::pair<ImageData*, ImageRepresentationRenderTarget*> outputTarget = ImageRepresentationRenderTarget::createWithImageData(_renderTargetSize.getValue());
107
108
109
110
111
112
113
114
115
116
117
            glPushAttrib(GL_ALL_ATTRIB_BITS);
            glEnable(GL_DEPTH_TEST);
            glDepthFunc(GL_ALWAYS);

            _shader->activate();
            tgt::TextureUnit colorUnit, depthUnit, pass2DepthUnit;
            inputImage->bind(_shader, &colorUnit, &depthUnit);
            inputImage->bind(_shader, 0, &pass2DepthUnit, "", "_depthPass2Texture");
            
            _shader->setUniform("_viewportSizeRCP", 1.f / tgt::vec2(_renderTargetSize.getValue()));
            _shader->setUniform("_direction", tgt::vec2(1.f, 0.f));
118
119
            _shader->setUniform("_sigma", p_sigma.getValue());
            _shader->setUniform("_lambda", p_lambda.getValue());
120
121
            _shader->setUniform("_minDepth", minDepth);
            _shader->setUniform("_maxDepth", maxDepth);
122
123
124
            if (p_useColorCoding.getValue()) {
                _shader->setUniform("_coldColor", p_coldColor.getValue());
                _shader->setUniform("_warmColor", p_warmColor.getValue());
125
            }
126

127
            tempTarget.second->activate();
128
129
130
            LGL_ERROR;
            glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
            QuadRdr.renderQuad();
131
            tempTarget.second->deactivate();
132
133

            inputImage->bind(_shader, &colorUnit, &depthUnit);
134
            tempTarget.second->bind(_shader, 0, &pass2DepthUnit, "", "_depthPass2Texture");
135
136
            _shader->setUniform("_direction", tgt::vec2(0.f, 1.f));
            
137
            outputTarget.second->activate();
138
139
140
            LGL_ERROR;
            glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
            QuadRdr.renderQuad();
141
            outputTarget.second->deactivate();
142
143
144
145
146
147

            _shader->deactivate();
            tgt::TextureUnit::setZeroUnit();
            glPopAttrib();
            LGL_ERROR;

148
149
            data.addData(p_outputImage.getValue() + "temp", tempTarget.first);
            data.addData(p_outputImage.getValue(), outputTarget.first);
150
            p_outputImage.issueWrite();
151
152
153
154
155
        }
        else {
            LERROR("No suitable input image found.");
        }

156
        applyInvalidationLevel(InvalidationLevel::VALID);
157
158
    }

159
    std::string DepthDarkening::generateHeader() const {
160
        if (p_useColorCoding.getValue())
161
162
163
164
165
            return "#define USE_COLORCODING\n";
        else
            return "";
    }

166
167
}