blob: fce10d9b190351fd29db159dc4bc991bf6b5460f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
import pyopencl as cl
from common import MomentsTextureBase
from lattice.AB import Lattice
class MomentsTexture(MomentsTextureBase):
pass
def collect(self):
cl.enqueue_acquire_gl_objects(self.lattice.queue, [self.cl_gl_moments])
if self.lattice.tick:
self.lattice.program.collect_moments_to_texture(
self.lattice.queue,
self.lattice.geometry.size(),
self.lattice.layout,
self.lattice.memory.cl_pop_a,
self.cl_gl_moments)
else:
self.lattice.program.collect_moments_to_texture(
self.lattice.queue,
self.lattice.geometry.size(),
self.lattice.layout,
self.lattice.memory.cl_pop_b,
self.cl_gl_moments)
|
