From e9606282237147a4626e3878c21709c47ea1209c Mon Sep 17 00:00:00 2001
From: Sam Hocevar <sam@hocevar.net>
Date: Mon, 28 Jan 2013 13:23:54 +0000
Subject: [PATCH] gpu: allow to set array uniform values.

---
 src/gpu/shader.cpp | 98 ++++++++++++++++++++++++++++++++++++++++++++++
 src/gpu/shader.h   |  5 +++
 2 files changed, 103 insertions(+)

diff --git a/src/gpu/shader.cpp b/src/gpu/shader.cpp
index 879434dd..ca25c434 100644
--- a/src/gpu/shader.cpp
+++ b/src/gpu/shader.cpp
@@ -379,6 +379,10 @@ ShaderUniform Shader::GetUniformLocation(char const *uni) const
     return ret;
 }
 
+/*
+ * Uniform setters for scalars
+ */
+
 void Shader::SetUniform(ShaderUniform const &uni, int i)
 {
 #if defined USE_D3D9 || defined _XBOX
@@ -557,6 +561,100 @@ void Shader::SetUniform(ShaderUniform const &uni, ShaderTexture tex, int index)
 #endif
 }
 
+/*
+ * Uniform setters for arrays
+ */
+
+void Shader::SetUniform(ShaderUniform const &uni, Array<float> const &v)
+{
+#if defined USE_D3D9 || defined _XBOX
+    /* FIXME: this will not work properly because we don't know how tell DX9
+     * it's a bunch of floats instead of vec4. */
+    if (uni.flags & 1)
+        g_d3ddevice->SetPixelShaderConstantF((UINT)uni.frag,
+                                             &v[0], v.Count() / 4);
+    if (uni.flags & 2)
+        g_d3ddevice->SetVertexShaderConstantF((UINT)uni.vert,
+                                              &v[0], v.Count() / 4);
+#elif !defined __CELLOS_LV2__
+    glUniform1fv(uni.frag, v.Count(), &v[0]);
+#else
+    if (uni.frag)
+        cgGLSetParameterArray1fv((CGparameter)uni.frag,
+                                 0, v.Count(), &v[0]);
+    if (uni.vert)
+        cgGLSetParameterArray1fv((CGparameter)uni.vert,
+                                 0, v.Count(), &v[0]);
+#endif
+}
+
+void Shader::SetUniform(ShaderUniform const &uni, Array<vec2> const &v)
+{
+#if defined USE_D3D9 || defined _XBOX
+    /* FIXME: this will not work properly because we don't know how tell DX9
+     * it's a bunch of vec2 instead of vec4. */
+    if (uni.flags & 1)
+        g_d3ddevice->SetPixelShaderConstantF((UINT)uni.frag,
+                                             &v[0][0], v.Count() / 2);
+    if (uni.flags & 2)
+        g_d3ddevice->SetVertexShaderConstantF((UINT)uni.vert,
+                                              &v[0][0], v.Count() / 2);
+#elif !defined __CELLOS_LV2__
+    glUniform2fv(uni.frag, v.Count(), &v[0][0]);
+#else
+    if (uni.frag)
+        cgGLSetParameterArray2fv((CGparameter)uni.frag,
+                                 0, v.Count(), &v[0][0]);
+    if (uni.vert)
+        cgGLSetParameterArray2fv((CGparameter)uni.vert,
+                                 0, v.Count(), &v[0][0]);
+#endif
+}
+
+void Shader::SetUniform(ShaderUniform const &uni, Array<vec3> const &v)
+{
+#if defined USE_D3D9 || defined _XBOX
+    /* FIXME: this will not work properly because we don't know how tell DX9
+     * it's a bunch of vec3 instead of vec4. */
+    if (uni.flags & 1)
+        g_d3ddevice->SetPixelShaderConstantF((UINT)uni.frag,
+                                             &v[0][0], v.Count());
+    if (uni.flags & 2)
+        g_d3ddevice->SetVertexShaderConstantF((UINT)uni.vert,
+                                              &v[0][0], v.Count());
+#elif !defined __CELLOS_LV2__
+    glUniform3fv(uni.frag, v.Count(), &v[0][0]);
+#else
+    if (uni.frag)
+        cgGLSetParameterArray3fv((CGparameter)uni.frag,
+                                 0, v.Count(), &v[0][0]);
+    if (uni.vert)
+        cgGLSetParameterArray3fv((CGparameter)uni.vert,
+                                 0, v.Count(), &v[0][0]);
+#endif
+}
+
+void Shader::SetUniform(ShaderUniform const &uni, Array<vec4> const &v)
+{
+#if defined USE_D3D9 || defined _XBOX
+    if (uni.flags & 1)
+        g_d3ddevice->SetPixelShaderConstantF((UINT)uni.frag,
+                                             &v[0][0], v.Count());
+    if (uni.flags & 2)
+        g_d3ddevice->SetVertexShaderConstantF((UINT)uni.vert,
+                                              &v[0][0], v.Count());
+#elif !defined __CELLOS_LV2__
+    glUniform4fv(uni.frag, v.Count(), &v[0][0]);
+#else
+    if (uni.frag)
+        cgGLSetParameterArray4fv((CGparameter)uni.frag,
+                                 0, v.Count(), &v[0][0]);
+    if (uni.vert)
+        cgGLSetParameterArray4fv((CGparameter)uni.vert,
+                                 0, v.Count(), &v[0][0]);
+#endif
+}
+
 void Shader::Bind() const
 {
 #if defined USE_D3D9 || defined _XBOX
diff --git a/src/gpu/shader.h b/src/gpu/shader.h
index 904c6084..cfc42706 100644
--- a/src/gpu/shader.h
+++ b/src/gpu/shader.h
@@ -84,6 +84,11 @@ public:
     void SetUniform(ShaderUniform const &uni, mat4 const &m);
     void SetUniform(ShaderUniform const &uni, ShaderTexture tex, int index);
 
+    void SetUniform(ShaderUniform const &uni, Array<float> const &v);
+    void SetUniform(ShaderUniform const &uni, Array<vec2> const &v);
+    void SetUniform(ShaderUniform const &uni, Array<vec3> const &v);
+    void SetUniform(ShaderUniform const &uni, Array<vec4> const &v);
+
     void Bind() const;
     void Unbind() const;