1 files changed, 16 insertions, 1 deletions
diff --git a/src/datavisualization/utils/shaderhelper.cpp b/src/datavisualization/utils/shaderhelper.cpp
index e27103af..7eb48945 100644
--- a/src/datavisualization/utils/shaderhelper.cpp
+++ b/src/datavisualization/utils/shaderhelper.cpp
@@ -90,7 +90,8 @@ void ShaderHelper::initialize()
     m_colorUniform = m_program->uniformLocation("color_mdl");
     m_textureUniform = m_program->uniformLocation("textureSampler");
     m_shadowUniform = m_program->uniformLocation("shadowMap");
-
+    m_gradientMinUniform = m_program->uniformLocation("gradMin");
+    m_gradientHeightUniform = m_program->uniformLocation("gradHeight");
     m_initialized = true;
 }
 
@@ -230,6 +231,20 @@ GLuint ShaderHelper::shadow()
     return m_shadowUniform;
 }
 
+GLuint ShaderHelper::gradientMin()
+{
+    if (!m_initialized)
+        qFatal("Shader not initialized");
+    return m_gradientMinUniform;
+}
+
+GLuint ShaderHelper::gradientHeight()
+{
+    if (!m_initialized)
+        qFatal("Shader not initialized");
+    return m_gradientHeightUniform;
+}
+
 GLuint ShaderHelper::posAtt()
 {
     if (!m_initialized)