Shader: Support setting float arrays

Author: einarf

demosys/opengl/shader.py

@@ -206,7 +206,7 @@ def uniform_check(self, name, expected_type):
             raise ShaderError("Incorrect data type: Uniform '{}' is of type {}".format(name, uniform.type.name))
         return uniform
 
-    # --- Setting uniforms ---
+    # --- Float uniforms ---
 
     def uniform_1f(self, name, value):
         """
@@ -254,6 +254,52 @@ def uniform_4f(self, name, x, y, z, w):
         uniform = self.uniform_check(name, GL.GL_FLOAT_VEC4)
         GL.glUniform4f(uniform.location, x, y, z, w)
 
+    # --- Float uniform arrays ---
+
+    def uniform_1fv(self, name, value, count=1):
+        """
+        Set a float uniform
+
+        :param name: Name of the uniform
+        :param count: Length of the uniform array (default 1)
+        :param value: float array
+        """
+        uniform = self.uniform_check(name, GL.GL_FLOAT)
+        GL.glUniform1fv(uniform.location, count, value)
+
+    def uniform_2fv(self, name, value, count=1):
+        """
+        Set a vec2 uniform
+
+        :param name: name of the uniform
+        :param count: Length of the uniform array (default 1)
+        :param value: float array
+        """
+        uniform = self.uniform_check(name, GL.GL_FLOAT_VEC2)
+        GL.glUniform2fv(uniform.location, count, value)
+
+    def uniform_3fv(self, name, value, count=1):
+        """
+        Set a vec3 uniform
+
+        :param name: Name of the uniform
+        :param count: Length of the uniform array (default 1)
+        :param value: float array
+        """
+        uniform = self.uniform_check(name, GL.GL_FLOAT_VEC3)
+        GL.glUniform3fv(uniform.location, count, value)
+
+    def uniform_4fv(self, name, value, count=1):
+        """
+        Set a vec4 uniform
+
+        :param name: Name of the uniform
+        :param count: Length of the uniform array (default 1)
+        :param value: float array
+        """
+        uniform = self.uniform_check(name, GL.GL_FLOAT_VEC4)
+        GL.glUniform4fv(uniform.location, count, value)
+
     # --- Signed Integers ---
 
     def uniform_1i(self, name, value):