aboutsummaryrefslogtreecommitdiff
path: root/sunrise.py
diff options
context:
space:
mode:
Diffstat (limited to 'sunrise.py')
-rw-r--r--sunrise.py34
1 files changed, 5 insertions, 29 deletions
diff --git a/sunrise.py b/sunrise.py
index 47457f6..7cde57e 100644
--- a/sunrise.py
+++ b/sunrise.py
@@ -1,13 +1,8 @@
import numpy
import matplotlib.pyplot as plt
-from string import Template
-import pyopencl as cl
-from pyopencl.cltypes import make_double3
-
-mf = cl.mem_flags
-
-from planets import earth
+from firmament import Renderer
+from firmament.planets import earth
config = {
'size_x': 1920//4,
@@ -25,31 +20,12 @@ config = {
sun_range = (-10, 90, 10)
-cl_platform = cl.get_platforms()[0]
-cl_context = cl.Context(properties=[(cl.context_properties.PLATFORM, cl_platform)])
-cl_queue = cl.CommandQueue(cl_context)
-
-cl_picture = cl.Buffer(cl_context, mf.WRITE_ONLY, size=config['size_x']*config['size_y']*3*numpy.float64(0).nbytes)
-program = None
-
-print('height: %d' % (earth['earth_radius']*config['eye_pos'][2] - earth['earth_radius']))
-
-with open('raymarch.cl') as f:
- program = cl.Program(cl_context, Template(f.read()).substitute(
- {**config, **earth}
- )).build()
+renderer = Renderer(config, earth)
for i in numpy.arange(*sun_range):
- sun = make_double3(0.0,numpy.cos(i*2*numpy.pi/360),numpy.sin(i*2*numpy.pi/360))
+ sun = (0.0, numpy.cos(i*2*numpy.pi/360), numpy.sin(i*2*numpy.pi/360))
print(sun)
- program.render_pinhole(
- cl_queue, (config['size_x'], config['size_y']), None, cl_picture,
- make_double3(*(config['eye_pos'] * earth['earth_radius'])),
- make_double3(*(config['eye_dir'] * earth['earth_radius'])),
- sun)
-
- np_picture = numpy.ndarray(shape=(config['size_y'], config['size_x'], 3), dtype=numpy.float64)
- cl.enqueue_copy(cl_queue, np_picture, cl_picture).wait();
+ np_picture = renderer.render_pinhole(sun)
plt.imsave("sky_%05.1f.png" % (i-sun_range[0]), np_picture, origin='lower')