aboutsummaryrefslogtreecommitdiff
path: root/utility/streamline.py
diff options
context:
space:
mode:
Diffstat (limited to 'utility/streamline.py')
-rw-r--r--utility/streamline.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/utility/streamline.py b/utility/streamline.py
index 5111fc2..9f0fb55 100644
--- a/utility/streamline.py
+++ b/utility/streamline.py
@@ -10,12 +10,11 @@ from OpenGL.GL import *
from OpenGL.arrays import vbo
class Streamlines:
- def __init__(self, lattice, moments, origins):
+ def __init__(self, lattice, origins):
self.lattice = lattice
self.context = self.lattice.context
self.queue = self.lattice.queue
self.float_type = self.lattice.memory.float_type
- self.moments = moments
self.count = len(origins)
self.np_origins = numpy.ndarray(shape=(self.count, 2), dtype=self.float_type)
@@ -66,7 +65,7 @@ class Streamlines:
self.program.draw_streamline(
self.queue, (self.count,1), None,
- self.moments.cl_gl_moments,
+ self.lattice.memory.cl_moments,
self.lattice.memory.cl_material,
self.cl_origins,
self.cl_gl_streamlines)