diff options
author | Adrian Kummerlaender | 2020-01-20 22:40:44 +0100 |
---|---|---|
committer | Adrian Kummerlaender | 2020-01-20 22:40:44 +0100 |
commit | abbdbe64d12af732983bc11aac772d3d32682bb5 (patch) | |
tree | fb89bd9818e33588cf295036c6feb681b92d0ff4 /firmament/renderer.py | |
parent | b0b52925fb96e417802f4b390fafc15c8536df2d (diff) | |
download | firmament-abbdbe64d12af732983bc11aac772d3d32682bb5.tar firmament-abbdbe64d12af732983bc11aac772d3d32682bb5.tar.gz firmament-abbdbe64d12af732983bc11aac772d3d32682bb5.tar.bz2 firmament-abbdbe64d12af732983bc11aac772d3d32682bb5.tar.lz firmament-abbdbe64d12af732983bc11aac772d3d32682bb5.tar.xz firmament-abbdbe64d12af732983bc11aac772d3d32682bb5.tar.zst firmament-abbdbe64d12af732983bc11aac772d3d32682bb5.zip |
Extract OpenCL setup into Renderer
Diffstat (limited to 'firmament/renderer.py')
-rw-r--r-- | firmament/renderer.py | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/firmament/renderer.py b/firmament/renderer.py new file mode 100644 index 0000000..d020cdb --- /dev/null +++ b/firmament/renderer.py @@ -0,0 +1,51 @@ +import numpy +from string import Template + +import pyopencl as cl +from pyopencl.cltypes import make_double3 + +mf = cl.mem_flags + +class Renderer: + def __init__(self, config, planet): + self.config = config + self.planet = planet + + self.cl_platform = cl.get_platforms()[0] + self.cl_context = cl.Context(properties=[(cl.context_properties.PLATFORM, self.cl_platform)]) + self.cl_queue = cl.CommandQueue(self.cl_context) + + self.cl_picture = cl.Buffer(self.cl_context, mf.WRITE_ONLY, size=self.config['size_x']*self.config['size_y']*3*numpy.float64(0).nbytes) + + with open('firmament/raymarch.cl') as f: + self.program = cl.Program(self.cl_context, Template(f.read()).substitute( + {**self.config, **self.planet} + )).build() + + def render_pinhole(self, sun): + self.program.render_pinhole( + self.cl_queue, + (self.config['size_x'], self.config['size_y']), + None, + self.cl_picture, + make_double3(*(self.config['eye_pos'] * self.planet['earth_radius'])), + make_double3(*(self.config['eye_dir'] * self.planet['earth_radius'])), + make_double3(*sun)) + + np_picture = numpy.ndarray(shape=(self.config['size_y'], self.config['size_x'], 3), dtype=numpy.float64) + cl.enqueue_copy(self.cl_queue, np_picture, self.cl_picture).wait(); + return np_picture + + def render_fisheye(self, sun): + self.program.render_fisheye( + self.cl_queue, + (self.config['size_x'], self.config['size_y']), + None, + self.cl_picture, + make_double3(*(self.config['eye_pos'] * self.planet['earth_radius'])), + make_double3(*(self.config['eye_dir'] * self.planet['earth_radius'])), + make_double3(*sun)) + + np_picture = numpy.ndarray(shape=(self.config['size_y'], self.config['size_x'], 3), dtype=numpy.float64) + cl.enqueue_copy(self.cl_queue, np_picture, self.cl_picture).wait(); + return np_picture |