2 files changed, 8 insertions, 8 deletions
diff --git a/gfx/gl/shader.cpp b/gfx/gl/shader.cpp
index 979d57e..8c58c39 100644
--- a/gfx/gl/shader.cpp
+++ b/gfx/gl/shader.cpp
@@ -6,8 +6,7 @@
 #include <stdexcept>
 #include <string>
 
-Shader::ProgramHandle::ProgramHandle(std::initializer_list<GLuint> srcs) :
-	viewProjection_uniform {}, model_uniform {}, lightDir_uniform {}
+Shader::ProgramHandle::ProgramHandle(std::initializer_list<GLuint> srcs) : viewProjection_uniform {}, model_uniform {}
 {
 	for (const auto & srcId : srcs) {
 		glAttachShader(m_program, srcId);
@@ -25,7 +24,6 @@ Shader::ProgramHandle::ProgramHandle(std::initializer_list<GLuint> srcs) :
 
 	viewProjection_uniform = glGetUniformLocation(m_program, "viewProjection");
 	model_uniform = glGetUniformLocation(m_program, "model");
-	lightDir_uniform = glGetUniformLocation(m_program, "lightDirection");
 }
 
 Shader::Shader() :
@@ -46,11 +44,13 @@ Shader::setView(glm::mat4 proj) const
 }
 
 void
-Shader::setLight(glm::vec3 lightDir) const
+Shader::setUniform(const GLchar * uniform, glm::vec3 v) const
 {
 	for (const auto & prog : programs) {
-		glUseProgram(prog.m_program);
-		glUniform3fv(prog.lightDir_uniform, 1, &lightDir[0]);
+		if (auto loc = glGetUniformLocation(prog.m_program, uniform); loc >= 0) {
+			glUseProgram(prog.m_program);
+			glUniform3fv(loc, 1, &v[0]);
+		}
 	}
 }
 
diff --git a/gfx/gl/shader.h b/gfx/gl/shader.h
index 98e0348..0282f21 100644
--- a/gfx/gl/shader.h
+++ b/gfx/gl/shader.h
@@ -15,7 +15,7 @@ public:
 
 	void setView(glm::mat4 view) const;
 	void setModel(glm::mat4 model, Program = Program::Basic) const;
-	void setLight(glm::vec3 dir) const;
+	void setUniform(const GLchar *, glm::vec3 dir) const;
 
 private:
 	class Source {
@@ -36,7 +36,7 @@ private:
 		using ProgramRef = glRef<GLuint, __glewCreateProgram, __glewDeleteProgram>;
 
 		ProgramRef m_program;
-		GLint viewProjection_uniform, model_uniform, lightDir_uniform;
+		GLint viewProjection_uniform, model_uniform;
 	};
 
 	std::array<ProgramHandle, 1> programs;