summaryrefslogtreecommitdiff
path: root/render.py
diff options
context:
space:
mode:
Diffstat (limited to 'render.py')
-rwxr-xr-xrender.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/render.py b/render.py
index 1890471..17a7289 100755
--- a/render.py
+++ b/render.py
@@ -21,6 +21,8 @@ def render(viewable, size=(800,600), name='', bits=8, make_movie=False):
scale = np.linalg.norm(upper_bound-lower_bound)
+ from pycuda import autoinit
+
print 'device %s' % autoinit.device.name()
module = SourceModule(src.kernel, options=['-I' + src.dir], no_extern_c=True, cache_dir=False)
@@ -68,15 +70,15 @@ def render(viewable, size=(800,600), name='', bits=8, make_movie=False):
rgb_lookup2[rgb_lookup2 > 1.0] = 1.0
rgb_lookup2_gpu.set(rgb_lookup2.view(gpuarray.vec.float3))
- camera = Camera(size)
+ #camera = Camera(size)
point = np.array([0, diagonal*1.75, (lower_bound[2]+upper_bound[2])/2])
axis1 = np.array([1,0,0], dtype=np.double)
axis2 = np.array([0,0,1], dtype=np.double)
- camera.position(point)
+ #camera.position(point)
- origins, directions = camera.get_rays()
+ origins, directions = get_rays(point, size)
origins_gpu = gpuarray.to_gpu(origins.astype(np.float32).view(gpuarray.vec.float3))
directions_gpu = gpuarray.to_gpu(directions.astype(np.float32).view(gpuarray.vec.float3))