From 8a384792b296d0c4c4a51d0a7679bc3c5f24c2c2 Mon Sep 17 00:00:00 2001 From: Anthony LaTorre Date: Mon, 16 May 2011 16:24:49 -0400 Subject: typo --- test.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index 9b0644d..236d1ec 100644 --- a/test.py +++ b/test.py @@ -12,6 +12,7 @@ from geometry import * from materials import * from camera import * from vector import * +import detectors import pygame @@ -40,15 +41,7 @@ origin, direction = make_vector(origin), make_vector(direction) origin_gpu = cuda.to_device(origin) direction_gpu = cuda.to_device(direction) -#solid = Solid(read_stl('models/lionsolid.stl'), vacuum, vacuum) -#geometry = Geometry() -#geometry.add_solid(solid) -#geometry.build(bits=3) - -import detectors - -geometry = detectors.build_lbne() -#geometry = detectors.load_lbne('./detectors/lbne_8bit.pkl') +geometry = detectors.build_lbne(bits=6) mesh = geometry.mesh mesh = mesh.reshape(mesh.shape[0]*3,3) @@ -90,11 +83,12 @@ states = np.empty(width*height, dtype=np.int32) states_gpu = cuda.to_device(states) block_size = 64 -for i in range(10): +for i in range(100000): rotate(np.int32(origin.size), origin_gpu, np.float32(np.pi/100), gpuarray.vec.make_float3(0,0,1), block=(block_size,1,1), grid=(width*height//block_size+1,1)) + rotate(np.int32(direction.size), direction_gpu, np.float32(np.pi/100), gpuarray.vec.make_float3(0,0,1), block=(block_size,1,1), grid=(width*height//block_size+1,1)) - translate(np.int32(origin.size), origin_gpu, gpuarray.vec.make_float3(-.5*np.sin(i*np.pi/100), -.5*np.cos(i*np.pi/100), 0), block=(block_size,1,1), grid=(width*height//block_size+1,1)) + #translate(np.int32(origin.size), origin_gpu, gpuarray.vec.make_float3(-np.sin(i*np.pi/100)/2, -np.cos(i*np.pi/100)/2, 0), block=(block_size,1,1), grid=(width*height//block_size+1,1)) t0 = time.time() intersect_mesh(np.int32(origin.size), origin_gpu, direction_gpu, pixels_gpu, first_leaf, states_gpu, block=(block_size,1,1), grid=(width*height//block_size+1,1), texrefs=[mesh_tex, lower_bound_tex, upper_bound_tex, child_map_tex, child_len_tex]) -- cgit