Separate classes into modules

1 files changed, 20 insertions, 0 deletions

diff --git a/boltzgas/visual/tracer.py b/boltzgas/visual/tracer.py
new file mode 100644
index 0000000..a5b2203
--- /dev/null
+++ b/boltzgas/visual/tracer.py
@@ -0,0 +1,20 @@
+from OpenGL.GL import *
+
+class Tracer:
+    def __init__(self, gas, iParticle):
+        self.gas = gas
+        self.iParticle = iParticle
+        self.trace = [ ]
+
+    def update(self):
+        position = self.gas.get_positions()[self.iParticle]
+        self.trace.append((position[0], position[1]))
+
+    def display(self, uniform):
+        glUniform3f(uniform['face_color'], 1., 0., 0.)
+        glLineWidth(2)
+        glBegin(GL_LINE_STRIP)
+        for v in self.trace:
+            glVertex(*v, 0.)
+        glEnd()
+