Initial public commit of this basic MD code

Simulation of interacting particles

1 files changed, 28 insertions, 0 deletions

diff --git a/interacticle/visual/shader.py b/interacticle/visual/shader.py
new file mode 100644
index 0000000..c6c035b
--- /dev/null
+++ b/interacticle/visual/shader.py
@@ -0,0 +1,28 @@
+from OpenGL.GL import GL_VERTEX_SHADER, GL_FRAGMENT_SHADER, GL_GEOMETRY_SHADER
+from OpenGL.GL import shaders
+
+class Shader:
+    def __init__(self, fragment_src, vertex_src, uniform):
+        self.program = shaders.compileProgram(
+            shaders.compileShader(vertex_src, GL_VERTEX_SHADER),
+            shaders.compileShader(fragment_src, GL_FRAGMENT_SHADER))
+        self.uniform = { }
+        for name in uniform:
+            self.uniform[name] = shaders.glGetUniformLocation(self.program, name)
+
+    def use(self):
+        shaders.glUseProgram(self.program)
+
+class GeometryShader:
+    def __init__(self, fragment_src, geometry_src, vertex_src, uniform):
+        self.program = shaders.compileProgram(
+            shaders.compileShader(vertex_src, GL_VERTEX_SHADER),
+            shaders.compileShader(geometry_src, GL_GEOMETRY_SHADER),
+            shaders.compileShader(fragment_src, GL_FRAGMENT_SHADER))
+        self.uniform = { }
+        for name in uniform:
+            self.uniform[name] = shaders.glGetUniformLocation(self.program, name)
+
+    def use(self):
+        shaders.glUseProgram(self.program)
+