diff options
Diffstat (limited to 'boltzgas/visual/tracer.py')
-rw-r--r-- | boltzgas/visual/tracer.py | 20 |
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() + |