diff options
Diffstat (limited to 'fieldicle.py')
| -rw-r--r-- | fieldicle.py | 222 |
1 files changed, 0 insertions, 222 deletions
diff --git a/fieldicle.py b/fieldicle.py deleted file mode 100644 index 42af810..0000000 --- a/fieldicle.py +++ /dev/null @@ -1,222 +0,0 @@ -import pyopencl as cl -mf = cl.mem_flags -from pyopencl.tools import get_gl_sharing_context_properties - -from string import Template - -from OpenGL.GL import * # OpenGL - GPU rendering interface -from OpenGL.GLU import * # OpenGL tools (mipmaps, NURBS, perspective projection, shapes) -from OpenGL.GLUT import * # OpenGL tool to make a visualization window -from OpenGL.arrays import vbo - -import numpy -import threading - -import gi -gi.require_version('Gtk', '3.0') -from gi.repository import Gtk - -class ParticleWindow: - window_width = 500 - window_height = 500 - - world_width = 20. - world_height = 20. - - num_particles = 100000 - time_step = .005 - - gtk_active = False - - def glut_window(self): - glutInit(sys.argv) - glutInitDisplayMode(GLUT_RGBA | GLUT_DOUBLE | GLUT_DEPTH) - glutInitWindowSize(self.window_width, self.window_height) - glutInitWindowPosition(0, 0) - window = glutCreateWindow("fieldicle") - - glutDisplayFunc(self.on_display) - glutSpecialFunc(self.on_keyboard) - glutTimerFunc(5, self.on_timer, 5) - glutReshapeFunc(self.on_window_resize) - - glViewport(0, 0, self.window_width, self.window_height) - glMatrixMode(GL_PROJECTION) - glLoadIdentity() - - glOrtho( - -(self.world_width/2), self.world_width/2, - -(self.world_height/2), self.world_height/2, - 0.1, 100.0 - ) - - return(window) - - def on_keyboard(self, key, x, y): - if key == GLUT_KEY_F1: - self.gtk_active = True - ParamWindow(self).show_all() - - def initial_buffers(self, num_particles): - self.np_position = numpy.ndarray((self.num_particles, 4), dtype=numpy.float32) - self.np_color = numpy.ndarray((num_particles, 4), dtype=numpy.float32) - - self.set_particle_start_positions() - - self.np_color[:,:] = [1.,1.,1.,1.] - self.np_color[:,3] = numpy.random.random_sample((self.num_particles,)) - - self.gl_position = vbo.VBO(data=self.np_position, usage=GL_DYNAMIC_DRAW, target=GL_ARRAY_BUFFER) - self.gl_position.bind() - self.gl_color = vbo.VBO(data=self.np_color, usage=GL_DYNAMIC_DRAW, target=GL_ARRAY_BUFFER) - self.gl_color.bind() - - return (self.np_position, self.gl_position, self.gl_color) - - def on_timer(self, t): - glutTimerFunc(t, self.on_timer, t) - glutPostRedisplay() - if self.gtk_active: - Gtk.main_iteration_do(False) - - def set_particle_start_positions(self): - self.np_position[:,0] = self.world_width * numpy.random.random_sample((self.num_particles,)) - (self.world_width/2) - self.np_position[:,1] = self.world_height * numpy.random.random_sample((self.num_particles,)) - (self.world_height/2) - self.np_position[:,2] = 0. - self.np_position[:,3] = 1. - self.cl_start_position = cl.Buffer(self.context, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=self.np_position) - - def on_window_resize(self, width, height): - self.window_width = width - self.window_height = height - self.world_height = self.world_width / self.window_width * self.window_height; - - glViewport(0, 0, self.window_width, self.window_height) - glLoadIdentity() - glOrtho( - -(self.world_width/2), self.world_width/2, - -(self.world_height/2), self.world_height/2, - 0.1, 100.0 - ) - - self.set_particle_start_positions() - - def update_field(self, fx, fy): - self.program = cl.Program(self.context, Template(self.kernel).substitute({ - 'fx': fx, - 'fy': fy, - 'time_step': self.time_step - })).build() - - def on_display(self): - # Update or particle positions by calling the OpenCL kernel - cl.enqueue_acquire_gl_objects(self.queue, [self.cl_gl_position, self.cl_gl_color]) - kernelargs = (self.cl_gl_position, self.cl_gl_color, self.cl_start_position) - self.program.update_particles(self.queue, (self.num_particles,), None, *(kernelargs)) - cl.enqueue_release_gl_objects(self.queue, [self.cl_gl_position, self.cl_gl_color]) - self.queue.finish() - glFlush() - - glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT) - glMatrixMode(GL_MODELVIEW) - glLoadIdentity() - - glTranslatef(0., 0., -1.) - - # Render the particles - glEnable(GL_POINT_SMOOTH) - glPointSize(1) - glEnable(GL_BLEND) - glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA) - - # Set up the VBOs - self.gl_color.bind() - glColorPointer(4, GL_FLOAT, 0, self.gl_color) - self.gl_position.bind() - glVertexPointer(4, GL_FLOAT, 0, self.gl_position) - glEnableClientState(GL_VERTEX_ARRAY) - glEnableClientState(GL_COLOR_ARRAY) - - # Draw the VBOs - glDrawArrays(GL_POINTS, 0, self.num_particles) - - glDisableClientState(GL_COLOR_ARRAY) - glDisableClientState(GL_VERTEX_ARRAY) - - glDisable(GL_BLEND) - - glutSwapBuffers() - - def run(self): - self.window = self.glut_window() - - self.platform = cl.get_platforms()[0] - self.context = cl.Context(properties=[(cl.context_properties.PLATFORM, self.platform)] + get_gl_sharing_context_properties()) - self.queue = cl.CommandQueue(self.context) - - (self.np_position, self.gl_position, self.gl_color) = self.initial_buffers(self.num_particles) - - self.cl_start_position = cl.Buffer(self.context, mf.READ_ONLY | mf.COPY_HOST_PTR, hostbuf=self.np_position) - - self.cl_gl_position = cl.GLBuffer(self.context, mf.READ_WRITE, int(self.gl_position)) - self.cl_gl_color = cl.GLBuffer(self.context, mf.READ_WRITE, int(self.gl_color)) - - self.kernel = """__kernel void update_particles(__global float4* position, - __global float4* color, - __global float4* start_position) - { - unsigned int i = get_global_id(0); - float4 p = position[i]; - - float life = color[i].w; - life -= $time_step; - - if (life <= 0.f) { - p = start_position[i]; - life = 1.0f; - } - - p.x += ($fx) * $time_step; - p.y += ($fy) * $time_step; - - position[i] = p; - color[i].w = life; - }""" - self.program = cl.Program(self.context, Template(self.kernel).substitute({ - 'fx': 'cos(p.x)', - 'fy': 'sin(p.y*p.x)', - 'time_step': self.time_step - })).build() - - glutMainLoop() - - -particleWindow = ParticleWindow() - -glfwThread = threading.Thread(target=particleWindow.run) -glfwThread.start() - -class ParamWindow(Gtk.Dialog): - def __init__(self, particleWin): - Gtk.Dialog.__init__(self, title="Field Parameters") - self.particleWin = particleWin - - self.updateBtn = Gtk.Button(label="Update field") - self.updateBtn.connect("clicked", self.on_update_clicked) - - self.entryFx = Gtk.Entry() - self.entryFx.set_text("cos(p.x)") - self.entryFy = Gtk.Entry() - self.entryFy.set_text("sin(p.y*p.x)") - - layout = self.get_content_area() - - layout.add(self.entryFx) - layout.add(self.entryFy) - layout.add(self.updateBtn) - - def on_update_clicked(self, widget): - self.particleWin.update_field( - self.entryFx.get_text(), - self.entryFy.get_text() - ) |